264 lines
8.2 KiB
Python
Executable file
264 lines
8.2 KiB
Python
Executable file
#!/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')
|
||
|
||
|