JMuonPostfit.sh

Go to the documentation of this file.

#!/bin/zsh
#
# \author mdejong, bofearraigh
#
script=${0##*/}

# ------------------------------------------------------------------------------------------
#
#  Utility script to plot fit results of track reconstruction chain for Monte Carlo data.
#
# ------------------------------------------------------------------------------------------


if [ -z $JPP_DIR ]; then
    echo "Variable JPP_DIR undefined."
    exit
fi


source $JPP_DIR/setenv.sh $JPP_DIR


# list of applications in the processing chain

APPLICATIONS=(prefit simplex gandalf energy start)

set_variable:  DEBUG               RECONSTRUCTION_DEBUG               1
set_variable:  NUMBER_OF_EVENTS    RECONSTRUCTION_NUMBER_OF_EVENTS    $((2**31))

set_variable:  PALETTE             GRAPHICS_PALETTE                   57
set_variable:  FORMAT              GRAPHICS_FORMAT                    gif
set_variable+  BATCH               GRAPHICS_BATCH                     -B

set_variable   OPTION                                                 N
set_variable   PRIMARY                                                muon
set_variable   WORKDIR                                                ./

if do_usage $*; then
    usage "$script <input file> [option [primary [working directory]]]"\
          "\nWhere option can be N, E, LINE, LOGE."\
          "\nNote that if more than one input file is specified, all other arguments must be provided."
fi

case $# in
    4)  set_variable WORKDIR      $4;&
    3)  set_variable PRIMARY      $3;&
    2)  set_variable OPTION       $2;&
    1)  set_array    INPUT_FILES  $1;;
    0)  fatal "Wrong number of arguments";;
    *)  set_array    INPUT_FILES  ${argv[1,-4]}
        set_variable OPTION       ${argv[-3]}
        set_variable PRIMARY      ${argv[-2]}
        set_variable WORKDIR      ${argv[-1]}
esac

# Number of prefits as a function of application (0 = all)

typeset -A NUMBER_OF_FIT

NUMBER_OF_FIT=(          \
    prefit     0         \
    simplex    0         \
    gandalf    1         \
    energy     1         \
    start      1)


# Fit application identifier

typeset -A ENUMERATION

ENUMERATION=(            \
    prefit   1       \
    simplex  2       \
    gandalf  3       \
    energy   4       \
    start    5)

case $OPTION in
    E)     set_variable  LABEL   "E [GeV]";
           set_variable  LOG     -XX;
           set_variable  LOGX    -XX;
           set_variable  LOGY    -YY;
           set_variable  E_E     "log10(E_{fit}/E_{#mu})";;
    LINE)  set_variable  LABEL   "E [GeV]";
           set_variable  LOG     ;
           set_variable  LOGX    ;
           set_variable  LOGY    ;
           set_variable  E_E     "E_{fit} - E_{#mu} [GeV]";;
    LOGE)  set_variable  LABEL   "E [GeV]";
           set_variable  LOG     -XX;
           set_variable  LOGX    -XX;
           set_variable  LOGY    -YY;
           set_variable  E_E     "log10(E_{fit}/E_{#mu})";;
    N)     set_variable  LABEL   "number of modules";
           set_variable  LOG     ;
           set_variable  LOGX    -XX;
           set_variable  LOGY    -YY;
           set_variable  E_E     "log10(E_{fit}/E_{#mu})";;
    *) fatal "Invalid option $OPTION"
esac


# user interaction

for APP in $APPLICATIONS[*]; do

    set_variable  OUTPUT_FILE    ${WORKDIR}/postfit\[${APP}\].root
        
    if [[ -f $OUTPUT_FILE ]]; then
        rm -i $OUTPUT_FILE
    fi
done


# processing

for APP in $APPLICATIONS[*]; do

    set_variable  OUTPUT_FILE    $WORKDIR/postfit\[${APP}\].root

    if [[ ! -f $OUTPUT_FILE ]]; then
        
        JMuonPostfit \
            -f"${INPUT_FILES[*]}"         \
            -o ${OUTPUT_FILE}             \
            -n ${NUMBER_OF_EVENTS}        \
            -N ${NUMBER_OF_FIT[${APP}]}   \
            -A ${ENUMERATION[${APP}]}     \
            -O ${OPTION}                  \
            -p ${PRIMARY}                 \
            -d ${DEBUG} --!
    fi 
done

if (( 1 )); then

    JPlot1D \
        -f $WORKDIR/postfit\[prefit\].root:hx          \
        -f $WORKDIR/postfit\[simplex\].root:hx         \
        -f $WORKDIR/postfit\[gandalf\].root:hx         \
        -\> "angle [deg]"                              \
        -\^ "number of events [a.u.]"                  \
        -XX                                            \
        -L TR -T ""                                    \
        -o $WORKDIR/angle.$FORMAT                      $BATCH
fi

if (( 1 )); then

    QUANTILES="0.33 0.5 0.66"
    
    for APP in prefit simplex gandalf; do

        JQuantiles2D \
            -f $WORKDIR/postfit\[${APP}\].root:h2      \
            -Q "$QUANTILES"                            \
            -o $WORKDIR/quantiles\[${APP}\].root
    done

    JPlot1D \
        -f $WORKDIR/quantiles\[prefit\].root:h21       \
        -f $WORKDIR/quantiles\[simplex\].root:h21      \
        -f $WORKDIR/quantiles\[gandalf\].root:h21      \
        -\> "$LABEL"            $LOG                   \
        -\^ "resolution (reco muon) [deg]"             \
        -y "1e-2 1e2"                                  \
        -Y                                             \
        -L TR -T ""                                    \
        -o $WORKDIR/resolution.$FORMAT                 $BATCH
fi

if (( 1 )); then

    JPlot1D \
        -f $WORKDIR/postfit\[energy\].root:E_0         \
        -f $WORKDIR/postfit\[energy\].root:E_1         \
        -f $WORKDIR/postfit\[energy\].root:E_2         \
        -\^ "number of events [a.u.]"                  \
        -\> "E_{#mu} [GeV]"   $LOGX                    \
        -T ""                                          \
        -o $WORKDIR/energy.$FORMAT                     $BATCH
fi

if (( 1 )); then

    JConvertToPDF2D \
        -f $WORKDIR/postfit\[energy\].root:ExE         \
        -O NY                                          \
        -o $WORKDIR/pdf.root

    JPlot2D \
        -f $WORKDIR/pdf.root:ExE                       \
        -O COLZ -p $PALETTE                            \
        -\> "E_{#mu} [GeV]"   $LOGX                    \
        -\< "E_{fit} [GeV]"   $LOGY                    \
        -Z                                             \
        -T ""                                          \
        -o $WORKDIR/ee.$FORMAT                         $BATCH
fi

if (( 1 )); then

    JFit \
        -f $WORKDIR/postfit\[energy\].root:E_E         \
        -o $WORKDIR/fit.root                           \
        -F "[0]*exp(-0.5*(x-[1])*(x-[1])/([2]*[2]))"   \
        -@ "p0 = 0.8*GetMaximum"                       \
        -@ "p1 = GetMean"                              \
        -@ "p2 = GetRMS"                                  

    JPlot1D \
        -f $WORKDIR/fit.root:E_E                       \
        -\> "$E_E"                                     \
        -T "" -o $WORKDIR/er.$FORMAT                   $BATCH
fi

if (( 1 )); then

    JPlot2D \
        -f $WORKDIR/postfit\[energy\].root:Va          \
        -O COLZ -p $PALETTE                            \
        -\> "R^{2} [m^{2}]"                            \
        -\< "z [m]"                                    \
        -N "X 505"                                     \
        -Z -T "" -o $WORKDIR/ha.$FORMAT                $BATCH
fi


if (( 1 )); then

    JPlot2D \
        -f $WORKDIR/postfit\[energy\].root:Vb          \
        -O COLZ -p $PALETTE                            \
        -\> "R^{2} [m^{2}]"                            \
        -\< "z [m]"                                    \
        -N "X 505"                                     \
        -Z -T "" -o $WORKDIR/hb.$FORMAT                $BATCH
fi

if (( 1 )); then

    JPlot1D \
        -f $WORKDIR/postfit\[start\].root:hz0          \
        -f $WORKDIR/postfit\[start\].root:hz1          \
        -\> "#Deltaz [m]"                              \
        -\^ "number of events [a.u.]"                  \
        -L TR -T ""                                    \
        -o $WORKDIR/z.$FORMAT                          $BATCH
fi