swOperation/node_modules/@antv/g2/lib/mark/sankey.js

"use strict";
var __rest = (this && this.__rest) || function (s, e) {
    var t = {};
    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
        t[p] = s[p];
    if (s != null && typeof Object.getOwnPropertySymbols === "function")
        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
                t[p[i]] = s[p[i]];
        }
    return t;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.Sankey = void 0;
const util_1 = require("@antv/util");
const sankey_1 = require("../data/sankey");
const helper_1 = require("../utils/helper");
const mark_1 = require("../utils/mark");
const utils_1 = require("./utils");
/**
 * @todo Add interaction
 * @todo Add source-link color mode
 */
const Sankey = (options) => {
    const DEFAULT_LAYOUT_OPTIONS = {
        nodeId: (d) => d.key,
        nodeWidth: 0.02,
        nodePadding: 0.02,
    };
    const DEFAULT_NODE_OPTIONS = {
        type: 'polygon',
        axis: false,
        legend: false,
        encode: {
            shape: 'polygon',
            x: 'x',
            y: 'y',
        },
        scale: {
            x: { type: 'identity' },
            y: { type: 'identity' },
        },
        style: {
            stroke: '#000',
        },
    };
    const DEFAULT_LINK_OPTIONS = {
        type: 'polygon',
        axis: false,
        legend: false,
        encode: {
            shape: 'ribbon',
            x: 'x',
            y: 'y',
        },
        style: {
            fillOpacity: 0.5,
            stroke: undefined,
        },
    };
    const DEFAULT_LABEL_OPTIONS = {
        textAlign: (d) => (d.x[0] < 0.5 ? 'start' : 'end'),
        position: (d) => (d.x[0] < 0.5 ? 'right' : 'left'),
        fontSize: 10,
    };
    return () => {
        const { data, encode = {}, scale, style = {}, layout = {}, nodeLabels = [], linkLabels = [], animate = {}, tooltip = {}, } = options;
        // Initialize data, generating nodes by link if is not specified.
        const { links, nodes } = (0, utils_1.initializeData)(data, encode);
        // Extract encode for node and link.
        const nodeEncode = (0, helper_1.subObject)(encode, 'node');
        const linkEncode = (0, helper_1.subObject)(encode, 'link');
        const { key: nodeKey = (d) => d.key, color = nodeKey } = nodeEncode;
        // Transform data, using nodeKey as nodeId.
        const { links: linkData, nodes: nodeData } = (0, sankey_1.Sankey)(Object.assign(Object.assign(Object.assign({}, DEFAULT_LAYOUT_OPTIONS), { nodeId: (0, utils_1.field)(nodeKey) }), layout))({ links, nodes });
        // Extract label style and apply defaults.
        const _a = (0, helper_1.subObject)(style, 'label'), { text = nodeKey, spacing = 5 } = _a, labelStyle = __rest(_a, ["text", "spacing"]);
        const key1 = (0, utils_1.field)(nodeKey);
        const nodeTooltip = (0, mark_1.subTooltip)(tooltip, 'node', {
            title: key1,
            items: [{ field: 'value' }],
        }, true);
        const linkTooltip = (0, mark_1.subTooltip)(tooltip, 'link', {
            title: '',
            items: [
                (d) => ({ name: 'source', value: key1(d.source) }),
                (d) => ({ name: 'target', value: key1(d.target) }),
            ],
        });
        return [
            (0, util_1.deepMix)({}, DEFAULT_NODE_OPTIONS, {
                data: nodeData,
                encode: Object.assign(Object.assign({}, nodeEncode), { color }),
                scale,
                style: (0, helper_1.subObject)(style, 'node'),
                labels: [
                    Object.assign(Object.assign(Object.assign({}, DEFAULT_LABEL_OPTIONS), { text, dx: (d) => (d.x[0] < 0.5 ? spacing : -spacing) }), labelStyle),
                    ...nodeLabels,
                ],
                tooltip: nodeTooltip,
                animate: (0, mark_1.maybeAnimation)(animate, 'node'),
                axis: false,
            }),
            (0, util_1.deepMix)({}, DEFAULT_LINK_OPTIONS, {
                data: linkData,
                encode: linkEncode,
                labels: linkLabels,
                style: Object.assign({ fill: linkEncode.color ? undefined : '#aaa', strokeWidth: 0 }, (0, helper_1.subObject)(style, 'link')),
                tooltip: linkTooltip,
                animate: (0, mark_1.maybeAnimation)(animate, 'link'),
            }),
        ];
    };
};
exports.Sankey = Sankey;
exports.Sankey.props = {};
//# sourceMappingURL=sankey.js.map