1 files changed, 133 insertions, 0 deletions

diff --git a/src/uk/ac/ox/cs/pagoda/reasoner/HermiTReasoner.java b/src/uk/ac/ox/cs/pagoda/reasoner/HermiTReasoner.java
new file mode 100644
index 0000000..5511691
--- /dev/null
+++ b/src/uk/ac/ox/cs/pagoda/reasoner/HermiTReasoner.java
@@ -0,0 +1,133 @@
+package uk.ac.ox.cs.pagoda.reasoner;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.HashSet;
+import java.util.Set;
+
+import org.semanticweb.HermiT.Reasoner;
+import org.semanticweb.owlapi.model.OWLClassExpression;
+import org.semanticweb.owlapi.model.OWLDataFactory;
+import org.semanticweb.owlapi.model.OWLNamedIndividual;
+import org.semanticweb.owlapi.model.OWLOntology;
+import org.semanticweb.owlapi.model.OWLOntologyCreationException;
+import org.semanticweb.owlapi.model.OWLOntologyStorageException;
+
+import uk.ac.ox.cs.JRDFox.model.Individual;
+import uk.ac.ox.cs.pagoda.multistage.MultiStageQueryEngine;
+import uk.ac.ox.cs.pagoda.owl.OWLHelper;
+import uk.ac.ox.cs.pagoda.owl.QueryRoller;
+import uk.ac.ox.cs.pagoda.query.AnswerTuple;
+import uk.ac.ox.cs.pagoda.query.AnswerTuples;
+import uk.ac.ox.cs.pagoda.query.AnswerTuplesImp;
+import uk.ac.ox.cs.pagoda.query.GapByStore4ID;
+import uk.ac.ox.cs.pagoda.query.QueryRecord;
+import uk.ac.ox.cs.pagoda.reasoner.light.BasicQueryEngine;
+import uk.ac.ox.cs.pagoda.rules.DatalogProgram;
+import uk.ac.ox.cs.pagoda.util.Utility;
+
+public class HermiTReasoner extends QueryReasoner {
+        
+        Reasoner hermit;
+        
+        BasicQueryEngine upperStore = null;
+        
+        OWLOntology onto;
+        OWLDataFactory factory; 
+        
+        String importedOntologyPath = null;
+
+        QueryRoller roller;
+        boolean toCheckSatisfiability;  
+        
+        public HermiTReasoner(boolean toCheckSatisfiability) {
+                this.toCheckSatisfiability = toCheckSatisfiability;  
+        }
+
+        @Override
+        public void loadOntology(OWLOntology ontology) {
+                onto = ontology; 
+        }
+
+        @Override
+        public boolean preprocess() {
+                OWLOntology tbox = onto;
+                try {
+                        onto = OWLHelper.getImportedOntology(tbox, importedData.toString().split(ImportDataFileSeparator));
+                        importedOntologyPath = OWLHelper.getOntologyPath(onto); 
+                } catch (OWLOntologyCreationException e) {
+                        // TODO Auto-generated catch block
+                        e.printStackTrace();
+                } catch (OWLOntologyStorageException e) {
+                        // TODO Auto-generated catch block
+                        e.printStackTrace();
+                } catch (IOException e) {
+                        // TODO Auto-generated catch block
+                        e.printStackTrace();
+                }
+                
+                DatalogProgram datalogProgram = new DatalogProgram(tbox, false);
+                importData(datalogProgram.getAdditionalDataFile()); 
+                upperStore = new MultiStageQueryEngine("rl-upper", false);
+                upperStore.importRDFData("data", importedData.toString());
+                GapByStore4ID gap = new GapByStore4ID(upperStore); 
+                upperStore.materialiseFoldedly(datalogProgram, gap);
+                gap.clear();
+
+                factory = onto.getOWLOntologyManager().getOWLDataFactory();
+                roller = new QueryRoller(factory);
+                
+                hermit = new Reasoner(onto);
+                return isConsistent();
+        }
+
+        @Override
+        public boolean isConsistent() {
+                if (toCheckSatisfiability)
+                        return hermit.isConsistent(); 
+                return true; 
+        }
+
+        @Override
+        public void evaluate(QueryRecord record) {
+                String[] disVars = record.getDistinguishedVariables();
+                Set<OWLNamedIndividual> individuals = onto.getIndividualsInSignature(true); 
+                if (disVars.length == 1) {
+                        OWLClassExpression clsExp = roller.rollUp(record.getClause(), record.getAnswerVariables()[0]);
+                        Set<AnswerTuple> answers = new HashSet<AnswerTuple>(); 
+                        for (OWLNamedIndividual individual: individuals) {
+                                Utility.logDebug("checking ... " + individual);
+                                if (hermit.isEntailed(factory.getOWLClassAssertionAxiom(clsExp, individual))) {
+                                        answers.add(new AnswerTuple(new Individual[] {Individual.create(individual.toStringID())})); 
+                                }
+                        }
+                        record.updateLowerBoundAnswers(new AnswerTuplesImp(record.getAnswerVariables(), answers));              
+                        record.markAsProcessed();
+                }
+                else {
+                        // FIXME join here
+                        record.markAsProcessed(); 
+                }
+        }
+
+        @Override
+        public void evaluateUpper(QueryRecord record) {
+                AnswerTuples rlAnswer = null; 
+                try {
+                        rlAnswer = upperStore.evaluate(record.getQueryText(), record.getAnswerVariables());
+                        record.updateUpperBoundAnswers(rlAnswer, true); 
+                } finally {
+                        if (rlAnswer != null) rlAnswer.dispose();
+                }
+        }
+
+        @Override
+        public void dispose() {
+                if (importedOntologyPath != null) {
+                        File tmp = new File(importedOntologyPath);
+                        if (tmp.exists()) tmp.delete(); 
+                }
+                super.dispose();
+        }
+
+}