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

the Grammar of Graphics in Javascript

import { deepMix } from '@antv/util';
import { subObject } from '../utils/helper';
import { Tree as TreeTransform, } from '../data/tree';
import { maybeAnimation, subTooltip } from '../utils/mark';
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,
};
export const Tree = (options) => {
    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