# This script runs Lumi macros for the full Sim

# reads number of Events, name of run and verbose Level
nEvts=10
nJobs=1
#path="test"
verbose=0
mom=0

echo Hey! Ready to sum analysis files of your data?
echo Do you remember tot.number of Events? Enter!
read nEvts 
echo "Number of Events: $nEvts!"

echo Do you remember number of Jobs? Enter!
read nJobs
echo "Number of Jobs: $nJobs!"

step=`expr $nEvts / $nJobs`

# echo And, yes, enter storage-folder:
# read folder
# echo "Storagepath: $folder!"

echo Enter momentum from "1.5, 4.06, 8.9, 11.91, 15 [GeV/c]":
read mom
echo "Beam mometum is: $mom!"

momStr=`echo $mom | sed -e '{ s/\./_/g; }'`
#dMom=(-1e-1 -1e-2 -1e-3 -1e-4 -1e-5 -1e-6 -1e-7 0 1e-7 1e-6 1e-5 1e-4 1e-3 1e-2 1e-1)
#dMom=(-1e-1 -1e-2 -1e-3 -1e-4 -1e-5 -1e-6 -1e-10 0)
dMom=(-1e-1 -1e-2 -1e-3 -1e-4 -1e-6 -1e-10 0)
#dMom=(1e-7 1e-6 1e-5 1e-4 1e-3 1e-2 1e-1 0)
#dMom=(0)
dMomN=${#dMom[@]}
#for dP in -1e-1 
for (( jp=0;jp<$dMomN;jp++)); do
    dP=${dMom[${jp}]}
    path="/home/karavdin/datastorage/BOXoutputJune2014_NewGeo5/all_"${momStr}"_dM_"${dP}"/PrecIsPbeam_allCutsON/"
# make data directory, if not there. all data file go there, makes them easier to move around

    if [ ! -d $path ]; then 
 # mkdir -p $path
	echo Ooops, seems like path does not exist yet!
    fi
#echo "Output files will be saved in $path"
### CONTINUE EDIT HERE !!!!!!!!

 #if [ ! -d jobs/all_${momStr}/${folder}/ ]; then 
#    mkdir -p jobs/all_${momStr}/${folder}/
# fi

# let's go

    last=`expr $nEvts - 1`
    source /home/karavdin/pandaroot13/buildPanda/config.sh
    if [ -r  ${path}/TOT_QA_jobs.log ]; then
	rm ${path}/TOT_QA_jobs.log
    fi
    cat > ${path}/TOT_QA_jobs.log <<EOF
EOF
#	cat >${path}/sum_${nEvts}events_BOX_MC_${momStr}.txt <<EOF
#EOF
    if [ -r  ${path}/sum_${nEvts}events_BOX_TrksQ_${momStr}.txt ]; then
	rm ${path}/sum_${nEvts}events_BOX_TrksQ_${momStr}.txt 
    fi
    cat >${path}/sum_${nEvts}events_BOX_TrksQ_${momStr}.txt <<EOF
EOF

    LEVEL=5
    FIRSTTRK=$(stat -c%s ${path}/Lumi_EffTrk_0.root)
    STANDARTTRK=$((${FIRSTTRK}*${LEVEL}/10))

    for start in `seq 0 $step $last` ; do
	
	if [ ! -r  ${path}/Lumi_EffTrk_${start}.root ]; then
	    iter=0
	    while [ $iter -lt 1 ];do 
		qsub jobs_box/all_${momStr}/job_dP_${dP}_${start}.sh
		done_test=$(echo $?)
		if [ "$done_test" -eq "0" ];then
		#    echo "Output files will be saved in $path"
		    iter=1
		else
		    echo "qsub did not work sleep for 60 seconds and then try again." 
		    sleep 60
		fi
	    done

	    echo "EffTrk doesn't exist: jobs_box/all_${momStr}/job_dP_${dP}_${start}.sh" >> ${path}/TOT_QA_jobs.log 
	else
	    FILESIZETRK=$(stat -c%s ${path}/Lumi_EffTrk_${start}.root )
	    if [[ ${FILESIZETRK} -lt ${STANDARTTRK} ]]; then
		iter=0
		while [ $iter -lt 1 ];do 
		    qsub jobs_box/all_${momStr}/job_dP_${dP}_${start}.sh
		    done_test=$(echo $?)
		    if [ "$done_test" -eq "0" ];then
		#    echo "Output files will be saved in $path"
			iter=1
		    else
			echo "qsub did not work sleep for 60 seconds and then try again." 
			sleep 60
		    fi
		done
		echo "too small EffTrk: jobs_box/all_${momStr}/job_dP_${dP}_${start}.sh"  >> ${path}/TOT_QA_jobs.log 
	    else
		echo -n " ${path}/Lumi_EffTrk_${start}.root" >> ${path}/sum_${nEvts}events_BOX_TrksQ_${momStr}.txt 
	    fi
	fi
 #   fi
    done

    cat >${PWD}/jobs_box/all_${momStr}/job_dP_${dP}_SUM.sh <<EOF
#!/bin/bash
#
#PBS -N lmdSumBOX${momStr}_${dP}
#PBS -j oe
#PBS -o ${path}/LOGsumBOX${momStr}_${dP}
#PBS -V
#PBS -l nodes=1:ppn=1,walltime=02:59:00,vmem=2GB
export PATH=\$PBS_O_PATH
cd \$PBS_O_WORKDIR
echo "Start BOX sum files  with $nEvts events @ ${momStr} GeV/c with dP=${dP}"
cd /home/karavdin/pandaroot13/macro/lmd
source /home/karavdin/pandaroot13/buildPanda/config.sh
### check files don't exist yet
if [ -r ${path}/TOT_Lumi_EffTrk.root ];then
  rm ${path}/TOT_Lumi_EffTrk.root
fi
${ROOTSYS}/bin/hadd -k -f ${path}/TOT_Lumi_EffTrk.root ${filenames} `cat ${path}/sum_${nEvts}events_BOX_TrksQ_${momStr}.txt`
exit 0
EOF

### submit job to batch system
    iter=0
    while [ $iter -lt 1 ];do 
	qsub ${PWD}/jobs_box/all_${momStr}/job_dP_${dP}_SUM.sh
	done_test=$(echo $?)
	if [ "$done_test" -eq "0" ];then
	    echo "Output files will be saved in $path"
	    iter=1
	else
	    echo "qsub did not work sleep for 60 seconds and then try again." 
	    sleep 60
	fi
    done
done