diff --git a/src/solutions/dae_solutions.jl b/src/solutions/dae_solutions.jl index 8cbbc839c2..0188fefb3a 100644 --- a/src/solutions/dae_solutions.jl +++ b/src/solutions/dae_solutions.jl @@ -27,13 +27,14 @@ https://docs.sciml.ai/DiffEqDocs/stable/basics/solution/ exited due to an error. For more details, see [the return code documentation](https://docs.sciml.ai/SciMLBase/stable/interfaces/Solutions/#retcodes). """ -struct DAESolution{T, N, uType, duType, uType2, DType, tType, P, A, ID, S} <: +struct DAESolution{T, N, uType, rateType, uType2, DType, tType, P, A, ID, S} <: AbstractDAESolution{T, N, uType} u::uType - du::duType + du::rateType u_analytic::uType2 errors::DType t::tType + k::Union{Nothing, rateType} prob::P alg::A interp::ID @@ -95,6 +96,7 @@ function build_solution(prob::AbstractDAEProblem, alg, t, u, du = nothing; u_analytic, errors, t, + k, prob, alg, interp, @@ -112,7 +114,7 @@ function build_solution(prob::AbstractDAEProblem, alg, t, u, du = nothing; DAESolution{T, N, typeof(u), typeof(du), Nothing, Nothing, typeof(t), typeof(prob), typeof(alg), typeof(interp), typeof(stats)}(u, du, nothing, - nothing, t, + nothing, t, k, prob, alg, interp, dense, 0, @@ -167,6 +169,7 @@ function build_solution(sol::AbstractDAESolution{T, N}, u_analytic, errors) wher u_analytic, errors, sol.t, + sol.k, sol.prob, sol.alg, sol.interp, @@ -184,6 +187,7 @@ function solution_new_retcode(sol::AbstractDAESolution{T, N}, retcode) where {T, sol.u_analytic, sol.errors, sol.t, + sol.k, sol.prob, sol.alg, sol.interp, @@ -201,6 +205,7 @@ function solution_new_tslocation(sol::AbstractDAESolution{T, N}, tslocation) whe sol.u_analytic, sol.errors, sol.t, + sol.k, sol.prob, sol.alg, sol.interp, @@ -221,6 +226,7 @@ function solution_slice(sol::AbstractDAESolution{T, N}, I) where {T, N} sol.u_analytic[I], sol.errors, sol.t[I], + sol.k[I], sol.prob, sol.alg, sol.interp,