4 files changed, 199 insertions, 0 deletions

diff --git a/store/src/main/java/org/eclipse/viatra/solver/data/query/view/FilteredRelationView.java b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/FilteredRelationView.java
new file mode 100644
index 00000000..edc534b7
--- /dev/null
+++ b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/FilteredRelationView.java
@@ -0,0 +1,48 @@
+package org.eclipse.viatra.solver.data.query.view;
+
+import java.util.function.BiPredicate;
+
+import org.eclipse.viatra.solver.data.model.Model;
+import org.eclipse.viatra.solver.data.model.Tuple;
+import org.eclipse.viatra.solver.data.model.Tuple.Tuple1;
+import org.eclipse.viatra.solver.data.model.representation.Relation;
+
+public class FilteredRelationView<D> extends RelationView<D>{
+        private final BiPredicate<Tuple,D> predicate;
+
+        public FilteredRelationView(Relation<D> representation, BiPredicate<Tuple,D> predicate) {
+                super(representation);
+                this.predicate = predicate;
+        }
+        @Override
+        protected Object[] forwardMap(Tuple key, D value) {
+                return toTuple1Array(key);
+        }
+        @Override
+        public boolean get(Model model, Object[] tuple) {
+                int[] content = new int[tuple.length];
+                for(int i = 0; i<tuple.length; i++) {
+                        content[i] =((Tuple1)tuple[i]).get(0);
+                }
+                Tuple key = Tuple.of(content);
+                D value = model.get(representation, key);
+                return filter(key, value);
+        }
+        
+        public static Object[] toTuple1Array(Tuple t) {
+                Object[] result = new Object[t.getSize()];
+                for(int i = 0; i<t.getSize(); i++) {
+                        result[i] = t.get(i);
+                }
+                return result;
+        }
+        
+        @Override
+        public int getArity() {
+                return this.representation.getArity();
+        }
+        @Override
+        protected boolean filter(Tuple key, D value) {
+                return this.predicate.test(key, value);
+        }
+}
diff --git a/store/src/main/java/org/eclipse/viatra/solver/data/query/view/FunctionalRelationView.java b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/FunctionalRelationView.java
new file mode 100644
index 00000000..4aa7cfd0
--- /dev/null
+++ b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/FunctionalRelationView.java
@@ -0,0 +1,50 @@
+package org.eclipse.viatra.solver.data.query.view;
+
+import org.eclipse.viatra.solver.data.model.Model;
+import org.eclipse.viatra.solver.data.model.Tuple;
+import org.eclipse.viatra.solver.data.model.Tuple.Tuple1;
+import org.eclipse.viatra.solver.data.model.representation.Relation;
+
+public class FunctionalRelationView<D> extends RelationView<D> {
+
+        public FunctionalRelationView(Relation<D> representation) {
+                super(representation);
+        }
+
+        @Override
+        protected boolean filter(Tuple key, D value) {
+                return true;
+        }
+
+        @Override
+        protected Object[] forwardMap(Tuple key, D value) {
+                return toTuple1ArrayPlusValue(key, value);
+        }
+
+        @Override
+        public boolean get(Model model, Object[] tuple) {
+                int[] content = new int[tuple.length-1];
+                for(int i = 0; i<tuple.length-1; i++) {
+                        content[i] =((Tuple1)tuple[i]).get(0);
+                }
+                Tuple key = Tuple.of(content);
+                @SuppressWarnings("unchecked")
+                D valueInTuple = (D) tuple[tuple.length-1];
+                D valueInMap = model.get(representation, key);
+                return valueInTuple.equals(valueInMap);
+        }
+        
+        public static <D> Object[] toTuple1ArrayPlusValue(Tuple t, D value) {
+                Object[] result = new Object[t.getSize()+1];
+                for(int i = 0; i<t.getSize(); i++) {
+                        result[i] = t.get(i);
+                }
+                result[t.getSize()] = value;
+                return result;
+        }
+        
+        @Override
+        public int getArity() {
+                return this.representation.getArity()+1;
+        }
+}
diff --git a/store/src/main/java/org/eclipse/viatra/solver/data/query/view/KeyOnlyRelationView.java b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/KeyOnlyRelationView.java
new file mode 100644
index 00000000..11a24fc8
--- /dev/null
+++ b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/KeyOnlyRelationView.java
@@ -0,0 +1,16 @@
+package org.eclipse.viatra.solver.data.query.view;
+
+import org.eclipse.viatra.solver.data.model.Tuple;
+import org.eclipse.viatra.solver.data.model.representation.Relation;
+
+public class KeyOnlyRelationView extends FilteredRelationView<Boolean>{
+
+        public KeyOnlyRelationView(Relation<Boolean> representation) {
+                super(representation, (k,v)->true);
+        }
+        @Override
+        protected boolean filter(Tuple key, Boolean value) {
+                return true;
+        }
+        
+}
diff --git a/store/src/main/java/org/eclipse/viatra/solver/data/query/view/RelationView.java b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/RelationView.java
new file mode 100644
index 00000000..c5bc5228
--- /dev/null
+++ b/store/src/main/java/org/eclipse/viatra/solver/data/query/view/RelationView.java
@@ -0,0 +1,85 @@
+package org.eclipse.viatra.solver.data.query.view;
+
+import java.util.Objects;
+
+import org.eclipse.viatra.query.runtime.matchers.context.common.BaseInputKeyWrapper;
+import org.eclipse.viatra.solver.data.map.CursorAsIterator;
+import org.eclipse.viatra.solver.data.model.Model;
+import org.eclipse.viatra.solver.data.model.Tuple;
+import org.eclipse.viatra.solver.data.model.representation.Relation;
+
+/**
+ * Represents a view of a {@link Relation} that can be queried.
+ * 
+ * @author Oszkar Semerath
+ *
+ * @param <D>
+ */
+public abstract class RelationView<D> extends BaseInputKeyWrapper<RelationView<D>> {
+        protected final Relation<D> representation;
+
+        protected RelationView(Relation<D> representation) {
+                super(null);
+                this.wrappedKey = this;
+                this.representation = representation;
+        }
+
+        @Override
+        public String getPrettyPrintableName() {
+                return representation.getName();
+        }
+
+        @Override
+        public String getStringID() {
+                return representation.getName() + this.getClass().getName();
+        }
+
+        public Relation<D> getRepresentation() {
+                return representation;
+        }
+
+        @Override
+        public boolean isEnumerable() {
+                return true;
+        }
+
+        protected abstract boolean filter(Tuple key, D value);
+
+        protected abstract Object[] forwardMap(Tuple key, D value);
+
+        public abstract boolean get(Model model, Object[] tuple);
+
+        public Object[] transform(Tuple tuple, D value) {
+                if (filter(tuple, value)) {
+                        return forwardMap(tuple, value);
+                } else
+                        return null;
+        }
+
+        public Iterable<Object[]> getAll(Model model) {
+                return (() -> new CursorAsIterator<>(model.getAll(representation), (k, v) -> forwardMap(k, v),
+                                (k, v) -> filter(k, v)));
+        }
+
+        @Override
+        public int hashCode() {
+                final int prime = 31;
+                int result = 1;
+                result = prime * result + Objects.hash(representation);
+                return result;
+        }
+
+        @Override
+        public boolean equals(Object obj) {
+                if (this == obj)
+                        return true;
+                if (!super.equals(obj))
+                        return false;
+                if (!(obj instanceof RelationView))
+                        return false;
+                @SuppressWarnings("unchecked")
+                RelationView<D> other = ((RelationView<D>) obj);
+                return Objects.equals(representation, other.representation);
+        }
+
+}