#!/bin/bash

set -e
#set -x

usage() {
    cat >&2 <<EOF
Usage: $0 [OPTION] <parameters file>
Options:
    -h  print this help
    -n  dry-run mode (for debugging)
    -c  (continue) do not overwrite previous results
    -z  compress output files with gzip
EOF
    exit $1
}

log() {
    echo "-#- $@" >&2
}

die() {
    echo "ERROR: $@" >&2
    exit 2
}

variable_check() {
    eval test -n "\${$1}" || die "undefined $1"
}

array_check() {
    eval test "\${#$1}" -gt 0 || die "undefined $1"
}

# read args
overwrite=1
debug=0
compress=0
while getopts "chnz" c; do
    case "$c" in
        'c') overwrite=0 ;;
        'h') usage 0 ;;
        'n') debug=1 ;;
        'z') compress=1 ;;
        '?') usage 1 ;;
    esac
done
shift $((OPTIND - 1))
[ $# -eq 1 ] || usage 1
parameters="$1"

log "Running: $0 $@"
log "Hostname: $(hostname -f)"

declare -a TOPOLOGIES ALGORITHMS PLATFORMS VARIANTS COMMON_OPTS MORE_ARGS

# read parameters
log "Reading parameters from \"$1\"."
source "$parameters" || die "cannot read parameters file: \"$parameters\""

array_check TOPOLOGIES
array_check ALGORITHMS
array_check PLATFORMS
variable_check NHOSTS
variable_check LOAD
variable_check DEADLINE

# default values
test -n "$VARIANTS" || VARIANTS=( "plain" "bookkeeping" )
test -n "$RESULTS"  || RESULTS="$PWD/results"
test -n "$LOBA"     || LOBA="$PWD/loba"

COMMON_OPTS=(
    "${MORE_ARGS[@]}"
)

test -x "$LOBA" || die "command not found: \"$LOBA\""

log "Results put in: \"$RESULTS\"."

[ $debug = 1 ] && log "Running in dry-run mode"

outfile() {
    echo "$*" | sed 's,[^ ]*/,,g;s/\.xml//;y/ /_/'
}

summary() {
    gzip -cdf "$@" \
    | sed -n '\!^\[main/INFO\] ,----\[ Results \]!,${
        /send\|recv\|wall clock\|Simulation succeeded/d;p;
      }'
}

if [ $compress = 1 ]; then
    outsuffix=".gz"
else
    outsuffix=""
fi

for plat in "${PLATFORMS[@]}"; do
    tmp=$(basename "$plat" ".xml")
    plat_output="$RESULTS/plat_$tmp"
    for topo in "${TOPOLOGIES[@]}"; do
        topo_output="$plat_output/topo_$topo"
        [ $debug = 0 ] && mkdir -p "$topo_output"
        for algo in "${ALGORITHMS[@]}"; do
            algo_output="$topo_output/algo_$algo"
            args=(
                -T"$topo"
                -a"$algo"
                -N"$NHOSTS"
                -L"$LOAD"
                -t"$DEADLINE"
                "$plat"
            )
            for variant in "${VARIANTS[@]}"; do
                cmd=( "$LOBA" "${COMMON_OPTS[@]}" )
                case "$variant" in
                    "plain") : ;;
                    "bookkeeping") cmd+=( "-b" ) ;;
                    *) die "unknown variant: \"$variant\"" ;;
                esac
                out="${algo_output}_${variant}.out"
                outf="$out$outsuffix"
                cmd+=( "${args[@]}" )
                log "Run: ${cmd[@]}"$'\n'"... &> $outf"
                if [ $overwrite = 0 -a -e "$outf" ]; then
                    log "already run !"
                    summary "$outf"
                    continue
                fi
                if [ $debug = 1 ]; then
                    log "skipped (dry-run)"
                    continue
                fi
                rm -f "$outf";
                echo "# ${cmd[@]}" > "$out"
                if "${cmd[@]}" >> "$out" 2>&1; then
                    summary "$out"
                else
                    grep -v '/INFO\]' "$out"
                fi
                if [ $compress = 1 ]; then
                    log "Compress output file."
                    gzip --best "$out"
                fi
            done
        done
    done
done