Mercurial > repos > public > sbplib
annotate +time/SBPInTime.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 | 38173ea263ed |
children | 8894e9c49e40 |
rev | line source |
---|---|
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
1 classdef SBPInTime < time.Timestepper |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
2 % The SBP in time method. |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
3 % Implemented for v_t = A*v + f(t) |
398 | 4 % |
5 % Each "step" takes one block step and thus advances | |
6 % k = k_local*(blockSize-1) in time. | |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
7 properties |
398 | 8 M % System matrix |
9 L,U,P,Q % LU factorization of M | |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
10 A |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
11 Et_r |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
12 penalty |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
13 f |
398 | 14 k_local % step size within a block |
15 k % Time size of a block k/(blockSize-1) = k_local | |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
16 t |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
17 v |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
18 m |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
19 n |
398 | 20 blockSize % number of points in each block |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
21 order |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
22 nodes |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
23 end |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
24 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
25 methods |
398 | 26 function obj = SBPInTime(A, f, k, t0, v0, TYPE, order, blockSize) |
406
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
27 |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
28 default_arg('TYPE','gauss'); |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
29 |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
30 if(strcmp(TYPE,'gauss')) |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
31 default_arg('order',4) |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
32 default_arg('blockSize',4) |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
33 else |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
34 default_arg('order', 8); |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
35 default_arg('blockSize',time.SBPInTime.smallestBlockSize(order,TYPE)); |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
36 end |
398 | 37 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
38 obj.A = A; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
39 obj.f = f; |
398 | 40 obj.k_local = k/(blockSize-1); |
41 obj.k = k; | |
42 obj.blockSize = blockSize; | |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
43 obj.t = t0; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
44 obj.m = length(v0); |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
45 obj.n = 0; |
398 | 46 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
47 %==== Build the time discretization matrix =====% |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
48 switch TYPE |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
49 case 'equidistant' |
398 | 50 ops = sbp.D2Standard(blockSize,{0,obj.k},order); |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
51 case 'optimal' |
398 | 52 ops = sbp.D1Nonequidistant(blockSize,{0,obj.k},order); |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
53 case 'minimal' |
398 | 54 ops = sbp.D1Nonequidistant(blockSize,{0,obj.k},order,'minimal'); |
406
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
55 case 'gauss' |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
56 ops = sbp.D1Gauss(blockSize,{0,obj.k}); |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
57 end |
398 | 58 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
59 D1 = ops.D1; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
60 HI = ops.HI; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
61 e_l = ops.e_l; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
62 e_r = ops.e_r; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
63 obj.nodes = ops.x; |
398 | 64 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
65 Ix = speye(size(A)); |
398 | 66 It = speye(blockSize,blockSize); |
67 | |
68 obj.Et_r = kron(e_r,Ix); | |
69 | |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
70 % Time derivative + penalty |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
71 tau = 1; |
398 | 72 Mt = D1 + tau*HI*(e_l*e_l'); |
73 | |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
74 % penalty to impose "data" |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
75 penalty = tau*HI*e_l; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
76 obj.penalty = kron(penalty,Ix); |
398 | 77 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
78 Mx = kron(It,A); |
398 | 79 Mt = kron(Mt,Ix); |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
80 obj.M = Mt - Mx; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
81 %==============================================% |
398 | 82 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
83 % LU factorization |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
84 [obj.L,obj.U,obj.P,obj.Q] = lu(obj.M); |
398 | 85 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
86 % Pretend that the initial condition is the last level |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
87 % of a previous step. |
407
e1db62d14835
Fixed bug with initial data and made gauss the default type
Martin Almquist <martin.almquist@it.uu.se>
parents:
406
diff
changeset
|
88 obj.v = 1/(e_r'*e_r) * obj.Et_r * v0; |
398 | 89 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
90 end |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
91 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
92 function [v,t] = getV(obj) |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
93 v = obj.Et_r' * obj.v; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
94 t = obj.t; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
95 end |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
96 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
97 function obj = step(obj) |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
98 obj.v = time.sbp.sbpintime(obj.v, obj.t, obj.nodes,... |
398 | 99 obj.penalty, obj.f, obj.blockSize,... |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
100 obj.Et_r,... |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
101 obj.L, obj.U, obj.P, obj.Q); |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
102 obj.t = obj.t + obj.k; |
398 | 103 obj.n = obj.n + 1; |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
104 end |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
105 end |
398 | 106 |
400
14f2be4fe9c1
Add function to center colorlimits.
Jonatan Werpers <jonatan@werpers.com>
parents:
365
diff
changeset
|
107 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
108 methods(Static) |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
109 function N = smallestBlockSize(order,TYPE) |
407
e1db62d14835
Fixed bug with initial data and made gauss the default type
Martin Almquist <martin.almquist@it.uu.se>
parents:
406
diff
changeset
|
110 default_arg('TYPE','gauss') |
398 | 111 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
112 switch TYPE |
398 | 113 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
114 case 'equidistant' |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
115 switch order |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
116 case 2 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
117 N = 2; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
118 case 4 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
119 N = 8; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
120 case 6 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
121 N = 12; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
122 case 8 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
123 N = 16; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
124 case 10 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
125 N = 20; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
126 case 12 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
127 N = 24; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
128 otherwise |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
129 error('Operator does not exist'); |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
130 end |
398 | 131 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
132 case 'optimal' |
398 | 133 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
134 switch order |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
135 case 4 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
136 N = 8; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
137 case 6 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
138 N = 12; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
139 case 8 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
140 N = 16; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
141 case 10 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
142 N = 20; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
143 case 12 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
144 N = 24; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
145 otherwise |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
146 error('Operator does not exist'); |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
147 end |
398 | 148 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
149 case 'minimal' |
398 | 150 |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
151 switch order |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
152 case 4 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
153 N = 6; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
154 case 6 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
155 N = 10; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
156 case 8 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
157 N = 12; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
158 case 10 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
159 N = 16; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
160 case 12 |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
161 N = 20; |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
162 otherwise |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
163 error('Operator does not exist'); |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
164 end |
406
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
165 case 'gauss' |
9fd9b1bea3d2
Added D1Gauss to SBPInTime
Martin Almquist <martin.almquist@it.uu.se>
parents:
398
diff
changeset
|
166 N = 4; |
365
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
167 end |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
168 end |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
169 end |
f908ce064f35
Added SBP in time timestepper.
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
170 end |