JARCAEnergyCorrection.sh

Go to the documentation of this file.

#!/bin/zsh
#
# \author mdejong
#
version=1.0
script=${0##*/}
 
# ------------------------------------------------------------------------------------------
#
#    Utility script to determine muon energy correction base on Monte Carlo simulations.
#
# ------------------------------------------------------------------------------------------
 
if [ -z $JPP_DIR ]; then
    echo "Variable JPP_DIR undefined."
    exit
fi
 
source $JPP_DIR/setenv.sh $JPP_DIR
 
zmodload zsh/mathfunc
 
set_variable   DEBUG              2
set_variable   WORKDIR            ${TMPDIR:-/tmp}/
set_variable   NUMBER_OF_RUNS     10
set_variable   NUMBER_OF_BINS     40
set_variable   RANGE_LOGE         3.0 6.5
set_variable:  FORMAT             GRAPHICS_FORMAT        gif
set_variable+  BATCH              GRAPHICS_BATCH         -B
 
if do_usage $*; then
    usage "$script <working directory> <option>"\
          "\nPossible options: run fit interpolate"
fi
 
if (( $# != 2 )); then
    fatal "Wrong number of arguments."
fi
 
set_variable   WORKDIR            $argv[1]
set_variable   OPTION             $argv[2]
 
if [[ $OPTION != "run" && $OPTION != "fit" && $OPTION != "interpolate" ]]; then
    fatal "Invalid option $OPTION."
fi
 
mkdir -p  $WORKDIR
 
if [[ $OPTION == "run" ]]; then
 
    set_variable  DETECTOR     $WORKDIR/detector.detx
 
    if [[ ! -f $DETECTOR ]]; then
        JDetector.sh $DETECTOR
    fi
 
    for (( i = 1; $i <= $NUMBER_OF_RUNS; ++i )); do
        JSirene.sh                 $DETECTOR  $JPP_DATA/genhen.km3net_wpd_V2_0.evt.gz  $WORKDIR/sirene_${i}.root
        JTriggerEfficiency.sh      $DETECTOR  $DETECTOR  $WORKDIR/sirene_${i}.root     $WORKDIR/trigger_efficiency_${i}.root
        JARCAMuonReconstruction.sh $DETECTOR  $WORKDIR/trigger_efficiency_${i}.root    $WORKDIR/%_${i}.root                   $JPP_DATA/PMT_parameters.txt
    done
fi
 
if [[ $OPTION == "fit" || $OPTION == "interpolate" ]]; then
 
    if ( ! reuse_file $WORKDIR/energy_correction.root ); then
 
        set_array  INPUT_FILES  `ls $WORKDIR/jchain.arca_*.root`
 
        if (( ${#INPUT_FILES} == 0 )); then
            fatal "No input file in directory $WORKDIR"
        fi
 
    else
 
        set_array  INPUT_FILES  `ls $WORKDIR/energy_correction.root`
    fi
 
    if   [[ $OPTION == "fit" ]]; then
 
        JEnergyCorrection \
            -f "$INPUT_FILES[*]"                      \
            -o $WORKDIR/energy_correction.root        \
            -F "[0] + [1]*x + [2]*x*x + [3]*x*x*x"    \
            -@ "p0 = 0.0"                             \
            -@ "p1 = 1.0"                             \
            -@ "p2 = 0.0"                             \
            -@ "p3 = 0.0"                             \
            -N $NUMBER_OF_BINS                        \
            -x "$RANGE_LOGE"                          \
            -d $DEBUG --!
 
    elif [[ $OPTION == "interpolate" ]]; then
 
        JEnergyCorrection \
            -f "$INPUT_FILES[*]"                      \
            -o $WORKDIR/energy_correction.root        \
            -N $NUMBER_OF_BINS                        \
            -x "$RANGE_LOGE"                          \
            -d $DEBUG --!
    fi
 
    JPlot2D \
        -f  $WORKDIR/energy_correction.root:h2    \
        -> "E_{fit} [GeV]"                       \
        -< "E_{true} [GeV]"                      \
        -XX                                       \
        -YY                                       \
        -Z                                        \
        -O COLZ                                   \
        -T ""                                     \
        -o ExE.$FORMAT                            $BATCH
 
    set_variable  YMIN    0
    set_variable  YMAX    `JPrintResult -f $WORKDIR/energy_correction.root:h1 -F "GetMaximum + 0.5"`
 
    JPlot1D \
        -f  $WORKDIR/energy_correction.root:h1    \
        -y "$YMIN $YMAX"                          \
        -> "log(E_{fit}/GeV)"                    \
        -\^ "log(E_{true}/GeV)"                   \
        -T ""                                     \
        -o E.$FORMAT                              $BATCH
 
    echo result stored in $WORKDIR/energy_correction.root
fi