Add support to save images in WebP format

Author: kinshukdua

lib/matplotlib/backend_bases.py

@@ -69,6 +69,7 @@
     'svgz': 'Scalable Vector Graphics',
     'tif': 'Tagged Image File Format',
     'tiff': 'Tagged Image File Format',
+    'webp': 'WebP Image Format',
 }
 _default_backends = {
     'eps': 'matplotlib.backends.backend_ps',
@@ -84,6 +85,7 @@
     'svgz': 'matplotlib.backends.backend_svg',
     'tif': 'matplotlib.backends.backend_agg',
     'tiff': 'matplotlib.backends.backend_agg',
+    'webp': 'matplotlib.backends.backend_agg',
 }
 
 

lib/matplotlib/backends/backend_agg.py

@@ -594,6 +594,29 @@ def print_tif(self, filename_or_obj, *, pil_kwargs=None):
 
     print_tiff = print_tif
 
+    @_check_savefig_extra_args
+    def print_webp(self, filename_or_obj, *, pil_kwargs=None):
+        """
+        Write the figure to a WebP file.
+
+        Parameters
+        ----------
+        filename_or_obj : str or path-like or file-like
+            The file to write to.
+
+        Other Parameters
+        ----------------
+        pil_kwargs : dict, optional
+            Additional keyword arguments that are passed to
+            `PIL.Image.Image.save` when saving the figure.
+        """
+        FigureCanvasAgg.draw(self)
+        if pil_kwargs is None:
+            pil_kwargs = {}
+        pil_kwargs.setdefault("dpi", (self.figure.dpi, self.figure.dpi))
+        return (Image.fromarray(np.asarray(self.buffer_rgba()))
+                .save(filename_or_obj, format='webp', **pil_kwargs))
+
 
 @_Backend.export
 class _BackendAgg(_Backend):