initial version

1 files changed, 98 insertions, 0 deletions

diff --git a/src/uk/ac/ox/cs/pagoda/reasoner/light/KarmaQueryEngine.java b/src/uk/ac/ox/cs/pagoda/reasoner/light/KarmaQueryEngine.java
new file mode 100644
index 0000000..f70dde9
--- /dev/null
+++ b/src/uk/ac/ox/cs/pagoda/reasoner/light/KarmaQueryEngine.java
@@ -0,0 +1,98 @@
+package uk.ac.ox.cs.pagoda.reasoner.light; 
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.util.*;
+
+import org.semanticweb.karma2.*; 
+import org.semanticweb.karma2.clausifier.OntologyProcesser;
+import org.semanticweb.karma2.exception.IllegalInputOntologyException;
+import org.semanticweb.karma2.model.ConjunctiveQuery;
+import org.semanticweb.owlapi.model.OWLOntology;
+
+import uk.ac.ox.cs.pagoda.query.*; 
+import uk.ac.ox.cs.pagoda.util.ConjunctiveQueryHelper;
+import uk.ac.ox.cs.pagoda.util.Utility;
+import uk.ac.ox.cs.JRDFox.JRDFStoreException;
+import uk.ac.ox.cs.JRDFox.store.DataStore;
+
+public class KarmaQueryEngine extends RDFoxQueryEngine {
+        
+        private MyKarma reasoner = null;
+        
+        String karmaDataFile = null, karmaRuleFile = null; 
+        
+        public KarmaQueryEngine(String name) {
+                super(name);
+                
+//              int Base = 1 << 6; 
+//              int index = (new Random().nextInt() % Base + Base) % Base;
+//              karmaDataFile = "karma_data" + index + ".ttl"; 
+//              karmaRuleFile = "karma_rule" + index + ".dlog";
+                karmaDataFile = Utility.TempDirectory + "karma_data.ttl"; 
+                karmaRuleFile = Utility.TempDirectory + "karma_rule.dlog";
+                
+                reasoner = new MyKarma(); 
+        }
+        
+        public MyKarma getReasoner() {
+                return reasoner; 
+        }
+        
+        public void processOntology(OWLOntology elhoOntology) {
+                try {
+                        OntologyProcesser.transformOntology(elhoOntology, new File(karmaDataFile), new File(karmaRuleFile));
+                } catch (IllegalInputOntologyException e) {
+                        e.printStackTrace();
+                }
+        }
+
+        @Override
+        public void dispose() {
+                reasoner.dispose(); 
+        }
+        
+        @Override
+        public AnswerTuples evaluate(String queryText) {
+                return evaluate(queryText, ConjunctiveQueryHelper.getAnswerVariables(queryText)[0], null); 
+        }
+        
+        @Override
+        public AnswerTuples evaluate(String queryText, String[] answerVars) {
+                return evaluate(queryText, answerVars, null); 
+        }
+        
+        public AnswerTuples evaluate(String queryText, AnswerTuples soundAnswerTuples) {
+                return evaluate(queryText, ConjunctiveQueryHelper.getAnswerVariables(queryText)[0], soundAnswerTuples); 
+        }
+        
+        public AnswerTuples evaluate(String queryText, String[] answerVars, AnswerTuples soundAnswerTuples) {
+                KarmaQuery karmaQuery = new KarmaQuery(queryText.replace("_:", "?"));
+                reasoner.setConcurrence(false);
+                ConjunctiveQuery cq = karmaQuery.getConjunctiveQuery(); 
+                if (cq == null) return null; 
+                Set<AnswerTuple> answers = reasoner.answerCQ(cq, soundAnswerTuples, !queryText.contains("_:"));
+                return new AnswerTuplesImp(answerVars, answers); 
+        }
+
+        @Override
+        public DataStore getDataStore() {
+                return reasoner.getStore();
+        }
+        
+        public void initialiseKarma() {
+                try {
+                        reasoner.initializeData(new File(karmaDataFile));
+                        reasoner.materialise(new File(karmaRuleFile));
+                        
+                        File tmp; 
+                        if (karmaDataFile != null && ((tmp = new File(karmaDataFile)).exists())) tmp.delete(); 
+                        if (karmaRuleFile != null && ((tmp = new File(karmaRuleFile)).exists())) tmp.delete(); 
+                } catch (FileNotFoundException e) {
+                        e.printStackTrace();
+                } catch (JRDFStoreException e) {
+                        e.printStackTrace();
+                }
+        }
+
+}