aboutsummaryrefslogtreecommitdiffstats
path: root/store/src/main/java/org/eclipse/viatra/solver/data/query/building/RelationAtom.java
diff options
context:
space:
mode:
Diffstat (limited to 'store/src/main/java/org/eclipse/viatra/solver/data/query/building/RelationAtom.java')
-rw-r--r--store/src/main/java/org/eclipse/viatra/solver/data/query/building/RelationAtom.java16
1 files changed, 10 insertions, 6 deletions
diff --git a/store/src/main/java/org/eclipse/viatra/solver/data/query/building/RelationAtom.java b/store/src/main/java/org/eclipse/viatra/solver/data/query/building/RelationAtom.java
index f7152bba..adf2f8c7 100644
--- a/store/src/main/java/org/eclipse/viatra/solver/data/query/building/RelationAtom.java
+++ b/store/src/main/java/org/eclipse/viatra/solver/data/query/building/RelationAtom.java
@@ -7,37 +7,41 @@ import java.util.Set;
7import org.eclipse.viatra.solver.data.query.view.FilteredRelationView; 7import org.eclipse.viatra.solver.data.query.view.FilteredRelationView;
8import org.eclipse.viatra.solver.data.query.view.RelationView; 8import org.eclipse.viatra.solver.data.query.view.RelationView;
9 9
10public class RelationAtom implements DNFAtom{ 10public class RelationAtom implements DNFAtom {
11 RelationView<?> view; 11 RelationView<?> view;
12 List<Variable> substitution; 12 List<Variable> substitution;
13 13
14 public RelationAtom(RelationView<?> view, List<Variable> substitution) { 14 public RelationAtom(RelationView<?> view, List<Variable> substitution) {
15 this.view = view; 15 this.view = view;
16 this.substitution = substitution; 16 this.substitution = substitution;
17 } 17 }
18
18 public RelationView<?> getView() { 19 public RelationView<?> getView() {
19 return view; 20 return view;
20 } 21 }
22
21 public void setView(FilteredRelationView<?> view) { 23 public void setView(FilteredRelationView<?> view) {
22 this.view = view; 24 this.view = view;
23 } 25 }
26
24 public List<Variable> getSubstitution() { 27 public List<Variable> getSubstitution() {
25 return substitution; 28 return substitution;
26 } 29 }
30
27 public void setSubstitution(List<Variable> substitution) { 31 public void setSubstitution(List<Variable> substitution) {
28 this.substitution = substitution; 32 this.substitution = substitution;
29 } 33 }
30 34
31 @Override 35 @Override
32 public void unifyVariables(Map<String, Variable> variables) { 36 public void unifyVariables(Map<String, Variable> variables) {
33 for(int i = 0; i<this.substitution.size(); i++) { 37 for (int i = 0; i < this.substitution.size(); i++) {
34 final Object term = this.substitution.get(i); 38 final Object term = this.substitution.get(i);
35 if(term instanceof Variable) { 39 if (term instanceof Variable variableReference) {
36 Variable variableReference = (Variable) term;
37 this.substitution.set(i, DNFAtom.unifyVariables(variables, variableReference)); 40 this.substitution.set(i, DNFAtom.unifyVariables(variables, variableReference));
38 } 41 }
39 } 42 }
40 } 43 }
44
41 @Override 45 @Override
42 public void collectAllVariables(Set<Variable> variables) { 46 public void collectAllVariables(Set<Variable> variables) {
43 DNFAtom.addToCollection(variables, substitution); 47 DNFAtom.addToCollection(variables, substitution);