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.diagram.synchronization.graph;
14 import org.simantics.db.Resource;
15 import org.simantics.db.WriteGraph;
16 import org.simantics.db.common.utils.OrderedSetUtils;
17 import org.simantics.db.exception.DatabaseException;
18 import org.simantics.diagram.content.ConnectionUtil;
19 import org.simantics.diagram.stubs.DiagramResource;
20 import org.simantics.diagram.synchronization.CopyAdvisor;
21 import org.simantics.diagram.synchronization.IModifiableSynchronizationContext;
22 import org.simantics.diagram.synchronization.ModificationAdapter;
23 import org.simantics.diagram.synchronization.SynchronizationHints;
24 import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager;
25 import org.simantics.diagram.ui.DiagramModelHints;
26 import org.simantics.g2d.diagram.DiagramHints;
27 import org.simantics.g2d.diagram.DiagramMutator;
28 import org.simantics.g2d.diagram.IDiagram;
29 import org.simantics.g2d.element.ElementHints;
30 import org.simantics.g2d.element.ElementUtils;
31 import org.simantics.g2d.element.IElement;
32 import org.simantics.layer0.Layer0;
33 import org.simantics.structural.stubs.StructuralResource2;
36 * @author Tuukka Lehtonen
38 public class AddConnection extends ModificationAdapter {
40 IModifiableSynchronizationContext context;
42 Resource diagramResource;
45 Resource connectionClassResource;
47 public AddConnection(IModifiableSynchronizationContext context, IDiagram diagram, IElement element) {
48 super(ADD_CONNECTION_PRIORITY);
49 this.context = context;
50 this.diagram = diagram;
51 this.diagramResource = diagram.getHint(DiagramModelHints.KEY_DIAGRAM_RESOURCE);
52 assert this.diagramResource != null;
53 this.element = element;
54 this.copyOf = element.getHint(ElementHints.KEY_COPY_OF_OBJECT);
56 this.connectionClassResource = ElementUtils.checkedAdapt(element.getElementClass(), Resource.class);
60 public void perform(WriteGraph graph) throws DatabaseException {
61 ConnectionUtil cu = new ConnectionUtil(graph);
62 DiagramMutator mutator = diagram.getHint(DiagramHints.KEY_MUTATOR);
64 Resource resource = null;
66 // 3. Try to copy the element from an existing element if requested.
68 CopyAdvisor ca = diagram.getHint(SynchronizationHints.COPY_ADVISOR);
70 Resource sourceDiagram = graph.getPossibleObject(copyOf, Layer0.getInstance(graph).PartOf);
71 if (sourceDiagram == null)
72 sourceDiagram = OrderedSetUtils.getSingleOwnerList(graph, copyOf, DiagramResource.getInstance(graph).Diagram);
73 resource = CopyAdvisorUtil.copy(context, graph, ca, copyOf, sourceDiagram, diagramResource);
74 if (resource != null) {
75 OrderedSetUtils.addFirst(graph, diagramResource, resource);
81 resource = cu.newConnection(diagramResource, connectionClassResource);
84 copyConnectionType(graph, copyOf, resource);
86 element.setHint(ElementHints.KEY_OBJECT, resource);
87 mutator.register(element, resource);
89 GraphLayerManager glm = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
91 glm.removeFromAllLayers(graph, resource);
92 glm.putElementOnVisibleLayers(diagram, graph, resource);
96 public static void copyConnectionType(WriteGraph graph, Resource source, Resource target) throws DatabaseException {
97 // Copy HasConnectionType relation
98 StructuralResource2 sr = StructuralResource2.getInstance(graph);
99 Resource connectionType = graph.getPossibleObject(source, sr.HasConnectionType);
100 if (connectionType != null) {
101 graph.deny(target, sr.HasConnectionType);
102 graph.claim(target, sr.HasConnectionType, null, connectionType);