sunburstVisual.js 3.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. /*
  2. * Licensed to the Apache Software Foundation (ASF) under one
  3. * or more contributor license agreements. See the NOTICE file
  4. * distributed with this work for additional information
  5. * regarding copyright ownership. The ASF licenses this file
  6. * to you under the Apache License, Version 2.0 (the
  7. * "License"); you may not use this file except in compliance
  8. * with the License. You may obtain a copy of the License at
  9. *
  10. * http://www.apache.org/licenses/LICENSE-2.0
  11. *
  12. * Unless required by applicable law or agreed to in writing,
  13. * software distributed under the License is distributed on an
  14. * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15. * KIND, either express or implied. See the License for the
  16. * specific language governing permissions and limitations
  17. * under the License.
  18. */
  19. /**
  20. * AUTO-GENERATED FILE. DO NOT MODIFY.
  21. */
  22. /*
  23. * Licensed to the Apache Software Foundation (ASF) under one
  24. * or more contributor license agreements. See the NOTICE file
  25. * distributed with this work for additional information
  26. * regarding copyright ownership. The ASF licenses this file
  27. * to you under the Apache License, Version 2.0 (the
  28. * "License"); you may not use this file except in compliance
  29. * with the License. You may obtain a copy of the License at
  30. *
  31. * http://www.apache.org/licenses/LICENSE-2.0
  32. *
  33. * Unless required by applicable law or agreed to in writing,
  34. * software distributed under the License is distributed on an
  35. * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  36. * KIND, either express or implied. See the License for the
  37. * specific language governing permissions and limitations
  38. * under the License.
  39. */
  40. import { lift } from 'zrender/lib/tool/color.js';
  41. import { extend, isString } from 'zrender/lib/core/util.js';
  42. import tokens from '../../visual/tokens.js';
  43. export default function sunburstVisual(ecModel) {
  44. var paletteScope = {};
  45. // Default color strategy
  46. function pickColor(node, seriesModel, treeHeight) {
  47. if (node.depth === 0) {
  48. // Don't use palette color for the root node, because it's displayed only when drilling down.
  49. return tokens.color.neutral50;
  50. }
  51. // Choose color from palette based on the first level.
  52. var current = node;
  53. while (current && current.depth > 1) {
  54. current = current.parentNode;
  55. }
  56. var color = seriesModel.getColorFromPalette(current.name || current.dataIndex + '', paletteScope);
  57. if (node.depth > 1 && isString(color)) {
  58. // Lighter on the deeper level.
  59. color = lift(color, (node.depth - 1) / (treeHeight - 1) * 0.5);
  60. }
  61. return color;
  62. }
  63. ecModel.eachSeriesByType('sunburst', function (seriesModel) {
  64. var data = seriesModel.getData();
  65. var tree = data.tree;
  66. tree.eachNode(function (node) {
  67. var model = node.getModel();
  68. var style = model.getModel('itemStyle').getItemStyle();
  69. if (!style.fill) {
  70. style.fill = pickColor(node, seriesModel, tree.root.height);
  71. }
  72. var existsStyle = data.ensureUniqueItemVisual(node.dataIndex, 'style');
  73. extend(existsStyle, style);
  74. });
  75. });
  76. }