#!usr/bin/env python
#encoding:utf-8
'''
__Author__:沂水寒城
功能： Python opencv調(diào)用攝像頭獲取個(gè)人圖片
使用方法：
    啟動(dòng)攝像頭后需要借助鍵盤(pán)輸入操作來(lái)完成圖片的獲取工作
    c(change): 生成存儲(chǔ)目錄
    p(photo): 執(zhí)行截圖
    q(quit): 退出拍攝
OpenCV簡(jiǎn)介
1、videoCapture()方法打開(kāi)攝像
  攝像頭變量 cv2.VideoCapture(n)  n為整數(shù)內(nèi)置攝像頭為0，若有其他攝像頭則依次為1,2,3,...
  cap=cv2.VideoCapture(0) 打開(kāi)內(nèi)置攝像頭
2、cap.isOpened()方法判斷攝像頭是否處于打開(kāi)，返回結(jié)果為：True、False
3、ret,img=cap.read() 讀取圖像
  布爾變量,圖像變量=cap.read() 
4、cap.release() 釋放攝像頭
5、action=cv2.waitKey(n) 獲取用戶輸入，同時(shí)可獲取按鍵的ASCLL碼值
'''
import os
import cv2
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
def cameraAutoForPictures(saveDir='data/'):
  '''
  調(diào)用電腦攝像頭來(lái)自動(dòng)獲取圖片
  '''
  if not os.path.exists(saveDir):
    os.makedirs(saveDir)
  count=1 #圖片計(jì)數(shù)索引
  cap=cv2.VideoCapture(0)
  width,height,w=640,480,360
  cap.set(cv2.CAP_PROP_FRAME_WIDTH,width)
  cap.set(cv2.CAP_PROP_FRAME_HEIGHT,height)
  crop_w_start=(width-w)//2
  crop_h_start=(height-w)//2
  print 'width: ',width
  print 'height: ',height
  while True:
    ret,frame=cap.read() #獲取相框
    frame=frame[crop_h_start:crop_h_start+w,crop_w_start:crop_w_start+w] #展示相框
    frame=cv2.flip(frame,1,dst=None) #前置攝像頭獲取的畫(huà)面是非鏡面的，即左手會(huì)出現(xiàn)在畫(huà)面的右側(cè)，此處使用flip進(jìn)行水平鏡像處理
    cv2.imshow("capture", frame)
    action=cv2.waitKey(1) & 0xFF
    if action==ord('c'):
      saveDir=raw_input(u"請(qǐng)輸入新的存儲(chǔ)目錄：")
      if not os.path.exists(saveDir):
        os.makedirs(saveDir)
    elif action==ord('p'):
      cv2.imwrite("%s/%d.jpg" % (saveDir,count),cv2.resize(frame, (224, 224),interpolation=cv2.INTER_AREA))
      print(u"%s: %d 張圖片" % (saveDir,count))
      count+=1
    if action==ord('q'):
      break
  cap.release() #釋放攝像頭
  cv2.destroyAllWindows() #丟棄窗口
if __name__=='__main__':  
  cameraAutoForPictures(saveDir='data/')