]> gerrit.simantics Code Review - simantics/platform.git/blobdiff - bundles/org.simantics.g2d/src/org/simantics/g2d/diagram/participant/TerminalPainter.java
Migrated source code from Simantics SVN
[simantics/platform.git] / bundles / org.simantics.g2d / src / org / simantics / g2d / diagram / participant / TerminalPainter.java
diff --git a/bundles/org.simantics.g2d/src/org/simantics/g2d/diagram/participant/TerminalPainter.java b/bundles/org.simantics.g2d/src/org/simantics/g2d/diagram/participant/TerminalPainter.java
new file mode 100644 (file)
index 0000000..54e4a8e
--- /dev/null
@@ -0,0 +1,304 @@
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.g2d.diagram.participant;\r
+\r
+import java.awt.BasicStroke;\r
+import java.awt.Color;\r
+import java.awt.Shape;\r
+import java.awt.Stroke;\r
+import java.awt.geom.AffineTransform;\r
+import java.awt.geom.Path2D;\r
+import java.awt.geom.Rectangle2D;\r
+import java.util.Collection;\r
+import java.util.List;\r
+\r
+import org.simantics.g2d.canvas.ICanvasContext;\r
+import org.simantics.g2d.canvas.impl.DependencyReflection.Dependency;\r
+import org.simantics.g2d.canvas.impl.SGNodeReflection.SGCleanup;\r
+import org.simantics.g2d.canvas.impl.SGNodeReflection.SGInit;\r
+import org.simantics.g2d.diagram.IDiagram;\r
+import org.simantics.g2d.diagram.participant.pointertool.PointerInteractor;\r
+import org.simantics.g2d.diagram.participant.pointertool.TerminalUtil;\r
+import org.simantics.g2d.diagram.participant.pointertool.TerminalUtil.TerminalInfo;\r
+import org.simantics.g2d.participant.MouseUtil;\r
+import org.simantics.g2d.participant.TransformUtil;\r
+import org.simantics.g2d.participant.MouseUtil.MouseInfo;\r
+import org.simantics.g2d.utils.GeometryUtils;\r
+import org.simantics.scenegraph.g2d.G2DParentNode;\r
+import org.simantics.scenegraph.g2d.events.EventHandlerReflection.EventHandler;\r
+import org.simantics.scenegraph.g2d.events.MouseEvent.MouseMovedEvent;\r
+import org.simantics.scenegraph.g2d.nodes.ShapeNode;\r
+import org.simantics.scenegraph.utils.ColorUtil;\r
+import org.simantics.utils.datastructures.hints.HintListenerAdapter;\r
+import org.simantics.utils.datastructures.hints.IHintListener;\r
+import org.simantics.utils.datastructures.hints.IHintObservable;\r
+import org.simantics.utils.datastructures.hints.IHintContext.Key;\r
+import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;\r
+\r
+/**\r
+ * Paints terminals of elements.\r
+ *\r
+ * @author Toni Kalajainen\r
+ */\r
+public class TerminalPainter extends AbstractDiagramParticipant {\r
+\r
+    public static final int PAINT_PRIORITY = ElementPainter.ELEMENT_PAINT_PRIORITY + 10;\r
+\r
+    public interface TerminalHoverStrategy {\r
+        /**\r
+         * \r
+         * @return <code>true</code> if highlighting is enabled at the moment in\r
+         *         general. This may depend on the current modifier key state\r
+         *         for example.\r
+         */\r
+        boolean highlightEnabled();\r
+\r
+        /**\r
+         * Checks whether the specified terminal should be highlighted at the\r
+         * moment or not. Whether to highlight or not may depend for example on\r
+         * the current modifier key state.\r
+         * \r
+         * @param ti\r
+         * @return\r
+         */\r
+        boolean highlight(TerminalInfo ti);\r
+    };\r
+\r
+    public static abstract class ChainedHoverStrategy implements TerminalHoverStrategy {\r
+        TerminalHoverStrategy orig;\r
+        public ChainedHoverStrategy(TerminalHoverStrategy orig) {\r
+            this.orig = orig;\r
+        }\r
+        @Override\r
+        public boolean highlightEnabled() {\r
+            return orig == null ? false : orig.highlightEnabled();\r
+        }\r
+        @Override\r
+        public final boolean highlight(TerminalInfo ti) {\r
+            boolean ret = canHighlight(ti);\r
+            return (ret || orig == null) ? ret : orig.highlight(ti);\r
+        }\r
+        public abstract boolean canHighlight(TerminalInfo ti);\r
+    }\r
+\r
+    /**\r
+     * If this hint is set to a Callable<Boolean>, the terminal painter will use\r
+     * the callable to evaluate whether it should highlight terminal hovers or\r
+     * not.\r
+     */\r
+    public static final Key          TERMINAL_HOVER_STRATEGY = new KeyOf(TerminalHoverStrategy.class);\r
+\r
+//    private static final Stroke      STROKE1                 = new BasicStroke(1.0f);\r
+//    private static final Stroke      STROKE15                = new BasicStroke(1.5f);\r
+    private static final Stroke      STROKE25                = new BasicStroke(2.5f);\r
+\r
+//    private final static Stroke      TERMINAL_STROKE         = new BasicStroke(1.0f);\r
+    public static final Shape        TERMINAL_SHAPE;\r
+\r
+    @Dependency\r
+       protected TransformUtil util;\r
+    @Dependency\r
+       protected MouseUtil mice;\r
+    \r
+    @Dependency\r
+    protected PointerInteractor pointerInteractor;\r
+\r
+    protected boolean paintPointTerminals;\r
+    protected boolean paintAreaTerminals;\r
+    protected boolean paintHoverPointTerminals;\r
+    protected boolean paintHoverAreaTerminals;\r
+\r
+    public TerminalPainter(boolean paintPointTerminals, boolean paintHoverPointTerminals, boolean paintAreaTerminals, boolean paintHoverAreaTerminals)\r
+    {\r
+        this.paintAreaTerminals = paintAreaTerminals;\r
+        this.paintPointTerminals = paintPointTerminals;\r
+        this.paintHoverAreaTerminals = paintHoverAreaTerminals;\r
+        this.paintHoverPointTerminals = paintHoverPointTerminals;\r
+    }\r
+\r
+    @Override\r
+    public void addedToContext(ICanvasContext ctx) {\r
+        super.addedToContext(ctx);\r
+\r
+        ctx.getHintStack().addKeyHintListener(getContext().getThreadAccess(), TERMINAL_HOVER_STRATEGY, hoverStrategyListener);\r
+    }\r
+\r
+    @Override\r
+    public void removedFromContext(ICanvasContext ctx) {\r
+        ctx.getHintStack().removeKeyHintListener(getContext().getThreadAccess(), TERMINAL_HOVER_STRATEGY, hoverStrategyListener);\r
+\r
+        super.removedFromContext(ctx);\r
+    }\r
+\r
+    public boolean highlightEnabled() {\r
+        TerminalHoverStrategy strategy = getHint(TERMINAL_HOVER_STRATEGY);\r
+        return strategy != null ? strategy.highlightEnabled() : true;\r
+    }\r
+\r
+    public boolean highlightTerminal(TerminalInfo ti) {\r
+        TerminalHoverStrategy strategy = getHint(TERMINAL_HOVER_STRATEGY);\r
+        return strategy != null ? strategy.highlight(ti) : true;\r
+    }\r
+\r
+    @EventHandler(priority = 0)\r
+    public boolean handleMove(MouseMovedEvent me) {\r
+        if ( (paintHoverAreaTerminals && paintAreaTerminals) ||\r
+                (paintHoverPointTerminals && paintPointTerminals) ) {\r
+            update(highlightEnabled());\r
+        }\r
+        return false;\r
+    }\r
+\r
+    protected G2DParentNode node = null;\r
+\r
+    @SGInit\r
+    public void initSG(G2DParentNode parent) {\r
+        node = parent.addNode("hovering terminals", G2DParentNode.class);\r
+        node.setZIndex(PAINT_PRIORITY);\r
+    }\r
+\r
+    @SGCleanup\r
+    public void cleanupSG() {\r
+        node.remove();\r
+        node = null;\r
+    }\r
+\r
+    public void update(boolean enabled) {\r
+        if (isRemoved())\r
+            return;\r
+\r
+        boolean repaint = false;\r
+        if(node == null) return;\r
+        if(node.getNodeCount() > 0) {\r
+            node.removeNodes();\r
+            repaint = true;\r
+        }\r
+        if (enabled) {\r
+\r
+            // Paint terminals normally\r
+            if (paintAreaTerminals || paintPointTerminals) {\r
+                List<TerminalInfo> pickedTerminals = TerminalUtil.pickTerminals(diagram, null, paintPointTerminals, paintAreaTerminals);\r
+                paintTerminals(node, Color.BLUE, diagram, null, pickedTerminals, null);\r
+                if(pickedTerminals.size() > 0) repaint = true;\r
+            }\r
+\r
+            if (paintHoverAreaTerminals || paintHoverPointTerminals) {\r
+                TerminalHoverStrategy strategy = getHint(TERMINAL_HOVER_STRATEGY);\r
+\r
+                AffineTransform invTx = util.getInverseTransform();\r
+                if (invTx == null) {\r
+                    System.err.println("NO CANVAS TRANSFORM INVERSE AVAILABLE, CANVAS TRANSFORM IS: " + util.getTransform());\r
+                    return;\r
+                }\r
+\r
+                // Pick terminals\r
+                for (MouseInfo mi : mice.getMiceInfo().values()) {\r
+                       Rectangle2D controlPickRect = getPickRectangle(mi.controlPosition.getX(), mi.controlPosition.getY());\r
+                    Shape       canvasPickRect  = GeometryUtils.transformShape(controlPickRect, invTx);\r
+\r
+                    List<TerminalInfo> tis = TerminalUtil.pickTerminals(diagram, canvasPickRect, paintHoverAreaTerminals, paintHoverPointTerminals);\r
+                    paintTerminals(node, Color.RED, diagram, canvasPickRect.getBounds2D(), tis, strategy);\r
+                    if(tis.size() > 0) repaint = true;\r
+                }\r
+            }\r
+        }\r
+        if (repaint) {\r
+            setDirty();\r
+        }\r
+    }\r
+\r
+    public void paintTerminals(G2DParentNode parent, Color color, IDiagram diagram, Rectangle2D pickRect, Collection<TerminalInfo> tis, TerminalHoverStrategy strategy) {\r
+        if (tis.isEmpty()) {\r
+            return;\r
+        }\r
+        G2DParentNode node = parent.getOrCreateNode(""+tis.hashCode(), G2DParentNode.class);\r
+\r
+        double minDist = Double.MAX_VALUE;\r
+        double maxDist = 0;\r
+        TerminalInfo nearest = null;\r
+        if (pickRect != null) {\r
+            for (TerminalInfo ti : tis) {\r
+                double dx = ti.posDia.getTranslateX() - pickRect.getCenterX();\r
+                double dy = ti.posDia.getTranslateY() - pickRect.getCenterY();\r
+                double dist = Math.sqrt(dx*dx+dy*dy);\r
+                if (dist > maxDist) {\r
+                    maxDist = dist;\r
+                }\r
+                if (dist < minDist) {\r
+                    minDist = dist;\r
+                    nearest = ti;\r
+                }\r
+            }\r
+        }\r
+\r
+        for (TerminalInfo ti : tis) {\r
+            if (strategy != null && !strategy.highlight(ti))\r
+                continue;\r
+            Shape shape = ti.shape != null ? ti.shape : getTerminalShape();\r
+            //System.out.println("painting terminal " + ti + ": " + shape);\r
+            ShapeNode sn = node.getOrCreateNode("terminal_"+ti.hashCode(), ShapeNode.class);\r
+            sn.setShape(shape);\r
+            sn.setStroke(STROKE25);\r
+            sn.setScaleStroke(true);\r
+            if (pickRect != null) {\r
+                Color blendedColor = color;\r
+                if (ti != nearest) {\r
+                    double dx = ti.posDia.getTranslateX() - pickRect.getCenterX();\r
+                    double dy = ti.posDia.getTranslateY() - pickRect.getCenterY();\r
+                    double dist = Math.sqrt(dx*dx+dy*dy);\r
+                    double normalizedDistance = dist / maxDist;\r
+                    final double maxFade = 0.5;\r
+                    float alpha = (float)(1 - normalizedDistance*maxFade);\r
+                    blendedColor = ColorUtil.withAlpha(ColorUtil.blend(color, Color.WHITE, normalizedDistance*maxFade), alpha);\r
+                }\r
+                sn.setColor(blendedColor);\r
+            } else {\r
+                sn.setColor(color);\r
+            }\r
+            sn.setTransform(ti.posDia);\r
+            sn.setFill(false);\r
+        }\r
+    }\r
+    \r
+    public Rectangle2D getTerminalShape() {\r
+       double pickDist = pointerInteractor.getPickDistance();\r
+       return new Rectangle2D.Double(-pickDist - 0.5, -pickDist - 0.5, pickDist * 2 + 1, pickDist * 2 + 1);\r
+    }\r
+    \r
+    public Rectangle2D getPickRectangle(double x, double y) {\r
+       double pickDist = pointerInteractor.getPickDistance();\r
+        Rectangle2D controlPickRect = new Rectangle2D.Double(x-pickDist, y-pickDist, pickDist*2+1, pickDist*2+1);\r
+        return controlPickRect;\r
+    }\r
+\r
+    static {\r
+        Path2D.Double cross = new Path2D.Double();\r
+        double s = 2;\r
+        cross.moveTo(-s, -s);\r
+        cross.lineTo(s, s);\r
+        cross.moveTo(-s, s);\r
+        cross.lineTo(s, -s);\r
+        TERMINAL_SHAPE = cross;\r
+    }\r
+\r
+    IHintListener hoverStrategyListener = new HintListenerAdapter() {\r
+        @Override\r
+        public void hintChanged(IHintObservable sender, Key key, Object oldValue, Object newValue) {\r
+            hoverStrategyChanged((TerminalHoverStrategy) newValue);\r
+        }\r
+    };\r
+\r
+    protected void hoverStrategyChanged(TerminalHoverStrategy strategy) {\r
+        update(strategy != null ? strategy.highlightEnabled() : false);\r
+    }\r
+\r
+}\r