Rollback erroring out at higher order derivatives

Author: sharanry

src/solutions/ode_solutions.jl

@@ -18,54 +18,50 @@ end
 (sol::ODESolution)(t,deriv::Type=Val{0};idxs=nothing,continuity=:left) = sol(t,deriv,idxs,continuity)
 (sol::ODESolution)(v,t,deriv::Type=Val{0};idxs=nothing,continuity=:left) = sol.interp(v,t,idxs,deriv,sol.prob.p,continuity)
 
-function (sol::ODESolution)(t::Real,::Type{Val{0}},idxs::Nothing,continuity)
-  sol.interp(t,idxs,Val{0},sol.prob.p,continuity)
+function (sol::ODESolution)(t::Real,deriv,idxs::Nothing,continuity)
+  sol.interp(t,idxs,deriv,sol.prob.p,continuity)
 end
 
-function (sol::ODESolution)(t::AbstractVector{<:Real},::Type{Val{0}},idxs::Nothing,continuity)
-  augment(sol.interp(t,idxs,Val{0},sol.prob.p,continuity), sol)
+function (sol::ODESolution)(t::AbstractVector{<:Real},deriv,idxs::Nothing,continuity)
+  augment(sol.interp(t,idxs,deriv,sol.prob.p,continuity), sol)
 end
 
-function (sol::ODESolution)(t::Real,::Type{Val{0}},idxs::Integer,continuity)
-  sol.interp(t,idxs,Val{0},sol.prob.p,continuity)
+function (sol::ODESolution)(t::Real,deriv,idxs::Integer,continuity)
+  sol.interp(t,idxs,deriv,sol.prob.p,continuity)
 end
-function (sol::ODESolution)(t::Real,::Type{Val{0}},idxs::AbstractVector{<:Integer},continuity)
-  sol.interp(t,idxs,Val{0},sol.prob.p,continuity)
+function (sol::ODESolution)(t::Real,deriv,idxs::AbstractVector{<:Integer},continuity)
+  sol.interp(t,idxs,deriv,sol.prob.p,continuity)
 end
-function (sol::ODESolution)(t::AbstractVector{<:Real},::Type{Val{0}},idxs::Integer,continuity)
-  sol.interp(t,idxs,Val{0},sol.prob.p,continuity)
+function (sol::ODESolution)(t::AbstractVector{<:Real},deriv,idxs::Integer,continuity)
+  sol.interp(t,idxs,deriv,sol.prob.p,continuity)
 end
-function (sol::ODESolution)(t::AbstractVector{<:Real},::Type{Val{0}},idxs::AbstractVector{<:Integer},continuity)
-  sol.interp(t,idxs,Val{0},sol.prob.p,continuity)
+function (sol::ODESolution)(t::AbstractVector{<:Real},deriv,idxs::AbstractVector{<:Integer},continuity)
+  sol.interp(t,idxs,deriv,sol.prob.p,continuity)
 end
 
-function (sol::ODESolution)(t::Real,::Type{Val{0}},idxs,continuity)
+function (sol::ODESolution)(t::Real,deriv,idxs,continuity)
   issymbollike(idxs) || error("Incorrect specification of `idxs`")
-  augment(sol.interp([t],nothing,Val{0},sol.prob.p,continuity), sol)[idxs][1]
+  augment(sol.interp([t],nothing,deriv,sol.prob.p,continuity), sol)[idxs][1]
 end
 
-function (sol::ODESolution)(t::Real,::Type{Val{0}},idxs::AbstractVector,continuity)
+function (sol::ODESolution)(t::Real,deriv,idxs::AbstractVector,continuity)
   all(issymbollike.(idxs)) || error("Incorrect specification of `idxs`")
-  interp_sol = augment(sol.interp([t],nothing,Val{0},sol.prob.p,continuity), sol)
+  interp_sol = augment(sol.interp([t],nothing,deriv,sol.prob.p,continuity), sol)
   [first(interp_sol[idx]) for idx in idxs]
 end
 
-function (sol::ODESolution)(t::AbstractVector{<:Real},::Type{Val{0}},idxs,continuity)
+function (sol::ODESolution)(t::AbstractVector{<:Real},deriv,idxs,continuity)
   issymbollike(idxs) || error("Incorrect specification of `idxs`")
-  interp_sol = augment(sol.interp(t,nothing,Val{0},sol.prob.p,continuity), sol)
+  interp_sol = augment(sol.interp(t,nothing,deriv,sol.prob.p,continuity), sol)
   DiffEqArray(interp_sol[idxs], t)
 end
 
-function (sol::ODESolution)(t::AbstractVector{<:Real},::Type{Val{0}},idxs::AbstractVector,continuity)
+function (sol::ODESolution)(t::AbstractVector{<:Real},deriv,idxs::AbstractVector,continuity)
   all(issymbollike.(idxs)) || error("Incorrect specification of `idxs`")
-  interp_sol = augment(sol.interp(t,nothing,Val{0},sol.prob.p,continuity), sol)
+  interp_sol = augment(sol.interp(t,nothing,deriv,sol.prob.p,continuity), sol)
   DiffEqArray([[interp_sol[idx][i] for idx in idxs] for i in 1:length(t)], t)
 end
 
-function (sol::ODESolution)(t,::Type{Val{N}},idxs,continuity) where N
-  N == 0 || error("Higher-order interpolation is not implemented.")
-end
-
 function build_solution(
         prob::Union{AbstractODEProblem,AbstractDDEProblem},
         alg,t,u;timeseries_errors=length(u)>2,

test/downstream/symbol_indexing.jl

@@ -110,8 +110,6 @@ sol9 = sol(0.0:1.0:10.0, idxs=2)
 sol10 = sol(0.1, idxs=2)
 @test sol10 isa Real
 
-@test_throws ErrorException sol(0.1, Val{1})
-@test_throws ErrorException sol([0.0:0.1:1.0], Val{2})
 
 #=
 using Plots