]> gerrit.simantics Code Review - simantics/platform.git/blobdiff - bundles/org.simantics.g2d/src/org/simantics/g2d/dnd/DragPainter.java
Migrated source code from Simantics SVN
[simantics/platform.git] / bundles / org.simantics.g2d / src / org / simantics / g2d / dnd / DragPainter.java
diff --git a/bundles/org.simantics.g2d/src/org/simantics/g2d/dnd/DragPainter.java b/bundles/org.simantics.g2d/src/org/simantics/g2d/dnd/DragPainter.java
new file mode 100644 (file)
index 0000000..208d4d0
--- /dev/null
@@ -0,0 +1,217 @@
+/*******************************************************************************\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.dnd;\r
+\r
+import java.awt.AlphaComposite;\r
+import java.awt.geom.AffineTransform;\r
+import java.awt.geom.Point2D;\r
+import java.awt.geom.Rectangle2D;\r
+\r
+import org.simantics.g2d.canvas.ICanvasContext;\r
+import org.simantics.g2d.canvas.impl.AbstractCanvasParticipant;\r
+import org.simantics.g2d.canvas.impl.SGNodeReflection.SGCleanup;\r
+import org.simantics.g2d.canvas.impl.SGNodeReflection.SGInit;\r
+import org.simantics.g2d.diagram.DiagramHints;\r
+import org.simantics.g2d.participant.TransformUtil;\r
+import org.simantics.scenegraph.g2d.G2DParentNode;\r
+import org.simantics.scenegraph.g2d.IG2DNode;\r
+import org.simantics.scenegraph.g2d.events.EventHandlerReflection.EventHandler;\r
+import org.simantics.scenegraph.g2d.events.command.CommandEvent;\r
+import org.simantics.scenegraph.g2d.events.command.Commands;\r
+import org.simantics.scenegraph.g2d.nodes.SingleElementNode;\r
+import org.simantics.scenegraph.g2d.snap.ISnapAdvisor;\r
+import org.simantics.utils.datastructures.context.IContext;\r
+import org.simantics.utils.datastructures.context.IContextListener;\r
+\r
+/**\r
+ * This participant paints an array of icons of a drag operation.\r
+ *\r
+ * @author Toni Kalajainen\r
+ */\r
+public class DragPainter extends AbstractCanvasParticipant {\r
+    // Priority of drag event, eats ecs key\r
+    public final static int DRAG_EVENT_PRIORITY = Integer.MAX_VALUE - 1500;\r
+    public final static int DRAG_PAINT_PRIORITY = Integer.MAX_VALUE - 1100;\r
+    public final static int MARGIN = 5;\r
+\r
+    final IDnDContext       dropCtx;\r
+    Point2D                 mouseControlPos;\r
+\r
+    public DragPainter(IDnDContext dropCtx, Point2D mouseControlPos) {\r
+        super();\r
+        this.dropCtx = dropCtx;\r
+        this.mouseControlPos = (Point2D) mouseControlPos.clone();\r
+    }\r
+\r
+    IContextListener<IDragItem> contextListener = new IContextListener<IDragItem>() {\r
+        @Override\r
+        public void itemAdded(IContext<IDragItem> sender, IDragItem item) {\r
+            updateItemPositions();\r
+            getContext().getContentContext().setDirty();\r
+        }\r
+        @Override\r
+        public void itemRemoved(IContext<IDragItem> sender, IDragItem item) {\r
+            updateItemPositions();\r
+            getContext().getContentContext().setDirty();\r
+        }\r
+    };\r
+\r
+    @Override\r
+    public void addedToContext(ICanvasContext ctx) {\r
+        super.addedToContext(ctx);\r
+        dropCtx.addContextListener(getContext().getThreadAccess(), contextListener);\r
+    }\r
+\r
+    @Override\r
+    public void removedFromContext(ICanvasContext ctx) {\r
+        dropCtx.removeContextListener(getContext().getThreadAccess(), contextListener);\r
+        super.removedFromContext(ctx);\r
+    }\r
+\r
+    public boolean setMousePos(Point2D mouseControlPos) {\r
+        return setMousePos(mouseControlPos, false);\r
+    }\r
+\r
+    public boolean setMousePos(Point2D mouseControlPos, boolean forceUpdate) {\r
+        if (!forceUpdate && mouseControlPos.equals(this.mouseControlPos))\r
+            return false;\r
+        this.mouseControlPos = (Point2D) mouseControlPos.clone();\r
+        getContext().getContentContext().setDirty();\r
+        updateItemPositions();\r
+        return true;\r
+    }\r
+\r
+    /**\r
+     * Get the size of the largest drop item\r
+     * @param items\r
+     * @return\r
+     */\r
+    private Point2D calcItemSize(IDragItem[] items)\r
+    {\r
+        double w = Double.MIN_VALUE;\r
+        double h = Double.MIN_VALUE;\r
+        for (IDragItem i : items) {\r
+            Rectangle2D bounds = i.getBounds();\r
+            if (bounds == null) bounds = new Rectangle2D.Double(0,0,1,1);\r
+            if (w < bounds.getWidth())\r
+                w = bounds.getWidth();\r
+            if (h < bounds.getHeight())\r
+                h = bounds.getHeight();\r
+        }\r
+        return new Point2D.Double(w, h);\r
+    }\r
+\r
+    private void updateItemPositions() {\r
+        // Calculate mouse position on diagram\r
+        TransformUtil vi = getContext().getSingleItem(TransformUtil.class);\r
+        AffineTransform controlToDiagram = vi.getInverseTransform();\r
+        if (controlToDiagram==null) return;\r
+        Point2D mouseDiagramPos = controlToDiagram.transform(mouseControlPos, new Point2D.Double());\r
+\r
+        //System.out.println("updateItemPositions: mousepos: " + mouseControlPos + " - " + mouseDiagramPos);\r
+\r
+        IDragItem items[] = dropCtx.toArray();\r
+        int count = items.length;\r
+        int columns = (int) Math.ceil( Math.sqrt( count) );\r
+        Integer columnsHint = dropCtx.getHints().getHint(DnDHints.KEY_DND_GRID_COLUMNS);\r
+        if (columnsHint != null)\r
+            columns = columnsHint;\r
+        int rows = (int) Math.ceil((double)count / (double)columns);\r
+        int margin = MARGIN;\r
+        // Size of the largest item\r
+        Point2D size = calcItemSize(items);\r
+\r
+        // Calculate center point of the whole item mass\r
+        double cx = ((columns-1)*(size.getX()+margin))/2 + size.getX()/2;\r
+        double cy = ((rows   -1)*(size.getY()+margin))/2 + size.getY()/2;\r
+        //double cx = ((columns)*(size.getX()+margin)-margin)/2 /*- size.getX()/2*/;\r
+        //double cy = ((rows   )*(size.getY()+margin)-margin)/2 /*- size.getY()/2*/;\r
+        //double cx = 0;\r
+        //double cy = 0;\r
+\r
+        int index = 0;\r
+        for (IDragItem i : items) {\r
+            Rectangle2D bounds = i.getBounds();\r
+            double x = (index % columns) * (margin + size.getX());\r
+            double y = (index / columns) * (margin + size.getY());\r
+//            Point2D pos = new Point2D.Double(x, y);\r
+            index++;\r
+\r
+            x -= bounds.getX();\r
+            y -= bounds.getY();\r
+            x += mouseDiagramPos.getX();\r
+            y += mouseDiagramPos.getY();\r
+\r
+            x -= cx;\r
+            y -= cy;\r
+\r
+            Point2D p = new Point2D.Double(x, y);\r
+            ISnapAdvisor snap = getHint(DiagramHints.SNAP_ADVISOR);\r
+            if (snap != null)\r
+                snap.snap(p);\r
+            dropCtx.setItemPosition(i, p);\r
+        }\r
+        updateSG();\r
+    }\r
+\r
+    protected G2DParentNode node = null;\r
+\r
+    @SGInit\r
+    public void initSG(G2DParentNode parent) {\r
+        node = parent.addNode(G2DParentNode.class);\r
+        node.setZIndex(DRAG_PAINT_PRIORITY);\r
+        updateSG();\r
+    }\r
+\r
+    @SGCleanup\r
+    public void cleanupSG() {\r
+        node.remove();\r
+        node = null;\r
+    }\r
+\r
+    private void updateSG() {\r
+        IDragItem items[] = dropCtx.toArray();\r
+//        updateItemPositions();\r
+        for (IDragItem item : items) {\r
+            Point2D pos = dropCtx.getItemPosition(item);\r
+            if(pos != null) { // Position can (or at least seems to be) be null on the first frame\r
+                AffineTransform subt = AffineTransform.getTranslateInstance(pos.getX(), pos.getY());\r
+\r
+                IG2DNode itemHolder = node.getNode(""+item.hashCode());\r
+                if (itemHolder != null && !(itemHolder instanceof SingleElementNode)) {\r
+                    System.err.println("BUG: item hash codes collide within the dragged item context - found unrecognized item " + itemHolder + " with node id '" + item.hashCode() + "'");\r
+                    continue;\r
+                }\r
+\r
+                SingleElementNode elementHolder = (SingleElementNode) itemHolder;\r
+                if (elementHolder == null) {\r
+                    elementHolder = node.getOrCreateNode(""+item.hashCode(), SingleElementNode.class);\r
+                    elementHolder.setTransform(subt);\r
+                    elementHolder.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.75f));\r
+                    item.paint(elementHolder);\r
+                } else {\r
+                    elementHolder.setTransform(subt);\r
+                }\r
+            }\r
+        }\r
+    }\r
+\r
+    @EventHandler(priority = DRAG_EVENT_PRIORITY)\r
+    public boolean handleEvent(CommandEvent e) {\r
+        if (e.command.equals( Commands.CANCEL) )\r
+        {\r
+            remove();\r
+            return true;\r
+        }\r
+        return false;\r
+    }\r
+}\r