Implement district features to Simantics platform

[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / nodes / DistrictNetworkVertexNode.java

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
index 7365060b0254bc8c6f62d5c4cb4aeef8e1dcd6d5..a126ee5942a841f50a2108684a364a78052a4554 100644 (file)
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
@@ -1,54 +1,83 @@
-package org.simantics.district.network.ui.nodes;\r
-\r
-import java.awt.Color;\r
-import java.awt.Graphics2D;\r
-import java.awt.RenderingHints;\r
-import java.awt.geom.AffineTransform;\r
-import java.awt.geom.Rectangle2D;\r
-\r
-import org.simantics.district.network.ui.adapters.DistrictNetworkVertex;\r
-import org.simantics.scenegraph.g2d.G2DNode;\r
-\r
-public class DistrictNetworkVertexNode extends G2DNode {\r
-\r
-    private static final long serialVersionUID = -2641639101400236719L;\r
-    private DistrictNetworkVertex vertex;\r
-\r
-    @Override\r
-    public void render(Graphics2D g2d) {\r
-        AffineTransform ot = null;\r
-        AffineTransform t = getTransform();\r
-        if (t != null && !t.isIdentity()) {\r
-            ot = g2d.getTransform();\r
-            g2d.transform(getTransform());\r
-        }\r
-        \r
-        Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING);\r
-        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF);\r
-        \r
-        Color oldColor = g2d.getColor();\r
-        g2d.setColor(Color.RED);\r
-        // render\r
-        \r
-        Rectangle2D.Double rect = new Rectangle2D.Double(-0.5, -0.5, 1, 1);\r
-        g2d.draw(rect);\r
-        \r
-        // Reset stats\r
-        g2d.setColor(oldColor);\r
-        \r
-        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint);\r
-        \r
-        if (ot != null)\r
-            g2d.setTransform(ot);\r
-    }\r
-\r
-    @Override\r
-    public Rectangle2D getBoundsInLocal() {\r
-        return new Rectangle2D.Double(-0.5, -0.5, 1, 1);\r
-    }\r
-\r
-    public void setVertex(DistrictNetworkVertex vertex) {\r
-        this.vertex = vertex;\r
-    }\r
-\r
-}\r
+package org.simantics.district.network.ui.nodes;
+
+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.Rectangle2D;
+
+import org.simantics.district.network.ui.adapters.DistrictNetworkVertex;
+import org.simantics.scenegraph.g2d.G2DNode;
+import org.simantics.scenegraph.utils.GeometryUtils;
+
+public class DistrictNetworkVertexNode extends G2DNode {
+
+    private static final long serialVersionUID = -2641639101400236719L;
+    private DistrictNetworkVertex vertex;
+    
+    private static final double scale = 1;
+    
+    private double left = -0.5;
+    private double top = -0.5;
+    private double width = 1;
+    private double height = 1;
+    
+    private Stroke stroke = new BasicStroke(2);
+    private boolean scaleStroke = true;
+
+    @Override
+    public void init() {
+        setZIndex(2);
+    }
+    
+    @Override
+    public void render(Graphics2D g2d) {
+        AffineTransform ot = null;
+        AffineTransform t = getTransform();
+        if (t != null && !t.isIdentity()) {
+            ot = g2d.getTransform();
+            g2d.transform(getTransform());
+        }
+        
+        Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING);
+        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF);
+        
+        Color oldColor = g2d.getColor();
+        Stroke oldStroke = g2d.getStroke();
+        
+        g2d.setColor(Color.RED);
+        if (stroke != null) {
+            if (scaleStroke && stroke instanceof BasicStroke) {
+                BasicStroke bs = GeometryUtils.scaleStroke(stroke, (float) (1.0 / GeometryUtils.getScale(g2d.getTransform())));
+                g2d.setStroke(bs);
+            } else {
+                g2d.setStroke(stroke);
+            }
+        }
+        // render
+        
+        Rectangle2D.Double rect = new Rectangle2D.Double(left / scale, top / scale, width / scale, height / scale);
+        g2d.draw(rect);
+        
+        // Reset stats
+        g2d.setColor(oldColor);
+        g2d.setStroke(oldStroke);
+        
+        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint);
+        
+        if (ot != null)
+            g2d.setTransform(ot);
+    }
+
+    @Override
+    public Rectangle2D getBoundsInLocal() {
+        return new Rectangle2D.Double(left / scale, top / scale, width / scale, height / scale);
+    }
+
+    public void setVertex(DistrictNetworkVertex vertex) {
+        this.vertex = vertex;
+    }
+
+}