1 files changed, 33 insertions, 1 deletions

diff --git a/src/main/scala/rsacomb/RSAOntology.scala b/src/main/scala/rsacomb/RSAOntology.scala
index 0acf2a6..e4348d8 100644
--- a/src/main/scala/rsacomb/RSAOntology.scala
+++ b/src/main/scala/rsacomb/RSAOntology.scala
@@ -46,12 +46,13 @@ trait RSAOntology {
         .toList
     }
 
-    private lazy val roles: Set[OWLObjectPropertyExpression] =
+    private val roles: Set[OWLObjectPropertyExpression] = {
       ontology
         .rboxAxioms(Imports.INCLUDED)
         .collect(Collectors.toSet())
         .asScala
         .flatMap(_.objectPropertyExpressionsInSignature)
+    }
 
     // OWLAPI reasoner for same easier tasks
     private val reasoner =
@@ -199,6 +200,37 @@ trait RSAOntology {
       (unsafe1 ++ unsafe2).toList
     }
 
+    lazy val canonicalModel: List[Rule] = {
+      // Compute program to generate canonical model
+      val tbox =
+        ontology
+          .tboxAxioms(Imports.INCLUDED)
+          .collect(Collectors.toList())
+          .asScala
+          .toList
+      val rbox =
+        ontology
+          .rboxAxioms(Imports.INCLUDED)
+          .collect(Collectors.toList())
+          .asScala
+          .toList
+      val axioms = tbox ++ rbox
+      val varX = Variable.create("X")
+      val visitor = ProgramGenerator(ontology, varX, unsafeRoles)
+      val facts = ProgramGenerator.NIs(individuals)
+      val rules1 = ProgramGenerator.generateRoleRules(
+        axioms
+          .flatMap(
+            _.objectPropertiesInSignature.collect(Collectors.toSet()).asScala
+          )
+          .toSet
+      )
+      val rules2 = axioms.flatMap(_.accept(visitor))
+
+      rules1 ++ rules2
+      // Call RDFox to generate the canonical model
+    }
+
     private def rsaGraph(
         data: DataStoreConnection
     ): Graph[Resource, UnDiEdge] = {