#!/bin/bash      

cd ${VMCWORKDIR}/macro/lmd/steve

#include some helper functions
. bashFunctions.sh

cd ..

if [ $# -eq '16' ]; then
  num_evts=$1
  mom=$2
  gen_input_file_stripped=$3
  dirname=$4
  pathname=$5
  filename_index=$6

  beamX0=$7
  beamY0=$8
  targetZ0=$9
  beam_widthX=${10}
  beam_widthY=${11}
  target_widthZ=${12}
  beam_gradX=${13}
  beam_gradY=${14}
  beam_grad_sigmaX=${15}
  beam_grad_sigmaY=${16}
fi

if [ ${PBS_ARRAYID} ]; then
  filename_index=${PBS_ARRAYID}
fi

gen_input_filename="${gen_input_file_stripped}_${filename_index}.root"

if [ ! -d $pathname ]; then
  mkdir -p $pathname
fi

verbositylvl=0
start_evt=$((${num_evts}*${filename_index})) #number of events * filename index is startevt

#switch on "missing plane" search algorithm
misspl=true
#use cuts during trk seacrh with "CA". Should be 'false' if sensors missaligned!
trkcut=true
#merge hits on sensors from different sides. true=yes
mergedHits=true
#Skip kinematic filter (before back-propagation)
#SkipFilt=true
## if SkipFilt=false (XThetaCut or YPhiCut) or BoxCut should be true:
## X-Theta kinematic cut before back-propagation
#XThetaCut=true
#YPhiCut=true
## BOX cut before back-propagation
BoxCut=false
## Clean after back-propagation (momentum or MVA cut)
#CleanSig=false

echo "xthetacut: $XThetaCut"
echo "yphicut: $YPhiCut"
echo "mcut: $CleanSig"

## Write all MC info in TrkQA array
WrAllMC=true

#ok we want to simulate only on the node so also the output files of the simulation so change the pathname to /local/scratch/dirname
dirname=`echo $dirname | sed -e 's/\//_/g'`
workpathname=/local/scratch/$dirname
if [ ! -d $workpathname ]; then
  mkdir -p $workpathname
fi

#simulation
check_stage_success "${pathname_base}/Lumi_MC_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel0SimDPM.C('${num_evts}','${start_evt}','${mom}',"'${gen_input_filename}'", "'${workpathname}'",'$beamX0', '$beamY0', '${targetZ0}', '${beam_widthX}', '${beam_widthY}', '${target_widthZ}', '${beam_gradX}', '${beam_gradY}', '${beam_grad_sigmaX}', '${beam_grad_sigmaY}','$verbositylvl')' > /dev/null 2>&1
  cp $workpathname/Lumi_MC_${start_evt}.root ${pathname_base}/Lumi_MC_${start_evt}.root
  cp $workpathname/Lumi_Params_${start_evt}.root ${pathname_base}/Lumi_Params_${start_evt}.root
else
  cp ${pathname_base}/Lumi_MC_${start_evt}.root $workpathname/Lumi_MC_${start_evt}.root
  cp ${pathname_base}/Lumi_Params_${start_evt}.root $workpathname/Lumi_Params_${start_evt}.root
fi

#digitization
check_stage_success "$workpathname/Lumi_digi_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel1Digi.C('${num_evts}','${start_evt}',"'${workpathname}'",'$verbositylvl')'
fi

check_stage_success "$workpathname/Lumi_reco_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel2Reco.C('${num_evts}','${start_evt}',"'${workpathname}'",'$verbositylvl',false)'
fi

#merge hits
check_stage_success "$workpathname/Lumi_recoMerged_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel2bHitMerge.C('${num_evts}','${start_evt}',"'${workpathname}'",'$verbositylvl')'
fi

### change "CA" --> "Follow" if you want to use Trk-Following as trk-search algorithm
### NB: CA can use merged or single(not merged) hits, Trk-Following can't
check_stage_success "$workpathname/Lumi_TCand_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel3Finder.C('${num_evts}','${start_evt}',"'${workpathname}'",'$verbositylvl',"CA",'${misspl}','${mergedHits}','${trkcut}','${mom}')'
fi

#track fit:
### Possible options: "Minuit", "KalmanGeane", "KalmanRK"
check_stage_success "$workpathname/Lumi_TrackNotFiltered_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel4Fitter.C('${num_evts}','${start_evt}',"'${workpathname}'",'$verbositylvl',"Minuit",'${mergedHits}')' > /dev/null 2>&1
  #this script output a Lumi_Track_... file. Rename that to the NotFiltered..
  mv ${workpathname}/Lumi_Track_${start_evt}.root ${workpathname}/Lumi_TrackNotFiltered_${start_evt}.root
fi

#track filter (on number of hits and chi2)
check_stage_success "$workpathname/Lumi_TrackFiltered_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  check_stage_success "$workpathname/Lumi_Track_${start_evt}.root"
  if [ 0 -eq "$?" ]; then
    #this macro needs Lumi_Track_... file as input so we need to link the unfiltered file
    ln -sf ${workpathname}/Lumi_TrackNotFiltered_${start_evt}.root ${workpathname}/Lumi_Track_${start_evt}.root

    root -l -b -q 'runLumiPixel4aFilter.C('${num_evts}', '${start_evt}', "'${workpathname}'", '$verbositylvl', '${mergedHits}', '${SkipFilt}', '${XThetaCut}', '${YPhiCut}', '${BoxCut}', '${rec_beamX0}', '${rec_beamY0}')'
   
    #now overwrite the Lumi_Track_ sym link with the filtered version
    ln -sf ${workpathname}/Lumi_TrackFiltered_${start_evt}.root ${workpathname}/Lumi_Track_${start_evt}.root
  fi
fi

# back-propgation GEANE
### Possible options: "Geane", "RK"
check_stage_success "$workpathname/Lumi_Geane_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel5BackProp.C('${num_evts}', '${start_evt}', "'${workpathname}'", '$verbositylvl', "Geane", '${mergedHits}', '${mom}', '${rec_beamX0}', '${rec_beamY0}', '${rec_targetZ0}')'
fi

# filter back-propagated tracks (momentum cut)
if [ $CleanSig == "true" ]; then 
  root -l -b -q 'runLumiPixel5bCleanSig.C('${num_evts}', '${start_evt}', "'${workpathname}'", '$verbositylvl', '${mom}', '${rec_beamX0}', '${rec_beamY0}')'
fi


# # Quality assurance task(s)
# combine MC and reco information
# the last parameter is mc all write flag and needs to be true
# so that all mc events are written even if geometrically missing the sensors
# this is required for the acceptance calculation
check_stage_success "$workpathname/Lumi_TrksQA_${start_evt}.root"
if [ 0 -eq "$?" ]; then
  root -l -b -q 'runLumiPixel7TrksQA.C('${num_evts}','${start_evt}',"'${workpathname}'",'0','${mom}', '$WrAllMC', '${CleanSig}')'
  cp $workpathname/Lumi_TrksQA_${start_evt}.root $pathname/Lumi_TrksQA_${start_evt}.root
fi

#remove everything in the local path
rm -f $workpathname/Lumi_*_${start_evt}.root
if [ ! "$(ls -A $workpathname)" ]; then
  rm -rf $workpathname
fi