rt-sim-training-client/src/jmap/painter.js

import * as zrUtil from 'zrender/src/core/util';
import Group from 'zrender/src/container/Group';
import deviceType from './config/deviceType';
import shapefactory from './shape/factory';
import ZoomHandle from './zoomHandle';

class Painter {
    constructor(jamp) {
        // 父级实例
        this.jmap = jamp;

        // 图层数据
        this.viewLevelMap = {};

        // 视图数据
        this.viewInstance = {};

        // 父级图层
        this.parentLevel = null;

        // 视图控制器
        this.zoomHandle = new ZoomHandle(this);

        // 初始绘图实例
        this.initPainterInstance();
    }

	/**
     * 初始绘图实例
     * @param {*} dom
     * @param {*} config
     */
    initPainterInstance() {
        // 添加父级图层
        this.parentLevel = new Group({ name: '__parent__' });
        this.jmap.getZr().add(this.parentLevel);

        // 添加子级图层
        zrUtil.each(Object.values(deviceType), (type) => {
            const level = new Group({ name: `__${type}__` });
            this.viewLevelMap[type] = level;
            this.parentLevel.add(level);
        });
    }

	/**
     * 重绘视图
     * @param {*} mapDevice
     */
    repaint(mapDevice) {
        // 清空视图
        this.clear();

        // 创建视图
        Object.values(mapDevice).forEach(device => {
            device && this.add(device);
        });
    }

	/**
     * 添加视图
     * @param {*} device
     */
    add(device) {
        const type = device._type;
        const code = device._code;
        const view = shapefactory(type, device, this.jmap.getStyleDict());
        if (view) {
            this.zoomHandle.transformView(view);
            this.viewInstance[code] = view;
            this.viewLevelMap[type].add(view);
        }
    }

	/**
     * 删除视图
     * @param {*} device
     */
    delete(device) {
        const code = device._code;
        const type = device._type;
        const view = this.viewInstance[code];
        if (view) {
            this.viewLevelMap[type].remove(view);
            delete this.viewInstance[code];
        }
    }

	/**
     * 更新视图
     * @param {*} device
     */
    update(device) {
        const code = device._code;
        const view = this.viewInstance[code];
        if (view) {
            view.setState(device);
        }
    }

    /**
     * 更新zoom
     * @param {*} zoom 
     */
    updateZoomTransform(zoom) {
        this.zoomHandle.updateTransform(zoom);
    }

    /**
     * 视图是否存在
     * @param {*} code 
     */
    hasViewInstance(code) {
        return this.viewInstance[code];
    }

    /**
     * 刷新画布，将在下一个渲染帧的时候被刷新。
     */
    refresh() {
        this.jmap.getZr() && this.jmap.getZr().refresh();
    }

	/**
     * 清除所有对象和画布
     */
    clear() {
        // 清空视图
        this.viewInstance = {};

        // 清空图层
        zrUtil.each(Object.values(this.viewLevelMap), (level) => {
            level && level.removeAll();
        })

        this.refresh();
    }

	/**
    * 销毁 ZRender 实例
    */
    dispose() {
        this.viewInstance = {};
        this.viewLevelMap = {};
        this.parentLevel = null;
    }
}

export default Painter;