Mercurial > repos > public > sbplib
annotate +time/Rungekutta4SecondOrder.m @ 283:728fdc431212 operator_remake
Close branch operator_remake
author | Jonatan Werpers <jonatan.werpers@it.uu.se> |
---|---|
date | Mon, 12 Sep 2016 12:53:02 +0200 |
parents | b18d3d201a71 |
children | e7e73173d44d b5e5b195da1e |
rev | line source |
---|---|
0 | 1 classdef Rungekutta4SecondOrder < time.Timestepper |
2 properties | |
3 F | |
4 k | |
5 t | |
6 w | |
7 m | |
8 D | |
9 E | |
10 S | |
11 M | |
12 C | |
13 n | |
14 end | |
15 | |
16 | |
17 methods | |
18 function obj = Rungekutta4SecondOrder(D, E, S, k, t0, v0, v0t) | |
19 obj.D = D; | |
20 obj.E = E; | |
21 obj.S = S; | |
22 obj.m = length(v0); | |
13
b18d3d201a71
Fixed initialization of step counter in timesteppers.
Jonatan Werpers <jonatan@werpers.com>
parents:
0
diff
changeset
|
23 obj.n = 0; |
0 | 24 |
25 I = speye(obj.m); | |
26 O = sparse(obj.m,obj.m); | |
27 obj.M = [O, I; D, E*I]; % Multiply with I to allow 0 as input. | |
28 | |
29 if S == 0 | |
30 obj.C = zeros(2*obj.m,1); | |
31 else | |
32 obj.C = [zeros(obj.m,1), S]; | |
33 end | |
34 | |
35 obj.k = k; | |
36 obj.t = t0; | |
37 obj.w = [v0; v0t]; | |
38 | |
39 obj.F = @(w,t)(obj.M*w + obj.C); | |
40 end | |
41 | |
42 function [v,t] = getV(obj) | |
43 v = obj.w(1:end/2); | |
44 t = obj.t; | |
45 end | |
46 | |
47 function [vt,t] = getVt(obj) | |
48 vt = obj.w(end/2+1:end); | |
49 t = obj.t; | |
50 end | |
51 | |
52 function obj = step(obj) | |
53 obj.w = time.rk4.rungekutta_4(obj.w, obj.t, obj.k, obj.F); | |
54 obj.t = obj.t + obj.k; | |
55 obj.n = obj.n + 1; | |
56 end | |
57 end | |
58 | |
59 | |
60 methods (Static) | |
61 function k = getTimeStep(lambda) | |
62 k = rk4.get_rk4_time_step(lambda); | |
63 end | |
64 end | |
65 | |
66 end |