Add support for specifying video codec profile

Author: jonathanhogg

src/flitter/cache.py

@@ -406,11 +406,11 @@ def write_image(self, image, quality=None, alpha=False):
         self._cache[key] = True
 
     def write_video_frame(self, frame, timestamp, codec='h264', pixfmt='yuv420p', fps=60, realtime=False,
-                          crf=None, preset=None, limit=None, alpha=False):
+                          crf=None, profile=None, preset=None, limit=None, alpha=False):
         import av
         self._touched = system_clock()
         width, height = frame.width, frame.height
-        key = 'video_output', width, height, alpha, codec, pixfmt, fps, crf, preset, limit
+        key = 'video_output', width, height, alpha, codec, profile, pixfmt, fps, crf, preset, limit
         writer, queue, start = self._cache.get(key, (None, None, None))
         if start is None:
             self.cleanup()
@@ -424,6 +424,8 @@ def write_video_frame(self, frame, timestamp, codec='h264', pixfmt='yuv420p', fp
                 options['crf'] = str(crf)
             if preset is not None:
                 options['preset'] = preset
+            if profile is not None:
+                options['profile'] = profile
             try:
                 codec = codec.lower()
                 if codec not in av.codecs_available:

src/flitter/render/window/record.py

@@ -15,12 +15,13 @@ def render(self, node, references, *, time=None, fps=None, realtime=None, **kwar
             codec = node.get('codec', 1, str, 'h264')
             keep_alpha = node.get('keep_alpha', 1, bool, False)
             if ext in ('.mp4', '.mov', '.m4v', '.mkv', '.webm', '.ogg') or (ext == '.gif' and codec == 'gif'):
-                pixfmt = node.get('pixfmt', 1, str, 'rgb8' if codec == 'gif' else 'yuv420p')
+                pixfmt = node.get('pixfmt', 1, str, 'rgb8' if codec == 'gif' else ('yuva420p' if keep_alpha else 'yuv420p'))
                 crf = node.get('crf', 1, int)
+                profile = node.get('profile', 1, str)
                 preset = node.get('preset', 1, str)
                 limit = node.get('limit', 1, float)
                 path.write_video_frame(self._target.video_frame, time, fps=int(fps), realtime=realtime, codec=codec,
-                                       pixfmt=pixfmt, crf=crf, preset=preset, limit=limit, alpha=keep_alpha)
+                                       pixfmt=pixfmt, crf=crf, profile=profile, preset=preset, limit=limit, alpha=keep_alpha)
             else:
                 quality = node.get('quality', 1, int)
                 path.write_image(self._target.image, quality=quality, alpha=keep_alpha)