Merged changes from feature/scl to master.

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

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/PostCommitOp.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/PostCommitOp.java
new file mode 100644 (file)
index 0000000..edbb975
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/chr/plan/PostCommitOp.java
@@ -0,0 +1,69 @@
+package org.simantics.scl.compiler.elaboration.chr.plan;\r
+\r
+import org.simantics.scl.compiler.compilation.CompilationContext;\r
+import org.simantics.scl.compiler.constants.IntegerConstant;\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.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
+\r
+public class PostCommitOp extends PlanOp {\r
+    int priority;\r
+    \r
+    public PostCommitOp(long location, int priority) {\r
+        super(location);\r
+        this.priority = priority;\r
+    }\r
+\r
+    @Override\r
+    public void toString(StringBuilder b) {\r
+        b.append("POST_COMMIT " + priority);\r
+    }\r
+\r
+    @Override\r
+    public void generateCode(CompilationContext context, PlanContext planContext, CodeWriter w) {\r
+        if(planContext.currentId != null) {\r
+            w.apply(location, planContext.ruleset.writeCurrentId, planContext.storeVar, planContext.currentId);\r
+            planContext.currentId = null;\r
+            w.apply(location, planContext.ruleset.activateProcedure, planContext.storeVar, new IntegerConstant(priority));\r
+        }\r
+        for(PartnerFact activeFact : planContext.partnerFacts) {\r
+            if(activeFact.killAfterMatch) {\r
+                if(activeFact.nextFact == null)\r
+                    w.jump(activeFact.finishCont);\r
+                else {\r
+                    CodeWriter iterateAlive = w.createBlock(activeFact.constraint.factType);\r
+                    w.jump(iterateAlive.getContinuation(), w.apply(location, activeFact.nextFact, activeFact.factVar));\r
+                    iterateUntilLiveFactFound(iterateAlive, activeFact);                    \r
+                }\r
+                break;\r
+            }\r
+            else if(activeFact.mayBeRemoved) {\r
+                if(activeFact.nextFact == null) {\r
+                    w.branchAwayUnless(activeFact.isAlive(location, w), activeFact.finishCont);\r
+                }\r
+                else {\r
+                    CodeWriter failure = w.createBlock();\r
+                    CodeWriter iterateAlive = w.createBlock(activeFact.constraint.factType);\r
+                    w.branchAwayUnless(activeFact.isAlive(location, w), failure.getContinuation());\r
+                    failure.jump(iterateAlive.getContinuation(), failure.apply(location, activeFact.nextFact, activeFact.factVar));\r
+                    iterateUntilLiveFactFound(iterateAlive, activeFact);     \r
+                }\r
+            }\r
+        }\r
+        // PostCommit does not call nextOp\r
+    }\r
+    \r
+    private void iterateUntilLiveFactFound(CodeWriter w, PartnerFact activeFact) {\r
+        ICont initialContinuation = w.getContinuation();\r
+        CHRConstraint constraint = activeFact.constraint;\r
+        IVal fact = w.getParameters()[0];\r
+        CodeWriter dead = w.createBlock();\r
+        w.branchAwayIf(w.apply(location, NullCheck.INSTANCE.createSpecialization(constraint.factType), fact), activeFact.finishCont);\r
+        w.branchAwayUnless(activeFact.isAlive(location, w, fact), dead.getContinuation());\r
+        w.jump(activeFact.continueCont, fact);\r
+        dead.jump(initialContinuation, dead.apply(location, activeFact.nextFact, fact));\r
+    }\r
+\r
+}\r