Merge pull request #959 from pypeclub/bugfix/3.0_prores_ks_in_burnin_script

Prores ks in burnin script
This commit is contained in:
Milan Kolar 2021-02-01 18:39:26 +01:00 committed by GitHub
commit 56f022012a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -543,21 +543,30 @@ def burnins_from_data(
ffmpeg_args = codec_data
else:
codec_name = burnin._streams[0].get("codec_name")
ffprobe_data = burnin._streams[0]
codec_name = ffprobe_data.get("codec_name")
if codec_name:
if codec_name == "prores":
tags = ffprobe_data.get("tags") or {}
encoder = tags.get("encoder") or ""
if encoder.endswith("prores_ks"):
codec_name = "prores_ks"
elif encoder.endswith("prores_aw"):
codec_name = "prores_aw"
ffmpeg_args.append("-codec:v {}".format(codec_name))
profile_name = burnin._streams[0].get("profile")
profile_name = ffprobe_data.get("profile")
if profile_name:
# lower profile name and repalce spaces with underscore
profile_name = profile_name.replace(" ", "_").lower()
ffmpeg_args.append("-profile:v {}".format(profile_name))
bit_rate = burnin._streams[0].get("bit_rate")
bit_rate = ffprobe_data.get("bit_rate")
if bit_rate:
ffmpeg_args.append("-b:v {}".format(bit_rate))
pix_fmt = burnin._streams[0].get("pix_fmt")
pix_fmt = ffprobe_data.get("pix_fmt")
if pix_fmt:
ffmpeg_args.append("-pix_fmt {}".format(pix_fmt))