import os
import cv2
from PIL import Image

def unlock_movie(path):
""" 將視頻轉(zhuǎn)換成圖片
    path: 視頻路徑 """
  cap = cv2.VideoCapture(path)
  suc = cap.isOpened()  # 是否成功打開
  frame_count = 0
  while suc:
      frame_count += 1
      suc, frame = cap.read()
      params = []
      params.append(2)  # params.append(1)
      cv2.imwrite('frames\\%d.jpg' % frame_count, frame, params)

  cap.release()
  print('unlock movie: ', frame_count)


def  jpg_to_video(path, fps):
  """ 將圖片合成視頻. path: 視頻路徑，fps: 幀率 """
  fourcc = cv2.VideoWriter_fourcc(*"MJPG")
  images = os.listdir('frames')#os.listdir()方法用于返回指定的文件夾包含的文件或文件夾的名字的列表
  image = Image.open('frames/' + images[0])
  vw = cv2.VideoWriter(path, fourcc, fps, image.size)

os.chdir('frames')
for i in range(len(images)):
    # Image.open(str(image)+'.jpg').convert("RGB").save(str(image)+'.jpg')
    jpgfile = str(i + 1) + '.jpg'
    try:
        new_frame = cv2.imread(jpgfile)
        vw.write(new_frame)
    except Exception as exc:
        print(jpgfile, exc)
vw.release()
print(path, 'Synthetic success!')


if __name__ == '__main__':
  PATH_TO_MOVIES = os.path.join('test_movies', 'beautiful_mind2.mp4')
  PATH_TO_OUTCOME = os.path.join('detection_movies', 'beautiful_mind2_detection_1.avi')
  unlock_movie(PATH_TO_MOVIES)  # 視頻轉(zhuǎn)圖片
  jpg_to_video(PATH_TO_OUTCOME, 24)  # 圖片轉(zhuǎn)視頻