#!/usr/bin/env python

import cv2
import sys
import time
import numpy as np
import time

if __name__ == "__main__":
  print (cv2.__version__)
  print (np.__version__)
  '''
  4.1.0
  1.16.3
  '''
  #//@ 打開(kāi)攝像頭 /dev/video0
  cap_1 = cv2.VideoCapture(0)
  # cap_1.set(3, 640)
  # cap_1.set(4, 480)
  #// 啟動(dòng)和停止錄像的標(biāo)志位
  write_ok = False
  #//@設(shè)置保存視頻的寬高、幀率、格式
  sz = (int(cap_1.get(cv2.CAP_PROP_FRAME_WIDTH)),
        int(cap_1.get(cv2.CAP_PROP_FRAME_HEIGHT)))
  fps = 30
  fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
  # fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', '2')
  # print ("###",fourcc,"###")
  #//@打開(kāi)視頻文件
  vout_1 = cv2.VideoWriter()
  vout_1.open('./output.mp4', fourcc, fps, sz, True)
  while (True):
    #//@ 保存
    if (write_ok):
      print("saving video...")
      (ret_1, frame_1) = cap_1.read()
      vout_1.write(frame_1)
    #//@ 不保存
    else:
      print("pass...")
      (ret_1, frame_1) = cap_1.read()
    #//@顯示
    cv2.imshow("cam_1", frame_1)
    #//@接收鍵盤(pán)指令
    key_input = cv2.waitKey(1) & 0xFF
    #// 當(dāng)輸入w時(shí)切換錄像/停止
    if key_input == ord("w") :
      write_ok = write_ok is not True
    #// 當(dāng)輸入為q時(shí)退出程序
    if key_input == ord("q"):
      vout_1.release()
      print("Will Stop!")
      sys.exit()