aboutsummaryrefslogtreecommitdiffstats
path: root/model-data/src/main/java/org/eclipse/viatra/solver/data/query/building/PredicateBuilder_string.java
blob: 41f85d392d12c232ee552ce673e7a5d484d9e842 (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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
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));
	}
}