X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2FNetworkDrawingParticipant.java;h=da2bf32aec1a6ab3170e1386da2dddda8a2e9b7b;hb=27f0f6dd22d6826ae85ea89b1916d17ca9f83c41;hp=6ae6097c183b5a3e8b4cef542a12794fd3cfc11f;hpb=8034693a5c86d690649a70a62709d29c3fa197aa;p=simantics%2Fdistrict.git

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 6ae6097c..da2bf32a 100644
--- 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,121 @@
-
-package org.simantics.district.network.ui;
-
-import org.simantics.district.network.ui.nodes.NetworkDrawingNode;
-import org.simantics.g2d.canvas.impl.SGNodeReflection.SGInit;
-import org.simantics.g2d.diagram.IDiagram;
-import org.simantics.g2d.diagram.participant.AbstractDiagramParticipant;
-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 {
-
-    /**
-     * 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);
-    }
-
-}
+
+package org.simantics.district.network.ui;
+
+import java.awt.geom.AffineTransform;
+import java.awt.geom.Point2D;
+import java.awt.geom.Rectangle2D;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.simantics.district.network.ui.adapters.DistrictNetworkEdgeElement;
+import org.simantics.district.network.ui.adapters.DistrictNetworkVertexElement;
+import org.simantics.district.network.ui.nodes.DistrictNetworkEdgeNode;
+import org.simantics.district.network.ui.nodes.DistrictNetworkVertexNode;
+import org.simantics.district.network.ui.nodes.HoverSensitiveNode;
+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.Node;
+import org.simantics.scenegraph.g2d.G2DParentNode;
+import org.simantics.scenegraph.g2d.IG2DNode;
+import org.simantics.utils.datastructures.hints.IHintContext.Key;
+import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
+
+public class NetworkDrawingParticipant extends AbstractDiagramParticipant {
+
+    public static final String NETWORK_DRAWING_NODE = "networkDrawingNode";
+
+	@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;
+
+    private AffineTransform transform;
+    
+    public NetworkDrawingParticipant(AffineTransform transform) {
+        this.transform = transform;
+    }
+
+    @SGInit
+    public void initSG(G2DParentNode parent) {
+        node = parent.addNode(NETWORK_DRAWING_NODE, NetworkDrawingNode.class);
+        node.setTransform(transform);
+        node.setNetworkDrawingParticipant(this);
+    }
+    
+    @Override
+    protected void onDiagramSet(IDiagram newDiagram, IDiagram oldDiagram) {
+        node.setDiagram(newDiagram);
+    }
+
+    public boolean pickHoveredElement(Point2D currentMousePos, boolean isConnectionTool) {
+        PickRequest req = new PickRequest(new Rectangle2D.Double(currentMousePos.getX(), currentMousePos.getY(), 1e-8, 1e-8)).context(getContext());
+        List<IElement> pickables = new ArrayList<>();
+        pick.pick(diagram, req, pickables);
+
+        List<IElement> snap = diagram.getSnapshot();
+
+        boolean changed = false;
+        changed |= hoverNodes(snap, false, isConnectionTool, currentMousePos);
+        changed |= hoverNodes(pickables, true, isConnectionTool, currentMousePos);
+        return changed;
+    }
+
+    private boolean hoverNodes(List<IElement> elements, boolean hover, boolean isConnectionTool, Point2D p) {
+        boolean changed = false;
+        for (IElement elem : elements) {
+            Node node = elem.getHint(DistrictNetworkVertexElement.KEY_DN_VERTEX_NODE);
+            if (node instanceof DistrictNetworkVertexNode) {
+                changed |= ((DistrictNetworkVertexNode) node).hover(hover, isConnectionTool);
+                if (hover)
+                    ((DistrictNetworkVertexNode) node).setMousePosition(p);
+            } else {
+                node = elem.getHint(DistrictNetworkEdgeElement.KEY_DN_EDGE_NODE);
+                if (node instanceof DistrictNetworkEdgeNode) {
+                    for (IG2DNode n : ((DistrictNetworkEdgeNode) node).getNodes()) {
+                        if (n instanceof HoverSensitiveNode) {
+                            changed |= ((HoverSensitiveNode)n).hover(hover, isConnectionTool);
+                            if (hover)
+                                ((HoverSensitiveNode)n).setMousePosition(p);
+                        }
+                    }
+                }
+            }
+        }
+        return changed;
+    }
+
+    public boolean isHoveringOverNode(Point2D currentMousePos) {
+        PickRequest req = new PickRequest(currentMousePos).context(getContext());
+        List<IElement> pickables = new ArrayList<IElement>();
+        pick.pick(diagram, req, pickables);
+        for (IElement elem : pickables) {
+            Node node = elem.getHint(DistrictNetworkVertexElement.KEY_DN_VERTEX_NODE);
+            if (node instanceof DistrictNetworkVertexNode) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public AffineTransform getTransform() {
+        return transform;
+    }
+}