Merge pull request #909 from SciML/init

Handle init with algorithm defaults

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "DifferentialEquations"
 uuid = "0c46a032-eb83-5123-abaf-570d42b7fbaa"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "7.5.0"
+version = "7.6.0"
 
 [deps]
 BoundaryValueDiffEq = "764a87c0-6b3e-53db-9096-fe964310641d"

src/default_solve.jl

@@ -9,6 +9,17 @@ function DiffEqBase.__solve(prob::DiffEqBase.DEProblem,
     DiffEqBase.__solve(prob, alg, args...; default_set = true, kwargs..., extra_kwargs...)
 end
 
+function DiffEqBase.__init(prob::DiffEqBase.DEProblem,
+                           alg::Union{Nothing, DiffEqBase.DEAlgorithm},
+                           args...; default_set = false, kwargs...)
+    if default_set == true
+        error("The chosen algorithm, $alg, does not exist. Please verify that the appropriate solver package has been installed.")
+    end
+    alg, extra_kwargs = default_algorithm(prob; kwargs...)
+    alg = DiffEqBase.prepare_alg(alg, prob.u0, prob.p, prob)
+    DiffEqBase.__init(prob, alg, args...; default_set = true, kwargs..., extra_kwargs...)
+end
+
 #=
 function DiffEqBase.__solve(prob::JumpProcesses.JumpProblem,alg::Nothing,
                args...;default_set=false,kwargs...)

test/default_ode_alg_test.jl

@@ -6,6 +6,7 @@ prob_ode_2Dlinear = ODEProblem(ODEFunction(f_2dlinear, analytic = f_2dlinear_ana
                                rand(4, 2), (0.0, 1.0), 1.01)
 
 alg, kwargs = default_algorithm(prob_ode_2Dlinear; dt = 1 // 2^(4))
+integ = init(prob_ode_2Dlinear; dt = 1 // 2^(4))
 sol = solve(prob_ode_2Dlinear; dt = 1 // 2^(4))
 
 @test typeof(sol.alg.algs[1]) <: Tsit5