detector-Zmul:fit.sh

Go to the documentation of this file.

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

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

source $JPP_DIR/setenv.sh $JPP_DIR >& /dev/null

set_variable:  DEBUG       ACOUSTICS_DEBUG        2
set_variable:  WORKDIR     ACOUSTICS_WORKDIR      ./
#set_variable   FORMULA     "[0] + [1] * log(1.0 + 0.5*(x-[2])*(x-[2])/([5]*[5]) + 0.5*(y-[3])*(y-[3])/([6]*[6]) + [4]*(x-[2])*(y-[3])/([5]*[6]))"  
set_variable   FORMULA     "[0] + [1] * ((x-[2])*(x-[2])/([5]*[5]) + (y-[3])*(y-[3])/([6]*[6]) + [4]*(x-[2])*(y-[3])/([5]*[6]))"
set_variable:  FORMAT      GRAPHICS_FORMAT        gif
set_variable+  BATCH       GRAPHICS_BATCH         -B

if do_usage $*; then
    usage "$script <string identifier> [<detector file>]"
fi

if (( $# == 0 || $# > 2 )); then
    fatal "Wrong number of arguments."
fi

set_variable  STRING              $argv[1]

set_variable  DETECTOR_TXT        $WORKDIR/detectorzmul_${STRING}.txt
set_variable  DETECTOR_ROOT       $WORKDIR/detectorzmul_${STRING}.root
set_variable  G1_TXT              $WORKDIR/g1.txt
set_variable  G1_ROOT             $WORKDIR/g1.root

if [[ ! -f $DETECTOR_TXT ]] then
    fatal "No detector data file $DETECTOR_TXT."
fi

JGraph2D \
    -f $DETECTOR_TXT                          \
    -o $DETECTOR_ROOT

sort -gr -k 3,3 $DETECTOR_TXT | tail -1 | read X Y Z RMS

echo $X $Y > $G1_TXT

JGraph \
    -f $G1_TXT                                \
    -o $G1_ROOT

awk '
    BEGIN {
    XMIN = +1e3
    XMAX = -1e3
    YMIN = +1e3
    YMAX = -1e3
    }
    END { print XMIN, XMAX, YMIN, YMAX }
    $1 !~ /^#/ {
    if ( $1 < XMIN ) { XMIN = $1 }
    if ( $1 > XMAX ) { XMAX = $1 }
    if ( $2 < YMIN ) { YMIN = $2 }
    if ( $2 > YMAX ) { YMAX = $2 }
    }
    ' $DETECTOR_TXT | read XMIN XMAX YMIN YMAX

if (( $X > $XMIN && $X < $XMAX && $Y > $YMIN && $Y < $YMAX )); then

    let XMIN="$X - 1.0"
    let XMAX="$X + 1.0"
    let YMIN="$Y - 0.005"
    let YMAX="$Y + 0.005"

    JFit2D \
        -f ${DETECTOR_ROOT}:\.\*                  \
        -o $WORKDIR/fit.root                      \
        -F "$FORMULA"                             \
        -x "$XMIN $XMAX"                          \
        -y "$YMIN $YMAX"                          \
        -@ "p0 =  $Z"                             \
        -@ "p1 =  1.0"                            \
        -@ "p2 =  $X"                             \
        -@ "p3 =  $Y"                             \
        -@ "p4 =  0.0"                            \
        -@ "p5 =  1.0"                            \
        -@ "p6 =  0.001"                          \
        -R "p2 =  $XMIN $XMAX"                    \
        -R "p3 =  $YMIN $YMAX"                    \
        -O LL                                     \
        -d $DEBUG

    set_array P2 `JPrintFit -f $WORKDIR/fit.root:user -@p2`
    set_array P3 `JPrintFit -f $WORKDIR/fit.root:user -@p3`

    set_variable X  $P2[1]
    set_variable Y  $P3[1]
    
    printf "nominal %4d %7.3f %7.3f %7.3f\n" $STRING 0.0 0.0 `JPrintFit -f $WORKDIR/fit.root:user -x "0.0 0.0 "`
    printf "optimal %4d %7.3f %7.3f %7.3f\n" $STRING $X  $Y  `JPrintFit -f $WORKDIR/fit.root:user -x "$X  $Y"`

    mv $WORKDIR/fit.root $DETECTOR_ROOT

else

    echo   "Optimum value at limit; -> no fit."
    printf "nominal %4d %7.3f %7.3f %7.3f\n" $STRING 0.0 0.0 `awk -v X=0.0 -v Y=0.0 '{ if ($1 == X && $2 == Y) { print $3 }}' $DETECTOR_TXT`
    printf "optimal %4d %7.3f %7.3f %7.3f\n" $STRING $X  $Y  $Z

fi

if (( $# == 2 )); then

    set_variable  DETECTOR        $argv[2]
    set_variable  MODULE          `getModule -a $DETECTOR -L "$STRING 0"`

    JEditDetector \
        -a $DETECTOR                     \
        -s "$STRING add 0.0 0.0 $X"      \
        -s "$STRING mul $Y"              \
        -o $DETECTOR
fi

typeset -Z 4 STRING

JPlot2D \
    -f ${DETECTOR_ROOT}:\.\*                      \
    -f ${G1_ROOT}:\.\*                            \
    -\> "#Deltaz [m]"                             \
    -\< "stretching"                              \
    -\^ "#chi^{2}/NDF"                            \
    -N "X 505"                                    \
    -N "Y 505"                                    \
    -O CONT3                                      \
    -T "[$STRING]"                                \
    -o $WORKDIR/detectorzmul_${STRING}.$FORMAT    $BATCH

rm -f  $DETECTOR_ROOT $G1_TXT $G1_ROOT