import os  # python標準庫，不需要安裝，用于系統文件操作相關
import cv2  # python非標準庫，pip install opencv-python 多媒體處理
from PIL import Image  # python非標準庫，pip install pillow，圖像處理
import moviepy.editor as mov  # python非標準庫，pip install moviepy，多媒體編輯

def image_to_video(image_path, media_path):
    '''
    圖片合成視頻函數
    :param image_path: 圖片路徑
    :param media_path: 合成視頻保存路徑
    :return:
    '''
    # 獲取圖片路徑下面的所有圖片名稱
    image_names = os.listdir(image_path)
    # 對提取到的圖片名稱進行排序
    image_names.sort(key=lambda n: int(n[:-4]))
    # 設置寫入格式
    fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')
    # 設置每秒幀數
    fps = 2  # 由于圖片數目較少，這里設置的幀數比較低
    # 讀取第一個圖片獲取大小尺寸，因為需要轉換成視頻的圖片大小尺寸是一樣的
    image = Image.open(image_path + image_names[0])
    # 初始化媒體寫入對象
    media_writer = cv2.VideoWriter(media_path, fourcc, fps, image.size)
    # 遍歷圖片，將每張圖片加入視頻當中
    for image_name in image_names:
        im = cv2.imread(os.path.join(image_path, image_name))
        media_writer.write(im)
        print(image_name, '合并完成！')
    # 釋放媒體寫入對象
    media_writer.release()
    print('無聲視頻寫入完成！')

def set_music():
    '''
    合成視頻設置背景音樂函數
    :return:
    '''
    print('開始添加背景音樂！')
    # 初始化視頻文件對象
    clip = mov.VideoFileClip('./media.mp4')
    # 從某個視頻中提取一段背景音樂
    audio = mov.AudioFileClip('./source.mp4').subclip(0, 83)
    # 將背景音樂寫入.mp3文件
    audio.write_audiofile('./background.mp3')
    # 向合成好的無聲視頻中添加背景音樂
    clip = clip.set_audio(audio)
    # 保存視頻
    clip.write_videofile('./media.mp4')
    print('背景音樂添加完成！')


# image_to_video('./images/', './media.mp4')
set_music()