sankeyVisual.js 3.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  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 * as zrUtil from 'zrender/lib/core/util.js';
  41. import VisualMapping from '../../visual/VisualMapping.js';
  42. export default function sankeyVisual(ecModel) {
  43. ecModel.eachSeriesByType('sankey', function (seriesModel) {
  44. var graph = seriesModel.getGraph();
  45. var nodes = graph.nodes;
  46. var edges = graph.edges;
  47. if (nodes.length) {
  48. var minValue_1 = Infinity;
  49. var maxValue_1 = -Infinity;
  50. zrUtil.each(nodes, function (node) {
  51. var nodeValue = node.getLayout().value;
  52. if (nodeValue < minValue_1) {
  53. minValue_1 = nodeValue;
  54. }
  55. if (nodeValue > maxValue_1) {
  56. maxValue_1 = nodeValue;
  57. }
  58. });
  59. zrUtil.each(nodes, function (node) {
  60. var mapping = new VisualMapping({
  61. type: 'color',
  62. mappingMethod: 'linear',
  63. dataExtent: [minValue_1, maxValue_1],
  64. visual: seriesModel.get('color')
  65. });
  66. var mapValueToColor = mapping.mapValueToVisual(node.getLayout().value);
  67. var customColor = node.getModel().get(['itemStyle', 'color']);
  68. if (customColor != null) {
  69. node.setVisual('color', customColor);
  70. node.setVisual('style', {
  71. fill: customColor
  72. });
  73. } else {
  74. node.setVisual('color', mapValueToColor);
  75. node.setVisual('style', {
  76. fill: mapValueToColor
  77. });
  78. }
  79. });
  80. }
  81. if (edges.length) {
  82. zrUtil.each(edges, function (edge) {
  83. var edgeStyle = edge.getModel().get('lineStyle');
  84. edge.setVisual('style', edgeStyle);
  85. });
  86. }
  87. });
  88. }