Merge pull request #39 from kris7t/partial-interpretation

Implement partial interpretation based model generation

1 files changed, 85 insertions, 0 deletions

diff --git a/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/multiobject/EqualsRelationRewriter.java b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/multiobject/EqualsRelationRewriter.java
new file mode 100644
index 00000000..61b9488c
--- /dev/null
+++ b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/multiobject/EqualsRelationRewriter.java
@@ -0,0 +1,85 @@
+/*
+ * SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ */
+package tools.refinery.store.reasoning.translator.multiobject;
+
+import tools.refinery.store.query.dnf.Query;
+import tools.refinery.store.query.dnf.RelationalQuery;
+import tools.refinery.store.query.literal.AbstractCallLiteral;
+import tools.refinery.store.query.literal.CallLiteral;
+import tools.refinery.store.query.literal.Literal;
+import tools.refinery.store.query.term.Variable;
+import tools.refinery.store.query.view.AnySymbolView;
+import tools.refinery.store.reasoning.interpretation.QueryBasedRelationRewriter;
+import tools.refinery.store.reasoning.literal.Concreteness;
+import tools.refinery.store.reasoning.literal.Modality;
+import tools.refinery.store.representation.cardinality.UpperCardinalities;
+import tools.refinery.store.representation.cardinality.UpperCardinality;
+
+import java.util.List;
+import java.util.Set;
+
+import static tools.refinery.store.query.literal.Literals.check;
+import static tools.refinery.store.query.term.uppercardinality.UpperCardinalityTerms.constant;
+import static tools.refinery.store.query.term.uppercardinality.UpperCardinalityTerms.lessEq;
+
+class EqualsRelationRewriter extends QueryBasedRelationRewriter {
+        private EqualsRelationRewriter(RelationalQuery may, RelationalQuery must) {
+                super(may, must, may, may);
+        }
+
+        @Override
+        public List<Literal> rewriteLiteral(Set<Variable> positiveVariables, AbstractCallLiteral literal,
+                                                                                Modality modality, Concreteness concreteness) {
+                if (!(literal instanceof CallLiteral callLiteral)) {
+                        return super.rewriteLiteral(positiveVariables, literal, modality, concreteness);
+                }
+                var left = callLiteral.getArguments().get(0).asNodeVariable();
+                var right = callLiteral.getArguments().get(1).asNodeVariable();
+                boolean useMay = modality == Modality.MAY || concreteness == Concreteness.CANDIDATE;
+                return switch (callLiteral.getPolarity()) {
+                        case POSITIVE, TRANSITIVE -> {
+                                if (useMay) {
+                                        if (positiveVariables.contains(left) || positiveVariables.contains(right)) {
+                                                // No need to enumerate arguments if at least one is already bound, since they will be unified.
+                                                yield List.of(left.isEquivalent(right));
+                                        } else {
+                                                yield List.of(
+                                                                left.isEquivalent(right),
+                                                                getMay().call(left, right)
+                                                );
+                                        }
+                                } else {
+                                        yield List.of(
+                                                        left.isEquivalent(right),
+                                                        getMust().call(left, right)
+                                        );
+                                }
+                        }
+                        case NEGATIVE -> {
+                                if (useMay) {
+                                        yield List.of(left.notEquivalent(right));
+                                } else {
+                                        yield super.rewriteLiteral(positiveVariables, literal, modality, concreteness);
+                                }
+                        }
+                };
+        }
+
+        public static EqualsRelationRewriter of(AnySymbolView upperCardinalityView) {
+                var may = Query.of("equals#may", (builder, p1, p2) -> builder
+                                .clause(
+                                                p1.isEquivalent(p2),
+                                                upperCardinalityView.call(p1, Variable.of(UpperCardinality.class))
+                                ));
+                var must = Query.of("equals#must", (builder, p1, p2) -> builder
+                                .clause(UpperCardinality.class, upper -> List.of(
+                                                p1.isEquivalent(p2),
+                                                upperCardinalityView.call(p1, upper),
+                                                check(lessEq(upper, constant(UpperCardinalities.ONE)))
+                                )));
+                return new EqualsRelationRewriter(may, must);
+        }
+}