Balanced Layout first level approach

This commit is contained in:
Gonzalo Bellver 2012-01-05 17:34:14 -03:00
parent 86e80e324c
commit f355c5a23a
6 changed files with 280 additions and 26 deletions

View File

@ -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;

View File

@ -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);
canvas.drawGrid(0, 0, size.w, size.h, size.w/squaresize, size.h/squaresize);
var canvas = Raphael(containerId, size.width, size.height);
canvas.drawGrid(0, 0, size.width, size.height, size.width/squaresize, size.height/squaresize);
this._treeSet.plot(canvas);
},

View File

@ -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();

View File

@ -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);

View File

@ -18,14 +18,16 @@
mindplot.nlayout.TestSuite = new Class({
Extends: mindplot.nlayout.ChildrenSorterStrategy,
initialize:function() {
this.testAligned();
this.testSymmetry();
this.testGrid();
this.testEvents();
this.testEventsComplex();
this.testDisconnect();
this.testReconnect();
this.testRemoveNode();
// this.testAligned();
// this.testSymmetry();
this.testBalanced();
// this.testGrid();
// this.testEvents();
// this.testEventsComplex();
// this.testDisconnect();
// 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.plot("testDisconnect1", {w:300, h:200});
manager.dump();
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.plot("testDisconnect2", {w:300, h:200});
manager.dump();
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.plot("testDisconnect3", {w:300, h:200});
manager.dump();
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");
}

View File

@ -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>