bundles/org.simantics.modeling/src/org/simantics/modeling/mapping/MappedElementCopyAdvisor.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.modeling.mapping;
  13
  14 import gnu.trove.map.hash.THashMap;
  15
  16 import java.util.Map;
  17
  18 import org.simantics.db.Resource;
  19 import org.simantics.db.WriteGraph;
  20 import org.simantics.db.exception.DatabaseException;
  21 import org.simantics.diagram.synchronization.CopyAdvisor;
  22 import org.simantics.diagram.synchronization.ISynchronizationContext;
  23 import org.simantics.diagram.synchronization.graph.GraphCopyAdvisor;
  24 import org.simantics.modeling.ModelingResources;
  25 import org.simantics.utils.datastructures.Pair;
  26
  27 /**
  28  * @author Tuukka Lehtonen
  29  */
  30 public class MappedElementCopyAdvisor extends GraphCopyAdvisor {
  31
  32     protected final CopyAdvisor elementAdvisor;
  33     protected final CopyAdvisor componentAdvisor;
  34
  35     public MappedElementCopyAdvisor(CopyAdvisor elementAdvisor, CopyAdvisor componentAdvisor) {
  36         this.elementAdvisor = elementAdvisor;
  37         this.componentAdvisor = componentAdvisor;
  38     }
  39
  40     @Override
  41     public Evaluation canCopy(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourceContainer, Resource targetContainer)
  42     throws DatabaseException {
  43         if (CopyAdvisor.SUPPORTED.contains(elementAdvisor.canCopy(context, source, sourceContainer, targetContainer))) {
  44             ModelingResources MOD = context.get(ModelingSynchronizationHints.MODELING_RESOURCE);
  45             //Layer0 L0 = Layer0.getInstance(graph);
  46
  47             Resource component = graph.getPossibleObject(source, MOD.ElementToComponent);
  48             if (component != null) {
  49                 Resource sourceComposite = sourceContainer != null ? graph.getPossibleObject(sourceContainer, MOD.DiagramToComposite) : null;
  50                 Resource targetComposite = targetContainer != null ? graph.getPossibleObject(targetContainer, MOD.DiagramToComposite) : null;
  51                 //if(targetComposite == null) targetComposite = graph.getPossibleObject(component, L0.PartOf);
  52
  53                 Evaluation componentEval = componentAdvisor.canCopy(context, component, sourceComposite, targetComposite);
  54                 if (componentEval != Evaluation.SUPPORTED)
  55                     return componentEval;
  56             }
  57
  58             Resource connection = graph.getPossibleObject(source, MOD.DiagramConnectionToConnection);
  59             if (connection != null) {
  60                 Resource sourceComposite = graph.getPossibleObject(sourceContainer, MOD.DiagramToComposite);
  61                 Resource targetComposite = graph.getPossibleObject(targetContainer, MOD.DiagramToComposite);
  62                 //if (targetComposite == null) targetComposite = graph.getPossibleObject(component, L0.PartOf);
  63
  64                 Evaluation connectionEval = componentAdvisor.canCopy(context, connection, sourceComposite, targetComposite);
  65                 if (connectionEval != Evaluation.SUPPORTED)
  66                     return connectionEval;
  67
  68                 Resource specialConnection = graph.getPossibleObject(source, MOD.DiagramConnectionToConnectionSpecial);
  69                 if (specialConnection != null) {
  70                     connectionEval = componentAdvisor.canCopy(context, specialConnection, sourceComposite, targetComposite);
  71                 }
  72
  73                 return connectionEval;
  74             }
  75             return Evaluation.SUPPORTED;
  76         }
  77         return Evaluation.NOT_SUPPORTED;
  78     }
  79
  80     @Override
  81     public Object copy(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourceContainer, Resource targetContainer) throws DatabaseException {
  82         return copy(context, graph, source, sourceContainer, targetContainer, new THashMap<Object, Object>());
  83     }
  84
  85     @Override
  86     public Object copy(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourceContainer, Resource targetContainer, Map<Object, Object> map) throws DatabaseException {
  87         ModelingResources MOD = context.get(ModelingSynchronizationHints.MODELING_RESOURCE);
  88         if (MOD == null)
  89             throw new IllegalStateException("no ModelingSynchronizationHints.MODELING_RESOURCE hint");
  90
  91         Resource sourceComposite = sourceContainer != null ? graph.getPossibleObject(sourceContainer, MOD.DiagramToComposite) : null;
  92         Resource targetComposite = targetContainer != null ? graph.getPossibleObject(targetContainer, MOD.DiagramToComposite) : null;
  93         //if(targetComposite == null) targetComposite = graph.getPossibleObject(component, L0.PartOf);
  94
  95         Resource elementCopy = (Resource) elementAdvisor.copy(context, source, sourceContainer, targetContainer, map);
  96
  97         Resource component = graph.getPossibleObject(source, MOD.ElementToComponent);
  98         Resource componentCopy = null;
  99         if (component != null) {
 100             componentCopy = (Resource) componentAdvisor.copy(context, component, sourceComposite, targetComposite, map);
 101             if (componentCopy != null) {
 102                 graph.claim(elementCopy, MOD.ElementToComponent, componentCopy);
 103             }
 104         }
 105
 106         Resource connection = graph.getPossibleObject(source, MOD.DiagramConnectionToConnection);
 107         if (connection != null) {
 108             Resource connectionCopy = (Resource) componentAdvisor.copy(context, connection, sourceComposite, targetComposite, map);
 109             if (connectionCopy != null) {
 110                 graph.claim(elementCopy, MOD.DiagramConnectionToConnection, connectionCopy);
 111             }
 112
 113             Resource specialConnection = graph.getPossibleObject(source, MOD.DiagramConnectionToConnectionSpecial);
 114             if (specialConnection != null) {
 115                 Resource specialConnectionCopy = (Resource) componentAdvisor.copy(context, specialConnection, sourceComposite, targetComposite, map);
 116                 if (specialConnectionCopy != null) {
 117                     graph.claim(elementCopy, MOD.DiagramConnectionToConnectionSpecial, specialConnectionCopy);
 118                 }
 119             }
 120
 121             // NOTICE: this does not copy MOD.ConnectorToComponent relations.
 122             // Those are unfortunately reconstructed in Paster since they would
 123             // be rather difficult to deterministically reconstruct here.
 124         }
 125
 126         return elementCopy;
 127     }
 128
 129     @Override
 130     public Object cut(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourceContainer,
 131             Resource targetContainer) throws DatabaseException {
 132         ModelingResources MOD = context.get(ModelingSynchronizationHints.MODELING_RESOURCE);
 133
 134         Resource sourceComposite = graph.getPossibleObject(sourceContainer, MOD.DiagramToComposite);
 135         Resource targetComposite = graph.getPossibleObject(targetContainer, MOD.DiagramToComposite);
 136
 137         Resource component = graph.getPossibleObject(source, MOD.ElementToComponent);
 138         Resource connection = graph.getPossibleObject(source, MOD.DiagramConnectionToConnection);
 139
 140         Object elementResult = elementAdvisor.cut(context, source, sourceContainer, targetContainer);
 141         Object componentResult = null;
 142
 143         if (sourceComposite != null && targetComposite != null) {
 144             if (component != null)
 145                 componentResult = componentAdvisor.cut(context, component, sourceComposite, targetComposite);
 146             if (connection != null)
 147                 componentResult = componentAdvisor.cut(context, connection, sourceComposite, targetComposite);
 148         }
 149
 150         if (elementResult == null && componentResult == null)
 151             return null;
 152         return Pair.make(elementResult, componentResult);
 153     }
 154
 155     @Override
 156     public void onFinish(ISynchronizationContext context) {
 157         super.onFinish(context);
 158         elementAdvisor.onFinish(context);
 159         componentAdvisor.onFinish(context);
 160     }
 161
 162
 163 }