q792602257/my-video-workflow
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

支持vaapi、支持定义使用哪种弹幕转换工具

Browse Source
This commit is contained in:
Jerry Yan
2022-10-10 14:35:28 +08:00
parent 7dbf9822c9
commit b74c6f0ef7
5 changed files with 98 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
config.py
View File

@ -3,12 +3,16 @@ import os.path
# [danmaku]
# use_danmu2ass
DANMAKU_USE_DANMU2ASS = False
# use_danmakufactory
DANMAKU_USE_DANMAKUFACTORY = True
# exec
DANMAKU_FACTORY_EXEC = "DanmakuFactory"
DANMAKU_EXEC = "DanmakuFactory"
# speed
DANMAKU_SPEED = 12
# font
DEFAULT_FONT_NAME = "Sarasa Term SC"
DANMAKU_FONT_NAME = "Sarasa Term SC"
# font_size
DANMAKU_FONT_SIZE = 40
# resolution
@ -22,6 +26,8 @@ FFMPEG_USE_HEVC = False
FFMPEG_USE_NVIDIA_GPU = False
# intel_gpu
FFMPEG_USE_INTEL_GPU = False
# vaapi
FFMPEG_USE_VAAPI = False
# bitrate
VIDEO_BITRATE = "2.5M"
# crf
@ -66,10 +72,13 @@ def load_config():
config.read("config.ini", encoding="utf-8")
if config.has_section("danmaku"):
section = config['danmaku']
global DANMAKU_FACTORY_EXEC, DANMAKU_SPEED, DEFAULT_FONT_NAME, VIDEO_RESOLUTION, DANMAKU_FONT_SIZE
DANMAKU_FACTORY_EXEC = section.get('exec', DANMAKU_FACTORY_EXEC)
global DANMAKU_EXEC, DANMAKU_SPEED, DANMAKU_FONT_NAME, VIDEO_RESOLUTION, DANMAKU_FONT_SIZE, \
DANMAKU_USE_DANMU2ASS, DANMAKU_USE_DANMAKUFACTORY
DANMAKU_USE_DANMU2ASS = section.getboolean('use_danmu2ass', DANMAKU_USE_DANMU2ASS)
DANMAKU_USE_DANMAKUFACTORY = section.getboolean('use_danmakufactory', DANMAKU_USE_DANMAKUFACTORY)
DANMAKU_EXEC = section.get('exec', DANMAKU_EXEC)
DANMAKU_SPEED = section.getfloat('speed', DANMAKU_SPEED)
DEFAULT_FONT_NAME = section.get('font', DEFAULT_FONT_NAME)
DANMAKU_FONT_NAME = section.get('font', DANMAKU_FONT_NAME)
DANMAKU_FONT_SIZE = section.getint('font_size', DANMAKU_FONT_SIZE)
VIDEO_RESOLUTION = section.get('resolution', VIDEO_RESOLUTION)
if config.has_section("video"):
@ -88,11 +97,12 @@ def load_config():
if config.has_section("ffmpeg"):
section = config['ffmpeg']
global FFMPEG_EXEC, FFMPEG_USE_HEVC, FFMPEG_USE_NVIDIA_GPU, FFMPEG_USE_INTEL_GPU, VIDEO_BITRATE, VIDEO_CRF, \
VIDEO_GOP
VIDEO_GOP, FFMPEG_USE_VAAPI
FFMPEG_EXEC = section.get('exec', FFMPEG_EXEC)
FFMPEG_USE_HEVC = section.getboolean('hevc', FFMPEG_USE_HEVC)
FFMPEG_USE_NVIDIA_GPU = section.getboolean('nvidia_gpu', FFMPEG_USE_NVIDIA_GPU)
FFMPEG_USE_INTEL_GPU = section.getboolean('intel_gpu', FFMPEG_USE_INTEL_GPU)
FFMPEG_USE_VAAPI = section.getboolean('vaapi', FFMPEG_USE_VAAPI)
VIDEO_BITRATE = section.get('bitrate', VIDEO_BITRATE)
VIDEO_CRF = section.getfloat('crf', VIDEO_CRF)
VIDEO_GOP = section.getfloat('gop', VIDEO_GOP)
@ -108,9 +118,11 @@ def load_config():
def get_config():
config = {
'danmaku': {
'exec': DANMAKU_FACTORY_EXEC,
'exec': DANMAKU_EXEC,
'use_danmu2ass': DANMAKU_USE_DANMU2ASS,
'use_danmakufactory': DANMAKU_USE_DANMAKUFACTORY,
'speed': DANMAKU_SPEED,
'font': DEFAULT_FONT_NAME,
'font': DANMAKU_FONT_NAME,
'font_size': DANMAKU_FONT_SIZE,
'resolution': VIDEO_RESOLUTION,
},
@ -130,6 +142,7 @@ def get_config():
'hevc': FFMPEG_USE_HEVC,
'nvidia_gpu': FFMPEG_USE_NVIDIA_GPU,
'intel_gpu': FFMPEG_USE_INTEL_GPU,
'vaapi': FFMPEG_USE_VAAPI,
'bitrate': VIDEO_BITRATE,
'crf': VIDEO_CRF,
'gop': VIDEO_GOP,