fix: add graph reference on model

antvis · May 14, 2020 · 9f5e88d · 9f5e88d
1 parent f510dc8
commit 9f5e88d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/packages/x6/src/core/graph.ts b/packages/x6/src/core/graph.ts
@@ -295,6 +295,7 @@ export class Graph extends View<Graph.EventArgs> {
 
   setModel(model: Model) {
     this.model = model == null ? this.createModel() : model
+    this.model.graph = this
   }
 
   init() {

diff --git a/packages/x6/src/core/model.ts b/packages/x6/src/core/model.ts
@@ -14,7 +14,7 @@ export class Model extends Basecoat<Model.EventArgs> {
   protected readonly batches: KeyValue<number> = {}
   protected readonly addings: WeakMap<Cell, boolean> = new WeakMap()
 
-  protected graph: Graph | null
+  public graph: Graph | null
   protected nodes: KeyValue<boolean> = {}
   protected edges: KeyValue<boolean> = {}