--- MITgcm_contrib/enderton/Diagnostics/DiagLoad.m 2005/02/01 18:49:28 1.3 +++ MITgcm_contrib/enderton/Diagnostics/DiagLoad.m 2005/09/05 18:48:27 1.13 @@ -1,7 +1,7 @@ function [data,xax,yax,time,pltslc] = ... - DiagLoad(fln,exp,dat,dad,grd,itr,tst,flu,ddf,gdf,avg,slc,pst,... - LoadGridData,DiagDebug,GridSuffix,ZcordFile,Index,Dim,... - Grads,Year0Iter,Months); + DiagLoad(fln,trl,dat,dad,grd,itr,tst,flu,ddf,gdf,avg,slc,pst,... + LoadGridData,DiagDebug,GridSuffix,ZcordFile,Index,Dim,Vector,Mate,... + Grads,Year0Iter,SecPerYear,Months,FieldName); % Load parameters. @@ -25,6 +25,11 @@ else , mnchandle = 'tave.*' ; end elseif isequal(dat,'Ins') filesuffix = ''; mnchandle = 'state.*'; +elseif isequal(dat(1:2),'Mn') + filesuffix = ''; + if ismember(dat,{'MnA','MnO'}), mnchandle = 'monitor.*'; end + if isequal(dat,'MnI'), mnchandle = 'monitor_sice.*'; end + if isequal(dat,'MnL'), mnchandle = 'monitor_land.*'; end elseif ismember(dat,{'','Non','None'}) filesuffix = ''; mnchandle = ''; else @@ -46,17 +51,23 @@ % Open grads files if ~isequal(Grads,0) if DiagDebug, disp([' Debug -- Loading GRADS field.']); end - [data,xax,yax,zax,months,time,dim] = DiagLoadGradsData(Grads,dad,fln,DiagDebug); - if DiagDebug, disp([' Debug -- ''data'' size after load: ',mat2str(size(data))]); end + [data,xax,yax,zax,months,time,dim] = ... + DiagLoadGradsData(Grads,dad,fln,DiagDebug); + if DiagDebug, disp([' Debug -- ''data'' size after load: ',... + mat2str(size(data))]); end data = DiagAverage(data,fln,avg,months,ddf,dim); - if DiagDebug, disp([' Debug -- ''data'' size after averaging: ',mat2str(size(data))]); end - if isequal(slc,'Sur') - data = data'; - pltslc='lonlat'; - else - error('Loading grads data only set to handle surface plots!'); - end - + if DiagDebug, disp([' Debug -- ''data'' size after averaging: ',... + mat2str(size(data))]); end + [data,xax,yax,pltslc] = ... + DiagSliceGradsData(fln,flu,slc,data,xax,yax,zax); + if DiagDebug, disp([' Debug -- ''data'' size after slicing: ',... + mat2str(size(data))]); end + +% Load monitor data. +elseif isequal(dat(1:2),'Mn') + [data,time] = ... + DiagLoadMonitor(fln,mnchandle,dad,itr,tst,SecPerYear,DiagDebug); + xax = time; yax = NaN; pltslc = 'timfld'; % Load AIM physics data. The data is all stuffed into one file called % 'aimPhy' which currently only comes in MDS format. When loading a field, @@ -69,7 +80,7 @@ % stresses (on atmosphere) must converted to a lat-lon grid. elseif ismember(fln,aimparameters) if isequal(ddf,'MNC') - error('Aim physics reader not set to hande ''MNC'' data format.') + error('Aim physics reader not set to handle ''MNC'' data format.') end data = DiagLoad_Local('aimPhy',dat,dad,grd,itr,ddf,filesuffix,mnchandle); if isequal(fln,'TotP') @@ -93,11 +104,11 @@ data = squeeze(data(:,:,1,:)); data = DiagAverage(data,fln,avg,months,ddf,Dim); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,flu,... - ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,flu,... + ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); % Load coupled fields. Like AIM, the data is all stuffed into one file -% called either 'cplFld.*' (MDS) or 'cpl_tave.*' (MNC). Variables within +% called either 'cplFldtave.*' (MDS) or 'cpl_tave.*' (MNC). Variables within % the MDS files given in 'DiagGenParam', accessed directly with MNC. TX % and TY (stress on ocean) must be converted to lat-lon grid. elseif ismember(fln,cplparameters) @@ -126,8 +137,8 @@ end data = DiagAverage(data,fln,avg,months,ddf,Dim); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,flu,... - ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,flu,... + ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); % Load ice parameters. Note that as the ice variable names appear to be in % constant flux, this part quickly becomes out of date. @@ -141,16 +152,16 @@ end data = DiagAverage(data,fln,avg,months,ddf,Dim); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,flu,... - ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,flu,... + ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); % Load generic fields where no modifications are needed. elseif ismember(fln,{'S','T','Temp','aim_RH','phiHyd','Conv'}) data = DiagLoad_Local(fln,dat,dad,grd,itr,ddf,filesuffix,mnchandle); data = DiagAverage(data,fln,avg,months,ddf,Dim); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,... - flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,... + flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); % Read vertical velocities, eta. Convert from [P] to [hPa] in atmosphere. elseif ismember(fln,{'W','wVel','Eta','ETA'}) @@ -158,17 +169,17 @@ if isequal(flu,'A'), data = data./100; end data = DiagAverage(data,fln,avg,months,ddf,Dim); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,... - flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,... + flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); % Load horizontal velocities, convert to lat-lon grid. -elseif ismember(fln,{'U','V','uVel','vVel'}) +elseif ismember(fln,{'U','V','uVel','vVel','fizhi_U','fizhi_V'}) if isequal(dat,'Tav') U = DiagLoad_Local('uVel',dat,dad,grd,itr,ddf,filesuffix,mnchandle); V = DiagLoad_Local('vVel',dat,dad,grd,itr,ddf,filesuffix,mnchandle); else isequal(dat,'Ins') - U = DiagLoad_Local('U',dat,dad,grd,itr,ddf,filesuffix,mnchandle); - V = DiagLoad_Local('V',dat,dad,grd,itr,ddf,filesuffix,mnchandle); + U = DiagLoad_Local([fln(1:end-1),'U'],dat,dad,grd,itr,ddf,filesuffix,mnchandle); + V = DiagLoad_Local([fln(1:end-1),'V'],dat,dad,grd,itr,ddf,filesuffix,mnchandle); end if isequal(ddf,'MNC') U = U(1:end-1,:,:,:); @@ -179,14 +190,14 @@ [XC,XG,YC,YG,Ylat,ZC,ZG,RAC,drC,drF,HFacC,HFacW,HFacS,dxG,dyG,dxC,dyC] = ... DiagLoadGridData(LoadGridData,grd,gdf,flu,GridSuffix,ZcordFile); [U,V]=uvcube2latlon(XC,YC,U,V,XL,YL); - if ismember(fln,{'U','uVel'}) + if ismember(fln,{'U','uVel','fizhi_U'}) data = U; - elseif ismember(fln,{'V','vVel'}) + elseif ismember(fln,{'V','vVel','fizhi_V'}) data = V; end [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,... - flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,... + flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); % Meridional overturning. elseif ismember(fln,{'Bol','Psi','Res'}) @@ -235,7 +246,43 @@ disp([' Calculated range: ',mat2str(datarange)]); disp([' Given range: ',mat2str(fixedrange)]); end - + + +% Actual temperature and moist potential temperature. Actual temperature +% is calculated as T=Theta*(p/po)^(R/cp). Moist potential temperature is +% calculated as Theta_e=Theta*e^(L*q*/cp*T) where q*=(R/Rv)*(es/p) and +% es=Ae^(beta*T). +elseif ismember(fln,{'ActT','ThetaEc'}) + if isequal('flu','O'), + error('Calculation may only be done for atmosphere!'); + end + [XC,XG,YC,YG,Ylat,ZC,ZG,RAC,drC,drF,... + HFacC,HFacW,HFacS,dxG,dyG,dxC,dyC] = ... + DiagLoadGridData(LoadGridData,grd,gdf,flu,GridSuffix,ZcordFile); + if isequal(ddf,'MDS') + theta = DiagLoad_Local('T',dat,dad,grd,itr,ddf,... + filesuffix,mnchandle); + elseif isequal(ddf,'MNC') + theta = DiagLoad_Local('Temp',dat,dad,grd,itr,... + ddf,filesuffix,mnchandle); + end + theta = DiagAverage(theta,fln,avg,months,ddf,Dim); + if ismember(fln,{'ActT','ThetaEc'}) + pres = NaN.*zeros(size(theta)); + for iz=1:length(ZC), pres(:,:,iz)=ZC(iz); end + temp=theta.*(pres./presrefA).^(RdA/cpA); + if isequal(fln,'ActT'), data=temp; end + end + if isequal(fln,'ThetaEc') + es=A_CC.*exp(Beta_CC.*(temp-K2C)); + qstar=(RdA/RvA).*(es./pres); + data=theta.*exp(LHvapA.*qstar./cpA./temp); + end + [data,xax,yax,pltslc] = ... + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,flu,ddf,gdf,... + avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); + + % Read in variabilities. Here we must make some simple calculations. For % all the variablilty fields there is are (field)^2 fields from which we % can calculate the desired variabilities: sqrt(field*field-(field)^2). @@ -248,58 +295,117 @@ data = sqrt(abs(ETA.*ETA-ETA2)); data = DiagAverage(data,fln,avg,months,ddf,Dim); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,... - flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,... + flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); elseif isequal(fln,'Tstd') T = DiagLoad_Local('T' ,dat,dad,grd,itr,ddf,filesuffix,mnchandle); TT = DiagLoad_Local('TT',dat,dad,grd,itr,ddf,filesuffix,mnchandle); data = sqrt(abs(T.*T-TT)); data = DiagAverage(data,fln,avg,months,ddf,Dim); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,... - flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,... + flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); elseif isequal(fln,'KEpri') U = DiagLoad_Local('uVel',dat,dad,grd,itr,ddf,filesuffix,mnchandle); V = DiagLoad_Local('vVel',dat,dad,grd,itr,ddf,filesuffix,mnchandle); UU = DiagLoad_Local('UU' ,dat,dad,grd,itr,ddf,filesuffix,mnchandle); VV = DiagLoad_Local('VV' ,dat,dad,grd,itr,ddf,filesuffix,mnchandle); if isequal(ddf,'MNC') - U = .5*( U(2:end,:,:,:) + U(1:end-1,:,:,:) ); - V = .5*( V(:,2:end,:,:) + V(:,1:end-1,:,:) ); - %UU = .5*( UU(2:end,:,:,:) + UU(1:end-1,:,:,:) ); % UU and VV might be incorrect. - %VV = .5*( VV(:,2:end,:,:) + VV(:,1:end-1,:,:) ); + U = U(1:end-1,:,:,:); UU = UU(1:end-1,:,:,:); + V = V(:,1:end-1,:,:); VV = VV(:,1:end-1,:,:); end - U = DiagAverage(U,fln,avg,months,ddf,Dim); - V = DiagAverage(V,fln,avg,months,ddf,Dim); + U = DiagAverage(U ,fln,avg,months,ddf,Dim); + V = DiagAverage(V ,fln,avg,months,ddf,Dim); UU = DiagAverage(UU,fln,avg,months,ddf,Dim); VV = DiagAverage(VV,fln,avg,months,ddf,Dim); data = sqrt(abs((U.*U + V.*V) - (UU + VV))); [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,... - flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,... + flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); % If the field is not recognized, try best to try and open it. else - if DiagDebug, disp([' Debug -- Unknown field, attempting to load.']); end + disp([' Unknown field, attempting to load.']); data = DiagLoad_Local(fln,dat,dad,grd,itr,ddf,filesuffix,mnchandle); if DiagDebug, disp([' Debug -- ''data'' size after load: ',mat2str(size(data))]); end - if ~isequal(Index,0) + if isequal(Vector,1) | isequal(Vector,2) | isequal(Vector,3) + if isequal(Vector,1) | isequal(Vector,3) + if ~isequal(Index,0) + if isequal(Dim,2), U = squeeze(data(:,:,Index,:)); + elseif isequal(Dim,3), U = squeeze(data(:,:,:,Index,:)); end + end + if ~isequal(Mate,0) + if isequal(Dim,2), V = squeeze(data(:,:,Mate,:)); + elseif isequal(Dim,3), V = squeeze(data(:,:,:,Mate,:)); end + end + elseif isequal(Vector,2) + if ~isequal(Index,0) + if isequal(Dim,2), V = squeeze(data(:,:,Index,:)); + elseif isequal(Dim,3), V = squeeze(data(:,:,:,Index,:)); end + end + if ~isequal(Mate,0) + if isequal(Dim,2), U = squeeze(data(:,:,Mate,:)); + elseif isequal(Dim,3), U = squeeze(data(:,:,:,Mate,:)); end + end + end + U = DiagAverage(U,fln,avg,months,ddf,Dim); + V = DiagAverage(V,fln,avg,months,ddf,Dim); + [XC,XG,YC,YG,Ylat,ZC,ZG,RAC,drC,drF,HFacC,HFacW,HFacS,dxG,dyG,dxC,dyC] = ... + DiagLoadGridData(LoadGridData,grd,gdf,flu,GridSuffix,ZcordFile); + if isequal(Vector,1) | isequal(Vector,2) + [U,V]=uvcube2latlon(XC,YC,U,V,XL,YL); + end + if isequal(Vector,1) + data = U; + elseif isequal(Vector,2) + data = V; + elseif isequal(Vector,3) + if isequal(flu,'A'), delM = - 1000*drF ./ g; + elseif isequal(flu,'O'), delM = drF; end + [psiavg,mskG,ylat]=calc_PsiCube(delM,U.*HFacW,V.*HFacS,dxG,dyG); + data = psiavg(2:end-1,:)'; + if isequal(flu,'A'), data = data./1e6; end + xax = ylat; yax = ZG; pltslc = 'lathgt'; + end + if DiagDebug, disp([' Debug -- ''data'' size after vector manipulation: ',mat2str(size(data))]); end + else +% Sequence here for Vector = 0 + if ~isequal(Index,0) & isequal(size(Index),[1 1]) & isequal(sign(Index),1) if isequal(Dim,2), data = squeeze(data(:,:,Index,:)); elseif isequal(Dim,3), data = squeeze(data(:,:,:,Index,:)); end + elseif isequal(size(Index),[1 1]) & ~isequal(sign(Index),1) + if isequal(Dim,2), data = sign(Index(1)).*squeeze(data(:,:,abs(Index(1)),:)); + elseif isequal(Dim,3), data = sign(Index(1)).*squeeze(data(:,:,:,abs(Index(1)),:)); end + elseif isequal(size(Index),[1 2]) + if isequal(Dim,2), data = sign(Index(1)).*squeeze(data(:,:,abs(Index(1)),:)) + sign(Index(2)).*squeeze(data(:,:,abs(Index(2)),:)); + elseif isequal(Dim,3), data = sign(Index(1)).*squeeze(data(:,:,:,abs(Index(1)),:)) + sign(Index(2)).*squeeze(data(:,:,:,abs(Index(2)),:)); end + end + if DiagDebug, disp([' Debug -- ''data'' size after indexing: ',mat2str(size(data))]); end + data = DiagAverage(data,fln,avg,months,ddf,Dim); +% Actual Temperature -- When the FieldName says ActT we assume that +% the field that is pointed to in Index is potential temperature + if isequal(FieldName,'ActT') + [XC,XG,YC,YG,Ylat,ZC,ZG,RAC,drC,drF,HFacC,HFacW,HFacS,dxG,dyG,dxC,dyC] = ... + DiagLoadGridData(LoadGridData,grd,gdf,flu,GridSuffix,ZcordFile); + pres = NaN.*zeros(size(data)); + for iz=1:length(ZC), pres(:,:,iz)=ZC(iz); end + temp=data.*(pres./presrefA).^(RdA/cpA); + data=temp; + end + + if DiagDebug, disp([' Debug -- ''data'' size after average: ',mat2str(size(data))]); end end - if DiagDebug, disp([' Debug -- ''data'' size after indexing: ',mat2str(size(data))]); end - data = DiagAverage(data,fln,avg,months,ddf,Dim); - if DiagDebug, disp([' Debug -- ''data'' size after average: ',mat2str(size(data))]); end + if ~isequal(Vector,3) [data,xax,yax,pltslc] = ... - DiagSlice(data,fln,exp,dat,dad,grd,itr,tst,... - flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile); + DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,... + flu,ddf,gdf,avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName); if DiagDebug, disp([' Debug -- ''data'' size after slice: ',mat2str(size(data))]); end if DiagDebug, disp([' Debug -- ''xax'' size: ',mat2str(size(xax))]); end if DiagDebug, disp([' Debug -- ''yax'' size: ',mat2str(size(yax))]); end if DiagDebug, disp([' Debug -- ''pltslc'': ',pltslc]); end + end end - %-------------------------------------------------------------------------% % Local functions % %-------------------------------------------------------------------------% @@ -307,17 +413,11 @@ % Thus is merely a local function that calls the load data functions % according to the DataFormat for the data specified by dfm. function data = DiagLoad_Local(fln,dat,dad,grd,itr,dfm,filesuffix,mnchandle) - if isequal(dfm,'MDS') data = rdmds([dad,'/',fln,filesuffix],itr); elseif isequal(dfm,'MNC') - %[dad,mnchandle] - iter = rdmnc_mod([dad,mnchandle],'T'); - iter = iter.T; - [dummy,indecies] = ismember(itr,iter); - data = rdmnc_mod([dad,mnchandle],[fln,filesuffix],'T',indecies); - iter = data.T; - if ~isequal(itr,iter'), error('Missing iterations in data!'); end + data = rdmnc_mod2([dad,mnchandle],[fln,filesuffix],'iter','T',itr); + %if ~isequal(itr,data.iter'), error('Missing iterations in data!'); end eval(['data = data.',fln,filesuffix,';']); else error(['Unrecognized data type: ',dfm]);