mirror of
https://github.com/python-pillow/Pillow.git
synced 2025-01-13 10:46:16 +03:00
Passes tests on python 3.2
Reorganized to remove ImagingMemoryInstance struct, only saving the three pointers that we need (image, image8, image32) and the x/ysize ints.
This commit is contained in:
parent
2921b5c5e5
commit
d8a88a5390
|
@ -26,34 +26,6 @@ from cffi import FFI
|
|||
DEBUG = 0
|
||||
|
||||
defs = """
|
||||
struct ImagingMemoryInstance {
|
||||
|
||||
/* Format */
|
||||
char mode[7]; /* Band names ("1", "L", "P", "RGB", "RGBA", "CMYK", "YCbCr", "BGR;xy") */
|
||||
int type; /* Data type (IMAGING_TYPE_*) */
|
||||
int depth; /* Depth (ignored in this version) */
|
||||
int bands; /* Number of bands (1, 2, 3, or 4) */
|
||||
int xsize; /* Image dimension. */
|
||||
int ysize;
|
||||
|
||||
/* Colour palette (for "P" images only) */
|
||||
void *palette;
|
||||
|
||||
/* Data pointers */
|
||||
unsigned char **image8; /* Set for 8-bit images (pixelsize=1). */
|
||||
int **image32; /* Set for 32-bit images (pixelsize=4). */
|
||||
|
||||
/* Internals */
|
||||
char **image; /* Actual raster data. */
|
||||
char *block; /* Set if data is allocated in a single block. */
|
||||
|
||||
int pixelsize; /* Size of a pixel, in bytes (1, 2 or 4) */
|
||||
int linesize; /* Size of a line, in bytes (xsize * pixelsize) */
|
||||
|
||||
/* Virtual methods */
|
||||
void (*destroy)(int im); /*keeping this for compatibility */
|
||||
};
|
||||
|
||||
struct Pixel_RGBA {
|
||||
unsigned char r,g,b,a;
|
||||
};
|
||||
|
@ -68,12 +40,18 @@ class PyAccess(object):
|
|||
def __init__(self, img, readonly = False):
|
||||
vals = dict(img.im.unsafe_ptrs)
|
||||
self.readonly = readonly
|
||||
for att in ['palette', 'image8', 'image32','image', 'block', 'destroy']:
|
||||
vals[att] = ffi.cast("void *", vals[att])
|
||||
self.image8 = ffi.cast('unsigned char **', vals['image8'])
|
||||
self.image32 = ffi.cast('int **', vals['image32'])
|
||||
self.image = ffi.cast('unsigned char **', vals['image'])
|
||||
self.xsize = vals['xsize']
|
||||
self.ysize = vals['ysize']
|
||||
|
||||
if DEBUG:
|
||||
print (vals)
|
||||
self.im = ffi.new("struct ImagingMemoryInstance *", vals)
|
||||
self._post_init()
|
||||
|
||||
def _post_init(): pass
|
||||
|
||||
def __setitem__(self, xy, color):
|
||||
"""
|
||||
Modifies the pixel at x,y. The color is given as a single
|
||||
|
@ -104,15 +82,14 @@ class PyAccess(object):
|
|||
|
||||
def check_xy(self, xy):
|
||||
(x,y) = xy
|
||||
if not (0 <= x < self.im.xsize and 0 <= y < self.im.ysize):
|
||||
if not (0 <= x < self.xsize and 0 <= y < self.ysize):
|
||||
raise Exception('ValueError- pixel location out of range') #undone
|
||||
return (x,y)
|
||||
|
||||
|
||||
class _PyAccess32_3(PyAccess):
|
||||
def __init__(self, *args, **kwargs):
|
||||
PyAccess.__init__(self, *args, **kwargs)
|
||||
self.pixels = ffi.cast("struct Pixel_RGBA **", self.im.image32)
|
||||
def _post_init(self, *args, **kwargs):
|
||||
self.pixels = ffi.cast("struct Pixel_RGBA **", self.image32)
|
||||
|
||||
def get_pixel(self, x,y):
|
||||
pixel = self.pixels[y][x]
|
||||
|
@ -120,19 +97,13 @@ class _PyAccess32_3(PyAccess):
|
|||
|
||||
def set_pixel(self, x,y, color):
|
||||
pixel = self.pixels[y][x]
|
||||
try:
|
||||
# tuple
|
||||
pixel.r, pixel.g, pixel.b = color
|
||||
except:
|
||||
# int, as a char[4]
|
||||
pixel.r = color >> 24
|
||||
pixel.g = (color >> 16) & 0xFF
|
||||
pixel.b = (color >> 8) & 0xFF
|
||||
# tuple
|
||||
pixel.r, pixel.g, pixel.b = color
|
||||
|
||||
|
||||
class _PyAccess32_4(PyAccess):
|
||||
def __init__(self, *args, **kwargs):
|
||||
PyAccess.__init__(self, *args, **kwargs)
|
||||
self.pixels = ffi.cast("struct Pixel_RGBA **", self.im.image32)
|
||||
def _post_init(self, *args, **kwargs):
|
||||
self.pixels = ffi.cast("struct Pixel_RGBA **", self.image32)
|
||||
|
||||
def get_pixel(self, x,y):
|
||||
pixel = self.pixels[y][x]
|
||||
|
@ -140,20 +111,12 @@ class _PyAccess32_4(PyAccess):
|
|||
|
||||
def set_pixel(self, x,y, color):
|
||||
pixel = self.pixels[y][x]
|
||||
try:
|
||||
# tuple
|
||||
pixel.r, pixel.g, pixel.b, pixel.a = color
|
||||
except:
|
||||
# int, as a char[4]
|
||||
pixel.r = color >> 24
|
||||
pixel.g = (color >> 16) & 0xFF
|
||||
pixel.b = (color >> 8) & 0xFF
|
||||
pixel.a = color & 0xFF
|
||||
# tuple
|
||||
pixel.r, pixel.g, pixel.b, pixel.a = color
|
||||
|
||||
class _PyAccess8(PyAccess):
|
||||
def __init__(self, *args, **kwargs):
|
||||
PyAccess.__init__(self, *args, **kwargs)
|
||||
self.pixels = self.im.image8
|
||||
def _post_init(self, *args, **kwargs):
|
||||
self.pixels = self.image8
|
||||
|
||||
def get_pixel(self, x,y):
|
||||
return self.pixels[y][x]
|
||||
|
|
Loading…
Reference in New Issue
Block a user