1 files changed, 35 insertions, 0 deletions

diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/query/relationView/FilteredRelationView.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/query/relationView/FilteredRelationView.java
new file mode 100644
index 00000000..b5660041
--- /dev/null
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/query/relationView/FilteredRelationView.java
@@ -0,0 +1,35 @@
+package org.eclipse.viatra.solver.data.query.relationView;
+
+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.RelationRepresentation;
+
+public abstract class FilteredRelationView<D> extends RelationView<D>{
+
+        protected FilteredRelationView(Model model, RelationRepresentation<D> representation) {
+                super(model, representation);
+        }
+        @Override
+        protected Object[] forwardMap(Tuple key, D value) {
+                return toTuple1Array(key);
+        }
+        @Override
+        public boolean get(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 = this.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;
+        }
+}