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       2
set_variable   WORKDIR     ./
set_variable   OPTION      N
set_variable:  FORMAT      GRAPHICS_FORMAT       gif
set_variable+  BATCH       GRAPHICS_BATCH        -B
     
if do_usage $*; then
    usage "$script (input file)+ [option]]"\
          "\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
    2)  set_variable OPTION      $2;&
    1)  set_variable INPUT_FILE  $1;;
    0)  fatal "Wrong number of arguments.";;
    *)  set_variable INPUT_FILE  $argv[1,-2]
        set_variable OPTION      $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)


typeset -A LABEL
typeset -A LOG
typeset -A LOGX
typeset -A LOGY

LABEL[E]="E [GeV]"
LABEL[LINE]="E [GeV]"
LABEL[LOGE]="E [GeV]"
LABEL[N]="number of modules"

LOG[E]=-XX
LOG[LINE]=
LOG[LOGE]=-XX
LOG[N]=

LOGX[E]=-XX
LOGX[LINE]=
LOGX[LOGE]=-XX
LOGX[N]=-XX

LOGY[E]=-YY
LOGY[LINE]=
LOGY[LOGE]=-YY
LOGY[N]=-YY

# 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_FILE}              \
        -o ${OUTPUT_FILE}             \
        -N ${NUMBER_OF_FIT[${APP}]}   \
        -A ${ENUMERATION[${APP}]}     \
        -O ${OPTION}                  \
        -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[${OPTION}]" $LOG[${OPTION}]        \
        -\^ "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[${OPTION}]         \
        -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                                        \
        -\> "E_{#mu} [GeV]"   $LOGX[${OPTION}]         \
        -\< "E_{fit} [GeV]"   $LOGY[${OPTION}]         \
        -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                       \
        -\> "log10(E_{fit}/E_{true}) muon"             \
        -T "" -o $WORKDIR/er.$FORMAT                   $BATCH
fi

if (( 1 )); then

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


if (( 1 )); then

    JPlot2D \
        -f $WORKDIR/postfit\[energy\].root:Vb          \
        -OCOLZ                                         \
        -\> "R^{2} [m^{2}]"                            \
        -\< "z [m]"                                    \
        -T "" -o $WORKDIR/hb.$FORMAT                   $BATCH

fi

if (( 1 )); then

    mv $WORKDIR/postfit\[start\].root  $WORKDIR/tmp.root

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

    mv $WORKDIR/tmp.root  $WORKDIR/postfit\[start\].root

fi