Merged changes from feature/scl to master.

[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / chr / plan / ClaimOp.java

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/ClaimOp.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/ClaimOp.java
new file mode 100644 (file)
index 0000000..663d237
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/ClaimOp.java
@@ -0,0 +1,38 @@
+package org.simantics.scl.compiler.elaboration.chr.plan;\r
+\r
+import org.simantics.scl.compiler.compilation.CompilationContext;\r
+import org.simantics.scl.compiler.elaboration.chr.relations.CHRConstraint;\r
+import org.simantics.scl.compiler.elaboration.expressions.Expression;\r
+import org.simantics.scl.compiler.internal.codegen.references.IVal;\r
+import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;\r
+\r
+public class ClaimOp extends PlanOp {\r
+\r
+    CHRConstraint constraint;\r
+    Expression[] parameters;\r
+\r
+    public ClaimOp(long location, CHRConstraint constraint, Expression[] parameters) {\r
+        super(location);\r
+        this.constraint = constraint;\r
+        this.parameters = parameters;\r
+    }\r
+\r
+    @Override\r
+    public void toString(StringBuilder b) {\r
+        b.append("CLAIM   ").append(constraint);\r
+        for (Expression parameter : parameters)\r
+            b.append(" (").append(parameter).append(')');\r
+    }\r
+\r
+    @Override\r
+    public void generateCode(CompilationContext context, PlanContext planContext, CodeWriter w) {\r
+        IVal[] parameterVars = new IVal[parameters.length+1];\r
+        parameterVars[0] = planContext.generateNewId(location, w);\r
+        for(int i=0;i<parameters.length;++i)\r
+            parameterVars[i+1] = parameters[i].toVal(context.environment, w);\r
+        IVal newFact = w.apply(location, constraint.constructor, parameterVars);\r
+        w.apply(location, constraint.addProcedure, planContext.storeVar, newFact);\r
+        planContext.nextOp(w);\r
+    }\r
+\r
+}\r