source: CONFIG/LMDZOR/branches/LMDZ4OR_v3/LMDZOR/COMP/lmdz.driver @ 959

# $Id$
#
#!/bin/ksh
#-----------------------------------------------------------------
function LMDZ_sed
{
    IGCM_debug_PushStack "LMDZ_sed"

    sed -e "s/^${2}\ *=.*/${2}= ${3}/" ${1} > ${1}.tmp
    RET=$?
    echo "LMDZ_sed : ${1} ${2} ${3}"
    \mv ${1}.tmp ${1}

    IGCM_debug_PopStack "LMDZ_sed"
    return $RET
}

function ATM_Initialize
{
    IGCM_debug_PushStack "ATM_Initialize"

    RESOL_ATM=$( echo $RESOL | awk "-F-" '{print $1}')

    [ -f ${SUBMIT_DIR}/../.resol ] && eval $(grep RESOL_ATM_3D ${SUBMIT_DIR}/../.resol) || RESOL_ATM_3D=96x95x19

    RESOL_ATM_Z=$( echo ${RESOL_ATM_3D} | awk "-Fx" '{print $3}' )

    ##-- Calendar type for LMDZ
    case ${config_UserChoices_CalendarType} in
        leap|gregorian)
            CalendarTypeForLmdz=earth_366d;;
        noleap)
            CalendarTypeForLmdz=earth_365d;;
        360d)
            CalendarTypeForLmdz=earth_360d;;
        *)
            CalendarTypeForLmdz=earth_360d
    esac

    ##- Create_etat0_limit version
    IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/lmdz.card UserChoices CREATE
    CREATE=${lmdz_UserChoices_CREATE}
    echo create_etat0_limit version : ${CREATE}

    ##- Default number of processor for lmdz
    NUM_PROC_ATM=1
    if [ X"${BATCH_NUM_PROC_TOT}" != X ] ; then
        NUM_PROC_ATM=${BATCH_NUM_PROC_TOT}
    fi
 
    ##--Frequency purpose ....
    ##--  Initialisation  ....
    OK_instan=n
    OK_journe=n
    OK_mensuel=n
    ok_hf=n

    case ${config_UserChoices_PeriodLength} in
        1Y|1y|1M|1m) OK_mensuel=y ;;
        5D|5d|1D|1d) OK_journe=y ;;
    esac

    for frequency in ${config_ATM_WriteFrequency} ; do
        case ${frequency} in
            5D|5d|1D|1d) OK_journe=y ;;
        esac
        case ${frequency} in
            HF|hf) ok_hf=y ;;
        esac
    done

    ##--Variables used by LMDZ in physiq.def --
    PAT_INST=$(        grep -v ^# ${SUBMIT_DIR}/PARAM/physiq.def_L${RESOL_ATM_Z}  | grep 'OK_instan='   )
    PAT_JOUR=$(        grep -v ^# ${SUBMIT_DIR}/PARAM/physiq.def_L${RESOL_ATM_Z}  | grep 'OK_journe='   )
    PAT_MOIS=$(        grep -v ^# ${SUBMIT_DIR}/PARAM/physiq.def_L${RESOL_ATM_Z}  | grep 'OK_mensuel='  )
    PAT_HFRE=$(        grep -v ^# ${SUBMIT_DIR}/PARAM/physiq.def_L${RESOL_ATM_Z}  | grep 'ok_hf='       )
    PAT_ecrit_mth=$(   grep -v ^# ${SUBMIT_DIR}/PARAM/physiq.def_L${RESOL_ATM_Z}  | grep 'ecrit_mth='   )
    PAT_ecrit_ISCCP=$( grep -v ^# ${SUBMIT_DIR}/PARAM/physiq.def_L${RESOL_ATM_Z}  | grep 'ecrit_ISCCP=' )

    ##--Variables used by LMDZ in gcm.def --
    PAT_iphysiq=$(   grep -v ^# ${SUBMIT_DIR}/PARAM/gcm.def_${RESOL_ATM_3D} | grep iphysiq  )
    PAT_iperiod=$(   grep -v ^# ${SUBMIT_DIR}/PARAM/gcm.def_${RESOL_ATM_3D} | grep iperiod  )
    PAT_day_step=$(  grep -v ^# ${SUBMIT_DIR}/PARAM/gcm.def_${RESOL_ATM_3D} | grep day_step )

    ## Read LMDZ_NbPeriod_adjust option in lmdz.card 
    IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/lmdz.card UserChoices LMDZ_NbPeriod_adjust
    eval LMDZ_NbPeriod_adjust=\${lmdz_UserChoices_LMDZ_NbPeriod_adjust} > /dev/null 2>&1

    if [ ${LMDZ_NbPeriod_adjust} -eq 0 ] ; then
        IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/lmdz.card UserChoices LMDZ_Bands_file_name
        eval LMDZ_Bands_file_name=\${lmdz_UserChoices_LMDZ_Bands_file_name} > /dev/null 2>&1
    fi

    ##-- This could be define in lmdz.card, inside section [UserChoices]
    ##-- Otherwise we get the value in *.def
    iperiod=$( echo ${PAT_iperiod} | awk -F= '{print $2}' )
    iphysiq=$( echo ${PAT_iphysiq} | awk -F= '{print $2}' )
    # day_step : number of steps per day (multiple of iperiod)
    #(( day_step = 48 * iphysiq ))
    day_step=$( echo ${PAT_day_step} | awk -F= '{print $2}' )

    ##  Read LMDZ_Freq_aero and LMDZ_Length_aero in lmdz.card
    eval LMDZ_Freq_aero=\${lmdz_UserChoices_LMDZ_Freq_aero} > /dev/null 2>&1
    eval LMDZ_Length_aero=\${lmdz_UserChoices_LMDZ_Length_aero} > /dev/null 2>&1

    IGCM_debug_PopStack "ATM_Initialize"
}

#-----------------------------------------------------------------
function ATM_Update
{
    IGCM_debug_PushStack "ATM_Update"

    case ${config_UserChoices_PeriodLength} in
        *Y|*y) 
               LMDZ_ecrit_mth=30. 
               LMDZ_ecrit_ISCCP=30.
               LMDZ_periodav=30.
               if [ "${config_UserChoices_CalendarType}" != "360d" ] ; then
                  echo Change calendartype in config.card. PeriodLength=1Y allowed only for CalendarType=360d 
                  exit
               fi
               ;;
        *)
               LMDZ_ecrit_mth=${PeriodLengthInDays}.
               LMDZ_ecrit_ISCCP=${PeriodLengthInDays}.
               LMDZ_periodav=${PeriodLengthInDays}.
               ;;
    esac

    ##-- Remise ou non a zero de la date initiale de LMDZ pour le fichier run.def
    if [ ${CumulPeriod} -eq 1 ] ; then
        RAZ_DATE=1
    else
        RAZ_DATE=0
    fi

    ## To have more variables in output files for aerosols. Parameters are set in lmdz.card
    if [ ${CumulPeriod}%${LMDZ_Freq_aero} -le ${LMDZ_Length_aero} ]; then
       if [ ${CumulPeriod}%${LMDZ_Freq_aero} -ne 0 ]; then 
          lev_histmth=4
       else
          lev_histmth=2    
       fi
    else
       lev_histmth=2
    fi    

    ## algo : For CumulPeriod=1 ; 
    ##        IF NbPeriod_Adjust = 0, Get Bands file directly from server, store it with _0 suffix
    ##        ELSE LMDZ_adjust=y ; save Bands file on file server ;
    ##        For CumulPeriod=2 to LMDZ_NbPeriod_adjust ; Get Bands file from server (CumulPeriod-1) ; LMDZ_adjust=y ; save Bands file on file server ;
    ##        For CumulPeriod > LMDZ_NbPeriod_adjust ; LMDZ_adjust=n ; Get Bands file from server (CumulPeriod=LMDZ_NbPeriod_adjust 0 to force it) ; Use Bands ;

    if [ ${CumulPeriod} -eq 1 ] ; then
        if [ ${LMDZ_NbPeriod_adjust} -eq 0 ] ; then
            IGCM_sys_Get ${LMDZ_Bands_file_name} Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat ; IGCM_sys_Chmod u+w Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat
            IGCM_sys_Put_Out Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat ${R_OUT_ATM_D}/${config_UserChoices_JobName}_Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat_0
            IGCM_debug_Print 1 "Bands file forced to ${LMDZ_Bands_file_name} and stored in ${config_UserChoices_JobName}_Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat_0"
        fi
    fi

    LMDZ_adjust=n
    [ ${CumulPeriod} -le ${LMDZ_NbPeriod_adjust} ] && LMDZ_adjust=y

    RefPeriod=${LMDZ_NbPeriod_adjust} #0 to use a Bands file from Restart
    [ ${CumulPeriod} -le ${LMDZ_NbPeriod_adjust} ] && (( RefPeriod = ${CumulPeriod} - 1 )) 

    [ ${CumulPeriod} -gt 1 ] && ( IGCM_sys_Get ${R_OUT_ATM_D}/${config_UserChoices_JobName}_Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat_${RefPeriod} Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat ; IGCM_sys_Chmod u+w Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat )

    ## physiq.def parameters modified from initial physiq_L${RESOL_ATM_Z}
    LMDZ_sed physiq.def OK_instan   ${OK_instan}
    LMDZ_sed physiq.def OK_journe   ${OK_journe}
    LMDZ_sed physiq.def OK_mensuel  ${OK_mensuel}
    LMDZ_sed physiq.def lev_histmth ${lev_histmth}
    LMDZ_sed physiq.def ok_hf       ${ok_hf}
    LMDZ_sed physiq.def ecrit_mth   ${LMDZ_ecrit_mth}
    LMDZ_sed physiq.def ecrit_ISCCP ${LMDZ_ecrit_ISCCP}

    ## gcm.def parameters : no change since gcm.def_${RESOL_ATM_3D} is used and already modified

    ## run.def parameters
    LMDZ_sed run.def dayref   ${InitDay}
    LMDZ_sed run.def anneeref ${InitYear}
    LMDZ_sed run.def calend   ${CalendarTypeForLmdz}
    LMDZ_sed run.def nday     ${PeriodLengthInDays}
    LMDZ_sed run.def raz_date ${RAZ_DATE}
    LMDZ_sed run.def periodav ${LMDZ_periodav}
    LMDZ_sed run.def adjust   ${LMDZ_adjust}

    ## Read ByPass_hgardfou_teta option in lmdz.card --> divide teta* by 2 if [ $ByPass_hgardfou_teta = 1 ]
    IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/lmdz.card UserChoices ByPass_hgardfou_teta
    eval ByPass_hgardfou_teta=\${lmdz_UserChoices_ByPass_hgardfou_teta} > /dev/null 2>&1
    if [ ${ByPass_hgardfou_teta} = y ] ; then
        awk '{ if ($0 ~ /^teta.*=/) {split($0,a,"=") ; print a[1]"="a[2]/2"."} else print $0}' gcm.def > gcm.def.tmp
        IGCM_sys_Mv gcm.def.tmp gcm.def
        echo
        IGCM_debug_Print 1 "ByPass_hgardfou_teta : ^teta*/2 in gcm.def"
        echo
        cat gcm.def
        ByPass_hgardfou_teta=n
        IGCM_card_WriteOption ${SUBMIT_DIR}/COMP/lmdz.card UserChoices ByPass_hgardfou_teta "${ByPass_hgardfou_teta}"
    fi

    ## Read ByPass_hgardfou_mats option in lmdz.card --> purmats=y 2 if [ $ByPass_hgardfou_mats = 1 ]
    IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/lmdz.card UserChoices ByPass_hgardfou_mats
    eval ByPass_hgardfou_mats=\${lmdz_UserChoices_ByPass_hgardfou_mats} > /dev/null 2>&1
    if [ ${ByPass_hgardfou_mats} = y ] ; then
        sed -e "s/^purmats=.*/purmats=y/" gcm.def > gcm.def.tmp
        IGCM_sys_Mv gcm.def.tmp gcm.def
        echo
        IGCM_debug_Print 1 "ByPass_hgardfou_mats : purmats=y in gcm.def"
        echo
        cat gcm.def
        ByPass_hgardfou_mats=n
        IGCM_card_WriteOption ${SUBMIT_DIR}/COMP/lmdz.card UserChoices ByPass_hgardfou_mats "${ByPass_hgardfou_mats}"
    fi

    IGCM_debug_PopStack "ATM_Update"
}

#-----------------------------------
function ATM_Finalize
{
    IGCM_debug_PushStack "ATM_Finalize"

    [ ${CumulPeriod} -le ${LMDZ_NbPeriod_adjust} ] && IGCM_sys_Put_Out Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat ${R_OUT_ATM_D}/${config_UserChoices_JobName}_Bands_${RESOL_ATM_3D}_${NUM_PROC_ATM}prc.dat_${CumulPeriod}

    echo FINALIZE ATM !

    IGCM_debug_PopStack "ATM_Finalize"
}