Mercurial > repos > public > sbplib

annotate +scheme/Elastic2dVariable.m @ 726:37d5d69b1a0d feature/poroelastic

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Refactor BC code in Elastic2dVariable
author Martin Almquist <malmquist@stanford.edu>
date Thu, 19 Apr 2018 17:06:27 -0700
parents 60eb7f46d8d9
children 6d5953fc090e
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
687
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
1 classdef Elastic2dVariable < scheme.Scheme
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
2
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
3 % Discretizes the elastic wave equation:
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
4 % rho u_{i,tt} = di lambda dj u_j + dj mu di u_j + dj mu dj u_i
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
5 % opSet should be cell array of opSets, one per dimension. This
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
6 % is useful if we have periodic BC in one direction.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
7
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
8 properties
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
9 m % Number of points in each direction, possibly a vector
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
10 h % Grid spacing
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
11
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
12 grid
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
13 dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
14
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
15 order % Order of accuracy for the approximation
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
16
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
17 % Diagonal matrices for varible coefficients
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
18 LAMBDA % Variable coefficient, related to dilation
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
19 MU % Shear modulus, variable coefficient
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
20 RHO, RHOi % Density, variable
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
21
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
22 D % Total operator
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
23 D1 % First derivatives
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
24
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
25 % Second derivatives
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
26 D2_lambda
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
27 D2_mu
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
28
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
29 % Traction operators used for BC
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
30 T_l, T_r
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
31 tau_l, tau_r
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
32
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
33 H, Hi % Inner products
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
34 phi % Borrowing constant for (d1 - e^T*D1) from R
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
35 gamma % Borrowing constant for d1 from M
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
36 H11 % First element of H
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
37 e_l, e_r
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
38 d1_l, d1_r % Normal derivatives at the boundary
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
39 E % E{i}^T picks out component i
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
40
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
41 H_boundary % Boundary inner products
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
42
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
43 % Kroneckered norms and coefficients
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
44 RHOi_kron
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
45 Hi_kron
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
46 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
47
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
48 methods
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
49
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
50 function obj = Elastic2dVariable(g ,order, lambda_fun, mu_fun, rho_fun, opSet)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
51 default_arg('opSet',{@sbp.D2Variable, @sbp.D2Variable});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
52 default_arg('lambda_fun', @(x,y) 0*x+1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
53 default_arg('mu_fun', @(x,y) 0*x+1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
54 default_arg('rho_fun', @(x,y) 0*x+1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
55 dim = 2;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
56
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
57 assert(isa(g, 'grid.Cartesian'))
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
58
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
59 lambda = grid.evalOn(g, lambda_fun);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
60 mu = grid.evalOn(g, mu_fun);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
61 rho = grid.evalOn(g, rho_fun);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
62 m = g.size();
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
63 m_tot = g.N();
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
64
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
65 h = g.scaling();
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
66 lim = g.lim;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
67
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
68 % 1D operators
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
69 ops = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
70 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
71 ops{i} = opSet{i}(m(i), lim{i}, order);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
72 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
73
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
74 % Borrowing constants
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
75 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
76 beta = ops{i}.borrowing.R.delta_D;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
77 obj.H11{i} = ops{i}.borrowing.H11;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
78 obj.phi{i} = beta/obj.H11{i};
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
79 obj.gamma{i} = ops{i}.borrowing.M.d1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
80 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
81
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
82 I = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
83 D1 = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
84 D2 = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
85 H = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
86 Hi = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
87 e_l = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
88 e_r = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
89 d1_l = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
90 d1_r = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
91
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
92 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
93 I{i} = speye(m(i));
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
94 D1{i} = ops{i}.D1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
95 D2{i} = ops{i}.D2;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
96 H{i} = ops{i}.H;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
97 Hi{i} = ops{i}.HI;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
98 e_l{i} = ops{i}.e_l;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
99 e_r{i} = ops{i}.e_r;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
100 d1_l{i} = ops{i}.d1_l;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
101 d1_r{i} = ops{i}.d1_r;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
102 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
103
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
104 %====== Assemble full operators ========
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
105 LAMBDA = spdiag(lambda);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
106 obj.LAMBDA = LAMBDA;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
107 MU = spdiag(mu);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
108 obj.MU = MU;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
109 RHO = spdiag(rho);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
110 obj.RHO = RHO;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
111 obj.RHOi = inv(RHO);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
112
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
113 obj.D1 = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
114 obj.D2_lambda = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
115 obj.D2_mu = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
116 obj.e_l = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
117 obj.e_r = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
118 obj.d1_l = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
119 obj.d1_r = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
120
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
121 % D1
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
122 obj.D1{1} = kron(D1{1},I{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
123 obj.D1{2} = kron(I{1},D1{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
124
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
125 % Boundary operators
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
126 obj.e_l{1} = kron(e_l{1},I{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
127 obj.e_l{2} = kron(I{1},e_l{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
128 obj.e_r{1} = kron(e_r{1},I{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
129 obj.e_r{2} = kron(I{1},e_r{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
130
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
131 obj.d1_l{1} = kron(d1_l{1},I{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
132 obj.d1_l{2} = kron(I{1},d1_l{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
133 obj.d1_r{1} = kron(d1_r{1},I{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
134 obj.d1_r{2} = kron(I{1},d1_r{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
135
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
136 % D2
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
137 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
138 obj.D2_lambda{i} = sparse(m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
139 obj.D2_mu{i} = sparse(m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
140 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
141 ind = grid.funcToMatrix(g, 1:m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
142
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
143 for i = 1:m(2)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
144 D_lambda = D2{1}(lambda(ind(:,i)));
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
145 D_mu = D2{1}(mu(ind(:,i)));
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
146
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
147 p = ind(:,i);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
148 obj.D2_lambda{1}(p,p) = D_lambda;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
149 obj.D2_mu{1}(p,p) = D_mu;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
150 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
151
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
152 for i = 1:m(1)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
153 D_lambda = D2{2}(lambda(ind(i,:)));
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
154 D_mu = D2{2}(mu(ind(i,:)));
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
155
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
156 p = ind(i,:);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
157 obj.D2_lambda{2}(p,p) = D_lambda;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
158 obj.D2_mu{2}(p,p) = D_mu;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
159 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
160
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
161 % Quadratures
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
162 obj.H = kron(H{1},H{2});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
163 obj.Hi = inv(obj.H);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
164 obj.H_boundary = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
165 obj.H_boundary{1} = H{2};
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
166 obj.H_boundary{2} = H{1};
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
167
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
168 % E{i}^T picks out component i.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
169 E = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
170 I = speye(m_tot,m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
171 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
172 e = sparse(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
173 e(i) = 1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
174 E{i} = kron(I,e);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
175 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
176 obj.E = E;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
177
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
178 % Differentiation matrix D (without SAT)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
179 D2_lambda = obj.D2_lambda;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
180 D2_mu = obj.D2_mu;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
181 D1 = obj.D1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
182 D = sparse(dim*m_tot,dim*m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
183 d = @kroneckerDelta; % Kronecker delta
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
184 db = @(i,j) 1-d(i,j); % Logical not of Kronecker delta
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
185 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
186 for j = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
187 D = D + E{i}*inv(RHO)*( d(i,j)*D2_lambda{i}*E{j}' +...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
188 db(i,j)*D1{i}*LAMBDA*D1{j}*E{j}' ...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
189 );
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
190 D = D + E{i}*inv(RHO)*( d(i,j)*D2_mu{i}*E{j}' +...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
191 db(i,j)*D1{j}*MU*D1{i}*E{j}' + ...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
192 D2_mu{j}*E{i}' ...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
193 );
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
194 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
195 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
196 obj.D = D;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
197 %=========================================%
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
198
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
199 % Numerical traction operators for BC.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
200 % Because d1 =/= e0^T*D1, the numerical tractions are different
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
201 % at every boundary.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
202 T_l = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
203 T_r = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
204 tau_l = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
205 tau_r = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
206 % tau^{j}_i = sum_k T^{j}_{ik} u_k
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
207
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
208 d1_l = obj.d1_l;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
209 d1_r = obj.d1_r;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
210 e_l = obj.e_l;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
211 e_r = obj.e_r;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
212 D1 = obj.D1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
213
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
214 % Loop over boundaries
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
215 for j = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
216 T_l{j} = cell(dim,dim);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
217 T_r{j} = cell(dim,dim);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
218 tau_l{j} = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
219 tau_r{j} = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
220
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
221 % Loop over components
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
222 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
223 tau_l{j}{i} = sparse(m_tot,dim*m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
224 tau_r{j}{i} = sparse(m_tot,dim*m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
225 for k = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
226 T_l{j}{i,k} = ...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
227 -d(i,j)*LAMBDA*(d(i,k)*e_l{k}*d1_l{k}' + db(i,k)*D1{k})...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
228 -d(j,k)*MU*(d(i,j)*e_l{i}*d1_l{i}' + db(i,j)*D1{i})...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
229 -d(i,k)*MU*e_l{j}*d1_l{j}';
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
230
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
231 T_r{j}{i,k} = ...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
232 d(i,j)*LAMBDA*(d(i,k)*e_r{k}*d1_r{k}' + db(i,k)*D1{k})...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
233 +d(j,k)*MU*(d(i,j)*e_r{i}*d1_r{i}' + db(i,j)*D1{i})...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
234 +d(i,k)*MU*e_r{j}*d1_r{j}';
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
235
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
236 tau_l{j}{i} = tau_l{j}{i} + T_l{j}{i,k}*E{k}';
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
237 tau_r{j}{i} = tau_r{j}{i} + T_r{j}{i,k}*E{k}';
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
238 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
239
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
240 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
241 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
242 obj.T_l = T_l;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
243 obj.T_r = T_r;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
244 obj.tau_l = tau_l;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
245 obj.tau_r = tau_r;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
246
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
247 % Kroneckered norms and coefficients
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
248 I_dim = speye(dim);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
249 obj.RHOi_kron = kron(obj.RHOi, I_dim);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
250 obj.Hi_kron = kron(obj.Hi, I_dim);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
251
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
252 % Misc.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
253 obj.m = m;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
254 obj.h = h;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
255 obj.order = order;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
256 obj.grid = g;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
257 obj.dim = dim;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
258
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
259 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
260
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
261
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
262 % Closure functions return the operators applied to the own domain to close the boundary
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
263 % Penalty functions return the operators to force the solution. In the case of an interface it returns the operator applied to the other doamin.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
264 % boundary is a string specifying the boundary e.g. 'l','r' or 'e','w','n','s'.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
265 % type is a cell array of strings specifying the type of boundary condition for each component.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
266 % data is a function returning the data that should be applied at the boundary.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
267 % neighbour_scheme is an instance of Scheme that should be interfaced to.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
268 % neighbour_boundary is a string specifying which boundary to interface to.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
269 function [closure, penalty] = boundary_condition(obj, boundary, type, parameter)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
270 default_arg('type',{'free','free'});
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
271 default_arg('parameter', []);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
272
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
273 % j is the coordinate direction of the boundary
726
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
274 j = obj.get_boundary_number(boundary);
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
275 [e, T, tau] = obj.get_boundary_operator({'e','T','tau'}, boundary);
687
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
276
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
277 E = obj.E;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
278 Hi = obj.Hi;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
279 H_gamma = obj.H_boundary{j};
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
280 LAMBDA = obj.LAMBDA;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
281 MU = obj.MU;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
282 RHOi = obj.RHOi;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
283
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
284 dim = obj.dim;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
285 m_tot = obj.grid.N();
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
286
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
287 RHOi_kron = obj.RHOi_kron;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
288 Hi_kron = obj.Hi_kron;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
289
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
290 % Preallocate
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
291 closure = sparse(dim*m_tot, dim*m_tot);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
292 penalty = cell(dim,1);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
293 for k = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
294 penalty{k} = sparse(dim*m_tot, m_tot/obj.m(j));
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
295 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
296
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
297 % Loop over components that we (potentially) have different BC on
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
298 for k = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
299 switch type{k}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
300
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
301 % Dirichlet boundary condition
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
302 case {'D','d','dirichlet','Dirichlet'}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
303
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
304 tuning = 1.2;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
305 phi = obj.phi{j};
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
306 h = obj.h(j);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
307 h11 = obj.H11{j}*h;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
308 gamma = obj.gamma{j};
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
309
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
310 a_lambda = dim/h11 + 1/(h11*phi);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
311 a_mu_i = 2/(gamma*h);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
312 a_mu_ij = 2/h11 + 1/(h11*phi);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
313
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
314 d = @kroneckerDelta; % Kronecker delta
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
315 db = @(i,j) 1-d(i,j); % Logical not of Kronecker delta
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
316 alpha = @(i,j) tuning*( d(i,j)* a_lambda*LAMBDA ...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
317 + d(i,j)* a_mu_i*MU ...
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
318 + db(i,j)*a_mu_ij*MU );
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
319
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
320 % Loop over components that Dirichlet penalties end up on
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
321 for i = 1:dim
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
322 C = T{k,i};
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
323 A = -d(i,k)*alpha(i,j);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
324 B = A + C;
726
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
325 closure = closure + E{i}*RHOi*Hi*B'*e*H_gamma*(e'*E{k}' );
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
326 penalty{k} = penalty{k} - E{i}*RHOi*Hi*B'*e*H_gamma;
687
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
327 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
328
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
329 % Free boundary condition
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
330 case {'F','f','Free','free','traction','Traction','t','T'}
726
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
331 closure = closure - E{k}*RHOi*Hi*e*H_gamma* (e'*tau{k} );
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
332 penalty{k} = penalty{k} + E{k}*RHOi*Hi*e*H_gamma;
687
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
333
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
334 % Unknown boundary condition
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
335 otherwise
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
336 error('No such boundary condition: type = %s',type);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
337 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
338 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
339 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
340
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
341 function [closure, penalty] = interface(obj,boundary,neighbour_scheme,neighbour_boundary)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
342 % u denotes the solution in the own domain
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
343 % v denotes the solution in the neighbour domain
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
344 tuning = 1.2;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
345 % tuning = 20.2;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
346 error('Interface not implemented');
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
347 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
348
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
349 % Returns the coordinate number and outward normal component for the boundary specified by the string boundary.
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
350 function [j, nj] = get_boundary_number(obj, boundary)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
351
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
352 switch boundary
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
353 case {'w','W','west','West', 'e', 'E', 'east', 'East'}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
354 j = 1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
355 case {'s','S','south','South', 'n', 'N', 'north', 'North'}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
356 j = 2;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
357 otherwise
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
358 error('No such boundary: boundary = %s',boundary);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
359 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
360
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
361 switch boundary
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
362 case {'w','W','west','West','s','S','south','South'}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
363 nj = -1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
364 case {'e', 'E', 'east', 'East','n', 'N', 'north', 'North'}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
365 nj = 1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
366 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
367 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
368
726
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
369 % Returns the boundary operator op for the boundary specified by the string boundary.
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
370 % op: may be a cell array of strings
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
371 function [varargout] = get_boundary_operator(obj, op, boundary)
687
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
372
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
373 switch boundary
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
374 case {'w','W','west','West', 'e', 'E', 'east', 'East'}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
375 j = 1;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
376 case {'s','S','south','South', 'n', 'N', 'north', 'North'}
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
377 j = 2;
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
378 otherwise
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
379 error('No such boundary: boundary = %s',boundary);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
380 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
381
726
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
382 if ~iscell(op)
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
383 op = {op};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
384 end
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
385
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
386 for i = 1:length(op)
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
387 switch op{i}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
388 case 'e'
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
389 switch boundary
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
390 case {'w','W','west','West','s','S','south','South'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
391 varargout{i} = obj.e_l{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
392 case {'e', 'E', 'east', 'East','n', 'N', 'north', 'North'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
393 varargout{i} = obj.e_r{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
394 end
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
395 case 'd'
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
396 switch boundary
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
397 case {'w','W','west','West','s','S','south','South'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
398 varargout{i} = obj.d1_l{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
399 case {'e', 'E', 'east', 'East','n', 'N', 'north', 'North'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
400 varargout{i} = obj.d1_r{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
401 end
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
402 case 'H'
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
403 varargout{i} = obj.H_boundary{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
404 case 'T'
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
405 switch boundary
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
406 case {'w','W','west','West','s','S','south','South'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
407 varargout{i} = obj.T_l{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
408 case {'e', 'E', 'east', 'East','n', 'N', 'north', 'North'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
409 varargout{i} = obj.T_r{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
410 end
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
411 case 'tau'
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
412 switch boundary
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
413 case {'w','W','west','West','s','S','south','South'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
414 varargout{i} = obj.tau_l{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
415 case {'e', 'E', 'east', 'East','n', 'N', 'north', 'North'}
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
416 varargout{i} = obj.tau_r{j};
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
417 end
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
418 otherwise
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
419 error(['No such operator: operator = ' op{i}]);
37d5d69b1a0d Refactor BC code in Elastic2dVariable
Martin Almquist <malmquist@stanford.edu>
parents: 689
diff changeset
420 end
687
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
421 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
422
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
423 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
424
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
425 function N = size(obj)
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
426 N = prod(obj.m);
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
427 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
428 end
e8fc3aa1faf6 Rename elastic scheme.
Martin Almquist <malmquist@stanford.edu>
parents:
diff changeset
429 end