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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/*
* SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.store.query.literal;
import tools.refinery.store.query.Constraint;
import tools.refinery.store.query.equality.LiteralEqualityHelper;
import tools.refinery.store.query.substitution.Substitution;
import tools.refinery.store.query.term.ParameterDirection;
import tools.refinery.store.query.term.Variable;
import java.util.*;
public final class CallLiteral extends AbstractCallLiteral implements CanNegate<CallLiteral> {
private final CallPolarity polarity;
public CallLiteral(CallPolarity polarity, Constraint target, List<Variable> arguments) {
super(target, arguments);
var parameters = target.getParameters();
int arity = target.arity();
if (polarity.isTransitive()) {
if (arity != 2) {
throw new IllegalArgumentException("Transitive closures can only take binary relations");
}
if (parameters.get(0).isDataVariable() || parameters.get(1).isDataVariable()) {
throw new IllegalArgumentException("Transitive closures can only be computed over nodes");
}
}
this.polarity = polarity;
}
public CallPolarity getPolarity() {
return polarity;
}
@Override
protected Literal doSubstitute(Substitution substitution, List<Variable> substitutedArguments) {
return new CallLiteral(polarity, getTarget(), substitutedArguments);
}
@Override
public Set<Variable> getOutputVariables() {
if (polarity.isPositive()) {
return getArgumentsOfDirection(ParameterDirection.OUT);
}
return Set.of();
}
@Override
public Set<Variable> getInputVariables(Set<? extends Variable> positiveVariablesInClause) {
if (polarity.isPositive()) {
return getArgumentsOfDirection(ParameterDirection.IN);
}
return super.getInputVariables(positiveVariablesInClause);
}
@Override
public Set<Variable> getPrivateVariables(Set<? extends Variable> positiveVariablesInClause) {
if (polarity.isPositive()) {
return Set.of();
}
return super.getPrivateVariables(positiveVariablesInClause);
}
@Override
public Literal reduce() {
var reduction = getTarget().getReduction();
var negatedReduction = polarity.isPositive() ? reduction : reduction.negate();
return switch (negatedReduction) {
case ALWAYS_TRUE -> BooleanLiteral.TRUE;
case ALWAYS_FALSE -> BooleanLiteral.FALSE;
default -> this;
};
}
@Override
public boolean equalsWithSubstitution(LiteralEqualityHelper helper, Literal other) {
if (!super.equalsWithSubstitution(helper, other)) {
return false;
}
var otherCallLiteral = (CallLiteral) other;
return polarity.equals(otherCallLiteral.polarity);
}
@Override
public CallLiteral negate() {
return new CallLiteral(polarity.negate(), getTarget(), getArguments());
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
if (!super.equals(o)) return false;
CallLiteral that = (CallLiteral) o;
return polarity == that.polarity;
}
@Override
public int hashCode() {
return Objects.hash(super.hashCode(), polarity);
}
@Override
public String toString() {
var builder = new StringBuilder();
if (!polarity.isPositive()) {
builder.append("!(");
}
builder.append(getTarget().toReferenceString());
if (polarity.isTransitive()) {
builder.append("+");
}
builder.append("(");
var argumentIterator = getArguments().iterator();
if (argumentIterator.hasNext()) {
builder.append(argumentIterator.next());
while (argumentIterator.hasNext()) {
builder.append(", ").append(argumentIterator.next());
}
}
builder.append(")");
if (!polarity.isPositive()) {
builder.append(")");
}
return builder.toString();
}
}
|
