#! bin/bash

if [ -z "$1" ]
  then
    echo "No first argument supplied"
    echo "Need of begin vector index"
    exit 1
fi

if [ -z "$2" ]
  then
    echo "No second argument supplied"
    echo "Need of end vector index"
    exit 1
fi

if [ -z "$3" ]
  then
    echo "No third argument supplied"
    echo "Need of model input"
    exit 1
fi

if [ -z "$4" ]
  then
    echo "No fourth argument supplied"
    echo "Need of mode file : 'svd', 'svdn', svdne"
    exit 1
fi

if [ -z "$5" ]
  then
    echo "No fifth argument supplied"
    echo "Need of metric : 'lab', 'mscn'"
    exit 1
fi

if [ -z "$6" ]
  then
    echo "No sixth argument supplied"
fi



INPUT_BEGIN=$1
INPUT_END=$2
INPUT_MODEL=$3
INPUT_MODE=$4
INPUT_METRIC=$5

zones="0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15"

echo "**Model :** ${INPUT_MODEL}"
echo "**Metric :** ${INPUT_METRIC}"
echo "**Mode :** ${INPUT_MODE}"
echo "**Vector range :** [${INPUT_BEGIN}, ${INPUT_END}]"
echo ""
echo " # | GLOBAL | NOISY | NOT NOISY"
echo "---|--------|-------|----------"

# only take maxwell scenes
for scene in {"A","D","G","H"}; do

  FILENAME="data/data_${INPUT_MODE}_${INPUT_METRIC}_B${INPUT_BEGIN}_E${INPUT_END}_scene${scene}"

  python generate/generate_data_model.py --output ${FILENAME} --interval "${INPUT_BEGIN},${INPUT_END}" --kind ${INPUT_MODE} --metric ${INPUT_METRIC} --scenes "${scene}" --zones "${zones}" --percent 1

  python prediction_scene.py --data "$FILENAME.train" --model ${INPUT_MODEL} --output "${INPUT_MODEL}_Scene${scene}_mode_${INPUT_MODE}_metric_${INPUT_METRIC}.prediction" --scene ${scene}

done
