#!/bin/bash
# 原图像位置
Dir_src_pics="./data/img"
Dir_result="./data/result"
Dir_ori_src_pics="/root/Dehaze/SRC_files/src"

mkdir -p $Dir_src_pics $Dir_result

# 进行绝对路径转化
Dir_src_pics=$(readlink -f "$Dir_src_pics")
Dir_result=$(readlink -f "$Dir_result")
if [ ! -d "$Dir_src_pics" ] && [ ! -d "$Dir_result" ]; then
    echo "image、label都不存在，程序退出"
    echo -e "\033[31mori_image_directory\033[0m: $Dir_src_pics"
    echo "$Dir_src_pics"
    echo -e "\033[31mori_label_directory\033[0m: $Dir_result"
    echo "$Dir_result"
    exit 1
fi

PS3='All in one choice : '
applications=("Delete_src_pics" "Delete_generate_pics" "Copy_src_pics_to_test" "Run_test_program" "quit")
select fav in "${applications[@]}"; do
    case $fav in
# 删除原始文件选项 
	  "Delete_src_pics")
        # 删除src文件
        echo "Delete all src files in $Dir_src_pics"
        rm $Dir_src_pics/*
        ;;
           
# 删除生成文件选项 
	  "Delete_generate_pics")
        # 删除result文件
        echo "Delete all src files in $Dir_result"
        rm $Dir_result/*
        ;;

# 复制待处理文件选项 
	  "Copy_src_pics_to_test")
        # 删除src文件
        echo "Copy all src files in $Dir_ori_src_pics"
        ln -s $Dir_ori_src_pics/* $Dir_src_pics
        ;;

# 运行程序 
	  "Run_test_program")
        source ~/miniconda/bin/activate Dehaze_DCP
        python ./test/test.py $Dir_src_pics $Dir_result
        ;;
            
# 退出选项     
	  "quit")
	        echo "User requested exit"
	        exit
	        ;;
	        
# 其他选项
       *) echo "invalid option $REPLY";;
    esac
done