Merge pull request #2377 from SciML/tstop_interface

add tstops API
SciML · Aug 18, 2024 · f43bcaf · f43bcaf
2 parents d88f255 + 6c6c838
commit f43bcaf
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -79,7 +79,7 @@ ADTypes = "0.2, 1"
 Adapt = "3.0, 4"
 ArrayInterface = "7"
 DataStructures = "0.18"
-DiffEqBase = "6.147"
+DiffEqBase = "6.154"
 DocStringExtensions = "0.9"
 EnumX = "1"
 ExponentialUtilities = "1"

diff --git a/lib/OrdinaryDiffEqCore/src/OrdinaryDiffEqCore.jl b/lib/OrdinaryDiffEqCore/src/OrdinaryDiffEqCore.jl
@@ -51,7 +51,9 @@ import DiffEqBase: resize!, deleteat!, addat!, full_cache, user_cache, u_cache,
                    add_tstop!, has_tstop, first_tstop, pop_tstop!,
                    add_saveat!, set_reltol!,
                    set_abstol!, postamble!, last_step_failed,
-                   isautodifferentiable
+                   isautodifferentiable,
+                   get_tstops, get_tstops_array, get_tstops_max
+
 
 using DiffEqBase: check_error!, @def, _vec, _reshape
 

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