591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
1 classdef Utux2D < scheme.Scheme
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
2 properties
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
3 m % Number of points in each direction, possibly a vector
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
4 h % Grid spacing
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
5 grid % Grid
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
6 order % Order accuracy for the approximation
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
7 v0 % Initial data
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
8
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
9 a % Wave speed a = [a1, a2];
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
10
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
11 H % Discrete norm
|
595
|
12 H_x, H_y % Norms in the x and y directions
|
|
13 Hi, Hx, Hy, Hxi, Hyi % Kroneckered norms
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
14
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
15 % Derivatives
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
16 Dx, Dy
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
17
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
18 % Boundary operators
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
19 e_w, e_e, e_s, e_n
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
20
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
21 D % Total discrete operator
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
22
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
23 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
24
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
25
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
26 methods
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
27 function obj = Utux2D(g ,order, opSet, a)
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
28
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
29 default_arg('a',1/sqrt(2)*[1, 1]);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
30 default_arg('opSet',@sbp.D2Standard);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
31 assert(isa(g, 'grid.Cartesian'))
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
32
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
33 m = g.size();
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
34 m_x = m(1);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
35 m_y = m(2);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
36 m_tot = g.N();
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
37
|
595
|
38 xlim = {g.x{1}(1), g.x{1}(end)};
|
|
39 ylim = {g.x{2}(1), g.x{2}(end)};
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
40 obj.grid = g;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
41
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
42 % Operator sets
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
43 ops_x = opSet(m_x, xlim, order);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
44 ops_y = opSet(m_y, ylim, order);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
45 Ix = speye(m_x);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
46 Iy = speye(m_y);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
47
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
48 % Norms
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
49 Hx = ops_x.H;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
50 Hy = ops_y.H;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
51 Hxi = ops_x.HI;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
52 Hyi = ops_y.HI;
|
595
|
53
|
|
54 obj.H_x = Hx;
|
|
55 obj.H_y = Hy;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
56 obj.H = kron(Hx,Hy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
57 obj.Hi = kron(Hxi,Hyi);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
58 obj.Hx = kron(Hx,Iy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
59 obj.Hy = kron(Ix,Hy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
60 obj.Hxi = kron(Hxi,Iy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
61 obj.Hyi = kron(Ix,Hyi);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
62
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
63 % Derivatives
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
64 Dx = ops_x.D1;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
65 Dy = ops_y.D1;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
66 obj.Dx = kron(Dx,Iy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
67 obj.Dy = kron(Ix,Dy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
68
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
69 % Boundary operators
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
70 obj.e_w = kr(ops_x.e_l, Iy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
71 obj.e_e = kr(ops_x.e_r, Iy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
72 obj.e_s = kr(Ix, ops_y.e_l);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
73 obj.e_n = kr(Ix, ops_y.e_r);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
74
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
75 obj.m = m;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
76 obj.h = [ops_x.h ops_y.h];
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
77 obj.order = order;
|
595
|
78 obj.a = a;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
79 obj.D = -(a(1)*obj.Dx + a(2)*obj.Dy);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
80
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
81 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
82 % Closure functions return the opertors applied to the own domain to close the boundary
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
83 % Penalty functions return the opertors to force the solution. In the case of an interface it returns the operator applied to the other doamin.
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
84 % boundary is a string specifying the boundary e.g. 'l','r' or 'e','w','n','s'.
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
85 % type is a string specifying the type of boundary condition if there are several.
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
86 % data is a function returning the data that should be applied at the boundary.
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
87 % neighbour_scheme is an instance of Scheme that should be interfaced to.
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
88 % neighbour_boundary is a string specifying which boundary to interface to.
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
89 function [closure, penalty] = boundary_condition(obj,boundary,type)
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
90 default_arg('type','dirichlet');
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
91
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
92 sigma = -1; % Scalar penalty parameter
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
93 switch boundary
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
94 case {'w','W','west','West'}
|
595
|
95 tau = sigma*obj.a(1)*obj.e_w*obj.H_y;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
96 closure = obj.Hi*tau*obj.e_w';
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
97
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
98 case {'s','S','south','South'}
|
595
|
99 tau = sigma*obj.a(2)*obj.e_s*obj.H_x;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
100 closure = obj.Hi*tau*obj.e_s';
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
101 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
102 penalty = -obj.Hi*tau;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
103
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
104 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
105
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
106 function [closure, penalty] = interface(obj,boundary,neighbour_scheme,neighbour_boundary)
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
107
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
108 % Get neighbour boundary operator
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
109 switch neighbour_boundary
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
110 case {'e','E','east','East'}
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
111 e_neighbour = neighbour_scheme.e_e;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
112 case {'w','W','west','West'}
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
113 e_neighbour = neighbour_scheme.e_w;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
114 case {'n','N','north','North'}
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
115 e_neighbour = neighbour_scheme.e_n;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
116 case {'s','S','south','South'}
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
117 e_neighbour = neighbour_scheme.e_s;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
118 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
119
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
120 % Upwind coupling
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
121 sigma_ds = -1; %"Downstream" penalty
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
122 sigma_us = 0; %"Upstream" penalty
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
123
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
124 switch boundary
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
125 case {'w','W','west','West'}
|
595
|
126 tau = sigma_ds*obj.a(1)*obj.e_w*obj.H_y;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
127 closure = obj.Hi*tau*obj.e_w';
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
128 case {'e','E','east','East'}
|
595
|
129 tau = sigma_us*obj.a(1)*obj.e_e*obj.H_y;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
130 closure = obj.Hi*tau*obj.e_e';
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
131 case {'s','S','south','South'}
|
595
|
132 tau = sigma_ds*obj.a(2)*obj.e_s*obj.H_x;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
133 closure = obj.Hi*tau*obj.e_s';
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
134 case {'n','N','north','North'}
|
595
|
135 tau = sigma_us*obj.a(2)*obj.e_n*obj.H_x;
|
591
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
136 closure = obj.Hi*tau*obj.e_n';
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
137 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
138 penalty = -obj.Hi*tau*e_neighbour';
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
139
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
140 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
141
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
142 function N = size(obj)
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
143 N = obj.m;
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
144 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
145
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
146 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
147
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
148 methods(Static)
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
149 % Calculates the matrices needed for the inteface coupling between boundary bound_u of scheme schm_u
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
150 % and bound_v of scheme schm_v.
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
151 % [uu, uv, vv, vu] = inteface_coupling(A,'r',B,'l')
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
152 function [uu, uv, vv, vu] = interface_coupling(schm_u,bound_u,schm_v,bound_v)
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
153 [uu,uv] = schm_u.interface(bound_u,schm_v,bound_v);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
154 [vv,vu] = schm_v.interface(bound_v,schm_u,bound_u);
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
155 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
156 end
|
Martin Almquist <martin.almquist@it.uu.se>
parents:
diff
changeset
|
157 end |