#!/bin/sh
set -e

# Name of this script
script=`basename "$0"`

# Handle case of manually running this script
if [ "$AUTOPKGTEST_TMP" = "" ]; then
    AUTOPKGTEST_TMP=`mktemp -d --tmpdir "$script"-XXX`;
    echo "AUTOPKGTEST_TMP is $AUTOPKGTEST_TMP";
fi
if [ "$ADT_ARTIFACTS" = "" ] ; then
    ADT_ARTIFACTS=${AUTOPKGTEST_TMP}/artifacts
fi
mkdir -p "${AUTOPKGTEST_TMP}"
mkdir -p "${ADT_ARTIFACTS}"

# Save current directory (must be top of source tree)
SRC_ROOT=`pwd`

# Setup Python example tree structure
mkdir -p ${AUTOPKGTEST_TMP}/doc
cp -a doc/examples ${AUTOPKGTEST_TMP}/doc/
mkdir -p ${AUTOPKGTEST_TMP}/python
cd ${AUTOPKGTEST_TMP}/python
echo "backend : PDF" > matplotlibrc
cp ${SRC_ROOT}/plugins/python/doc/examples/* ./
cp ${SRC_ROOT}/python/example-mpi.py ./

# Get list of Python examples
BASE_NAMES=`basename -s.xml *.xml`

# Build Gyoto command line
GYOTO="/usr/bin/gyoto --nprocesses=4 --nthreads=1 --resolution=32"

# Determine default MPI implementation for this arch
HOSTFILE="${AUTOPKGTEST_TMP}"/hostfile.${script}
eval `grep ARCH_DEFAULT_MPI_IMPL /usr/share/mpi-default-dev/debian_defaults`
if [ "x$ARCH_DEFAULT_MPI_IMPL" = x"openmpi" ]
# Setup MPI environment to run on localhost with enough slots
then export OMPI_MCA_plm_rsh_agent=/bin/false
     export OMPI_MCA_btl_tcp_if_include=lo
     HOSTFILE="${AUTOPKGTEST_TMP}"/hostfile.${script}
     echo localhost slots=5 > "${HOSTFILE}"
     MPIRUN_CMD="mpirun.openmpi -np 1 --hostfile ${HOSTFILE} --oversubscribe"
elif [ "x$ARCH_DEFAULT_MPI_IMPL" = x"mpich" ]
then echo localhost > "${HOSTFILE}"
     MPIRUN_CMD="mpirun.mpich -np 1 --hostfile ${HOSTFILE}"
else echo "unknown MPI implementation: " $ARCH_DEFAULT_MPI_IMPL
     exit 1
fi

# Run each example in this environment
for python in python3 `py3versions -s` ; do
    echo "**** Testing Python version: $python ****" 
    unset GYOTO_PLUGINS
    ${MPIRUN_CMD} $python example-mpi.py ;
    mkdir -p ${ADT_ARTIFACTS}/${python}-mpi
    export GYOTO_PLUGINS=$python,stdplug
    for base in ${BASE_NAMES} ; do
	${MPIRUN_CMD} ${GYOTO} \
		      ${base}.xml \
		      \!${ADT_ARTIFACTS}/${python}-mpi/${base}.fits ;
    done ;
done

exit 0
