cione-comp-lib
Version:
`$ yarn add cione-comp-lib` 或者 `$ npm i cione-comp-lib -S`
97 lines (96 loc) • 3.34 kB
JavaScript
import { eachAfter, eachBefore } from "./traversalHelper.mjs";
import { getViewRect, init, separation, firstWalk, secondWalk, radialCoordinate } from "./layoutHelper.mjs";
function treeLayout(ecModel, api) {
ecModel.eachSeriesByType("tree", function(seriesModel) {
commonLayout(seriesModel, api);
});
}
function commonLayout(seriesModel, api) {
var layoutInfo = getViewRect(seriesModel, api);
seriesModel.layoutInfo = layoutInfo;
var layout = seriesModel.get("layout");
var width = 0;
var height = 0;
var separation$1 = null;
if (layout === "radial") {
width = 2 * Math.PI;
height = Math.min(layoutInfo.height, layoutInfo.width) / 2;
separation$1 = separation(function(node1, node2) {
return (node1.parentNode === node2.parentNode ? 1 : 2) / node1.depth;
});
} else {
width = layoutInfo.width;
height = layoutInfo.height;
separation$1 = separation();
}
var virtualRoot = seriesModel.getData().tree.root;
var realRoot = virtualRoot.children[0];
if (realRoot) {
init(virtualRoot);
eachAfter(realRoot, firstWalk, separation$1);
virtualRoot.hierNode.modifier = -realRoot.hierNode.prelim;
eachBefore(realRoot, secondWalk);
var left_1 = realRoot;
var right_1 = realRoot;
var bottom_1 = realRoot;
eachBefore(realRoot, function(node) {
var x = node.getLayout().x;
if (x < left_1.getLayout().x) {
left_1 = node;
}
if (x > right_1.getLayout().x) {
right_1 = node;
}
if (node.depth > bottom_1.depth) {
bottom_1 = node;
}
});
var delta = left_1 === right_1 ? 1 : separation$1(left_1, right_1) / 2;
var tx_1 = delta - left_1.getLayout().x;
var kx_1 = 0;
var ky_1 = 0;
var coorX_1 = 0;
var coorY_1 = 0;
if (layout === "radial") {
kx_1 = width / (right_1.getLayout().x + delta + tx_1);
ky_1 = height / (bottom_1.depth - 1 || 1);
eachBefore(realRoot, function(node) {
coorX_1 = (node.getLayout().x + tx_1) * kx_1;
coorY_1 = (node.depth - 1) * ky_1;
var finalCoor = radialCoordinate(coorX_1, coorY_1);
node.setLayout({
x: finalCoor.x,
y: finalCoor.y,
rawX: coorX_1,
rawY: coorY_1
}, true);
});
} else {
var orient_1 = seriesModel.getOrient();
if (orient_1 === "RL" || orient_1 === "LR") {
ky_1 = height / (right_1.getLayout().x + delta + tx_1);
kx_1 = width / (bottom_1.depth - 1 || 1);
eachBefore(realRoot, function(node) {
coorY_1 = (node.getLayout().x + tx_1) * ky_1;
coorX_1 = orient_1 === "LR" ? (node.depth - 1) * kx_1 : width - (node.depth - 1) * kx_1;
node.setLayout({
x: coorX_1,
y: coorY_1
}, true);
});
} else if (orient_1 === "TB" || orient_1 === "BT") {
kx_1 = width / (right_1.getLayout().x + delta + tx_1);
ky_1 = height / (bottom_1.depth - 1 || 1);
eachBefore(realRoot, function(node) {
coorX_1 = (node.getLayout().x + tx_1) * kx_1;
coorY_1 = orient_1 === "TB" ? (node.depth - 1) * ky_1 : height - (node.depth - 1) * ky_1;
node.setLayout({
x: coorX_1,
y: coorY_1
}, true);
});
}
}
}
}
export { treeLayout as default };