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拆分成三段的一個(gè)列表，三個(gè)元素分別是：模式串前面的字符串、模式串（含百分號(hào)、數(shù)字和d)、模式串后面的字符串，如果沒有找到模式串則返回None
    """
    patternIndex = patternString.index("%")
    if patternIndex<0:
        print("模式字符串中未發(fā)現(xiàn)模式匹配符：%")
        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("模式字符串中的模式存在非數(shù)字")
                return None
    if matched: return [patternString[:patternIndex],patternString[patternIndex:patternEnd+1],patternString[patternEnd+1:]]
    else:
        print("模式字符串中的模式不是字母d結(jié)尾")
        return None

def composePPTImgsAndSounds(imgFilePattern,soundFilePattern,startSeq=1,endSeq=None):
    """
    :param imgFilePattern: 圖像文件模式，要求圖像文件中必須含文件序號(hào)，序號(hào)采用%d、%02d/%03d的方式來識(shí)別
    :param soundFilePattern:音頻文件格式，要求文件中必須含文件序號(hào)，序號(hào)采用%d、%02d/%03d的方式來識(shí)別
    :param startSeq:文件序列號(hào)其實(shí)值，一般為0或1
    :param endSeq:文件序列號(hào)結(jié)束值，可以不填
    :return:生成的視頻
    """
    if endSeq is None:endSeq=9999999  #當(dāng)對(duì)應(yīng)序號(hào)找不到文件時(shí)結(jié)束
    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"音視和圖像文件加載結(jié)束，最后一個(gè)文件的序號(hào)為{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("視頻文件已經(jīng)生成")
    return clip


composePPTImgsAndSounds(r'F:\bak\pr\大H\十百千\視頻制作\5G在線計(jì)費(fèi)動(dòng)態(tài)授權(quán)機(jī)制介紹3.9\幻燈片%d.PNG',r'C:\temp\錄音文件\p%d.m4a')