JGraph2D.sh

Go to the documentation of this file.

#!/bin/zsh
#
# \author mdejong
#
version=1.0
script=${0##*/}

# ------------------------------------------------------------------------------------------
#
#                         Utility script to test JGraph2D, JFit2D and JPlot2D.
#
# ------------------------------------------------------------------------------------------


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


source $JPP_DIR/setenv.sh $JPP_DIR

zmodload zsh/mathfunc

set_variable   DEBUG    2      
set_variable   WORKDIR  ${TMPDIR:-/tmp}/  

if do_usage $*; then
    usage "$script [working directory]"
fi

case $# in
    1) set_variable  WORKDIR  $1;;
esac


set_variable GRAPH_TXT   graph.txt
set_variable GRAPH_ROOT  graph.root


let "TOP   = 100.0"
let "X0    =   0.0"
let "Y0    =   0.0"
let "SIGMA =   1.0"

function f2()
{
    let "X  = $1"
    let "Y  = $2"
    
    echo $(( $TOP * exp(-0.5*(($X - $X0)/$SIGMA)**2) * exp(-0.5*(($Y - $Y0)/$SIGMA)**2) ))
}

rm -f $WORKDIR/$GRAPH_TXT >& /dev/null

for (( X = -3.0; $X <= +3.5; X += 1.0 )); do
    for (( Y = -3.0; $Y <= +3.5; Y += 1.0 )); do

        let "U = `f2 $X $Y`"
        let "V = `$JPP_DIR/examples/JMath/getPoisson -e $U`"

        printf "%7.3f %7.3f %7.3f\n"  $X  $Y  $V  >>  $WORKDIR/$GRAPH_TXT
    done
done

echo "Data points:"
cat $WORKDIR/$GRAPH_TXT

JGraph2D \
    -f $WORKDIR/$GRAPH_TXT                  \
    -o $WORKDIR/$GRAPH_ROOT              

set_variable   FORMULA  "[0] * exp(-0.5 * (x-[1])*(x-[1]) / ([2]*[2])) * exp(-0.5 * (y-[3])*(y-[3]) / ([4]*[4]))"

JFit2D \
    -f $WORKDIR/${GRAPH_ROOT}:${GRAPH_TXT}  \
    -o $WORKDIR/fit.root                    \
    -F "$FORMULA"                           \
    -@ "p0 = $TOP;"                         \
    -@ "p1 = 0.0;"                          \
    -@ "p2 = $SIGMA;"                       \
    -@ "p3 = 0.0;"                          \
    -@ "p4 = $SIGMA;"                       \
    -O LL

JPlot2D \
    -f $WORKDIR/fit.root:${GRAPH_TXT}       \
    -O COLZ                                 \