Pillow/PIL/WebPImagePlugin.py

34 lines
978 B
Python
Raw Normal View History

2013-03-14 06:03:38 +04:00
from PIL import Image
from PIL import ImageFile
2013-03-14 06:42:21 +04:00
from io import BytesIO
2013-03-12 18:30:59 +04:00
import _webp
def _accept(prefix):
2013-03-14 10:28:30 +04:00
return prefix[:4] == b"RIFF" and prefix[8:16] == b"WEBPVP8 "
2013-03-12 18:30:59 +04:00
class WebPImageFile(ImageFile.ImageFile):
format = "WEBP"
format_description = "WebP image"
def _open(self):
self.mode = "RGB"
data, width, height = _webp.WebPDecodeRGB(self.fp.read())
self.size = width, height
2013-03-14 06:42:21 +04:00
self.fp = BytesIO(data)
2013-03-12 18:30:59 +04:00
self.tile = [("raw", (0, 0) + self.size, 0, 'RGB')]
def _save(im, fp, filename):
if im.mode != "RGB":
raise IOError("cannot write mode %s as WEBP" % im.mode)
quality = im.encoderinfo.get("quality", 80)
2013-03-20 00:53:37 +04:00
data = _webp.WebPEncodeRGB(im.tobytes(), im.size[0], im.size[1], im.size[0] * 3, float(quality))
2013-03-12 18:30:59 +04:00
fp.write(data)
Image.register_open("WEBP", WebPImageFile, _accept)
Image.register_save("WEBP", _save)
Image.register_extension("WEBP", ".webp")
Image.register_mime("WEBP", "image/webp")