1 files changed, 139 insertions, 0 deletions

diff --git a/test/uk/ac/ox/cs/pagoda/summary/SummaryTester.java b/test/uk/ac/ox/cs/pagoda/summary/SummaryTester.java
new file mode 100644
index 0000000..18b6090
--- /dev/null
+++ b/test/uk/ac/ox/cs/pagoda/summary/SummaryTester.java
@@ -0,0 +1,139 @@
+package uk.ac.ox.cs.pagoda.summary;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.Scanner;
+
+import org.semanticweb.HermiT.Reasoner;
+import org.semanticweb.owlapi.model.AxiomType;
+import org.semanticweb.owlapi.model.IRI;
+import org.semanticweb.owlapi.model.OWLClassExpression;
+import org.semanticweb.owlapi.model.OWLDataFactory;
+import org.semanticweb.owlapi.model.OWLOntology;
+import org.semanticweb.owlapi.model.OWLOntologyCreationException;
+import org.semanticweb.owlapi.model.OWLOntologyStorageException;
+
+import uk.ac.ox.cs.pagoda.hermit.DLClauseHelper;
+import uk.ac.ox.cs.pagoda.owl.OWLHelper;
+import uk.ac.ox.cs.pagoda.owl.QueryRoller;
+import uk.ac.ox.cs.pagoda.query.QueryManager;
+import uk.ac.ox.cs.pagoda.query.QueryRecord;
+import uk.ac.ox.cs.pagoda.summary.Summary;
+
+public class SummaryTester {
+
+        static String FILE_BREAK = System.getProperty("file.separator");
+        static String LINE_BREAK = System.getProperty("line.separator");
+        
+        public static void main(String[] args) throws Exception {
+//              String arg = "ontologies/claros/all-in-one-manually.owl"; 
+//              String arg = "ontologies/claros/Claros.owl ontologies/claros/data"; 
+                String arg =  "../uobmGenerator/univ-bench-dl.owl " + 
+                                "../uobmGenerator/uobm1 " + //"a " + 
+                                "ontologies/uobm/queries/uobm_standard_less.sparql"; 
+                                
+                testSummarisedUpperBound(arg.split("\\ ")); 
+        }
+        
+        /**
+         * args[0] ontology file location
+         * args[1] data directory
+         * args[2] sparql query file location
+         * 
+         * @param args 
+         * @throws OWLOntologyCreationException
+         * @throws FileNotFoundException
+         * @throws OWLOntologyStorageException 
+         */
+        public static void testSummarisedUpperBound(String[] args) throws OWLOntologyCreationException, FileNotFoundException, OWLOntologyStorageException {
+                OWLOntology onto = OWLHelper.loadOntology(args[0]); 
+                try {
+                        onto = OWLHelper.getImportedOntology(onto, args[1]); 
+                } catch (IOException e) {
+                        e.printStackTrace();
+                } 
+                 
+                Summary sum = testSummary(onto);
+                System.out.println("Summarisation Done."); 
+                
+                System.out.println(args[2]); 
+                Scanner scanner = new Scanner(new File(args[2])); 
+                OWLOntology summary = sum.getSummary();
+                OWLDataFactory factory = summary.getOWLOntologyManager().getOWLDataFactory(); 
+                QueryRoller r = new QueryRoller(factory);
+                OWLClassExpression summarisedQueryExp;
+                Reasoner reasoner = new Reasoner(summary); 
+                QueryManager queryManager = new QueryManager(); 
+                int upperBoundCounter, queryID = 0;
+                StringBuilder queryText = new StringBuilder();
+                String[] vars; 
+                
+                for (String line; ; ) {
+                        queryText.setLength(0);
+                        while (scanner.hasNextLine() && (line = scanner.nextLine()) != null && !line.startsWith("^[query"));
+                        if (!scanner.hasNextLine()) break; 
+                        
+                        while (scanner.hasNextLine() && (line = scanner.nextLine()) != null && !line.isEmpty())
+                                queryText.append(line).append(LINE_BREAK);
+                        if (!scanner.hasNextLine()) break; 
+                                
+                        System.out.println("------------ starting computing for Query " + ++queryID + "------------"); 
+                        
+                        System.out.println(queryText); 
+                        
+                        QueryRecord record = queryManager.create(queryText.toString(), queryID);
+                        vars = record.getAnswerVariables(); 
+                        if (vars.length > 1) {
+                                System.out.println("The query cannot be processed by HermiT ... More than one answer variable"); 
+                                continue; 
+                        }
+                        
+                        summarisedQueryExp = r.rollUp(DLClauseHelper.getQuery(sum.getSummary(record), null), vars[0]);
+                        
+                        upperBoundCounter = 0; 
+                        for (String representative: sum.getRepresentatives()) 
+                                if (reasoner.isEntailed(factory.getOWLClassAssertionAxiom(summarisedQueryExp, factory.getOWLNamedIndividual(IRI.create(representative))))) {
+                                        upperBoundCounter += sum.getGroup(representative).size(); 
+                                }
+                        
+                        System.out.println("There are " + upperBoundCounter + " individual(s) in the upper bound computed by summary."); 
+                }
+                scanner.close();
+        }
+        
+        public static Summary testSummary(OWLOntology ontology) throws OWLOntologyCreationException, FileNotFoundException {
+                Summary sum = new Summary(ontology);
+                
+                System.out.println("original ontology data: "); 
+                outputStatistics(ontology); 
+                
+                OWLOntology summary = sum.getSummary();
+                
+                System.out.println("summarised ontology data: "); 
+                outputStatistics(summary);
+                
+                try {
+                        FileOutputStream out = new FileOutputStream("summary.owl"); 
+                        summary.getOWLOntologyManager().saveOntology(summary, out);
+                        out.close();
+                } catch (OWLOntologyStorageException e) {
+                        e.printStackTrace();
+                } catch (IOException e) {
+                        // TODO Auto-generated catch block
+                        e.printStackTrace();
+                }
+                
+                return sum; 
+        }
+
+        private static void outputStatistics(OWLOntology onto) {
+                System.out.println("TBox: " + onto.getTBoxAxioms(true).size() +
+                                "\tRBox: " + onto.getRBoxAxioms(true).size() + 
+                                "\tABox: " + onto.getABoxAxioms(true).size());
+                System.out.println("Class Assertions: " + onto.getAxiomCount(AxiomType.CLASS_ASSERTION, true) + 
+                                "\tObject Property Assertions: " + onto.getAxiomCount(AxiomType.OBJECT_PROPERTY_ASSERTION, true)); 
+        }
+
+}