#!/bin/sh

# Load some commonly used functions
. ${VMCWORKDIR}/macro/littrack/scripts/common.sh

# Define that ROOT macro was executed via shell script
export LIT_SCRIPT=yes

# Set input parameters
test_name=$1 # Test name
much_geo=$2 # MUCH geometry
much_digi=$3 # MUCH digi

# Create output directory for the files
create_output_dir events_${test_name}/

# Number of events
nevents=100

#     NMU+ NMU- NE- NE+ NPI+ NPI- NJPSIMU NJPSIE AU  URQMD UNIGEN
pars=(0    0    0   0   0    0    10       0      0  no    no)
set_simulation_parameters ${pars}

# Set default geometries
set_default_muon_geometry
# Override default geometry of the MUCH system
export LIT_MUCH_GEOM=${much_geo}
export LIT_MUCH_DIGI=${much_digi}

# Hit producers to be used in STS
export LIT_STS_HITPRODUCER_TYPE=real

# Set default names for the output files
set_default_file_names ${LIT_DIR} 0000

# Path to the UrQMD file to be used
export LIT_URQMD_FILE=${VMCWORKDIR}/input/urqmd.auau.25gev.centr.root
    
# First run MC simulation
${ROOTSYS}/bin/root -b -q -l "${VMCWORKDIR}/macro/littrack/global_sim.C(${nevents})"

# Function performs event reconstruction
# First parameter is simple or advanced hit producer in MUCH
# Second parameter is MUCH tracking type nn/branch
function run_reco() {
   # MUCH hit producer type simple/advanced 
   export LIT_MUCH_HITPRODUCER_TYPE=$1
   # Global tracking type
   export LIT_GLOBAL_TRACKING_TYPE=$2
   # Name of the output reconstruction file
   # Since it differs from the default 
   export LIT_GLOBAL_RECO_FILE=${LIT_DIR}/global.reco.${LIT_MUCH_HITPRODUCER_TYPE}.${LIT_GLOBAL_TRACKING_TYPE}.0000.root 
   # Create directory for output results
   create_result_dir ${test_name}_${LIT_MUCH_HITPRODUCER_TYPE}_${LIT_GLOBAL_TRACKING_TYPE}/
   # Run event reconstruction
   ${ROOTSYS}/bin/root -b -q -l "${VMCWORKDIR}/macro/littrack/global_reco.C(${nevents}, \"all\")"
}

# Run 4 different event reconstruction 
# simple or advanced hit producers in MUCH
# and nn or branch global tracking
run_reco simple nn
run_reco simple branch
run_reco advanced nn
run_reco advanced branch

export LIT_SCRIPT=no