/[MITgcm]/MITgcm_contrib/gael/matlab_class/gcmfaces_IO/convert2gcmfaces.m
ViewVC logotype

Contents of /MITgcm_contrib/gael/matlab_class/gcmfaces_IO/convert2gcmfaces.m

Parent Directory Parent Directory | Revision Log Revision Log | View Revision Graph Revision Graph


Revision 1.7 - (show annotations) (download)
Tue Jul 5 17:51:13 2011 UTC (14 years ago) by gforget
Branch: MAIN
Changes since 1.6: +27 -1 lines
- introduce mygrid.gcm2facesFast that speeds up convert2gcmfaces.
- mygrid.gcm2facesFast is originally set up to false in grid_load;
  then the grid is loaded, which allows the computation of
  mygrid.gcm2faces, mygrid.faces2gcmSize and mygrid.faces2gcm;
  at this point mygrid.gcm2facesFast is set to true, and any
  other call to convert2gcmfaces will use the faster approach.

1 function [v1]=convert2gcmfaces(v0,varargin);
2 %object: converts model output to gcmfaces object
3 % or vice versa
4 %input: v0 model output array (resp. gcmfaces object)
5 %output: v1 gcmfaces object (model output array)
6 %
7 %note: global mygrid parameters (nFaces, fileFormat) are used
8
9 input_list_check('convert2gcmfaces',nargin);
10
11 aa=whos('v0'); doGcm2Faces=strcmp(aa.class,'double');
12
13 global mygrid;
14
15 if doGcm2Faces&mygrid.gcm2facesFast;
16
17 [n1,n2,n3,n4,n5]=size(v0);
18
19 v0=reshape(v0,[n1*n2 n3*n4*n5]);
20 for iFace=1:mygrid.nFaces;
21 v1{iFace}=reshape(mygrid.gcm2faces{iFace}*v0,[size(mygrid.XC{iFace}) n3 n4 n5]);
22 end;
23
24 if mygrid.nFaces==1; gridType='ll';
25 elseif mygrid.nFaces==5; gridType='llc';
26 elseif mygrid.nFaces==6; gridType='cube';
27 end;
28 v1=gcmfaces(v1,gridType);
29
30 elseif ~doGcm2Faces&mygrid.gcm2facesFast;
31
32 [n1,n2,n3,n4,n5]=size(v0{1});
33
34 v1=zeros(mygrid.faces2gcmSize(1)*mygrid.faces2gcmSize(2),n3*n4*n5);
35 for iFace=1:mygrid.nFaces;
36 nn=size(mygrid.XC{iFace});
37 v1(mygrid.faces2gcm{iFace},:)=reshape(v0{iFace},[nn(1)*nn(2) n3*n4*n5]);
38 end;
39 v1=reshape(v1,[mygrid.faces2gcmSize n3 n4 n5]);
40
41 elseif doGcm2Faces;
42
43 [n1,n2,n3,n4,n5]=size(v0);
44
45 if strcmp(mygrid.fileFormat,'straight');
46 v1={v0};
47 elseif strcmp(mygrid.fileFormat,'cube');
48 for ii=1:6; v1{ii}=v0(n2*(ii-1)+[1:n2],:,:,:,:); end;
49 elseif strcmp(mygrid.fileFormat,'compact');
50
51 nn=size(v0,1);
52 pp=size(v0,2)/nn;
53 mm=(pp+4-mygrid.nFaces)/4*nn;
54
55 v00=reshape(v0,[nn*nn*pp n3*n4*n5]);
56 i0=1; i1=nn*mm; v1{1}=reshape(v00(i0:i1,:),[nn mm n3 n4 n5]);
57 i0=i1+1; i1=i1+nn*mm; v1{2}=reshape(v00(i0:i1,:),[nn mm n3 n4 n5]);
58 i0=i1+1; i1=i1+nn*nn; v1{3}=reshape(v00(i0:i1,:),[nn nn n3 n4 n5]);
59 i0=i1+1; i1=i1+nn*mm; v1{4}=reshape(v00(i0:i1,:),[mm nn n3 n4 n5]);
60 i0=i1+1; i1=i1+nn*mm; v1{5}=reshape(v00(i0:i1,:),[mm nn n3 n4 n5]);
61 if mygrid.nFaces==6;
62 i0=i1+1; i1=i1+nn*nn; v1{6}=reshape(v00(i0:i1,:),[nn nn n3 n4 n5]);
63 end;
64
65 end;
66
67 if mygrid.nFaces==1; gridType='ll';
68 elseif mygrid.nFaces==5; gridType='llc';
69 elseif mygrid.nFaces==6; gridType='cube';
70 end;
71 v1=gcmfaces(v1,gridType);
72
73 else;
74
75 [n1,n2,n3,n4,n5]=size(v0{1});
76
77 if strcmp(mygrid.fileFormat,'straight');
78 v1=v0{1};
79 elseif strcmp(mygrid.fileFormat,'cube');
80 v1=zeros(n2*6,n2,n3,n4,n5);
81 for ii=1:6; v1([1:n2]+(ii-1)*n2,:,:,:,:)=v0{ii}; end;
82 elseif strcmp(mygrid.fileFormat,'compact');
83
84 v0_faces=v0; clear v0;
85 for iFace=1:mygrid.nFaces; eval(['v0{iFace}=get(v0_faces,''f' num2str(iFace) ''');']); end;
86
87 nn=size(v0{1},1); mm=size(v0{1},2);
88 pp=mm/nn*4+mygrid.nFaces-4;
89
90 n3=size(v0{1},3); n4=size(v0{1},4); n5=size(v0{1},5);
91 v1=NaN*zeros(nn,nn*pp,n3,n4,n5);
92
93 v11=NaN*zeros(nn*nn*pp,n3*n4*n5);
94
95 i0=1; i1=nn*mm; tmp1=reshape(v0{1},[nn*mm n3*n4*n5]); v11(i0:i1,:)=tmp1(:,:);
96 i0=i1+1; i1=i1+nn*mm; tmp1=reshape(v0{2},[nn*mm n3*n4*n5]); v11(i0:i1,:)=tmp1(:,:);
97 i0=i1+1; i1=i1+nn*nn; tmp1=reshape(v0{3},[nn*nn n3*n4*n5]); v11(i0:i1,:)=tmp1(:,:);
98 i0=i1+1; i1=i1+nn*mm; tmp1=reshape(v0{4},[mm*nn n3*n4*n5]); v11(i0:i1,:)=tmp1(:,:);
99 i0=i1+1; i1=i1+nn*mm; tmp1=reshape(v0{5},[mm*nn n3*n4*n5]); v11(i0:i1,:)=tmp1(:,:);
100 if mygrid.nFaces==6;
101 i0=i1+1; i1=i1+nn*nn; tmp1=reshape(v0{6},[nn*nn n3*n4*n5]); v11(i0:i1,:)=tmp1(:,:);
102 end;
103
104 v1=reshape(v11,[nn nn*pp n3 n4 n5]);
105
106 end;
107
108 end;
109
110

  ViewVC Help
Powered by ViewVC 1.1.22