Simple collective cursors for models

2 files changed, 51 insertions, 0 deletions

diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/ModelCursor.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/ModelCursor.java
new file mode 100644
index 00000000..3157c9f0
--- /dev/null
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/ModelCursor.java
@@ -0,0 +1,25 @@
+package org.eclipse.viatra.solver.data.model;
+
+import java.util.Map;
+
+import org.eclipse.viatra.solver.data.map.Cursor;
+import org.eclipse.viatra.solver.data.model.representation.DataRepresentation;
+
+public class ModelCursor {
+        final Map<DataRepresentation<?, ?>,Cursor<?,?>> cursors;
+
+        public ModelCursor(Map<DataRepresentation<?, ?>, Cursor<?, ?>> cursors) {
+                super();
+                this.cursors = cursors;
+        }
+        
+        @SuppressWarnings("unchecked")
+        public <K,V> Cursor<K,V> getCursor(DataRepresentation<K, V> representation) {
+                Cursor<?, ?> cursor = cursors.get(representation);
+                if(cursor != null) {
+                        return (Cursor<K, V>) cursor;
+                } else {
+                        throw new IllegalArgumentException("ModelCursor does not contain cursor for representation "+representation);
+                }
+        }
+}
diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/ModelDiffCursor.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/ModelDiffCursor.java
new file mode 100644
index 00000000..d3551e47
--- /dev/null
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/ModelDiffCursor.java
@@ -0,0 +1,26 @@
+package org.eclipse.viatra.solver.data.model;
+
+import java.util.Map;
+
+import org.eclipse.viatra.solver.data.map.Cursor;
+import org.eclipse.viatra.solver.data.map.DiffCursor;
+import org.eclipse.viatra.solver.data.model.representation.DataRepresentation;
+
+public class ModelDiffCursor {
+        final Map<DataRepresentation<?, ?>,DiffCursor<?,?>> diffcursors;
+
+        public ModelDiffCursor(Map<DataRepresentation<?, ?>, DiffCursor<?, ?>> diffcursors) {
+                super();
+                this.diffcursors = diffcursors;
+        }
+        
+        @SuppressWarnings("unchecked")
+        public <K,V> DiffCursor<K,V> getCursor(DataRepresentation<K, V> representation) {
+                Cursor<?, ?> cursor = diffcursors.get(representation);
+                if(cursor != null) {
+                        return (DiffCursor<K, V>) cursor;
+                } else {
+                        throw new IllegalArgumentException("ModelCursor does not contain cursor for representation "+representation);
+                }
+        }
+}