add precompile workload

Author: hexaeder

NetworkDynamicsInspector/Project.toml

@@ -13,6 +13,7 @@ Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 NetworkDynamics = "22e9dc34-2a0d-11e9-0de0-8588d035468b"
 Observables = "510215fc-4207-5dde-b226-833fc4488ee2"
 OrderedCollections = "bac558e1-5e72-5ebc-8fee-abe8a469f55d"
+PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 SciMLBase = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 WGLMakie = "276b4fcb-3e11-5398-bf8b-a0c2d153d008"
 
@@ -36,6 +37,7 @@ Graphs = "1.12.0"
 NetworkDynamics = "0.9.13"
 Observables = "0.5.5"
 OrderedCollections = "1.8.0"
+PrecompileTools = "1.2.1"
 SciMLBase = "2.75.1"
 WGLMakie = "0.11"
 julia = "1.10"

NetworkDynamicsInspector/src/NetworkDynamicsInspector.jl

@@ -24,6 +24,7 @@ using Colors: Colors, @colorant_str, RGB, color
 using ColorSchemes: ColorSchemes, ColorScheme
 using SciMLBase: SciMLBase
 using Downloads: Downloads
+using PrecompileTools: @compile_workload
 
 # defined based on the julia version
 using NetworkDynamics: AnnotatedIOBuffer, AnnotatedString, @styled_str
@@ -256,6 +257,34 @@ function element_info_card(app, session)
     )
 end
 
-
+@compile_workload begin
+    using NetworkDynamics: EdgeModel, VertexModel, AntiSymmetric, Network, dim, pdim
+    using Graphs: complete_graph
+    Base.@propagate_inbounds function diffusionedge!(e, v_s, v_d, (p,), _)
+        e .= p * (v_s[1] .- v_d[1])
+    end
+    e = EdgeModel(g=AntiSymmetric(diffusionedge!), outdim=1, pdim=1, pdef=[1], name=:diff_edge, indim=1)
+    Base.@propagate_inbounds function diffusionvertex!(dv, _, acc, _, _)
+        dv[1] = acc[1]
+        nothing
+    end
+    v = VertexModel(f=diffusionvertex!, dim=1, g=1:1, indim=1)
+
+    g = complete_graph(3)
+    nw = Network(g, v, e)
+    prob = SciMLBase.ODEProblem(nw, zeros(dim(nw)), (0, 0.1), zeros(pdim(nw)))
+
+    t = [0, 0.1]
+    u = [zeros(dim(nw)) for _ in t]
+    sol = SciMLBase.build_solution(prob, nothing, t, u)
+
+    app = NetworkDynamicsInspector.AppState(sol);
+    session = Bonito.Session()
+    NetworkDynamicsInspector.graphplot_card(app, session);
+    NetworkDynamicsInspector.gpstate_control_card(app, :vertex);
+    NetworkDynamicsInspector.element_info_card(app, session);
+    NetworkDynamicsInspector.timeseries_col(app, session);
+    NetworkDynamicsInspector.clear_obs_and_close!(app);
+end
 
 end # module NetworkDynamicsInspector

NetworkDynamicsInspector/src/timeseries.jl

@@ -401,7 +401,9 @@ function timeseries_card(app, key, session)
                 @error "$key: Plotting failed for idx $(valid_idxs[])" e
             end
         end
-        put!(plotqueue, task)
+        if !is_precompiling()
+            put!(plotqueue, task)
+        end
         nothing
     end |> track_obsf
 
@@ -474,7 +476,7 @@ function _plot_queue(key)
             rmt = take!(ch) # remove the executed task
             @assert istaskdone(rmt)
         end
-        @info "Plot queue for $key closed"
+        @debug "Plot queue for $key closed"
     end
 end
 

NetworkDynamicsInspector/src/utils.jl

@@ -161,3 +161,5 @@ function wait_for(queue)
         sleep(0.01)
     end
 end
+
+is_precompiling() = ccall(:jl_generating_output, Cint, ()) == 1