torch

#!/usr/bin/env bash
POSITIONAL=()
set -e

function usage() {
    #TODO : rewrite Usage
    printf 'Usage: torch [-b/--bind dirname] [-s/--shell] [-p/--python filename] [-c commands] \n'
}

singularity="/usr/local/bin/singularity"
torch_img="/opt/singularity/torch-jskim.sif"
cmd="help"
bpath="run"
pyexec="python --version"
testexec="pytest"
cmexec="ls /mnt"

while [[ $# -gt 0 ]]
do
    cmd="$1"
    case $cmd in
        -b|--bind)
            bpath="$2"
            shift
            shift
            ;;
        -c|--command)
            cmd="cmd" 
            shift
            cmexec="$*"
            break
            ;;
        -p|--python)
            cmd="python"
            pyexec="python $2"
            shift
            shift
            break
            ;;
        -s|--shell)
            cmd="shell" 
            shift
            break
            ;;
        -t|--test)
            cmd="test"
            testexec="pytest --pyargs $2"
            shift
            break
            ;;
        -h|--help)
            cmd="help"
            shift
            break
            ;;
        *)    # unknown option
            POSITIONAL+=("$1") # save it in an array for later
            shift # past argument
            ;;
    esac
done

set -- "${POSITIONAL[@]}" # restore positional parameters


# create directory for mount
mkdir -p /data/appleparan/"${bpath}"

case $cmd in
    cmd)
        ${singularity} exec --nv --bind "${HOME}"/input:/input,/data/appleparan/"${bpath}":/mnt/data --nv ${torch_img} "${cmexec}"
        ;;
    python) 
        source activate
        ${singularity} exec --nv --bind "${HOME}"/input:/input,/data/appleparan/"${bpath}":/mnt/data --nv ${torch_img} ${pyexec}
        ;;
    shell)
        ${singularity} shell --nv --bind "${HOME}"/input:/input,/data/appleparan/"${bpath}":/mnt/data --nv ${torch_img} 
        ;;
    test) 
        source activate
        ${singularity} exec --nv --bind "${HOME}"/input:/input,/data/appleparan/"${bpath}":/mnt/data --nv ${torch_img} ${testexec}
        ;;
    help)
        usage
        exit 0
        ;;
esac