matlab_class/gcmfaces_calc/calc_overturn.m

function [FLD]=calc_overturn(fldU,fldV,doFlip,list_factors);
%object:    compute meridional overturning streamfunction
%inputs:    fldU and fldV are the fields of grid point transport
%optional:  doFlip (default is 1). If 1 then flip the vertical 
%               axis back and forth, hence intergrating from the 
%               'bottom'. If 0 then dont.
%           list_factors (default is {'dh','dz'})
%output:    FLD is the streamfunction
%
%notes:     mygrid.LATS_MASKS is the set of quasi longitudinal lines along which
%               transports will integrated, as computed in gcmfaces_lines_zonal
%           the result is converted to Sv, and sign is changed.

global mygrid;

if nargin<3; doFlip=1; end;
if nargin<4; list_factors={'dh','dz'}; end;

%initialize output:
n3=max(size(fldU.f1,3),1); n4=max(size(fldV.f1,4),1);
FLD=NaN*squeeze(zeros(length(mygrid.LATS_MASKS),n3+1,n4));

%prepare fldU/fldV:
fldU(isnan(fldU))=0; fldV(isnan(fldV))=0;

dxg=mk3D(mygrid.DXG,fldU); dyg=mk3D(mygrid.DYG,fldU);
if size(fldU.f1,3)==length(mygrid.DRF); drf=mk3D(mygrid.DRF,fldU); else; drf=fldU; drf(:)=1; end;
facW=drf; facW(:)=1; facS=facW;
for ii=1:length(list_factors);
  tmp1=list_factors{ii};
  if strcmp(tmp1,'dh'); facW=facW.*dyg; facS=facS.*dxg;
  elseif strcmp(tmp1,'dz'); facW=facW.*drf; facS=facS.*drf;
  elseif strcmp(tmp1,'hfac'); facW=facW.*mygrid.hFacW; facS=facS.*mygrid.hFacS;
  elseif isempty(tmp1); 1;
  else; fprintf('error in calc_overturn : non supported factor\n'); return;
  end;
end;

for k4=1:n4;
fldU(:,:,:,k4)=fldU(:,:,:,k4).*facW;
fldV(:,:,:,k4)=fldV(:,:,:,k4).*facS;
end;

%use array format to speed up computation below:
fldU=convert2array(fldU); fldV=convert2array(fldV);

for iy=1:length(mygrid.LATS_MASKS); 

   %get list ofpoints that form a zonal band:
   mskW=mygrid.LATS_MASKS(iy).mskWedge;
   vecW=gcmfaces_subset(mskW,fldU);
   mskS=mygrid.LATS_MASKS(iy).mskSedge;
   vecS=gcmfaces_subset(mskS,fldV);
   trsp=nansum(vecW,1)+nansum(vecS,1);

   %store:
   if doFlip;
     FLD(iy,1:n3,:)=flipdim(cumsum(flipdim(trsp,2),2),2);
   else;
     FLD(iy,2:n3+1,:)=cumsum(trsp,2);
   end;

   %convert to Sv and change sign:
   FLD(iy,:,:)=-1e-6*FLD(iy,:,:);

end; 

if doFlip;
  FLD(:,end,:)=0;
else;
  FLD(:,1,:)=0;
end;

1	gforget	1.6	function [FLD]=calc_overturn(fldU,fldV,doFlip,list_factors);
2	gforget	1.2	%object: compute meridional overturning streamfunction
3			%inputs: fldU and fldV are the fields of grid point transport
4	gforget	1.6	%optional: doFlip (default is 1). If 1 then flip the vertical
5			% axis back and forth, hence intergrating from the
6			% 'bottom'. If 0 then dont.
7			% list_factors (default is {'dh','dz'})
8	gforget	1.2	%output: FLD is the streamfunction
9	gforget	1.3	%
10			%notes: mygrid.LATS_MASKS is the set of quasi longitudinal lines along which
11			% transports will integrated, as computed in gcmfaces_lines_zonal
12	gforget	1.6	% the result is converted to Sv, and sign is changed.
13	gforget	1.1
14			global mygrid;
15
16	gforget	1.6	if nargin<3; doFlip=1; end;
17			if nargin<4; list_factors={'dh','dz'}; end;
18
19	gforget	1.1	%initialize output:
20			n3=max(size(fldU.f1,3),1); n4=max(size(fldV.f1,4),1);
21	gforget	1.3	FLD=NaN*squeeze(zeros(length(mygrid.LATS_MASKS),n3+1,n4));
22	gforget	1.1
23			%prepare fldU/fldV:
24			fldU(isnan(fldU))=0; fldV(isnan(fldV))=0;
25
26	gforget	1.6	dxg=mk3D(mygrid.DXG,fldU); dyg=mk3D(mygrid.DYG,fldU);
27			if size(fldU.f1,3)==length(mygrid.DRF); drf=mk3D(mygrid.DRF,fldU); else; drf=fldU; drf(:)=1; end;
28			facW=drf; facW(:)=1; facS=facW;
29			for ii=1:length(list_factors);
30			tmp1=list_factors{ii};
31			if strcmp(tmp1,'dh'); facW=facW.dyg; facS=facS.dxg;
32			elseif strcmp(tmp1,'dz'); facW=facW.drf; facS=facS.drf;
33			elseif strcmp(tmp1,'hfac'); facW=facW.mygrid.hFacW; facS=facS.mygrid.hFacS;
34			elseif isempty(tmp1); 1;
35	gforget	1.9	else; fprintf('error in calc_overturn : non supported factor\n'); return;
36	gforget	1.6	end;
37			end;
38
39	gforget	1.1	for k4=1:n4;
40	gforget	1.6	fldU(:,:,:,k4)=fldU(:,:,:,k4).*facW;
41			fldV(:,:,:,k4)=fldV(:,:,:,k4).*facS;
42	gforget	1.1	end;
43
44			%use array format to speed up computation below:
45			fldU=convert2array(fldU); fldV=convert2array(fldV);
46
47	gforget	1.3	for iy=1:length(mygrid.LATS_MASKS);
48	gforget	1.1
49			%get list ofpoints that form a zonal band:
50	gforget	1.5	mskW=mygrid.LATS_MASKS(iy).mskWedge;
51			vecW=gcmfaces_subset(mskW,fldU);
52			mskS=mygrid.LATS_MASKS(iy).mskSedge;
53			vecS=gcmfaces_subset(mskS,fldV);
54	gforget	1.7	trsp=nansum(vecW,1)+nansum(vecS,1);
55	gforget	1.1
56			%store:
57	gforget	1.8	if doFlip;
58			FLD(iy,1:n3,:)=flipdim(cumsum(flipdim(trsp,2),2),2);
59			else;
60			FLD(iy,2:n3+1,:)=cumsum(trsp,2);
61			end;
62	gforget	1.6
63			%convert to Sv and change sign:
64	gforget	1.8	FLD(iy,:,:)=-1e-6*FLD(iy,:,:);
65	gforget	1.1
66			end;
67
68	gforget	1.8	if doFlip;
69			FLD(:,end,:)=0;
70			else;
71			FLD(:,1,:)=0;
72			end;
73	gforget	1.1