#!/bin/bash

usage() {
	echo "This script is used to recalculate the effectiveness measures for ARQMath task 2"
	echo "usage: $0 [-run] [-ldir2] [-ldir3] [type]"
  echo "  -run  directory having all the run files"
  echo "  -ldir2  latex file directory v2"
  echo "  -ldir3  latex file directory v3"
  echo "  -type  1 for arqmath-1, 2 for arqmath-2"
	}

# check whether user had supplied -h or --help . If yes display usage
if [[ $# == "--help" || $# == "-h" ]];
then
  usage
  exit 1
fi

# if less than three arguments supplied, display usage
if [  $# -le 3 ]
then
  usage
  exit 1
fi

run=$1
ldir2=$2
ldir3=$3
type=$4

echo "[Generating deduplicated results]"
if [ "$type" == "1" ]; then
    python3 de_duplicate.py -qre "./qrel_t2_a1_v2.tsv" -tsv "$ldir2" -sub "$run" -v 4 -pri "./prime_runs_v2"
    python3 de_duplicate.py -qre "./qrel_t2_a1_v3.tsv" -tsv "$ldir3" -sub "$run" -v 6 -pri "./prime_runs_v3"
else
    python3 de_duplicate.py -qre "./qrel_t2_a2_v2.tsv" -tsv "$ldir2" -sub "$run" -v 4 -pri "./prime_runs_v2"
    python3 de_duplicate.py -qre "./qrel_t2_a2_v3.tsv" -tsv "$ldir3" -sub "$run" -v 6 -pri "./prime_runs_v3"
fi

echo "[Done]"