enderton/Diagnostics/DiagSlice.m

function [data,xax,yax,pltslc] = ...
    DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,flu,ddf,gdf,...
              avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName);

% Function: DiagSlice
% Author:   Daniel Enderton
%
% Input Fields:
%
%   Field        Type        (Brief) Description
%   -----------------------------------------------------------------------
%   data         array       Field name
%   fln          string      Field name
%   dad          string      Data directory.
%   grd          string      Grid data directory.
%   avg          string      Averaging scheme ('Ann','DJF', 'JJA',...)
%   slc          string      Slice type ('Zon','k=#',...)
%   pst          string      Plot style ('Con','Sur',...)
%   flu          string      Fluid ('A','O')
%   dfm          string      Data format ('MDS' or 'MNC')
%   LoadGridData 0/1         Optionally load grid data
%
% Output Fields:
%
%   Field       Type        (Brief) Description
%   -----------------------------------------------------------------------
%   data        array       Sliced data.
%
% Descripton:
%   This function slices the data according to SliceType 'slc'.  Depending
%   on the SliceType and PlotStyle, the data may be converted to a lat-lon
%   grid.  At the end there is also a range check.  If the value is out of
%   the user specified range given in 'DaigFieldParamA'.

% Load diagnostics parameters, grid data, mark data size
DiagGenParam;
DiagFieldParamA;
DiagFieldParamO;
DiagFieldParamC;
DiagFieldParamI;
[XC,XG,YC,YG,Ylat,ZC,ZG,RAC,drC,drF,HFacC,HFacW,HFacS,dxG,dyG,dxC,dyC] = ...
    DiagLoadGridData(LoadGridData,grd,gdf,flu,GridSuffix,ZcordFile);
datasize = size(data);


%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%                               Surface plot                              %
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

% All surface plots are direct cube sphere outputs, and therefore must have
% the dimensions of the cube grid data.  If the 'pst' is 'Grd' or 'Int',
% everything is already all set since those handle the cubed sphere data.
% If the 'pst' is 'Con' or 'Cnf', convert data to a lat-lon grid so that it
% can easily be contoured.
if isequal(slc,'Sur')
    if ismember(fln,{'TX','TY','USTR','VSTR'}) | ~isequal(Vector,0)
        data = data'; xax = XL; yax = YL;
    elseif ~isequal(datasize,size(XC)) 
        error('Incorrect dimensions for slc:  ',slc);
    else
        if ismember(pst,{'Grd','Int'})
            if     isequal(pst,'Grd'),
                xax = XG;
                yax = YG;
            elseif isequal(pst,'Int'),
                xax = XC;
                yax = YC;
            end
        elseif ismember(pst,{'Con','Cnf'})
            data = cube2latlon(XC,YC,data,XL,YL)';
            xax = XL; yax = YL;
        else
            error(['slc = ''Sur'' can not use pst:  ',pst]);
        end
    end
    pltslc='lonlat';
    
    
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%                            Zonal average plot                           %
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

% When computing a zonal average, there are many different options for the
% initial data.  If the data is cube-sphere, it could be the shape of XC,
% possibly also with a vertical axis.  In either case, 'calc_ZonAv_CS' (a
% nifty zonally averaging script from JMC) is called to compute the zonal
% average for raw cube sphere data.  For horizontal velocities, the data
% could be of the size [length(XL),length(YL),z-axis length], in which case
% you just need to take the mean over the longitude axis (always 1).
elseif isequal(slc,'Zon')
    if isequal(datasize(1:2),size(XC))
        if isequal(flu,'O'), nBas = 0; end
        if isequal(flu,'A'), nBas = 0; end
            [data,dump1,dump2] = ...
                calc_ZonAv_CS(data,kpr,kwr,nBas,XC,YC,XG,YG,RAC,dad,HFacC);
        if isequal(avg,'Tse')
            data=data(:,:,1);  xax=years;  yax=Ylat; pltslc='timlat';
        elseif isequal(avg,'Tyr')
            data=data(:,:,1);  xax=[0:12]; yax=Ylat; pltslc='timlat';
        elseif isequal(pst,'Lin')
            data=data(:,:,1)'; xax=Ylat;   yax=NaN;  pltslc='latfld';
        else
            data=data(:,:,1)'; xax=Ylat;   yax=ZC;   pltslc='lathgt';
        end
    elseif isequal(datasize(1:2),[length(XL),length(YL)])
        if ~isequal(pst,'Lin')
            data = squeeze(mean(data,1))'; xax=YL; yax=ZC;  pltslc='lathgt';
        else
            data = squeeze(mean(data,1))'; xax=YL; yax=NaN; pltslc='latfld';
        end
    else
        error('Incorrect dimensions for slc = ''Zon''');
    end

    
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%                                   i,j,k=#                               %
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

% These slicing settings just take a slice on one of the axes.  If the
% slice option is 'i=' or 'j=' and the data is cube-sphere, it is converted
% to lat-lon first and indexed then (thus the index number should be based
% off of the XL and YL interpolated grids defined in 'DiagGenParam'.  It
% would be nice to add functions that can slice by longitude, latitude, and
% height values directly.  Note that there is transposing of all the data
% to prepare it for contour plots (harmless is line plots).
elseif isequal(slc(1:2),'i=')
    ii = str2num(slc(3:end));
    data = cube2latlon(XC,YC,data,XL,YL);
    if ismember(fln,fields2D)
        data=squeeze(data(ii,:));
        xax=YL; yax=NaN; pltslc='latfld';
    elseif ismember(fln,fields3D)
        data=squeeze(data(ii,:,:))';
        xax=YL; yax=ZC;  pltslc='lathgt';
    end
    
elseif isequal(slc(1:2),'j=')
    jj = str2num(slc(3:end));
    data = cube2latlon(XC,YC,data,XL,YL);
    if ismember(fln,fields2D)
        data = squeeze(data(:,jj));
        xax=XL; yax=NaN; pltslc='lonfld';
    elseif ismember(fln,fields3D)
        data = squeeze(data(:,jj,:))';
        xax=XL; yax=ZC;  pltslc='lonhgt';
    end
    
elseif isequal(slc(1:2),'k=')
    kk = str2num(slc(3:end));
    data = squeeze(data(:,:,kk));
    if ismember(fln,{'U','V','uVel','vVel','fizhi_U','fizhi_V'})
        data = data'; xax = XL; yax = YL;
    elseif ismember(pst,{'Grd','Int'})
        if isequal(pst,'Grd'),
            xax = XG;
            yax = YG;
        elseif isequal(pst,'Int'),
            xax = XC;
            yax = YC;
        end
    elseif ismember(pst,{'Con','Cnf'})
        data = cube2latlon(XC,YC,data,XL,YL)'; xax = XL; yax = YL;
    end
    pltslc='lonlat';
    
else
    error(['Unrecognized SliceType:  ',slc]);
end


%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%                               Check range                               %
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

% Load fixed and data ranges, throw a warning is data out of range.
datarange = [min(data(:)),max(data(:))];
try
    eval(['fixedrange = ',fln,'range',flu,';']);
    if datarange(1) < fixedrange(1) | ...
       datarange(2) > fixedrange(2)
        disp(['***Warning***  Value out of range for ',fln]);
        disp(['               Data range:  ',mat2str(datarange)]);
        disp(['               Fixed range:  ',mat2str(fixedrange)]);
    end
catch
    disp(['***Warning***  No range information found for ',fln]);
    disp(['               Data range:  ',mat2str(datarange)]);
end
1	enderton	1.1	function [data,xax,yax,pltslc] = ...
2	enderton	1.2	DiagSlice(data,fln,trl,dat,dad,grd,itr,tst,flu,ddf,gdf,...
3	molod	1.5	avg,slc,pst,LoadGridData,GridSuffix,ZcordFile,Vector,FieldName);
4	enderton	1.1
5			% Function: DiagSlice
6			% Author: Daniel Enderton
7			%
8			% Input Fields:
9			%
10			% Field Type (Brief) Description
11			% -----------------------------------------------------------------------
12			% data array Field name
13			% fln string Field name
14			% dad string Data directory.
15			% grd string Grid data directory.
16			% avg string Averaging scheme ('Ann','DJF', 'JJA',...)
17			% slc string Slice type ('Zon','k=#',...)
18			% pst string Plot style ('Con','Sur',...)
19			% flu string Fluid ('A','O')
20			% dfm string Data format ('MDS' or 'MNC')
21			% LoadGridData 0/1 Optionally load grid data
22			%
23			% Output Fields:
24			%
25			% Field Type (Brief) Description
26			% -----------------------------------------------------------------------
27			% data array Sliced data.
28			%
29			% Descripton:
30			% This function slices the data according to SliceType 'slc'. Depending
31			% on the SliceType and PlotStyle, the data may be converted to a lat-lon
32			% grid. At the end there is also a range check. If the value is out of
33			% the user specified range given in 'DaigFieldParamA'.
34
35			% Load diagnostics parameters, grid data, mark data size
36			DiagGenParam;
37			DiagFieldParamA;
38			DiagFieldParamO;
39			DiagFieldParamC;
40			DiagFieldParamI;
41			[XC,XG,YC,YG,Ylat,ZC,ZG,RAC,drC,drF,HFacC,HFacW,HFacS,dxG,dyG,dxC,dyC] = ...
42			DiagLoadGridData(LoadGridData,grd,gdf,flu,GridSuffix,ZcordFile);
43			datasize = size(data);
44
45
46			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
47			% Surface plot %
48			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
49
50			% All surface plots are direct cube sphere outputs, and therefore must have
51	enderton	1.6	% the dimensions of the cube grid data. If the 'pst' is 'Grd' or 'Int',
52			% everything is already all set since those handle the cubed sphere data.
53			% If the 'pst' is 'Con' or 'Cnf', convert data to a lat-lon grid so that it
54			% can easily be contoured.
55	enderton	1.1	if isequal(slc,'Sur')
56	molod	1.5	if ismember(fln,{'TX','TY','USTR','VSTR'}) \| ~isequal(Vector,0)
57	enderton	1.1	data = data'; xax = XL; yax = YL;
58	enderton	1.6	elseif ~isequal(datasize,size(XC))
59	enderton	1.1	error('Incorrect dimensions for slc: ',slc);
60			else
61			if ismember(pst,{'Grd','Int'})
62			if isequal(pst,'Grd'),
63	enderton	1.6	xax = XG;
64			yax = YG;
65	enderton	1.1	elseif isequal(pst,'Int'),
66	enderton	1.6	xax = XC;
67			yax = YC;
68	enderton	1.1	end
69			elseif ismember(pst,{'Con','Cnf'})
70			data = cube2latlon(XC,YC,data,XL,YL)';
71			xax = XL; yax = YL;
72			else
73			error(['slc = ''Sur'' can not use pst: ',pst]);
74			end
75			end
76			pltslc='lonlat';
77
78
79			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
80			% Zonal average plot %
81			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
82
83			% When computing a zonal average, there are many different options for the
84	enderton	1.6	% initial data. If the data is cube-sphere, it could be the shape of XC,
85			% possibly also with a vertical axis. In either case, 'calc_ZonAv_CS' (a
86	enderton	1.1	% nifty zonally averaging script from JMC) is called to compute the zonal
87			% average for raw cube sphere data. For horizontal velocities, the data
88			% could be of the size [length(XL),length(YL),z-axis length], in which case
89			% you just need to take the mean over the longitude axis (always 1).
90			elseif isequal(slc,'Zon')
91	enderton	1.6	if isequal(datasize(1:2),size(XC))
92	enderton	1.1	if isequal(flu,'O'), nBas = 0; end
93	molod	1.5	if isequal(flu,'A'), nBas = 0; end
94	enderton	1.1	[data,dump1,dump2] = ...
95			calc_ZonAv_CS(data,kpr,kwr,nBas,XC,YC,XG,YG,RAC,dad,HFacC);
96			if isequal(avg,'Tse')
97			data=data(:,:,1); xax=years; yax=Ylat; pltslc='timlat';
98			elseif isequal(avg,'Tyr')
99			data=data(:,:,1); xax=[0:12]; yax=Ylat; pltslc='timlat';
100			elseif isequal(pst,'Lin')
101			data=data(:,:,1)'; xax=Ylat; yax=NaN; pltslc='latfld';
102			else
103			data=data(:,:,1)'; xax=Ylat; yax=ZC; pltslc='lathgt';
104			end
105			elseif isequal(datasize(1:2),[length(XL),length(YL)])
106	molod	1.4	if ~isequal(pst,'Lin')
107	enderton	1.1	data = squeeze(mean(data,1))'; xax=YL; yax=ZC; pltslc='lathgt';
108	molod	1.4	else
109	enderton	1.1	data = squeeze(mean(data,1))'; xax=YL; yax=NaN; pltslc='latfld';
110			end
111			else
112			error('Incorrect dimensions for slc = ''Zon''');
113			end
114
115
116			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
117			% i,j,k=# %
118			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
119
120			% These slicing settings just take a slice on one of the axes. If the
121			% slice option is 'i=' or 'j=' and the data is cube-sphere, it is converted
122			% to lat-lon first and indexed then (thus the index number should be based
123			% off of the XL and YL interpolated grids defined in 'DiagGenParam'. It
124			% would be nice to add functions that can slice by longitude, latitude, and
125			% height values directly. Note that there is transposing of all the data
126			% to prepare it for contour plots (harmless is line plots).
127			elseif isequal(slc(1:2),'i=')
128			ii = str2num(slc(3:end));
129			data = cube2latlon(XC,YC,data,XL,YL);
130			if ismember(fln,fields2D)
131			data=squeeze(data(ii,:));
132			xax=YL; yax=NaN; pltslc='latfld';
133			elseif ismember(fln,fields3D)
134			data=squeeze(data(ii,:,:))';
135			xax=YL; yax=ZC; pltslc='lathgt';
136			end
137
138			elseif isequal(slc(1:2),'j=')
139			jj = str2num(slc(3:end));
140			data = cube2latlon(XC,YC,data,XL,YL);
141			if ismember(fln,fields2D)
142			data = squeeze(data(:,jj));
143			xax=XL; yax=NaN; pltslc='lonfld';
144			elseif ismember(fln,fields3D)
145			data = squeeze(data(:,jj,:))';
146			xax=XL; yax=ZC; pltslc='lonhgt';
147			end
148
149			elseif isequal(slc(1:2),'k=')
150			kk = str2num(slc(3:end));
151			data = squeeze(data(:,:,kk));
152	molod	1.3	if ismember(fln,{'U','V','uVel','vVel','fizhi_U','fizhi_V'})
153	enderton	1.1	data = data'; xax = XL; yax = YL;
154			elseif ismember(pst,{'Grd','Int'})
155			if isequal(pst,'Grd'),
156	enderton	1.6	xax = XG;
157			yax = YG;
158	enderton	1.1	elseif isequal(pst,'Int'),
159	enderton	1.6	xax = XC;
160			yax = YC;
161	enderton	1.1	end
162			elseif ismember(pst,{'Con','Cnf'})
163			data = cube2latlon(XC,YC,data,XL,YL)'; xax = XL; yax = YL;
164			end
165			pltslc='lonlat';
166
167			else
168			error(['Unrecognized SliceType: ',slc]);
169			end
170
171
172			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
173			% Check range %
174			%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
175
176			% Load fixed and data ranges, throw a warning is data out of range.
177			datarange = [min(data(:)),max(data(:))];
178			try
179			eval(['fixedrange = ',fln,'range',flu,';']);
180			if datarange(1) < fixedrange(1) \| ...
181			datarange(2) > fixedrange(2)
182			disp(['*Warning* Value out of range for ',fln]);
183			disp([' Data range: ',mat2str(datarange)]);
184			disp([' Fixed range: ',mat2str(fixedrange)]);
185			end
186			catch
187			disp(['*Warning* No range information found for ',fln]);
188			disp([' Data range: ',mat2str(datarange)]);
189	molod	1.3	end