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

mkdir -p $Dir_src_pics $Dir_result

PS3='All in one choice : '
applications=("Delete_src_pics" "Delete_generate_pics" "Copy_src_pics" "Run_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")
        # 删除src文件
        echo "Copy all src files in $Dir_ori_src_pics"
        ln -s $Dir_ori_src_pics/* $Dir_src_pics
        ;;

# 运行程序 
	  "Run_program")
        source ~/miniconda/bin/activate Dehaze_GCANet
        python test.py --task dehaze --gpu_id 0 --indir ./imgs/src --outdir ./imgs/result
        ;;
            
# 退出选项     
	  "quit")
	        echo "User requested exit"
	        exit
	        ;;
	        
# 其他选项
       *) echo "invalid option $REPLY";;
    esac
done