sankeyVisual.js 3.4 KB

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