sankeyVisual.js 3.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  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. if (nodes.length) {
  47. var minValue_1 = Infinity;
  48. var maxValue_1 = -Infinity;
  49. zrUtil.each(nodes, function (node) {
  50. var nodeValue = node.getLayout().value;
  51. if (nodeValue < minValue_1) {
  52. minValue_1 = nodeValue;
  53. }
  54. if (nodeValue > maxValue_1) {
  55. maxValue_1 = nodeValue;
  56. }
  57. });
  58. zrUtil.each(nodes, function (node) {
  59. var mapping = new VisualMapping({
  60. type: 'color',
  61. mappingMethod: 'linear',
  62. dataExtent: [minValue_1, maxValue_1],
  63. visual: seriesModel.get('color')
  64. });
  65. var mapValueToColor = mapping.mapValueToVisual(node.getLayout().value);
  66. var customColor = node.getModel().get(['itemStyle', 'color']);
  67. if (customColor != null) {
  68. node.setVisual('color', customColor);
  69. node.setVisual('style', {
  70. fill: customColor
  71. });
  72. } else {
  73. node.setVisual('color', mapValueToColor);
  74. node.setVisual('style', {
  75. fill: mapValueToColor
  76. });
  77. }
  78. });
  79. }
  80. });
  81. }