1 files changed, 71 insertions, 0 deletions

diff --git a/subprojects/store-reasoning-scope/src/main/java/tools/refinery/store/reasoning/scope/TypeScopePropagator.java b/subprojects/store-reasoning-scope/src/main/java/tools/refinery/store/reasoning/scope/TypeScopePropagator.java
new file mode 100644
index 00000000..bb50656b
--- /dev/null
+++ b/subprojects/store-reasoning-scope/src/main/java/tools/refinery/store/reasoning/scope/TypeScopePropagator.java
@@ -0,0 +1,71 @@
+/*
+ * SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ */
+package tools.refinery.store.reasoning.scope;
+
+import com.google.ortools.linearsolver.MPConstraint;
+import tools.refinery.store.model.ModelStoreBuilder;
+import tools.refinery.store.query.ModelQueryBuilder;
+import tools.refinery.store.query.dnf.AnyQuery;
+import tools.refinery.store.query.dnf.RelationalQuery;
+import tools.refinery.store.query.resultset.ResultSet;
+import tools.refinery.store.tuple.Tuple;
+
+import java.util.Collection;
+
+abstract class TypeScopePropagator {
+        private final BoundScopePropagator adapter;
+        private final ResultSet<Boolean> allNodes;
+        private final ResultSet<Boolean> multiNodes;
+        protected final MPConstraint constraint;
+
+        protected TypeScopePropagator(BoundScopePropagator adapter, RelationalQuery allQuery,
+                                                                  RelationalQuery multiQuery) {
+                this.adapter = adapter;
+                var queryEngine = adapter.getQueryEngine();
+                allNodes = queryEngine.getResultSet(allQuery);
+                multiNodes = queryEngine.getResultSet(multiQuery);
+                constraint = adapter.makeConstraint();
+                constraint.setBounds(0, Double.POSITIVE_INFINITY);
+                var cursor = multiNodes.getAll();
+                while (cursor.move()) {
+                        var variable = adapter.getVariable(cursor.getKey().get(0));
+                        constraint.setCoefficient(variable, 1);
+                }
+                allNodes.addListener(this::allChanged);
+                multiNodes.addListener(this::multiChanged);
+        }
+
+        protected abstract void doUpdateBounds();
+
+        public boolean updateBounds() {
+                doUpdateBounds();
+                return constraint.lb() <= constraint.ub();
+        }
+
+        protected int getSingleCount() {
+                return allNodes.size() - multiNodes.size();
+        }
+
+        private void allChanged(Tuple ignoredKey, Boolean ignoredOldValue, Boolean ignoredNewValue) {
+                adapter.markAsChanged();
+        }
+
+        private void multiChanged(Tuple key, Boolean ignoredOldValue, Boolean newValue) {
+                var variable = adapter.getVariable(key.get(0));
+                constraint.setCoefficient(variable, Boolean.TRUE.equals(newValue) ? 1 : 0);
+                adapter.markAsChanged();
+        }
+
+        public abstract static class Factory {
+                public abstract TypeScopePropagator createPropagator(BoundScopePropagator adapter);
+
+                protected abstract Collection<AnyQuery> getQueries();
+
+                public void configure(ModelStoreBuilder storeBuilder) {
+                        storeBuilder.getAdapter(ModelQueryBuilder.class).queries(getQueries());
+                }
+        }
+}