#!/bin/ksh
#-----------------------------------------------------------------
function ATM_Initialize
{
    IGCM_debug_PushStack "ATM_Initialize"

    RESOL_ATM=$( echo $RESOL | awk "-Fx" '{print $2}' | 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

    ##--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 ${compname}.card 
    IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/${compname}.card UserChoices LMDZ_NbPeriod_adjust
    eval LMDZ_NbPeriod_adjust=\${${compname}_UserChoices_LMDZ_NbPeriod_adjust} > /dev/null 2>&1
    
    ## algo : CumulPeriod=1 ; LMDZ_adjust=y ; save Bands file on file server ;
    ##        For CumulPeriod=2 to LMDZ_NbPeriod_adjust ; Fetch Bands file from server (CumulPeriod-1) ; LMDZ_adjust=y ; save Bands file on file server ;
    ##        For CumulPeriod > LMDZ_NbPeriod_adjust ; LMDZ_adjust=n ; Fetch Bands file from server (CumulPeriod=LMDZ_NbPeriod_adjust) ; Use Bands ;

    ##-- 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}' )

    IGCM_debug_PopStack "ATM_Initialize"
}

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

    LMDZ_ecrit_mth=${PeriodLengthInDays}.
    LMDZ_ecrit_ISCCP=${PeriodLengthInDays}.
    LMDZ_periodav=${PeriodLengthInDays}.

    ##-- 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

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

    RefPeriod=${LMDZ_NbPeriod_adjust}
    [ ${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

    ## Mise en forme du fichier physiq.def
    sed -e "s/${PAT_INST}/OK_instan=${OK_instan}/"                 \
	-e "s/${PAT_JOUR}/OK_journe=${OK_journe}/"                 \
	-e "s/${PAT_MOIS}/OK_mensuel=${OK_mensuel}/"               \
	-e "s/${PAT_HFRE}/ok_hf=${ok_hf}/"                         \
        -e "s/${PAT_ecrit_mth}/ecrit_mth=${LMDZ_ecrit_mth}/"       \
        -e "s/${PAT_ecrit_ISCCP}/ecrit_ISCCP=${LMDZ_ecrit_ISCCP}/" \
	physiq.def > physiq.def.tmp
    IGCM_sys_Mv physiq.def.tmp physiq.def

    ## Mise en forme du fichier gcm.def
    sed -e "s/${PAT_day_step}/day_step=${day_step}/"    \
	-e "s/${PAT_iperiod}/iperiod=${iperiod}/"       \
	-e "s/${PAT_iphysiq}/iphysiq=${iphysiq}/"       \
	gcm.def > gcm.def.tmp
    IGCM_sys_Mv gcm.def.tmp gcm.def

    ## Mise en forme du fichier run.def
    sed -e "s/_dayref_/${InitDay}/"                    \
	-e "s/_anneeref_/${InitYear}/"                 \
        -e "s/_calend_/${CalendarTypeForLmdz}/"        \
	-e "s/_nday_/${PeriodLengthInDays}/"           \
	-e "s/_raz_date_/${RAZ_DATE}/"                 \
        -e "s/_periodav_/${LMDZ_periodav}/"            \
	-e "s/_adjust_/${LMDZ_adjust}/"                \
	run.def > run.def.tmp
    IGCM_sys_Mv run.def.tmp run.def

    ## Read ByPass_hgardfou_teta option in ${compname}.card --> divide teta* by 2 if [ $ByPass_hgardfou_teta = 1 ]
    IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/${compname}.card UserChoices ByPass_hgardfou_teta
    eval ByPass_hgardfou_teta=\${${compname}_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/${compname}.card UserChoices ByPass_hgardfou_teta "${ByPass_hgardfou_teta}"
    fi

    ## Read ByPass_hgardfou_mats option in ${compname}.card --> purmats=y 2 if [ $ByPass_hgardfou_mats = 1 ]
    IGCM_card_DefineVariableFromOption ${SUBMIT_DIR}/COMP/${compname}.card UserChoices ByPass_hgardfou_mats
    eval ByPass_hgardfou_mats=\${${compname}_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/${compname}.card UserChoices ByPass_hgardfou_mats "${ByPass_hgardfou_mats}"
    fi

    if ( ${FirstInitialize} ) ; then

	if ( [ "${config_Restarts_OverRule}" = "n" ] && [ "${config_ATM_Restart}" = "n" ] ); then
	    if [ ${DRYRUN} -le 2 ] ; then
		IGCM_sys_Cp ${R_EXE}/create_etat0_limit.e ${RUN_DIR}/.
		if [ ${DRYRUN} -le 1 ] ; then
		    ./create_etat0_limit.e
		fi
	    else
		echo "EXECUTION of ./create_etat0_limit.e simulated"
		echo "EXECUTION of ./create_etat0_limit.e simulated for DRYRUN = " $DRYRUN >> stack
	    fi
	fi

    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"
}
