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