-package org.simantics.scl.compiler.elaboration.chr;\r
-\r
-import org.simantics.scl.compiler.elaboration.chr.plan.PostCommitOp;\r
-import org.simantics.scl.compiler.elaboration.chr.plan.PreCommitOp;\r
-import org.simantics.scl.compiler.elaboration.chr.planning.QueryPlanningContext;\r
-import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
-import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
-import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
-import org.simantics.scl.compiler.elaboration.expressions.Expression;\r
-import org.simantics.scl.compiler.elaboration.expressions.Variable;\r
-import org.simantics.scl.compiler.elaboration.expressions.VariableProcedure;\r
-import org.simantics.scl.compiler.elaboration.expressions.printing.ExpressionToStringVisitor;\r
-import org.simantics.scl.compiler.errors.Locations;\r
-import org.simantics.scl.compiler.internal.parsing.Symbol;\r
-\r
-import gnu.trove.map.hash.TObjectIntHashMap;\r
-import gnu.trove.set.hash.THashSet;\r
-import gnu.trove.set.hash.TIntHashSet;\r
-\r
-public class CHRQuery extends Symbol {\r
- public CHRLiteral[] literals;\r
-\r
- public CHRQuery(CHRLiteral[] literals) {\r
- this.literals = literals;\r
- }\r
-\r
- public void resolve(TranslationContext context) {\r
- for(CHRLiteral literal : literals)\r
- literal.resolve(context);\r
- }\r
-\r
- public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {\r
- for(CHRLiteral literal : literals)\r
- literal.collectRefs(allRefs, refs);\r
- }\r
-\r
- public void checkType(TypingContext context) {\r
- for(CHRLiteral literal : literals)\r
- literal.checkType(context);\r
- }\r
- \r
- public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {\r
- for(CHRLiteral literal : literals)\r
- literal.collectVars(allVars, vars);\r
- }\r
-\r
- public void forVariables(VariableProcedure procedure) {\r
- for(CHRLiteral literal : literals)\r
- literal.forVariables(procedure);\r
- }\r
-\r
- public void collectFreeVariables(THashSet<Variable> vars) {\r
- for(CHRLiteral literal : literals)\r
- literal.collectFreeVariables(vars);\r
- }\r
-\r
- public void setLocationDeep(long loc) {\r
- if(location == Locations.NO_LOCATION) {\r
- this.location = loc;\r
- for(CHRLiteral literal : literals)\r
- literal.setLocationDeep(loc);\r
- }\r
- }\r
- \r
- public void createQueryPlan(QueryPlanningContext context, Expression inputFact, int activeLiteralId) {\r
- for(int i=0;i<literals.length;++i) {\r
- CHRLiteral literal = literals[i];\r
- if(i == activeLiteralId)\r
- context.activate(literal, inputFact, i);\r
- else\r
- context.add(literal, i);\r
- }\r
- context.createQueryPlan();\r
- }\r
- \r
- public void simplify(SimplificationContext context) {\r
- for(CHRLiteral literal : literals)\r
- literal.simplify(context);\r
- }\r
-\r
- public void createEnforcePlan(QueryPlanningContext context, int priority) {\r
- context.addPlanOp(new PreCommitOp(location));\r
- for(CHRLiteral literal : literals)\r
- context.claim(context, literal);\r
- context.addPlanOp(new PostCommitOp(location, priority));\r
- }\r
- \r
- public String toString() {\r
- StringBuilder b = new StringBuilder();\r
- ExpressionToStringVisitor visitor = new ExpressionToStringVisitor(b);\r
- visitor.visit(this);\r
- return b.toString();\r
- }\r
-}\r
+package org.simantics.scl.compiler.elaboration.chr;
+
+import org.simantics.scl.compiler.elaboration.chr.plan.PostCommitOp;
+import org.simantics.scl.compiler.elaboration.chr.plan.PreCommitOp;
+import org.simantics.scl.compiler.elaboration.chr.planning.QueryPlanningContext;
+import org.simantics.scl.compiler.elaboration.chr.relations.CHRConstraint;
+import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
+import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
+import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
+import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
+import org.simantics.scl.compiler.elaboration.expressions.Expression;
+import org.simantics.scl.compiler.elaboration.expressions.Variable;
+import org.simantics.scl.compiler.elaboration.expressions.printing.ExpressionToStringVisitor;
+import org.simantics.scl.compiler.errors.Locations;
+import org.simantics.scl.compiler.internal.parsing.Symbol;
+
+import gnu.trove.map.hash.TObjectIntHashMap;
+import gnu.trove.set.hash.TIntHashSet;
+
+public class CHRQuery extends Symbol {
+ public CHRLiteral[] literals;
+
+ public CHRQuery(CHRLiteral[] literals) {
+ this.literals = literals;
+ }
+
+ public void resolve(TranslationContext context) {
+ for(CHRLiteral literal : literals)
+ literal.resolve(context);
+ }
+
+ public void checkType(TypingContext context) {
+ for(CHRLiteral literal : literals)
+ literal.checkType(context);
+ }
+
+ public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
+ for(CHRLiteral literal : literals)
+ literal.collectVars(allVars, vars);
+ }
+
+ public void setLocationDeep(long loc) {
+ if(location == Locations.NO_LOCATION) {
+ this.location = loc;
+ for(CHRLiteral literal : literals)
+ literal.setLocationDeep(loc);
+ }
+ }
+
+ public boolean createQueryPlan(QueryPlanningContext context, Expression inputFact, int activeLiteralId, CHRConstraint initConstraint) {
+ for(int i=0;i<literals.length;++i) {
+ CHRLiteral literal = literals[i];
+ if(i == activeLiteralId)
+ context.activate(literal, inputFact, i);
+ else
+ context.add(literal, i);
+ }
+ if(activeLiteralId == -1 && inputFact != null) {
+ context.addInitFact(initConstraint, inputFact);
+ }
+ return context.createQueryPlan();
+ }
+
+ public void simplify(SimplificationContext context) {
+ for(CHRLiteral literal : literals)
+ literal.simplify(context);
+ }
+
+ public void createEnforcePlan(QueryPlanningContext context, int priority) {
+ context.addPlanOp(new PreCommitOp(location));
+ for(CHRLiteral literal : literals)
+ context.claim(context, literal);
+ context.addPlanOp(new PostCommitOp(location, priority));
+ }
+
+ public String toString() {
+ StringBuilder b = new StringBuilder();
+ ExpressionToStringVisitor visitor = new ExpressionToStringVisitor(b);
+ visitor.visit(this);
+ return b.toString();
+ }
+
+ public CHRQuery replace(ReplaceContext context) {
+ CHRLiteral[] newLiterals = new CHRLiteral[literals.length];
+ for(int i=0;i<literals.length;++i)
+ newLiterals[i] = literals[i].replace(context);
+ return new CHRQuery(newLiterals);
+ }
+}