@antv/g2plot
Version:
An interactive and responsive charting library
58 lines • 2.54 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.transformViolinData = exports.toViolinValue = exports.toBoxValue = void 0;
var tslib_1 = require("tslib");
var util_1 = require("@antv/util");
var pdfast_1 = tslib_1.__importDefault(require("pdfast"));
var quantile_1 = require("../../utils/transform/quantile");
exports.toBoxValue = function (values) {
return {
low: util_1.min(values),
high: util_1.max(values),
q1: quantile_1.quantile(values, 0.25),
q3: quantile_1.quantile(values, 0.75),
median: quantile_1.quantile(values, [0.5]),
minMax: [util_1.min(values), util_1.max(values)],
quantile: [quantile_1.quantile(values, 0.25), quantile_1.quantile(values, 0.75)],
};
};
exports.toViolinValue = function (values, pdfOptions) {
var pdfResults = pdfast_1.default.create(values, pdfOptions);
return {
violinSize: pdfResults.map(function (result) { return result.y; }),
violinY: pdfResults.map(function (result) { return result.x; }),
};
};
exports.transformViolinData = function (options) {
var xField = options.xField, yField = options.yField, seriesField = options.seriesField, data = options.data, kde = options.kde;
/** 生成概率密度函数的配置 */
var pdfOptions = {
min: kde.min,
max: kde.max,
size: kde.sampleSize,
width: kde.width,
};
// 无拆分
if (!seriesField) {
var group_1 = util_1.groupBy(data, xField);
return Object.keys(group_1).map(function (x) {
var records = group_1[x];
var values = records.map(function (record) { return record[yField]; });
return tslib_1.__assign(tslib_1.__assign({ x: x }, exports.toViolinValue(values, pdfOptions)), exports.toBoxValue(values));
});
}
// 有拆分
var resultList = [];
var seriesGroup = util_1.groupBy(data, seriesField);
Object.keys(seriesGroup).forEach(function (series) {
var group = util_1.groupBy(seriesGroup[series], xField);
return Object.keys(group).forEach(function (key) {
var _a;
var records = group[key];
var values = records.map(function (record) { return record[yField]; });
resultList.push(tslib_1.__assign(tslib_1.__assign((_a = { x: key }, _a[seriesField] = series, _a), exports.toViolinValue(values, pdfOptions)), exports.toBoxValue(values)));
});
});
return resultList;
};
//# sourceMappingURL=utils.js.map