123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157 |
- import * as zrUtil from 'zrender/lib/core/util.js';
- import * as numberUtil from '../../util/number.js';
- export default function themeRiverLayout(ecModel, api) {
- ecModel.eachSeriesByType('themeRiver', function (seriesModel) {
- var data = seriesModel.getData();
- var single = seriesModel.coordinateSystem;
- var layoutInfo = {};
-
- var rect = single.getRect();
- layoutInfo.rect = rect;
- var boundaryGap = seriesModel.get('boundaryGap');
- var axis = single.getAxis();
- layoutInfo.boundaryGap = boundaryGap;
- if (axis.orient === 'horizontal') {
- boundaryGap[0] = numberUtil.parsePercent(boundaryGap[0], rect.height);
- boundaryGap[1] = numberUtil.parsePercent(boundaryGap[1], rect.height);
- var height = rect.height - boundaryGap[0] - boundaryGap[1];
- doThemeRiverLayout(data, seriesModel, height);
- } else {
- boundaryGap[0] = numberUtil.parsePercent(boundaryGap[0], rect.width);
- boundaryGap[1] = numberUtil.parsePercent(boundaryGap[1], rect.width);
- var width = rect.width - boundaryGap[0] - boundaryGap[1];
- doThemeRiverLayout(data, seriesModel, width);
- }
- data.setLayout('layoutInfo', layoutInfo);
- });
- }
- function doThemeRiverLayout(data, seriesModel, height) {
- if (!data.count()) {
- return;
- }
- var coordSys = seriesModel.coordinateSystem;
-
- var layerSeries = seriesModel.getLayerSeries();
-
- var timeDim = data.mapDimension('single');
- var valueDim = data.mapDimension('value');
- var layerPoints = zrUtil.map(layerSeries, function (singleLayer) {
- return zrUtil.map(singleLayer.indices, function (idx) {
- var pt = coordSys.dataToPoint(data.get(timeDim, idx));
- pt[1] = data.get(valueDim, idx);
- return pt;
- });
- });
- var base = computeBaseline(layerPoints);
- var baseLine = base.y0;
- var ky = height / base.max;
-
- var n = layerSeries.length;
- var m = layerSeries[0].indices.length;
- var baseY0;
- for (var j = 0; j < m; ++j) {
- baseY0 = baseLine[j] * ky;
- data.setItemLayout(layerSeries[0].indices[j], {
- layerIndex: 0,
- x: layerPoints[0][j][0],
- y0: baseY0,
- y: layerPoints[0][j][1] * ky
- });
- for (var i = 1; i < n; ++i) {
- baseY0 += layerPoints[i - 1][j][1] * ky;
- data.setItemLayout(layerSeries[i].indices[j], {
- layerIndex: i,
- x: layerPoints[i][j][0],
- y0: baseY0,
- y: layerPoints[i][j][1] * ky
- });
- }
- }
- }
- function computeBaseline(data) {
- var layerNum = data.length;
- var pointNum = data[0].length;
- var sums = [];
- var y0 = [];
- var max = 0;
- for (var i = 0; i < pointNum; ++i) {
- var temp = 0;
- for (var j = 0; j < layerNum; ++j) {
- temp += data[j][i][1];
- }
- if (temp > max) {
- max = temp;
- }
- sums.push(temp);
- }
- for (var k = 0; k < pointNum; ++k) {
- y0[k] = (max - sums[k]) / 2;
- }
- max = 0;
- for (var l = 0; l < pointNum; ++l) {
- var sum = sums[l] + y0[l];
- if (sum > max) {
- max = sum;
- }
- }
- return {
- y0: y0,
- max: max
- };
- }
|