1 /*******************************************************************************
\r
2 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
\r
3 * in Industry THTH ry.
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.modeling.mapping;
\r
14 import gnu.trove.map.hash.THashMap;
\r
16 import java.util.Map;
\r
18 import org.simantics.db.ReadGraph;
\r
19 import org.simantics.db.Resource;
\r
20 import org.simantics.db.Statement;
\r
21 import org.simantics.db.WriteGraph;
\r
22 import org.simantics.db.common.utils.NameUtils;
\r
23 import org.simantics.db.common.utils.OrderedSetUtils;
\r
24 import org.simantics.db.exception.CancelTransactionException;
\r
25 import org.simantics.db.exception.DatabaseException;
\r
26 import org.simantics.db.layer0.request.PossibleModel;
\r
27 import org.simantics.diagram.stubs.DiagramResource;
\r
28 import org.simantics.diagram.stubs.G2DResource;
\r
29 import org.simantics.diagram.synchronization.ISynchronizationContext;
\r
30 import org.simantics.diagram.synchronization.StatementEvaluation;
\r
31 import org.simantics.diagram.synchronization.graph.BasicResources;
\r
32 import org.simantics.diagram.synchronization.graph.CopyAdvisorUtil;
\r
33 import org.simantics.diagram.synchronization.graph.GraphCopyAdvisor;
\r
34 import org.simantics.diagram.synchronization.graph.GraphSynchronizationHints;
\r
35 import org.simantics.layer0.Layer0;
\r
36 import org.simantics.utils.datastructures.BinaryFunction;
\r
39 * @author Tuukka Lehtonen
\r
41 public class ElementCopyAdvisor extends GraphCopyAdvisor {
\r
44 DiagramResource DIA;
\r
47 * This is necessary to have DIA.Flag type copied over to the copied flag.
\r
48 * Diagram mapping will have problems and potentially break the
\r
49 * configuration if the type is not the same as in the source.
\r
51 BinaryFunction<StatementEvaluation, ReadGraph, Statement> statementAdvisor =
\r
52 new BinaryFunction<StatementEvaluation, ReadGraph, Statement>() {
\r
54 public StatementEvaluation call(ReadGraph graph, Statement stm) {
\r
55 if (DIA.HasFlagType.equals(stm.getPredicate()))
\r
56 return StatementEvaluation.INCLUDE;
\r
57 if (G2D.HasFontStyle.equals(stm.getPredicate()))
\r
58 return StatementEvaluation.INCLUDE;
\r
59 return StatementEvaluation.USE_DEFAULT;
\r
64 public Evaluation canCopy(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourcecontainer, Resource targetContainer) throws DatabaseException {
\r
65 BasicResources br = context.get(GraphSynchronizationHints.BASIC_RESOURCES);
\r
66 return graph.isInstanceOf(source, br.DIA.Element) ? Evaluation.SUPPORTED : Evaluation.NOT_SUPPORTED;
\r
70 public Object copy(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourceContainer, Resource targetContainer) throws DatabaseException {
\r
71 return copy(context, graph, source, sourceContainer, targetContainer, new THashMap<Object, Object>());
\r
75 public Object copy(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourceContainer,
\r
76 Resource targetContainer, Map<Object, Object> map) throws DatabaseException {
\r
77 Resource model = graph.syncRequest(new PossibleModel(targetContainer));
\r
78 //if (model == null)
\r
79 // throw new IllegalArgumentException("no model found for copy target container " + NameUtils.getSafeName(graph, targetContainer, true));
\r
81 Layer0 L0 = Layer0.getInstance(graph);
\r
82 G2D = G2DResource.getInstance(graph);
\r
83 DIA = DiagramResource.getInstance(graph);
\r
85 Resource copy = (model != null) ?
\r
86 CopyAdvisorUtil.copy3(graph, source, model, statementAdvisor, map) :
\r
87 CopyAdvisorUtil.copy2(graph, source, statementAdvisor, map);
\r
89 if (graph.hasStatement(sourceContainer, L0.ConsistsOf, source))
\r
90 graph.claim(targetContainer, L0.ConsistsOf, copy);
\r
96 public Object cut(ISynchronizationContext context, WriteGraph graph, Resource source, Resource sourceContainer,
\r
97 Resource targetContainer) throws DatabaseException {
\r
99 Layer0 L0 = Layer0.getInstance(graph);
\r
100 DiagramResource DIA = DiagramResource.getInstance(graph);
\r
102 if (sourceContainer.equals(targetContainer))
\r
105 // disconnect from source diagram
\r
106 if (!OrderedSetUtils.remove(graph, sourceContainer, source))
\r
107 // Unlink failed for some reason.
\r
108 throw new CancelTransactionException("Failed to unlink element "
\r
109 + NameUtils.getSafeName(graph, source) + " from source diagram "
\r
110 + NameUtils.getSafeName(graph, sourceContainer));
\r
112 // connect to target diagram
\r
113 boolean prepend = graph.isInstanceOf(source, DIA.Connection);
\r
114 boolean add = true;
\r
116 add = OrderedSetUtils.addFirst(graph, targetContainer, source);
\r
118 add = OrderedSetUtils.add(graph, targetContainer, source);
\r
120 // Link failed for some reason.
\r
121 throw new CancelTransactionException("Failed to link element "
\r
122 + NameUtils.getSafeName(graph, source) + " to target diagram "
\r
123 + NameUtils.getSafeName(graph, targetContainer));
\r
125 // This handles e.g. connections, which are not part of container
\r
126 if(graph.hasStatement(sourceContainer, L0.ConsistsOf, source)) {
\r
127 graph.deny(sourceContainer, L0.ConsistsOf, source);
\r
128 graph.claim(targetContainer, L0.ConsistsOf, source);
\r