org.simantics.district.network.ui/src/org/simantics/district/network/ui/NetworkDrawingParticipant.java

   1
   2 package org.simantics.district.network.ui;
   3
   4 import java.awt.geom.Point2D;
   5 import java.util.ArrayList;
   6 import java.util.List;
   7
   8 import org.simantics.district.network.ui.adapters.DistrictNetworkVertexElement;
   9 import org.simantics.district.network.ui.nodes.DistrictNetworkVertexNode;
  10 import org.simantics.district.network.ui.nodes.NetworkDrawingNode;
  11 import org.simantics.g2d.canvas.impl.DependencyReflection.Dependency;
  12 import org.simantics.g2d.canvas.impl.SGNodeReflection.SGInit;
  13 import org.simantics.g2d.diagram.IDiagram;
  14 import org.simantics.g2d.diagram.handler.PickContext;
  15 import org.simantics.g2d.diagram.handler.PickRequest;
  16 import org.simantics.g2d.diagram.participant.AbstractDiagramParticipant;
  17 import org.simantics.g2d.element.IElement;
  18 import org.simantics.scenegraph.g2d.G2DNode;
  19 import org.simantics.scenegraph.g2d.G2DParentNode;
  20 import org.simantics.utils.datastructures.hints.IHintContext.Key;
  21 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
  22
  23 public class NetworkDrawingParticipant extends AbstractDiagramParticipant {
  24
  25     @Dependency
  26     PickContext pick;
  27
  28     /**
  29      * A hint key for terminal pick distance in control pixels.
  30      * @see #PICK_DIST
  31      */
  32     public static final Key KEY_PICK_DISTANCE = new KeyOf(Double.class, "PICK_DISTANCE");
  33
  34     /**
  35      * Default terminal pick distance in control pixels.
  36      * @see #DEFAULT_PICK_DISTANCE
  37      */
  38     public static final double PICK_DIST = 10;
  39
  40     private NetworkDrawingNode node;
  41
  42     @SGInit
  43     public void initSG(G2DParentNode parent) {
  44         node = parent.addNode("networkDrawingNode", NetworkDrawingNode.class);
  45         node.setNetworkDrawingParticipant(this);
  46     }
  47
  48     @Override
  49     protected void onDiagramSet(IDiagram newDiagram, IDiagram oldDiagram) {
  50         node.setDiagram(newDiagram);
  51     }
  52
  53     public boolean pickHoveredElement(Point2D currentMousePos) {
  54         PickRequest req = new PickRequest(currentMousePos.getX(), currentMousePos.getY());
  55         List<IElement> pickables = new ArrayList<IElement>();
  56         pick.pick(diagram, req, pickables);
  57
  58         List<IElement> snap = new ArrayList<>(diagram.getSnapshot());
  59
  60         snap.removeAll(pickables);
  61
  62         boolean changed = false;
  63         for (IElement sn : snap) {
  64             G2DNode node = sn.getHint(DistrictNetworkVertexElement.KEY_DN_VERTEX_NODE);
  65             if (node instanceof DistrictNetworkVertexNode) {
  66                 if (((DistrictNetworkVertexNode) node).hover(false) && !changed) {
  67                     changed = true;
  68                 }
  69             }
  70         }
  71
  72         for (IElement elem : pickables) {
  73             G2DNode node = elem.getHint(DistrictNetworkVertexElement.KEY_DN_VERTEX_NODE);
  74             if (node instanceof DistrictNetworkVertexNode) {
  75                 if (((DistrictNetworkVertexNode) node).hover(true) && !changed) {
  76                     changed = true;
  77                 }
  78             }
  79         }
  80         return changed;
  81     }
  82
  83 }