anim: Only pre-composite formats that are opaque

On video formats like `.mp4`, it makes sense to do our own
pre-compositing to white. The animation backends would otherwise do it
themselves, and tend to just make semi-transparent pixels either fully
transparent or fully opaque before placing on white. So doing it
ourselves corrects that error.

However, formts like `.gif`, or `.webm` _do_ support transparency, and
we shouldn't do the compositing then.

Fixes matplotlib#27173

Author: QuLogic

lib/matplotlib/animation.py

@@ -295,6 +295,15 @@ def _adjust_frame_size(self):
         _log.debug('frame size in pixels is %s x %s', *self.frame_size)
         return w, h
 
+    @property
+    def _supports_transparency(self):
+        """
+        Whether this writer supports transparency.
+
+        Writers may consult output file type and codec to determine this at runtime.
+        """
+        return False
+
     def setup(self, fig, outfile, dpi=None):
         # docstring inherited
         super().setup(fig, outfile, dpi=dpi)
@@ -468,6 +477,10 @@ def finish(self):
 
 @writers.register('pillow')
 class PillowWriter(AbstractMovieWriter):
+    @property
+    def _supports_transparency(self):
+        return True
+
     @classmethod
     def isAvailable(cls):
         return True
@@ -503,6 +516,15 @@ class FFMpegBase:
     _exec_key = 'animation.ffmpeg_path'
     _args_key = 'animation.ffmpeg_args'
 
+    @property
+    def _supports_transparency(self):
+        suffix = Path(self.outfile).suffix
+        if suffix in {'.apng', '.avif', '.gif', '.webm', '.webp'}:
+            return True
+        elif suffix == '.mov':
+            return self.codec == 'png'
+        return False
+
     @property
     def output_args(self):
         args = []
@@ -519,11 +541,17 @@ def output_args(self):
         # macOS). Also fixes internet explorer. This is as of 2015/10/29.
         if self.codec == 'h264' and '-pix_fmt' not in extra_args:
             args.extend(['-pix_fmt', 'yuv420p'])
-        # For GIF, we're telling FFMPEG to split the video stream, to generate
+        # For GIF, we're telling FFmpeg to split the video stream, to generate
         # a palette, and then use it for encoding.
         elif self.codec == 'gif' and '-filter_complex' not in extra_args:
             args.extend(['-filter_complex',
                          'split [a][b];[a] palettegen [p];[b][p] paletteuse'])
+        # For AVIF, we're telling FFmpeg to split the video stream, extract the alpha,
+        # in order to place it in a secondary stream, as needed by AVIF-in-FFmpeg.
+        elif self.codec == 'avif' and '-filter_complex' not in extra_args:
+            args.extend(['-filter_complex',
+                         'split [rgb][rgba]; [rgba] alphaextract [alpha]',
+                         '-map', '[rgb]', '-map', '[alpha]'])
         if self.bitrate > 0:
             args.extend(['-b', '%dk' % self.bitrate])  # %dk: bitrate in kbps.
         for k, v in self.metadata.items():
@@ -611,6 +639,11 @@ class ImageMagickBase:
     _exec_key = 'animation.convert_path'
     _args_key = 'animation.convert_args'
 
+    @property
+    def _supports_transparency(self):
+        suffix = Path(self.outfile).suffix
+        return suffix in {'.apng', '.avif', '.gif', '.webm', '.webp'}
+
     def _args(self):
         # ImageMagick does not recognize "raw".
         fmt = "rgba" if self.frame_format == "raw" else self.frame_format
@@ -1046,22 +1079,23 @@ def func(current_frame: int, total_frames: int) -> Any
         # since GUI widgets are gone. Either need to remove extra code to
         # allow for this non-existent use case or find a way to make it work.
 
-        facecolor = savefig_kwargs.get('facecolor',
-                                       mpl.rcParams['savefig.facecolor'])
-        if facecolor == 'auto':
-            facecolor = self._fig.get_facecolor()
-
         def _pre_composite_to_white(color):
             r, g, b, a = mcolors.to_rgba(color)
             return a * np.array([r, g, b]) + 1 - a
 
-        savefig_kwargs['facecolor'] = _pre_composite_to_white(facecolor)
-        savefig_kwargs['transparent'] = False   # just to be safe!
         # canvas._is_saving = True makes the draw_event animation-starting
         # callback a no-op; canvas.manager = None prevents resizing the GUI
         # widget (both are likewise done in savefig()).
         with (writer.saving(self._fig, filename, dpi),
               cbook._setattr_cm(self._fig.canvas, _is_saving=True, manager=None)):
+            if not getattr(writer, '_supports_transparency', False):
+                facecolor = savefig_kwargs.get('facecolor',
+                                               mpl.rcParams['savefig.facecolor'])
+                if facecolor == 'auto':
+                    facecolor = self._fig.get_facecolor()
+                savefig_kwargs['facecolor'] = _pre_composite_to_white(facecolor)
+                savefig_kwargs['transparent'] = False   # just to be safe!
+
             for anim in all_anim:
                 anim._init_draw()  # Clear the initial frame
             frame_number = 0