Some more added functionality to simantics district editor etc

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

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/NetworkDrawingParticipant.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/NetworkDrawingParticipant.java
index 6ae6097c183b5a3e8b4cef542a12794fd3cfc11f..a340d8591f625f0fdb78a7de27edaec6b34c3112 100644 (file)
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/NetworkDrawingParticipant.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/NetworkDrawingParticipant.java
@@ -1,39 +1,83 @@
-\r
-package org.simantics.district.network.ui;\r
-\r
-import org.simantics.district.network.ui.nodes.NetworkDrawingNode;\r
-import org.simantics.g2d.canvas.impl.SGNodeReflection.SGInit;\r
-import org.simantics.g2d.diagram.IDiagram;\r
-import org.simantics.g2d.diagram.participant.AbstractDiagramParticipant;\r
-import org.simantics.scenegraph.g2d.G2DParentNode;\r
-import org.simantics.utils.datastructures.hints.IHintContext.Key;\r
-import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;\r
-\r
-public class NetworkDrawingParticipant extends AbstractDiagramParticipant {\r
-\r
-    /**\r
-     * A hint key for terminal pick distance in control pixels.\r
-     * @see #PICK_DIST\r
-     */\r
-    public static final Key KEY_PICK_DISTANCE = new KeyOf(Double.class, "PICK_DISTANCE");\r
-\r
-    /**\r
-     * Default terminal pick distance in control pixels.\r
-     * @see #DEFAULT_PICK_DISTANCE\r
-     */\r
-    public static final double PICK_DIST = 10;\r
-    \r
-    private NetworkDrawingNode node;\r
-    \r
-    @SGInit\r
-    public void initSG(G2DParentNode parent) {\r
-        node = parent.addNode("networkDrawingNode", NetworkDrawingNode.class);\r
-        node.setNetworkDrawingParticipant(this);\r
-    }\r
-    \r
-    @Override\r
-    protected void onDiagramSet(IDiagram newDiagram, IDiagram oldDiagram) {\r
-        node.setDiagram(newDiagram);\r
-    }\r
-\r
-}\r
+
+package org.simantics.district.network.ui;
+
+import java.awt.geom.Point2D;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.simantics.district.network.ui.adapters.DistrictNetworkVertexElement;
+import org.simantics.district.network.ui.nodes.DistrictNetworkVertexNode;
+import org.simantics.district.network.ui.nodes.NetworkDrawingNode;
+import org.simantics.g2d.canvas.impl.DependencyReflection.Dependency;
+import org.simantics.g2d.canvas.impl.SGNodeReflection.SGInit;
+import org.simantics.g2d.diagram.IDiagram;
+import org.simantics.g2d.diagram.handler.PickContext;
+import org.simantics.g2d.diagram.handler.PickRequest;
+import org.simantics.g2d.diagram.participant.AbstractDiagramParticipant;
+import org.simantics.g2d.element.IElement;
+import org.simantics.scenegraph.g2d.G2DNode;
+import org.simantics.scenegraph.g2d.G2DParentNode;
+import org.simantics.utils.datastructures.hints.IHintContext.Key;
+import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
+
+public class NetworkDrawingParticipant extends AbstractDiagramParticipant {
+
+    @Dependency
+    PickContext pick;
+    
+    /**
+     * A hint key for terminal pick distance in control pixels.
+     * @see #PICK_DIST
+     */
+    public static final Key KEY_PICK_DISTANCE = new KeyOf(Double.class, "PICK_DISTANCE");
+
+    /**
+     * Default terminal pick distance in control pixels.
+     * @see #DEFAULT_PICK_DISTANCE
+     */
+    public static final double PICK_DIST = 10;
+    
+    private NetworkDrawingNode node;
+    
+    @SGInit
+    public void initSG(G2DParentNode parent) {
+        node = parent.addNode("networkDrawingNode", NetworkDrawingNode.class);
+        node.setNetworkDrawingParticipant(this);
+    }
+    
+    @Override
+    protected void onDiagramSet(IDiagram newDiagram, IDiagram oldDiagram) {
+        node.setDiagram(newDiagram);
+    }
+
+    public boolean pickHoveredElement(Point2D currentMousePos) {
+        PickRequest req = new PickRequest(currentMousePos.getX(), currentMousePos.getY());
+        List<IElement> pickables = new ArrayList<IElement>();
+        pick.pick(diagram, req, pickables);
+        
+        List<IElement> snap = new ArrayList<>(diagram.getSnapshot());
+        
+        snap.removeAll(pickables);
+        
+        boolean changed = false;
+        for (IElement sn : snap) {
+            G2DNode node = sn.getHint(DistrictNetworkVertexElement.KEY_DN_VERTEX_NODE);
+            if (node instanceof DistrictNetworkVertexNode) {
+                if (((DistrictNetworkVertexNode) node).hover(false) && !changed) {
+                    changed = true;
+                }
+            }
+        }
+        
+        for (IElement elem : pickables) {
+            G2DNode node = elem.getHint(DistrictNetworkVertexElement.KEY_DN_VERTEX_NODE);
+            if (node instanceof DistrictNetworkVertexNode) {
+                if (((DistrictNetworkVertexNode) node).hover(true) && !changed) {
+                    changed = true;
+                }
+            }
+        }
+        return changed;
+    }
+
+}