1 |
function []=process2nctiles(dirModel,fileModel,fldModel,tileSize); |
2 |
%process2nctiles(dirModel); |
3 |
%object : convert MITgcm binary output to netcdf files (tiled) |
4 |
%inputs : dirModel is the MITgcm run directory |
5 |
% It is expected to contain binaries in |
6 |
% 'diags/STATE/', 'diags/TRSP/', etc. as well |
7 |
% as the 'available_diagnostics.log' text file. |
8 |
% fileModel the file name base e.g. 'state_2d_set1' |
9 |
% By default : all variables in e.g. 'state_2d_set1*' |
10 |
% files will be processed, and writen individually to |
11 |
% nctiles (tiled netcdf) that will be located in 'nctiles/' |
12 |
% fldModel (by default []) can be specified (as e.g. 'ETAN') |
13 |
% when fldModel is empty, all fields are processed |
14 |
% tileSize (optional) is e.g. [90 90] (by default tiles=faces) |
15 |
%output : (netcdf files) |
16 |
|
17 |
gcmfaces_global; |
18 |
|
19 |
%listFiles={'state_2d_set1','state_2d_set2','state_3d_set1','state_3d_set2'}; |
20 |
%listFiles={'trsp_3d_set1','trsp_3d_set2','trsp_3d_set3'}; |
21 |
%for ff=1:length(listFiles); process2nctiles('iter12/',listFiles{ff},[],[90 90]); end; |
22 |
|
23 |
%directory names |
24 |
listDirs={'STATE/','TRSP/'}; |
25 |
filAvailDiag=[dirModel 'available_diagnostics.log']; |
26 |
filReadme=[dirModel 'README']; |
27 |
dirOut=[dirModel 'nctiles/']; |
28 |
if ~isdir(dirOut); mkdir(dirOut); end; |
29 |
|
30 |
%seacrh in subdirectories |
31 |
subDir=[]; |
32 |
for ff=1:length(listDirs); |
33 |
tmp1=dir([dirModel 'diags/' listDirs{ff} fileModel '*']); |
34 |
if ~isempty(tmp1); subDir=listDirs{ff}; end; |
35 |
end; |
36 |
|
37 |
if isempty(subDir); |
38 |
error(['file ' fileModel ' was not found']); |
39 |
else; |
40 |
dirIn=[dirModel 'diags/' subDir]; |
41 |
nn=length(dir([dirIn fileModel '*data'])); |
42 |
fprintf('%s (%d files) was found in \n %s \n',fileModel,nn,dirIn); |
43 |
end; |
44 |
|
45 |
%set list of variables to process |
46 |
if ~isempty(fldModel); |
47 |
if ischar(fldModel); listFlds={fldModel}; |
48 |
else; listFlds=fldModel; |
49 |
end; |
50 |
else; |
51 |
meta=read_meta([dirIn fileModel '*']); |
52 |
listFlds=meta.fldList; |
53 |
end; |
54 |
|
55 |
%determine map of tile indices (by default tiles=faces) |
56 |
if isempty(whos('tileSize')); |
57 |
tileNo=mygrid.XC; |
58 |
for ff=1:mygrid.nFaces; tileNo{ff}(:)=ff; end; |
59 |
else; |
60 |
tileNo=gcmfaces_loc_tile(tileSize(1),tileSize(2)); |
61 |
end; |
62 |
|
63 |
%now do the actual processing |
64 |
for vv=1:length(listFlds); |
65 |
nameDiag=deblank(listFlds{vv}) |
66 |
|
67 |
%get meta information |
68 |
meta=read_meta([dirIn fileModel '*']); |
69 |
irec=find(strcmp(deblank(meta.fldList),nameDiag)); |
70 |
if length(irec)~=1; error('field not in file\n'); end; |
71 |
|
72 |
%read time series |
73 |
myDiag=rdmds2gcmfaces([dirIn fileModel '*'],NaN,'rec',irec); |
74 |
|
75 |
%set ancilliary time variable |
76 |
nn=length(size(myDiag{1})); |
77 |
nn=size(myDiag{1},nn); |
78 |
%tim=[1:nn]; |
79 |
tim=[1992*ones(nn,1) [1:nn]' 15*ones(nn,1)]; |
80 |
tim=datenum(tim)-datenum([1992 1 0]); |
81 |
timUnits='days since 1992-1-1 0:0:0'; |
82 |
|
83 |
%get units and long name from available_diagnostics.log |
84 |
[avail_diag]=read_avail_diag(filAvailDiag,nameDiag); |
85 |
|
86 |
%get description of estimate from README |
87 |
[rdm]=read_readme(filReadme); |
88 |
disp(rdm'); |
89 |
|
90 |
%convert to MITgcm format (90x1170 array) |
91 |
myFile=[dirOut nameDiag];%first instance is for subdirectory name |
92 |
if ~isdir(myFile); mkdir(myFile); end; |
93 |
myFile=[myFile '/' nameDiag];%second instance is for file name base |
94 |
|
95 |
%get grid params |
96 |
[grid_diag]=set_grid_diag(avail_diag); |
97 |
|
98 |
%apply mask, and convert to land mask |
99 |
msk=grid_diag.msk; |
100 |
msk=repmat(msk,[1 1 size(myDiag{1},3) size(myDiag{1},4)]); |
101 |
myDiag=myDiag.*msk; |
102 |
clear msk; |
103 |
% |
104 |
land=isnan(grid_diag.msk); |
105 |
|
106 |
%set 'coord' attribute |
107 |
if avail_diag.nr~=1; |
108 |
coord='lon lat dep tim'; |
109 |
else; |
110 |
coord='lon lat tim'; |
111 |
end; |
112 |
|
113 |
%create netcdf file using write2nctiles (works only with old matlab, thus far ...) |
114 |
doCreate=1; |
115 |
dimlist=write2nctiles(myFile,myDiag,doCreate,{'tileNo',tileNo},... |
116 |
{'fldName',nameDiag},{'longName',avail_diag.longNameDiag},... |
117 |
{'units',avail_diag.units},{'descr',nameDiag},{'coord',coord},{'rdm',rdm}); |
118 |
|
119 |
%determine relevant dimensions |
120 |
for ff=1:length(dimlist); |
121 |
dim.tim{ff}={dimlist{ff}{1}}; |
122 |
dim.twoD{ff}={dimlist{ff}{end-1:end}}; |
123 |
if avail_diag.nr~=1; |
124 |
dim.threeD{ff}={dimlist{ff}{end-2:end}}; |
125 |
dim.dep{ff}={dimlist{ff}{end-2}}; |
126 |
else; |
127 |
dim.threeD{ff}=dim.twoD{ff}; |
128 |
dim.dep{ff}=[]; |
129 |
end; |
130 |
end; |
131 |
|
132 |
%prepare to add fields |
133 |
doCreate=0; |
134 |
|
135 |
%now add fields |
136 |
write2nctiles(myFile,tim,doCreate,{'tileNo',tileNo},... |
137 |
{'fldName','tim'},{'longName','time'},... |
138 |
{'units',timUnits},{'dimIn',dim.tim}); |
139 |
write2nctiles(myFile,grid_diag.lon,doCreate,{'tileNo',tileNo},... |
140 |
{'fldName','lon'},{'units','degrees_east'},{'dimIn',dim.twoD}); |
141 |
write2nctiles(myFile,grid_diag.lat,doCreate,{'tileNo',tileNo},... |
142 |
{'fldName','lat'},{'units','degrees_north'},{'dimIn',dim.twoD}); |
143 |
write2nctiles(myFile,grid_diag.msk,doCreate,{'tileNo',tileNo},... |
144 |
{'fldName','land'},{'units','1'},{'longName','land mask'},{'dimIn',dim.threeD}); |
145 |
write2nctiles(myFile,grid_diag.RAC,doCreate,{'tileNo',tileNo},... |
146 |
{'fldName','area'},{'units','m^2'},{'longName','grid cell area'},{'dimIn',dim.twoD}); |
147 |
if isfield(grid_diag,'dep'); |
148 |
write2nctiles(myFile,grid_diag.dep,doCreate,{'tileNo',tileNo},... |
149 |
{'fldName','dep'},{'units','m'},{'dimIn',dim.dep}); |
150 |
write2nctiles(myFile,grid_diag.dz,doCreate,{'tileNo',tileNo},... |
151 |
{'fldName','thic'},{'units','m'},{'dimIn',dim.dep}); |
152 |
end; |
153 |
|
154 |
end;%for vv=1:length(listFlds); |
155 |
|
156 |
function [meta]=read_meta(fileName); |
157 |
|
158 |
%read meta file |
159 |
tmp1=dir([fileName '*.meta']); tmp1=tmp1(1).name; |
160 |
tmp2=strfind(fileName,'/'); |
161 |
if ~isempty(tmp2); tmp2=tmp2(end); else; tmp2=0; end; |
162 |
tmp1=[fileName(1:tmp2) tmp1]; fid=fopen(tmp1); |
163 |
while 1; |
164 |
tline = fgetl(fid); |
165 |
if ~ischar(tline), break, end |
166 |
if isempty(whos('tmp3')); tmp3=tline; else; tmp3=[tmp3 ' ' tline]; end; |
167 |
end |
168 |
fclose(fid); |
169 |
|
170 |
%add meta variables to workspace |
171 |
eval(tmp3); |
172 |
|
173 |
%reformat to meta structure |
174 |
meta.dataprec=dataprec; |
175 |
meta.nDims=nDims; |
176 |
meta.nFlds=nFlds; |
177 |
meta.nrecords=nrecords; |
178 |
meta.fldList=fldList; |
179 |
meta.dimList=dimList; |
180 |
if ~isempty(who('timeInterval')); meta.timeInterval=timeInterval; end; |
181 |
if ~isempty(who('timeStepNumber')); meta.timeStepNumber=timeStepNumber; end; |
182 |
|
183 |
%% |
184 |
|
185 |
function [rdm]=read_readme(filReadme); |
186 |
|
187 |
gcmfaces_global; |
188 |
|
189 |
rdm=[]; |
190 |
|
191 |
fid=fopen(filReadme,'rt'); |
192 |
while ~feof(fid); |
193 |
nn=length(rdm); |
194 |
rdm{nn+1} = fgetl(fid); |
195 |
end; |
196 |
fclose(fid); |
197 |
|
198 |
%% |
199 |
|
200 |
function [avail_diag]=read_avail_diag(filAvailDiag,nameDiag); |
201 |
|
202 |
gcmfaces_global; |
203 |
|
204 |
avail_diag=[]; |
205 |
|
206 |
fid=fopen(filAvailDiag,'rt'); |
207 |
while ~feof(fid); |
208 |
tline = fgetl(fid); |
209 |
tmp1=8-length(nameDiag); tmp1=repmat(' ',[1 tmp1]); |
210 |
tname = ['|' sprintf('%s',nameDiag) tmp1 '|']; |
211 |
if ~isempty(strfind(tline,tname)); |
212 |
%e.g. tline=' 235 |SIatmQnt| 1 | |SM U1|W/m^2 |Net atmospheric heat flux, >0 decreases theta'; |
213 |
% |
214 |
tmp1=strfind(tline,'|'); tmp1=tmp1(end-1:end); |
215 |
avail_diag.units=strtrim(tline(tmp1(1)+1:tmp1(2)-1)); |
216 |
avail_diag.longNameDiag=tline(tmp1(2)+1:end); |
217 |
% |
218 |
tmp1=strfind(tline,'|'); tmp1=tmp1(4:5); |
219 |
pars=tline(tmp1(1)+1:tmp1(2)-1); |
220 |
% |
221 |
if strcmp(pars(2),'M'); avail_diag.loc_h='C'; |
222 |
elseif strcmp(pars(2),'U'); avail_diag.loc_h='W'; |
223 |
elseif strcmp(pars(2),'V'); avail_diag.loc_h='S'; |
224 |
end; |
225 |
% |
226 |
avail_diag.loc_z=pars(9); |
227 |
% |
228 |
if strcmp(pars(10),'1'); avail_diag.nr=1; |
229 |
else; avail_diag.nr=length(mygrid.RC); |
230 |
end; |
231 |
end; |
232 |
end; |
233 |
fclose(fid); |
234 |
|
235 |
%% |
236 |
|
237 |
function [grid_diag]=set_grid_diag(avail_diag); |
238 |
|
239 |
gcmfaces_global; |
240 |
|
241 |
%switch for non-tracer point values |
242 |
if strcmp(avail_diag.loc_h,'C'); |
243 |
grid_diag.lon=mygrid.XC; |
244 |
grid_diag.lat=mygrid.YC; |
245 |
grid_diag.msk=mygrid.mskC(:,:,1:avail_diag.nr); |
246 |
elseif strcmp(avail_diag.loc_h,'W'); |
247 |
grid_diag.lon=mygrid.XG; |
248 |
grid_diag.lat=mygrid.YC; |
249 |
grid_diag.msk=mygrid.mskW(:,:,1:avail_diag.nr); |
250 |
elseif strcmp(avail_diag.loc_h,'S'); |
251 |
grid_diag.lon=mygrid.XC; |
252 |
grid_diag.lat=mygrid.YG; |
253 |
grid_diag.msk=mygrid.mskS(:,:,1:avail_diag.nr); |
254 |
end; |
255 |
grid_diag.RAC=mygrid.RAC; |
256 |
|
257 |
%vertical grid |
258 |
if avail_diag.nr~=1; |
259 |
if strcmp(avail_diag.loc_z,'M'); |
260 |
grid_diag.dep=-mygrid.RC; |
261 |
grid_diag.dz=mygrid.DRF; |
262 |
elseif strcmp(avail_diag.loc_z,'L'); |
263 |
grid_diag.dep=-mygrid.RF(2:end); |
264 |
grid_diag.dz=[mygrid.DRC(2:end) ; 228.25];%quick fix |
265 |
else; |
266 |
error('unknown vertical grid'); |
267 |
end; |
268 |
grid_diag.dep=reshape(grid_diag.dep,[1 1 avail_diag.nr]); |
269 |
grid_diag.dz=reshape(grid_diag.dz,[1 1 avail_diag.nr]); |
270 |
end; |