Merge pull request #39 from kris7t/partial-interpretation

Implement partial interpretation based model generation

1 files changed, 15 insertions, 0 deletions

diff --git a/subprojects/store-query/src/main/java/tools/refinery/store/query/view/AbstractFunctionView.java b/subprojects/store-query/src/main/java/tools/refinery/store/query/view/AbstractFunctionView.java
index fd37604e..f130fa59 100644
--- a/subprojects/store-query/src/main/java/tools/refinery/store/query/view/AbstractFunctionView.java
+++ b/subprojects/store-query/src/main/java/tools/refinery/store/query/view/AbstractFunctionView.java
@@ -5,6 +5,7 @@
  */
 package tools.refinery.store.query.view;
 
+import tools.refinery.store.map.CursorAsIterator;
 import tools.refinery.store.model.Model;
 import tools.refinery.store.query.dnf.FunctionalDependency;
 import tools.refinery.store.query.term.Parameter;
@@ -83,6 +84,20 @@ public abstract class AbstractFunctionView<T> extends SymbolView<T> {
         }
 
         @Override
+        public boolean canIndexSlot(int slot) {
+                return slot >= 0 && slot < getSymbol().arity();
+        }
+
+        @Override
+        public Iterable<Object[]> getAdjacent(Model model, int slot, Object value) {
+                if (!(value instanceof Tuple1 tuple1)) {
+                        return Set.of();
+                }
+                return (() -> new CursorAsIterator<>(model.getInterpretation(getSymbol()).getAdjacent(slot, tuple1.get(0)),
+                                this::forwardMap, this::filter));
+        }
+
+        @Override
         public List<Parameter> getParameters() {
                 return parameters;
         }