-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
LayerModel.ts
105 lines (90 loc) · 2.52 KB
/
LayerModel.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import { BaseModel, BaseModelGenerics, BaseModelOptions } from '../../core-models/BaseModel';
import { CanvasModel } from '../canvas/CanvasModel';
import * as _ from 'lodash';
import { CanvasEngine } from '../../CanvasEngine';
import { FactoryBank } from '../../core/FactoryBank';
import { AbstractModelFactory } from '../../core/AbstractModelFactory';
import { DeserializeEvent } from '../../core-models/BaseEntity';
export interface LayerModelOptions extends BaseModelOptions {
isSvg?: boolean;
transformed?: boolean;
}
export interface LayerModelGenerics extends BaseModelGenerics {
OPTIONS: LayerModelOptions;
PARENT: CanvasModel;
CHILDREN: BaseModel;
ENGINE: CanvasEngine;
}
export abstract class LayerModel<G extends LayerModelGenerics = LayerModelGenerics> extends BaseModel<G> {
protected models: { [id: string]: G['CHILDREN'] };
protected repaintEnabled: boolean;
constructor(options: G['OPTIONS'] = {}) {
super(options);
this.models = {};
this.repaintEnabled = true;
}
/**
* This is used for deserialization
*/
abstract getChildModelFactoryBank(engine: G['ENGINE']): FactoryBank<AbstractModelFactory<BaseModel>>;
deserialize(event: DeserializeEvent<this>) {
super.deserialize(event);
this.options.isSvg = !!event.data.isSvg;
this.options.transformed = !!event.data.transformed;
_.forEach(event.data.models, (model) => {
const modelOb = this.getChildModelFactoryBank(event.engine).getFactory(model.type).generateModel({
initialConfig: model
});
modelOb.deserialize({
...event,
data: model
});
this.addModel(modelOb);
});
}
serialize() {
return {
...super.serialize(),
isSvg: this.options.isSvg,
transformed: this.options.transformed,
models: _.mapValues(this.models, (model) => {
return model.serialize();
})
};
}
isRepaintEnabled() {
return this.repaintEnabled;
}
allowRepaint(allow: boolean = true) {
this.repaintEnabled = allow;
}
remove() {
if (this.parent) {
this.parent.removeLayer(this);
}
super.remove();
}
addModel(model: G['CHILDREN']) {
model.setParent(this);
this.models[model.getID()] = model;
}
getSelectionEntities(): Array<BaseModel> {
return _.flatMap(this.models, (model) => {
return model.getSelectionEntities();
});
}
getModels() {
return this.models;
}
getModel(id: string) {
return this.models[id];
}
removeModel(id: string | G['CHILDREN']): boolean {
const _id = typeof id === 'string' ? id : id.getID();
if (this.models[_id]) {
delete this.models[_id];
return true;
}
return false;
}
}