Mercurial > repos > public > sbplib
annotate +scheme/Schrodinger2dCurve.m @ 499:f1465e6aeb26 feature/quantumTriangles
Commit before changing branch
author | Ylva Rydin <ylva.rydin@telia.com> |
---|---|
date | Mon, 06 Mar 2017 17:23:52 +0100 |
parents | 324c927d8b1d |
children | 83734c26b8e3 |
rev | line source |
---|---|
490 | 1 classdef Schrodinger2dCurve < scheme.Scheme |
2 properties | |
3 m % Number of points in each direction, possibly a vector | |
4 h % Grid spacing | |
5 | |
6 grid | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
7 xm, ym |
490 | 8 |
9 order % Order accuracy for the approximation | |
10 | |
11 D % non-stabalized scheme operator | |
12 M % Derivative norm | |
13 H % Discrete norm | |
14 Hi | |
15 H_u, H_v % Norms in the x and y directions | |
16 Hu,Hv % Kroneckerd norms. 1'*Hx*v corresponds to integration in the x dir. | |
17 Hi_u, Hi_v | |
18 Hiu, Hiv | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
19 D1_v, D1_u |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
20 D2_v, D2_u |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
21 Du, Dv |
492
6b95a894cbd7
fixed a bug in the metric coefficients but something is wrong at the boundaries
Ylva Rydin <ylva.rydin@telia.com>
parents:
491
diff
changeset
|
22 x,y |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
23 b1, b2 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
24 b1_u,b2_v |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
25 DU, DV, DUU, DUV, DVU, DVV |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
26 |
490 | 27 e_w, e_e, e_s, e_n |
28 du_w, dv_w | |
29 du_e, dv_e | |
30 du_s, dv_s | |
31 du_n, dv_n | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
32 g_1, g_2 |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
33 c |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
34 ind |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
35 t_up |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
36 |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
37 a11, a12, a22 |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
38 m_tot, m_u, m_v |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
39 p |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
40 Ji |
490 | 41 end |
42 | |
43 methods | |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
44 function obj = Schrodinger2dCurve(g ,order, opSet,p) |
490 | 45 default_arg('opSet',@sbp.D2Variable); |
46 default_arg('c', 1); | |
47 | |
48 obj.p=p; | |
49 obj.c=1; | |
50 | |
51 m = g.size(); | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
52 obj.m_u = m(1); |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
53 obj.m_v = m(2); |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
54 obj.m_tot = g.N(); |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
55 obj.grid=g; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
56 |
490 | 57 h = g.scaling(); |
58 h_u = h(1); | |
59 h_v = h(2); | |
60 | |
61 % Operators | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
62 ops_u = opSet(obj.m_u, {0, 1}, order); |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
63 ops_v = opSet(obj.m_v, {0, 1}, order); |
490 | 64 |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
65 I_u = speye(obj.m_u); |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
66 I_v = speye(obj.m_v); |
490 | 67 |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
68 obj.D1_u = ops_u.D1; |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
69 obj.D2_u = ops_u.D2; |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
70 |
490 | 71 H_u = ops_u.H; |
72 Hi_u = ops_u.HI; | |
73 e_l_u = ops_u.e_l; | |
74 e_r_u = ops_u.e_r; | |
75 d1_l_u = ops_u.d1_l; | |
76 d1_r_u = ops_u.d1_r; | |
77 | |
78 obj.D1_v = ops_v.D1; | |
79 obj.D2_v = ops_v.D2; | |
80 H_v = ops_v.H; | |
81 Hi_v = ops_v.HI; | |
82 e_l_v = ops_v.e_l; | |
83 e_r_v = ops_v.e_r; | |
84 d1_l_v = ops_v.d1_l; | |
85 d1_r_v = ops_v.d1_r; | |
86 | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
87 obj.Du = kr(obj.D1_u,I_v); |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
88 obj.Dv = kr(I_u,obj.D1_v); |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
89 |
490 | 90 obj.H = kr(H_u,H_v); |
91 obj.Hi = kr(Hi_u,Hi_v); | |
92 obj.Hu = kr(H_u,I_v); | |
93 obj.Hv = kr(I_u,H_v); | |
94 obj.Hiu = kr(Hi_u,I_v); | |
95 obj.Hiv = kr(I_u,Hi_v); | |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
96 |
490 | 97 obj.e_w = kr(e_l_u,I_v); |
98 obj.e_e = kr(e_r_u,I_v); | |
99 obj.e_s = kr(I_u,e_l_v); | |
100 obj.e_n = kr(I_u,e_r_v); | |
101 obj.du_w = kr(d1_l_u,I_v); | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
102 obj.dv_w = (obj.e_w'*obj.Dv)'; |
490 | 103 obj.du_e = kr(d1_r_u,I_v); |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
104 obj.dv_e = (obj.e_e'*obj.Dv)'; |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
105 obj.du_s = (obj.e_s'*obj.Du)'; |
490 | 106 obj.dv_s = kr(I_u,d1_l_v); |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
107 obj.du_n = (obj.e_n'*obj.Du)'; |
490 | 108 obj.dv_n = kr(I_u,d1_r_v); |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
109 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
110 obj.DUU = sparse(obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
111 obj.DVV = sparse(obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
112 obj.ind = grid.funcToMatrix(obj.grid, 1:obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
113 |
490 | 114 obj.m = m; |
115 obj.h = [h_u h_v]; | |
116 obj.order = order; | |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
117 obj.D = @(t)obj.d_fun(t); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
118 obj.variable_update(0); |
490 | 119 end |
120 | |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
121 function [D] = d_fun(obj,t) |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
122 % obj.update_vairables(t); In driscretization? |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
123 D = obj.Ji*(-1/2*(obj.b1*obj.Du-obj.b1_u+obj.Du*obj.b1) - 1/2*(obj.b2*obj.Dv - obj.b2_v +obj.Dv*obj.b2) + 1i*obj.c^2*(obj.DUU + obj.DUV + obj.DVU + obj.DVV)); |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
124 |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
125 end |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
126 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
127 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
128 function [D ]= variable_update(obj,t) |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
129 % Metric derivatives |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
130 if(obj.t_up == t) |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
131 return |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
132 else |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
133 ti = parametrization.Ti.points(obj.p{1}(t),obj.p{2}(t),obj.p{3}(t),obj.p{4}(t)); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
134 ti_tau = parametrization.Ti.points(obj.p{5}(t),obj.p{6}(t),obj.p{7}(t),obj.p{8}(t)); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
135 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
136 lcoords=points(obj.grid); |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
137 [obj.xm,obj.ym]= ti.map(lcoords(1:obj.m_v:end,1),lcoords(1:obj.m_v,2)); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
138 [x_tau,y_tau]= ti_tau.map(lcoords(1:obj.m_v:end,1),lcoords(1:obj.m_v,2)); |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
139 x = reshape(obj.xm,obj.m_tot,1); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
140 y = reshape(obj.ym,obj.m_tot,1); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
141 obj.x = x; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
142 obj.y = y; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
143 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
144 x_tau = reshape(x_tau,obj.m_tot,1); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
145 y_tau = reshape(y_tau,obj.m_tot,1); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
146 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
147 x_u = obj.Du*x; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
148 x_v = obj.Dv*x; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
149 y_u = obj.Du*y; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
150 y_v = obj.Dv*y; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
151 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
152 J = x_u.*y_v - x_v.*y_u; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
153 a11 = 1./J.* (x_v.^2 + y_v.^2); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
154 a12 = -1./J .* (x_u.*x_v + y_u.*y_v); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
155 a22 = 1./J .* (x_u.^2 + y_u.^2); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
156 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
157 obj.a11 = a11; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
158 obj.a12 = a12; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
159 obj.a22 = a22; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
160 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
161 % Assemble full operators |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
162 L_12 = spdiags(a12, 0, obj.m_tot, obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
163 obj.DUV = obj.Du*L_12*obj.Dv; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
164 obj.DVU = obj.Dv*L_12*obj.Du; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
165 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
166 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
167 for i = 1:obj.m_v |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
168 D = obj.D2_u(a11(obj.ind(:,i))); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
169 p = obj.ind(:,i); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
170 obj.DUU(p,p) = D; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
171 end |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
172 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
173 for i = 1:obj.m_u |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
174 D = obj.D2_v(a22(obj.ind(i,:))); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
175 p = obj.ind(i,:); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
176 obj.DVV(p,p) = D; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
177 end |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
178 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
179 Ji = spdiags(1./J, 0, obj.m_tot, obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
180 obj.Ji = Ji; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
181 obj.g_1 = x_v.*y_tau-x_tau.*y_v; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
182 obj.g_2 = x_tau.*y_u - y_tau.*x_u; |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
183 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
184 obj.b1 = spdiags(obj.g_1, 0, obj.m_tot, obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
185 obj.b2 = spdiags(obj.g_2, 0, obj.m_tot, obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
186 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
187 obj.b1_u = spdiags(obj.Du*obj.g_1, 0, obj.m_tot, obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
188 obj.b2_v = spdiags(obj.Dv*obj.g_2, 0, obj.m_tot, obj.m_tot); |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
189 obj.t_up=t; |
490 | 190 end |
191 end | |
192 | |
193 % Closure functions return the opertors applied to the own doamin to close the boundary | |
194 % Penalty functions return the opertors to force the solution. In the case of an interface it returns the operator applied to the other doamin. | |
195 % boundary is a string specifying the boundary e.g. 'l','r' or 'e','w','n','s'. | |
196 % type is a string specifying the type of boundary condition if there are several. | |
197 % data is a function returning the data that should be applied at the boundary. | |
198 % neighbour_scheme is an instance of Scheme that should be interfaced to. | |
199 % neighbour_boundary is a string specifying which boundary to interface to. | |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
200 function [closure, penalty] = boundary_condition(obj, boundary,~) |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
201 [e, d_n, d_t, coeff_t, coeff_n s, halfnorm_inv_n, halfnorm_inv_t, halfnorm_t,g] = obj.get_boundary_ops(boundary); |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
202 |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
203 a_t = @(t) spdiag(coeff_t(t)); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
204 a_n = @(t) spdiag(coeff_n(t)); |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
205 |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
206 |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
207 F = @(t)(s * a_n(t)*d_n' + s * a_t(t) *d_t')'; |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
208 tau1 = 1; |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
209 a = @(t)spdiag(g(t)); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
210 tau2 = @(t) (-1*s*a(t) - abs(a(t)))/4; |
490 | 211 |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
212 penalty_parameter_1 = @(t) 1*1i*halfnorm_inv_n*halfnorm_inv_t*F(t)*e'*halfnorm_t*e; |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
213 penalty_parameter_2 = @(t) halfnorm_inv_n*e*tau2(t); |
490 | 214 |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
215 closure = @(t) obj.Ji*obj.c^2 * penalty_parameter_1(t)*e' + obj.Ji* penalty_parameter_2(t)*e'; |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
216 penalty = @(t) -obj.Ji*obj.c^2 * penalty_parameter_1(t)*e'- obj.Ji*penalty_parameter_2(t)*e'; |
490 | 217 |
218 end | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
219 |
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
220 function [closure, penalty] = interface(obj,boundary,neighbour_scheme,neighbour_boundary) |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
221 % u denotes the solution in the own domain |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
222 % v denotes the solution in the neighbour domain |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
223 [e_u, d_n_u, d_t_u, coeff_t_u, coeff_n_u,s_u, halfnorm_inv_u_n, halfnorm_inv_u_t, halfnorm_u_t,gamm_u] = obj.get_boundary_ops(boundary); |
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
224 [e_v, d_n_v, d_t_v, coeff_t_v, coeff_n_v s_v, halfnorm_inv_v_n, halfnorm_inv_v_t, halfnorm_v_t,gamm_v] = neighbour_scheme.get_boundary_ops(neighbour_boundary); |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
225 |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
226 a_n_u = @(t) spdiag(coeff_n_u(t)); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
227 a_t_u = @(t) spdiag(coeff_t_u(t)); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
228 a_n_v = @(t) spdiag(coeff_n_v(t)); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
229 a_t_v = @(t) spdiag(coeff_t_v(t)); |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
230 |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
231 |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
232 F_u = @(t)(a_n_u(t)*d_n_u' + a_t_u(t)*d_t_u')'; |
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
233 F_v = @(t)(a_n_v(t)*d_n_v' + a_t_v(t)*d_t_v')'; |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
234 |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
235 a = @(t)spdiag(gamm_u(t)); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
236 |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
237 tau = s_u*1*1i/2; |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
238 sig = -s_u*1*1i/2; |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
239 gamm = @(t) (-s_u*a(t))/2; |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
240 |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
241 penalty_parameter_1 = @(t) halfnorm_inv_u_n*(tau*halfnorm_inv_u_t*F_u(t)*e_u'*halfnorm_u_t*e_u); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
242 penalty_parameter_2 = @(t) halfnorm_inv_u_n * e_u * (sig ); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
243 penalty_parameter_3 = @(t) halfnorm_inv_u_n * e_u * (gamm(t) ); |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
244 |
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
245 |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
246 closure =@(t) obj.Ji*obj.c^2 * ( penalty_parameter_1(t)*e_u' + penalty_parameter_2(t)*F_u(t)' + penalty_parameter_3(t)*e_u'); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
247 penalty =@(t) obj.Ji*obj.c^2 * ( -penalty_parameter_1(t)*e_v' - penalty_parameter_2(t)*F_v(t)' - penalty_parameter_3(t)*e_v'); |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
248 end |
490 | 249 |
497
4905446f165e
Added 2D interface to shrodinger
Ylva Rydin <ylva.rydin@telia.com>
parents:
493
diff
changeset
|
250 |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
251 function [e, d_n, d_t, coeff_t,coeff_n, s, halfnorm_inv_n, halfnorm_inv_t, halfnorm_t,g] = get_boundary_ops(obj, boundary) |
490 | 252 |
253 % gridMatrix = zeros(obj.m(2),obj.m(1)); | |
254 % gridMatrix(:) = 1:numel(gridMatrix); | |
255 | |
256 ind = grid.funcToMatrix(obj.grid, 1:prod(obj.m)); | |
257 | |
258 switch boundary | |
259 case 'w' | |
260 e = obj.e_w; | |
261 d_n = obj.du_w; | |
262 d_t = obj.dv_w; | |
263 s = -1; | |
264 | |
265 I = ind(1,:); | |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
266 coeff_t = @(t)obj.a12(I); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
267 coeff_n =@(t) obj.a11(I); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
268 g = @(t)obj.g_1(I); |
490 | 269 case 'e' |
270 e = obj.e_e; | |
271 d_n = obj.du_e; | |
272 d_t = obj.dv_e; | |
273 s = 1; | |
274 | |
275 I = ind(end,:); | |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
276 coeff_t = @(t)obj.a12(I); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
277 coeff_n = @(t)obj.a11(I); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
278 g = @(t)obj.g_1(I); |
490 | 279 case 's' |
280 e = obj.e_s; | |
281 d_n = obj.dv_s; | |
282 d_t = obj.du_s; | |
283 s = -1; | |
284 | |
285 I = ind(:,1)'; | |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
286 coeff_t = @(t)obj.a12(I); |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
287 coeff_n = @(t)obj.a22(I); |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
288 g = @(t)obj.g_2(I); |
490 | 289 case 'n' |
290 e = obj.e_n; | |
291 d_n = obj.dv_n; | |
292 d_t = obj.du_n; | |
293 s = 1; | |
294 | |
295 I = ind(:,end)'; | |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
296 coeff_t = @(t)obj.a12(I); |
499
f1465e6aeb26
Commit before changing branch
Ylva Rydin <ylva.rydin@telia.com>
parents:
498
diff
changeset
|
297 coeff_n = @(t)obj.a22(I); |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
298 g = @(t)obj.g_2(I); |
490 | 299 otherwise |
300 error('No such boundary: boundary = %s',boundary); | |
301 end | |
302 | |
303 switch boundary | |
304 case {'w','e'} | |
305 halfnorm_inv_n = obj.Hiu; | |
306 halfnorm_inv_t = obj.Hiv; | |
307 halfnorm_t = obj.Hv; | |
491
26125cfefe11
Schrodinger 2d runs without exploding with fixed coordinates
Ylva Rydin <ylva.rydin@telia.com>
parents:
490
diff
changeset
|
308 |
490 | 309 case {'s','n'} |
310 halfnorm_inv_n = obj.Hiv; | |
311 halfnorm_inv_t = obj.Hiu; | |
312 halfnorm_t = obj.Hu; | |
313 end | |
314 end | |
315 | |
316 function N = size(obj) | |
317 N = prod(obj.m); | |
318 end | |
319 end | |
498
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
320 methods(Static) |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
321 % Calculates the matrcis need for the inteface coupling between boundary bound_u of scheme schm_u |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
322 % and bound_v of scheme schm_v. |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
323 % [uu, uv, vv, vu] = inteface_couplong(A,'r',B,'l') |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
324 function [uu, uv, vv, vu] = interface_coupling(schm_u,bound_u,schm_v,bound_v) |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
325 [uu,uv] = schm_u.interface(bound_u,schm_v,bound_v); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
326 [vv,vu] = schm_v.interface(bound_v,schm_u,bound_u); |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
327 end |
324c927d8b1d
chnaged sbp interfacein 1d among many things
Ylva Rydin <ylva.rydin@telia.com>
parents:
497
diff
changeset
|
328 end |
490 | 329 end |