'''
Program: MT_imgWatermark.py (Report comments/bugs to chikh@yuntech.edu.tw)
Function:	
'''

import cv2						#引入OpenCV影像處理的相關套件
from datetime import datetime	#引入讀取機器時間相關套件
import threading				#引入多執行緒相關套件
import time					#for time.sleep() 

threadList, imgList = [], []
sem = threading.Semaphore(1)

#----- 設定要添加的文字：開始 -----#
position = (10,40) 	#文字左下角的座標
font = cv2.FONT_HERSHEY_COMPLEX
fontScale = 0.5		#指定文字大小的比例
color = (0,255,0) 	#以RGB三原色設定綠色文字；RGB色表可見 https://bit.ly/2ymSOJF
thickness = 1  		#文字的線寬
#----- 設定要添加的文字：結束 -----#

def camCapture(ID):
	global imgList
	cap = cv2.VideoCapture(ID) 			#選擇攝像頭設備
	i = 0
	while True:
		ret, frame = cap.read()			#從攝像頭讀取一幀影像
		cv2.imshow("Window %d"%ID,frame)	#重要!若未加入ID區分，二個鏡頭擷取的影像將重疊在同一視窗上
		if cv2.waitKey(1) & 0xFF in (27,81,113): break #每隔1毫秒偵測是否有按鍵，若有，則讀取最末位元組內容，比對是否為<Esc>、'q'或'Q'，倘符合，則退出迴圈
		i += 1
		if (i%40 == 0): 
			i = 0	#重設i，避免持續增加i值造成溢位(overflow)
			sem.acquire()
			imgList.append((ID,datetime.now().strftime("%H-%M-%S"),frame))
			cv2.imwrite('./opencvData/[%d] img-%s.jpg'%(imgList[-1][0],imgList[-1][1]),frame)	#影像輸出為檔案儲存
			sem.release()
	cap.release()	#釋放攝像頭設備

def procImage():
	global imgList
	while True:
		sem.acquire()
		if len(imgList) > 0:
			img = imgList[0][2]
			text = "Cam%d-%s"%(imgList[0][0],imgList[0][1]) 
			cv2.putText(img,text,position,font,fontScale,color,thickness)	#在影像上添加文字
			cv2.imshow('Camera%d'%imgList[0][0],img) 	#顯示影像
			cv2.imwrite('./opencvData/Cam%d-%s.jpg'%(imgList[0][0],imgList[0][1]),img)
			imgList.pop(0)
		sem.release()
		if cv2.waitKey(1) & 0xFF in (27,81,113): break
		time.sleep(0.1)

for i in range(2):
	threadList.append(threading.Thread(target=camCapture,args=(i,))) 
	threadList[i].start()

threadList.append(threading.Thread(target=procImage))
threadList[-1].start()

for i in range(3):
	threadList[i].join() 	#wait for the thread specified by threadList[i] to terminate

cv2.destroyAllWindows()	#關閉所有視窗