0
我有一個類的一些ODE求解器的自定義實現。我遇到問題,如果我的時間步dt
小於.2
,程序將停止。但是,如果我註釋掉一個龍格 - 庫塔求解器,它會很快執行,我可以切換哪一個被註釋掉,這樣我就可以從兩個求解器中獲得解決方案。我想知道如何解決這個問題。我一直在試圖找到某種解決方法可能會干擾另一方的方法,但我不明白這是怎麼發生的。ODE求解器在一起運行時緩慢運行
實現:
global dt;
dt = .5; % going below ~.25 makes the program take a very long time to exit
g = 1;
c_d = 2;
m = 3;
tf = 15;
dudt = @(t, u) g - (c_d/m) * u.^2
[t_euler, u_euler] = euler(dudt, [0, tf], 0);
[t_rk4, u_rk4] = rk4(dudt, [0, tf], 0); % either of this one or rk2
% can be commented out to make the program
% run quickly, but rk2 and rk4 cannot be
% run at the same time
[t_rk2, u_rk2] = rk2(dudt, [0, tf], 0);
%% rk4.m %%
function [t, u] = rk4(odefun, tspan, u0)
t0 = tspan(1);
t = [ t0 ];
t_new = t0;
global dt;
tf = tspan(2);
u_new = u0;
u = [ u0 ];
while (t_new < tf)
if (t_new + dt > tf)
dt = tf - t_new;
end
k1 = dt * odefun(t_new, u_new);
k2 = dt * odefun(t_new + dt/2, u_new + k1/2);
k3 = dt * odefun(t_new + dt/2, u_new + k2/2);
k4 = dt * odefun(t_new + dt, u_new + k3);
u_new = u_new + 1/6 * (k1 + 2*k2 + 2*k3 + k4);
% rk2.m is the same as rk4.m, except u_new = u_new + k2
% euler.m is also the same, except u_new = u_new + k1
u = [ u, u_new ];
t_new = t_new + dt;
t = [ t, t_new ];
end
end
就是這樣。非常感謝你! –