Balanced Layout first level approach

mindplot/src/main/javascript/nlayout/BalancedSorter.js

@@ -0,0 +1,175 @@
+mindplot.nlayout.BalancedSorter = new Class({
+    Extends: mindplot.nlayout.ChildrenSorterStrategy,
+
+    initialize: function() {
+
+    },
+
+    computeChildrenIdByHeights: function(treeSet, node) {
+        var result = {};
+        this._computeChildrenHeight(treeSet, node, result);
+        return result;
+    },
+
+    _computeChildrenHeight : function(treeSet, node, heightCache) {
+        var height = node.getSize().height + (mindplot.nlayout.BalancedSorter.INTERNODE_VERTICAL_PADDING * 2); // 2* Top and down padding;
+
+        var result;
+        var children = treeSet.getChildren(node);
+        if (children.length == 0) {
+            result = height;
+        } else {
+            var childrenHeight = 0;
+            children.forEach(function(child) {
+                childrenHeight += this._computeChildrenHeight(treeSet, child, heightCache);
+            }, this);
+
+            result = Math.max(height, childrenHeight);
+        }
+
+        if (heightCache) {
+            heightCache[node.getId()] = result;
+        }
+
+        return result;
+    },
+
+    predict : function(parent, graph, position) {
+
+        // No children...
+        var children = graph.getChildren(parent);
+        if (children.length == 0) {
+            return [0,parent.getPosition()];  // @Todo:Change x ...
+        }
+
+        // Try to fit within ...
+        //
+        // - Order is change if the position top position is changed ...
+        // - Suggested position is the middle between the two topics...
+        //
+        var result = null;
+        children.forEach(function(child) {
+            var cpos = child.getPosition();
+            if (position.y > cpos.y) {
+                result = [child.getOrder(),{x:cpos.x,y:cpos.y + child.getSize().height}];
+            }
+        });
+
+        // Ok, no overlap. Suggest a new order.
+        if (result) {
+            var last = children.getLast();
+            result = [last.getOrder() + 1,{x:cpos.x,y:cpos.y - (mindplot.nlayout.BalancedSorter.INTERNODE_VERTICAL_PADDING * 4)}];
+        }
+
+        return result;
+    },
+
+    insert: function(treeSet, parent, child, order) {
+        var children = this._getSortedChildren(treeSet, parent);
+        $assert(order <= children.length, "Order must be continues and can not have holes. Order:" + order);
+        $assert(order <= children.length, "Order must be continues and can not have holes. Order:" + order);
+
+        // Shift all the elements in one .
+        for (var i = order; i < children.length; i++) {
+            var node = children[i];
+            node.setOrder(i + 1);
+        }
+        child.setOrder(order);
+    },
+
+    detach:function(treeSet, node) {
+        var parent = treeSet.getParent(node);
+        var children = this._getSortedChildren(treeSet, parent);
+        var order = node.getOrder();
+        $assert(children[order] === node, "Node seems not to be in the right position");
+
+        // Shift all the nodes ...
+        for (var i = node.getOrder() + 1; i < children.length; i++) {
+            var child = children[i];
+            child.setOrder(child.getOrder() - 1);
+        }
+        node.setOrder(0);
+    },
+
+    _getSortedChildren:function(treeSet, node) {
+        var result = treeSet.getChildren(node);
+        result.sort(function(a, b) {
+            return a.getOrder() - b.getOrder()
+        });
+        return result;
+    },
+
+    verify:function(treeSet, node) {
+        // Check that all is consistent ...
+        var children = this._getSortedChildren(treeSet, node);
+
+        for (var i = 0; i < children.length; i++) {
+            $assert(children[i].getOrder() == i, "missing order elements");
+        }
+    },
+
+    computeOffsets:function(treeSet, node) {
+        $assert(treeSet, "treeSet can no be null.");
+        $assert(node, "node can no be null.");
+
+        var children = this._getSortedChildren(treeSet, node);
+
+        // Compute heights ...
+        var heights = children.map(function(child) {
+            return {id:child.getId(), order:child.getOrder(), height:this._computeChildrenHeight(treeSet, child)};
+        }, this);
+
+
+        // Compute the center of the branch ...
+        var totalPHeight = 0;
+        var totalNHeight = 0;
+        //TODO(gb): optimize using filter
+        heights.forEach(function(elem) {
+            if (elem.order % 2 == 0) {
+                totalPHeight += elem.height;
+            } else {
+                totalNHeight += elem.height;
+            }
+        });
+        var psum = totalPHeight/2;
+        var nsum = totalNHeight/2;
+        var totalDifference = totalPHeight - totalNHeight;
+        var ysum = 0;
+
+        // Calculate the offsets ...
+        var result = {};
+        for (var i = 0; i < heights.length; i++) {
+            var direction = heights[i].order % 2 ? -1 : 1;
+
+            if (direction > 0) {
+                psum = psum - heights[i].height;
+                ysum = psum;
+            } else {
+                nsum = nsum - heights[i].height;
+                ysum = nsum;
+            }
+
+//            console.log("node {id: " + heights[i].id + ", order: " + heights[i].order + ", psum: " + psum + ", nsum: " + nsum + "}");        //TODO(gb): Remove trace!!!
+//            console.log("totalPHeight: " + totalPHeight + ", totalNHeight: " + totalNHeight);        //TODO(gb): Remove trace!!!
+            console.log("node {id: " + heights[i].id + ", totalDifference: " + totalDifference + ", direction: " + direction + "}");        //TODO(gb): Remove trace!!!
+
+            var yOffset = ysum + heights[i].height/2;
+            var xOffset = direction * (node.getSize().width + mindplot.nlayout.SymetricSorter.INTERNODE_HORIZONTAL_PADDING);
+
+            $assert(!isNaN(xOffset), "xOffset can not be null");
+            $assert(!isNaN(yOffset), "yOffset can not be null");
+
+            result[heights[i].id] = {x:xOffset,y:yOffset};
+        }
+        console.log("----------------------\n");        //TODO(gb): Remove trace!!!
+        return result;
+    },
+
+    toString:function() {
+        return "Balanced Sorter";
+    }
+});
+
+mindplot.nlayout.BalancedSorter.INTERNODE_VERTICAL_PADDING = 5;
+mindplot.nlayout.BalancedSorter.INTERNODE_HORIZONTAL_PADDING = 5;
+mindplot.nlayout.BalancedSorter.INITIAL_HEIGHT = 100;

mindplot/src/main/javascript/nlayout/LayoutManager.js

@@ -39,10 +39,14 @@ mindplot.nlayout.LayoutManager = new Class({
         // @Todo: finish...
     },
 
-    find:  function(id) {
+    find: function(id) {
         return this._treeSet.find(id);
     },
 
+    move: function() {
+      //TODO(gb): implement
+    },
+
     connectNode: function(parentId, childId, order) {
         $assert($defined(parentId), "parentId can not be null");
         $assert($defined(childId), "childId can not be null");
@@ -92,10 +96,10 @@ mindplot.nlayout.LayoutManager = new Class({
 
     plot: function(containerId, size) {
         $assert(containerId, "containerId cannot be null");
-        size = size || {w:200,h:200};
+        size = size || {width:200,height:200};
         var squaresize = 10;
-        var canvas = Raphael(containerId, size.w, size.h);
+        var canvas = Raphael(containerId, size.width, size.height);
-        canvas.drawGrid(0, 0, size.w, size.h, size.w/squaresize, size.h/squaresize);
+        canvas.drawGrid(0, 0, size.width, size.height, size.width/squaresize, size.height/squaresize);
         this._treeSet.plot(canvas);
     },
 

mindplot/src/main/javascript/nlayout/OriginalLayout.js

@@ -26,7 +26,9 @@ mindplot.nlayout.OriginalLayout = new Class({
         $assert(position, "position can not be null");
         $assert(type, "type can not be null");
 
-        var strategy = type === 'root' ? mindplot.nlayout.OriginalLayout.GRID_SORTER : mindplot.nlayout.OriginalLayout.SYMETRIC_SORTER;
+        var strategy = type === 'root' ?
+            mindplot.nlayout.OriginalLayout.BALANCED_SORTER :
+            mindplot.nlayout.OriginalLayout.SYMETRIC_SORTER;
         return new mindplot.nlayout.Node(id, size, position, strategy);
     },
 
@@ -120,6 +122,7 @@ mindplot.nlayout.OriginalLayout = new Class({
 
 mindplot.nlayout.OriginalLayout.SYMETRIC_SORTER = new mindplot.nlayout.SymetricSorter();
 mindplot.nlayout.OriginalLayout.GRID_SORTER = new mindplot.nlayout.GridSorter();
+mindplot.nlayout.OriginalLayout.BALANCED_SORTER = new mindplot.nlayout.BalancedSorter();
 
 
 

mindplot/src/main/javascript/nlayout/SymetricSorter.js

@@ -127,7 +127,6 @@ mindplot.nlayout.SymetricSorter = new Class({
     computeOffsets:function(treeSet, node) {
         $assert(treeSet, "treeSet can no be null.");
         $assert(node, "node can no be null.");
-        $assert("order can no be null.");
 
         var children = this._getSortedChildren(treeSet, node);
 

mindplot/src/main/javascript/nlayout/TestSuite.js

@@ -18,14 +18,16 @@
 mindplot.nlayout.TestSuite = new Class({
     Extends: mindplot.nlayout.ChildrenSorterStrategy,
     initialize:function() {
-        this.testAligned();
+//        this.testAligned();
-        this.testSymmetry();
+//        this.testSymmetry();
-        this.testGrid();
+        this.testBalanced();
-        this.testEvents();
+//        this.testGrid();
-        this.testEventsComplex();
+//        this.testEvents();
-        this.testDisconnect();
+//        this.testEventsComplex();
-        this.testReconnect();
+//        this.testDisconnect();
-        this.testRemoveNode();
+//        this.testReconnect();
+//        this.testRemoveNode();
+//        this.testFreePosition();
     },
 
     testAligned: function() {
@@ -45,7 +47,7 @@ mindplot.nlayout.TestSuite = new Class({
 
         manager.layout();
         manager.dump();
-        manager.plot("testAligned", {w:300,h:200});
+        manager.plot("testAligned", {width:300,height:200});
 
         // All nodes should be vertically aligned
         $assert(manager.find(0).getPosition().y == manager.find(1).getPosition().y, "Nodes are not aligned");
@@ -90,11 +92,50 @@ mindplot.nlayout.TestSuite = new Class({
 
         manager.layout();
         manager.dump();
-        manager.plot("testSymmetry",{w:500, h:300});
+        manager.plot("testSymmetry",{width:500, height:300});
 
         //TODO(gb): make asserts
     },
 
+    testBalanced: function() {
+//        var size = {width:80, height:30};
+        var size = {width:25, height:25};
+        var plotsize = {width: 400, height:280};
+        var position = {x:0, y:0};
+//        var manager = new mindplot.nlayout.LayoutManager(0, {width:120, height:40});
+        var manager = new mindplot.nlayout.LayoutManager(0, size);
+
+        manager.addNode(1, size, position);
+        manager.connectNode(0, 1, 0);
+        manager.layout();
+        manager.plot("testBalanced1", plotsize);
+
+        manager.addNode(2, size, position);
+        manager.connectNode(0, 2, 1);
+        manager.layout();
+        manager.plot("testBalanced2", plotsize);
+
+        manager.addNode(3, size, position);
+        manager.connectNode(0, 3, 2);
+        manager.layout();
+        manager.plot("testBalanced3", plotsize);
+
+        manager.addNode(4, size, position);
+        manager.connectNode(0, 4, 3);
+        manager.layout();
+        manager.plot("testBalanced4", plotsize);
+
+        manager.addNode(5, size, position);
+        manager.connectNode(0, 5, 4);
+        manager.layout();
+        manager.plot("testBalanced5", plotsize);
+
+        manager.addNode(6, size, position);
+        manager.connectNode(0, 6, 5);
+        manager.layout();
+        manager.plot("testBalanced6", plotsize);
+    },
+
     testGrid: function() {
         var size = {width:25,height:25};
         var position = {x:0,y:0};
@@ -127,7 +168,7 @@ mindplot.nlayout.TestSuite = new Class({
         manager.connectNode(2, 6, 1);
         manager.connectNode(6, 7, 0);
         manager.layout();
-        manager.plot("testGrid5", {w:300, h:300});
+        manager.plot("testGrid5", {width:300, height:300});
 
         manager.dump();
 
@@ -236,16 +277,16 @@ mindplot.nlayout.TestSuite = new Class({
             events.push(event);
         });
         manager.layout(true);
-//        manager.dump();
+        manager.dump();
-        manager.plot("testDisconnect1", {w:300, h:200});
+        manager.plot("testDisconnect1", {width:300, height:200});
 
         // Now, disconnect one node ...
         console.log("--- Disconnect a single node ---");
         events.empty();
         manager.disconnectNode(2);
         manager.layout(true);
-//        manager.dump();
+        manager.dump();
-        manager.plot("testDisconnect2", {w:300, h:200});
+        manager.plot("testDisconnect2", {width:300, height:200});
 
         $assert(events.some(
             function(event) {
@@ -256,8 +297,8 @@ mindplot.nlayout.TestSuite = new Class({
         console.log("--- Disconnect a node with children ---");
         manager.disconnectNode(3);
         manager.layout(true);
-//        manager.dump();
+        manager.dump();
-        manager.plot("testDisconnect3", {w:300, h:200});
+        manager.plot("testDisconnect3", {width:300, height:200});
 
         $assert(events.some(
             function(event) {
@@ -297,14 +338,14 @@ mindplot.nlayout.TestSuite = new Class({
 
         manager.layout();
         manager.dump();
-        manager.plot("testReconnect1",{w:400, h:300});
+        manager.plot("testReconnect1",{width:400, height:300});
 
         // Reconnect node 6 to node 4
         manager.disconnectNode(6);
         manager.connectNode(4,6,0);
         manager.layout();
         manager.dump();
-        manager.plot("testReconnect2",{w:400, h:300});
+        manager.plot("testReconnect2",{width:400, height:300});
 
         //TODO(gb): make asserts
     },
@@ -334,7 +375,7 @@ mindplot.nlayout.TestSuite = new Class({
         });
         manager.layout(true);
         manager.dump();
-        manager.plot("testRemoveNode1", {w:300, h:200});
+        manager.plot("testRemoveNode1", {width:300, height:200});
 
         // Test removal of a connected node ...
         console.log("--- Remove node 3  ---");
@@ -342,6 +383,26 @@ mindplot.nlayout.TestSuite = new Class({
         manager.layout(true);
         manager.dump();
         manager.plot("testRemoveNode2");
+    },
+
+    testFreePosition: function() {
+        var size = {width:25,height:25};
+        var position = {x:0,y:0};
+        var manager = new mindplot.nlayout.LayoutManager(0, size);
+
+        // Prepare a sample graph ...
+        manager.addNode(1, size, position);
+        manager.addNode(2, size, position);
+        manager.addNode(3, size, position);
+        manager.addNode(4, size, position);
+
+        manager.connectNode(0, 4, 0);
+        manager.connectNode(4, 1, 0);
+        manager.connectNode(4, 2, 1);
+        manager.connectNode(4, 3, 2);
+
+        manager.layout();
+        manager.plot("testFreePosition");
     }
 
 

mindplot/src/test/javascript/static/layout.html

@@ -14,6 +14,7 @@
     <script type='text/javascript' src='../../../main/javascript/nlayout/ChildrenSorterStrategy.js'></script>
     <script type='text/javascript' src='../../../main/javascript/nlayout/SymetricSorter.js'></script>
     <script type='text/javascript' src='../../../main/javascript/nlayout/GridSorter.js'></script>
+    <script type='text/javascript' src='../../../main/javascript/nlayout/BalancedSorter.js'></script>
     <script type='text/javascript' src='../../../main/javascript/nlayout/OriginalLayout.js'></script>
     <script type='text/javascript' src='../../../main/javascript/nlayout/ChangeEvent.js'></script>
 
@@ -57,6 +58,14 @@
     <div id="testGrid5" class="col last"></div>
 </div>
 
+<h2>testBalanced:</h2>
+<div id="testBalanced1" class="col"></div>
+<div id="testBalanced2" class="col"></div>
+<div id="testBalanced3" class="col last"></div>
+<div id="testBalanced4" class="col"></div>
+<div id="testBalanced5" class="col"></div>
+<div id="testBalanced6" class="col last"></div>
+
 <h2>testEvents:</h2>
 <div id="testEvents1" class="col"></div>
 <div id="testEvents2" class="col last"></div>
@@ -78,6 +87,9 @@
 <div id="testRemoveNode1" class="col"></div>
 <div id="testRemoveNode2" class="col last"></div>
 
+<h2>testFreePosition:</h2>
+<div id="testFreePosition"></div>
+
 
 </body>
 </html>