JGraph2D.sh

Go to the documentation of this file.

#!/usr/bin/env 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                                 \