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")
|