Extend the API a bit more

SciML · Oct 30, 2023 · 9b2890f · 9b2890f
1 parent 7e3610e
commit 9b2890f
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "DiffEqBase"
 uuid = "2b5f629d-d688-5b77-993f-72d75c75574e"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "6.135.0"
+version = "6.136.0"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

diff --git a/src/DiffEqBase.jl b/src/DiffEqBase.jl
@@ -171,9 +171,13 @@ export initialize!, finalize!
 
 export SensitivityADPassThrough
 
+export SteadyStateDiffEqTerminationMode, SimpleNonlinearSolveTerminationMode,
+    NormTerminationMode, RelTerminationMode, RelNormTerminationMode, AbsTerminationMode,
+    AbsNormTerminationMode, RelSafeTerminationMode, AbsSafeTerminationMode,
+    RelSafeBestTerminationMode, AbsSafeBestTerminationMode
+# Deprecated API
 export NLSolveTerminationMode,
-    NLSolveSafeTerminationOptions, NLSolveTerminationCondition,
-    NLSolveSafeTerminationResult
+    NLSolveSafeTerminationOptions, NLSolveTerminationCondition, NLSolveSafeTerminationResult
 
 export KeywordArgError, KeywordArgWarn, KeywordArgSilent
 

diff --git a/src/termination_conditions.jl b/src/termination_conditions.jl
@@ -55,6 +55,10 @@ mutable struct NonlinearTerminationModeCache{uType, T,
     nsteps::Int
 end
 
+get_termination_mode(cache::NonlinearTerminationModeCache) = cache.mode
+get_abstol(cache::NonlinearTerminationModeCache) = cache.abstol
+get_reltol(cache::NonlinearTerminationModeCache) = cache.reltol
+
 function __update_u!!(cache::NonlinearTerminationModeCache, u)
     cache.u === nothing && return
     if ArrayInterface.can_setindex(cache.u)
@@ -73,8 +77,9 @@ function _get_tolerance(::Nothing, ::Type{T}) where {T}
     return _get_tolerance(η, T)
 end
 
-function SciMLBase.init(u::AbstractArray{T}, mode::AbstractNonlinearTerminationMode;
-    abstol = nothing, reltol = nothing, kwargs...) where {T}
+function SciMLBase.init(u::Union{AbstractArray{T}, T},
+    mode::AbstractNonlinearTerminationMode; abstol = nothing, reltol = nothing,
+    kwargs...) where {T <: Number}
     abstol = _get_tolerance(abstol, T)
     reltol = _get_tolerance(reltol, T)
     best_value = __cvt_real(T, Inf)