# -*- coding: utf-8 -*-

#将图片转换为模板
__author__ = '上海色宁'

import os
import sys
import cv2
import numpy as np
import uuid
import getopt


def start_check_argv_and_convert(argv):
    """
        通过 getopt模块 来识别参数demo,  http://blog.csdn.net/ouyang_peng/
    """
    input_dir = ""
    dest_dir = ""
    
    try:
        """
            options, args = getopt.getopt(args, shortopts, longopts=[])

            参数args：一般是sys.argv[1:]。过滤掉sys.argv[0]，它是执行脚本的名字，不算做命令行参数。
            参数shortopts：短格式分析串。例如："hp:i:"，h后面没有冒号，表示后面不带参数；p和i后面带有冒号，表示后面带参数。
            参数longopts：长格式分析串列表。例如：["help", "ip=", "port="]，help后面没有等号，表示后面不带参数；ip和port后面带冒号，表示后面带参数。
            返回值options是以元组为元素的列表，每个元组的形式为：(选项串, 附加参数)，如：('-i', '192.168.0.1')
            返回值args是个列表，其中的元素是那些不含'-'或'--'的参数。
        """
        opts, args = getopt.getopt(argv, "hi:d:", ["help", "input_dir=","dest_dir="])
    except getopt.GetoptError:
        print('Error: 1_convert_image_into_template.py -i <input_dir> -d <dest_dir>')
        print('   or: 1_convert_image_into_template.py --input_dir=<numbers_image> --dest_dir=<template_image>')
        sys.exit(2)

    # 处理 返回值options是以元组为元素的列表。
    if len(opts) != 2 :
        print('Error: Parameters not enough!')
        print('Error: 1_convert_image_into_template.py -i <input_dir> -d <dest_dir>')
        print('   or: 1_convert_image_into_template.py --input_dir=<numbers_image> --dest_dir=<template_image>')
        sys.exit()
    for opt, arg in opts:
        if opt in ("-h", "--h","-help","--help"):
            print('Error: 1_convert_image_into_template.py -i <input_dir> -d <dest_dir>')
            print('   or: 1_convert_image_into_template.py --input_dir=<numbers_image> --dest_dir=<template_image>')
            sys.exit()
        elif opt in ("-i", "--input_dir"):
            input_dir = arg
        elif opt in ("-d", "--dest_dir"):
            dest_dir = arg
    print('input_dir:', input_dir)
    print('dest_dir:', dest_dir)
    
    if not os.path.exists(input_dir):
        print( input_dir+ " is  not exist!!!")
        sys.exit()
    if not os.path.exists(dest_dir):
        print( dest_dir+ " is  not exist, we are going to creat it !!!")
        os.makedirs(dest_dir)

    # 打印 返回值args列表，即其中的元素是那些不含'-'或'--'的参数。
    #for i in range(0, len(args)):
    #   print('参数 %s 为：%s' % (i + 1, args[i]))
    
    convert_into_template(input_dir,dest_dir)
    

def convert_into_template(input_dir,dest_dir):
        
    # 转换图片为匹配模板
    filename = os.listdir(input_dir)    # 获取文件夹文件
    for file in filename:
        img = cv2.imread(input_dir + "/" + file)    # 读取图片
        img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)# 灰度处理
        # img_gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
        cv2.imshow('GrayThings',img_gray)
        # plt.imshow(img, cmap = 'gray', interpolation = 'bicubic')


        t,dst1 = cv2.threshold(img,255,0,cv2.THRESH_BINARY)
        HSV = cv2.cvtColor(dst1, cv2.COLOR_BGR2HSV)
        H, S, V = cv2.split(HSV)
        LowerYellow = np.array([20, 20, 20])
        UpperYellow = np.array([30, 255, 255])
        mask = cv2.inRange(HSV, LowerYellow, UpperYellow)
        YellowThings = cv2.bitwise_and(dst1, dst1, mask=mask)
        cv2.imshow('YellowThings',YellowThings)
        cv2.imwrite(dest_dir+"/t_"+file, YellowThings)
        

#程序入口
if __name__ == '__main__':

    # 这个的-i  -d  
    start_check_argv_and_convert(sys.argv[1:])