1 files changed, 116 insertions, 0 deletions
diff --git a/source/com/c2kernel/graph/layout/DefaultGraphLayoutGenerator.java b/source/com/c2kernel/graph/layout/DefaultGraphLayoutGenerator.java
new file mode 100755
index 0000000..e243817
--- /dev/null
+++ b/source/com/c2kernel/graph/layout/DefaultGraphLayoutGenerator.java
@@ -0,0 +1,116 @@
+package com.c2kernel.graph.layout;
+
+import java.util.Vector;
+
+import com.c2kernel.graph.model.DirectedEdge;
+import com.c2kernel.graph.model.GraphModel;
+import com.c2kernel.graph.model.GraphPoint;
+import com.c2kernel.graph.model.Vertex;
+import com.c2kernel.utils.Logger;
+
+public class DefaultGraphLayoutGenerator {
+    private static int mTopMargin = 100;
+    private static int mLeftMargin = 100;
+    private static int mHorzGap = 180;
+    private static int mVertGap = 100;
+
+    private DefaultGraphLayoutGenerator() {
+    }
+
+    public static void layoutGraph(GraphModel graphModel) {
+        Vertex start = graphModel.getStartVertex();
+        Vector rowVector = new Vector(10, 10);
+        int[] midPoints = null;
+        IntegerWrapper valueOfLargestMidPoint = new IntegerWrapper(0);
+        if (start == null) {
+            Logger.msg(1,"Error graph must have a starting vertex to be layed out");
+            return;
+        }
+        graphModel.clearTags(start);
+        visitVertex(graphModel, start, 0, rowVector, start);
+        midPoints = new int[rowVector.size()];
+        calculateRowMidPoints(rowVector, midPoints, valueOfLargestMidPoint);
+        fillInVertexLocations(graphModel, rowVector, valueOfLargestMidPoint, midPoints);
+        fillInEdgeLocations(graphModel);
+        graphModel.forceNotify();
+    }
+
+    private static void visitVertex(GraphModel graphModel, Vertex vertex, int rowIndex, Vector rowVector, Object tag) {
+        int i = 0;
+        Vertex[] children = graphModel.getOutVertices(vertex);
+        vertex.setTag(tag);
+        addVertexToRow(vertex, rowIndex, rowVector);
+        for (i = 0; i < children.length; i++) {
+            if (!(children[i].hasTag(tag))) {
+                visitVertex(graphModel, children[i], rowIndex + 1, rowVector, tag);
+            }
+        }
+    }
+
+    private static void addVertexToRow(Vertex vertex, int rowIndex, Vector rowVector) {
+        Vector rowsVertices = null;
+        // If there is no vector of vertices already created for this row,
+        // then create one
+        if (rowVector.size() == rowIndex) {
+            rowVector.add(new Vector(10, 10));
+        }
+        // Add the vertex to the row's vector of vertices
+        rowsVertices = (Vector)rowVector.elementAt(rowIndex);
+        rowsVertices.add(vertex);
+    }
+
+    private static void calculateRowMidPoints(Vector rowVector, int[] midPoints, IntegerWrapper valueOfLargestMidPoint) {
+        Vector rowsVertices = null;
+        int rowsWidth = 0;
+        int i = 0;
+        for (i = 0; i < midPoints.length; i++) {
+            rowsVertices = (Vector)rowVector.elementAt(i);
+            rowsWidth = mHorzGap * (rowsVertices.size() - 1);
+            midPoints[i] = rowsWidth / 2;
+            if (midPoints[i] > valueOfLargestMidPoint.mValue) {
+                valueOfLargestMidPoint.mValue = midPoints[i];
+            }
+        }
+    }
+
+    private static void fillInVertexLocations(GraphModel graphModel, Vector rowVector,
+        IntegerWrapper valueOfLargestMidPoint, int[] midPoints) {
+            Vector rowsVertices = null;
+            Vertex vertex = null;
+            int rowIndex = 0;
+            int column = 0;
+            int rowsLeftMargin = 0;
+            GraphPoint point = new GraphPoint(0, 0);
+            for (rowIndex = 0; rowIndex < rowVector.size(); rowIndex++) {
+                rowsVertices = (Vector)rowVector.elementAt(rowIndex);
+                rowsLeftMargin = mLeftMargin + valueOfLargestMidPoint.mValue - midPoints[rowIndex];
+                for (column = 0; column < rowsVertices.size(); column++) {
+                    vertex = (Vertex)rowsVertices.elementAt(column);
+                    point.x = rowsLeftMargin + column * mHorzGap;
+                    point.y = mTopMargin + rowIndex * mVertGap;
+                    vertex.moveAbsolute(point);
+                    graphModel.checkSize(vertex);
+                }
+            }
+    }
+
+    private static void fillInEdgeLocations(GraphModel graphModel) {
+        Vertex[] vertices = graphModel.getVertices();
+        GraphPoint centrePoint = null;
+        DirectedEdge[] inEdges = null;
+        DirectedEdge[] outEdges = null;
+        int i = 0;
+        int j = 0;
+        for (i = 0; i < vertices.length; i++) {
+            centrePoint = vertices[i].getCentrePoint();
+            inEdges = graphModel.getInEdges(vertices[i]);
+            outEdges = graphModel.getOutEdges(vertices[i]);
+            for (j = 0; j < inEdges.length; j++) {
+                inEdges[j].setTerminusPoint(centrePoint);
+            }
+            for (j = 0; j < outEdges.length; j++) {
+                outEdges[j].setOriginPoint(centrePoint);
+            }
+        }
+    }
+}