bundles/org.simantics.diagram/src/org/simantics/diagram/synchronization/graph/SetConnectionType.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.diagram.synchronization.graph;
  13
  14 import org.simantics.db.Resource;
  15 import org.simantics.db.WriteGraph;
  16 import org.simantics.db.exception.DatabaseException;
  17 import org.simantics.diagram.synchronization.ISynchronizationContext;
  18 import org.simantics.diagram.synchronization.ModificationAdapter;
  19 import org.simantics.diagram.ui.DiagramModelHints;
  20 import org.simantics.g2d.diagram.DiagramHints;
  21 import org.simantics.g2d.diagram.DiagramMutator;
  22 import org.simantics.g2d.diagram.IDiagram;
  23 import org.simantics.g2d.element.ElementHints;
  24 import org.simantics.g2d.element.IElement;
  25 import org.simantics.structural2.modelingRules.ConnectionJudgement;
  26 import org.simantics.structural2.modelingRules.IModelingRules;
  27
  28 /**
  29  * @author Tuukka Lehtonen
  30  */
  31 public class SetConnectionType extends ModificationAdapter {
  32
  33     ISynchronizationContext context;
  34     IDiagram                diagram;
  35     IElement                element;
  36
  37     public SetConnectionType(ISynchronizationContext context, IDiagram diagram, IElement element) {
  38         super(SET_CONNECTION_TYPE_PRIORITY);
  39         this.context = context;
  40         this.diagram = diagram;
  41         this.element = element;
  42     }
  43
  44     Resource getDesiredConnectionType(IElement e) {
  45         Object obj = e.getHint(ElementHints.KEY_CONNECTION_TYPE);
  46         if (obj instanceof Resource)
  47             return (Resource) obj;
  48         if (obj instanceof ConnectionJudgement) {
  49             ConnectionJudgement judgement = (ConnectionJudgement) obj;
  50             return judgement.connectionType;
  51         }
  52         return null;
  53     }
  54
  55     @Override
  56     public void perform(WriteGraph graph) throws DatabaseException {
  57         DiagramMutator mutator = diagram.getHint(DiagramHints.KEY_MUTATOR);
  58         IModelingRules modelingRules = diagram.getHint(DiagramModelHints.KEY_MODELING_RULES);
  59
  60         Resource connection = mutator.backendObject(element);
  61         Resource connectionType = getDesiredConnectionType(element);
  62
  63         if (connectionType != null && modelingRules != null) {
  64             modelingRules.setConnectionType(graph, connection, connectionType);
  65         }
  66     }
  67
  68 }