keep track of async and wait for gui

Author: hexaeder

NetworkDynamicsInspector/src/appstate.jl

@@ -40,6 +40,7 @@ struct AppState
     active_tsplot::Observable{String}
     graphplot::GraphPlot
     tsplots::Observable{OrderedDict{String, TimeseriesPlot}}
+    _plotqueue::Channel{Task}
 end
 function AppState(sol::SciMLBase.AbstractODESolution)
     t = sol.t[begin]
@@ -50,7 +51,8 @@ function AppState(sol::SciMLBase.AbstractODESolution)
         "ts-1" => TimeseriesPlot(),
     )
     active_tsplot = "ts-1"
-    AppState(sol, t, tmin, tmax, active_tsplot, graphplot, tsplots)
+    _plotqueue = Channel{Task}(Inf)
+    AppState(sol, t, tmin, tmax, active_tsplot, graphplot, tsplots, _plotqueue)
 end
 
 function free_ts_key()

NetworkDynamicsInspector/src/timeseries.jl

@@ -358,7 +358,7 @@ function timeseries_card(app, key, session)
     last_autolimits = Ref((eltype(valid_idxs)(), tsplot.rel[]))
     # plot the thing
     onany(data, replot; update=true) do _dat, _
-        @async begin
+        task = @async begin
             try
                 empty!(ax)
                 vlines!(ax.scene, app.t; color=:black)
@@ -381,6 +381,7 @@ function timeseries_card(app, key, session)
                 @error "Plotting failed" e
             end
         end
+        push!(app._plotqueue, task)
         nothing
     end |> track_obsf
 

NetworkDynamicsInspector/src/utils.jl

@@ -141,3 +141,12 @@ function download_assets()
         download(url, joinpath(ASSETS, name))
     end
 end
+
+function wait_for()
+    isnothing(APPSTATE[]) && return
+    queue = APPSTATE[]._plotqueue
+    while !isempty(queue)
+        wait(take!(queue))
+    end
+    nothing
+end