@visactor/vgrammar-plot 0.6.0

package/es/rule.js

@@ -0,0 +1,45 @@
+import { SemanticMark } from "./semantic-mark";
+
+import { getPalette, GrammarMarkType } from "@visactor/vgrammar";
+
+import { isArray } from "@visactor/vutils";
+
+import { PlotMakType } from "./enums";
+
+export class Rule extends SemanticMark {
+    constructor(id) {
+        super(PlotMakType.rule, id);
+    }
+    setMarkType() {
+        return GrammarMarkType.rule;
+    }
+    parseScaleOfEncodeX(option) {
+        const res = super.parseScaleOfEncodeX(option);
+        return res.type = "point", res;
+    }
+    parseScaleByEncode(channel, option) {
+        return this.parseScaleOfCommonEncode(channel, option);
+    }
+    convertMarkEncode(encode) {
+        var _a, _b, _c, _d, _e;
+        const markEncoder = this.convertSimpleMarkEncode(encode);
+        let res;
+        return isArray(null === (_a = markEncoder.x) || void 0 === _a ? void 0 : _a.field) ? (res.x = {
+            field: markEncoder.x.field[0],
+            scale: markEncoder.x.scale
+        }, res.x1 = {
+            field: markEncoder.x.field[1],
+            scale: markEncoder.x.scale
+        }) : res.x = res.x1 = markEncoder.x, isArray(null === (_b = markEncoder.y) || void 0 === _b ? void 0 : _b.field) ? (res.y = {
+            field: markEncoder.y.field[0],
+            scale: markEncoder.y.scale
+        }, res.y1 = {
+            field: markEncoder.y.field[1],
+            scale: markEncoder.y.scale
+        }) : res.y = res.y1 = markEncoder.y, markEncoder.color || markEncoder.group ? res.stroke = null !== (_c = markEncoder.color) && void 0 !== _c ? _c : markEncoder.group : res.stroke = null !== (_e = null === (_d = this.spec.style) || void 0 === _d ? void 0 : _d.stroke) && void 0 !== _e ? _e : getPalette()[0], 
+        res;
+    }
+}
+
+Rule.type = PlotMakType.rule;
+//# sourceMappingURL=rule.js.map

package/es/rule.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/rule.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAEtC,MAAM,OAAO,IAAK,SAAQ,YAAqD;IAE7E,YAAY,EAAoB;QAC9B,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC9B,CAAC;IAED,WAAW;QACT,OAAO,eAAe,CAAC,IAAI,CAAC;IAC9B,CAAC;IAES,mBAAmB,CAC3B,MAA+F;QAE/F,MAAM,GAAG,GAAG,KAAK,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAE9C,GAAG,CAAC,IAAI,GAAG,OAAO,CAAC;QACnB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,kBAAkB,CAChB,OAA2B,EAC3B,MAA+F;QAE/F,OAAO,IAAI,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACxD,CAAC;IAED,iBAAiB,CACf,MAAkE;;QAElE,MAAM,WAAW,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;QAEzD,IAAI,GAA6C,CAAC;QAElD,IAAI,OAAO,CAAC,MAAA,WAAW,CAAC,CAAC,0CAAE,KAAK,CAAC,EAAE;YACjC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;YACtE,GAAG,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;SACxE;aAAM;YACL,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC;SAChC;QAED,IAAI,OAAO,CAAC,MAAA,WAAW,CAAC,CAAC,0CAAE,KAAK,CAAC,EAAE;YACjC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;YACtE,GAAG,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;SACxE;aAAM;YACL,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC;SAChC;QAED,IAAI,WAAW,CAAC,KAAK,IAAI,WAAW,CAAC,KAAK,EAAE;YAC1C,GAAG,CAAC,MAAM,GAAG,MAAA,WAAW,CAAC,KAAK,mCAAI,WAAW,CAAC,KAAK,CAAC;SACrD;aAAM;YACL,GAAG,CAAC,MAAM,GAAG,MAAA,MAAA,IAAI,CAAC,IAAI,CAAC,KAAK,0CAAE,MAAM,mCAAI,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;SACzD;QAED,OAAO,GAAG,CAAC;IACb,CAAC;;AArDe,SAAI,GAAG,WAAW,CAAC,IAAI,CAAC","file":"rule.js","sourcesContent":["import type {\n  RuleEncodeChannels,\n  WithDefaultEncode,\n  PlotRuleEncoderSpec,\n  GenerateBaseEncodeSpec,\n  GenerateEncoderSpec,\n  Nil,\n  ScaleSpec,\n  ValueOf\n} from '@visactor/vgrammar';\nimport { SemanticMark } from './semantic-mark';\nimport { getPalette, GrammarMarkType } from '@visactor/vgrammar';\nimport { isArray } from '@visactor/vutils';\nimport { PlotMakType } from './enums';\n\nexport class Rule extends SemanticMark<PlotRuleEncoderSpec, RuleEncodeChannels> {\n  static readonly type = PlotMakType.rule;\n  constructor(id?: string | number) {\n    super(PlotMakType.rule, id);\n  }\n\n  setMarkType() {\n    return GrammarMarkType.rule;\n  }\n\n  protected parseScaleOfEncodeX(\n    option: ValueOf<WithDefaultEncode<PlotRuleEncoderSpec, RuleEncodeChannels>, RuleEncodeChannels>\n  ): ScaleSpec | Nil {\n    const res = super.parseScaleOfEncodeX(option);\n\n    res.type = 'point';\n    return res;\n  }\n\n  parseScaleByEncode(\n    channel: RuleEncodeChannels,\n    option: ValueOf<WithDefaultEncode<PlotRuleEncoderSpec, RuleEncodeChannels>, RuleEncodeChannels>\n  ): ScaleSpec | Nil {\n    return this.parseScaleOfCommonEncode(channel, option);\n  }\n\n  convertMarkEncode(\n    encode: WithDefaultEncode<PlotRuleEncoderSpec, RuleEncodeChannels>\n  ): GenerateBaseEncodeSpec<PlotRuleEncoderSpec> {\n    const markEncoder = this.convertSimpleMarkEncode(encode);\n\n    let res: GenerateEncoderSpec<PlotRuleEncoderSpec>;\n\n    if (isArray(markEncoder.x?.field)) {\n      res.x = { field: markEncoder.x.field[0], scale: markEncoder.x.scale };\n      res.x1 = { field: markEncoder.x.field[1], scale: markEncoder.x.scale };\n    } else {\n      res.x = res.x1 = markEncoder.x;\n    }\n\n    if (isArray(markEncoder.y?.field)) {\n      res.y = { field: markEncoder.y.field[0], scale: markEncoder.y.scale };\n      res.y1 = { field: markEncoder.y.field[1], scale: markEncoder.y.scale };\n    } else {\n      res.y = res.y1 = markEncoder.y;\n    }\n\n    if (markEncoder.color || markEncoder.group) {\n      res.stroke = markEncoder.color ?? markEncoder.group;\n    } else {\n      res.stroke = this.spec.style?.stroke ?? getPalette()[0];\n    }\n\n    return res;\n  }\n}\n"]}

package/es/sankey.d.ts

@@ -0,0 +1,46 @@
+import type { SankeyEncodeChannels, WithDefaultEncode, PlotSankeyEncoderSpec, SemanticLabelOption, ChannelEncodeType, DataSpec, GenerateBaseEncodeSpec, Nil, ScaleSpec, ValueOf, TransformSpec, IElement } from '@visactor/vgrammar';
+import { SemanticMark } from './semantic-mark';
+import { GrammarMarkType } from '@visactor/vgrammar';
+import { PlotMakType } from './enums';
+import type { ITextAttribute } from '@visactor/vrender';
+export declare class SankeySemanticMark extends SemanticMark<PlotSankeyEncoderSpec, SankeyEncodeChannels> {
+    static readonly type = PlotMakType.sankey;
+    constructor(id?: string | number);
+    setMarkType(): GrammarMarkType;
+    setDefaultDataTransform(): TransformSpec[];
+    getDataIdOfLink(): string;
+    setMultipleData(): DataSpec[];
+    parseScaleByEncode(channel: SankeyEncodeChannels, option: ValueOf<WithDefaultEncode<PlotSankeyEncoderSpec, SankeyEncodeChannels>, SankeyEncodeChannels>): ScaleSpec | Nil;
+    convertMarkEncode(encode: WithDefaultEncode<PlotSankeyEncoderSpec, SankeyEncodeChannels>): GenerateBaseEncodeSpec<PlotSankeyEncoderSpec>;
+    setMainMarkSpec(): {
+        key: string;
+    };
+    protected setLabelTextGetter(channel: string, option: SemanticLabelOption | boolean): ChannelEncodeType<ITextAttribute['text']>;
+    setMultiMarksSpec(): {
+        id: string;
+        type: string;
+        glyphType: string;
+        from: {
+            data: string;
+        };
+        layout: {
+            position: string;
+            skipBeforeLayouted: boolean;
+        };
+        key: string;
+        dependency: string[];
+        animation: import("@visactor/vgrammar").MarkAnimationSpec;
+        encode: Record<string, Partial<Partial<import("@visactor/vgrammar").LinkPathEncoderSpec>>> & {
+            enter: any;
+            update: (datum: any, el: IElement, params: any) => {
+                direction: string;
+                x0: any;
+                x1: any;
+                y0: any;
+                y1: any;
+                thickness: any;
+                fill: any;
+            };
+        };
+    }[];
+}

package/es/sankey.js

@@ -0,0 +1,137 @@
+import { SemanticMark } from "./semantic-mark";
+
+import { getPalette, GrammarMarkType, SIGNAL_VIEW_BOX, getTransform, getGlyph } from "@visactor/vgrammar";
+
+import { PlotMakType } from "./enums";
+
+import { field as getFieldAccessor } from "@visactor/vgrammar-util";
+
+export class SankeySemanticMark extends SemanticMark {
+    constructor(id) {
+        super(PlotMakType.sankey, id), getTransform(PlotMakType.sankey) || this._logger.error("Please add this line of code: import { registerSankeyTransforms } from '@visactor/vgrammar-sankey'; \n        and run registerSankeyTransforms() before use sankey"), 
+        getGlyph("linkPath") || this._logger.error("\n      Please add this line of code: import { registerLinkPathGlyph } from '@visactor/vgrammar';\n      add run registerLinkPathGlyph() before use sankey\n      ");
+    }
+    setMarkType() {
+        return GrammarMarkType.rect;
+    }
+    setDefaultDataTransform() {
+        var _a;
+        return [ {
+            type: "sankey",
+            width: {
+                signal: "viewWidth"
+            },
+            height: {
+                signal: "viewHeight"
+            },
+            nodeKey: null === (_a = this.spec.encode) || void 0 === _a ? void 0 : _a.node
+        } ];
+    }
+    getDataIdOfLink() {
+        var _a, _b;
+        return `${null !== (_b = null === (_a = this.spec.data) || void 0 === _a ? void 0 : _a.id) && void 0 !== _b ? _b : this.spec.id}-data-link`;
+    }
+    setMultipleData() {
+        return [ {
+            id: this.getDataIdOfFiltered(),
+            transform: [ {
+                type: "map",
+                all: !0,
+                callback: datum => datum[0].nodes
+            } ]
+        }, {
+            source: this.getDataIdOfMain(),
+            id: this.getDataIdOfLink(),
+            transform: [ {
+                type: "map",
+                all: !0,
+                callback: datum => datum[0].links
+            } ]
+        } ];
+    }
+    parseScaleByEncode(channel, option) {
+        return "color" === channel ? {
+            type: "ordinal",
+            id: this.getScaleId("color"),
+            domain: {
+                data: this.getDataIdOfFiltered(),
+                field: option
+            },
+            range: getPalette()
+        } : null;
+    }
+    convertMarkEncode(encode) {
+        var _a, _b, _c;
+        const markEncoder = this.convertSimpleMarkEncode(encode), res = {
+            x: {
+                field: "x0"
+            },
+            x1: {
+                field: "x1"
+            },
+            y: {
+                field: "y0"
+            },
+            y1: {
+                field: "y1"
+            }
+        };
+        if (markEncoder.color) {
+            const scaleColorId = this.getScaleId("color"), colorAccessor = getFieldAccessor(markEncoder.color.field);
+            res.fill = (datum, el, params) => params[scaleColorId].scale(colorAccessor(null == datum ? void 0 : datum.datum));
+        } else res.fill = null !== (_c = null === (_b = null === (_a = this.spec.style) || void 0 === _a ? void 0 : _a.nodeStyle) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : getPalette()[0];
+        return res;
+    }
+    setMainMarkSpec() {
+        return {
+            key: "key"
+        };
+    }
+    setLabelTextGetter(channel, option) {
+        const textGetter = getFieldAccessor(channel);
+        return (datum, el, params) => textGetter(datum.datum);
+    }
+    setMultiMarksSpec() {
+        var _a;
+        return [ {
+            id: `${this.getMarkId()}-link`,
+            type: "glyph",
+            glyphType: "linkPath",
+            from: {
+                data: this.getDataIdOfLink()
+            },
+            layout: {
+                position: "content",
+                skipBeforeLayouted: !0
+            },
+            key: "index",
+            dependency: this.viewSpec.scales.map((scale => scale.id)).concat(SIGNAL_VIEW_BOX),
+            animation: this.convertMarkAnimation(),
+            encode: Object.assign({}, this.spec.state, {
+                enter: Object.assign({
+                    backgroundStyle: {
+                        fillColor: "#ccc",
+                        fillOpacity: .2
+                    },
+                    fillOpacity: .8,
+                    round: !0
+                }, null === (_a = this.spec.style) || void 0 === _a ? void 0 : _a.linkStyle),
+                update: (datum, el, params) => {
+                    var _a, _b, _c;
+                    return {
+                        direction: datum.vertical ? "vertical" : "horizontal",
+                        x0: datum.x0,
+                        x1: datum.x1,
+                        y0: datum.y0,
+                        y1: datum.y1,
+                        thickness: datum.thickness,
+                        fill: null !== (_c = null === (_b = null === (_a = this.spec.style) || void 0 === _a ? void 0 : _a.linkStyle) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : getPalette()[0]
+                    };
+                }
+            })
+        } ];
+    }
+}
+
+SankeySemanticMark.type = PlotMakType.sankey;
+//# sourceMappingURL=sankey.js.map

package/es/sankey.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/sankey.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,eAAe,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC1G,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAAE,KAAK,IAAI,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAGpE,MAAM,OAAO,kBAAmB,SAAQ,YAAyD;IAE/F,YAAY,EAAoB;QAC9B,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAE9B,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE;YACrC,IAAI,CAAC,OAAO,CAAC,KAAK,CAChB;6DACqD,CACtD,CAAC;SACH;QAED,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE;YACzB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;;;OAGlB,CAAC,CAAC;SACJ;IACH,CAAC;IAED,WAAW;QACT,OAAO,eAAe,CAAC,IAAI,CAAC;IAC9B,CAAC;IAED,uBAAuB;;QACrB,OAAO;YACL;gBACE,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE;gBAC9B,MAAM,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE;gBAChC,OAAO,EAAE,MAAA,IAAI,CAAC,IAAI,CAAC,MAAM,0CAAE,IAAI;aAChC;SACF,CAAC;IACJ,CAAC;IAED,eAAe;;QACb,OAAO,GAAG,MAAA,MAAA,IAAI,CAAC,IAAI,CAAC,IAAI,0CAAE,EAAE,mCAAI,IAAI,CAAC,IAAI,CAAC,EAAE,YAAY,CAAC;IAC3D,CAAC;IAED,eAAe;QACb,OAAO;YACL;gBACE,EAAE,EAAE,IAAI,CAAC,mBAAmB,EAAE;gBAC9B,SAAS,EAAE;oBACT;wBACE,IAAI,EAAE,KAAK;wBACX,GAAG,EAAE,IAAI;wBACT,QAAQ,EAAE,CAAC,KAAU,EAAE,EAAE;4BACvB,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;wBACxB,CAAC;qBACF;iBACF;aACF;YACD;gBACE,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE;gBAC9B,EAAE,EAAE,IAAI,CAAC,eAAe,EAAE;gBAC1B,SAAS,EAAE;oBACT;wBACE,IAAI,EAAE,KAAK;wBACX,GAAG,EAAE,IAAI;wBACT,QAAQ,EAAE,CAAC,KAAU,EAAE,EAAE;4BAEvB,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;wBACxB,CAAC;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,kBAAkB,CAChB,OAA6B,EAC7B,MAAqG;QAErG,IAAI,OAAO,KAAK,OAAO,EAAE;YACvB,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC5B,MAAM,EAAE;oBACN,IAAI,EAAE,IAAI,CAAC,mBAAmB,EAAE;oBAChC,KAAK,EAAE,MAAgB;iBACxB;gBACD,KAAK,EAAE,UAAU,EAAE;aACpB,CAAC;SACH;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,iBAAiB,CACf,MAAsE;;QAEtE,MAAM,WAAW,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;QAEzD,MAAM,GAAG,GAA+C;YACtD,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;YAClB,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;YACnB,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;YAClB,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;SACpB,CAAC;QAEF,IAAI,WAAW,CAAC,KAAK,EAAE;YACrB,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,aAAa,GAAG,gBAAgB,CAAC,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAEhE,GAAG,CAAC,IAAI,GAAG,CAAC,KAAU,EAAE,EAAY,EAAE,MAAW,EAAE,EAAE;gBACnD,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;gBACnC,OAAO,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,CAAC,CAAC,CAAC;YAClD,CAAC,CAAC;SACH;aAAM;YACL,GAAG,CAAC,IAAI,GAAG,MAAA,MAAA,MAAA,IAAI,CAAC,IAAI,CAAC,KAAK,0CAAE,SAAS,0CAAE,IAAI,mCAAI,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;SAChE;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAED,eAAe;QACb,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC;IACxB,CAAC;IAES,kBAAkB,CAC1B,OAAe,EACf,MAAqC;QAErC,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC7C,OAAO,CAAC,KAAU,EAAE,EAAY,EAAE,MAAW,EAAE,EAAE;YAC/C,OAAO,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACjC,CAAC,CAAC;IACJ,CAAC;IAED,iBAAiB;;QAEf,OAAO;YACL;gBACE,EAAE,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,OAAO;gBAC9B,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,UAAU;gBACrB,IAAI,EAAE;oBACJ,IAAI,EAAE,IAAI,CAAC,eAAe,EAAE;iBAC7B;gBACD,MAAM,EAAE;oBACN,QAAQ,EAAE,SAAS;oBACnB,kBAAkB,EAAE,IAAI;iBACzB;gBACD,GAAG,EAAE,OAAO;gBACZ,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC;gBAC/E,SAAS,EAAE,IAAI,CAAC,oBAAoB,EAAE;gBACtC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;oBACzC,KAAK,EAAE,MAAM,CAAC,MAAM,CAClB;wBACE,eAAe,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE;wBACxD,WAAW,EAAE,GAAG;wBAChB,KAAK,EAAE,IAAI;qBACZ,EACD,MAAA,IAAI,CAAC,IAAI,CAAC,KAAK,0CAAE,SAAS,CAC3B;oBACD,MAAM,EAAE,CAAC,KAAU,EAAE,EAAY,EAAE,MAAW,EAAE,EAAE;;wBAChD,OAAO;4BACL,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,YAAY;4BACrD,EAAE,EAAE,KAAK,CAAC,EAAE;4BACZ,EAAE,EAAE,KAAK,CAAC,EAAE;4BACZ,EAAE,EAAE,KAAK,CAAC,EAAE;4BACZ,EAAE,EAAE,KAAK,CAAC,EAAE;4BACZ,SAAS,EAAE,KAAK,CAAC,SAAS;4BAC1B,IAAI,EAAE,MAAA,MAAA,MAAA,IAAI,CAAC,IAAI,CAAC,KAAK,0CAAE,SAAS,0CAAE,IAAI,mCAAI,UAAU,EAAE,CAAC,CAAC,CAAC;yBAC1D,CAAC;oBACJ,CAAC;iBACF,CAAC;aACH;SACF,CAAC;IACJ,CAAC;;AAzKe,uBAAI,GAAG,WAAW,CAAC,MAAM,CAAC","file":"sankey.js","sourcesContent":["import type {\n  SankeyEncodeChannels,\n  WithDefaultEncode,\n  PlotSankeyEncoderSpec,\n  SemanticLabelOption,\n  ChannelEncodeType,\n  DataSpec,\n  GenerateBaseEncodeSpec,\n  GenerateEncoderSpec,\n  Nil,\n  ScaleSpec,\n  ValueOf,\n  TransformSpec,\n  IElement\n} from '@visactor/vgrammar';\nimport { SemanticMark } from './semantic-mark';\nimport { getPalette, GrammarMarkType, SIGNAL_VIEW_BOX, getTransform, getGlyph } from '@visactor/vgrammar';\nimport { PlotMakType } from './enums';\nimport { field as getFieldAccessor } from '@visactor/vgrammar-util';\nimport type { ITextAttribute } from '@visactor/vrender';\n\nexport class SankeySemanticMark extends SemanticMark<PlotSankeyEncoderSpec, SankeyEncodeChannels> {\n  static readonly type = PlotMakType.sankey;\n  constructor(id?: string | number) {\n    super(PlotMakType.sankey, id);\n\n    if (!getTransform(PlotMakType.sankey)) {\n      this._logger.error(\n        `Please add this line of code: import { registerSankeyTransforms } from '@visactor/vgrammar-sankey'; \n        and run registerSankeyTransforms() before use sankey`\n      );\n    }\n\n    if (!getGlyph('linkPath')) {\n      this._logger.error(`\n      Please add this line of code: import { registerLinkPathGlyph } from '@visactor/vgrammar';\n      add run registerLinkPathGlyph() before use sankey\n      `);\n    }\n  }\n\n  setMarkType() {\n    return GrammarMarkType.rect;\n  }\n\n  setDefaultDataTransform(): TransformSpec[] {\n    return [\n      {\n        type: 'sankey',\n        width: { signal: 'viewWidth' },\n        height: { signal: 'viewHeight' },\n        nodeKey: this.spec.encode?.node\n      }\n    ];\n  }\n\n  getDataIdOfLink() {\n    return `${this.spec.data?.id ?? this.spec.id}-data-link`;\n  }\n\n  setMultipleData(): DataSpec[] {\n    return [\n      {\n        id: this.getDataIdOfFiltered(),\n        transform: [\n          {\n            type: 'map',\n            all: true,\n            callback: (datum: any) => {\n              return datum[0].nodes;\n            }\n          }\n        ]\n      },\n      {\n        source: this.getDataIdOfMain(),\n        id: this.getDataIdOfLink(),\n        transform: [\n          {\n            type: 'map',\n            all: true,\n            callback: (datum: any) => {\n              // return formatLinkPath(datum[0].links);\n              return datum[0].links;\n            }\n          }\n        ]\n      }\n    ];\n  }\n\n  parseScaleByEncode(\n    channel: SankeyEncodeChannels,\n    option: ValueOf<WithDefaultEncode<PlotSankeyEncoderSpec, SankeyEncodeChannels>, SankeyEncodeChannels>\n  ): ScaleSpec | Nil {\n    if (channel === 'color') {\n      return {\n        type: 'ordinal',\n        id: this.getScaleId('color'),\n        domain: {\n          data: this.getDataIdOfFiltered(),\n          field: option as string\n        },\n        range: getPalette()\n      };\n    }\n\n    return null;\n  }\n\n  convertMarkEncode(\n    encode: WithDefaultEncode<PlotSankeyEncoderSpec, SankeyEncodeChannels>\n  ): GenerateBaseEncodeSpec<PlotSankeyEncoderSpec> {\n    const markEncoder = this.convertSimpleMarkEncode(encode);\n\n    const res: GenerateEncoderSpec<PlotSankeyEncoderSpec> = {\n      x: { field: 'x0' },\n      x1: { field: 'x1' },\n      y: { field: 'y0' },\n      y1: { field: 'y1' }\n    };\n\n    if (markEncoder.color) {\n      const scaleColorId = this.getScaleId('color');\n      const colorAccessor = getFieldAccessor(markEncoder.color.field);\n\n      res.fill = (datum: any, el: IElement, params: any) => {\n        const scale = params[scaleColorId];\n        return scale.scale(colorAccessor(datum?.datum));\n      };\n    } else {\n      res.fill = this.spec.style?.nodeStyle?.fill ?? getPalette()[0];\n    }\n\n    return res;\n  }\n\n  setMainMarkSpec() {\n    return { key: 'key' };\n  }\n\n  protected setLabelTextGetter(\n    channel: string,\n    option: SemanticLabelOption | boolean\n  ): ChannelEncodeType<ITextAttribute['text']> {\n    const textGetter = getFieldAccessor(channel);\n    return (datum: any, el: IElement, params: any) => {\n      return textGetter(datum.datum);\n    };\n  }\n\n  setMultiMarksSpec() {\n    // TODO: optimize the default style\n    return [\n      {\n        id: `${this.getMarkId()}-link`,\n        type: 'glyph',\n        glyphType: 'linkPath',\n        from: {\n          data: this.getDataIdOfLink()\n        },\n        layout: {\n          position: 'content',\n          skipBeforeLayouted: true\n        },\n        key: 'index',\n        dependency: this.viewSpec.scales.map(scale => scale.id).concat(SIGNAL_VIEW_BOX),\n        animation: this.convertMarkAnimation(),\n        encode: Object.assign({}, this.spec.state, {\n          enter: Object.assign(\n            {\n              backgroundStyle: { fillColor: '#ccc', fillOpacity: 0.2 },\n              fillOpacity: 0.8,\n              round: true\n            },\n            this.spec.style?.linkStyle\n          ),\n          update: (datum: any, el: IElement, params: any) => {\n            return {\n              direction: datum.vertical ? 'vertical' : 'horizontal',\n              x0: datum.x0,\n              x1: datum.x1,\n              y0: datum.y0,\n              y1: datum.y1,\n              thickness: datum.thickness,\n              fill: this.spec.style?.linkStyle?.fill ?? getPalette()[0]\n            };\n          }\n        })\n      }\n    ];\n  }\n}\n"]}

package/es/semantic-mark.d.ts

@@ -0,0 +1,88 @@
+import type { ILogger } from '@visactor/vutils';
+import type { ISemanticMark, ISemanticMarkSpec, ParsedSimpleEncode, WithDefaultEncode, SemanticTooltipOption, SemanticAxisOption, SemanticPlayerOption, SemanticLabelOption, SemanticDataZoomOption, SemanticSliderOption, SemanticLegendOption, SemanticCrosshairOption, CoordinateOption, AxisSpec, CrosshairSpec, DatazoomSpec, DimensionTooltipSpec, LabelSpec, LegendSpec, PlayerSpec, SliderSpec, TooltipSpec, CoordinateSpec, DataSpec, ChannelEncodeType, ScaleSpec, TransformSpec, MarkAnimationSpec, ViewSpec, MarkType, Nil, ValueOf, GenerateBaseEncodeSpec, IAnimationConfig, MarkSpec, MarkRelativeItemSpec } from '@visactor/vgrammar';
+import type { ITextAttribute } from '@visactor/vrender';
+export declare abstract class SemanticMark<EncodeSpec, K extends string> implements ISemanticMark<EncodeSpec, K> {
+    spec: ISemanticMarkSpec<EncodeSpec, K>;
+    viewSpec?: ViewSpec;
+    readonly uid: number;
+    protected _logger: ILogger;
+    protected _coordinate: CoordinateOption;
+    readonly type: string;
+    constructor(type: string, id?: string | number);
+    parseSpec(spec: Partial<ISemanticMarkSpec<EncodeSpec, K>>): this;
+    coordinate(option: CoordinateOption): this;
+    data(values: any, transform?: TransformSpec[], id?: string): this;
+    encode(channel: K, option: ValueOf<WithDefaultEncode<EncodeSpec, K>, K>): this;
+    scale(channel: string, option: Partial<ScaleSpec>): this;
+    style(style: Partial<EncodeSpec & any>): this;
+    transform(option: TransformSpec | TransformSpec[]): this;
+    state(state: string, option: Partial<EncodeSpec>): this;
+    animate(state: string, option: IAnimationConfig | IAnimationConfig[]): this;
+    axis(channel: string, option?: SemanticAxisOption | boolean, layout?: MarkRelativeItemSpec): this;
+    legend(channel: string, option?: SemanticLegendOption | boolean, layout?: MarkRelativeItemSpec): this;
+    crosshair(channel: string, option?: SemanticCrosshairOption | boolean): this;
+    tooltip(option: SemanticTooltipOption | boolean): this;
+    slider(channel: string, option?: SemanticSliderOption | boolean, layout?: MarkRelativeItemSpec): this;
+    datazoom(channel: string, option?: SemanticDataZoomOption | boolean, layout?: MarkRelativeItemSpec): this;
+    label(channel: string, option?: SemanticLabelOption | boolean): this;
+    player(data?: any[], option?: SemanticPlayerOption | boolean, layout?: MarkRelativeItemSpec): this;
+    abstract setMarkType(): MarkType;
+    abstract parseScaleByEncode(channel: K, option: ValueOf<WithDefaultEncode<EncodeSpec, K>, K>): ScaleSpec | Nil;
+    abstract convertMarkEncode(encode: WithDefaultEncode<EncodeSpec, K>): GenerateBaseEncodeSpec<EncodeSpec>;
+    protected setDefaultDataTransform(): TransformSpec[];
+    protected setMultipleData(): DataSpec[];
+    protected setDefaultMarkTransform(): TransformSpec[];
+    protected convertMarkTransform(userTransform: TransformSpec[], defaultTransform?: TransformSpec[]): TransformSpec[];
+    protected convertMarkAnimation(): MarkAnimationSpec;
+    protected convertSimpleMarkEncode(encode: WithDefaultEncode<EncodeSpec, K>): ParsedSimpleEncode<EncodeSpec, K>;
+    protected getDataIdOfFiltered(): string;
+    protected getDataIdOfMain(): string;
+    protected getDataIdOfPlayer(): string;
+    protected getDataZoomScaleId(channel: string): {
+        x: string;
+        y: string;
+    };
+    protected getScaleId(channel: string): any;
+    protected getMarkId(): string;
+    protected getScaleSpec(scaleId: string): ScaleSpec;
+    protected parseScaleOfEncodeX(option: ValueOf<WithDefaultEncode<EncodeSpec, K>, K>): ScaleSpec | Nil;
+    protected parseScaleOfEncodeY(option: ValueOf<WithDefaultEncode<EncodeSpec, K>, K>): ScaleSpec | Nil;
+    protected parseScaleOfEncodeColor(option: ValueOf<WithDefaultEncode<EncodeSpec, K>, K>): ScaleSpec | Nil;
+    protected parseScaleOfEncodeGroup(option: ValueOf<WithDefaultEncode<EncodeSpec, K>, K>): ScaleSpec | Nil;
+    protected parseScaleOfCommonEncode(channel: K, option: ValueOf<WithDefaultEncode<EncodeSpec, K>, K>): ScaleSpec | Nil;
+    protected setDefaultAxis(): Record<string, Partial<AxisSpec>>;
+    protected parseAxisSpec(): AxisSpec[];
+    protected parseOption<T>(spec: {
+        option: T | boolean;
+        layout?: MarkRelativeItemSpec;
+    } | T | boolean): {
+        option: boolean | T;
+        layout: MarkRelativeItemSpec;
+    };
+    protected setDefaultLegend(): Record<string, Partial<LegendSpec>>;
+    protected parseLegendSpec(): LegendSpec[];
+    protected setDefaultCrosshair(): Record<string, Pick<CrosshairSpec, 'crosshairShape' | 'crosshairType'>>;
+    protected getVisualChannel(channel: 'x' | 'y'): 'x' | 'y' | 'angle' | 'radius';
+    protected parseCrosshairSpec(): CrosshairSpec[];
+    protected setDefaultTooltip(): SemanticTooltipOption | Nil;
+    protected parseTooltipSpec(): Array<TooltipSpec | DimensionTooltipSpec> | Nil;
+    protected setDefaultSlider(): Record<string, Partial<SliderSpec>>;
+    protected parseSliderSpec(): SliderSpec[];
+    protected setDefaultDataZoom(): Record<string, Partial<DatazoomSpec>>;
+    protected getVisiualPositionByDimension(channel: string): "left" | "bottom";
+    protected parseDataZoomSpec(): DatazoomSpec[];
+    protected setDefaultLabel(): Record<string, Partial<LabelSpec>>;
+    protected getLabelPosition(): string;
+    protected setLabelTextGetter(channel: string, option: SemanticLabelOption | boolean): ChannelEncodeType<ITextAttribute['text']>;
+    protected parseLabelSpec(): LabelSpec[];
+    protected setDefaultPlayer(): Record<string, Partial<PlayerSpec>>;
+    protected parsePlayerSpec(): PlayerSpec[];
+    protected parseDataSpec(): DataSpec[];
+    protected parseScaleSpec(): ScaleSpec[];
+    protected parseCoordinateSpec(): CoordinateSpec[];
+    protected setMainMarkEnterEncode(): Partial<any>;
+    protected setMainMarkSpec(): {};
+    protected setMultiMarksSpec(): MarkSpec[];
+    toViewSpec(): ViewSpec;
+    clear(): void;
+}