pdf.sh

#!/bin/bash                

############# FUNCTION CALCULATE_3_LAST_MONTHS

calculate_3_last_months ()
{

if [ $current_month -eq 2 ]
then
    let next_year=$current_year
    let next_month=$current_month+1

    let one_year_ago=$current_year
    let one_month_ago=1
    
    let two_year_ago=$current_year-1
    let two_month_ago=12
elif [ $current_month -eq 1 ]
then
    let next_year=$current_year
    let next_month=$current_month+1

    let one_year_ago=$current_year-1
    let one_month_ago=12

    let two_year_ago=$current_year-1
    let two_month_ago=11
elif [ $current_month -eq 12 ]
then
    let next_year=$current_year+1
    let next_month=1

    let one_year_ago=$current_year
    let one_month_ago=$current_month-1

    let two_year_ago=$current_year
    let two_month_ago=$current_month-2

else
    let next_year=$current_year
    let next_month=$current_month+1

    let one_year_ago=$current_year
#$((10#$item))
    let one_month_ago=$current_month-1
#    let one_month_ago=$((10#$current_month))-1

    let two_year_ago=$current_year
    let two_month_ago=$current_month-2
fi
}

############## END OF FUNCTION CALCULATE_3_LAST_MONTHS


#export IDL_STARTUP=~/idlstartup.pro

{

mkdir ./data
mkdir ./out 
mkdir ./out/plots
mkdir ./out/files  


current_year_to_compare=`date -u +"%Y"`
current_month_to_compare=`date -u +"%m"`
current_day_to_compare=`date -u +"%d"`
current_hour_to_compare=`date -u +"%H"`

current_month_to_compare=`echo $current_month_to_compare|sed 's/^0*//'`
current_day_to_compare=`echo $current_day_to_compare|sed 's/^0*//'`
current_hour_to_compare=`echo $current_hour_to_compare|sed 's/^0*//'`


#echo "enter YYYYMMDDHHEPF"
date_ensemble=""$1""
year_pred=${date_ensemble:0:4}
month_pred=${date_ensemble:4:2}
day_pred=${date_ensemble:6:2}
hour_pred=${date_ensemble:8:2}
let ensemble_or_not=${date_ensemble:10:1}
let user_want_to_plot=${date_ensemble:11:1}
let user_want_file=${date_ensemble:12:1}
let user_want_error_file=${date_ensemble:13:1}
# echo "let user_want_to_plot=${date_ensemble:11:1}"
# echo "let user_want_file=${date_ensemble:12:1}"
# echo "let user_want_error_file=${date_ensemble:13:1}"

if [ $user_want_error_file -eq 0 ]
then
    redirect_output="/dev/null"
else
    redirect_output="./out/files/execution_output"
fi

} &>/dev/null

{


echo "**************************************"
echo "****** WELCOME TO THE PDF MODEL ******" 
echo "**************************************"
echo ""


month_pred=`echo $month_pred|sed 's/^0*//'`
day_pred=`echo $day_pred|sed 's/^0*//'`
if [ $((10#$hour_pred)) -ne 0 ] 
then
    hour_pred=`echo $hour_pred|sed 's/^0*//'`
else
    hour_pred='0'
fi

let year_pred_nb=$year_pred
let month_pred_nb=$month_pred

let current_year_to_compare_nb=$current_year_to_compare
let current_month_to_compare_nb=$current_month_to_compare
let previous_month_nb=$current_month_to_compare-1


if [ $year_pred_nb = $current_year_to_compare_nb ] 
then
    let same_year_as_today=1
else
    let same_year_as_today=0
fi
if [ $month_pred_nb = $previous_month_nb ]  || [ $month_pred = $current_month_to_compare ] 
then
    let same_month_or_one_month_ago=1
else
    let same_month_or_one_month_ago=0
fi

if [ $same_year_as_today == 1 ]  && [ $same_month_or_one_month_ago == 1 ] 
then ############### USE DATA FROM SWPC

    current_year=$year_pred
    current_month=$month_pred

    calculate_3_last_months

    if [ $current_month -lt 10 ] 
    then
	name_current_month="$current_year""0""$current_month""_ace_swepam_1h.txt"
    else
	name_current_month="$current_year""$current_month""_ace_swepam_1h.txt"
    fi

    if [ $one_month_ago -lt 10 ] 
    then
	name_one_month_ago="$one_year_ago""0""$one_month_ago""_ace_swepam_1h.txt"
    else
	name_one_month_ago="$one_year_ago""$one_month_ago""_ace_swepam_1h.txt"
    fi

    if [ $two_month_ago -lt 10 ] 
    then
	name_two_month_ago="$two_year_ago""0""$two_month_ago""_ace_swepam_1h.txt"
    else
	name_two_month_ago="$two_year_ago""$two_month_ago""_ace_swepam_1h.txt"
    fi


    if [ $next_month -lt 10 ] 
    then
	name_next_month="$next_year""0""$next_month""_ace_swepam_1h.txt"
    else
	name_next_month="$next_year""$next_month""_ace_swepam_1h.txt"
    fi



    echo ""
    echo "Downloading data from SWPC..."
    echo ""

    cd ./data/

    dt0=`curl -u anonymous:cbv@umich.edu -o current_month.txt ftp://ftp.swpc.noaa.gov/pub/lists/ace2/$name_current_month` 
    dt1=`curl -u anonymous:cbv@umich.edu -o one_month_ago.txt ftp://ftp.swpc.noaa.gov/pub/lists/ace2/$name_one_month_ago` 
    dt2=`curl -u anonymous:cbv@umich.edu -o two_months_ago.txt ftp://ftp.swpc.noaa.gov/pub/lists/ace2/$name_two_month_ago`

    if [ $month_pred_nb = $previous_month_nb ]
    then
	dt3=`curl -u anonymous:cbv@umich.edu -o next_month.txt ftp://ftp.swpc.noaa.gov/pub/lists/ace2/$name_next_month`
	let next_month_or_not=1
    else
	let next_month_or_not=0
    fi

    if [ $two_month_ago -lt 10 ]
    then
	name_two_month_ago="0""$two_month_ago"
    else 
	name_two_month_ago="$two_month_ago"
    fi

    if [ $month_pred -lt 10 ]
    then
	month_pred="0""$month_pred"
    else 
	month_pred="$month_pred"
    fi

    if [ $day_pred -lt 10 ]
    then
	day_pred="0""$day_pred"
    else 
	day_pred="$day_pred"
    fi

    if [ $hour_pred -lt 10 ]
    then
	hour_pred="0""$hour_pred"
    else 
	hour_pred="$hour_pred"
    fi

    start_date_chosen="$two_year_ago""$name_two_month_ago""$day_pred""$hour_pred"
    end_date_chosen="$year_pred""$month_pred""$day_pred""$hour_pred"

    cd ../src/

#/Applications/exelis/idl82/bin/
idl -e "convert_spwc_to_omni,'""$end_date_chosen""',""$next_month_or_not"
cd ../data/
rm "current_month.txt" "one_month_ago.txt" "two_months_ago.txt" 
if [ $month_pred_nb = $previous_month_nb ] 
then
    rm "next_month.txt"
fi

cd ../


############### END OF USE DATA FROM SWPC
else ############### USE DATA FROM OMNIWEB

    echo ""
    echo "Downloading data from OmniWeb..."
    echo ""

    current_year=$year_pred
    current_month=$month_pred

    calculate_3_last_months

   
    if [ $two_month_ago -lt 10 ]
    then
	name_two_month_ago="0""$two_month_ago"
    else 
	name_two_month_ago="$two_month_ago"
    fi

    if [ $month_pred -lt 10 ]
    then
	month_pred="0""$month_pred"
    else 
	month_pred="$month_pred"
    fi

    if [ $day_pred -lt 10 ]
    then
	day_pred="0""$day_pred"
    else 
	day_pred="$day_pred"
    fi

    if [ $hour_pred -lt 10 ]
    then
	hour_pred="0""$hour_pred"
    else 
	hour_pred="$hour_pred"
    fi


    if [ $next_month -lt 10 ]
    then
	name_next_month="0""$next_month"
    else
	name_next_month="$next_month"
    fi

    start_date_chosen_download="$two_year_ago""$name_two_month_ago""$day_pred"
    end_date_chosen_for_download="$next_year""$name_next_month""$day_pred"

    end_date_chosen="$year_pred""$month_pred""$day_pred"

    start_date_chosen="$two_year_ago""$name_two_month_ago""$day_pred""$hour_pred"
    end_date_chosen="$year_pred""$month_pred""$day_pred""$hour_pred"    
    name_file_downloaded="PDF_speed""_""$year_pred""-$month_pred""-$day_pred""-$hour_pred""00"".txt"

    let var_chosen="24"

    curl -d  "activity=retrieve&res=hour&spacecraft=omni2&start_date=""$start_date_chosen_download""&end_date=""$end_date_chosen_for_download""&vars=""$var_chosen""&scale=Linear&ymin=&ymax=&view=0&charsize=&xstyle=0&ystyle=0&symbol=0&symsize=&linestyle=solid&table=0&imagex=640&imagey=480&color=&back=" https://omniweb.gsfc.nasa.gov/cgi/nx1.cgi > ./data/$name_file_downloaded
   
# Test if the file from onmiWeb has been successfully downloaded

    word_error=Error
    grep -q "\<$word_error\>" "./data/""$name_file_downloaded" && echo "The data has not been successfully downloaded, probably due to an error in the start/end date(s). Please try to run the script again." && exit 1
   

############### END OF USE DATA FROM OMNIWEB
fi
echo ""
echo "Downloading successfully completed."
echo ""
echo "The PDF Model is now predicting the speeds for the next 5 days..."
echo ""    


cd ./src/
#/Applications/exelis/idl82/bin/
idl -e "main,""$ensemble_or_not"",""$user_want_to_plot"",""$user_want_file"",'""$end_date_chosen" 

cd ../

# if [ $user_want_to_plot -eq 1 ]
# then
#     name_plot_ps="./out/plots/PDF_speed""_""$year_pred""-$month_pred""-$day_pred""-$hour_pred""00"".ps"
#     name_plot_pdf="./out/plots/PDF_speed""_""$year_pred""-$month_pred""-$day_pred""-$hour_pred""00"".pdf"
#     ps2pdf $name_plot_ps $name_plot_pdf 
#     rm $name_plot_ps 
#  #  open $name_plot_pdf
# fi




echo ""
echo "**********************************************"
echo "THANK YOU FOR USING THE PDF MODEL. A BIENTOT !"
echo "**********************************************"
echo ""

name_execution_output="./out/files/execution_output_PDF_speed""_""$year_pred""-$month_pred""-$day_pred""-$hour_pred""00"

}  &> "$redirect_output"

if [ $user_want_error_file -eq 1 ]
then
    mv ./out/files/execution_output "$name_execution_output" 
fi