1 files changed, 107 insertions, 0 deletions

diff --git a/store/src/main/java/org/eclipse/viatra/solver/data/query/building/PredicateBuilder_string.java b/store/src/main/java/org/eclipse/viatra/solver/data/query/building/PredicateBuilder_string.java
new file mode 100644
index 00000000..41f85d39
--- /dev/null
+++ b/store/src/main/java/org/eclipse/viatra/solver/data/query/building/PredicateBuilder_string.java
@@ -0,0 +1,107 @@
+package org.eclipse.viatra.solver.data.query.building;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.List;
+
+import org.eclipse.viatra.solver.data.query.view.RelationView;
+
+public class PredicateBuilder_string {
+        private PredicateBuilder_string() {}
+        
+        public static PredicateBuild1 predicate(String name) {
+                return new PredicateBuild1(name);
+        }
+        public static class PredicateBuild1 {
+                private String name;
+                public PredicateBuild1(String name) {
+                        this.name = name;
+                }
+                public PredicateBuild2 parameters(String... parameters) {
+                        return new PredicateBuild2(name, parameters);
+                }
+        }
+        public static class PredicateBuild2 {
+                private String name;
+                private String[] parameters;
+                public PredicateBuild2(String name, String[] parameters) {
+                        this.name = name;
+                        this.parameters = parameters;
+                }
+                
+                public PredicateBuild3 clause(DNFAtom...constraints) {
+                        return new PredicateBuild3(name,parameters,List.<DNFAtom[]>of(constraints));
+                }
+        }
+        public static class PredicateBuild3 {
+                String name;
+                String[] parameters;
+                List<DNFAtom[]> clauses;
+                public PredicateBuild3(String name, String[] parameters, List<DNFAtom[]> clauses) {
+                        super();
+                        this.name = name;
+                        this.parameters = parameters;
+                        this.clauses = clauses;
+                }
+                
+                public PredicateBuild3 clause(DNFAtom...constraints) {
+                        List<DNFAtom[]> newClauses = new ArrayList<>();
+                        newClauses.addAll(clauses);
+                        newClauses.add(constraints);
+                        return new PredicateBuild3(name, parameters, newClauses);
+                }
+                public DNFPredicate build() {
+                        List<Variable> newParameters = new ArrayList<>(this.parameters.length);
+                        for(int i = 0; i<this.parameters.length; i++) {
+                                newParameters.add(new Variable(parameters[i]));
+                        }
+                        
+                        List<DNFAnd> newClauses = new ArrayList<>(this.clauses.size());
+                        for(DNFAtom[] clause : this.clauses) {
+                                List<DNFAtom> constraints = new ArrayList<>(clause.length);
+                                Collections.addAll(constraints, clause);
+                                newClauses.add(new DNFAnd(new HashSet<>(), constraints));
+                        }
+                        
+                        return new DNFPredicate(name,newParameters,newClauses);
+                }
+        }
+        
+        private static Variable stringToVariable(String name) {
+                if(name != null) {
+                        return new Variable(name);
+                } else {
+                        return null;
+                }
+        }
+        private static List<Variable> stringToVariable(String[] names) {
+                List<Variable> variables = new ArrayList<>();
+                for(int i = 0; i<names.length; i++) {
+                        variables.add(stringToVariable(names[i]));
+                }
+                return variables;
+        }
+        
+        public static EquivalenceAtom cEquals(String v1, String v2) {
+                return new EquivalenceAtom(true,stringToVariable(v1),stringToVariable(v2));
+        }
+        public static EquivalenceAtom cNotEquals(String v1, String v2) {
+                return new EquivalenceAtom(false,stringToVariable(v1),stringToVariable(v2));
+        }
+        
+        public static RelationAtom cInRelation(RelationView<?> view, String... variables) {
+                
+                return new RelationAtom(view, stringToVariable(variables));
+        }
+        
+        public static PredicateAtom cInPredicate(DNFPredicate referred, String... variables) {
+                return new PredicateAtom(true, false, referred, stringToVariable(variables));
+        }
+        public static PredicateAtom cInTransitivePredicate(DNFPredicate referred, String... variables) {
+                return new PredicateAtom(true, true, referred, stringToVariable(variables));
+        }
+        public static PredicateAtom cNotInPredicate(DNFPredicate referred, String... variables) {
+                return new PredicateAtom(false, false, referred, stringToVariable(variables));
+        }
+}