--- /dev/null
+/*******************************************************************************\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