Mercurial > repos > public > sbplib

annotate +multiblock/stitchSchemes.m @ 496:437fad4a47e1 feature/quantumTriangles

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add 1d interface for shrodinger in 1d
author Ylva Rydin <ylva.rydin@telia.com>
date Fri, 24 Feb 2017 13:58:18 +0100
parents 419ec303e97d
children 324c927d8b1d
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 % Stitch schemes together given connection matrix and BC vector.
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
2 % schmHand - function_handle to a Scheme constructor
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
3 % order - order of accuracy
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4 % schmParam - cell array of extra parameters sent to each Scheme stored as cell arrays
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5 % blocks - block definitions, On whatever form the scheme expects.
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
6 % ms - grid points in each direction for each block. Ex {[10,10], [10, 20]}
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 % conn - connection matrix
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 % bound - boundary condition vector, array of structs with fields w,e,s,n
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 % each field with a parameter array that is sent to schm.boundary_condition
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
10 %
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11 % Output parameters are cell arrays and cell matrices.
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
12 %
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
13 % Ex: [schms, D, H] = stitchSchemes(schmHand, order, schmParam, blocks, ms, conn, bound)
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
14 function [schms, D, H] = stitchSchemes(schmHand, order, schmParam, grids, conn, bound,timeDep)
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
15 default_arg('schmParam',[]);
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
16 default_arg('timeDep','N');
181
419ec303e97d Made some local changes to stichSchemes.
Jonatan Werpers <jonatan@werpers.com>
parents: 120
diff changeset
17 n_blocks = numel(grids);
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19 % Creating Schemes
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
20 for i = 1:n_blocks
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
21 if isempty(schmParam);
181
419ec303e97d Made some local changes to stichSchemes.
Jonatan Werpers <jonatan@werpers.com>
parents: 120
diff changeset
22 schms{i} = schmHand(grids{i},order,[]);
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
23 elseif ~iscell(schmParam)
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
24 param = schmParam(i);
181
419ec303e97d Made some local changes to stichSchemes.
Jonatan Werpers <jonatan@werpers.com>
parents: 120
diff changeset
25 schms{i} = schmHand(grids{i},order,param);
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
26 else
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
27 param = schmParam{i};
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
28 if iscell(param)
181
419ec303e97d Made some local changes to stichSchemes.
Jonatan Werpers <jonatan@werpers.com>
parents: 120
diff changeset
29 schms{i} = schmHand(grids{i},order,param{:});
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
30 else
181
419ec303e97d Made some local changes to stichSchemes.
Jonatan Werpers <jonatan@werpers.com>
parents: 120
diff changeset
31 schms{i} = schmHand(grids{i},order,param);
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
32 end
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33 end
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
35 % class(schmParam)
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36 % class(ms)
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37 % class(blocks)
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
38 % class(schmParam{i})
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
39 % class(ms)
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40
76
5c569cbef49e Added more input parameter handling and fixed some bugs.
Jonatan Werpers <jonatan@werpers.com>
parents: 0
diff changeset
41
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42 end
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
43
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
44
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45 % Total norm
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46 H = cell(n_blocks,n_blocks);
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47 for i = 1:n_blocks
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48 H{i,i} = schms{i}.H;
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49 end
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
50
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 %% Total system matrix
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
52
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53 % Differentiation terms
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
54 D = cell(n_blocks,n_blocks);
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
55 for i = 1:n_blocks
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
56 D{i,i} = schms{i}.D;
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
57 end
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
58
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
59 % Boundary penalty terms
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
60 switch timeDep
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
61 case {'n','no','N','No'}
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
62 for i = 1:n_blocks
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
63 if ~isstruct(bound{i})
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
64 continue
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
65 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
66
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
67 fn = fieldnames(bound{i});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
68 for j = 1:length(fn)
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
69 bc = bound{i}.(fn{j});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
70 if isempty(bc)
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
71 continue
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
72 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
73
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
74 [closure, ~] = schms{i}.boundary_condition(fn{j},bc{:});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
75 D{i,i} = D{i,i}+closure;
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
76 end
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
77 end
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
78
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
79 % Interface penalty terms
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
80 for i = 1:n_blocks
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
81 for j = 1:n_blocks
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
82 intf = conn{i,j};
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
83 if isempty(intf)
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
84 continue
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
85 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
86
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
87 [uu,uv,vv,vu] = schms{i}.interface_coupling(schms{i},intf{1},schms{j},intf{2});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
88 D{i,i} = D{i,i} + uu;
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
89 D{i,j} = uv;
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
90 D{j,j} = D{j,j} + vv;
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
91 D{j,i} = vu;
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
92 end
0
48b6fb693025 Initial commit.
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
93 end
496
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
94 case {'y','yes','Y','Yes'}
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
95 for i = 1:n_blocks
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
96 if ~isstruct(bound{i})
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
97 continue
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
98 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
99
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
100 fn = fieldnames(bound{i});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
101 for j = 1:length(fn)
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
102 bc = bound{i}.(fn{j});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
103 if isempty(bc)
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
104 continue
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
105 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
106
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
107 [closure, ~] = schms{i}.boundary_condition(fn{j},bc{:});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
108 D{i,i} =@(t) D{i,i}(t) + closure(t);
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
109 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
110 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
111
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
112 % Interface penalty terms
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
113 for i = 1:n_blocks
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
114 for j = 1:n_blocks
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
115 intf = conn{i,j};
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
116 if isempty(intf)
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
117 continue
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
118 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
119
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
120 [uu,uv,vv,vu] = schms{i}.interface_coupling(schms{i},intf{1},schms{j},intf{2});
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
121 D{i,i} = @(t) D{i,i}(t) + uu(t);
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
122 D{i,j} = uv;
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
123 D{j,j} = @(t)D{j,j}(t) + vv(t);
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
124 D{j,i} = vu;
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
125 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
126 end
437fad4a47e1 Add 1d interface for shrodinger in 1d
Ylva Rydin <ylva.rydin@telia.com>
parents: 181
diff changeset
127 end