157 lines
3.4 KiB
JavaScript
157 lines
3.4 KiB
JavaScript
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;
|