@antv/g2/esm/transform/filter.js

the Grammar of Graphics in Javascript

import { deepMix } from '@antv/util';
import { defined } from '../utils/helper';
import { columnOf } from './utils/helper';
function normalizeValue(value) {
    if (typeof value === 'object')
        return [value.value, value.ordinal];
    else
        return [value, true];
}
/**
 * The Filter transform filter channels.
 */
export const Filter = (options = {}) => {
    return (I, mark) => {
        const { encode, data } = mark;
        const filters = Object.entries(options)
            .map(([key, v]) => {
            const [V] = columnOf(encode, key);
            // Skip empty channel.
            if (!V)
                return null;
            const [value, ordinal = true] = normalizeValue(v);
            if (typeof value === 'function')
                return (i) => value(V[i]);
            if (ordinal) {
                const expectedValues = Array.isArray(value) ? value : [value];
                // Skip empty expected values.
                if (expectedValues.length === 0)
                    return null;
                return (i) => expectedValues.includes(V[i]);
            }
            else {
                const [start, end] = value;
                return (i) => V[i] >= start && V[i] <= end;
            }
        })
            .filter(defined);
        // Skip empty filters.
        if (filters.length === 0)
            return [I, mark];
        // Filter index and channels.
        const totalFilter = (i) => filters.every((f) => f(i));
        const FI = I.filter(totalFilter);
        const newIndex = FI.map((_, i) => i);
        const newEncodes = Object.entries(encode).map(([key, encode]) => {
            return [
                key,
                Object.assign(Object.assign({}, encode), { value: newIndex
                        .map((i) => encode.value[FI[i]])
                        .filter((v) => v !== undefined) }),
            ];
        });
        return [
            newIndex,
            deepMix({}, mark, {
                encode: Object.fromEntries(newEncodes),
                // Filter data for tooltip item.
                data: FI.map((i) => data[i]),
            }),
        ];
    };
};
Filter.props = {};
//# sourceMappingURL=filter.js.map