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(jmap) {
		// 父级实例
		this.$jmap = jmap;

		// 视图控制器
		this.$zoomHandle = new ZoomHandle(this, jmap.getZr());

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

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

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

		// 初始绘图实例
		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;
	}

	/**
     * 获取父级图层
     */
	getParentLevel() {
		return this.parentLevel;
	}
}

export default Painter;