add ffmpeg only render for cccamp19
This commit is contained in:
derchris
parent
02b3ed3fa0
commit
aac613e920
5 changed files with 389 additions and 0 deletions
264
make-ffmpeg.py
Executable file
264
make-ffmpeg.py
Executable file
|
|
@ -0,0 +1,264 @@
|
|||
#!/usr/bin/env python3
|
||||
# vim: tabstop=4 shiftwidth=4 expandtab
|
||||
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
import renderlib
|
||||
import argparse
|
||||
import shlex
|
||||
from PIL import ImageFont
|
||||
from configparser import ConfigParser
|
||||
|
||||
# Parse arguments
|
||||
parser = argparse.ArgumentParser(
|
||||
description='C3VOC Intro-Outro-Generator - Variant which renders only using video filters in ffmpeg',
|
||||
usage="./make-ffmpeg.py yourproject/ https://url/to/schedule.xml",
|
||||
formatter_class=argparse.RawTextHelpFormatter)
|
||||
|
||||
parser.add_argument('project', action="store", metavar='Project folder', type=str, help='''
|
||||
Path to your project folder
|
||||
''')
|
||||
|
||||
parser.add_argument('schedule', action="store", metavar='Schedule-URL', type=str, nargs='?', help='''
|
||||
URL or Path to your schedule.xml
|
||||
''')
|
||||
|
||||
parser.add_argument('--debug', action="store_true", default=False, help='''
|
||||
Run script in debug mode and render with placeholder texts,
|
||||
not parsing or accessing a schedule. Schedule-URL can be left blank when
|
||||
used with --debug
|
||||
This argument must not be used together with --id
|
||||
Usage: ./make-ffmpeg.py yourproject/ --debug
|
||||
''')
|
||||
|
||||
parser.add_argument('--id', dest='ids', nargs='+', action="store", type=int, help='''
|
||||
Only render the given ID(s) from your projects schedule.
|
||||
This argument must not be used together with --debug
|
||||
Usage: ./make-adobe-after-effects.py yourproject/ --id 4711 0815 4223 1337
|
||||
''')
|
||||
|
||||
parser.add_argument('--room', dest='rooms', nargs='+', action="store", type=str, help='''
|
||||
Only render the given room(s) from your projects schedule.
|
||||
This argument must not be used together with --debug
|
||||
Usage: ./make-adobe-after-effects.py yourproject/ --room "HfG_Studio" "ZKM_Vortragssaal"
|
||||
''')
|
||||
|
||||
parser.add_argument('--skip', nargs='+', action="store", type=str, help='''
|
||||
Skip ID(s) not needed to be rendered.
|
||||
Usage: ./make-ffmpeg.py yourproject/ --skip 4711 0815 4223 1337
|
||||
''')
|
||||
|
||||
parser.add_argument('--force', action="store_true", default=False, help='''
|
||||
Force render if file exists.
|
||||
''')
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
if (args.skip is None):
|
||||
args.skip = []
|
||||
|
||||
|
||||
def headline(str):
|
||||
print("##################################################")
|
||||
print(str)
|
||||
print("##################################################")
|
||||
print()
|
||||
|
||||
|
||||
def error(str):
|
||||
headline(str)
|
||||
parser.print_help()
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if not (os.path.exists(os.path.join(args.project, 'config.ini'))):
|
||||
error("config.ini file in Project Path is missing")
|
||||
|
||||
if not args.project:
|
||||
error("The Project Path is a required argument")
|
||||
|
||||
if not args.debug and not args.schedule:
|
||||
error("Either specify --debug or supply a schedule")
|
||||
|
||||
if args.debug:
|
||||
persons = ['Thomas Roth', 'Dmitry Nedospasov', 'Josh Datko']
|
||||
events = [{
|
||||
'id': 'debug',
|
||||
'title': 'wallet.fail',
|
||||
'subtitle': 'Hacking the most popular cryptocurrency hardware wallets',
|
||||
'persons': persons,
|
||||
'personnames': ', '.join(persons),
|
||||
'room': 'Borg',
|
||||
}]
|
||||
|
||||
else:
|
||||
events = list(renderlib.events(args.schedule))
|
||||
|
||||
parser = ConfigParser()
|
||||
parser.read(os.path.join(os.path.dirname(args.project), 'config.ini'))
|
||||
template = parser['default']['template']
|
||||
|
||||
title_in = parser['title']['in']
|
||||
title_out = parser['title']['out']
|
||||
title_font = parser['title']['font']
|
||||
title_fontsize = parser['title']['fontsize']
|
||||
title_fontcolor = parser['title']['fontcolor']
|
||||
title_x = parser['title']['x']
|
||||
title_y = parser['title']['y']
|
||||
|
||||
speaker_in = parser['speaker']['in']
|
||||
speaker_out = parser['speaker']['out']
|
||||
speaker_font = parser['speaker']['font']
|
||||
speaker_fontsize = parser['speaker']['fontsize']
|
||||
speaker_fontcolor = parser['speaker']['fontcolor']
|
||||
speaker_x = parser['speaker']['x']
|
||||
speaker_y = parser['speaker']['y']
|
||||
|
||||
text_in = parser['text']['in']
|
||||
text_out = parser['text']['out']
|
||||
text_font = parser['text']['font']
|
||||
text_fontsize = parser['text']['fontsize']
|
||||
text_fontcolor = parser['text']['fontcolor']
|
||||
text_x = parser['text']['x']
|
||||
text_y = parser['text']['y']
|
||||
text_text = parser['text']['text']
|
||||
|
||||
font_t = os.path.join(os.path.dirname(args.project), title_font)
|
||||
font_s = os.path.join(os.path.dirname(args.project), speaker_font)
|
||||
font_tt = os.path.join(os.path.dirname(args.project), text_font)
|
||||
|
||||
fileformat = os.path.splitext(template)[1]
|
||||
infile = os.path.join(os.path.dirname(args.project), template)
|
||||
|
||||
def describe_event(event):
|
||||
return "#{}: {}".format(event['id'], event['title'])
|
||||
|
||||
|
||||
def event_print(event, message):
|
||||
print("{} – {}".format(describe_event(event), message))
|
||||
|
||||
|
||||
def fmt_command(command, **kwargs):
|
||||
args = {}
|
||||
for key, value in kwargs.items():
|
||||
args[key] = shlex.quote(value)
|
||||
|
||||
command = command.format(**args)
|
||||
return shlex.split(command)
|
||||
|
||||
|
||||
def run(command, **kwargs):
|
||||
return subprocess.check_call(
|
||||
fmt_command(command, **kwargs),
|
||||
stderr=subprocess.STDOUT,
|
||||
stdout=subprocess.DEVNULL)
|
||||
|
||||
|
||||
def fit_text(string: str, frame_width):
|
||||
split_line = [x.strip() for x in string.split()]
|
||||
lines = ""
|
||||
w = 0
|
||||
line_num = 0
|
||||
line = ""
|
||||
for word in split_line:
|
||||
w, _ = translation_font.getsize(" ".join([line, word]))
|
||||
if w > (frame_width - (2 * 6)):
|
||||
lines += line.strip() + "\n"
|
||||
line = ""
|
||||
|
||||
line += word + " "
|
||||
|
||||
lines += line.strip()
|
||||
return lines
|
||||
|
||||
|
||||
def fit_title(string: str):
|
||||
global translation_font
|
||||
translation_font = ImageFont.truetype(
|
||||
font_t, size=80, encoding="unic")
|
||||
title = fit_text(string, 1080)
|
||||
|
||||
return title
|
||||
|
||||
|
||||
def fit_speaker(string: str):
|
||||
global translation_font
|
||||
translation_font = ImageFont.truetype(
|
||||
font_s, size=50, encoding="unic")
|
||||
speaker = fit_text(string, 1080)
|
||||
|
||||
return speaker
|
||||
|
||||
|
||||
def enqueue_job(event):
|
||||
event_id = str(event['id'])
|
||||
if event_id in args.skip:
|
||||
event_print(event, "skipping " + str(event['id']))
|
||||
return
|
||||
if (os.path.exists(os.path.join(args.project, event_id + '.ts')) or os.path.exists(os.path.join(args.project, event_id + '.mov'))) and not args.force:
|
||||
event_print(event, "file exist, skipping " + str(event['id']))
|
||||
return
|
||||
|
||||
event_title = str(event['title'])
|
||||
event_personnames = str(event['personnames'])
|
||||
|
||||
t = fit_title(event_title)
|
||||
s = fit_speaker(event_personnames)
|
||||
|
||||
if args.debug:
|
||||
print('Title: ', t)
|
||||
print('Speaker: ', s)
|
||||
|
||||
if fileformat == '.mov':
|
||||
outfile = os.path.join(os.path.dirname(args.project), event_id + '.mov')
|
||||
else:
|
||||
outfile = os.path.join(os.path.dirname(args.project), event_id + '.ts')
|
||||
|
||||
videofilter = "drawtext=enable='between(n,{0},{1})':fontfile={2}:fontsize={3}:fontcolor={4}:x={5}:y={6}:text='{7}',".format(title_in, title_out, font_t, title_fontsize, title_fontcolor, title_x, title_y, t)
|
||||
videofilter += "drawtext=enable='between(n,{0},{1})':fontfile={2}:fontsize={3}:fontcolor={4}:x={5}:y={6}:text='{7}',".format(speaker_in, speaker_out, font_s, speaker_fontsize, speaker_fontcolor, speaker_x, speaker_y, s)
|
||||
videofilter += "drawtext=enable='between(n,{0},{1})':fontfile={2}:fontsize={3}:fontcolor={4}:x={5}:y={6}:text={7}".format(text_in, text_out, font_tt, text_fontsize, text_fontcolor, text_x, text_y, text_text)
|
||||
|
||||
if fileformat == '.mov':
|
||||
cmd = 'ffmpeg -y -i "{0}" -vf "{1}" -vcodec prores_ks -pix_fmt yuva444p10le -profile:v 4444 -shortest -movflags faststart -f mov "{2}"'.format(infile, videofilter, outfile)
|
||||
else:
|
||||
cmd = 'ffmpeg -y -i "{0}" -vf "{1}" -map 0:0 -c:v mpeg2video -q:v 2 -aspect 16:9 -map 0:1 -c:a mp2 -b:a 384k -shortest -f mpegts "{2}"'.format(infile, videofilter, outfile)
|
||||
if args.debug:
|
||||
print(cmd)
|
||||
|
||||
run(cmd)
|
||||
|
||||
return event_id
|
||||
|
||||
|
||||
if args.ids:
|
||||
if len(args.ids) == 1:
|
||||
print("enqueuing {} job".format(len(args.ids)))
|
||||
else:
|
||||
print("enqueuing {} jobs".format(len(args.ids)))
|
||||
else:
|
||||
if len(events) == 1:
|
||||
print("enqueuing {} job".format(len(events)))
|
||||
else:
|
||||
print("enqueuing {} jobs".format(len(events)))
|
||||
|
||||
|
||||
for event in events:
|
||||
if args.ids and event['id'] not in args.ids:
|
||||
continue
|
||||
|
||||
if args.rooms and event['room'] not in args.rooms:
|
||||
print("skipping room %s (%s)" % (event['room'], event['title']))
|
||||
continue
|
||||
|
||||
event_print(event, "enqueued as " + str(event['id']))
|
||||
|
||||
job_id = enqueue_job(event)
|
||||
if not job_id:
|
||||
event_print(event, "job was not enqueued successfully, skipping postprocessing")
|
||||
continue
|
||||
|
||||
|
||||
print('all done')
|
||||
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue