! Comparing Era Interim Data to Coarse Bling and Aeolus output: latent heat flux surface, sensible heat flux surface
!
! Arguments
! $1 = Season index: 1=DJF, 2=MAM, 3=JJA, 4=SON, ANNUAL=1:4@ave
! $2 = Season: DJF, MAM, JJA, SON, ANNUAL
! $3 = dataset

cancel data/all

!spawn cdo -seldate\,0052-01-01T00:00:00\,0104-12-31T23:59:59 $3 Heatflux-data.nc
!spawn ncatted -a axis\,level\,c\,c\,Z Heatflux-data.nc
!spawn cdo yseasmean Heatflux-data.nc Heatflux-data_sm.nc
!spawn cdo settaxis,2013-02-01,12:00,3months Heatflux-data_sm.nc Heatflux-data_sm_taxis.nc 
!spawn ncks -A -v levelb $3 Heatflux-data_sm_taxis.nc
!spawn ncatted -a edges\,level\,c\,c\,levelb Heatflux-data_sm_taxis.nc
!spawn rm -f Heatflux-data.nc Heatflux-data_sm.nc
!use "Heatflux-data_sm_taxis.nc" !d=1 AEOLUS DATA
use $3 !d=1 AEOLUS DATA
use "/p/projects/climber3/caesar/CM2M/multi_year_seasonal_means/00530101.atmos-dly_sm.nc" !d=2 CM2M DATA
use "/p/projects/climber3/caesar/CM2M/multi_year_seasonal_means/00530101.flux-dly_sm.nc" !d=3 CM2M DATA
use "/p/projects/climber3/caesar/Era_interim_data/noleap_data/multi_year_season_mean/latent_heat_flux_sf_sm.nc"	!d=4
use "/p/projects/climber3/caesar/Era_interim_data/noleap_data/multi_year_season_mean/sens_heat_sf_sm.nc" !d=5

set memory/size=10000

if `"($PROGRAM_NAME)" eq "Ferret"` then
set mode metafile: heatflux.plt
endif

!!!!!!!!!!!!!!!!!!!!!!!!!!
!!!!! Make the plots! !!!!
!!!!!!!!!!!!!!!!!!!!!!!!!!

set win 1
! Latent heat flux at the surface
set viewport UR
shade/nolabels/levels=20 EVAPO[d=1,l=$1]*2.5e6; go land
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Longitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Latitude"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Aeolus $2 mean"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Latent heat (W/m^2)"
!---timestemp--------------------------------------------------
let xpos=`($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.1 "@ASPIK/L.Caesar<nl>($SESSION_DATE)"
PPL SHASET RESET

set viewport UL
shade/nolabels/levels=(-inf)(`($lev_min)+($lev_del)`,`($lev_max)-($lev_del)`,($lev_del))(inf) SLHF[d=4,l=$1]/43200*-1; go land
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Longitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Latitude"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Era Interim $2 climatology (1979-2013)"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Latent heat (W/m^2)"
PPL SHASET RESET

set viewport LL
shade/nolabels/levels=($lev_text) EVAP[d=3,l=$1]*2.5e6; go land
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Longitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Latitude"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "CM2M $2 mean"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Latent heat (W/m^2)"
PPL SHASET RESET

set viewport LR
plot/nolabels SLHF[d=4,l=$1,i=@ave]/43200*-1, EVAPO[d=1,l=$1,i=@ave]*2.5e6,EVAP[d=3,i=@ave,l=$1]*2.5e6
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Latitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Latent heat"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "$2 Zonal Means"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Latent heat - positive upwards (W/m^2)"
!---key--------------------------------------------------
let xpos=`0.05*($PPL$XLEN)`
let ypos=`0.9*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,-1,1,.1 "@P1Era Interim"
let xpos=`0.05*($PPL$XLEN)`
let ypos=`0.85*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,-1,1,.1 "@P2Aeolus"
let xpos=`0.05*($PPL$XLEN)`
let ypos=`0.8*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,-1,1,.1 "@P3CM2M"
PPL SHASET RESET

if `"($PROGRAM_NAME)" eq "PyFerret"` then
!frame/file="heatflux_1.ps" Sigh. Postscript export is broken with Qt5
frame/file="heatflux_1.pdf"
spawn pdf2ps heatflux_1.pdf heatflux_1.ps
spawn rm heatflux_1.pdf
endif

!!

set win 2
! Sensible heat flux at the surface
set viewport UR
shade/nolabels/levels=20 F_SENS[d=1,k=3,l=$1]; go land
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Longitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Latitude"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Aeolus $2 mean"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Sensible heat (W/m^2)"
!---timestemp--------------------------------------------------
let xpos=`($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.1 "@ASPIK/L.Caesar<nl>($SESSION_DATE)"
PPL SHASET RESET

set viewport UL
shade/nolabels/levels=(-inf)(`($lev_min)+($lev_del)`,`($lev_max)-($lev_del)`,($lev_del))(inf) SSHF[d=5,l=$1]/43200*-1; go land
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Longitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Latitude"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Era Interim $2 climatology (1979-2013)"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Sensible heat (W/m^2)"
PPL SHASET RESET

set viewport LL
shade/nolabels/levels=($lev_text) SHFLX[d=3,k=10,l=$1]; go land
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Longitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Latitude"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "CM2M $2 mean"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Sensible heat (W/m^2)"
PPL SHASET RESET

set viewport LR
plot/nolabels SSHF[d=5,l=$1,i=@ave]/43200*-1, F_SENS[d=1,k=3,l=$1,i=@ave],SHFLX[d=3,k=10,i=@ave,l=$1]
!---xlabel-------------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.5*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Latitude"
!---ylabel-------------------------------------------------
let xpos=`-0.5*($PPL$XORG)`
let ypos=`0.5*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,0,90,.15 "Sensible heat"
!---titel 1 - above--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`($PPL$YLEN)+0.2*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "$2 Zonal Means"
!---titel 2 - below--------------------------------------------
let xpos=`0.5*($PPL$XLEN)`
let ypos=`-0.8*($PPL$YORG)`
label/nouser `xpos`,`ypos`,0,0,.15 "Sensible heat - positive upwards (W/m^2)"
!---key--------------------------------------------------
let xpos=`0.05*($PPL$XLEN)`
let ypos=`0.9*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,-1,1,.1 "@P1Era Interim"
let xpos=`0.05*($PPL$XLEN)`
let ypos=`0.85*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,-1,1,.1 "@P2Aeolus"
let xpos=`0.05*($PPL$XLEN)`
let ypos=`0.8*($PPL$YLEN)`
label/nouser `xpos`,`ypos`,-1,1,.1 "@P3CM2M"
PPL SHASET RESET

if `"($PROGRAM_NAME)" eq "PyFerret"` then
!frame/file="heatflux_2.ps" Sigh. Postscript export is broken with Qt5
frame/file="heatflux_2.pdf"
spawn pdf2ps heatflux_2.pdf heatflux_2.ps
spawn rm heatflux_2.pdf
endif

!!

!!!!! The end !!!!!
if `"($PROGRAM_NAME)" eq "Ferret"` then
  CANCEL MODE METAFILE
  spawn Fprint -l cps -o heatflux_1.ps -R heatflux.plt
  spawn Fprint -l cps -o heatflux.ps -R heatflux.plt.~*~
  spawn rm heatflux.plt*
  spawn cat heatflux_1.ps >> heatflux.ps
else
  spawn cat heatflux_?.ps > heatflux.ps
endif
spawn rm heatflux_?.ps
spawn ps2pdf heatflux.ps heatflux_seas_$2.pdf
spawn rm heatflux.ps
spawn rm ferret.jnl
cancel data/all
!spawn rm -f Heatflux-data_sm_taxis.nc