clear

%grid
lats=-89:89; nz=90;

load FWHT_year2.mat %from (U*T)^bar(at T point)
aFW_yr2=aFW; gFW_yr2=gFW; pFW_yr2=pFW;
aHT_yr2=aHT; gHT_yr2=gHT; pHT_yr2=pHT;
kk=find(lats<-35|lats>70); aFW_yr2(kk,:)=nan; aHT_yr2(kk,:)=nan;
kk=find(lats<-35|lats>65); pFW_yr2(kk,:)=nan; pHT_yr2(kk,:)=nan;


figure(1)
fig_ps=[752 56 1537 1038];
set(gcf,'Position',fig_ps)

subplot(211)
plot(lats,[gFW_yr2 aFW_yr2 pFW_yr2],'linewidth',2)
grid on
xlabel('Latitude')
xlim([-90 90])
legend('Global','Atlantic','Pacific+Indian');
title('Meridional seawater transport (in Sv)');

subplot(212)
plot(lats,[gHT_yr2 aHT_yr2 pHT_yr2],'linewidth',2)
grid on
xlabel('Latitude')
xlim([-90 90])
legend('Global','Atlantic','Pacific+Indian');
title('Meridional Heat Transport (in PW)');

load FWHT_month2.mat
kk=find(lats<-35|lats>70); aFW(kk,:)=nan; aHT(kk,:)=nan;
kk=find(lats<-35|lats>65); pFW(kk,:)=nan; pHT(kk,:)=nan;
aFW_yrr=mean(aFW,2); gFW_yrr=mean(gFW,2); pFW_yrr=mean(pFW,2);
aHT_yrr=mean(aHT,2); gHT_yrr=mean(gHT,2); pHT_yrr=mean(pHT,2);

figure(4)
set(gcf,'Position',fig_ps)
subplot(211)
plot(lats,[gFW_yrr-gFW_yr2 aFW_yrr-aFW_yr2 pFW_yrr-pFW_yr2],'linewidth',2)
grid on
xlabel('Latitude')
xlim([-90 90])
legend('Global','Atlantic','Pacific+Indian');
title('Meridional seawater transport (in Sv)');

subplot(212)
plot(lats,[gHT_yrr-gHT_yr2 aHT_yrr-aHT_yr2 pHT_yrr-pHT_yr2],'linewidth',2)
grid on
xlabel('Latitude')
xlim([-90 90])
legend('Global','Atlantic','Pacific+Indian');
title('Meridional Heat Transport (in PW)');

figure(5)
set(gcf,'Position',fig_ps)
TT=3:14;
lat=-89:89;
subplot(211)
pcolor(TT,lat,gFW)
caxis([-3 3])
shading flat, thincb(1)
title('Meridional freshwater transport (Sv)')
xlabel('month from 2020/01')
subplot(212)
pcolor(TT,lat,gHT)
caxis([-4 4])
shading flat, thincb(1)
title('Meridional heat transport (PW)')
xlabel('month from 2020/01')

colormap(jet)

figure(1)
print -dpng lk_trsp_MT_lat
figure(5)
print -dpng lk_trsp_MT_time