Skip to content

ENH: Add support to save images in WebP format #21274

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Oct 15, 2021
Merged

ENH: Add support to save images in WebP format #21274

merged 4 commits into from
Oct 15, 2021

Conversation

kinshukdua
Copy link
Contributor

PR Summary

Fixes #21162
This adds support to save images in WebP format.
This works fine without any problem since Pillow supports WebP and the code is analogous to the support for tiff images added in PR #15193.

PR Checklist

  • Has pytest style unit tests (and pytest passes).
  • Is Flake 8 compliant (run flake8 on changed files to check).
  • [N/A] New features are documented, with examples if plot related.
  • Documentation is sphinx and numpydoc compliant (the docs should build without error).
  • Conforms to Matplotlib style conventions (install flake8-docstrings and run flake8 --docstring-convention=all).
  • [N/A] New features have an entry in doc/users/next_whats_new/ (follow instructions in README.rst there).
  • [N/A] API changes documented in doc/api/next_api_changes/ (follow instructions in README.rst there).

Copy link

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for opening your first PR into Matplotlib!

If you have not heard from us in a while, please feel free to ping @matplotlib/developers or anyone who has commented on the PR. Most of our reviewers are volunteers and sometimes things fall through the cracks.

You can also join us on gitter for real-time discussion.

For details on testing, writing docs, and our review process, please see the developer guide

We strive to be a welcoming and open project. Please follow our Code of Conduct.

@tacaswell tacaswell added this to the v3.6.0 milestone Oct 4, 2021
@tacaswell
Copy link
Member

It is nice how simple this patch is to pick up a new format!

Could you add a test that saves an image to ByteIO buffer? That will be enough to exercise the code that you added. Given that this is using the Agg backend we definitely do not need to run it against the full test suite as we would then just be testing pillow. It looks like print_tiff is covered, maybe find that test and adapt / parameterize it?

@kinshukdua
Copy link
Contributor Author

@tacaswell Yes, I will add a test using ByteIO buffer.

@@ -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):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Are all these pillow save formats the same? If so, I wonder if we just want a generic print_pillow(format=pillow_format) rather than adding these one-by-one?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As far as I can tell, it is the same for all the fully supported formats- BMP, DDS, DIB, EPS, GIF, ICNS, ICO, IM, JPEG, JPEG 2000, MSP, PCX, PNG, PPM, SGI, SPIDER, TGA, TIFF, WebP, XBM.
I think adding a generic print_pillow does make a lot of sense especially if we want to support all of these formats. The tests however will have to be created for each format, as they all have their specific options that need to be tested.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thats the approach I would take, but maybe its more complex than that...

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe we should leave it for a different PR then?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I guess that depends on whether we think webp is used enough to merit a standalone method.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I suspect the way we would have to do this in general would be some mild-meta-programming as

print_method = getattr(canvas, 'print_%s' % format)
is how we look up the method to use and above it we get the format via
if format is None:
# get format from filename, or from backend's default filetype
if isinstance(filename, os.PathLike):
filename = os.fspath(filename)
if isinstance(filename, str):
format = os.path.splitext(filename)[1][1:]
if format is None or format == '':
format = self.get_default_filetype()
if isinstance(filename, str):
filename = filename.rstrip('.') + '.' + format
format = format.lower()
so we have to have a print_foo method for every format.

However, we can write ourselves a print function factory function and in backend_agg.py do something like:

for fmt in LIST_OF_PILLOW_FORMATS:
   setattr(BackendAgg, f'print_{fmt}', pillow_printer_factor(fmt)

@tacaswell
Copy link
Member

@kinshukdua Would you be willing to rebase this PR? We prefer to not merge the default branch back into feature branches.

@kinshukdua
Copy link
Contributor Author

@tacaswell I've rebased the PR, I'm new to git so please let me know if I need to do something else.

Copy link
Member

@jklymak jklymak left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think we need to decide whether we want a new webp method, a general "pass to pillow" method or a factory. If in the end, we don't want a new webp method, I dont think we can add this.

I'm not against a webp method if it really is something many folks want, but it seems obscure to me, and maybe just a general method to pillow takes care of the obscure use cases.

I'm blocking until we can sort this out, but if another core dev wants to support a standalone webp method, feel free to dismiss my review.

@jklymak
Copy link
Member

jklymak commented Oct 12, 2021

@tacaswell said that this is what we want, which is fine with me. Thanks for your patience!

Copy link
Contributor

@anntzer anntzer left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please consider clarifying the situation wrt. dpi. Anyone can dismiss this.

@anntzer
Copy link
Contributor

anntzer commented Oct 15, 2021

Thanks, anyone can merge after CI passes. Congrats on getting your first PR in :)

@QuLogic QuLogic merged commit 60466fb into matplotlib:master Oct 15, 2021
@QuLogic
Copy link
Member

QuLogic commented Oct 15, 2021

Thanks @kinshukdua! Congratulations on your first PR to Matplotlib 🎉 We hope to hear from you again.

@kinshukdua
Copy link
Contributor Author

Thanks! It was really great contributing back to such an important open source project, I hope to continue contributing.

tacaswell pushed a commit that referenced this pull request Oct 20, 2021
ENH: Add support to save images in WebP format
@QuLogic QuLogic mentioned this pull request Sep 9, 2022
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[ENH]: saving images in webp format
6 participants