#!/bin/bash

source ./test_tipc/common_func.sh

FILENAME=$1
# MODE be one of ['lite_train_lite_infer']
MODE=$2

dataline=$(cat ${FILENAME})

# parser params
IFS=$'\n'
lines=(${dataline})

# The training params
model_name=$(func_parser_value "${lines[1]}")

trainer_list=$(func_parser_value "${lines[14]}")

# MODE be one of ['lite_train_lite_infer']
if [ ${MODE} = "lite_train_lite_infer" ]; then
    if [ ${model_name} = "UNETR" ]; then
        mkdir -p ./test_tipc/data
        rm -rf ./test_tipc/data/mini_levir_dataset
        cd ./test_tipc/data/
        wget https://bj.bcebos.com/paddleseg/dataset/mini_brainT_dataset.zip
        unzip mini_brainT_dataset.zip  && cd ../../
    elif [ ${model_name} = "TransUNet" ] || [ ${model_name} = "SwinUNet" ]; then
        mkdir -p ./test_tipc/data
        rm -rf ./test_tipc/data/mini_synapse_dataset
        cd ./test_tipc/data/
        wget https://paddleseg.bj.bcebos.com/dataset/mini_synapse_dataset.zip
        unzip mini_synapse_dataset.zip  && cd ../../
    elif [ "${model_name}" = "nnFormer" ]; then
        mkdir -p ./test_tipc/data
        rm -rf ./test_tipc/data/mini_acdc
        cd ./test_tipc/data/
        wget https://paddleseg.bj.bcebos.com/dataset/mini_acdc.zip
        unzip mini_acdc.zip  && cd ../../
    elif [ "${model_name}" = "VNet" ]; then
        mkdir -p ./test_tipc/data
        rm -rf ./test_tipc/data/mini_lung_coronavirus
        cd ./test_tipc/data/
        wget https://bj.bcebos.com/paddleseg/dataset/mini_lung_coronavirus.zip
        unzip mini_lung_coronavirus.zip  && cd ../../
    else 
        echo "Not added into TIPC yet."
    fi
fi