#!/bin/bash

set -eu

pyprog=~/open-eprem/eprem-analysis/video3D.py

savedir=images
common="
    --fps 3.0 \
    --config eprem.cfg \
    --time-step 100 \
    --elevation 45 --azimuth 0 \
    --on-error abort \
    --stream-row 2 3 4 --stream-column 2 3 4 --stream-face 2 3 4 5 \
    --data-lims -3 +3 \
"

all=(
    Vr
    Bphi
    B
    rho
    flux
)
quantities=${@:-${all[@]}}
for quantity in ${quantities[@]}; do
    echo "Creating $quantity.gif"

    if [ $quantity == "Vr" ]; then
        python ${pyprog} \
            --verbose \
            --quantity $quantity \
            --min 300 --max 1200 --unit km/s \
            --save-path ${savedir}/$quantity.gif \
            --save-pngs ${savedir}/$quantity-pngs \
            $common
    fi

    if [ $quantity == "rho" ]; then
        python ${pyprog} \
            --verbose \
            --quantity $quantity \
            --min 5 --max 20 --unit "cm^-3" \
            --save-path ${savedir}/$quantity.gif \
            --save-pngs ${savedir}/$quantity-pngs \
            $common
    fi

    if [ $quantity == "Bphi" ]; then
        python ${pyprog} \
            --verbose \
            --quantity $quantity \
            --min -20 --max 0 --unit nT \
            --save-path ${savedir}/$quantity.gif \
            --save-pngs ${savedir}/$quantity-pngs \
            $common
    fi

    if [ $quantity == "B" ]; then
        python ${pyprog} \
            --verbose \
            --quantity $quantity \
            --min 0 --max 20 --unit nT \
            --save-path ${savedir}/$quantity.gif \
            --save-pngs ${savedir}/$quantity-pngs \
            $common
    fi

    if [ $quantity == "flux" ]; then
        for energy in 1 10 100; do
            echo "[$energy MeV]"
            python ${pyprog} \
                --verbose \
                --quantity $quantity \
                --min 1e-7 --max 1e3 --log --unit "1/(cm^2 s sr MeV/nuc)" \
                --save-path ${savedir}/$quantity-${energy}MeV.gif \
                --save-pngs ${savedir}/$quantity-${energy}MeV-pngs \
                --energy $energy MeV \
                $common
        done
    fi
    echo 
done

