refactor: move type-pirated function from BoundaryValueDiffEq here

SciML · May 31, 2024 · ab153bd · ab153bd
1 parent 47a9224
commit ab153bd
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/solutions/ode_solutions.jl b/src/solutions/ode_solutions.jl
@@ -519,6 +519,14 @@ function solution_new_tslocation(sol::ODESolution{T, N}, tslocation) where {T, N
         sol.original)
 end
 
+function solution_new_original_retcode(
+        sol::ODESolution{T, N}, original, retcode, resid) where {T, N}
+    return ODESolution{T, N}(
+        sol.u, sol.u_analytic, sol.errors, sol.t, sol.k, sol.discretes, sol.prob, sol.alg,
+        sol.interp, sol.dense, sol.tslocation, sol.stats, sol.alg_choice, retcode, resid,
+        original)
+end
+
 function solution_slice(sol::ODESolution{T, N}, I) where {T, N}
     new_t = sol.t[I]
     discretes = mask_discretes(sol.discretes, new_t, I)