Generalise prefixes

1 files changed, 14 insertions, 5 deletions

diff --git a/src/main/scala/rsacomb/RSA.scala b/src/main/scala/rsacomb/RSA.scala
index 1fa0fc1..79617c7 100644
--- a/src/main/scala/rsacomb/RSA.scala
+++ b/src/main/scala/rsacomb/RSA.scala
@@ -2,17 +2,26 @@ package rsacomb
 
 /* Java imports */
 import java.io.File
+import java.util.Map
 
+import tech.oxfordsemantic.jrdfox.Prefixes
+import tech.oxfordsemantic.jrdfox.logic.IRI
 import org.semanticweb.owlapi.apibinding.OWLManager
 import org.semanticweb.owlapi.model.OWLOntology
 
 object RSA extends RSAOntology {
 
-  val PrefixBase = "http://example.com/rsa_example.owl#"
-  val PrefixInternal = "http://127.0.0.1/"
-  val PredicatePE = PrefixInternal + "PE"
-  val PredicateU = PrefixInternal + "U"
-  val PredicateE = PrefixInternal + "E"
+  val Prefixes = new Prefixes()
+  Prefixes.declarePrefix(":", "http://example.com/rsa_example.owl#")
+  Prefixes.declarePrefix("internal:", "http://127.0.0.1/")
+  Prefixes.declarePrefix("rdf:", "http://www.w3.org/1999/02/22-rdf-syntax-ns#")
+  Prefixes.declarePrefix("rdfs:", "http://www.w3.org/2000/01/rdf-schema#")
+  Prefixes.declarePrefix("owl:", "http://www.w3.org/2002/07/owl#")
+
+  def internal(name: String): IRI =
+    IRI.create(
+      Prefixes.getPrefixIRIsByPrefixName.get("internal:").getIRI + name
+    )
 
   // TODO: move this somewhere else... maybe an OntoUtils class or something.
   def loadOntology(onto: File): OWLOntology = {