#!/bin/bash
# 原图像位置
Dir_src_pics="./img/src"
Dir_Trans_Refine="./img/Trans_Refine"
Dir_Trans_Esti="./img/Trans_Esti"
Dir_result="./img/result"
Dir_ori_src_pics="/root/Dehaze/SRC_files/src_1280_720"

mkdir -p $Dir_src_pics $Dir_Trans_Refine $Dir_Trans_Esti $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")
	    # 删除dark文件
        echo "Delete all src files in $Dir_Trans_Refine"
        rm $Dir_Trans_Refine/*
        # 删除result文件
        echo "Delete all src files in $Dir_result"
        rm $Dir_result/*
        # 删除trans文件
        echo "Delete all src files in $Dir_Trans_Esti"
        rm $Dir_Trans_Esti/*
        ;;

# 复制待处理文件选项 
	  "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_DCP
        python DehazeNet.py ./img
        ;;
            
# 退出选项     
	  "quit")
	        echo "User requested exit"
	        exit
	        ;;
	        
# 其他选项
       *) echo "invalid option $REPLY";;
    esac
done