Expose ABox in RSAOntology class

1 files changed, 9 insertions, 2 deletions

diff --git a/src/main/scala/rsacomb/RSAOntology.scala b/src/main/scala/rsacomb/RSAOntology.scala
index 8d6fe8c..e273dc5 100644
--- a/src/main/scala/rsacomb/RSAOntology.scala
+++ b/src/main/scala/rsacomb/RSAOntology.scala
@@ -47,7 +47,7 @@ trait RSAOntology {
    */
   implicit class RSAOntology(ontology: OWLOntology) extends RSAAxiom {
 
-    // Gather TBox+RBox from original ontology
+    // Gather TBox/RBox/ABox from original ontology
     lazy val tbox: List[OWLAxiom] =
       ontology
         .tboxAxioms(Imports.INCLUDED)
@@ -62,7 +62,14 @@ trait RSAOntology {
         .asScala
         .toList
 
-    lazy val axioms: List[OWLAxiom] = tbox ++ rbox
+    lazy val abox: List[OWLAxiom] =
+      ontology
+        .aboxAxioms(Imports.INCLUDED)
+        .collect(Collectors.toList())
+        .asScala
+        .toList
+
+    lazy val axioms: List[OWLAxiom] = abox ++ tbox ++ rbox
 
     /* Retrieve individuals in the original ontology
      */