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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/*
* SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.store.query.term;
import tools.refinery.store.query.InvalidQueryException;
import tools.refinery.store.query.equality.LiteralEqualityHelper;
import tools.refinery.store.query.equality.LiteralHashCodeHelper;
import tools.refinery.store.query.substitution.Substitution;
import tools.refinery.store.query.valuation.Valuation;
import java.util.Collections;
import java.util.HashSet;
import java.util.Objects;
import java.util.Set;
// {@link Object#equals(Object)} is implemented by {@link AbstractTerm}.
@SuppressWarnings("squid:S2160")
public abstract class BinaryTerm<R, T1, T2> extends AbstractTerm<R> {
private final Class<T1> leftType;
private final Class<T2> rightType;
private final Term<T1> left;
private final Term<T2> right;
protected BinaryTerm(Class<R> type, Class<T1> leftType, Class<T2> rightType, Term<T1> left, Term<T2> right) {
super(type);
if (!left.getType().equals(leftType)) {
throw new InvalidQueryException("Expected left %s to be of type %s, got %s instead".formatted(
left, leftType.getName(), left.getType().getName()));
}
if (!right.getType().equals(rightType)) {
throw new InvalidQueryException("Expected right %s to be of type %s, got %s instead".formatted(
right, rightType.getName(), right.getType().getName()));
}
this.leftType = leftType;
this.rightType = rightType;
this.left = left;
this.right = right;
}
public Class<T1> getLeftType() {
return leftType;
}
public Class<T2> getRightType() {
return rightType;
}
public Term<T1> getLeft() {
return left;
}
public Term<T2> getRight() {
return right;
}
@Override
public R evaluate(Valuation valuation) {
var leftValue = left.evaluate(valuation);
if (leftValue == null) {
return null;
}
var rightValue = right.evaluate(valuation);
if (rightValue == null) {
return null;
}
return doEvaluate(leftValue, rightValue);
}
protected abstract R doEvaluate(T1 leftValue, T2 rightValue);
@Override
public boolean equalsWithSubstitution(LiteralEqualityHelper helper, AnyTerm other) {
if (!super.equalsWithSubstitution(helper, other)) {
return false;
}
var otherBinaryTerm = (BinaryTerm<?, ?, ?>) other;
return leftType.equals(otherBinaryTerm.leftType) &&
rightType.equals(otherBinaryTerm.rightType) &&
left.equalsWithSubstitution(helper, otherBinaryTerm.left) &&
right.equalsWithSubstitution(helper, otherBinaryTerm.right);
}
@Override
public int hashCodeWithSubstitution(LiteralHashCodeHelper helper) {
return Objects.hash(super.hashCodeWithSubstitution(helper), leftType.hashCode(), rightType.hashCode(),
left.hashCodeWithSubstitution(helper), right.hashCodeWithSubstitution(helper));
}
@Override
public Term<R> substitute(Substitution substitution) {
return doSubstitute(substitution, left.substitute(substitution), right.substitute(substitution));
}
public abstract Term<R> doSubstitute(Substitution substitution, Term<T1> substitutedLeft,
Term<T2> substitutedRight);
@Override
public Set<AnyDataVariable> getInputVariables() {
var inputVariables = new HashSet<>(left.getInputVariables());
inputVariables.addAll(right.getInputVariables());
return Collections.unmodifiableSet(inputVariables);
}
}
|
