1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- import * as zrUtil from 'zrender/lib/core/util.js';
- import VisualMapping from '../../visual/VisualMapping.js';
- export default function sankeyVisual(ecModel) {
- ecModel.eachSeriesByType('sankey', function (seriesModel) {
- var graph = seriesModel.getGraph();
- var nodes = graph.nodes;
- var edges = graph.edges;
- if (nodes.length) {
- var minValue_1 = Infinity;
- var maxValue_1 = -Infinity;
- zrUtil.each(nodes, function (node) {
- var nodeValue = node.getLayout().value;
- if (nodeValue < minValue_1) {
- minValue_1 = nodeValue;
- }
- if (nodeValue > maxValue_1) {
- maxValue_1 = nodeValue;
- }
- });
- zrUtil.each(nodes, function (node) {
- var mapping = new VisualMapping({
- type: 'color',
- mappingMethod: 'linear',
- dataExtent: [minValue_1, maxValue_1],
- visual: seriesModel.get('color')
- });
- var mapValueToColor = mapping.mapValueToVisual(node.getLayout().value);
- var customColor = node.getModel().get(['itemStyle', 'color']);
- if (customColor != null) {
- node.setVisual('color', customColor);
- node.setVisual('style', {
- fill: customColor
- });
- } else {
- node.setVisual('color', mapValueToColor);
- node.setVisual('style', {
- fill: mapValueToColor
- });
- }
- });
- }
- if (edges.length) {
- zrUtil.each(edges, function (edge) {
- var edgeStyle = edge.getModel().get('lineStyle');
- edge.setVisual('style', edgeStyle);
- });
- }
- });
- }
|