#!/bin/bash

set -e

scripts="$(dirname "$(realpath  --canonicalize-existing ${0})")"

pyprog=~/open-eprem/eprem-analysis/plot-multipanel.py

layout="${1}"

if [ -z $layout ]; then
    python ${pyprog} --help
    echo
    echo "Layout argument is required for ${0}"
    exit 1
fi

args=""
if [ $layout == "groups" ]; then
    args+=" --positions ${scripts}/multipanel-positions.json"
    args+=" --hide-x 0 1 2 6 7 12 --hide-y 2 5 7 10 11 15"
    args+=" --label-x 8 --label-y 13"
    args+=" --x-fontsize 16 --y-fontsize 16"
else
    args+=" --layout $layout"
fi

if [ $layout == "circle" -o $layout == "groups" ]; then
    args+=" --draw-sun"
fi

if [ $layout == "rectangle" ]; then
    figstem="flux-in-rows"
elif [ $layout == "circle" ]; then
    figstem="flux-in-circle"
elif [ $layout == "sectors" ]; then
    figstem="flux-in-sectors"
elif [ $layout == "groups" ]; then
    figstem="flux-in-groups"
else
    figstem="flux-multipanel"
fi

python ${pyprog} \
        --verbose \
        --quantity flux \
        --unit "1/(cm^2 s sr MeV/nuc)" \
        --ylim 1e-5 1e3 --log \
        --energy 0.5 1 5 10 50 100 \
        --plot-shock-time \
        --subplot-radii 0.2 0.4 \
        --dpi 200.0 \
        --output images/${figstem}.png \
        $args

