Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.debug.graphical / src / org / simantics / debug / graphical / layout / ExtensionLayoutAlgorithm.java

diff --git a/bundles/org.simantics.debug.graphical/src/org/simantics/debug/graphical/layout/ExtensionLayoutAlgorithm.java b/bundles/org.simantics.debug.graphical/src/org/simantics/debug/graphical/layout/ExtensionLayoutAlgorithm.java
new file mode 100644 (file)
index 0000000..9865e28
--- /dev/null
+++ b/bundles/org.simantics.debug.graphical/src/org/simantics/debug/graphical/layout/ExtensionLayoutAlgorithm.java
@@ -0,0 +1,125 @@
+package org.simantics.debug.graphical.layout;\r
+\r
+public class ExtensionLayoutAlgorithm {\r
+    final double OPTIMAL_DISTANCE = 150.0;   \r
+    final double C = 0.1;\r
+    final double K = C / OPTIMAL_DISTANCE;\r
+    final double G = K;\r
+    final double Q = C * OPTIMAL_DISTANCE * OPTIMAL_DISTANCE;\r
+    final double TOLERANCE = 1e-3;\r
+    final int MAX_ITERATIONS = 10000;\r
+    final long TIMEOUT = 1000000000L; // 1 sec\r
+    final double MAX_MOVE2 = 40000.0;\r
+    \r
+    int size;\r
+    double[] posX;\r
+    double[] posY;\r
+    double[] forceX;\r
+    double[] forceY;\r
+    \r
+    int repulsiveSize;\r
+    double[] fixedRepulsiveX;\r
+    double[] fixedRepulsiveY;\r
+    double[][] neighbors;\r
+    \r
+    public ExtensionLayoutAlgorithm(double[][] neighbors, \r
+            double[] fixedRepulsiveX, double[] fixedRepulsiveY) {\r
+        this.size = neighbors.length;\r
+        this.neighbors = neighbors;\r
+        this.repulsiveSize = fixedRepulsiveX.length;\r
+        this.fixedRepulsiveX = fixedRepulsiveX;\r
+        this.fixedRepulsiveY = fixedRepulsiveY;\r
+        this.posX = new double[size];\r
+        this.posY = new double[size];\r
+        this.forceX = new double[size];\r
+        this.forceY = new double[size];\r
+    }\r
+    \r
+    public double[] getPosX() {\r
+        return posX;\r
+    }\r
+    \r
+    public double[] getPosY() {\r
+        return posY;\r
+    }\r
+    \r
+    private void computeForces() {\r
+        for(int i=0;i<size;++i) {\r
+            double x = posX[i];\r
+            double y = posY[i];\r
+            double fx = -G * x;\r
+            double fy = -G * y;\r
+            \r
+            for(int j=0;j<size;++j)\r
+                if(j != i) {\r
+                    double dx = posX[j] - x;\r
+                    double dy = posY[j] - y;\r
+                    double l2 = dx*dx + dy*dy;\r
+                    fx -= Q * dx / l2;\r
+                    fy -= Q * dy / l2;\r
+                }\r
+            for(int j=0;j<repulsiveSize;++j)\r
+                if(j != i) {\r
+                    double dx = fixedRepulsiveX[j] - x;\r
+                    double dy = fixedRepulsiveY[j] - y;\r
+                    double l2 = dx*dx + dy*dy;\r
+                    fx -= Q * dx / l2;\r
+                    fy -= Q * dy / l2;\r
+                }\r
+            \r
+            double[] ns = neighbors[i];\r
+            for(int j=0;j<ns.length;j+=2) {\r
+                double dx = ns[j] - x;\r
+                double dy = ns[j+1] - y;       \r
+                double l2 = dx*dx + dy*dy;\r
+                double l = Math.sqrt(l2);\r
+                fx += K * l * dx;\r
+                fy += K * l * dy;\r
+            }\r
+            \r
+            forceX[i] = fx;\r
+            forceY[i] = fy;\r
+        }\r
+    }\r
+    \r
+    private boolean converged() {\r
+        for(int i=0;i<size;++i) {\r
+            double fx = forceX[i];\r
+            double fy = forceY[i];\r
+            double f2 = fx*fx + fy*fy;\r
+            if(f2 > TOLERANCE)\r
+                return false;\r
+        }\r
+        return true;\r
+    }\r
+    \r
+    private void moveAll() {\r
+        for(int i=0;i<size;++i) {\r
+            double fx = forceX[i];\r
+            double fy = forceY[i];\r
+            double f2 = fx*fx + fy*fy;\r
+            if(f2 > MAX_MOVE2) {\r
+                double s = Math.sqrt(MAX_MOVE2 / f2);\r
+                fx *= s;\r
+                fy *= s;\r
+            }\r
+            posX[i] += fx;\r
+            posY[i] += fy;\r
+        }\r
+        computeForces();\r
+    }\r
+    \r
+    public void optimize() {\r
+        computeForces();\r
+\r
+        long beginTime = System.nanoTime();\r
+        int iter;\r
+        for(iter=0;iter<MAX_ITERATIONS;++iter) {\r
+            if(converged() || ((iter&0xf)==0 && System.nanoTime()-beginTime > TIMEOUT))\r
+                break;            \r
+            moveAll();\r
+        }\r
+        System.out.println("Elapsed: " + (System.nanoTime()-beginTime)*1e-6 + " ms");\r
+        System.out.println("Iterations: " + iter);\r
+    }\r
+}\r