#!/bin/sh

# Common part
MAX_PPN=60
MAX_CPU_SHARING_FACTOR=4
MAX_CUDA_SHARING_FACTOR=16

# Default
NODE_COUNT=1
MAX_NODES_PER_TASK=1
INTERACTIVE=1
HAS_RES_MANAGER=0

# Specializations
if [ `hostname` = "k100" ]; then
    NODE_COUNT=64
    MAX_NODES_PER_TASK=8
    INTERACTIVE=0
    # Since launch isn't interactive - one must provide is_launched, is_finished, get_elapsed_time, stdout_fn, stderr_fn calls
    get_task_dir() {
        local n
        for n in 1 2 3 4 5 6 7 8 9; do
            if [ -d "$1.$n" ]; then
                printf %s "$1.$n"
                return
            fi
        done
        printf %s "$1"
    }
    is_launched() {
        local STDOUT_FN
        STDOUT_FN="$1"
        local STDERR_FN
        STDERR_FN="$2"
        # Add handling for refuses from SUPPZ
        echo 1
    }
    is_finished() {
        if [ "$(tail -n 1 $(get_task_dir "$1")/manager.log)" = "Exiting..." ]; then
            echo 1
        else
            echo 0
        fi
    }
    get_elapsed_time() {
        local da
        local mo
        local ye
        local dat
        local tim
        local sec1
        local sec2
        local task_dir
        task_dir="$(get_task_dir "$1")"
        dat=`grep "started at" <"$task_dir/manager.log" | awk '{print $5}' | sed 's/\./ /g'`
        tim=`grep "started at" <"$task_dir/manager.log" | awk '{print $6}'`
        da=`echo "$dat" | awk '{print $1}'`
        mo=`echo "$dat" | awk '{print $2}'`
        ye=`echo "$dat" | awk '{print $3}'`
        dat="$ye-$mo-$da $tim"
        sec1=`date -d "$dat" +%s`
        dat=`grep "done at" <"$task_dir/manager.log" | awk '{print $6}' | sed 's/\./ /g'`
        tim=`grep "done at" <"$task_dir/manager.log" | awk '{print $7}'`
        da=`echo "$dat" | awk '{print $1}'`
        mo=`echo "$dat" | awk '{print $2}'`
        ye=`echo "$dat" | awk '{print $3}'`
        dat="$ye-$mo-$da $tim"
        sec2=`date -d "$dat" +%s`
        echo $(( sec2 - sec1 ))
    }
    stdout_fn() {
        echo "$(get_task_dir "$1")/output"
    }
    stderr_fn() {
        echo "$(get_task_dir "$1")/errors"
    }
    HAS_RES_MANAGER=1
    # Since machine has resource manager (task queue) - one must provide can_launch call
    can_launch() {
        if [ `mps 2>/dev/null | tail -n +3 | wc -l` -lt 6 ]; then
            echo 1
        else
            echo 0
        fi
    }
fi