feat: generate multiple solutions

Switch to partial interpretation based neighborhood calculation when multiple
models are request to avoid returning isomorphic models.

1 files changed, 63 insertions, 0 deletions

diff --git a/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/interpretation/PartialNeighbourhoodCalculator.java b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/interpretation/PartialNeighbourhoodCalculator.java
new file mode 100644
index 00000000..859cf7c1
--- /dev/null
+++ b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/interpretation/PartialNeighbourhoodCalculator.java
@@ -0,0 +1,63 @@
+/*
+ * SPDX-FileCopyrightText: 2024 The Refinery Authors <https://refinery.tools/>
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ */
+package tools.refinery.store.reasoning.interpretation;
+
+import org.eclipse.collections.api.set.primitive.IntSet;
+import tools.refinery.store.map.Cursor;
+import tools.refinery.store.model.Model;
+import tools.refinery.store.query.ModelQueryAdapter;
+import tools.refinery.store.reasoning.ReasoningAdapter;
+import tools.refinery.store.reasoning.literal.Concreteness;
+import tools.refinery.store.reasoning.representation.PartialSymbol;
+import tools.refinery.store.statecoding.StateCodeCalculatorFactory;
+import tools.refinery.store.statecoding.StateCoderResult;
+import tools.refinery.store.statecoding.neighbourhood.AbstractNeighbourhoodCalculator;
+import tools.refinery.store.tuple.Tuple;
+
+import java.util.List;
+
+public class PartialNeighbourhoodCalculator extends AbstractNeighbourhoodCalculator<PartialInterpretation<?, ?>> {
+        private final ModelQueryAdapter queryAdapter;
+
+        public static final StateCodeCalculatorFactory FACTORY =
+                        (model, ignoredInterpretations, individuals) -> new PartialNeighbourhoodCalculator(model, individuals);
+
+        protected PartialNeighbourhoodCalculator(Model model, IntSet individuals) {
+                super(model, individuals);
+                queryAdapter = model.getAdapter(ModelQueryAdapter.class);
+        }
+
+        @Override
+        public StateCoderResult calculateCodes() {
+                queryAdapter.flushChanges();
+                return super.calculateCodes();
+        }
+
+        @Override
+        protected List<PartialInterpretation<?, ?>> getInterpretations() {
+                var adapter = getModel().getAdapter(ReasoningAdapter.class);
+                var partialSymbols = adapter.getStoreAdapter().getPartialSymbols();
+                return partialSymbols.stream()
+                                .<PartialInterpretation<?, ?>>map(partialSymbol ->
+                                                adapter.getPartialInterpretation(Concreteness.PARTIAL, (PartialSymbol<?, ?>) partialSymbol))
+                                .toList();
+        }
+
+        @Override
+        protected int getArity(PartialInterpretation<?, ?> interpretation) {
+                return interpretation.getPartialSymbol().arity();
+        }
+
+        @Override
+        protected Object getNullValue(PartialInterpretation<?, ?> interpretation) {
+                return interpretation.get(Tuple.of());
+        }
+
+        @Override
+        protected Cursor<Tuple, ?> getCursor(PartialInterpretation<?, ?> interpretation) {
+                return interpretation.getAll();
+        }
+}