StructV2/src/Model/SVModel.ts

import { Util } from "../Common/util";
import { Style } from "../options";
import { BoundingRect } from "../Common/boundingRect";
import { EdgeConfig, Item, NodeConfig } from "@antv/g6-core";
import { Graph } from "_@antv_g6-pc@0.5.0@@antv/g6-pc";
import merge from 'merge';




export class SVModel {
    public id: string;
    public sourceType: string;

    public g6Instance: Graph;
    public shadowG6Instance: Graph;
    public group: string;
    public layout: string;
    public G6ModelProps: NodeConfig | EdgeConfig;
    public shadowG6Item: Item;
    public G6Item: Item;

    public preLayout: boolean;   // 是否进入预备布局阶段
    public discarded: boolean;
    public freed: boolean;
    public leaked: boolean;
    public generalStyle: Partial<Style>;

    private transformMatrix: number[];
    private modelType: string;

    public layoutX: number;
    public layoutY: number;

    constructor(id: string, type: string, group: string, layout: string, modelType: string) {
        this.id = id;
        this.sourceType = type;
        this.group = group;
        this.layout = layout;
        this.shadowG6Item = null;
        this.G6Item = null;
        this.preLayout = false;
        this.discarded = false;
        this.freed = false;
        this.leaked = false;
        this.transformMatrix = [1, 0, 0, 0, 1, 0, 0, 0, 1];
        this.modelType = modelType;
    }

    /**
     * @override
     * 定义 G6 model 的属性
     * @param option 
     */
    generateG6ModelProps(options: unknown): NodeConfig | EdgeConfig {
        return null;
    }

    /**
     * 获取 G6 model 的属性
     * @param attr 
     */
    get(attr: string): any {
        return this.G6ModelProps[attr];
    }

    /**
     * 设置 G6 model 的属性
     * @param attr 
     * @param value 
     * @returns 
     */
    set(attr: string | object, value?: any) {
        if (this.discarded) {
            return;
        }

        if (typeof attr === 'object') {
            Object.keys(attr).map(item => {
                this.set(item, attr[item]);
            });
            return;
        }

        if (this.G6ModelProps[attr] === value) {
            return;
        }

        if (attr === 'style' || attr === 'labelCfg') {
            this.G6ModelProps[attr] = merge(this.G6ModelProps[attr] || {}, value);
        }
        else {
            this.G6ModelProps[attr] = value;
        }

        if (attr === 'rotation') {
            const matrix = Util.calcRotateMatrix(this.getMatrix(), value);
            this.setMatrix(matrix);
        }

        // 更新G6Item
        if (this.G6Item) {
            if (this.preLayout) {
                const G6ItemModel = this.G6Item.getModel();
                G6ItemModel[attr] = value;
            }
            else {
                this.g6Instance.updateItem(this.G6Item, this.G6ModelProps);
            }
        }

        // 更新shadowG6Item
        if (this.shadowG6Item) {
            this.shadowG6Instance.updateItem(this.shadowG6Item, this.G6ModelProps);
        }
    }

    /**
     * 
     * @param G6ModelProps 
     */
    updateG6ModelStyle(G6ModelProps: NodeConfig | EdgeConfig) {
        const newG6ModelProps = { 
            style: {
                ...G6ModelProps.style
            },
            labelCfg: {
                ...G6ModelProps.labelCfg
            }
        };

        this.G6ModelProps = merge(this.G6ModelProps, newG6ModelProps);

        if (this.G6Item) {
            this.g6Instance.updateItem(this.G6Item, this.G6ModelProps);
        }

        if (this.shadowG6Item) {
            this.shadowG6Instance.updateItem(this.shadowG6Item, this.G6ModelProps);
        }
    }

    /**
     * 获取包围盒
     * @returns 
     */
    getBound(): BoundingRect {
        return this.shadowG6Item.getBBox();
    }

    /**
     * 获取变换矩阵
     */
    getMatrix(): number[] {
        return [...this.transformMatrix];
    }

    /**
     * 设置变换矩阵
     * @param matrix 
     */
    setMatrix(matrix: number[]) {
        this.transformMatrix = matrix;
        this.set('style', { matrix });
    }

    /**
     * 设置是否被选中的状态
     * @param isSelected 
     */
    setSelectedState(isSelected: boolean) {
        if(this.G6Item === null) {
            return;
        }

        this.G6Item.setState('selected', isSelected);
    }

    /**
     * 
     * @returns 
     */
    getG6ModelProps(): NodeConfig | EdgeConfig {
        return Util.objectClone(this.G6ModelProps);
    }

    /**
     * 获取 model 类型
     * @returns 
     */
    getModelType(): string {
        return this.modelType;
    }

    /**
     * 判断是否为节点model（SVNode）
     */
    isNode(): boolean {
        return false;
    }


}