bundles/org.simantics.diagram/src/org/simantics/diagram/elements/SVGNode.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.elements;
  13
  14 import java.awt.Graphics2D;
  15 import java.util.List;
  16 import java.util.Set;
  17
  18 import org.simantics.g2d.canvas.ICanvasContext;
  19 import org.simantics.g2d.diagram.participant.Selection;
  20 import org.simantics.g2d.element.IElement;
  21 import org.simantics.scenegraph.ExportableWidget.RasterOutputWidget;
  22 import org.simantics.scenegraph.INode;
  23 import org.simantics.scenegraph.g2d.IG2DNode;
  24 import org.simantics.scenegraph.g2d.events.EventTypes;
  25 import org.simantics.scenegraph.g2d.events.KeyEvent.KeyPressedEvent;
  26 import org.simantics.scenegraph.g2d.events.MouseEvent.MouseClickEvent;
  27 import org.simantics.scenegraph.g2d.events.command.CommandEvent;
  28 import org.simantics.scenegraph.g2d.nodes.SVGNodeAssignment;
  29 import org.simantics.scenegraph.g2d.nodes.SingleElementNode;
  30 import org.simantics.scenegraph.utils.NodeUtil;
  31
  32 import com.kitfox.svg.SVGDiagram;
  33 import com.kitfox.svg.SVGException;
  34
  35 @RasterOutputWidget
  36 public class SVGNode extends org.simantics.scenegraph.g2d.nodes.SVGNode {
  37
  38     private static final long serialVersionUID = 4735066193941274186L;
  39
  40     transient EditorStateManager esm  = null;
  41     transient boolean parametersProcessed = false;
  42
  43     @Override
  44     public void cleanup() {
  45         if (esm != null)
  46             removeEventHandler(this);
  47         super.cleanup();
  48     }
  49
  50     @Override
  51     public void render(Graphics2D g2d) {
  52         if (!parametersProcessed) {
  53             SingleElementNode sne = getSingleElementNode();
  54             if (sne != null) {
  55                 Boolean enableEditing = sne.getParameter("enableEditing");
  56                 if (enableEditing != null && enableEditing) {
  57                     esm = new EditorStateManager(this);
  58                     addEventHandler(this);
  59                 }
  60                 parametersProcessed = true;
  61             }
  62         }
  63
  64         super.render(g2d);
  65     }
  66
  67     @Override
  68     protected int dynamicHash() {
  69         return esm != null ? esm.currentHash() : 0;
  70     }
  71
  72     @Override
  73     protected boolean applyAssignments(SVGDiagram diagram, List<SVGNodeAssignment> assignments) throws SVGException {
  74         boolean changed = super.applyAssignments(diagram, assignments);
  75         if (esm != null && !esm.applyEditMode(diagram))
  76             diagram.updateTime(0);
  77         return changed;
  78     }
  79
  80     private boolean isSelected(IG2DNode node) {
  81         ICanvasContext ctx = DiagramNodeUtil.getPossibleCanvasContext(this);
  82         if (ctx == null)
  83             return false;
  84         IElement ie = DiagramNodeUtil.getElement(ctx, node);
  85         if (ie == null)
  86             return false;
  87         Selection sel = ctx.getAtMostOneItemOfClass(Selection.class);
  88         if (sel != null) {
  89             Set<IElement> elems = sel.getSelection(0);
  90             if (elems.size() == 1 && elems.contains(ie))
  91                 return true;
  92         }
  93         return false;
  94     }
  95
  96     private boolean isSelected() {
  97         SingleElementNode sne = getSingleElementNode();
  98         if (isSelected(sne))
  99             return true;
 100         INode n = sne.getParent();
 101         while (n != null) {
 102             if (isSelected((IG2DNode)n))
 103                 return true;
 104             n = n.getParent();
 105         }
 106         return false;
 107     }
 108
 109     @Override
 110     protected boolean mouseClicked(MouseClickEvent e) {
 111         if(esm != null) {
 112             if(esm.isEditMode()) {
 113                 esm.applyEdit();
 114                 esm.deactivateEdit();
 115                 return true;
 116             }
 117             if(isSelected()) {
 118                 if(esm.tryToStartEditMode(diagramCache, e))
 119                     return true;
 120             }
 121         }
 122         return super.mouseClicked(e);
 123     }
 124
 125     SingleElementNode getSingleElementNode() {
 126         return (SingleElementNode)NodeUtil.getNearestParentOfType(this, SingleElementNode.class);
 127     }
 128
 129     @Override
 130     protected boolean keyPressed(KeyPressedEvent e) {
 131         if (esm != null && esm.keyPressed(e))
 132             return true;
 133         return super.keyPressed(e);
 134     }
 135
 136     @Override
 137     protected boolean handleCommand(CommandEvent e) {
 138         if (esm != null
 139                 && !esm.isEditMode()
 140                 && "org.eclipse.ui.edit.rename".equals(e.command.id)
 141                 && isSelected()
 142                 && esm.tryToStartEditMode(diagramCache)) {
 143             return true;
 144         }
 145         return super.handleCommand(e);
 146     }
 147
 148     @Override
 149     public int getEventMask() {
 150         return EventTypes.KeyPressedMask | EventTypes.MouseClickMask | EventTypes.CommandMask;
 151     }
 152
 153 }