path: root/src/uk/ac/ox/cs/pagoda/util/Utility.java

package uk.ac.ox.cs.pagoda.util;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintStream;
import java.text.SimpleDateFormat;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Collection;
import java.util.Date;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Scanner;
import java.util.Set;
import java.util.Stack;

import org.apache.log4j.Logger;
import org.semanticweb.HermiT.model.Atom;

public class Utility {
	
	private static final Logger LOGS = Logger.getLogger(""); // null; //  
	
	public static final String JAVA_FILE_SEPARATOR = "/";
	public static final String FILE_SEPARATOR = System.getProperty("file.separator");
	public static final String LINE_SEPARATOR = System.getProperty("line.separator");
	
	public static final String TempDirectory = (new File("tmp" + DateTimeFormatter.ISO_LOCAL_DATE_TIME.format(LocalDateTime.now()))).getAbsolutePath() + FILE_SEPARATOR;
	
	public static final int TEST = -1; 
	public static final int FLY = 0;
	public static final int UOBM = 1;
	public static final int LUBM = 2;
	public static final int AEO = 3;
	public static final int WINE = 4;

	public static Set<Atom> toSet(Atom[] data)
	{
		HashSet<Atom> ret = new HashSet<Atom>();
		for (Atom element: data) 
			ret.add(element);
		return ret;
	}
	
	static Stack<PrintStream> outs = new Stack<PrintStream>();
	
	static {
		outs.push(System.out); 
	}
	
	public static boolean redirectSystemOut()
	{
		String stamp = new SimpleDateFormat( "HH:mm:ss").format(new Date());
		return redirectCurrentOut("./console" + stamp + ".txt"); 
	}
	
	public static boolean redirectCurrentOut(String fileName)
	{
		File file = new File(fileName);
		PrintStream out; 
		try {
			out = new PrintStream(new FileOutputStream(file));
		} catch (FileNotFoundException e) {
			e.printStackTrace();
			return false;
		}
		outs.push(out); 
		System.setOut(out);
		return true; 
	}
	
	public static void closeCurrentOut() {
		if (!outs.isEmpty())
			outs.pop().close();
		
		if (!outs.isEmpty()) 
			System.setOut(outs.peek());
	}
	
	public static void sparql2expression(String input, String output) throws IOException {
		BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(input)));
		BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(output)));
		boolean first;
		String line, query;
		while ((line = reader.readLine()) != null) {
			if (line.startsWith("^")) {
				for (int i = 0; i < 4; ++i)
					line = reader.readLine();
				first = true;
				query = "";
				while ((line = reader.readLine()) != null && !line.startsWith("}"))
					if (first) {
						first = false;
						query = expression(line.trim()); 
					}
					else query += ", " + expression(line.trim());
				writer.write(query);
				writer.newLine();
			}
		}
		reader.close();
		writer.close();
	}

	private static String expression(String line) {
		String[] parts = line.split(" ");
		if (parts[1].equals("rdf:type")) {
			return parts[2] + "(?" + variableIndex(parts[0]) + ")";
		}
		else return parts[1] + "(?" + variableIndex(parts[0]) + ",?" + variableIndex(parts[2]) + ")";
	}
	
	private static int asciiX = (int)'X';
	
	private static int variableIndex(String exp) {
		char var = exp.charAt(1);
		return (int)var - asciiX;
	}
	
	public static String readLine(BufferedReader reader) throws IOException {
		String line = reader.readLine();
		if (line == null)
			return null;
		return line.trim();
	}
	
	public static String getTextfromFile(String fileName) throws FileNotFoundException {
		Scanner scanner = new Scanner(new File(fileName));
		String program = scanner.useDelimiter("\\Z").next();
		scanner.close();
		return program; 
	}

	public static String[] getPattern(BufferedReader answerReader) throws IOException {
		String lastLine = readLine(answerReader), line;
		while ((line = readLine(answerReader)) != null && !line.startsWith("---------"))
			lastLine = line;
		return lastLine.split(" ");
	}

	public static void removeRecursively(File file) {
		if (!file.exists()) return;
		
		if (file.isDirectory())
			for (File tFile: file.listFiles())
				removeRecursively(tFile);
		file.delete();
	}
	
	public static void removeRecursively(String fileName) {
		removeRecursively(new File(fileName));
	}

	public static Collection<String> getQueryTexts(String fileName) throws IOException {
		BufferedReader queryReader = new BufferedReader(new InputStreamReader(new FileInputStream(fileName))); 
		String line; 
		Collection<String> queryTexts = new LinkedList<String>(); 
		while (true) {
			while ((line = queryReader.readLine()) != null && ((line = line.trim()).isEmpty() || line.startsWith("#"))); 
			if (line == null) {
				queryReader.close();
				return queryTexts;
			}
			
			StringBuffer query = new StringBuffer();
			if (!line.startsWith("^["))
				query.append(line).append(LINE_SEPARATOR);
			
			while ((line = queryReader.readLine()) != null && !line.trim().endsWith("}")) 
				query.append(line).append(LINE_SEPARATOR);
			query.append(line); 
			queryTexts.add(query.toString());
		}
	}

	/**
	 * 
	 * @param answerReader
	 * @return all lines before the next empty line
	 * @throws IOException
	 */
	public static Collection<String> getLines(BufferedReader answerReader) throws IOException {
		Collection<String> answerTuples = new LinkedList<String>(); 
		String line;
		while ((line = answerReader.readLine()) != null) {
			line = line.trim();
			if (line.isEmpty())
				break; 
			answerTuples.add(line);
		}
		return answerTuples;
	}

	private static StringBuilder logMessage = new StringBuilder();
	
	private static String getLogMessage(Object[] messages) {
		if (messages.length == 1) return messages[0].toString(); 
		else {
			logMessage.setLength(0);
			for (int i = 0; i < messages.length; ++i) { 
				if (logMessage.length() != 0)
					logMessage.append(LINE_SEPARATOR); 
				logMessage.append(messages[i]); 
			}
			return logMessage.toString(); 		
		}

	}
	
	public static void logInfo(Object... messages) {
		if (LOGS != null)
			LOGS.info(getLogMessage(messages)); 
	}
	
	public static void logTrace(Object... messages) {
		if (LOGS != null)
			LOGS.trace(getLogMessage(messages)); 
	}
	
	public static void logDebug(Object... messages) {
		if (LOGS != null)
			LOGS.debug(getLogMessage(messages));
	}
	
	public static void logError(Object... messages) {
		if (LOGS != null)
			LOGS.error(getLogMessage(messages));
	}
	
	public static void initialise() {
		File tmp = new File(TempDirectory); 
		if (!tmp.exists()) tmp.mkdirs(); 
	}

	public static void cleanup() {
		File tmp = new File(TempDirectory);
		if (tmp.exists()) {
			for (File file: tmp.listFiles())
				file.delete(); 
			tmp.delete();
		}
	}

	public static String toFileIRI(String path) {
		String iri; 
		if (path.startsWith(FILE_SEPARATOR)) iri = "file:" + path; 
		else iri = "file:\\\\\\" + path;
		return iri.replace(FILE_SEPARATOR, JAVA_FILE_SEPARATOR).replace(" ", "%20");
	}

}