JShowerPostFit.sh

Go to the documentation of this file.

#!/bin/zsh
#
# \author adomi
#
script=${0##*/}

# ------------------------------------------------------------------------------------------
#
#                         Utility script to plot fit results.
#
# ------------------------------------------------------------------------------------------


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 positionfit completefit)

set_variable  DEBUG            2
set_variable  WORKDIR          ./
set_variable  OPTION           N
set_variable  INPUT_FILE       $WORKDIR/${APPLICATIONS[-1]}.root
set_variable  MUONS            
     
if ( do_usage $* ); then
    usage "$script [input file [working directory [muons]]]"
fi

case $# in
    3) set_variable MUONS       $3;&
    2) set_variable WORKDIR     $2;&
    1) set_variable INPUT_FILE  $1;;
esac


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

typeset -A NUMBER_OF_FIT

NUMBER_OF_FIT=(          \
    prefit     1         \
    positionfit    1     \
    completefit    1)

# Fit application identifier

typeset -A ENUMERATION

ENUMERATION=(            \
   prefit   101       \
   positionfit  102       \
   completefit  103)

# user interaction

for APP in $APPLICATIONS[*]; do

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

    if [[ -f $OUTPUT_FILE ]]; then
   rm -i $OUTPUT_FILE
    fi
done


# processing

for APP in $APPLICATIONS[*]; do

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

    if [[ ! -f $OUTPUT_FILE ]]; then

   echo "************************************************ " $MUONS 
   if [[ -z $MUONS ]]; then
       JShowerPostfit \
           -f $INPUT_FILE                \
           -o $OUTPUT_FILE               \
           -N ${NUMBER_OF_FIT[${APP}]}   \
           -A ${ENUMERATION[${APP}]}     \
           -O ${OPTION}                  \
           -d ${DEBUG} --!
   else 
       JShowerPostfit \
           -f $INPUT_FILE                \
           -o $OUTPUT_FILE               \
           -N ${NUMBER_OF_FIT[${APP}]}   \
           -A ${ENUMERATION[${APP}]}     \
           -I \
           -O ${OPTION}                  \
           -d ${DEBUG} --!
   fi      
    fi 

done


if (( 1 )); then

    JPlot1D \
   -f $WORKDIR/jshowerpostfit\[completefit\].root:hx   \
   -> "angle wrt electron [deg]"                        \
   -\^ "number of events [a.u]"             \
   -XX                                      \
   -L TR -T ""                              \
   -o $WORKDIR/angle.gif

fi


if (( 1 )); then

    QUANTILES="0.33 0.5 0.66"
    
    for APP in prefit positionfit completefit; do

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

    typeset -A LABEL
    typeset -A LOGX

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

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

    
    JPlot1D \
        -f $WORKDIR/jshowerquantiles\[completefit\].root:h21\.\*  \
        -> $LABEL[${OPTION}] $LOGX[${OPTION}]         \
        -\^ " [deg]"                                   \
        -y "1e-2 1e1"                                  \
   -Y                                             \
        -L TR -T ""                                    \
   -o $WORKDIR/jshowerresolution.gif

fi

if (( 0 )); then

    JPlot1D \
   -f $WORKDIR/jshowerpostfit\[prefit].root:he       \
   -f $WORKDIR/jshowerpostfit\[positionfi].root:he      \
   -f $WORKDIR/jshowerpostfit\[completefit].root:he      \
   -XX                                        \
   -y "0 1.1"                                 \
   -> "E [GeV]"                              \
   -\^"efficiency"                            \
   -L TR -T ""                                \
   -o $WORKDIR/jshowerefficiency.gif

fi


if (( 1 )); then

    JPlot1D \
   -f $WORKDIR/jshowerpostfit\[completefit\].root:e0      \
   -f $WORKDIR/jshowerpostfit\[completefit\].root:e1      \
   -f $WORKDIR/jshowerpostfit\[completefit\].root:e2      \
   -\^ "number of events [a.u.]"              \
   -> "log10(E_{fit}/GeV)"                   \
   -XX -T ""                                  \
   -o $WORKDIR/jshowerenergy.gif

fi

if (( 1 )); then

    JPlot2D \
   -f $WORKDIR/jshowerpostfit\[completefit\].root:ee     \
   -O COLZ                                   \
   -XX -YY -T ""                             \
   -> "E_{true} [GeV]"                      \
   -< "E_{fit} [GeV]"                       \
   -o $WORKDIR/jshoweree.gif
   
fi

if (( 1 )); then

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

    JPlot1D \
   -f $WORKDIR/fit.root:er                         \
   -> "log10(E_{fit}/E_{true})"                   \
   -T "" -o $WORKDIR/er.gif

fi


if (( 1 )); then

    JPlot2D \
   -f $WORKDIR/postfit\[start\].root:ha   \
   -OCOLZ                                 \
   -> "R^{2} [m^{2}]"                    \
   -< "z [m]"                            \
   -T "" -o $WORKDIR/ha.gif

fi


if (( 1 )); then

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

fi


if (( 1 )); then

    JPlot1D \
   -f $WORKDIR/postfit\[start\].root:hz   \
   -> "z [m]"                            \
   -\^ "number of events [a.u]"           \
   -L TR -T ""                            \
   -o $WORKDIR/z.gif

fi