public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix transparent background videos

Browse source
This commit is contained in:
Grant Sanderson 2023-02-15 09:38:35 -08:00
parent 01c51dbc6d
commit 557cb66c52
2 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
manimlib/config.py
View file

@ -420,11 +420,12 @@ def get_file_writer_config(args: Namespace, custom_config: dict) -> dict:
result["video_codec"] = args.vcodec result["video_codec"] = args.vcodec
elif args.transparent: elif args.transparent:
result["video_codec"] = 'prores_ks' result["video_codec"] = 'prores_ks'
result["pixel_format"] = ''
elif args.gif: elif args.gif:
result["video_codec"] = '' result["video_codec"] = ''
if args.pix_fmt: if args.pix_fmt:
result["pix_fmt"] = args.pix_fmt result["pixel_format"] = args.pix_fmt
return result return result

6
manimlib/scene/scene_file_writer.py
View file

@ -258,6 +258,10 @@ class SceneFileWriter(object):
fps = self.scene.camera.fps fps = self.scene.camera.fps
width, height = self.scene.camera.get_pixel_shape() width, height = self.scene.camera.get_pixel_shape()
vf_arg = 'vflip'
if self.pixel_format.startswith("yuv"):
vf_arg += f',eq=saturation={self.saturation}:gamma={self.gamma}'
command = [ command = [
FFMPEG_BIN, FFMPEG_BIN,
'-y', # overwrite output file if it exists '-y', # overwrite output file if it exists
@ -266,7 +270,7 @@ class SceneFileWriter(object):
'-pix_fmt', 'rgba', '-pix_fmt', 'rgba',
'-r', str(fps), # frames per second '-r', str(fps), # frames per second
'-i', '-', # The input comes from a pipe '-i', '-', # The input comes from a pipe
'-vf', f'vflip,eq=saturation={self.saturation}:gamma={self.gamma}', '-vf', vf_arg,
'-an', # Tells FFMPEG not to expect any audio '-an', # Tells FFMPEG not to expect any audio
'-loglevel', 'error', '-loglevel', 'error',
] ]