1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
% Initialize structure for packing and unpacking given their mask.
%
% s = statevector_init(mask1, mask2, ...)
%
% Initialize structure for packing and unpacking
% multiple variables given their corresponding land-sea mask.
%
% Input:
% mask1, mask2,...: land-sea mask for variable 1,2,... Sea grid points correspond to one and land grid points to zero.
% Every mask can have a different shape.
%
% Output:
% s: structure to be used with statevector_pack and statevector_unpack.
%
% Note:
% see also statevector_pack, statevector_unpack
% Author: Alexander Barth, 2009 <a.barth@ulg.ac.be>
% License: GPL 2 or later
function s = statevector_init(varargin)
s.nvar = nargin;
for i=1:nargin
mask = varargin{i};
s.mask{i} = mask;
s.numels(i) = sum(mask(:) == 1);
s.numels_all(i) = numel(mask);
s.size{i} = size(mask);
end
s.ind = [0 cumsum(s.numels)];
s.n = s.ind(end);
%!test
%! mask = rand(10,10) > .5;
%! mask_u = rand(9,10) > .5;
%! mask_v = rand(10,9) > .5;
%! sv = statevector_init(mask,mask_u,mask_v);
%! var = rand(10,10);
%! var(mask==0) = 0;
%! var_u = rand(9,10);
%! var_u(mask_u==0) = 0;
%! var(mask==0) = 0;
%! var_v = rand(10,9);
%! var_v(mask_v==0) = 0;
%! [E] = statevector_pack(sv,var,var_u,var_v);
%! [Ezeta2,Eu2,Ev2] = statevector_unpack(sv,E);
%! assert(Ezeta2,var)
%! assert(Eu2,var_u)
%! assert(Ev2,var_v)
% Copyright (C) 2009 Alexander Barth <a.barth@ulg.ac.be>
%
% This program is free software; you can redistribute it and/or modify
% it under the terms of the GNU General Public License as published by
% the Free Software Foundation; either version 2 of the License, or
% (at your option) any later version.
%
% This program is distributed in the hope that it will be useful,
% but WITHOUT ANY WARRANTY; without even the implied warranty of
% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
% GNU General Public License for more details.
%
% You should have received a copy of the GNU General Public License
% along with this program; If not, see <http://www.gnu.org/licenses/>.
% LocalWords: statevector init GPL
|
