aboutsummaryrefslogtreecommitdiff
path: root/src/uk/ac/ox/cs/pagoda/reasoner/light/KarmaQueryEngine.java
blob: f70dde90ea1e80b62a2bdb7f0767a02bdab10a3a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
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();
		}
	}

}