bundles/org.simantics.mapping/src/org/simantics/mapping/rule/instructions/ClaimRuleInstruction.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.mapping.rule.instructions;
  13
  14 import gnu.trove.map.hash.TIntIntHashMap;
  15 import gnu.trove.set.hash.TIntHashSet;
  16
  17 import java.util.Arrays;
  18
  19 import org.simantics.db.ReadGraph;
  20 import org.simantics.db.WriteGraph;
  21 import org.simantics.db.exception.DatabaseException;
  22 import org.simantics.layer0.utils.triggers.IModification;
  23 import org.simantics.mapping.IContextualModification;
  24 import org.simantics.mapping.constraint.instructions.IInstruction;
  25
  26 public class ClaimRuleInstruction implements IRuleInstruction {
  27
  28         IInstruction instruction;
  29         IRuleInstruction continuation;
  30
  31         public ClaimRuleInstruction(IInstruction instruction,
  32                 IRuleInstruction continuation) {
  33                 this.instruction = instruction;
  34                 this.continuation = continuation;
  35         }
  36
  37         public ClaimRuleInstruction(IInstruction instruction) {
  38                 this(instruction, null);
  39         }
  40
  41         @Override
  42         public IModification execute(ReadGraph g, Object[] bindings) throws DatabaseException {
  43                 final IContextualModification modi = instruction.claim(g, bindings);
  44                 if(modi != null) {
  45                         final Object[] curBindings = Arrays.copyOf(bindings, bindings.length);
  46                         return new IModification() {
  47
  48                                 @Override
  49                                 public void perform(WriteGraph g) throws DatabaseException {
  50                                         modi.perform(g, curBindings);
  51                                         if(continuation != null) {
  52                                                 IModification modi2 = continuation.execute(g, curBindings);
  53                                                 if(modi2 != null)
  54                                                         modi2.perform(g);
  55                                         }
  56                                 }
  57
  58                         };
  59                 }
  60                 if(continuation != null)
  61                         return continuation.execute(g, bindings);
  62                 return null;
  63         }
  64
  65         @Override
  66         public void doExecute(WriteGraph g, Object[] bindings) throws DatabaseException {
  67                 instruction.doClaim(g, bindings);
  68         }
  69
  70         @Override
  71         public void collectVariables(TIntHashSet reads, TIntHashSet writes) {
  72                 instruction.collectVariables(reads, writes);
  73                 if(continuation != null)
  74                         continuation.collectVariables(reads, writes);
  75         }
  76
  77         @Override
  78         public void mapVariables(TIntIntHashMap map) {
  79                 instruction.mapVariables(map);
  80                 if(continuation != null)
  81                         continuation.mapVariables(map);
  82         }
  83         @Override
  84         public void toString(StringBuilder b, int indent) {
  85                 b.append("claim ");
  86                 instruction.toString(b, indent+1);
  87                 if(continuation != null) {
  88                         b.append('\n');
  89                         for(int i=0;i<indent;++i)
  90                                 b.append(INDENTATION);
  91                         b.append("and ");
  92                         continuation.toString(b, indent+1);
  93                 }
  94         }
  95 }