@antv/g2
Version:
the Grammar of Graphics in Javascript
100 lines • 3.98 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.StackY = void 0;
const util_1 = require("@antv/util");
const helper_1 = require("./utils/helper");
const order_1 = require("./utils/order");
/**
* The stack transform group marks into series by color channel,
* and then produce new y channel for each series by specified order,
* say to form vertical "stacks" by specified channels.
*/
const StackY = (options = {}) => {
const { groupBy = 'x', orderBy = null, reverse = false, y: fromY = 'y', y1: fromY1 = 'y1', series = true, } = options;
return (I, mark) => {
var _a;
const { data, encode, style = {} } = mark;
const [Y, fy] = (0, helper_1.columnOf)(encode, 'y');
const [Y1, fy1] = (0, helper_1.columnOf)(encode, 'y1');
const [S] = series
? (0, helper_1.maybeColumnOf)(encode, 'series', 'color')
: (0, helper_1.columnOf)(encode, 'color');
// Create groups and apply specified order for each group.
const groups = (0, order_1.createGroups)(groupBy, I, mark);
const createComparator = (_a = (0, order_1.normalizeComparator)(orderBy)) !== null && _a !== void 0 ? _a : (() => null);
const comparator = createComparator(data, Y, S);
if (comparator)
(0, order_1.applyOrder)(groups, comparator);
// Stack y channels to produce new y and y1 channel.
const newY = new Array(I.length);
const newY1 = new Array(I.length);
const TY = new Array(I.length);
const F = [];
const L = [];
for (const G of groups) {
if (reverse)
G.reverse();
// For range interval with specified y and y1.
const start = Y1 ? +Y1[G[0]] : 0;
// Split positive indices of Y and negative Y.
const PG = [];
const NG = [];
for (const i of G) {
const y = (TY[i] = +Y[i] - start);
if (y < 0)
NG.push(i);
else if (y >= 0)
PG.push(i);
}
// Store the first and last layer.
const FG = PG.length > 0 ? PG : NG;
const LG = NG.length > 0 ? NG : PG;
let i = PG.length - 1;
let j = 0;
// Find the last non-zero index.
while (i > 0 && Y[FG[i]] === 0)
i--;
// Find the first non-zero index.
while (j < LG.length - 1 && Y[LG[j]] === 0)
j++;
F.push(FG[i]);
L.push(LG[j]);
// Stack negative y in reverse order.
let ny = start;
for (const i of NG.reverse()) {
const y = TY[i];
ny = newY[i] = (newY1[i] = ny) + y;
}
// Stack positive y in input order.
let py = start;
for (const i of PG) {
const y = TY[i];
if (y > 0)
py = newY[i] = (newY1[i] = py) + y;
else
newY[i] = newY1[i] = py;
}
}
// Only set top radius for the first layer,
// and set bottom radius for the last layer.
const FS = new Set(F);
const LS = new Set(L);
// Choose new y or y1 channel as the new y channel.
const V = fromY === 'y' ? newY : newY1;
const V1 = fromY1 === 'y' ? newY : newY1;
return [
I,
(0, util_1.deepMix)({}, mark, {
encode: {
y0: (0, helper_1.inferredColumn)(Y, fy),
y: (0, helper_1.column)(V, fy),
y1: (0, helper_1.column)(V1, fy1),
},
style: Object.assign({ first: (_, i) => FS.has(i), last: (_, i) => LS.has(i) }, style),
}),
];
};
};
exports.StackY = StackY;
exports.StackY.props = {};
//# sourceMappingURL=stackY.js.map