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

SIMD Bands. access violation in getband

Browse Source
This commit is contained in:
Alexander 2020-01-17 03:19:06 +03:00 committed by Alexander Karpinsky
parent d74ab21ac9
commit be70e9eded
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/libImaging/Bands.c
View File

@ -22,6 +22,7 @@ Imaging
ImagingGetBand(Imaging imIn, int band) { ImagingGetBand(Imaging imIn, int band) {
Imaging imOut; Imaging imOut;
int x, y; int x, y;
__m128i shuffle_mask;
/* Check arguments */ /* Check arguments */
if (!imIn || imIn->type != IMAGING_TYPE_UINT8) { if (!imIn || imIn->type != IMAGING_TYPE_UINT8) {
@ -47,20 +48,22 @@ ImagingGetBand(Imaging imIn, int band) {
return NULL; return NULL;
} }
shuffle_mask = _mm_set_epi8(
-1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, 12+band,8+band,4+band,0+band);
/* Extract band from image */ /* Extract band from image */
for (y = 0; y < imIn->ysize; y++) { for (y = 0; y < imIn->ysize; y++) {
UINT8 *in = (UINT8 *)imIn->image[y] + band; UINT8* in = (UINT8*) imIn->image[y];
UINT8 *out = imOut->image8[y]; UINT8* out = imOut->image8[y];
x = 0; x = 0;
for (; x < imIn->xsize - 3; x += 4) { for (; x < imIn->xsize - 3; x += 4) {
__m128i source = _mm_loadu_si128((__m128i *) in); __m128i source = _mm_loadu_si128((__m128i *) in);
*((UINT32*) (out + x)) = _mm_cvtsi128_si32( *((UINT32*) (out + x)) = _mm_cvtsi128_si32(
_mm_shuffle_epi8(source, _mm_set_epi8( _mm_shuffle_epi8(source, shuffle_mask));
-1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, 12,8,4,0)));
in += 16; in += 16;
} }
for (; x < imIn->xsize; x++) { for (; x < imIn->xsize; x++) {
out[x] = *in; out[x] = *(in + band);
in += 4; in += 4;
} }
} }