Merge pull request #186 from SciML/precompile

Copy of "Add precompilation"

Author: pogudingleb

Project.toml

@@ -15,6 +15,7 @@ Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 ModelingToolkit = "961ee093-0014-501f-94e3-6117800e7a78"
 Nemo = "2edaba10-b0f1-5616-af89-8c11ac63239a"
+PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 Primes = "27ebfcd6-29c5-5fa9-bf4b-fb8fc14df3ae"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 SymbolicUtils = "d1185830-fcd6-423d-90d6-eec64667417b"
@@ -29,6 +30,7 @@ IterTools = "1"
 MacroTools = "0.5"
 ModelingToolkit = "7, 8"
 Nemo = "0.24, 0.25, 0.26, 0.27, 0.28, 0.29, 0.30, 0.31, 0.32, 0.33, 0.34"
+PrecompileTools = "1"
 Primes = "0.5"
 SpecialFunctions = "1, 2"
 SymbolicUtils = "1"

src/ODE.jl

@@ -387,11 +387,13 @@ macro ODEmodel(ex::Expr...)
             )
         end
     end
-    @info "Summary of the model:"
-    @info "State variables: " * join(map(string, collect(x_vars)), ", ")
-    @info "Parameters: " * join(map(string, collect(params)), ", ")
-    @info "Inputs: " * join(map(string, collect(u_vars)), ", ")
-    @info "Outputs: " * join(map(string, collect(y_vars)), ", ")
+    logging_exprs = [
+        :(@info "Summary of the model:"),
+        :(@info "State variables: " * $(join(map(string, collect(x_vars)), ", "))),
+        :(@info "Parameters: " * $(join(map(string, collect(params)), ", "))),
+        :(@info "Inputs: " * $(join(map(string, collect(u_vars)), ", "))),
+        :(@info "Outputs: " * $(join(map(string, collect(y_vars)), ", "))),
+    ]
 
     # creating the ode object
     ode_expr = :(ODE{StructuralIdentifiability.Nemo.fmpq_mpoly}(
@@ -402,6 +404,7 @@ macro ODEmodel(ex::Expr...)
 
     result = Expr(
         :block,
+        logging_exprs...,
         exp_ring,
         assignments...,
         x_dict_create_expr,

src/StructuralIdentifiability.jl

@@ -188,4 +188,7 @@ function assess_identifiability(
     return out_dict
 end
 
+using PrecompileTools
+include("precompile.jl")
+
 end

src/precompile.jl

@@ -0,0 +1,27 @@
+
+@setup_workload begin
+    # Putting some things in `@setup_workload` instead of `@compile_workload` can reduce the size of the
+    # precompile file and potentially make loading faster.
+    using Logging
+    using ModelingToolkit
+    @parameters a01 a21 a12
+    @variables t x0(t) x1(t) y(t)
+    D = Differential(t)
+    eqs = [D(x0) ~ -(a01 + a21) * x0 + a12 * x1, D(x1) ~ a21 * x0 - a12 * x1]
+    de = ODESystem(eqs, t, name = :Test)
+    @compile_workload begin
+        # all calls in this block will be precompiled, regardless of whether
+        # they belong to your package or not (on Julia 1.8 and higher)
+        with_logger(ConsoleLogger(stdout, Logging.Warn)) do
+            ode = @ODEmodel(
+                x1'(t) = -(a01 + a21) * x1(t) + a12 * x2(t) + u(t),
+                x2'(t) = a21 * x1(t) - a12 * x2(t) - x3(t) / b,
+                x3'(t) = x3(t),
+                y(t) = x2(t)
+            )
+            assess_identifiability(ode)
+            assess_identifiability(de; measured_quantities = [x0])
+            assess_identifiability(de; measured_quantities = [y ~ x0])
+        end
+    end
+end