現有字幕程式

openshot_word.py

@@ -0,0 +1,111 @@
+import openshot
+import re
+from PIL import Image,ImageDraw,ImageFont
+
+def cKey(r,g,b,fuzz):
+    col=openshot.Color()
+    col.red=openshot.Keyframe(r)
+    col.green=openshot.Keyframe(g)
+    col.blue=openshot.Keyframe(b)
+    return openshot.ChromaKey(col, openshot.Keyframe(fuzz))
+
+def video_writer_init(path):
+    w = openshot.FFmpegWriter(path)
+    w.SetAudioOptions(True, "aac", 44100, 2, openshot.LAYOUT_STEREO, 3000000)
+    w.SetVideoOptions(True, "libx264", openshot.Fraction(30000, 1000), 1280, 720,
+        openshot.Fraction(1, 1), False, False, 3000000)
+    return w
+
+def video_photo_clip(video=None,layer=None, position=None, end=None
+    ,scale_x=1,scale_y=1,location_x=0,location_y=0,ck=None,audio=True):
+    clip = openshot.Clip(video)
+    clip.Layer(layer)
+    clip.Position(position)
+    clip.End(end)
+    clip.scale_x=openshot.Keyframe(scale_x)
+    clip.scale_y=openshot.Keyframe(scale_y)
+    clip.location_x=openshot.Keyframe(location_x)
+    clip.location_y=openshot.Keyframe(location_y)
+    
+    if ck!=None:
+        clip.AddEffect(ck)
+    if audio==True:
+        clip.has_audio=openshot.Keyframe(1)
+    else:
+        clip.has_audio=openshot.Keyframe(0)
+    return clip
+
+def trim_punctuation(s):
+    pat_block = u'[^\u4e00-\u9fff0-9a-zA-Z]+'
+    pattern = u'([0-9]+{0}[0-9]+)|{0}'.format(pat_block)
+    res = re.sub(pattern, lambda x: x.group(1) if x.group(1) else u" " ,s)
+    return res
+
+def txt2image(content, save_target,lang='zh',size=38):
+    unicode_text = trim_punctuation(content)
+    font = ''
+    if lang=='zh':
+        font = ImageFont.truetype(font="font/DFT_B7.ttc", size=size)
+    else :
+        font = ImageFont.truetype(font="font/arial.ttf", size=size)
+    text_width, text_height = font.getsize(unicode_text)
+    canvas = Image.new('RGBA', (700, 500), (255, 255, 255, 0) )
+    draw = ImageDraw.Draw(canvas)
+    text= unicode_text
+    draw.text((5,5), text,'black', font)
+    canvas.save(save_target, "PNG")
+
+def text_to_short_vedio(bg,title_bg,text,title,time):
+    t = openshot.Timeline(1280, 720, openshot.Fraction(30000, 1000), 44100, 2, openshot.LAYOUT_STEREO)
+    t.Open()
+    
+    # 去背參數
+    ck=cKey(0,254,0,270)
+    ck_anchor=cKey(0,255,0,320)
+
+    anchor = openshot.QtImageReader(bg)
+    anchor.Open()
+    anchor_clip = video_photo_clip(video=anchor,layer=2,scale_x=1,scale_y=1,
+            location_x=0,location_y=0,position=0, end=5,ck=ck_anchor,audio=True)
+    t.AddClip(anchor_clip)
+    anchor.Close()
+
+    anchor3 = openshot.FFmpegReader(title_bg)
+    anchor3.Open()
+    anchor_clip3 = video_photo_clip(video=anchor3,layer=3,scale_x=0.34,scale_y=0.34,
+            location_x=0.33,location_y=-0.34,position=0, end=5,ck=ck_anchor,audio=True)
+    t.AddClip(anchor_clip3)
+    anchor3.Close()
+
+    txt2image(text, "tmp/save_target.png",lang='zh',size = 26)
+    text_anchor = openshot.QtImageReader("tmp/save_target.png")
+    text_anchor.Open()
+    text_anchor_clip = video_photo_clip(video=text_anchor,layer=4,scale_x=1,scale_y=1,
+            location_x=0.27,location_y=0.84,position=0, end=5,ck=ck_anchor,audio=True)
+    t.AddClip(text_anchor_clip)
+    text_anchor.Close()
+
+    txt2image(title, "tmp/save_title.png",lang='zh')
+    text_title = openshot.QtImageReader("tmp/save_title.png")
+    text_title.Open()
+    text_title_clip = video_photo_clip(video=text_title,layer=5,scale_x=0.5,scale_y=0.5,
+            location_x=0.5,location_y=-0.145,position=0, end=5,ck=ck_anchor,audio=True)
+    t.AddClip(text_title_clip)
+    text_title.Close()
+
+
+    w = video_writer_init("output/test2.mp4")
+    w.Open()
+
+    frames = int(t.info.fps)*int(time)
+    for n in range(frames):
+        # tmp = n%(int(t.info.fps)*3) +int(t.info.fps)*int(2)
+        f=t.GetFrame(n)
+        w.WriteFrame(f)
+
+    t.Close()
+    w.Close()
+
+
+if __name__ == '__main__':
+    text_to_short_vedio("input/bg/客家_BG.png","input/title/右上角桐花.MOV","長度測試長度測試長度測試","標題測試",3)