aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/translator/multiobject/EqualsRelationRewriter.java
blob: 61b9488ccf892b0674cbd8239c01cbe89965c80b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
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);
	}
}