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 pointsimplex positionfit energyprefit directionprefit 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
    pointsimplex      1 
    positionfit       1
    energyprefit      1 
    directionprefit   1
    completefit       1)

# Fit application identifier

typeset -A ENUMERATION

ENUMERATION=(
    prefit            101
    pointsimplex      106
    positionfit       102
    energyprefit      105
    directionprefit   107
    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}                  \
                -w                            \
                -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:ha     \
        -\> "angle wrt selected fermion [deg]"                \
        -\^ "number of events [a.u.]"                         \
        -s "111111"                                           \
        -o $WORKDIR/angle.gif

fi


if (( 1 )); then

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

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

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