Fix a mode == 'RGBA'

This commit is contained in:
hugovk 2014-04-14 14:08:29 +03:00
parent 329fd00d3f
commit 98e4dfeefb

View File

@ -15,10 +15,11 @@
__version__ = "0.1"
from PIL import Image, ImageFile, _binary
import struct
import os
from PIL import Image, ImageFile
import io
import os
import struct
def _parse_codestream(fp):
"""Parse the JPEG 2000 codestream to extract the size and component
@ -45,12 +46,13 @@ def _parse_codestream(fp):
elif csiz == 3:
mode = 'RGB'
elif csiz == 4:
mode == 'RGBA'
mode = 'RGBA'
else:
mode = None
return (size, mode)
def _parse_jp2_header(fp):
"""Parse the JP2 header box to extract size, component count and
color space information, returning a PIL (size, mode) tuple."""
@ -127,6 +129,7 @@ def _parse_jp2_header(fp):
return (size, mode)
##
# Image plugin for JPEG2000 images.
@ -180,10 +183,12 @@ class Jpeg2KImageFile(ImageFile.ImageFile):
ImageFile.ImageFile.load(self)
def _accept(prefix):
return (prefix[:4] == b'\xff\x4f\xff\x51'
or prefix[:12] == b'\x00\x00\x00\x0cjP \x0d\x0a\x87\x0a')
# ------------------------------------------------------------
# Save support