1 /*******************************************************************************
2 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
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
10 * VTT Technical Research Centre of Finland - initial API and implementation
11 *******************************************************************************/
12 package org.simantics.modeling.mapping;
14 import gnu.trove.map.hash.THashMap;
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;
28 * @author Tuukka Lehtonen
30 public class MappedElementCopyAdvisor extends GraphCopyAdvisor {
32 protected final CopyAdvisor elementAdvisor;
33 protected final CopyAdvisor componentAdvisor;
35 public MappedElementCopyAdvisor(CopyAdvisor elementAdvisor, CopyAdvisor componentAdvisor) {
36 this.elementAdvisor = elementAdvisor;
37 this.componentAdvisor = componentAdvisor;
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);
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);
53 Evaluation componentEval = componentAdvisor.canCopy(context, component, sourceComposite, targetComposite);
54 if (componentEval != Evaluation.SUPPORTED)
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);
64 Evaluation connectionEval = componentAdvisor.canCopy(context, connection, sourceComposite, targetComposite);
65 if (connectionEval != Evaluation.SUPPORTED)
66 return connectionEval;
68 Resource specialConnection = graph.getPossibleObject(source, MOD.DiagramConnectionToConnectionSpecial);
69 if (specialConnection != null) {
70 connectionEval = componentAdvisor.canCopy(context, specialConnection, sourceComposite, targetComposite);
73 return connectionEval;
75 return Evaluation.SUPPORTED;
77 return Evaluation.NOT_SUPPORTED;
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>());
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);
89 throw new IllegalStateException("no ModelingSynchronizationHints.MODELING_RESOURCE hint");
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);
95 Resource elementCopy = (Resource) elementAdvisor.copy(context, source, sourceContainer, targetContainer, map);
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);
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);
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);
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.
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);
134 Resource sourceComposite = graph.getPossibleObject(sourceContainer, MOD.DiagramToComposite);
135 Resource targetComposite = graph.getPossibleObject(targetContainer, MOD.DiagramToComposite);
137 Resource component = graph.getPossibleObject(source, MOD.ElementToComponent);
138 Resource connection = graph.getPossibleObject(source, MOD.DiagramConnectionToConnection);
140 Object elementResult = elementAdvisor.cut(context, source, sourceContainer, targetContainer);
141 Object componentResult = null;
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);
150 if (elementResult == null && componentResult == null)
152 return Pair.make(elementResult, componentResult);
156 public void onFinish(ISynchronizationContext context) {
157 super.onFinish(context);
158 elementAdvisor.onFinish(context);
159 componentAdvisor.onFinish(context);