add precompile workload back

Author: hexaeder

ext/MTKExt.jl

@@ -475,11 +475,9 @@ end
 
 _all_rhs_symbols(eqs) = mapreduce(eq->get_variables(eq.rhs), ∪, eqs, init=Set{Symbolic}())
 
-using PrecompileTools: @setup_workload, @compile_workload
-@setup_workload begin
-    @compile_workload begin
-        # include("precompile_workload.jl")
-    end
+using PrecompileTools: @compile_workload
+@compile_workload begin
+    include("precompile_workload.jl")
 end
 
 end

ext/precompile_workload.jl

@@ -1,30 +1,28 @@
-is_precompiling() = ccall(:jl_generating_output, Cint, ()) == 1
-
 using NetworkDynamics
 using ModelingToolkit
 using ModelingToolkit: t_nounits as t, D_nounits as Dt
 
 @connector Terminal begin
-    u_r(t), [description="d-voltage"]
-    u_i(t), [description="q-voltage"]
-    i_r(t), [description="d-current", connect=Flow]
-    i_i(t), [description="q-current", connect=Flow]
+    u_r(t)
+    u_i(t)
+    i_r(t), [connect=Flow]
+    i_i(t), [connect=Flow]
 end
 @mtkmodel Swing begin
     @components begin
         terminal = Terminal()
     end
     @variables begin
-        ω(t)=0.0, [description="Rotor frequency"]
-        θ(t)=0.0, [description="Rotor angle"]
-        Pel(t), [description="Electrical Power injected into the grid"]
+        ω(t), [description="Rotor frequency", guess=0]
+        θ(t), [description="Rotor angle", guess=0]
+        Pel(t)
     end
     @parameters begin
-        M=1, [description="Inertia"]
-        D=0.1, [description="Damping"]
-        V=1.0, [description="Voltage magnitude"]
-        ω_ref=0, [description="Reference frequency"]
-        Pm, [description="Mechanical Power"]
+        M=0.005
+        D=0.1
+        V, [guess=1]
+        ω_ref=0
+        Pm, [description="Mechanical Power", guess=-1]
     end
     @equations begin
         Dt(θ) ~ ω - ω_ref
@@ -36,26 +34,10 @@ end
 end
 @mtkmodel BusBase begin
     @variables begin
-        u_r(t)=1, [description="bus d-voltage", output=true]
-        u_i(t)=0, [description="bus q-voltage", output=true]
-        i_r(t), [description="bus d-current (flowing into bus)", input=true]
-        i_i(t), [description="bus d-current (flowing into bus)", input=true]
-        P(t), [description="bus active power (flowing into network)"]
-        Q(t), [description="bus reactive power (flowing into network)"]
-        u_mag(t), [description="bus voltage magnitude"]
-        u_arg(t), [description="bus voltage argument"]
-        i_mag(t), [description="bus current magnitude"]
-        i_arg(t), [description="bus current argument"]
-        ω(t), [description="bus angular frequency"]
-    end
-    @equations begin
-        P ~ u_r * (-i_r) + u_i * (-i_i)
-        Q ~ u_i * (-i_r) - u_r * (-i_i)
-        u_mag ~ sqrt(u_r^2 + u_i^2)
-        u_arg ~ atan(u_i, u_r)
-        i_mag ~ sqrt(i_r^2 + i_i^2)
-        i_arg ~ atan(i_i, i_r)
-        ω ~ Dt(u_arg)
+        u_r(t)=1.0
+        u_i(t)=0.0
+        i_r(t)=1.0
+        i_i(t)=0.0
     end
 end
 @mtkmodel BusBar begin
@@ -80,10 +62,5 @@ end
     end
 end
 @named swingbus = Bus()
-
-if is_precompiling()
-    VertexModel(swingbus, [:busbar₊i_r, :busbar₊i_i], [:busbar₊u_r, :busbar₊u_i]; verbose=false)
-else
-    @info "ODEVertex"
-    @time @eval VertexModel(swingbus, [:busbar₊i_r, :busbar₊i_i], [:busbar₊u_r, :busbar₊u_i]; verbose=false)
-end
+vm = VertexModel(swingbus, [:busbar₊i_r, :busbar₊i_i], [:busbar₊u_r, :busbar₊u_i]; verbose=false)
+NetworkDynamics.initialize_component!(vm; verbose=false)

src/NetworkDynamics.jl

@@ -125,11 +125,9 @@ function reloadfaces!()
 end
 # NetworkDynamics.reloadfaces!()
 
-using PrecompileTools: @setup_workload, @compile_workload
-@setup_workload begin
-    @compile_workload begin
-        # include("precompile_workload.jl")
-    end
+using PrecompileTools: @compile_workload
+@compile_workload begin
+    include("precompile_workload.jl")
 end
 
 end

src/precompile_workload.jl

@@ -1,6 +1,3 @@
-is_precompiling() = ccall(:jl_generating_output, Cint, ()) == 1
-
-using TimerOutputs: TimerOutput, @timeit, print_timer
 using NetworkDynamics
 using Graphs: complete_graph
 
@@ -13,21 +10,11 @@ Base.@propagate_inbounds function diffusionvertex!(dv, _, acc, _, _)
     nothing
 end
 
-to = TimerOutput();
-@timeit to "precompile workload" begin
-    @timeit to "create graph" begin
-        g = complete_graph(3)
-    end
-    @timeit to "create network" begin
-        edge = EdgeModel(g=AntiSymmetric(diffusionedge!), outdim=1, pdim=1, pdef=[1])
-        vert = VertexModel(f=diffusionvertex!, g=1, dim=1, pdim=0)
-        nd = Network(g, vert, edge)
-    end
-    u = rand(dim(nd))
-    du = similar(u)
-    p = rand(pdim(nd))
-    @timeit to "coreloop" begin
-        nd(du,u,p,0.0)
-    end
-end
-is_precompiling() || print_timer(to)
+g = complete_graph(3)
+edge = EdgeModel(g=AntiSymmetric(diffusionedge!), outdim=1, pdim=1, pdef=[1])
+vert = VertexModel(f=diffusionvertex!, g=1, dim=1, pdim=0)
+nd = Network(g, vert, edge)
+u = rand(dim(nd))
+du = similar(u)
+p = rand(pdim(nd))
+nd(du,u,p,0.0)