bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagram/style/DocumentDecorationStyle.java

   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
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.modeling.ui.diagram.style;\r
  13 \r
  14 import java.awt.geom.AffineTransform;\r
  15 import java.awt.geom.Rectangle2D;\r
  16 import java.util.HashSet;\r
  17 import java.util.Set;\r
  18 \r
  19 import org.simantics.db.ReadGraph;\r
  20 import org.simantics.db.Resource;\r
  21 import org.simantics.db.exception.DatabaseException;\r
  22 import org.simantics.db.layer0.variable.Variable;\r
  23 import org.simantics.diagram.profile.StyleBase;\r
  24 import org.simantics.diagram.synchronization.graph.DiagramGraphUtil;\r
  25 import org.simantics.document.DocumentResource;\r
  26 import org.simantics.modeling.ModelingResources;\r
  27 import org.simantics.modeling.ui.Activator;\r
  28 import org.simantics.scenegraph.INode;\r
  29 import org.simantics.scenegraph.g2d.nodes.Decoration;\r
  30 import org.simantics.scenegraph.g2d.nodes.DecorationSVGNode;\r
  31 import org.simantics.scenegraph.g2d.nodes.SVGNode;\r
  32 import org.simantics.scenegraph.profile.EvaluationContext;\r
  33 import org.simantics.scenegraph.profile.common.ProfileVariables;\r
  34 import org.simantics.scenegraph.utils.NodeUtil;\r
  35 import org.simantics.utils.datastructures.map.Tuple;\r
  36 \r
  37 /**\r
  38  * @author Antti Villberg\r
  39  */\r
  40 public class DocumentDecorationStyle extends StyleBase<DocumentResult> {\r
  41 \r
  42     private static final String DECORATION_NODE_NAME = "documentDecorations";\r
  43 \r
  44     private Set<Resource> getContexts(ReadGraph graph, Resource element) throws DatabaseException {\r
  45 \r
  46         ModelingResources MOD = ModelingResources.getInstance(graph);\r
  47         HashSet<Resource> result = new HashSet<Resource>();\r
  48         result.add(element);\r
  49         Resource config = graph.getPossibleObject(element, MOD.ElementToComponent);\r
  50         if (config != null) result.add(config);\r
  51         config = graph.getPossibleObject(element, MOD.DiagramConnectionToConnection);\r
  52         if (config != null) result.add(config);\r
  53         return result;\r
  54         \r
  55     }\r
  56     \r
  57     @Override\r
  58     public DocumentResult calculateStyle(ReadGraph graph, Resource runtimeDiagram, Resource entry, Resource element, Variable configuration) throws DatabaseException {\r
  59         \r
  60         DocumentResource DOC = DocumentResource.getInstance(graph);\r
  61         for(Resource r : getContexts(graph, element)) {\r
  62                 if(graph.hasStatement(r, DOC.HasDocumentation)) {\r
  63                 AffineTransform transform = DiagramGraphUtil.getAffineTransform(graph, element);\r
  64                         return new DocumentResult(transform);\r
  65                 }\r
  66         }\r
  67         \r
  68         return null;\r
  69         \r
  70     }\r
  71 \r
  72     @Override\r
  73     public void applyStyleForNode(EvaluationContext observer, INode node, DocumentResult result) {\r
  74         if (result == null) {\r
  75             ProfileVariables.denyChild(node, "", DECORATION_NODE_NAME);\r
  76             return;\r
  77         }\r
  78 \r
  79         SVGNode svgNode = ProfileVariables.claimChild(node, "", DECORATION_NODE_NAME, DecorationSVGNode.class, observer);\r
  80 \r
  81         Rectangle2D bounds = NodeUtil.getLocalBounds(node, Decoration.class);\r
  82 \r
  83         double tx = bounds.getX();\r
  84         double ty = bounds.getY();\r
  85         double h = bounds.getHeight();\r
  86         \r
  87         svgNode.setZIndex( Integer.MAX_VALUE );\r
  88         svgNode.setTransform( AffineTransform.getTranslateInstance(tx, ty+h) ); \r
  89         svgNode.setData(Activator.DOCUMENT_SVG_TEXT);\r
  90 \r
  91     }\r
  92 \r
  93     @Override\r
  94     protected void cleanupStyleForNode(INode node) {\r
  95         ProfileVariables.denyChild(node, "", DECORATION_NODE_NAME);\r
  96     }\r
  97 \r
  98     @Override\r
  99     public String toString() {\r
 100         return "Document decoration";\r
 101     }\r
 102 \r
 103 }\r
 104 \r
 105 /**\r
 106  * This is needed to keep the issue decoration up-to-date when its parent\r
 107  * element moves.\r
 108  */\r
 109 class DocumentResult extends Tuple {\r
 110     public DocumentResult(AffineTransform transform) {\r
 111         super(transform);\r
 112     }\r
 113 }\r