swOperation/node_modules/@antv/g2/esm/mark/tree.js

import { deepMix } from '@antv/util';
import { subObject } from '../utils/helper';
import { Tree as TreeTransform, } from '../data/tree';
import { maybeAnimation, subTooltip } from '../utils/mark';
export const Tree = (options) => {
    const DEFAULT_LAYOUT_OPTIONS = {
        sortBy: (a, b) => b.value - a.value,
    };
    const DEFAULT_NODE_OPTIONS = {
        axis: false,
        legend: false,
        type: 'point',
        encode: {
            x: 'x',
            y: 'y',
            size: 2,
            shape: 'point',
        },
    };
    const DEFAULT_LINK_OPTIONS = {
        type: 'link',
        encode: {
            x: 'x',
            y: 'y',
            shape: 'smooth',
        },
    };
    const DEFAULT_LABEL_OPTIONS = {
        text: '',
        fontSize: 10,
    };
    return () => {
        const { data, encode = {}, scale = {}, style = {}, layout = {}, nodeLabels = [], linkLabels = [], animate = {}, tooltip = {}, } = options;
        const valueEncode = encode === null || encode === void 0 ? void 0 : encode.value;
        const { nodes, edges } = TreeTransform(Object.assign(Object.assign(Object.assign({}, DEFAULT_LAYOUT_OPTIONS), layout), { field: valueEncode }))(data);
        const nodeTooltip = subTooltip(tooltip, 'node', {
            title: 'name',
            items: ['value'],
        }, true);
        const linkTooltip = subTooltip(tooltip, 'link', {
            title: '',
            items: [
                (d) => ({ name: 'source', value: d.source.name }),
                (d) => ({ name: 'target', value: d.target.name }),
            ],
        });
        return [
            deepMix({}, DEFAULT_LINK_OPTIONS, {
                data: edges,
                encode: subObject(encode, 'link'),
                scale: subObject(scale, 'link'),
                labels: linkLabels,
                style: Object.assign({ stroke: '#999' }, subObject(style, 'link')),
                tooltip: linkTooltip,
                animate: maybeAnimation(animate, 'link'),
            }),
            deepMix({}, DEFAULT_NODE_OPTIONS, {
                data: nodes,
                scale: subObject(scale, 'node'),
                encode: subObject(encode, 'node'),
                labels: [
                    Object.assign(Object.assign({}, DEFAULT_LABEL_OPTIONS), subObject(style, 'label')),
                    ...nodeLabels,
                ],
                style: Object.assign({}, subObject(style, 'node')),
                tooltip: nodeTooltip,
                animate: maybeAnimation(animate, 'node'),
            }),
        ];
    };
};
Tree.props = {};
//# sourceMappingURL=tree.js.map