From 237166fae55cbf1386530c7c656456d7e567af1b Mon Sep 17 00:00:00 2001 From: ParamThakkar123 Date: Tue, 19 Nov 2024 00:15:32 +0530 Subject: [PATCH] Added Dispatch --- ext/TaylorIntegrationDiffEqExt.jl | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/ext/TaylorIntegrationDiffEqExt.jl b/ext/TaylorIntegrationDiffEqExt.jl index 9de736664..3528eb711 100644 --- a/ext/TaylorIntegrationDiffEqExt.jl +++ b/ext/TaylorIntegrationDiffEqExt.jl @@ -196,6 +196,22 @@ function OrdinaryDiffEq.perform_step!(integrator, cache::TaylorMethodCache) integrator.stats.nf += 1 end +function OrdinaryDiffEq.get_fsalfirstlast!(cache::TaylorMethodCache, f, t, + u, uprev, p, tT, uT, duT, uauxT + ) + fsalfirst = zero(rate_prototype) + fsallast = zero(rate_prototype) + if parse_eqs[] + fsalfirst = f(t, u, p) + fsallast = f(t, u, p) + else + TaylorIntegration.__jetcoeffs!(Val(false), f, tT, uT, duT, uauxT, p, cache.rv) + fsalfirst = constant_term.(duT) + fsallast = constant_term.(duT) + end + return fsalfirst, fsallast +end + OrdinaryDiffEq.stepsize_controller!(integrator,alg::TaylorMethodParams) = TaylorIntegration.stepsize(integrator.cache.uT, integrator.opts.abstol) OrdinaryDiffEq.step_accept_controller!(integrator, alg::TaylorMethodParams, q) = q