123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216 |
- import { parsePercent } from '../../util/number.js';
- import * as zrUtil from 'zrender/lib/core/util.js';
- var RADIAN = Math.PI / 180;
- export default function sunburstLayout(seriesType, ecModel, api) {
- ecModel.eachSeriesByType(seriesType, function (seriesModel) {
- var center = seriesModel.get('center');
- var radius = seriesModel.get('radius');
- if (!zrUtil.isArray(radius)) {
- radius = [0, radius];
- }
- if (!zrUtil.isArray(center)) {
- center = [center, center];
- }
- var width = api.getWidth();
- var height = api.getHeight();
- var size = Math.min(width, height);
- var cx = parsePercent(center[0], width);
- var cy = parsePercent(center[1], height);
- var r0 = parsePercent(radius[0], size / 2);
- var r = parsePercent(radius[1], size / 2);
- var startAngle = -seriesModel.get('startAngle') * RADIAN;
- var minAngle = seriesModel.get('minAngle') * RADIAN;
- var virtualRoot = seriesModel.getData().tree.root;
- var treeRoot = seriesModel.getViewRoot();
- var rootDepth = treeRoot.depth;
- var sort = seriesModel.get('sort');
- if (sort != null) {
- initChildren(treeRoot, sort);
- }
- var validDataCount = 0;
- zrUtil.each(treeRoot.children, function (child) {
- !isNaN(child.getValue()) && validDataCount++;
- });
- var sum = treeRoot.getValue();
-
- var unitRadian = Math.PI / (sum || validDataCount) * 2;
- var renderRollupNode = treeRoot.depth > 0;
- var levels = treeRoot.height - (renderRollupNode ? -1 : 1);
- var rPerLevel = (r - r0) / (levels || 1);
- var clockwise = seriesModel.get('clockwise');
- var stillShowZeroSum = seriesModel.get('stillShowZeroSum');
-
-
-
- var dir = clockwise ? 1 : -1;
-
- var renderNode = function (node, startAngle) {
- if (!node) {
- return;
- }
- var endAngle = startAngle;
-
- if (node !== virtualRoot) {
-
- var value = node.getValue();
- var angle = sum === 0 && stillShowZeroSum ? unitRadian : value * unitRadian;
- if (angle < minAngle) {
- angle = minAngle;
-
- }
-
-
-
- endAngle = startAngle + dir * angle;
- var depth = node.depth - rootDepth - (renderRollupNode ? -1 : 1);
- var rStart = r0 + rPerLevel * depth;
- var rEnd = r0 + rPerLevel * (depth + 1);
- var levelModel = seriesModel.getLevelModel(node);
- if (levelModel) {
- var r0_1 = levelModel.get('r0', true);
- var r_1 = levelModel.get('r', true);
- var radius_1 = levelModel.get('radius', true);
- if (radius_1 != null) {
- r0_1 = radius_1[0];
- r_1 = radius_1[1];
- }
- r0_1 != null && (rStart = parsePercent(r0_1, size / 2));
- r_1 != null && (rEnd = parsePercent(r_1, size / 2));
- }
- node.setLayout({
- angle: angle,
- startAngle: startAngle,
- endAngle: endAngle,
- clockwise: clockwise,
- cx: cx,
- cy: cy,
- r0: rStart,
- r: rEnd
- });
- }
-
- if (node.children && node.children.length) {
-
- var siblingAngle_1 = 0;
- zrUtil.each(node.children, function (node) {
- siblingAngle_1 += renderNode(node, startAngle + siblingAngle_1);
- });
- }
- return endAngle - startAngle;
- };
-
- if (renderRollupNode) {
- var rStart = r0;
- var rEnd = r0 + rPerLevel;
- var angle = Math.PI * 2;
- virtualRoot.setLayout({
- angle: angle,
- startAngle: startAngle,
- endAngle: startAngle + angle,
- clockwise: clockwise,
- cx: cx,
- cy: cy,
- r0: rStart,
- r: rEnd
- });
- }
- renderNode(treeRoot, startAngle);
- });
- }
- function initChildren(node, sortOrder) {
- var children = node.children || [];
- node.children = sort(children, sortOrder);
-
- if (children.length) {
- zrUtil.each(node.children, function (child) {
- initChildren(child, sortOrder);
- });
- }
- }
- function sort(children, sortOrder) {
- if (zrUtil.isFunction(sortOrder)) {
- var sortTargets = zrUtil.map(children, function (child, idx) {
- var value = child.getValue();
- return {
- params: {
- depth: child.depth,
- height: child.height,
- dataIndex: child.dataIndex,
- getValue: function () {
- return value;
- }
- },
- index: idx
- };
- });
- sortTargets.sort(function (a, b) {
- return sortOrder(a.params, b.params);
- });
- return zrUtil.map(sortTargets, function (target) {
- return children[target.index];
- });
- } else {
- var isAsc_1 = sortOrder === 'asc';
- return children.sort(function (a, b) {
- var diff = (a.getValue() - b.getValue()) * (isAsc_1 ? 1 : -1);
- return diff === 0 ? (a.dataIndex - b.dataIndex) * (isAsc_1 ? -1 : 1) : diff;
- });
- }
- }
|