Merged changes from feature/scl to master.

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

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/IterateConstraintOp.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/IterateConstraintOp.java
new file mode 100644 (file)
index 0000000..ec7dd9d
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/IterateConstraintOp.java
@@ -0,0 +1,91 @@
+package org.simantics.scl.compiler.elaboration.chr.plan;\r
+\r
+import java.util.ArrayList;\r
+\r
+import org.simantics.scl.compiler.compilation.CompilationContext;\r
+import org.simantics.scl.compiler.constants.Constant;\r
+import org.simantics.scl.compiler.constants.JavaComparisonOperation;\r
+import org.simantics.scl.compiler.constants.singletons.NullCheck;\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.elaboration.expressions.Variable;\r
+import org.simantics.scl.compiler.internal.codegen.continuations.ICont;\r
+import org.simantics.scl.compiler.internal.codegen.references.IVal;\r
+import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;\r
+import org.simantics.scl.compiler.types.Types;\r
+\r
+public class IterateConstraintOp extends PlanOp {\r
+    \r
+    public CHRConstraint constraint;\r
+    public Variable[] variables;\r
+    public Expression[] expressions;\r
+    public int boundMask;\r
+    public boolean killAfterMatch;\r
+    public boolean passive;\r
+    \r
+    public IterateConstraintOp(long location, CHRConstraint constraint, Variable[] variables, Expression[] expressions,\r
+            int boundMask, boolean killAfterMatch, boolean passive) {\r
+        super(location);\r
+        this.constraint = constraint;\r
+        this.variables = variables;\r
+        this.expressions = expressions;\r
+        this.boundMask = boundMask;\r
+        this.killAfterMatch = killAfterMatch;\r
+        this.passive = passive;\r
+        \r
+    }\r
+\r
+    @Override\r
+    public void toString(StringBuilder b) {\r
+        b.append("ITERATE ").append(constraint);\r
+        for(int i=0;i<expressions.length;++i)\r
+            if((boundMask & (1 << i)) != 0) {\r
+                b.append(" (").append(expressions[i]).append(")");\r
+            }\r
+            else {\r
+                b.append(" ").append(variables[i]);\r
+            }\r
+    }\r
+\r
+    @Override\r
+    public void generateCode(CompilationContext context, PlanContext planContext, CodeWriter w) {\r
+        CodeWriter body = w.createBlock(constraint.factType);\r
+        CodeWriter nextFact = w.createBlock();\r
+        ICont bodyContinuation = body.getContinuation();\r
+        CodeWriter end = w.createBlock();\r
+        IVal fact = body.getParameters()[0];\r
+\r
+        ArrayList<IVal> parameters = new ArrayList<IVal>(expressions.length+1);\r
+        parameters.add(planContext.storeVar);\r
+        for(int i=0;i<expressions.length;++i)\r
+            if(((boundMask>>i)&1)==1)\r
+                parameters.add(expressions[i].toVal(context.environment, w));\r
+        w.jump(bodyContinuation, w.apply(location,\r
+                constraint.fetchFromIndex(context, boundMask), parameters.toArray(new IVal[parameters.size()])));\r
+\r
+        body.branchAwayIf(body.apply(location, NullCheck.INSTANCE.createSpecialization(constraint.factType), fact),\r
+                end.getContinuation());\r
+        IVal id = body.apply(location, constraint.accessId, fact);\r
+        for(PartnerFact partnerFact : planContext.partnerFacts)\r
+            if(partnerFact.active && !passive) {\r
+                body.branchAwayUnless(body.apply(location, JavaComparisonOperation.ILESS, id, partnerFact.id),\r
+                        nextFact.getContinuation());\r
+            }\r
+            else if(partnerFact.constraint == constraint) {\r
+                body.branchAwayIf(body.apply(location, JavaComparisonOperation.IEQUAL, id, partnerFact.id),\r
+                        nextFact.getContinuation());\r
+            }\r
+\r
+        for(int i=0;i<variables.length;++i)\r
+            if(((boundMask>>i)&1)==0)\r
+                variables[i].setVal(body.apply(location, constraint.accessComponent(i), fact));\r
+        Constant nextElement = constraint.nextElement(context, boundMask);\r
+        planContext.partnerFacts.add(new PartnerFact(false, id, constraint, fact, constraint.mayBeRemoved(), killAfterMatch, nextElement, bodyContinuation, end.getContinuation()));\r
+        planContext.nextOp(body);\r
+        if(body.isUnfinished()) \r
+            body.jump(nextFact.getContinuation());\r
+        nextFact.jump(bodyContinuation, nextFact.apply(location, nextElement, fact));\r
+\r
+        w.continueAs(end);\r
+    }   \r
+}\r