From ecf62ce9828a1fdf45e81d9bf91655513a0e26d6 Mon Sep 17 00:00:00 2001
From: Kristóf Marussy <kristof@marussy.com>
Date: Fri, 18 Aug 2023 20:35:21 +0200
Subject: feat: predicate semantics

---
 .../translator/predicate/PredicateTranslator.java  | 85 ++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/predicate/PredicateTranslator.java

(limited to 'subprojects/store-reasoning')

diff --git a/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/predicate/PredicateTranslator.java b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/predicate/PredicateTranslator.java
new file mode 100644
index 00000000..ee022f2d
--- /dev/null
+++ b/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/predicate/PredicateTranslator.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.predicate;
+
+import tools.refinery.store.model.ModelStoreBuilder;
+import tools.refinery.store.model.ModelStoreConfiguration;
+import tools.refinery.store.query.dnf.Query;
+import tools.refinery.store.query.dnf.RelationalQuery;
+import tools.refinery.store.query.literal.Literal;
+import tools.refinery.store.query.term.NodeVariable;
+import tools.refinery.store.query.term.Variable;
+import tools.refinery.store.query.view.ForbiddenView;
+import tools.refinery.store.query.view.MayView;
+import tools.refinery.store.query.view.MustView;
+import tools.refinery.store.reasoning.representation.PartialRelation;
+import tools.refinery.store.reasoning.translator.PartialRelationTranslator;
+import tools.refinery.store.representation.Symbol;
+import tools.refinery.store.representation.TruthValue;
+
+import static tools.refinery.store.query.literal.Literals.not;
+import static tools.refinery.store.reasoning.literal.PartialLiterals.may;
+import static tools.refinery.store.reasoning.literal.PartialLiterals.must;
+
+public class PredicateTranslator implements ModelStoreConfiguration {
+	private final PartialRelation relation;
+	private final RelationalQuery query;
+	private final boolean mutable;
+	private final TruthValue defaultValue;
+
+	public PredicateTranslator(PartialRelation relation, RelationalQuery query, boolean mutable,
+							   TruthValue defaultValue) {
+		if (relation.arity() != query.arity()) {
+			throw new IllegalArgumentException("Expected arity %d query for partial relation %s, got %d instead"
+					.formatted(relation.arity(), relation, query.arity()));
+		}
+		if (defaultValue.must()) {
+			throw new IllegalArgumentException("Default value must be UNKNOWN or FALSE");
+		}
+		this.relation = relation;
+		this.query = query;
+		this.mutable = mutable;
+		this.defaultValue = defaultValue;
+	}
+
+	@Override
+	public void apply(ModelStoreBuilder storeBuilder) {
+		var translator = PartialRelationTranslator.of(relation)
+				.query(query);
+		if (mutable) {
+			var symbol = Symbol.of(relation.name(), relation.arity(), TruthValue.class, defaultValue);
+			translator.symbol(symbol);
+
+			var parameters = new NodeVariable[relation.arity()];
+			for (int i = 0; i < parameters.length; i++) {
+				parameters[i] = Variable.of("p" + i);
+			}
+
+			var must = Query.builder()
+					.parameters(parameters)
+					.clause(must(query.call(parameters)))
+					.clause(new MustView(symbol).call(parameters))
+					.build();
+			translator.must(must);
+
+			var mayLiterals = new Literal[2];
+			mayLiterals[0] = may(query.call(parameters));
+			if (defaultValue.may()) {
+				mayLiterals[1] = not(new ForbiddenView(symbol).call(parameters));
+			} else {
+				mayLiterals[1] = new MayView(symbol).call(parameters);
+			}
+			var may = Query.builder()
+					.parameters(parameters)
+					.clause(mayLiterals)
+					.build();
+			translator.may(may);
+		} else if (!defaultValue.may()) {
+			translator.mayNever();
+		}
+		storeBuilder.with(translator);
+	}
+}
-- 
cgit v1.2.3-70-g09d2