#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
STEP3_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)"
REPRO_ROOT="$(cd "${STEP3_DIR}/.." && pwd)"

PYTHON_BIN="${PYTHON_BIN:-python}"
DATASET_ROOT="${DATASET_ROOT:-${REPRO_ROOT}/../dataset/OCTA-60}"
CROP_ALIGN_DIR="${CROP_ALIGN_DIR:-${DATASET_ROOT}/crop_align_images}"
OUTPUT_DIR="${OUTPUT_DIR:-${STEP3_DIR}/Output/interfaceA_retry_noinvertext}"
CONFIG_PATH="${CONFIG_PATH:-${STEP3_DIR}/Src/config/test.yaml}"
MODEL_PATH="${MODEL_PATH:-${STEP3_DIR}/Src/save/crop_vseg_vessel_111.pth}"
DEVICE="${DEVICE:-auto}"
INLIER_METHOD="${INLIER_METHOD:-RANSAC}"
RETRY_FILTERED_INLIER_THRESHOLD="${RETRY_FILTERED_INLIER_THRESHOLD:-20}"
LIMIT="${LIMIT:-}"

ARGS=(
  --dataset_root "${DATASET_ROOT}"
  --input_source crop_align
  --crop_align_dir "${CROP_ALIGN_DIR}"
  --output_dir "${OUTPUT_DIR}"
  --config_path "${CONFIG_PATH}"
  --model_path "${MODEL_PATH}"
  --device "${DEVICE}"
  --inlier_method "${INLIER_METHOD}"
  --legacy_script_crop
  --retry_filtered_inlier_threshold "${RETRY_FILTERED_INLIER_THRESHOLD}"
  --force_rerun
)

if [[ -n "${LIMIT}" ]]; then
  ARGS+=(--limit "${LIMIT}")
fi

exec "${PYTHON_BIN}" "${STEP3_DIR}/run_eval_octa60_with_retry_no_invert_ext.py" "${ARGS[@]}" "$@"