source: CONFIG/publications/ICOLMDZORINCA_CO2_Transport_GMD_2023/INCA/tools/TEMPLATES/extract_bcpom_budget.job @ 6610

#!/usr/bin/ksh 
# script for new INCAparallel  LMDZ4 output files
set -ex

user=p86cozic
producer=p86cozic

config=LMDZORINCA
model=LOI_01

set -A ANN 2000
set -A MTH '01'  '02'  '03' '04' '05'  '06' '07' '08' '09' '10' '11' '12'
firstmth='01'
lastmth='12'
lastday=29 # =30 si time_counter=31 / =29 si time_counter=30

compute_budgetfiles=1


dirout=/dmnfs/cont003/${producer}/IGCM_OUT/${config}/${model}/CHM/Output/MO
dirscr=/scratchdir/cont003/${producer}/INCA_OUTPUT/${model}

dirproc=/scratchdir/cont003/${user}/INCA_OUTPUT/${model}
mkdir -p $dirproc
cd $dirproc

for AN in ${ANN[*]} ; do 
    im=0
    for IMTH in ${MTH[*]} ; do 


        tagname=_${AN}${IMTH}01_${AN}${IMTH}30_1M_
        infile=${dirproc}/${AN}${IMTH}/MM_${model}${tagname}inca_avgr.nc
        moy_file=${dirout}/MM_${model}${tagname}inca_avgr.nc
        
        if [ ! -f ${moy_file} ] ; then
          if [ ! -f ${infile} ]; then 
            cd  ${dirproc}/
            mkdir -p ${AN}${IMTH}
            cd ${dirproc}/${AN}${IMTH}/
            cp ${dirout}/${model}${tagname}inca_avgr.nc .
            ncra -O ${model}${tagname}inca_avgr.nc MM_${model}${tagname}inca_avgr.nc
          fi    
          cp MM_${model}${tagname}inca_avgr.nc ${moy_file}
        fi      
        cd $dirproc

        file=bcpombudget_${model}.an${AN}.m${IMTH}.nc

if [ 1 = $compute_budgetfiles ] ; then         
        ncks -O -v area,lat,lon,presnivs,time_counter,AIRMASS,\
'Emi_.*BCM','Emi_.*POMM','DRY.*POMM','WET.*POMM','DRY.*BCM','SED.*BCM','SED.*POMM','WET.*BCM',\
'Emi_.*CIDUSTM','DRY.*CIDUSTM','SED.*CIDUSTM','WET.*CIDUSTM',\
'Emi_.*SSM','DRY.*SSM','SED.*SSM','WET.*SSM',\
AIPOMM,ASPOMM,AIBCM,ASBCM,CIDUSTM,ASSSM,CSSSM,SSSSM $infile $file

        ncwa -O -a presnivs -y ttl -v  'Emi_alt.*' $file tmp.nc
        ncks -O -x -v 'Emi_alt.*' $file $file
        ncks -A -v  'Emi_alt.*' tmp.nc $file
        ncap -O -s "\
            EMI_BC=Emi_alt_AIBCM+Emi_alt_ASBCM+Emi_AIBCM+Emi_ASBCM;\
            EMI_POM=Emi_alt_AIPOMM+Emi_alt_ASPOMM+Emi_AIPOMM+Emi_ASPOMM;\
            EMI_AIBC=Emi_alt_AIBCM+Emi_AIBCM;\
            EMI_ASBC=Emi_alt_ASBCM+Emi_ASBCM;\
            EMI_AIPOM=Emi_alt_AIPOMM+Emi_AIPOMM;\
            EMI_ASPOM=Emi_alt_ASPOMM+Emi_ASPOMM;\
            EMI_DUST=Emi_CIDUSTM;\
            EMI_SS=Emi_ASSSM+Emi_CSSSM;\
            EMI_CSS=Emi_CSSSM;\
            EMI_ASS=Emi_ASSSM;\
            EMI_SSS=Emi_SSSSM;\
            WET_POM=WET_AIPOMM+WET_ASPOMM;\
            WET_BC=WET_AIBCM+WET_ASBCM;\
            WET_DUST=WET_CIDUSTM;\
            WET_SS=WET_ASSSM+WET_CSSSM;\
            SEDDRY_POM=DRY_AIPOMM+DRY_ASPOMM+SED_AIPOMM+SED_ASPOMM;\
            SEDDRY_BC=DRY_AIBCM+DRY_ASBCM+SED_AIBCM+SED_ASBCM;\
            SEDDRY_AIPOM=SED_AIPOMM+DRY_AIPOMM;\
            SEDDRY_ASPOM=SED_ASPOMM+DRY_ASPOMM;\
            SEDDRY_AIBC=SED_AIBCM+DRY_AIBCM;\
            SEDDRY_ASBC=SED_ASBCM+DRY_ASBCM;\
            SEDDRY_DUST=SED_CIDUSTM+DRY_CIDUSTM;\
            SEDDRY_SS=SED_ASSSM+DRY_ASSSM+SED_CSSSM+DRY_CSSSM;\
            " $file $file

        ncap -O -s "LOAD_BC[time_counter,presnivs,lat,lon]=(AIBCM+ASBCM)*AIRMASS/area;\
            LOAD_POM[time_counter,presnivs,lat,lon]=(AIPOMM+ASPOMM)*AIRMASS/area;\
            LOAD_AIBC[time_counter,presnivs,lat,lon]=AIBCM*AIRMASS/area;\
            LOAD_AIPOM[time_counter,presnivs,lat,lon]=AIPOMM*AIRMASS/area;\
            LOAD_ASBC[time_counter,presnivs,lat,lon]=ASBCM*AIRMASS/area;\
            LOAD_ASPOM[time_counter,presnivs,lat,lon]=ASPOMM*AIRMASS/area;\
            LOAD_DUST[time_counter,presnivs,lat,lon]=CIDUSTM*AIRMASS/area;\
            LOAD_SS[time_counter,presnivs,lat,lon]=(ASSSM+CSSSM)*AIRMASS/area;\
            LOAD_CSS[time_counter,presnivs,lat,lon]=CSSSM*AIRMASS/area;\
            LOAD_ASS[time_counter,presnivs,lat,lon]=ASSSM*AIRMASS/area;\
            LOAD_SSS[time_counter,presnivs,lat,lon]=SSSSM*AIRMASS/area;\
            " $file tmp.nc
        ncwa -O -a presnivs -y ttl -v  'LOAD.*' tmp.nc tmp.nc
        ncks -A -v  'LOAD.*' tmp.nc $file

fi      
    ((im=$im+1))    
    done

# accumulation during run
    infilestart=${dirout}/${model}_${AN}${firstmth}01_${AN}${firstmth}30_1M_inca_avgr.nc
    infileend=${dirout}/${model}_${AN}${lastmth}01_${AN}${lastmth}30_1M_inca_avgr.nc
    ncks -O -d time_counter,0 -v area,AIRMASS,AIPOMM,ASPOMM,AIBCM,ASBCM,CIDUSTM,ASSSM,CSSSM,SSSSM $infilestart accu.nc
    ncrename -v AIPOMM,AIPOMMST -v ASPOMM,ASPOMMST -v AIBCM,AIBCMST -v ASBCM,ASBCMST \
-v CIDUSTM,CIDUSTMST -v ASSSM,ASSSMST -v CSSSM,CSSSMST -v SSSSM,SSSSMST accu.nc
    ncks -A -d time_counter,$lastday -v AIPOMM,ASPOMM,AIBCM,ASBCM,CIDUSTM,ASSSM,CSSSM,SSSSM $infileend accu.nc
    ncap -O -s "ACCU_BC[time_counter,presnivs,lat,lon]=(AIBCM+ASBCM-AIBCMST-ASBCMST)*AIRMASS/area;\
            ACCU_POM[time_counter,presnivs,lat,lon]=(AIPOMM+ASPOMM-AIPOMMST-ASPOMMST)*AIRMASS/area;\
            ACCU_AIBC[time_counter,presnivs,lat,lon]=(AIBCM-AIBCMST)*AIRMASS/area;\
            ACCU_AIPOM[time_counter,presnivs,lat,lon]=(AIPOMM-AIPOMMST)*AIRMASS/area;\
            ACCU_ASBC[time_counter,presnivs,lat,lon]=(ASBCM-ASBCMST)*AIRMASS/area;\
            ACCU_ASPOM[time_counter,presnivs,lat,lon]=(ASPOMM-ASPOMMST)*AIRMASS/area;\
            ACCU_DUST[time_counter,presnivs,lat,lon]=(CIDUSTM-CIDUSTMST)*AIRMASS/area;\
            ACCU_SS[time_counter,presnivs,lat,lon]=(ASSSM+CSSSM-ASSSMST-CSSSMST)*AIRMASS/area;\
            ACCU_CSS[time_counter,presnivs,lat,lon]=(CSSSM-CSSSMST)*AIRMASS/area;\
            ACCU_ASS[time_counter,presnivs,lat,lon]=(ASSSM-ASSSMST)*AIRMASS/area;\
            ACCU_SSS[time_counter,presnivs,lat,lon]=(SSSSM-SSSSMST)*AIRMASS/area;\
            " accu.nc accu.nc
    ncwa -O -a lat,lon -w area -v 'ACCU.*' accu.nc tmp0.nc
    ncap -O -s "FAC=1.e6f" tmp0.nc tmp0.nc      
    ncwa -O -N -w FAC tmp0.nc tmp0.nc

    filean=bcpombudget_${model}.an${AN}.nc
    ncra -O bcpombudget_${model}.an${AN}.m??.nc $filean 

    ncwa -O -a lat,lon -w area -v 'SEDDRY.*','WET.*','EMI.*' $filean tmp.nc
    
    ncap -O -s "FAC=86400.f*365.f*1e6f" tmp.nc tmp.nc   
    ncwa -O -N -w FAC tmp.nc tmp.nc
    

    ncwa -O -a lat,lon -w area -v 'LOAD.*BC$','LOAD.*POM$','LOAD.*DUST$','LOAD.*SS$' $filean t.nc
    ncap -O -s "FAC=1.e6f" t.nc t.nc    
    ncwa -O -N -w FAC t.nc t.nc
    ncks -A -v 'LOAD.*BC$','LOAD.*POM$','LOAD.*DUST$','LOAD.*SS$'  t.nc  tmp.nc

    ncks -O -x -v FAC tmp.nc tmp.nc
    ncks -A -v 'ACCU.*' tmp0.nc tmp.nc

    echo "fluxes in mg m-2 y-1 // loads in mg m-2"
    echo ${model}
    ncks -H  tmp.nc

done

exit