@visactor/vgrammar-core/es/transforms/data/unfold.js

VGrammar is a visual grammar library

import { array, maxInArray, minInArray } from "@visactor/vutils";

const aggregateFuncs = {
    sum: arr => arr.reduce(((sum, val) => sum + val), 0),
    min: arr => minInArray(arr),
    max: arr => maxInArray(arr),
    count: arr => arr.length,
    mean: arr => arr.reduce(((sum, val) => sum + val), 0) / arr.length
};

export const transform = (options, upstreamData) => {
    var _a;
    if (!upstreamData || !upstreamData.length) return [];
    const res = [], groups = {}, keyField = options.keyField, valueField = options.valueField, aggregate = aggregateFuncs[null !== (_a = options.aggregateType) && void 0 !== _a ? _a : "sum"];
    if (options.groupBy) {
        const groupByFields = array(options.groupBy);
        upstreamData.forEach((entry => {
            if (!entry) return;
            const datum = {}, keys = [];
            groupByFields.forEach((field => {
                datum[field] = entry[field], keys.push(entry[field]);
            }));
            const groupKey = keys.join("~");
            groups[groupKey] ? groups[groupKey].values[entry[keyField]] ? groups[groupKey].values[entry[keyField]].push(entry[valueField]) : groups[groupKey].values[entry[keyField]] = [ entry[valueField] ] : groups[groupKey] = {
                datum: datum,
                values: {
                    [entry[keyField]]: [ entry[valueField] ]
                }
            };
        }));
    } else groups[0] = {
        datum: {},
        values: {}
    }, upstreamData.forEach((entry => {
        entry && (groups[0].values[entry[keyField]] ? groups[0].values[entry[keyField]].push(entry[valueField]) : groups[0].values[entry[keyField]] = [ entry[valueField] ]);
    }));
    return Object.keys(groups).forEach((groupKey => {
        const datum = groups[groupKey].datum, values = groups[groupKey].values;
        Object.keys(values).forEach((key => {
            const rows = values[key];
            datum[key] = aggregate(rows);
        })), res.push(datum);
    })), res;
};
//# sourceMappingURL=unfold.js.map