Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.g2d / src / org / simantics / g2d / utils / geom / DirectionSet.java

diff --git a/bundles/org.simantics.g2d/src/org/simantics/g2d/utils/geom/DirectionSet.java b/bundles/org.simantics.g2d/src/org/simantics/g2d/utils/geom/DirectionSet.java
new file mode 100644 (file)
index 0000000..e1475c4
--- /dev/null
+++ b/bundles/org.simantics.g2d/src/org/simantics/g2d/utils/geom/DirectionSet.java
@@ -0,0 +1,153 @@
+/*******************************************************************************\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.utils.geom;\r
+\r
+import java.awt.geom.Point2D;\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.HashSet;\r
+import java.util.Set;\r
+\r
+import org.simantics.g2d.utils.GeometryUtils;\r
+\r
+/**\r
+ * A set of compass [0..360) directions\r
+ */\r
+public class DirectionSet extends HashSet<Double> {\r
+       \r
+       public static final DirectionSet NESW = new DirectionSet(0.0, 90.0, 180.0, 270.0);\r
+       \r
+       public static final DirectionSet N      = new DirectionSet(0.0);\r
+       public static final DirectionSet NE     = new DirectionSet(0.0, 90.0);\r
+       public static final DirectionSet E      = new DirectionSet(90.0);\r
+       public static final DirectionSet SE     = new DirectionSet(90.0,180.0);\r
+       public static final DirectionSet S      = new DirectionSet(180.0);\r
+       public static final DirectionSet SW     = new DirectionSet(180.0, 270.0);\r
+       public static final DirectionSet W      = new DirectionSet(270.0);\r
+       public static final DirectionSet NW     = new DirectionSet(270.0, 0.0);\r
+\r
+       public static final DirectionSet NE2    = new DirectionSet(0.0, 90.0, 45.0);\r
+       public static final DirectionSet SE2    = new DirectionSet(90.0,180.0, 135.0);\r
+       public static final DirectionSet SW2    = new DirectionSet(180.0, 270.0, 225.0);\r
+       public static final DirectionSet NW2    = new DirectionSet(270.0, 0.0, 315.0);\r
+       \r
+       public static final DirectionSet HORIZ = new DirectionSet(90.0, 270.0);\r
+       public static final DirectionSet VERT = new DirectionSet(0.0, 180.0);\r
+       public static final DirectionSet ANY = \r
+               new DirectionSet(\r
+                               0, 15, 30, 45, 60, 75, 90, 105, 120, 135, 150, 165, 180,\r
+                               195, 210, 225, 240, 255, 270, 285, 300, 315, 330, 345);                         \r
+\r
+       private Set<Point2D> unitVectors;\r
+       \r
+       public DirectionSet(double ... directions)\r
+       {\r
+               for (double d : directions)\r
+                       add(d);\r
+       }\r
+       \r
+       private static final long serialVersionUID = 1L;\r
+       private boolean add(double e) {\r
+               if (e<0 || e>=360.0) e = Math.IEEEremainder(e, 360.0);\r
+               if (e<0) e = 360.0-e;\r
+               return super.add(e);\r
+       }\r
+       @Override\r
+       public boolean add(Double e) {\r
+               unitVectors = null;\r
+               if (e==null) return false;\r
+               return add((double)e);\r
+       }\r
+       @Override\r
+       public boolean addAll(Collection<? extends Double> c) {\r
+               unitVectors = null;\r
+               return super.addAll(c);\r
+       }\r
+       @Override\r
+       public boolean remove(Object o) {\r
+               unitVectors = null;\r
+               return super.remove(o);\r
+       }\r
+       @Override\r
+       public boolean removeAll(Collection<?> c) {\r
+               unitVectors = null;\r
+               return super.removeAll(c);\r
+       }\r
+       @Override\r
+       public void clear() {\r
+               unitVectors = null;\r
+               super.clear();\r
+       }\r
+       \r
+       public Double getClosestDirection(double d2)\r
+       {\r
+               Double closest = null;\r
+               for (Double d1 : this)\r
+               {\r
+                       double diff = d1>180==d2>180 ? Math.abs(d1-d2) : d1<d2 ? d1-d2+360 : d2-d1+360;\r
+                       if (closest==null || closest<diff)\r
+                               closest = diff;\r
+               }\r
+               return closest;\r
+       }\r
+       public Double getClosestDirection(double d2, double tolerance)\r
+       {\r
+               Double closest = null;\r
+               for (Double d1 : this)\r
+               {\r
+                       double diff = d1>180==d2>180 ? Math.abs(d1-d2) : d1<d2 ? d1-d2+360 : d2-d1+360;\r
+                       if (diff>tolerance) continue;\r
+                       if (closest==null || closest<diff)\r
+                               closest = diff;\r
+               }\r
+               return closest;\r
+       }       \r
+       \r
+       /**\r
+        * Create direction set of inverse directions\r
+        * @return\r
+        */\r
+       public DirectionSet createInverse() \r
+       {\r
+               DirectionSet result = new DirectionSet();\r
+               for (double dir : this)\r
+               {\r
+                       dir = dir + 180.0;\r
+                       if (dir>360.0) dir-=360.0;\r
+                       result.add(dir);\r
+               }\r
+               return result;\r
+       }\r
+       \r
+       \r
+       /**\r
+        * Get directions as unit vectors\r
+        * @return collection where to add unit vectors\r
+        */\r
+       private Set<Point2D> createUnitVectors() {\r
+               HashSet<Point2D> result = new HashSet<Point2D>(size());\r
+               for (double d : this)\r
+                       result.add( GeometryUtils.toUnitVector(d, new Point2D.Double()) );\r
+               return result;\r
+       }\r
+\r
+       public Set<Point2D> getUnitVectors() {\r
+               if (unitVectors==null) {\r
+                       unitVectors = createUnitVectors();\r
+                       unitVectors = Collections.unmodifiableSet(unitVectors);\r
+               }\r
+               return unitVectors;\r
+       }\r
+\r
+       \r
+       \r
+}
\ No newline at end of file