from  moviepy.editor import *
import os,traceback

def isdigit(c):
    return c>='0' and c='9'

def getPattern(patternString):
    """
    :param patternString: 含有模式串的字符串，模式串仿C語言整型輸出格式，類似于：%d、%02d等
    :return: 返回將patternString拆分成三段的一個列表，三個元素分別是：模式串前面的字符串、模式串（含百分號、數字和d)、模式串后面的字符串，如果沒有找到模式串則返回None
    """
    patternIndex = patternString.index("%")
    if patternIndex0:
        print("模式字符串中未發現模式匹配符：%")
        return None
    matched = False
    for patternEnd in range(patternIndex+1,len(patternString)):
        if patternString[patternEnd]in'dD':
            matched = True
            break
        else:
            if  not isdigit(patternString[patternEnd]):
                print("模式字符串中的模式存在非數字")
                return None
    if matched: return [patternString[:patternIndex],patternString[patternIndex:patternEnd+1],patternString[patternEnd+1:]]
    else:
        print("模式字符串中的模式不是字母d結尾")
        return None

def composePPTImgsAndSounds(imgFilePattern,soundFilePattern,startSeq=1,endSeq=None):
    """
    :param imgFilePattern: 圖像文件模式，要求圖像文件中必須含文件序號，序號采用%d、%02d/%03d的方式來識別
    :param soundFilePattern:音頻文件格式，要求文件中必須含文件序號，序號采用%d、%02d/%03d的方式來識別
    :param startSeq:文件序列號其實值，一般為0或1
    :param endSeq:文件序列號結束值，可以不填
    :return:生成的視頻
    """
    if endSeq is None:endSeq=9999999  #當對應序號找不到文件時結束
    patterns = getPattern(imgFilePattern)
    if patterns is None:
        print('圖像文件名不符合模式格式要求')
        return None
    else:
        imgPrefix,imgPattern,imgPostfix = patterns
    patterns = getPattern(soundFilePattern)
    if patterns is None:
        print('音頻文件名不符合模式格式要求')
        return None
    else:
        soundPrefix, soundPattern, soundPostfix = patterns
    clips = []
    try:
        for seq in range(startSeq,endSeq):
            imgFName = f"{imgPrefix}{seq:{imgPattern[1:]}}{imgPostfix}"
            soundFName = f"{soundPrefix}{seq:{soundPattern[1:]}}{soundPostfix}"

            if os.path.exists(soundFName) and os.path.exists(imgFName):
                soundClip = AudioFileClip(soundFName)
                imgClip = ImageClip(imgFName).set_duration(soundClip.duration).set_fps(2).set_audio(soundClip)
                clips.append(imgClip)
            else:
                if seq:
                    print(f"音視和圖像文件加載結束，最后一個文件的序號為{seq}")
                    break
                else:
                    print(f"圖像文件{imgClip}或音頻文件{soundClip}不存在")
                    return None

    except Exception as e:
        print(f"加載音頻和圖像文件失敗：\n{e}\n{traceback.format_exc()}")
        return None

    print("輸出視頻文件....")
    clip = concatenate_videoclips(clips,'compose')
    clip.write_videofile(r"C:\temp\ppt.mp4",threads=8)
    print("視頻文件已經生成")
    return clip


composePPTImgsAndSounds(r'F:\bak\pr\大H\十百千\視頻制作\5G在線計費動態授權機制介紹3.9\幻燈片%d.PNG',r'C:\temp\錄音文件\p%d.m4a')