1 package org.simantics.scl.compiler.elaboration.chr;
3 import org.simantics.scl.compiler.elaboration.chr.plan.PostCommitOp;
4 import org.simantics.scl.compiler.elaboration.chr.plan.PreCommitOp;
5 import org.simantics.scl.compiler.elaboration.chr.planning.QueryPlanningContext;
6 import org.simantics.scl.compiler.elaboration.chr.relations.CHRConstraint;
7 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
8 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
9 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
10 import org.simantics.scl.compiler.elaboration.expressions.Expression;
11 import org.simantics.scl.compiler.elaboration.expressions.Variable;
12 import org.simantics.scl.compiler.elaboration.expressions.printing.ExpressionToStringVisitor;
13 import org.simantics.scl.compiler.errors.Locations;
14 import org.simantics.scl.compiler.internal.parsing.Symbol;
16 import gnu.trove.map.hash.TObjectIntHashMap;
17 import gnu.trove.set.hash.TIntHashSet;
19 public class CHRQuery extends Symbol {
20 public CHRLiteral[] literals;
22 public CHRQuery(CHRLiteral[] literals) {
23 this.literals = literals;
26 public void resolve(TranslationContext context) {
27 for(CHRLiteral literal : literals)
28 literal.resolve(context);
31 public void checkType(TypingContext context) {
32 for(CHRLiteral literal : literals)
33 literal.checkType(context);
36 public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
37 for(CHRLiteral literal : literals)
38 literal.collectVars(allVars, vars);
41 public void setLocationDeep(long loc) {
42 if(location == Locations.NO_LOCATION) {
44 for(CHRLiteral literal : literals)
45 literal.setLocationDeep(loc);
49 public boolean createQueryPlan(QueryPlanningContext context, Expression inputFact, int activeLiteralId, CHRConstraint initConstraint) {
50 for(int i=0;i<literals.length;++i) {
51 CHRLiteral literal = literals[i];
52 if(i == activeLiteralId)
53 context.activate(literal, inputFact, i);
55 context.add(literal, i);
57 if(activeLiteralId == -1 && inputFact != null) {
58 context.addInitFact(initConstraint, inputFact);
60 return context.createQueryPlan();
63 public void simplify(SimplificationContext context) {
64 for(CHRLiteral literal : literals)
65 literal.simplify(context);
68 public void createEnforcePlan(QueryPlanningContext context, int priority) {
69 context.addPlanOp(new PreCommitOp(location));
70 for(CHRLiteral literal : literals)
71 context.claim(context, literal);
72 context.addPlanOp(new PostCommitOp(location, priority));
75 public String toString() {
76 StringBuilder b = new StringBuilder();
77 ExpressionToStringVisitor visitor = new ExpressionToStringVisitor(b);