--- /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.diagram.handler;\r
+\r
+import java.awt.Shape;\r
+import java.awt.geom.AffineTransform;\r
+import java.awt.geom.Area;\r
+import java.awt.geom.Point2D;\r
+import java.awt.geom.Rectangle2D;\r
+import java.util.Collections;\r
+import java.util.Comparator;\r
+import java.util.List;\r
+\r
+import org.simantics.g2d.connection.handler.ConnectionHandler;\r
+import org.simantics.g2d.element.IElement;\r
+import org.simantics.g2d.element.handler.BendsHandler;\r
+import org.simantics.g2d.element.handler.TerminalTopology;\r
+import org.simantics.g2d.element.handler.impl.ConnectionSelectionOutline;\r
+import org.simantics.g2d.elementclass.BranchPoint;\r
+import org.simantics.g2d.elementclass.MonitorHandler;\r
+import org.simantics.g2d.utils.GeometryUtils;\r
+import org.simantics.scenegraph.utils.TransformedRectangle;\r
+\r
+/**\r
+ *\r
+ * @See {@link GeometryUtils} for intersect and contains tests\r
+ * @See {@link TransformedRectangle}\r
+ * @See {@link Area} intersects operations for complex shapes\r
+ * @author Toni Kalajainen\r
+ */\r
+public class PickRequest {\r
+\r
+ public static enum PickPolicy {\r
+ PICK_INTERSECTING_OBJECTS,\r
+ PICK_CONTAINED_OBJECTS\r
+ }\r
+\r
+ public Shape pickArea;\r
+ public PickPolicy pickPolicy = PickPolicy.PICK_INTERSECTING_OBJECTS;\r
+ /** Pick filter (null == everything is accepted) */\r
+ public PickFilter pickFilter = null;\r
+ public PickSorter pickSorter = null;\r
+\r
+ public PickRequest(double x, double y)\r
+ {\r
+ pickArea = new Rectangle2D.Double(x, y, 1, 1);\r
+ }\r
+ public PickRequest(Point2D p)\r
+ {\r
+ pickArea = new Rectangle2D.Double(p.getX(), p.getY(), 0.0001, 0.0001);\r
+ }\r
+ public PickRequest(Shape shape)\r
+ {\r
+ pickArea = shape;\r
+ }\r
+ public PickRequest(Shape shape, AffineTransform transform)\r
+ {\r
+ pickArea = GeometryUtils.transformShape(shape, transform);\r
+ }\r
+\r
+ public static interface PickFilter {\r
+ boolean accept(IElement e);\r
+\r
+ public static final PickFilter FILTER_ALL = new PickFilter() {\r
+ @Override\r
+ public boolean accept(IElement e) {\r
+ return true;\r
+ }\r
+ };\r
+ // Connections\r
+ public static final PickFilter FILTER_CONNECTIONS = new PickFilter() {\r
+ @Override\r
+ public boolean accept(IElement e) {\r
+ return e.getElementClass().containsClass(ConnectionHandler.class);\r
+ }\r
+ @Override\r
+ public String toString() { return "FILTER_CONNECTIONS"; }\r
+ };\r
+ // Connection edges\r
+ public static final PickFilter FILTER_CONNECTION_EDGES = new PickFilter() {\r
+ @Override\r
+ public boolean accept(IElement e) {\r
+ return e.getElementClass().containsClass(BendsHandler.class) || e.getElementClass().containsClass(ConnectionSelectionOutline.class);\r
+ }\r
+ @Override\r
+ public String toString() { return "FILTER_CONNECTION_EDGES"; }\r
+ };\r
+ // Connections\r
+ public static final PickFilter FILTER_BRANCH_POINT = new PickFilter() {\r
+ @Override\r
+ public boolean accept(IElement e) {\r
+ return e.getElementClass().containsClass(BranchPoint.class);\r
+ }\r
+ @Override\r
+ public String toString() { return "FILTER_BRANCH_POINTS"; }\r
+ };\r
+ // Anything that has terminals\r
+ public static final PickFilter FILTER_NODES = new PickFilter() {\r
+ @Override\r
+ public boolean accept(IElement e) {\r
+ return e.getElementClass().containsClass(TerminalTopology.class);\r
+ }\r
+ @Override\r
+ public String toString() { return "FILTER_NODES"; }\r
+ };\r
+ public static final PickFilter FILTER_MONITORS = new PickFilter() {\r
+ @Override\r
+ public boolean accept(IElement e) {\r
+ return e.getElementClass().containsClass(MonitorHandler.class);\r
+ }\r
+ @Override\r
+ public String toString() { return "FILTER_MONITORS"; }\r
+ };\r
+ }\r
+\r
+ public static interface PickSorter {\r
+ void sort(List<IElement> elements);\r
+\r
+ //\r
+ public static final PickSorter CONNECTIONS_LAST = new PickSorter() {\r
+ @Override\r
+ public void sort(List<IElement> elements) {\r
+ Collections.sort(elements, new Comparator<IElement>() {\r
+ @Override\r
+ public int compare(IElement e1, IElement e2) {\r
+ boolean isConn1 = PickFilter.FILTER_CONNECTION_EDGES.accept(e1);\r
+ boolean isConn2 = PickFilter.FILTER_CONNECTION_EDGES.accept(e2);\r
+ if (!isConn1 && isConn2)\r
+ return -1;\r
+ if (isConn1 && !isConn2)\r
+ return 1;\r
+ return 0;\r
+ }\r
+ });\r
+ }\r
+ };\r
+ public static final PickSorter CONNECTIONS_FIRST = new PickSorter() {\r
+ @Override\r
+ public void sort(List<IElement> elements) {\r
+ Collections.sort(elements, new Comparator<IElement>() {\r
+ @Override\r
+ public int compare(IElement e1, IElement e2) {\r
+ boolean isConn1 = PickFilter.FILTER_CONNECTION_EDGES.accept(e1);\r
+ boolean isConn2 = PickFilter.FILTER_CONNECTION_EDGES.accept(e2);\r
+ if (!isConn1 && isConn2)\r
+ return 1;\r
+ if (isConn1 && !isConn2)\r
+ return -1;\r
+ return 0;\r
+ }\r
+ });\r
+ }\r
+ };\r
+ }\r
+\r
+}\r