X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fnodes%2FDistrictNetworkEdgeNode.java;h=00a1427f5f2e6d07f72c6770364f94892afb64c1;hb=835710a4f22ec2d91c287295fb176a0af62d1186;hp=1e577e97a4553faebcfebd82659f56c4a8c96894;hpb=3354c578fb8e65421d0fdb123310232ccdc597cf;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkEdgeNode.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkEdgeNode.java
index 1e577e97..00a1427f 100644
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkEdgeNode.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkEdgeNode.java
@@ -4,37 +4,50 @@ import java.awt.BasicStroke;
 import java.awt.Color;
 import java.awt.Graphics2D;
 import java.awt.RenderingHints;
-import java.awt.Stroke;
 import java.awt.geom.AffineTransform;
 import java.awt.geom.Line2D;
 import java.awt.geom.Path2D;
+import java.awt.geom.Point2D;
 import java.awt.geom.Rectangle2D;
 
 import org.simantics.district.network.ModelledCRS;
 import org.simantics.district.network.ui.DistrictNetworkEdge;
+import org.simantics.scenegraph.INode;
 import org.simantics.scenegraph.ISelectionPainterNode;
 import org.simantics.scenegraph.g2d.G2DNode;
+import org.simantics.scenegraph.g2d.G2DParentNode;
+import org.simantics.scenegraph.g2d.nodes.SVGNode;
 import org.simantics.scenegraph.utils.GeometryUtils;
 import org.simantics.scenegraph.utils.NodeUtil;
 
-public class DistrictNetworkEdgeNode extends G2DNode implements ISelectionPainterNode {
+public class DistrictNetworkEdgeNode extends G2DParentNode implements ISelectionPainterNode {
 
     private static final long serialVersionUID = 8049769475036519806L;
 
-    private static final Stroke     SELECTION_STROKE = new BasicStroke(1f, BasicStroke.CAP_BUTT, BasicStroke.JOIN_MITER);
-    private static final Color      SELECTION_COLOR  = new Color(255, 0, 255, 96);
+    private static final BasicStroke STROKE           = new BasicStroke(4, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND);
+    private static final Color       SELECTION_COLOR  = new Color(255, 0, 255, 96);
 
     private DistrictNetworkEdge edge;
     private Rectangle2D bounds;
-    private Line2D path;
+    private transient Line2D path;
 
-    private static final BasicStroke STROKE = new BasicStroke(4, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND);
     private boolean scaleStroke = true;
-
     private Color color;
-
     private Double stroke;
-
+    private transient Color dynamicColor = null;
+
+    // Dimensions for shut-off valve symbol
+    private static final double left = -0.25;
+    private static final double top = -0.25;
+    private static final double width = 0.5;
+    private static final double height = 0.5;
+
+    private static final Rectangle2D NORMAL = new Rectangle2D.Double(left, top, width, height);
+    
+    private transient Point2D centerPoint;
+    private transient Rectangle2D symbolRect;
+    private transient AffineTransform symbolTransform;
+    
     @Override
     public void init() {
     }
@@ -54,7 +67,6 @@ public class DistrictNetworkEdgeNode extends G2DNode implements ISelectionPainte
         Color oldColor = g2d.getColor();
         BasicStroke oldStroke = (BasicStroke) g2d.getStroke();
 
-        g2d.setColor(color);
         BasicStroke bs = null;
         if (scaleStroke) {
             double scale = GeometryUtils.getScale(g2d.getTransform());
@@ -73,18 +85,47 @@ public class DistrictNetworkEdgeNode extends G2DNode implements ISelectionPainte
             g2d.draw(path);
         }
 
+        g2d.setColor(dynamicColor != null ? dynamicColor : color);
         g2d.setStroke(bs);
         g2d.draw(path);
 
         // Reset
         g2d.setStroke(oldStroke);
         g2d.setColor(oldColor);
+        
+        // Render SVG symbol
+        double viewScaleRecip = 10;
+        if (scaleStroke) {
+            double scale = GeometryUtils.getScale(g2d.getTransform());
+            scale = Math.max(10000, Math.min(scale, 50000));
+            viewScaleRecip /= scale;
+        }
+        
+        Point2D p = getCenterPoint();
+        symbolRect = DistrictNetworkNodeUtils.calculateDrawnGeometry(p, NORMAL, symbolRect, viewScaleRecip);
+        symbolTransform = DistrictNetworkNodeUtils.getTransformToRectangle(symbolRect, symbolTransform);
+        
+        for (INode nn : getNodes()) {
+            G2DNode g2dNode = (G2DNode)nn;
+            g2dNode.setTransform(symbolTransform);
+            g2dNode.render(g2d);
+        }
+        
         g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint);
 
         if (ot != null)
             g2d.setTransform(ot);
     }
 
+    private Point2D getCenterPoint() {
+        if (centerPoint == null)
+            centerPoint = new Point2D.Double();
+        
+        Rectangle2D bounds = path.getBounds2D();
+        centerPoint.setLocation(bounds.getCenterX(), bounds.getCenterY());
+        return centerPoint;
+    }
+
     public static Line2D calculateLine(DistrictNetworkEdge edge, Line2D result) {
         // Convert to screen coordinates
         double startX = ModelledCRS.longitudeToX(edge.getStartPoint().getX());
@@ -153,4 +194,16 @@ public class DistrictNetworkEdgeNode extends G2DNode implements ISelectionPainte
         this.stroke = stroke;
     }
 
+    @PropertySetter(value = "dynamicColor")
+    public void setDynamicColor(Color color) {
+        this.dynamicColor = color;
+    }
+
+    @PropertySetter(value = "SVG")
+    public void setSVG(String value) {
+        for (INode nn : this.getNodes())
+            if (nn instanceof SVGNode)
+                ((SVGNode)nn).setData(value);
+    }
+    
 }