Mercurial > repos > public > sbplib
annotate +time/Rk4SecondOrderNonlin.m @ 1:5ae4f23d9130
Added CdiffNonlin timestepper. Probably fixed a bug with Cdiff. Added default arguments to Rk4SecondOrderNonlin.
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Thu, 17 Sep 2015 19:19:58 +0200 |
| parents | 48b6fb693025 |
| children | b18d3d201a71 |
| rev | line source |
|---|---|
| 0 | 1 classdef Rk4SecondOrderNonlin < time.Timestepper |
| 2 properties | |
| 3 F | |
| 4 k | |
| 5 t | |
| 6 w | |
| 7 m | |
| 8 | |
| 9 D | |
| 10 E | |
| 11 S | |
| 12 | |
| 13 n | |
| 14 end | |
| 15 | |
| 16 | |
| 17 methods | |
| 18 function obj = Rk4SecondOrderNonlin(D, E, S, k, t0, v0, v0t) | |
|
1
5ae4f23d9130
Added CdiffNonlin timestepper. Probably fixed a bug with Cdiff. Added default arguments to Rk4SecondOrderNonlin.
Jonatan Werpers <jonatan@werpers.com>
parents:
0
diff
changeset
|
19 default_arg('S',0); |
|
5ae4f23d9130
Added CdiffNonlin timestepper. Probably fixed a bug with Cdiff. Added default arguments to Rk4SecondOrderNonlin.
Jonatan Werpers <jonatan@werpers.com>
parents:
0
diff
changeset
|
20 default_arg('E',0); |
| 0 | 21 |
| 22 if S == 0 | |
| 23 S = @(v,t)0; | |
| 24 end | |
| 25 | |
| 26 if E == 0 | |
| 27 E = @(v,t)0; | |
| 28 end | |
| 29 | |
| 30 obj.k = k; | |
| 31 obj.t = t0; | |
| 32 obj.w = [v0; v0t]; | |
| 33 | |
| 34 m = length(v0); | |
| 35 function wt = F(w,t) | |
| 36 v = w(1:m); | |
| 37 vt = w(m+1:end); | |
| 38 | |
| 39 % Def: w = [v; vt] | |
| 40 wt(1:m,1) = vt; | |
| 41 wt(m+1:2*m,1) = D(v)*v + E(v)*vt + S(v,t); | |
| 42 | |
| 43 end | |
| 44 | |
| 45 obj.F = @F; | |
| 46 obj.D = D; | |
| 47 obj.E = E; | |
| 48 obj.S = S; | |
| 49 obj.m = m; | |
| 50 end | |
| 51 | |
| 52 function [v,t] = getV(obj) | |
| 53 v = obj.w(1:end/2); | |
| 54 t = obj.t; | |
| 55 end | |
| 56 | |
| 57 function [vt,t] = getVt(obj) | |
| 58 vt = obj.w(end/2+1:end); | |
| 59 t = obj.t; | |
| 60 end | |
| 61 | |
| 62 function obj = step(obj) | |
| 63 obj.w = time.rk4.rungekutta_4(obj.w, obj.t, obj.k, obj.F); | |
| 64 obj.t = obj.t + obj.k; | |
| 65 obj.n = obj.n + 1; | |
| 66 end | |
| 67 end | |
| 68 | |
| 69 | |
| 70 methods (Static) | |
| 71 function k = getTimeStep(lambda) | |
| 72 k = rk4.get_rk4_time_step(lambda); | |
| 73 end | |
| 74 end | |
| 75 | |
| 76 end |