copyneb.sh

#!/usr/bin/env bash

set -e

NIMAGES=8

WD=`pwd`

while [ $# -gt 0 ]
do
    case "$1" in 
        (-nimages) NIMAGES="$2"; shift;;
        (-*) echo "$0: error - unrecognized option $1" 1>&2; exit 1;;
        (*) break;;
    esac
    shift
done

JOBID=$1

OUTPUT=`squeue -j $JOBID -o %R -h`
if [ "${OUTPUT:0:4}" != "aci-" ]
then
    echo $OUTPUT 1>&2
    exit 1
fi

NODES=${OUTPUT:4}

if [ "${NODES:0:1}" == "[" ]
then
    LEN=`expr length $NODES`
    NODES=${NODES:1:$(($LEN - 2))}
    IFS=','
    read -ra NODES <<< "$NODES"
    NNODES=0
    NODELIST=()
    for NODE in "${NODES[@]}"
    do
        SUBLEN=`expr length $NODE`
        if [ "$SUBLEN" -eq "7" ]
        then
            for SUBNODE in `seq -s ',' ${NODE:0:3} ${NODE:4:3}`
            do
                NODELIST+=($SUBNODE)
                NNODES=$(($NNODES + 1))
            done
        elif [ "$SUBLEN" -eq "3" ]
        then
            NODELIST+=($NODE)
            NNODES=$(($NNODES + 1))
        else
            echo "Error parsing output!" 1>&2
            echo $OUTPUT 1>&2
            exit 1
        fi
    done
else
    NNODES=1
    NODELIST=($NODES)
fi

if [ $NNODES -eq $NIMAGES ]
then
    for IMAGE in `seq -w -s ',' 01 $NIMAGES`
    do
        ssh aci-${NODELIST[$((${IMAGE#0} - 1))]} rsync -ravz /scratch/local/ehermes/$JOBID/$IMAGE $WD
    done
else
    echo "NNODES != NIMAGES not yet implemented" 1>&2
    exit 1
fi