4 files changed, 66 insertions, 12 deletions

diff --git a/src/main/scala/uk/ac/ox/cs/rsacomb/RSAOntology.scala b/src/main/scala/uk/ac/ox/cs/rsacomb/RSAOntology.scala
index 42a5b87..b0b52c7 100644
--- a/src/main/scala/uk/ac/ox/cs/rsacomb/RSAOntology.scala
+++ b/src/main/scala/uk/ac/ox/cs/rsacomb/RSAOntology.scala
@@ -58,6 +58,7 @@ import tech.oxfordsemantic.jrdfox.logic._
 import org.semanticweb.owlapi.model.OWLObjectInverseOf
 
 import uk.ac.ox.cs.rsacomb.converter._
+import uk.ac.ox.cs.rsacomb.filtering.{FilteringProgram, FilterType}
 import uk.ac.ox.cs.rsacomb.suffix._
 import uk.ac.ox.cs.rsacomb.sparql._
 import uk.ac.ox.cs.rsacomb.util.{RDFoxUtil, RSA}
@@ -409,7 +410,7 @@ class RSAOntology(_ontology: File, val datafiles: File*) {
 
   def filteringProgram(query: ConjunctiveQuery): FilteringProgram =
     Logger.timed(
-      new FilteringProgram(query),
+      FilteringProgram(FilterType.FILTER_NAIVE)(query),
       "Generating filtering program",
       Logger.DEBUG
     )
diff --git a/src/main/scala/uk/ac/ox/cs/rsacomb/filtering/FilteringProgram.scala b/src/main/scala/uk/ac/ox/cs/rsacomb/filtering/FilteringProgram.scala
new file mode 100644
index 0000000..9c8cbaa
--- /dev/null
+++ b/src/main/scala/uk/ac/ox/cs/rsacomb/filtering/FilteringProgram.scala
@@ -0,0 +1,41 @@
+package uk.ac.ox.cs.rsacomb.filtering
+
+import tech.oxfordsemantic.jrdfox.logic.datalog.Rule
+import uk.ac.ox.cs.rsacomb.sparql.ConjunctiveQuery
+import uk.ac.ox.cs.rsacomb.util.Versioned
+
+sealed trait FilterType
+object FilterType {
+  case object FILTER_NAIVE extends FilterType
+  case object FILTER_REVISED_V1 extends FilterType
+}
+
+object FilteringProgram extends Versioned[FilterType] {
+
+  import FilterType._
+
+  type Result = (ConjunctiveQuery) => FilteringProgram
+
+  def apply(t: FilterType): (ConjunctiveQuery) => FilteringProgram =
+    t match {
+      case FILTER_NAIVE      => NaiveFilteringProgram(_)
+      case FILTER_REVISED_V1 => NaiveFilteringProgram(_)
+    }
+}
+
+/** Filtering Program generator
+  *
+  * Handles the conversion of a CQ into a set of logic rules,
+  * representing the filtering step of the RSA combined approach.
+  */
+trait FilteringProgram {
+
+  /** Query from which the filtering program is generated */
+  val query: ConjunctiveQuery
+
+  /** Collection of filtering program rules. */
+  def rules: List[Rule]
+
+  /** Pretty-print filtering rule */
+  override def toString(): String = rules mkString "\n"
+}
diff --git a/src/main/scala/uk/ac/ox/cs/rsacomb/FilteringProgram.scala b/src/main/scala/uk/ac/ox/cs/rsacomb/filtering/NaiveFilteringProgram.scala
index 06224e7..57898a8 100644
--- a/src/main/scala/uk/ac/ox/cs/rsacomb/FilteringProgram.scala
+++ b/src/main/scala/uk/ac/ox/cs/rsacomb/filtering/NaiveFilteringProgram.scala
@@ -1,4 +1,4 @@
-package uk.ac.ox.cs.rsacomb
+package uk.ac.ox.cs.rsacomb.filtering
 
 //import scala.collection.JavaConverters._
 import tech.oxfordsemantic.jrdfox.logic.Datatype
@@ -14,17 +14,14 @@ import uk.ac.ox.cs.rsacomb.suffix.{Forward, Backward}
 import uk.ac.ox.cs.rsacomb.util.{RSA, RDFoxUtil}
 
 /** Factory for [[uk.ac.ox.cs.rsacomb.FilteringProgram FilteringProgram]] */
-object FilteringProgram {
+object NaiveFilteringProgram {
 
   /** Create a new FilteringProgram instance.
     *
     * @param query CQ to be converted into logic rules.
-    * @param constants constants in the original ontology. They will be
-    * used to initialize predicate `rsa:Named`.
     */
   def apply(query: ConjunctiveQuery): FilteringProgram =
-    new FilteringProgram(query)
-
+    new NaiveFilteringProgram(query)
 }
 
 /** Filtering Program generator
@@ -34,7 +31,8 @@ object FilteringProgram {
   *
   * Instances can be created using the companion object.
   */
-class FilteringProgram(query: ConjunctiveQuery) {
+class NaiveFilteringProgram(val query: ConjunctiveQuery)
+    extends FilteringProgram {
 
   /** Extends capabilities of
     * [[tech.oxfordsemantic.jrdfox.logic.datalog.TupleTableAtom TupleTableAtom]]
@@ -306,8 +304,4 @@ class FilteringProgram(query: ConjunctiveQuery) {
         r8a ::: r8b :: r8c :::
         r9 :: List()) map RDFoxUtil.reify
     }
-
-  /** Pretty-print filtering rule */
-  override def toString(): String = rules mkString "\n"
-
 }
diff --git a/src/main/scala/uk/ac/ox/cs/rsacomb/util/Versioned.scala b/src/main/scala/uk/ac/ox/cs/rsacomb/util/Versioned.scala
new file mode 100644
index 0000000..aa2a963
--- /dev/null
+++ b/src/main/scala/uk/ac/ox/cs/rsacomb/util/Versioned.scala
@@ -0,0 +1,18 @@
+package uk.ac.ox.cs.rsacomb.util
+
+/** Utility to allow hussle free version switching of blocks of code
+  *
+  * This allows for example testing different implementations of a
+  * module or algorithm.
+  */
+trait Versioned[T] {
+
+  /** Type of the returned versioned object */
+  type Result
+
+  /** Returns correct instance of the versioned object
+    *
+    * @param t object uniquely identifing the requested instance.
+    */
+  def apply(t: T): Result
+}