feat: solve initialization problem for null solutions/integrators

SciML · May 28, 2024 · aa3a3c3 · aa3a3c3
1 parent 08fbaf0
commit aa3a3c3
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/solve.jl b/src/solve.jl
@@ -676,7 +676,21 @@ function build_null_solution(prob::AbstractDEProblem, args...;
 
     timeseries = [Float64[] for i in 1:length(ts)]
 
-    build_solution(prob, nothing, ts, timeseries, retcode = ReturnCode.Success)
+    sol = build_solution(prob, nothing, ts, timeseries, retcode = ReturnCode.Success)
+    if SciMLBase.has_initializeprob(prob.f)
+        integ = NullODEIntegrator{
+            isinplace(prob), typeof(prob), eltype(prob.tspan), typeof(sol),
+            typeof(prob.f), typeof(prob.p)
+        }(Float64[],
+            Float64[],
+            prob.tspan[1],
+            prob,
+            sol,
+            prob.f,
+            prob.p)
+        initialize_dae!(integ)
+    end
+    return sol
 end
 
 function build_null_solution(