update interface to catch errors.

QuEraComputing · Jan 3, 2024 · 10b2f68 · 10b2f68
1 parent ccdcc74
commit 10b2f68
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/src/dp5/solver.jl b/src/dp5/solver.jl
@@ -3,7 +3,7 @@
 #include("checks.jl")
 #include("helpers.jl")
 
-function  DormandPrince.integrate!(
+function  DormandPrince.integrate_core!(
    solver::DP5Solver{T},
    xend::T
 ) where T

diff --git a/src/dp8/solver.jl b/src/dp8/solver.jl
@@ -3,7 +3,7 @@
 #include("checks.jl")
 #include("helpers.jl")
 
-function  DormandPrince.integrate!(
+function  DormandPrince.integrate_core!(
    solver::DP8Solver{T},
    xend::T
 ) where T
@@ -21,6 +21,10 @@ function  DormandPrince.integrate!(
     check_beta(solver.options) || return Report(solver.vars.x, DormandPrince.CURIOUS_BETA, DormandPrince.INPUT_NOT_CONSISTENT, 0, 0, 0, 0)
     check_safety_factor(solver.options) || return Report(solver.vars.x, DormandPrince.CURIOUS_SAFETY_FACTOR, DormandPrince.INPUT_NOT_CONSISTENT, 0, 0, 0, 0)
 
+    if solver.vars.x == xend
+        return Report(solver.vars.x, DormandPrince.INPUT_CHECKS_SUCCESSFUL, DormandPrince.COMPUTATION_SUCCESSFUL, 0, 0, 0, 0)
+    end
+
     ###### nstiff -  parameters for stiffness detection
     # nstiff = solver_options.stiffness_test_activation_step
 

diff --git a/src/interface.jl b/src/interface.jl
@@ -30,13 +30,21 @@ end
 # 3. integrate(callback, solver, times) -> vector of states with callback applied
 
 get_current_state(::AbstractDPSolver) = error("not implemented")
-integrate!(::AbstractDPSolver{T}, ::T) where T = error("not implemented")
+integrate_core!(::AbstractDPSolver{T}, ::T) where T = error("not implemented")
+
+function integrate!(solver::AbstractDPSolver{T}, time::T) where T <: Real
+    report = integrate_core!(solver, time)
+    if report.idid != COMPUTATION_SUCCESSFUL
+        error("integration failed at time $time with report $report")
+    end
+end
 function integrate!(callback, solver::AbstractDPSolver{T}, times::AbstractVector{T}; sort_times::Bool = true) where {T <: Real}
     times = sort_times ? sort(collect(times)) : times
 
     result = []
     for time in times
         integrate!(solver, time)
+
         push!(result, callback(time, get_current_state(solver)))
     end