@@ -16,6 +16,10 @@ export class TipInfo {
1616 is_tip : boolean ;
1717}
1818
19+ class history {
20+ vertices : Array < Vertex > ;
21+ }
22+
1923const vertexSize = 20 ;
2024
2125export class VisualizerStore {
@@ -24,7 +28,7 @@ export class VisualizerStore {
2428 @observable solid_count = 0 ;
2529 @observable tips_count = 0 ;
2630 verticesIncomingOrder = [ ] ;
27- collect : boolean = false ;
31+ draw : boolean = false ;
2832 routerStore : RouterStore ;
2933
3034 // the currently selected vertex via hover
@@ -45,8 +49,22 @@ export class VisualizerStore {
4549
4650 constructor ( routerStore : RouterStore ) {
4751 this . routerStore = routerStore ;
52+ this . fetchHistory ( ) ;
4853 registerHandler ( WSMsgType . Vertex , this . addVertex ) ;
49- registerHandler ( WSMsgType . TipInfo , this . addTipInfo ) ;
54+ registerHandler ( WSMsgType . TipInfo , this . addTipInfo ) ;
55+ }
56+
57+ fetchHistory = async ( ) => {
58+ try {
59+ let res = await fetch ( `/api/visualizer/history` ) ;
60+ let history : history = await res . json ( ) ;
61+ history . vertices . forEach ( v => {
62+ this . addVertex ( v ) ;
63+ } ) ;
64+ } catch ( err ) {
65+ console . log ( "Fail to fetch history in visualizer" , err ) ;
66+ }
67+ return
5068 }
5169
5270 @action
@@ -87,9 +105,7 @@ export class VisualizerStore {
87105 }
88106
89107 @action
90- addVertex = ( vert : Vertex ) => {
91- if ( ! this . collect ) return ;
92-
108+ addVertex = ( vert : Vertex ) => {
93109 let existing = this . vertices . get ( vert . id ) ;
94110 if ( existing ) {
95111 // can only go from unsolid to solid
@@ -108,35 +124,16 @@ export class VisualizerStore {
108124 }
109125 this . verticesIncomingOrder . push ( vert . id ) ;
110126 this . checkLimit ( ) ;
111-
112- if ( vert . strongParentIDs ) {
113- // clear tip status of strong and weak parents
114- vert . strongParentIDs . forEach ( ( value , index ) => {
115- let strongParentVert = this . vertices . get ( value ) ;
116- if ( strongParentVert ) {
117- strongParentVert . is_tip = false ;
118- this . vertices . set ( strongParentVert . id , strongParentVert )
119- }
120- } ) ;
121- }
122- if ( vert . weakParentIDs ) {
123- vert . weakParentIDs . forEach ( ( value , index ) => {
124- let weakParentVert = this . vertices . get ( value ) ;
125- if ( weakParentVert ) {
126- weakParentVert . is_tip = false ;
127- this . vertices . set ( weakParentVert . id , weakParentVert )
128- }
129- } ) ;
130- }
131127 }
132128
133129 this . vertices . set ( vert . id , vert ) ;
134- this . drawVertex ( vert ) ;
130+ if ( this . draw ) {
131+ this . drawVertex ( vert ) ;
132+ }
135133 } ;
136134
137135 @action
138136 addTipInfo = ( tipInfo : TipInfo ) => {
139- if ( ! this . collect ) return ;
140137 let vert = this . vertices . get ( tipInfo . id ) ;
141138 if ( ! vert ) {
142139 // create a new empty one for now
@@ -148,7 +145,9 @@ export class VisualizerStore {
148145 this . tips_count += tipInfo . is_tip ? 1 : vert . is_tip ? - 1 : 0 ;
149146 vert . is_tip = tipInfo . is_tip ;
150147 this . vertices . set ( vert . id , vert ) ;
151- this . drawVertex ( vert ) ;
148+ if ( this . draw ) {
149+ this . drawVertex ( vert ) ;
150+ }
152151 } ;
153152
154153 @action
@@ -161,7 +160,9 @@ export class VisualizerStore {
161160 this . clearSelected ( ) ;
162161 }
163162 this . vertices . delete ( deleteId ) ;
164- this . graph . removeNode ( deleteId ) ;
163+ if ( this . draw ) {
164+ this . graph . removeNode ( deleteId ) ;
165+ }
165166 if ( ! vert ) {
166167 continue ;
167168 }
@@ -198,7 +199,9 @@ export class VisualizerStore {
198199 }
199200 this . vertices . delete ( approveeId ) ;
200201 }
201- this . graph . removeNode ( approveeId ) ;
202+ if ( this . draw ) {
203+ this . graph . removeNode ( approveeId ) ;
204+ }
202205 }
203206
204207 drawVertex = ( vert : Vertex ) => {
@@ -243,7 +246,7 @@ export class VisualizerStore {
243246 }
244247
245248 start = ( ) => {
246- this . collect = true ;
249+ this . draw = true ;
247250 this . graph = Viva . Graph . graph ( ) ;
248251
249252 let graphics : any = Viva . Graph . View . webglGraphics ( ) ;
@@ -280,17 +283,18 @@ export class VisualizerStore {
280283 } ) ;
281284 this . graphics = graphics ;
282285 this . renderer . run ( ) ;
286+
287+ this . vertices . forEach ( ( vertex ) => {
288+ this . drawVertex ( vertex )
289+ } )
283290 }
284291
285292 stop = ( ) => {
286- this . collect = false ;
293+ this . draw = false ;
287294 this . renderer . dispose ( ) ;
288295 this . graph = null ;
289296 this . paused = false ;
290297 this . selected = null ;
291- this . solid_count = 0 ;
292- this . tips_count = 0 ;
293- this . vertices . clear ( ) ;
294298 }
295299
296300 @action
0 commit comments