From 9ce65c5a963b03ee97fe9cb6c5aa65a3c04a80a8 Mon Sep 17 00:00:00 2001
From: yzhou <yujiao.zhou@gmail.com>
Date: Tue, 21 Apr 2015 10:34:27 +0100
Subject: initial version

---
 .../cs/pagoda/reasoner/light/RDFoxQueryEngine.java | 110 +++++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100644 src/uk/ac/ox/cs/pagoda/reasoner/light/RDFoxQueryEngine.java

(limited to 'src/uk/ac/ox/cs/pagoda/reasoner/light/RDFoxQueryEngine.java')

diff --git a/src/uk/ac/ox/cs/pagoda/reasoner/light/RDFoxQueryEngine.java b/src/uk/ac/ox/cs/pagoda/reasoner/light/RDFoxQueryEngine.java
new file mode 100644
index 0000000..30771ab
--- /dev/null
+++ b/src/uk/ac/ox/cs/pagoda/reasoner/light/RDFoxQueryEngine.java
@@ -0,0 +1,110 @@
+package uk.ac.ox.cs.pagoda.reasoner.light;
+
+import java.io.File;
+import java.util.Collection;
+
+import uk.ac.ox.cs.pagoda.MyPrefixes;
+import uk.ac.ox.cs.pagoda.query.AnswerTuples;
+import uk.ac.ox.cs.pagoda.reasoner.QueryEngine;
+import uk.ac.ox.cs.pagoda.reasoner.QueryReasoner;
+import uk.ac.ox.cs.pagoda.tracking.AnswerTuplesWriter;
+import uk.ac.ox.cs.pagoda.util.Timer;
+import uk.ac.ox.cs.pagoda.util.Utility;
+import uk.ac.ox.cs.JRDFox.JRDFStoreException;
+import uk.ac.ox.cs.JRDFox.Prefixes;
+import uk.ac.ox.cs.JRDFox.store.DataStore;
+import uk.ac.ox.cs.JRDFox.store.DataStore.StoreType;
+
+public abstract class RDFoxQueryEngine implements QueryEngine {
+	
+	public static final int matNoOfThreads = Runtime.getRuntime().availableProcessors() * 2;
+	
+	protected String name; 
+	protected Prefixes prefixes = MyPrefixes.PAGOdAPrefixes.getRDFoxPrefixes();
+
+	public RDFoxQueryEngine(String name) {
+		this.name = name; 
+	} 
+	
+	public abstract DataStore getDataStore();
+	
+	public abstract void dispose(); 
+	
+	public void importRDFData(String fileName, String importedFile) {
+		if (importedFile == null || importedFile.isEmpty()) return ; 
+		Timer t = new Timer(); 
+		DataStore store = getDataStore(); 
+		try {
+			long oldTripleCount = store.getTriplesCount(), tripleCount;
+			for (String file: importedFile.split(QueryReasoner.ImportDataFileSeparator))
+				store.importTurtleFile(new File(file), prefixes);
+			tripleCount = store.getTriplesCount(); 
+			Utility.logDebug(name + " store after importing " + fileName + ": " + tripleCount + " (" + (tripleCount - oldTripleCount) + " new)");
+			store.clearRulesAndMakeFactsExplicit();
+		} catch (JRDFStoreException e) {
+			e.printStackTrace();
+		}
+		Utility.logDebug(name + " store finished importing " + fileName + " in " + t.duration() + " seconds.");
+	}
+	
+	public void materialise(String programName, String programText) {
+		if (programText == null) return ; 
+		Timer t = new Timer();
+		DataStore store = getDataStore(); 
+		try {
+			long oldTripleCount = store.getTriplesCount(), tripleCount;
+//			store.addRules(new String[] {programText});
+			store.importRules(programText);
+			store.applyReasoning();
+			tripleCount = store.getTriplesCount(); 
+			Utility.logDebug(name + " store after materialising " + programName + ": " + tripleCount + " (" + (tripleCount - oldTripleCount) + " new)");
+			store.clearRulesAndMakeFactsExplicit();
+		} catch (JRDFStoreException e) {
+			e.printStackTrace();
+		}
+		Utility.logDebug(name + " store finished the materialisation of " + programName + " in " + t.duration() + " seconds.");
+	}
+
+	@Override
+	public void evaluate(Collection<String> queryTexts, String answerFile) {
+		if (queryTexts == null)
+			return ;
+		
+		int queryID = 0; 
+		AnswerTuplesWriter answerWriter = new AnswerTuplesWriter(answerFile); 
+		AnswerTuples answerTuples;
+		Timer t = new Timer(); 
+		try {
+			for (String query: queryTexts) {
+				t.reset();
+				answerTuples = null; 
+				try {
+					answerTuples = evaluate(query); 
+					Utility.logDebug("time to answer Query " + ++queryID + ": " + t.duration());
+					answerWriter.write(answerTuples.getAnswerVariables(), answerTuples);
+				} finally {
+					if (answerTuples != null) answerTuples.dispose();
+				}
+			}
+		} finally {
+			answerWriter.close();
+		}
+		
+		Utility.logDebug("done computing query answers by RDFox.");
+		
+	}
+	
+	public static DataStore createDataStore() {
+		DataStore instance = null; 
+		try {
+//			instance = new DataStore("par-head-n");
+			instance = new DataStore(StoreType.NarrowParallelHead);
+			instance.setNumberOfThreads(matNoOfThreads);
+			instance.initialize();
+		} catch (JRDFStoreException e) {
+			e.printStackTrace();
+		}
+		return instance;
+	}
+
+}
-- 
cgit v1.2.3