You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
2.8 KiB
90 lines
2.8 KiB
4 years ago
|
/**
|
||
|
|
* Tree layout
|
||
|
|
* @module echarts/layout/Tree
|
||
|
|
* @author pissang(http://github.com/pissang)
|
||
|
|
*/
|
||
|
|
define(function (require) {
|
||
|
|
|
||
|
|
var vec2 = require('zrender/tool/vector');
|
||
|
|
|
||
|
|
function TreeLayout(opts) {
|
||
|
|
|
||
|
|
opts = opts || {};
|
||
|
|
|
||
|
|
this.nodePadding = opts.nodePadding || 30;
|
||
|
|
|
||
|
|
this.layerPadding = opts.layerPadding || 100;
|
||
|
|
|
||
|
|
this._layerOffsets = [];
|
||
|
|
|
||
|
|
this._layers = [];
|
||
|
|
}
|
||
|
|
|
||
|
|
TreeLayout.prototype.run = function (tree) {
|
||
|
|
this._layerOffsets.length = 0;
|
||
|
|
for (var i = 0; i < tree.root.height + 1; i++) {
|
||
|
|
this._layerOffsets[i] = 0;
|
||
|
|
this._layers[i] = [];
|
||
|
|
}
|
||
|
|
this._updateNodeXPosition(tree.root);
|
||
|
|
var root = tree.root;
|
||
|
|
this._updateNodeYPosition(root, 0, root.layout.height);
|
||
|
|
};
|
||
|
|
|
||
|
|
TreeLayout.prototype._updateNodeXPosition = function (node) {
|
||
|
|
var minX = Infinity;
|
||
|
|
var maxX = -Infinity;
|
||
|
|
node.layout.position = node.layout.position || vec2.create();
|
||
|
|
for (var i = 0; i < node.children.length; i++) {
|
||
|
|
var child = node.children[i];
|
||
|
|
this._updateNodeXPosition(child);
|
||
|
|
var x = child.layout.position[0];
|
||
|
|
if (x < minX) {
|
||
|
|
minX = x;
|
||
|
|
}
|
||
|
|
if (x > maxX) {
|
||
|
|
maxX = x;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (node.children.length > 0) {
|
||
|
|
node.layout.position[0] = (minX + maxX) / 2;
|
||
|
|
} else {
|
||
|
|
node.layout.position[0] = 0;
|
||
|
|
}
|
||
|
|
var off = this._layerOffsets[node.depth] || 0;
|
||
|
|
if (off > node.layout.position[0]) {
|
||
|
|
var shift = off - node.layout.position[0];
|
||
|
|
this._shiftSubtree(node, shift);
|
||
|
|
for (var i = node.depth + 1; i < node.height + node.depth; i++) {
|
||
|
|
this._layerOffsets[i] += shift;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
this._layerOffsets[node.depth] = node.layout.position[0] + node.layout.width + this.nodePadding;
|
||
|
|
|
||
|
|
this._layers[node.depth].push(node);
|
||
|
|
};
|
||
|
|
|
||
|
|
TreeLayout.prototype._shiftSubtree = function (root, offset) {
|
||
|
|
root.layout.position[0] += offset;
|
||
|
|
for (var i = 0; i < root.children.length; i++) {
|
||
|
|
this._shiftSubtree(root.children[i], offset);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
TreeLayout.prototype._updateNodeYPosition = function (node, y, prevLayerHeight) {
|
||
|
|
node.layout.position[1] = y;
|
||
|
|
var layerHeight = 0;
|
||
|
|
for (var i = 0; i < node.children.length; i++) {
|
||
|
|
layerHeight = Math.max(node.children[i].layout.height, layerHeight);
|
||
|
|
}
|
||
|
|
var layerPadding = this.layerPadding;
|
||
|
|
if (typeof(layerPadding) === 'function') {
|
||
|
|
layerPadding = layerPadding(node.depth);
|
||
|
|
}
|
||
|
|
for (var i = 0; i < node.children.length; i++) {
|
||
|
|
this._updateNodeYPosition(node.children[i], y + layerPadding + prevLayerHeight, layerHeight);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
return TreeLayout;
|
||
|
|
});
|