private IRI executeRule(final Rule r, final IRI inputIRI) {
try {
PelletOptions.USE_ANNOTATION_SUPPORT = true;
PelletOptions.TREAT_ALL_VARS_DISTINGUISHED = controller.isTreatAllVariablesDistinguished();
QueryEngineType type = (QueryEngineType) controller.getQueryEngineType();
final QueryExecution qe;
final ByteArrayOutputStream w = new ByteArrayOutputStream();
final Query qSelect = getSelectExampleQuery(r.getQuery());
if (type.toPellet() != null) {
final OWLOntology queryOntology = getInputOntologyForRule(inputIRI);
final PelletReasoner reasoner =
PelletReasonerFactory.getInstance().createReasoner(queryOntology);
log.info("Ontology size: " + reasoner.getKB().getInfo());
final Dataset ds = kb2ds(reasoner.getKB());
final QueryExecution qeSelect =
SparqlDLExecutionFactory.create(qSelect, ds, null, type.toPellet());
final ResultSet rs = qeSelect.execSelect();
controller.setSelect(r, rs.getResultVars(), ResultSetFormatter.toList(rs));
qe =
SparqlDLExecutionFactory.create(
r.getQuery(), kb2ds(reasoner.getKB()), null, type.toPellet());
qe.execConstruct().write(w);
} else {
final ByteArrayOutputStream w2 = new ByteArrayOutputStream();
final Model model = ModelFactory.createDefaultModel();
try {
controller
.getOWLOntologyManager()
.saveOntology(queryOntology, new TurtleOntologyFormat(), w2);
model.read(new ByteArrayInputStream(w2.toByteArray()), "", "TURTLE");
final QueryExecution qeSelect = QueryExecutionFactory.create(qSelect, model);
final ResultSet rs = qeSelect.execSelect();
controller.setSelect(r, rs.getResultVars(), ResultSetFormatter.toList(rs));
qe = QueryExecutionFactory.create(r.getQuery(), model);
qe.execConstruct().write(w);
} catch (OWLOntologyStorageException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
final IRI outputIRI = getOntologyIRIForRuleName(r.getName());
// loaded generated ontology
final OWLOntology generatedOntology =
controller
.getOWLOntologyManager()
.loadOntologyFromOntologyDocument(new ByteArrayInputStream(w.toByteArray()));
controller.updateOntology(
generatedOntology,
outputIRI,
inputIRI,
controller.getRuleSpec().getResultFile(r).toURI());
controller.setStatus("Rule " + r.getName() + " successfully executed");
return outputIRI;
} catch (OWLOntologyCreationException e1) {
controller.setStatus(e1.getMessage());
return null;
}
}
