Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.mapping / src / org / simantics / mapping / rule / instructions / IfRuleInstruction.java

diff --git a/bundles/org.simantics.mapping/src/org/simantics/mapping/rule/instructions/IfRuleInstruction.java b/bundles/org.simantics.mapping/src/org/simantics/mapping/rule/instructions/IfRuleInstruction.java
new file mode 100644 (file)
index 0000000..c5b3070
--- /dev/null
+++ b/bundles/org.simantics.mapping/src/org/simantics/mapping/rule/instructions/IfRuleInstruction.java
@@ -0,0 +1,133 @@
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.mapping.rule.instructions;\r
+\r
+import gnu.trove.map.hash.TIntIntHashMap;\r
+import gnu.trove.set.hash.TIntHashSet;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.layer0.utils.triggers.IModification;\r
+import org.simantics.mapping.constraint.instructions.IInstruction;\r
+\r
+public class IfRuleInstruction implements IRuleInstruction {\r
+       IInstruction condition;\r
+       IRuleInstruction rule;\r
+       IRuleInstruction elseRule;      \r
+       \r
+       public IfRuleInstruction(IInstruction condition, IRuleInstruction rule, IRuleInstruction elseRule) {\r
+               this.condition = condition;\r
+               this.rule = rule;\r
+               this.elseRule = elseRule;\r
+       }\r
+       \r
+       public IfRuleInstruction(IInstruction condition, IRuleInstruction rule) {\r
+               this(condition, rule, null);\r
+       }       \r
+       \r
+       @Override\r
+       public IModification execute(ReadGraph g, final Object[] bindings) throws DatabaseException {\r
+               if(DEBUG)\r
+                       System.out.println("IfRuleInstruction: condition");\r
+               Object cont = condition.query(g, bindings);\r
+               if(cont == IInstruction.FAILURE) {\r
+                       if(DEBUG)\r
+                         System.out.println("IfRuleInstruction: else");\r
+                       if(elseRule!=null)\r
+                               return elseRule.execute(g, bindings);\r
+                       else\r
+                               return null;\r
+               }\r
+               do {\r
+                       if(DEBUG)\r
+                               System.out.println("IfRuleInstruction: modification");\r
+                       final IModification modi = rule.execute(g, bindings);\r
+                       if(modi != null) {\r
+                               if(cont==null)\r
+                                       return modi;\r
+                               final Object curCont = condition.next(g, bindings, cont);\r
+                               if(curCont==IInstruction.FAILURE)\r
+                                       return modi;\r
+                               return new IModification() {\r
+                                       @Override\r
+                                       public void perform(WriteGraph g) throws DatabaseException {\r
+                                               modi.perform(g);\r
+                                               Object cont = curCont;\r
+                                               do {\r
+                                                       rule.doExecute(g, bindings);\r
+                                                       if(cont==null)\r
+                                                               break;\r
+                                                       cont = condition.next(g, bindings, cont);\r
+                                               } while(cont != IInstruction.FAILURE);\r
+                                       }                               \r
+                               };      \r
+                       }\r
+                       if(cont==null)\r
+                               break;\r
+                       cont = condition.next(g, bindings, cont);\r
+               } while(cont != IInstruction.FAILURE);\r
+               return null;\r
+       }\r
+       \r
+       @Override\r
+       public void doExecute(WriteGraph g, Object[] bindings) throws DatabaseException {\r
+               Object cont = condition.query(g, bindings);\r
+               if(cont == IInstruction.FAILURE) {\r
+                       if(elseRule!=null) \r
+                               elseRule.doExecute(g, bindings);\r
+                       return;\r
+               }\r
+               do {\r
+                       rule.doExecute(g, bindings);\r
+                       if(cont==null)\r
+                               break;\r
+                       cont = condition.next(g, bindings, cont);\r
+               } while(cont != IInstruction.FAILURE);\r
+       }\r
+       \r
+       @Override\r
+       public void collectVariables(TIntHashSet reads, TIntHashSet writes) {\r
+               condition.collectVariables(reads, writes);\r
+               rule.collectVariables(reads, writes);\r
+               if(elseRule != null)\r
+                       elseRule.collectVariables(reads, writes);\r
+       }\r
+\r
+       @Override\r
+       public void mapVariables(TIntIntHashMap map) {\r
+               condition.mapVariables(map);\r
+               rule.mapVariables(map);\r
+               if(elseRule != null)\r
+                       elseRule.mapVariables(map);\r
+       }\r
+\r
+       @Override\r
+       public void toString(StringBuilder b, int indent) {             \r
+               b.append("if   ");\r
+               condition.toString(b, indent+1);\r
+               \r
+               b.append('\n');         \r
+               for(int i=0;i<indent;++i)\r
+                       b.append(INDENTATION);\r
+               b.append("then ");\r
+               rule.toString(b, indent+1);             \r
+               \r
+               if(elseRule != null) {\r
+                       b.append('\n');\r
+                       for(int i=0;i<indent;++i)\r
+                               b.append(INDENTATION);\r
+                       b.append("else ");\r
+                       elseRule.toString(b, indent+1);\r
+               }\r
+       }\r
+}\r