Merge pull request #580 from SciML/add_tstops_api

add tstops API
SciML · Aug 18, 2024 · 7358a9b · 7358a9b
2 parents c147c7a + eae6f16
commit 7358a9b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -34,7 +34,7 @@ UnPack = "3a884ed6-31ef-47d7-9d2a-63182c4928ed"
 Adapt = "3, 4"
 ArrayInterface = "6, 7"
 DataStructures = "0.18"
-DiffEqBase = "6.130.1"
+DiffEqBase = "6.154"
 DiffEqNoiseProcess = "5.13"
 DocStringExtensions = "0.8, 0.9"
 FiniteDiff = "2"

diff --git a/src/StochasticDiffEq.jl b/src/StochasticDiffEq.jl
@@ -47,7 +47,8 @@ using DocStringExtensions
                      resize_non_user_cache!,deleteat_non_user_cache!,addat_non_user_cache!,
                      terminate!,get_du, get_dt,get_proposed_dt,set_proposed_dt!,
                      u_modified!,savevalues!,add_tstop!,add_saveat!,set_reltol!,
-                     set_abstol!, postamble!, last_step_failed, has_Wfact, has_jac
+                     set_abstol!, postamble!, last_step_failed, has_Wfact, has_jac,
+                     get_tstops, get_tstops_array, get_tstops_max
 
   using DiffEqBase: check_error!, is_diagonal_noise, @..
 

diff --git a/src/integrators/integrator_interface.jl b/src/integrators/integrator_interface.jl
@@ -390,3 +390,7 @@ function DiffEqBase.set_u!(integrator::SDEIntegrator, u)
   integrator.u = u
   u_modified!(integrator, true)
 end
+
+DiffEqBase.get_tstops(integ::SDEIntegrator) = integ.opts.tstops
+DiffEqBase.get_tstops_array(integ::SDEIntegrator) = get_tstops(integ).valtree
+DiffEqBase.get_tstops_max(integ::SDEIntegrator) = maximum(get_tstops_array(integ))