view +time/+rk/rungekutta.m @ 1223:9fddc8749445 rv_diffOp_test

Closing branch
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Mon, 05 Aug 2019 10:48:37 +0200
parents c6fcee3fcf1b
children
line wrap: on
line source

% Takes one time step of size dt using the rungekutta method
% starting from v_0 and where the function F(v,t) gives the
% time derivatives. coeffs is a struct holding the RK coefficients
% for the specific method.
function v = rungekutta(v, t , dt, F, coeffs)
    % Compute the intermediate stages k
    k = zeros(length(v), coeffs.s);
    for i = 1:coeffs.s
        u = v;
        for j = 1:i-1
            u = u + dt*coeffs.a(i,j)*k(:,j);
        end
        k(:,i) = F(u,t+coeffs.c(i)*dt);
    end
    % Compute the updated solution as a linear combination
    % of the intermediate stages.
    for i = 1:coeffs.s
        v = v + dt*coeffs.b(i)*k(:,i);
    end
end