sunburstVisual.js 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  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. export default function sunburstVisual(ecModel) {
  43. var paletteScope = {};
  44. // Default color strategy
  45. function pickColor(node, seriesModel, treeHeight) {
  46. // Choose color from palette based on the first level.
  47. var current = node;
  48. while (current && current.depth > 1) {
  49. current = current.parentNode;
  50. }
  51. var color = seriesModel.getColorFromPalette(current.name || current.dataIndex + '', paletteScope);
  52. if (node.depth > 1 && isString(color)) {
  53. // Lighter on the deeper level.
  54. color = lift(color, (node.depth - 1) / (treeHeight - 1) * 0.5);
  55. }
  56. return color;
  57. }
  58. ecModel.eachSeriesByType('sunburst', function (seriesModel) {
  59. var data = seriesModel.getData();
  60. var tree = data.tree;
  61. tree.eachNode(function (node) {
  62. var model = node.getModel();
  63. var style = model.getModel('itemStyle').getItemStyle();
  64. if (!style.fill) {
  65. style.fill = pickColor(node, seriesModel, tree.root.height);
  66. }
  67. var existsStyle = data.ensureUniqueItemVisual(node.dataIndex, 'style');
  68. extend(existsStyle, style);
  69. });
  70. });
  71. }