123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288 |
- import * as layout from '../../util/layout.js';
- export function init(inRoot) {
- var root = inRoot;
- root.hierNode = {
- defaultAncestor: null,
- ancestor: root,
- prelim: 0,
- modifier: 0,
- change: 0,
- shift: 0,
- i: 0,
- thread: null
- };
- var nodes = [root];
- var node;
- var children;
- while (node = nodes.pop()) {
-
- children = node.children;
- if (node.isExpand && children.length) {
- var n = children.length;
- for (var i = n - 1; i >= 0; i--) {
- var child = children[i];
- child.hierNode = {
- defaultAncestor: null,
- ancestor: child,
- prelim: 0,
- modifier: 0,
- change: 0,
- shift: 0,
- i: i,
- thread: null
- };
- nodes.push(child);
- }
- }
- }
- }
- export function firstWalk(node, separation) {
- var children = node.isExpand ? node.children : [];
- var siblings = node.parentNode.children;
- var subtreeW = node.hierNode.i ? siblings[node.hierNode.i - 1] : null;
- if (children.length) {
- executeShifts(node);
- var midPoint = (children[0].hierNode.prelim + children[children.length - 1].hierNode.prelim) / 2;
- if (subtreeW) {
- node.hierNode.prelim = subtreeW.hierNode.prelim + separation(node, subtreeW);
- node.hierNode.modifier = node.hierNode.prelim - midPoint;
- } else {
- node.hierNode.prelim = midPoint;
- }
- } else if (subtreeW) {
- node.hierNode.prelim = subtreeW.hierNode.prelim + separation(node, subtreeW);
- }
- node.parentNode.hierNode.defaultAncestor = apportion(node, subtreeW, node.parentNode.hierNode.defaultAncestor || siblings[0], separation);
- }
- export function secondWalk(node) {
- var nodeX = node.hierNode.prelim + node.parentNode.hierNode.modifier;
- node.setLayout({
- x: nodeX
- }, true);
- node.hierNode.modifier += node.parentNode.hierNode.modifier;
- }
- export function separation(cb) {
- return arguments.length ? cb : defaultSeparation;
- }
- export function radialCoordinate(rad, r) {
- rad -= Math.PI / 2;
- return {
- x: r * Math.cos(rad),
- y: r * Math.sin(rad)
- };
- }
- export function getViewRect(seriesModel, api) {
- return layout.getLayoutRect(seriesModel.getBoxLayoutParams(), {
- width: api.getWidth(),
- height: api.getHeight()
- });
- }
- function executeShifts(node) {
- var children = node.children;
- var n = children.length;
- var shift = 0;
- var change = 0;
- while (--n >= 0) {
- var child = children[n];
- child.hierNode.prelim += shift;
- child.hierNode.modifier += shift;
- change += child.hierNode.change;
- shift += child.hierNode.shift + change;
- }
- }
- function apportion(subtreeV, subtreeW, ancestor, separation) {
- if (subtreeW) {
- var nodeOutRight = subtreeV;
- var nodeInRight = subtreeV;
- var nodeOutLeft = nodeInRight.parentNode.children[0];
- var nodeInLeft = subtreeW;
- var sumOutRight = nodeOutRight.hierNode.modifier;
- var sumInRight = nodeInRight.hierNode.modifier;
- var sumOutLeft = nodeOutLeft.hierNode.modifier;
- var sumInLeft = nodeInLeft.hierNode.modifier;
- while (nodeInLeft = nextRight(nodeInLeft), nodeInRight = nextLeft(nodeInRight), nodeInLeft && nodeInRight) {
- nodeOutRight = nextRight(nodeOutRight);
- nodeOutLeft = nextLeft(nodeOutLeft);
- nodeOutRight.hierNode.ancestor = subtreeV;
- var shift = nodeInLeft.hierNode.prelim + sumInLeft - nodeInRight.hierNode.prelim - sumInRight + separation(nodeInLeft, nodeInRight);
- if (shift > 0) {
- moveSubtree(nextAncestor(nodeInLeft, subtreeV, ancestor), subtreeV, shift);
- sumInRight += shift;
- sumOutRight += shift;
- }
- sumInLeft += nodeInLeft.hierNode.modifier;
- sumInRight += nodeInRight.hierNode.modifier;
- sumOutRight += nodeOutRight.hierNode.modifier;
- sumOutLeft += nodeOutLeft.hierNode.modifier;
- }
- if (nodeInLeft && !nextRight(nodeOutRight)) {
- nodeOutRight.hierNode.thread = nodeInLeft;
- nodeOutRight.hierNode.modifier += sumInLeft - sumOutRight;
- }
- if (nodeInRight && !nextLeft(nodeOutLeft)) {
- nodeOutLeft.hierNode.thread = nodeInRight;
- nodeOutLeft.hierNode.modifier += sumInRight - sumOutLeft;
- ancestor = subtreeV;
- }
- }
- return ancestor;
- }
- function nextRight(node) {
- var children = node.children;
- return children.length && node.isExpand ? children[children.length - 1] : node.hierNode.thread;
- }
- function nextLeft(node) {
- var children = node.children;
- return children.length && node.isExpand ? children[0] : node.hierNode.thread;
- }
- function nextAncestor(nodeInLeft, node, ancestor) {
- return nodeInLeft.hierNode.ancestor.parentNode === node.parentNode ? nodeInLeft.hierNode.ancestor : ancestor;
- }
- function moveSubtree(wl, wr, shift) {
- var change = shift / (wr.hierNode.i - wl.hierNode.i);
- wr.hierNode.change -= change;
- wr.hierNode.shift += shift;
- wr.hierNode.modifier += shift;
- wr.hierNode.prelim += shift;
- wl.hierNode.change += change;
- }
- function defaultSeparation(node1, node2) {
- return node1.parentNode === node2.parentNode ? 1 : 2;
- }
|