/*** 创建连线* @param {ht.DataModel} dataModel - 数据容器* @param {ht.Node} source - 起点* @param {ht.Node} target - 终点*/
function createEdge(dataModel, source, target) {// 创建连线，链接父亲节点及孩子节点var edge = new ht.Edge();edge.setSource(source);edge.setTarget(target);dataModel.add(edge);
}/*** 创建节点对象* @param {ht.DataModel} dataModel - 数据容器* @param {ht.Node} [parent] - 父亲节点* @returns {ht.Node} 节点对象*/
function createNode(dataModel, parent) {var node = new ht.Node();if (parent) {// 设置父亲节点node.setParent(parent);createEdge(dataModel, parent, node);}// 添加到数据容器中dataModel.add(node);return node;
}/*** 创建结构树* @param {ht.DataModel} dataModel - 数据容器* @param {ht.Node} parent - 父亲节点* @param {Number} level - 深度* @param {Array} count - 每层节点个数* @param {function(ht.Node, Number, Number)} callback - 回调函数(节点对象，节点对应的层级，节点在层级中的编号)*/
function createTreeNodes(dataModel, parent, level, count, callback) {level--;var num = (typeof count === 'number' ? count : count[level]);while (num--) {var node = createNode(dataModel, parent);// 调用回调函数，用户可以在回调里面设置节点相关属性callback(node, level, num);if (level === 0) continue;// 递归调用创建孩子节点createTreeNodes(dataModel, node, level, count, callback);}
}

/*** 布局树* @param {ht.Node} root - 根节点* @param {Number} [minR] - 末端节点的最小半径*/
function layout(root, minR) {// 设置默认半径minR = (minR == null ? 25 : minR);// 获取到所有的孩子节点对象数组var children = root.getChildren().toArray();// 获取孩子节点个数var len = children.length;// 计算张角var degree = Math.PI * 2 / len;// 根据三角函数计算绕父亲节点的半径var sin = Math.sin(degree / 2),r = minR / sin;// 获取父亲节点的位置坐标var rootPosition = root.p();children.forEach(function(child, index) {// 根据三角函数计算每个节点相对于父亲节点的偏移量var s = Math.sin(degree * index),c = Math.cos(degree * index),x = s * r,y = c * r;// 设置孩子节点的位置坐标child.p(x + rootPosition.x, y + rootPosition.y);});
}

/*** 就按节点领域半径* @param {ht.Node} root - 根节点对象* @param {Number} minR - 最小半径*/
function countRadius(root, minR) {minR = (minR == null ? 25 : minR);// 若果是末端节点，则设置其半径为最小半径if (!root.hasChildren()) {root.a('radius', minR);return;}// 遍历孩子节点递归计算半径var children = root.getChildren();children.each(function(child) {countRadius(child, minR);});var child0 = root.getChildAt(0);// 获取孩子节点半径var radius = child0.a('radius');// 计算子节点的1/2张角var degree = Math.PI / children.size();// 计算父亲节点的半径var pRadius = radius / Math.sin(degree);// 设置父亲节点的半径及其孩子节点的布局张角root.a('radius', pRadius);root.a('degree', degree * 2);
}

/*** 布局树* @param {ht.Node} root - 根节点*/
function layout(root) {// 获取到所有的孩子节点对象数组var children = root.getChildren().toArray();// 获取孩子节点个数var len = children.length;// 计算张角var degree = root.a('degree');// 根据三角函数计算绕父亲节点的半径var r = root.a('radius');// 获取父亲节点的位置坐标var rootPosition = root.p();children.forEach(function(child, index) {// 根据三角函数计算每个节点相对于父亲节点的偏移量var s = Math.sin(degree * index),c = Math.cos(degree * index),x = s * r,y = c * r;// 设置孩子节点的位置坐标child.p(x + rootPosition.x, y + rootPosition.y);// 递归调用布局孩子节点layout(child);});
}

/*** 就按节点领域半径* @param {ht.Node} root - 根节点对象* @param {Number} minR - 最小半径*/
function countRadius(root, minR) {……var child0 = root.getChildAt(0);// 获取孩子节点半径var radius = child0.a('radius');var child00 = child0.getChildAt(0);// 半径加上孙子节点半径，避免节点重叠if (child00) radius += child00.a('radius');……
}

/*** 就按节点领域半径及布局半径* @param {ht.Node} root - 根节点对象* @param {Number} minR - 最小半径*/
function countRadius(root, minR) {minR = (minR == null ? 25 : minR);// 若果是末端节点，则设置其布局半径及领域半径为最小半径if (!root.hasChildren()) {root.a('radius', minR);root.a('totalRadius', minR);return;}// 遍历孩子节点递归计算半径var children = root.getChildren();children.each(function(child) {countRadius(child, minR);});var child0 = root.getChildAt(0);// 获取孩子节点半径var radius = child0.a('radius'),totalRadius = child0.a('totalRadius');// 计算子节点的1/2张角var degree = Math.PI / children.size();// 计算父亲节点的布局半径var pRadius = totalRadius / Math.sin(degree);// 缓存父亲节点的布局半径root.a('radius', pRadius);// 缓存父亲节点的领域半径root.a('totalRadius', pRadius + totalRadius);// 缓存其孩子节点的布局张角root.a('degree', degree * 2);
}

/*** 布局树* @param {ht.Node} root - 根节点*/
function layout(root) {// 获取到所有的孩子节点对象数组var children = root.getChildren().toArray();// 获取孩子节点个数var len = children.length;// 计算张角var degree = root.a('degree');// 根据三角函数计算绕父亲节点的半径var r = root.a('radius');// 获取父亲节点的位置坐标var rootPosition = root.p3();children.forEach(function(child, index) {// 根据三角函数计算每个节点相对于父亲节点的偏移量var s = Math.sin(degree * index),c = Math.cos(degree * index),x = s * r,z = c * r;// 设置孩子节点的位置坐标child.p3(x + rootPosition[0], rootPosition[1] - 100, z + rootPosition[2]);// 递归调用布局孩子节点layout(child);});
}

var level = 4,size = (level + 1) * 20;var root = createNode(dataModel);
root.setName('root');
root.p(100, 100);root.s('shape3d', 'sphere');
root.s('shape3d.color', randomColor());
root.s3(size, size, size);var colors = {},sizes = {};
createTreeNodes(dataModel, root, level - 1, 5, function(data, level, num) {if (!colors[level]) {colors[level] = randomColor();sizes[level] = (level + 1) * 20;}size = sizes[level];data.setName('item-' + level + '-' + num);// 设置节点形状为球形data.s('shape3d', 'sphere');data.s('shape3d.color', colors[level]);data.s3(size, size, size);
});