import os  # python標(biāo)準(zhǔn)庫，不需要安裝，用于系統(tǒng)文件操作相關(guān)
import cv2  # python非標(biāo)準(zhǔn)庫，pip install opencv-python 多媒體處理
from PIL import Image  # python非標(biāo)準(zhǔn)庫，pip install pillow，圖像處理
import moviepy.editor as mov  # python非標(biāo)準(zhǔn)庫，pip install moviepy，多媒體編輯

def image_to_video(image_path, media_path):
    '''
    圖片合成視頻函數(shù)
    :param image_path: 圖片路徑
    :param media_path: 合成視頻保存路徑
    :return:
    '''
    # 獲取圖片路徑下面的所有圖片名稱
    image_names = os.listdir(image_path)
    # 對提取到的圖片名稱進(jìn)行排序
    image_names.sort(key=lambda n: int(n[:-4]))
    # 設(shè)置寫入格式
    fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')
    # 設(shè)置每秒幀數(shù)
    fps = 2  # 由于圖片數(shù)目較少，這里設(shè)置的幀數(shù)比較低
    # 讀取第一個(gè)圖片獲取大小尺寸，因?yàn)樾枰D(zhuǎn)換成視頻的圖片大小尺寸是一樣的
    image = Image.open(image_path + image_names[0])
    # 初始化媒體寫入對象
    media_writer = cv2.VideoWriter(media_path, fourcc, fps, image.size)
    # 遍歷圖片，將每張圖片加入視頻當(dāng)中
    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():
    '''
    合成視頻設(shè)置背景音樂函數(shù)
    :return:
    '''
    print('開始添加背景音樂！')
    # 初始化視頻文件對象
    clip = mov.VideoFileClip('./media.mp4')
    # 從某個(gè)視頻中提取一段背景音樂
    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()