Add support to save images in WebP format

kinshukdua · kinshukdua · commit 145984f63df6 · 2021-10-04T13:29:07.000+05:30
diff --git a/lib/matplotlib/backend_bases.py b/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',
 }
 
 
diff --git a/lib/matplotlib/backends/backend_agg.py b/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):

Original file line number	Diff line number	Diff line change
`@@ -69,6 +69,7 @@`
`69`	`69`	`'svgz': 'Scalable Vector Graphics',`
`70`	`70`	`'tif': 'Tagged Image File Format',`
`71`	`71`	`'tiff': 'Tagged Image File Format',`
	`72`	`+ 'webp': 'WebP Image Format',`
`72`	`73`	`}`
`73`	`74`	`_default_backends = {`
`74`	`75`	`'eps': 'matplotlib.backends.backend_ps',`
`@@ -84,6 +85,7 @@`
`84`	`85`	`'svgz': 'matplotlib.backends.backend_svg',`
`85`	`86`	`'tif': 'matplotlib.backends.backend_agg',`
`86`	`87`	`'tiff': 'matplotlib.backends.backend_agg',`
	`88`	`+ 'webp': 'matplotlib.backends.backend_agg',`
`87`	`89`	`}`
`88`	`90`
`89`	`91`