bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/CompositeImage.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.adapter;
  13
  14 import java.awt.Shape;
  15 import java.awt.geom.Rectangle2D;
  16 import java.util.Collection;
  17 import java.util.EnumSet;
  18 import java.util.List;
  19
  20 import org.simantics.g2d.element.ElementClass;
  21 import org.simantics.g2d.element.ElementHints;
  22 import org.simantics.g2d.element.ElementUtils;
  23 import org.simantics.g2d.element.IElement;
  24 import org.simantics.g2d.element.SceneGraphNodeKey;
  25 import org.simantics.g2d.element.handler.SceneGraph;
  26 import org.simantics.g2d.image.Image;
  27 import org.simantics.scenegraph.INode;
  28 import org.simantics.scenegraph.Node;
  29 import org.simantics.scenegraph.g2d.G2DParentNode;
  30 import org.simantics.scenegraph.g2d.nodes.SingleElementNode;
  31 import org.simantics.utils.datastructures.hints.IHintContext.Key;
  32
  33 /**
  34  * Element Composition -> Image
  35  *
  36  * @author Toni Kalajainen <toni.kalajainen@vtt.fi>
  37  */
  38 public class CompositeImage implements Image {
  39
  40     Collection<IElement> elements;
  41     private Shape shape;
  42     private Rectangle2D bounds;
  43     //IHintContext parentHints;
  44     static EnumSet<Feature> feats = VOLATILE_VECTOR;
  45
  46     public static final Key  KEY_SG_NODE             = new SceneGraphNodeKey(Node.class, "COMPOSITE_IMAGE_SG_NODE");
  47
  48     public CompositeImage(Collection<IElement> elements)
  49     {
  50         //this.nodeIdentifier = nodeIdentifier;
  51         this.elements = elements;
  52 //        ITask task = ThreadLogger.getInstance().begin("getElementShapesOnDiagram");
  53         // getElementShapesOnDiagram is ridiculously slow with some input data..
  54         // With Balas it took almost 2sec to calculate one Area.add() for the last terminal element.
  55         //shape = ElementUtils.getElementShapesOnDiagram(elements);
  56 //        shape = ElementUtils.getElementBoundsOnDiagram(elements);
  57   //      task.finish();
  58 //        bounds = shape.getBounds2D();
  59     }
  60
  61     @Override
  62     public Rectangle2D getBounds() {
  63         if(bounds == null) {
  64             Shape shape = getShape();
  65             // Shape may be null.
  66             if (shape == null)
  67                 return new Rectangle2D.Double();
  68             bounds = shape.getBounds2D();
  69         }
  70         return bounds;
  71     }
  72
  73     @Override
  74     public EnumSet<Feature> getFeatures() {
  75         return feats;
  76     }
  77
  78     private final Shape getShape() {
  79         if(shape == null) {
  80             shape = ElementUtils.getElementBoundsOnDiagram(elements);
  81         }
  82         return shape;
  83     }
  84
  85     @Override
  86     public Shape getOutline() {
  87         return getShape();
  88     }
  89
  90     public Collection<IElement> getElements() {
  91         return elements;
  92     }
  93
  94     @Override
  95     public void addImageListener(ImageListener listener) {
  96     }
  97
  98     @Override
  99     public void removeImageListener(ImageListener listener) {
 100     }
 101
 102 //    public void setParentHints(IHintContext hints) {
 103 //        this.parentHints = hints;
 104 //    }
 105
 106     // Rendering is single-threaded, this is used while rendering.
 107     //Rectangle2D tempBounds = new Rectangle2D.Double(0, 0, 0, 0);
 108
 109     @Override
 110     public Node init(G2DParentNode parent) {
 111         if (elements.size() < 2) {
 112             // Optimization for 0..1 element composites
 113             for (IElement e : elements) {
 114                 ElementClass ec = e.getElementClass();
 115                 G2DParentNode node = getOrCreateParentNode(parent);
 116                 List<SceneGraph> nodeHandlers = ec.getItemsByClass(SceneGraph.class);
 117                 for (SceneGraph n : nodeHandlers) {
 118 //                    n.init(e, parent);
 119                     n.init(e, node);
 120                 }
 121                 return node; // Valid node must be returned because transform is set afterwards
 122             }
 123         }
 124
 125         // For N element composites
 126
 127         G2DParentNode node = getOrCreateParentNode(parent);
 128 //        Rectangle2D bounds = tempBounds;
 129
 130         int zIndex = 0;
 131         for (IElement e : elements) {
 132             ElementClass ec = e.getElementClass();
 133 //            InternalSize size = ec.getSingleItem(InternalSize.class);
 134 //            size.getBounds(e, bounds);
 135
 136 //            Transform transform = e.getElementClass().getSingleItem(Transform.class);
 137 //            AffineTransform at2 = transform.getTransform(e);
 138 //            if (at2 == null)
 139 //                continue;
 140             SingleElementNode holder = node.getOrCreateNode(ElementUtils.generateNodeId(e), SingleElementNode.class);
 141             //SingleElementNode holder = parent.getOrCreateNode(ElementUtils.generateNodeId(e), SingleElementNode.class);
 142             holder.setZIndex(++zIndex);
 143             holder.setKey(e.getHint(ElementHints.KEY_OBJECT));
 144             holder.setTypeClass(e.getHint(ElementHints.KEY_TYPE_CLASS));
 145
 146             List<SceneGraph> nodeHandlers = ec.getItemsByClass(SceneGraph.class);
 147             for(SceneGraph n : nodeHandlers) {
 148                 n.init(e, holder);
 149             }
 150         }
 151
 152         return node; // Valid node must be returned because transform is set afterwards
 153     }
 154
 155     private G2DParentNode getOrCreateParentNode(G2DParentNode parent) {
 156         G2DParentNode node = (G2DParentNode)parent.getNode("composite_image_"+this.hashCode());
 157         if (node == null) {
 158             for (INode n : parent.getNodes()) {
 159                 // #7663: prevent the code from removing other element graphics from directly under another element
 160                 if (!(n instanceof SingleElementNode))
 161                     n.remove();
 162             }
 163             // Removed this grouping node as unnecessary, just use the given parent node
 164             node = parent.getOrCreateNode("composite_image_"+this.hashCode(), G2DParentNode.class);
 165         }
 166         return node;
 167     }
 168
 169 }