python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-08-14 17:24:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

update ImagingGetColors() for up to 4-byte pixels

Browse Source
This commit is contained in:
Yay295 2022-08-23 20:00:22 -05:00
parent 49099beb3d
commit eba429c9a3
1 changed files with 9 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/libImaging/GetBBox.c
View File

@ -240,22 +240,12 @@ ImagingGetExtrema(Imaging im, void *extrema) {
return 1; /* ok */
}
/* static ImagingColorItem* getcolors8(Imaging im, int maxcolors, int* size);*/
static ImagingColorItem *
getcolors32(Imaging im, int maxcolors, int *size);
ImagingColorItem *
ImagingGetColors(Imaging im, int maxcolors, int *size) {
/* FIXME: add support for 8-bit images */
return getcolors32(im, maxcolors, size);
}
static ImagingColorItem *
getcolors32(Imaging im, int maxcolors, int *size) {
unsigned int h;
unsigned int i, incr;
int colors;
INT32 pixel_mask;
INT32 pixel;
int x, y;
ImagingColorItem *table;
ImagingColorItem *v;
@ -287,36 +277,27 @@ getcolors32(Imaging im, int maxcolors, int *size) {
}
}
/* printf("code_size=%d\n", code_size); */
/* printf("code_poly=%d\n", code_poly); */
if (!code_size) {
return ImagingError_MemoryError(); /* just give up */
}
if (!im->image32) {
return ImagingError_ModeError();
}
table = calloc(code_size + 1, sizeof(ImagingColorItem));
if (!table) {
return ImagingError_MemoryError();
}
pixel_mask = 0xffffffff;
if (im->bands == 3) {
((UINT8 *)&pixel_mask)[3] = 0;
}
colors = 0;
for (y = 0; y < im->ysize; y++) {
INT32 *p = im->image32[y];
for (x = 0; x < im->xsize; x++) {
INT32 pixel = p[x] & pixel_mask;
UINT8 *in = (UINT8 *)im->image[y];
for (x = 0; x < im->xsize; x++, in += im->pixelsize) {
pixel = 0;
memcpy(&pixel, in, im->pixelsize);
h = (pixel); /* null hashing */
i = (~h) & code_mask;
v = &table[i];
if (!v->count) {
/* add to table */
if (colors++ == maxcolors) {
@ -363,13 +344,14 @@ getcolors32(Imaging im, int maxcolors, int *size) {
overflow:
/* pack the table */
for (x = y = 0; x < (int)code_size; x++)
for (x = y = 0; x < (int)code_size; x++) {
if (table[x].count) {
if (x != y) {
table[y] = table[x];
}
y++;
}
}
table[y].count = 0; /* mark end of table */
*size = colors;