1 files changed, 46 insertions, 0 deletions

diff --git a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/update/ModelUpdateListener.java b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/update/ModelUpdateListener.java
new file mode 100644
index 00000000..1ae3daa7
--- /dev/null
+++ b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/update/ModelUpdateListener.java
@@ -0,0 +1,46 @@
+package tools.refinery.store.query.viatra.internal.update;
+
+import org.eclipse.viatra.query.runtime.matchers.context.IInputKey;
+import org.eclipse.viatra.query.runtime.matchers.context.IQueryRuntimeContextListener;
+import org.eclipse.viatra.query.runtime.matchers.tuple.ITuple;
+import tools.refinery.store.model.Model;
+import tools.refinery.store.query.view.AnyRelationView;
+import tools.refinery.store.query.view.RelationView;
+
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.Map;
+
+public class ModelUpdateListener {
+        private final Map<AnyRelationView, RelationViewUpdateListener<?>> relationViewUpdateListeners;
+
+        public ModelUpdateListener(Model model, Collection<AnyRelationView> relationViews) {
+                relationViewUpdateListeners = new HashMap<>(relationViews.size());
+                for (var relationView : relationViews) {
+                        registerView(model, (RelationView<?>) relationView);
+                }
+        }
+
+        private <T> void registerView(Model model, RelationView<T> relationView) {
+                var listener = RelationViewUpdateListener.of(relationView);
+                var interpretation = model.getInterpretation(relationView.getSymbol());
+                interpretation.addListener(listener, true);
+                relationViewUpdateListeners.put(relationView, listener);
+        }
+
+        public boolean containsRelationView(AnyRelationView relationView) {
+                return relationViewUpdateListeners.containsKey(relationView);
+        }
+
+        public void addListener(IInputKey key, AnyRelationView relationView, ITuple seed,
+                                                        IQueryRuntimeContextListener listener) {
+                var relationViewUpdateListener = relationViewUpdateListeners.get(relationView);
+                relationViewUpdateListener.addFilter(key, seed, listener);
+        }
+
+        public void removeListener(IInputKey key, AnyRelationView relationView, ITuple seed,
+                                                           IQueryRuntimeContextListener listener) {
+                var relationViewUpdateListener = relationViewUpdateListeners.get(relationView);
+                relationViewUpdateListener.removeFilter(key, seed, listener);
+        }
+}