Pillow/src/libImaging/Geometry.c

1070 lines
29 KiB
C
Raw Normal View History

2010-07-31 06:52:47 +04:00
#include "Imaging.h"
2014-11-07 11:48:47 +03:00
/* For large images rotation is an inefficient operation in terms of CPU cache.
One row in the source image affects each column in destination.
Rotating in chunks that fit in the cache can speed up rotation
8x on a modern CPU. A chunk size of 128 requires only 65k and is large enough
that the overhead from the extra loops are not apparent. */
2017-09-11 17:01:38 +03:00
#define ROTATE_CHUNK 512
#define ROTATE_SMALL_CHUNK 8
2014-11-07 04:46:16 +03:00
2010-07-31 06:52:47 +04:00
#define COORD(v) ((v) < 0.0 ? -1 : ((int)(v)))
#define FLOOR(v) ((v) < 0.0 ? ((int)floor(v)) : ((int)(v)))
/* -------------------------------------------------------------------- */
2014-11-07 03:21:38 +03:00
/* Transpose operations */
2010-07-31 06:52:47 +04:00
Imaging
ImagingFlipLeftRight(Imaging imOut, Imaging imIn)
{
ImagingSectionCookie cookie;
int x, y, xr;
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
if (imIn->xsize != imOut->xsize || imIn->ysize != imOut->ysize)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_Mismatch();
2010-07-31 06:52:47 +04:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
#define FLIP_LEFT_RIGHT(INT, image) \
for (y = 0; y < imIn->ysize; y++) { \
INT* in = (INT *)imIn->image[y]; \
INT* out = (INT *)imOut->image[y]; \
xr = imIn->xsize-1; \
for (x = 0; x < imIn->xsize; x++, xr--) \
out[xr] = in[x]; \
}
2010-07-31 06:52:47 +04:00
ImagingSectionEnter(&cookie);
2017-09-10 12:45:45 +03:00
if (imIn->image8) {
if (strncmp(imIn->mode, "I;16", 4) == 0) {
FLIP_LEFT_RIGHT(UINT16, image8)
} else {
FLIP_LEFT_RIGHT(UINT8, image8)
}
2017-09-10 12:45:45 +03:00
} else {
FLIP_LEFT_RIGHT(INT32, image32)
2017-09-10 12:45:45 +03:00
}
2010-07-31 06:52:47 +04:00
ImagingSectionLeave(&cookie);
#undef FLIP_LEFT_RIGHT
2010-07-31 06:52:47 +04:00
return imOut;
}
Imaging
ImagingFlipTopBottom(Imaging imOut, Imaging imIn)
{
ImagingSectionCookie cookie;
int y, yr;
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
if (imIn->xsize != imOut->xsize || imIn->ysize != imOut->ysize)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_Mismatch();
2010-07-31 06:52:47 +04:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
ImagingSectionEnter(&cookie);
yr = imIn->ysize - 1;
2010-07-31 06:52:47 +04:00
for (y = 0; y < imIn->ysize; y++, yr--)
2014-11-07 03:21:38 +03:00
memcpy(imOut->image[yr], imIn->image[y], imIn->linesize);
2010-07-31 06:52:47 +04:00
ImagingSectionLeave(&cookie);
return imOut;
}
Imaging
ImagingRotate90(Imaging imOut, Imaging imIn)
{
ImagingSectionCookie cookie;
2014-11-07 04:15:18 +03:00
int x, y, xx, yy, xr, xxsize, yysize;
2017-09-11 17:01:38 +03:00
int xxx, yyy, xxxsize, yyysize;
2010-07-31 06:52:47 +04:00
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
if (imIn->xsize != imOut->ysize || imIn->ysize != imOut->xsize)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_Mismatch();
2010-07-31 06:52:47 +04:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
2017-09-11 17:01:38 +03:00
#define ROTATE_90(INT, image) \
2014-11-07 04:46:16 +03:00
for (y = 0; y < imIn->ysize; y += ROTATE_CHUNK) { \
for (x = 0; x < imIn->xsize; x += ROTATE_CHUNK) { \
yysize = y + ROTATE_CHUNK < imIn->ysize ? y + ROTATE_CHUNK : imIn->ysize; \
xxsize = x + ROTATE_CHUNK < imIn->xsize ? x + ROTATE_CHUNK : imIn->xsize; \
2017-09-11 17:01:38 +03:00
for (yy = y; yy < yysize; yy += ROTATE_SMALL_CHUNK) { \
for (xx = x; xx < xxsize; xx += ROTATE_SMALL_CHUNK) { \
yyysize = yy + ROTATE_SMALL_CHUNK < imIn->ysize ? yy + ROTATE_SMALL_CHUNK : imIn->ysize; \
xxxsize = xx + ROTATE_SMALL_CHUNK < imIn->xsize ? xx + ROTATE_SMALL_CHUNK : imIn->xsize; \
for (yyy = yy; yyy < yyysize; yyy++) { \
INT* in = (INT *)imIn->image[yyy]; \
2017-09-11 17:01:38 +03:00
xr = imIn->xsize - 1 - xx; \
for (xxx = xx; xxx < xxxsize; xxx++, xr--) { \
INT* out = (INT *)imOut->image[xr]; \
out[yyy] = in[xxx]; \
2017-09-11 17:01:38 +03:00
} \
} \
2014-11-07 04:15:18 +03:00
} \
} \
} \
2010-07-31 06:52:47 +04:00
}
ImagingSectionEnter(&cookie);
if (imIn->image8) {
if (strncmp(imIn->mode, "I;16", 4) == 0) {
ROTATE_90(UINT16, image8);
} else {
ROTATE_90(UINT8, image8);
}
} else {
ROTATE_90(INT32, image32);
}
2010-07-31 06:52:47 +04:00
2014-10-25 04:12:24 +04:00
ImagingSectionLeave(&cookie);
#undef ROTATE_90
2014-10-25 04:12:24 +04:00
return imOut;
}
Imaging
ImagingTranspose(Imaging imOut, Imaging imIn)
{
ImagingSectionCookie cookie;
int x, y, xx, yy, xxsize, yysize;
2017-09-11 17:01:38 +03:00
int xxx, yyy, xxxsize, yyysize;
2014-10-25 04:12:24 +04:00
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
return (Imaging) ImagingError_ModeError();
if (imIn->xsize != imOut->ysize || imIn->ysize != imOut->xsize)
return (Imaging) ImagingError_Mismatch();
ImagingCopyPalette(imOut, imIn);
2017-09-11 17:01:38 +03:00
#define TRANSPOSE(INT, image) \
2014-11-07 04:46:16 +03:00
for (y = 0; y < imIn->ysize; y += ROTATE_CHUNK) { \
for (x = 0; x < imIn->xsize; x += ROTATE_CHUNK) { \
yysize = y + ROTATE_CHUNK < imIn->ysize ? y + ROTATE_CHUNK : imIn->ysize; \
xxsize = x + ROTATE_CHUNK < imIn->xsize ? x + ROTATE_CHUNK : imIn->xsize; \
2017-09-11 17:01:38 +03:00
for (yy = y; yy < yysize; yy += ROTATE_SMALL_CHUNK) { \
for (xx = x; xx < xxsize; xx += ROTATE_SMALL_CHUNK) { \
yyysize = yy + ROTATE_SMALL_CHUNK < imIn->ysize ? yy + ROTATE_SMALL_CHUNK : imIn->ysize; \
xxxsize = xx + ROTATE_SMALL_CHUNK < imIn->xsize ? xx + ROTATE_SMALL_CHUNK : imIn->xsize; \
for (yyy = yy; yyy < yyysize; yyy++) { \
INT* in = (INT *)imIn->image[yyy]; \
2017-09-11 17:01:38 +03:00
for (xxx = xx; xxx < xxxsize; xxx++) { \
INT* out = (INT *)imOut->image[xxx]; \
out[yyy] = in[xxx]; \
2017-09-11 17:01:38 +03:00
} \
} \
2014-10-25 04:12:24 +04:00
} \
} \
} \
}
ImagingSectionEnter(&cookie);
2016-09-03 05:23:42 +03:00
if (imIn->image8) {
if (strncmp(imIn->mode, "I;16", 4) == 0) {
TRANSPOSE(UINT16, image8);
} else {
TRANSPOSE(UINT8, image8);
}
} else {
TRANSPOSE(INT32, image32);
}
2010-07-31 06:52:47 +04:00
ImagingSectionLeave(&cookie);
#undef TRANSPOSE
2010-07-31 06:52:47 +04:00
return imOut;
}
2017-09-11 01:47:47 +03:00
Imaging
ImagingTransverse(Imaging imOut, Imaging imIn)
2017-09-11 01:47:47 +03:00
{
ImagingSectionCookie cookie;
2017-09-11 17:01:38 +03:00
int x, y, xr, yr, xx, yy, xxsize, yysize;
int xxx, yyy, xxxsize, yyysize;
2017-09-11 01:47:47 +03:00
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
return (Imaging) ImagingError_ModeError();
if (imIn->xsize != imOut->ysize || imIn->ysize != imOut->xsize)
return (Imaging) ImagingError_Mismatch();
2017-09-19 21:01:34 +03:00
ImagingCopyPalette(imOut, imIn);
2017-09-11 01:47:47 +03:00
#define TRANSVERSE(INT, image) \
2017-09-11 01:47:47 +03:00
for (y = 0; y < imIn->ysize; y += ROTATE_CHUNK) { \
for (x = 0; x < imIn->xsize; x += ROTATE_CHUNK) { \
yysize = y + ROTATE_CHUNK < imIn->ysize ? y + ROTATE_CHUNK : imIn->ysize; \
xxsize = x + ROTATE_CHUNK < imIn->xsize ? x + ROTATE_CHUNK : imIn->xsize; \
2017-09-11 17:01:38 +03:00
for (yy = y; yy < yysize; yy += ROTATE_SMALL_CHUNK) { \
for (xx = x; xx < xxsize; xx += ROTATE_SMALL_CHUNK) { \
yyysize = yy + ROTATE_SMALL_CHUNK < imIn->ysize ? yy + ROTATE_SMALL_CHUNK : imIn->ysize; \
xxxsize = xx + ROTATE_SMALL_CHUNK < imIn->xsize ? xx + ROTATE_SMALL_CHUNK : imIn->xsize; \
yr = imIn->ysize - 1 - yy; \
for (yyy = yy; yyy < yyysize; yyy++, yr--) { \
INT* in = (INT *)imIn->image[yyy]; \
2017-09-11 17:01:38 +03:00
xr = imIn->xsize - 1 - xx; \
for (xxx = xx; xxx < xxxsize; xxx++, xr--) { \
INT* out = (INT *)imOut->image[xr]; \
out[yr] = in[xxx]; \
2017-09-11 17:01:38 +03:00
} \
} \
2017-09-11 01:47:47 +03:00
} \
} \
} \
}
ImagingSectionEnter(&cookie);
if (imIn->image8) {
if (strncmp(imIn->mode, "I;16", 4) == 0) {
TRANSVERSE(UINT16, image8);
} else {
TRANSVERSE(UINT8, image8);
}
} else {
TRANSVERSE(INT32, image32);
}
2017-09-11 01:47:47 +03:00
ImagingSectionLeave(&cookie);
#undef TRANSVERSE
2017-09-11 01:47:47 +03:00
return imOut;
}
2010-07-31 06:52:47 +04:00
Imaging
ImagingRotate180(Imaging imOut, Imaging imIn)
{
ImagingSectionCookie cookie;
int x, y, xr, yr;
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
if (imIn->xsize != imOut->xsize || imIn->ysize != imOut->ysize)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_Mismatch();
2010-07-31 06:52:47 +04:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
#define ROTATE_180(INT, image) \
for (y = 0; y < imIn->ysize; y++, yr--) { \
INT* in = (INT *)imIn->image[y]; \
INT* out = (INT *)imOut->image[yr]; \
xr = imIn->xsize-1; \
for (x = 0; x < imIn->xsize; x++, xr--) \
out[xr] = in[x]; \
2010-07-31 06:52:47 +04:00
}
ImagingSectionEnter(&cookie);
yr = imIn->ysize-1;
2017-09-11 01:07:38 +03:00
if (imIn->image8) {
if (strncmp(imIn->mode, "I;16", 4) == 0) {
ROTATE_180(UINT16, image8)
} else {
ROTATE_180(UINT8, image8)
}
2017-09-11 01:07:38 +03:00
} else {
ROTATE_180(INT32, image32)
2017-09-11 01:07:38 +03:00
}
2010-07-31 06:52:47 +04:00
ImagingSectionLeave(&cookie);
#undef ROTATE_180
2010-07-31 06:52:47 +04:00
return imOut;
}
Imaging
ImagingRotate270(Imaging imOut, Imaging imIn)
{
ImagingSectionCookie cookie;
2014-11-07 04:15:18 +03:00
int x, y, xx, yy, yr, xxsize, yysize;
2017-09-11 17:01:38 +03:00
int xxx, yyy, xxxsize, yyysize;
2010-07-31 06:52:47 +04:00
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
if (imIn->xsize != imOut->ysize || imIn->ysize != imOut->xsize)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_Mismatch();
2010-07-31 06:52:47 +04:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
2017-09-11 17:01:38 +03:00
#define ROTATE_270(INT, image) \
2014-11-07 04:46:16 +03:00
for (y = 0; y < imIn->ysize; y += ROTATE_CHUNK) { \
for (x = 0; x < imIn->xsize; x += ROTATE_CHUNK) { \
yysize = y + ROTATE_CHUNK < imIn->ysize ? y + ROTATE_CHUNK : imIn->ysize; \
xxsize = x + ROTATE_CHUNK < imIn->xsize ? x + ROTATE_CHUNK : imIn->xsize; \
2017-09-11 17:01:38 +03:00
for (yy = y; yy < yysize; yy += ROTATE_SMALL_CHUNK) { \
for (xx = x; xx < xxsize; xx += ROTATE_SMALL_CHUNK) { \
yyysize = yy + ROTATE_SMALL_CHUNK < imIn->ysize ? yy + ROTATE_SMALL_CHUNK : imIn->ysize; \
xxxsize = xx + ROTATE_SMALL_CHUNK < imIn->xsize ? xx + ROTATE_SMALL_CHUNK : imIn->xsize; \
yr = imIn->ysize - 1 - yy; \
for (yyy = yy; yyy < yyysize; yyy++, yr--) { \
INT* in = (INT *)imIn->image[yyy]; \
2017-09-11 17:01:38 +03:00
for (xxx = xx; xxx < xxxsize; xxx++) { \
INT* out = (INT *)imOut->image[xxx]; \
out[yr] = in[xxx]; \
2017-09-11 17:01:38 +03:00
} \
} \
2014-11-07 04:15:18 +03:00
} \
} \
} \
}
2010-07-31 06:52:47 +04:00
ImagingSectionEnter(&cookie);
if (imIn->image8) {
if (strncmp(imIn->mode, "I;16", 4) == 0) {
ROTATE_270(UINT16, image8);
} else {
ROTATE_270(UINT8, image8);
}
} else {
ROTATE_270(INT32, image32);
}
2010-07-31 06:52:47 +04:00
ImagingSectionLeave(&cookie);
#undef ROTATE_270
2010-07-31 06:52:47 +04:00
return imOut;
}
/* -------------------------------------------------------------------- */
2014-11-07 03:21:38 +03:00
/* Transforms */
2010-07-31 06:52:47 +04:00
/* transform primitives (ImagingTransformMap) */
static int
affine_transform(double* xout, double* yout, int x, int y, void* data)
2010-07-31 06:52:47 +04:00
{
/* full moon tonight. your compiler will generate bogus code
for simple expressions, unless you reorganize the code, or
install Service Pack 3 */
double* a = (double*) data;
double a0 = a[0]; double a1 = a[1]; double a2 = a[2];
double a3 = a[3]; double a4 = a[4]; double a5 = a[5];
double xin = x + 0.5;
double yin = y + 0.5;
xout[0] = a0*xin + a1*yin + a2;
yout[0] = a3*xin + a4*yin + a5;
2010-07-31 06:52:47 +04:00
return 1;
}
static int
perspective_transform(double* xout, double* yout, int x, int y, void* data)
2010-07-31 06:52:47 +04:00
{
double* a = (double*) data;
double a0 = a[0]; double a1 = a[1]; double a2 = a[2];
double a3 = a[3]; double a4 = a[4]; double a5 = a[5];
double a6 = a[6]; double a7 = a[7];
double xin = x + 0.5;
double yin = y + 0.5;
xout[0] = (a0*xin + a1*yin + a2) / (a6*xin + a7*yin + 1);
yout[0] = (a3*xin + a4*yin + a5) / (a6*xin + a7*yin + 1);
2010-07-31 06:52:47 +04:00
return 1;
}
static int
quad_transform(double* xout, double* yout, int x, int y, void* data)
2010-07-31 06:52:47 +04:00
{
/* quad warp: map quadrilateral to rectangle */
double* a = (double*) data;
double a0 = a[0]; double a1 = a[1]; double a2 = a[2]; double a3 = a[3];
double a4 = a[4]; double a5 = a[5]; double a6 = a[6]; double a7 = a[7];
double xin = x + 0.5;
double yin = y + 0.5;
xout[0] = a0 + a1*xin + a2*yin + a3*xin*yin;
yout[0] = a4 + a5*xin + a6*yin + a7*xin*yin;
2010-07-31 06:52:47 +04:00
return 1;
}
/* transform filters (ImagingTransformFilter) */
static int
2016-06-02 18:33:17 +03:00
nearest_filter8(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
int x = COORD(xin);
int y = COORD(yin);
2010-07-31 06:52:47 +04:00
if (x < 0 || x >= im->xsize || y < 0 || y >= im->ysize)
return 0;
((UINT8*)out)[0] = im->image8[y][x];
return 1;
}
static int
2016-06-02 18:33:17 +03:00
nearest_filter16(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
int x = COORD(xin);
int y = COORD(yin);
2010-07-31 06:52:47 +04:00
if (x < 0 || x >= im->xsize || y < 0 || y >= im->ysize)
return 0;
memcpy(out, im->image8[y] + x * sizeof(INT16), sizeof(INT16));
2010-07-31 06:52:47 +04:00
return 1;
}
static int
2016-06-02 18:33:17 +03:00
nearest_filter32(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
int x = COORD(xin);
int y = COORD(yin);
2010-07-31 06:52:47 +04:00
if (x < 0 || x >= im->xsize || y < 0 || y >= im->ysize)
return 0;
memcpy(out, &im->image32[y][x], sizeof(INT32));
2010-07-31 06:52:47 +04:00
return 1;
}
#define XCLIP(im, x) ( ((x) < 0) ? 0 : ((x) < im->xsize) ? (x) : im->xsize-1 )
#define YCLIP(im, y) ( ((y) < 0) ? 0 : ((y) < im->ysize) ? (y) : im->ysize-1 )
#define BILINEAR(v, a, b, d)\
(v = (a) + ( (b) - (a) ) * (d))
#define BILINEAR_HEAD(type)\
int x, y;\
int x0, x1;\
double v1, v2;\
double dx, dy;\
type* in;\
if (xin < 0.0 || xin >= im->xsize || yin < 0.0 || yin >= im->ysize)\
return 0;\
xin -= 0.5;\
yin -= 0.5;\
x = FLOOR(xin);\
y = FLOOR(yin);\
dx = xin - x;\
dy = yin - y;
#define BILINEAR_BODY(type, image, step, offset) {\
in = (type*) ((image)[YCLIP(im, y)] + offset);\
x0 = XCLIP(im, x+0)*step;\
x1 = XCLIP(im, x+1)*step;\
BILINEAR(v1, in[x0], in[x1], dx);\
if (y+1 >= 0 && y+1 < im->ysize) {\
in = (type*) ((image)[y+1] + offset);\
BILINEAR(v2, in[x0], in[x1], dx);\
} else\
v2 = v1;\
BILINEAR(v1, v1, v2, dy);\
}
static int
2016-06-02 18:33:17 +03:00
bilinear_filter8(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
BILINEAR_HEAD(UINT8);
BILINEAR_BODY(UINT8, im->image8, 1, 0);
((UINT8*)out)[0] = (UINT8) v1;
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bilinear_filter32I(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
INT32 k;
2010-07-31 06:52:47 +04:00
BILINEAR_HEAD(INT32);
BILINEAR_BODY(INT32, im->image32, 1, 0);
k = v1;
memcpy(out, &k, sizeof(k));
2010-07-31 06:52:47 +04:00
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bilinear_filter32F(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
FLOAT32 k;
2010-07-31 06:52:47 +04:00
BILINEAR_HEAD(FLOAT32);
BILINEAR_BODY(FLOAT32, im->image32, 1, 0);
k = v1;
memcpy(out, &k, sizeof(k));
2010-07-31 06:52:47 +04:00
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bilinear_filter32LA(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
BILINEAR_HEAD(UINT8);
BILINEAR_BODY(UINT8, im->image, 4, 0);
((UINT8*)out)[0] = (UINT8) v1;
((UINT8*)out)[1] = (UINT8) v1;
((UINT8*)out)[2] = (UINT8) v1;
BILINEAR_BODY(UINT8, im->image, 4, 3);
((UINT8*)out)[3] = (UINT8) v1;
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bilinear_filter32RGB(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
int b;
BILINEAR_HEAD(UINT8);
for (b = 0; b < im->bands; b++) {
BILINEAR_BODY(UINT8, im->image, 4, b);
((UINT8*)out)[b] = (UINT8) v1;
}
return 1;
}
#undef BILINEAR
#undef BILINEAR_HEAD
#undef BILINEAR_BODY
2010-07-31 06:52:47 +04:00
#define BICUBIC(v, v1, v2, v3, v4, d) {\
double p1 = v2;\
double p2 = -v1 + v3;\
double p3 = 2*(v1 - v2) + v3 - v4;\
double p4 = -v1 + v2 - v3 + v4;\
v = p1 + (d)*(p2 + (d)*(p3 + (d)*p4));\
}
#define BICUBIC_HEAD(type)\
int x = FLOOR(xin);\
int y = FLOOR(yin);\
int x0, x1, x2, x3;\
double v1, v2, v3, v4;\
double dx, dy;\
type* in;\
if (xin < 0.0 || xin >= im->xsize || yin < 0.0 || yin >= im->ysize)\
return 0;\
xin -= 0.5;\
yin -= 0.5;\
x = FLOOR(xin);\
y = FLOOR(yin);\
dx = xin - x;\
dy = yin - y;\
x--; y--;
#define BICUBIC_BODY(type, image, step, offset) {\
in = (type*) ((image)[YCLIP(im, y)] + offset);\
x0 = XCLIP(im, x+0)*step;\
x1 = XCLIP(im, x+1)*step;\
x2 = XCLIP(im, x+2)*step;\
x3 = XCLIP(im, x+3)*step;\
BICUBIC(v1, in[x0], in[x1], in[x2], in[x3], dx);\
if (y+1 >= 0 && y+1 < im->ysize) {\
in = (type*) ((image)[y+1] + offset);\
BICUBIC(v2, in[x0], in[x1], in[x2], in[x3], dx);\
} else\
v2 = v1;\
if (y+2 >= 0 && y+2 < im->ysize) {\
in = (type*) ((image)[y+2] + offset);\
BICUBIC(v3, in[x0], in[x1], in[x2], in[x3], dx);\
} else\
v3 = v2;\
if (y+3 >= 0 && y+3 < im->ysize) {\
in = (type*) ((image)[y+3] + offset);\
BICUBIC(v4, in[x0], in[x1], in[x2], in[x3], dx);\
} else\
v4 = v3;\
BICUBIC(v1, v1, v2, v3, v4, dy);\
}
static int
2016-06-02 18:33:17 +03:00
bicubic_filter8(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
BICUBIC_HEAD(UINT8);
BICUBIC_BODY(UINT8, im->image8, 1, 0);
if (v1 <= 0.0)
((UINT8*)out)[0] = 0;
else if (v1 >= 255.0)
((UINT8*)out)[0] = 255;
else
((UINT8*)out)[0] = (UINT8) v1;
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bicubic_filter32I(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
INT32 k;
2010-07-31 06:52:47 +04:00
BICUBIC_HEAD(INT32);
BICUBIC_BODY(INT32, im->image32, 1, 0);
k = v1;
memcpy(out, &k, sizeof(k));
2010-07-31 06:52:47 +04:00
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bicubic_filter32F(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
FLOAT32 k;
2010-07-31 06:52:47 +04:00
BICUBIC_HEAD(FLOAT32);
BICUBIC_BODY(FLOAT32, im->image32, 1, 0);
k = v1;
memcpy(out, &k, sizeof(k));
2010-07-31 06:52:47 +04:00
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bicubic_filter32LA(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
BICUBIC_HEAD(UINT8);
BICUBIC_BODY(UINT8, im->image, 4, 0);
if (v1 <= 0.0) {
((UINT8*)out)[0] = 0;
((UINT8*)out)[1] = 0;
((UINT8*)out)[2] = 0;
} else if (v1 >= 255.0) {
((UINT8*)out)[0] = 255;
((UINT8*)out)[1] = 255;
((UINT8*)out)[2] = 255;
} else {
((UINT8*)out)[0] = (UINT8) v1;
((UINT8*)out)[1] = (UINT8) v1;
((UINT8*)out)[2] = (UINT8) v1;
}
BICUBIC_BODY(UINT8, im->image, 4, 3);
if (v1 <= 0.0)
((UINT8*)out)[3] = 0;
else if (v1 >= 255.0)
((UINT8*)out)[3] = 255;
else
((UINT8*)out)[3] = (UINT8) v1;
return 1;
}
static int
2016-06-02 18:33:17 +03:00
bicubic_filter32RGB(void* out, Imaging im, double xin, double yin)
2010-07-31 06:52:47 +04:00
{
int b;
BICUBIC_HEAD(UINT8);
for (b = 0; b < im->bands; b++) {
BICUBIC_BODY(UINT8, im->image, 4, b);
if (v1 <= 0.0)
((UINT8*)out)[b] = 0;
else if (v1 >= 255.0)
((UINT8*)out)[b] = 255;
else
((UINT8*)out)[b] = (UINT8) v1;
}
return 1;
}
#undef BICUBIC
#undef BICUBIC_HEAD
#undef BICUBIC_BODY
2010-07-31 06:52:47 +04:00
static ImagingTransformFilter
getfilter(Imaging im, int filterid)
{
switch (filterid) {
case IMAGING_TRANSFORM_NEAREST:
if (im->image8)
switch (im->type) {
case IMAGING_TYPE_UINT8:
2016-06-02 18:33:17 +03:00
return nearest_filter8;
2010-07-31 06:52:47 +04:00
case IMAGING_TYPE_SPECIAL:
switch (im->pixelsize) {
case 1:
2016-06-02 18:33:17 +03:00
return nearest_filter8;
2010-07-31 06:52:47 +04:00
case 2:
2016-06-02 18:33:17 +03:00
return nearest_filter16;
2010-07-31 06:52:47 +04:00
case 4:
2016-06-02 18:33:17 +03:00
return nearest_filter32;
2010-07-31 06:52:47 +04:00
}
}
else
2016-06-02 18:33:17 +03:00
return nearest_filter32;
2010-07-31 06:52:47 +04:00
break;
case IMAGING_TRANSFORM_BILINEAR:
if (im->image8)
2016-06-02 18:33:17 +03:00
return bilinear_filter8;
2010-07-31 06:52:47 +04:00
else if (im->image32) {
switch (im->type) {
case IMAGING_TYPE_UINT8:
if (im->bands == 2)
2016-06-02 18:33:17 +03:00
return bilinear_filter32LA;
2010-07-31 06:52:47 +04:00
else
2016-06-02 18:33:17 +03:00
return bilinear_filter32RGB;
2010-07-31 06:52:47 +04:00
case IMAGING_TYPE_INT32:
2016-06-02 18:33:17 +03:00
return bilinear_filter32I;
2010-07-31 06:52:47 +04:00
case IMAGING_TYPE_FLOAT32:
2016-06-02 18:33:17 +03:00
return bilinear_filter32F;
2010-07-31 06:52:47 +04:00
}
}
break;
case IMAGING_TRANSFORM_BICUBIC:
if (im->image8)
2016-06-02 18:33:17 +03:00
return bicubic_filter8;
2010-07-31 06:52:47 +04:00
else if (im->image32) {
switch (im->type) {
case IMAGING_TYPE_UINT8:
if (im->bands == 2)
2016-06-02 18:33:17 +03:00
return bicubic_filter32LA;
2010-07-31 06:52:47 +04:00
else
2016-06-02 18:33:17 +03:00
return bicubic_filter32RGB;
2010-07-31 06:52:47 +04:00
case IMAGING_TYPE_INT32:
2016-06-02 18:33:17 +03:00
return bicubic_filter32I;
2010-07-31 06:52:47 +04:00
case IMAGING_TYPE_FLOAT32:
2016-06-02 18:33:17 +03:00
return bicubic_filter32F;
2010-07-31 06:52:47 +04:00
}
}
break;
}
/* no such filter */
return NULL;
}
/* transformation engines */
Imaging
ImagingGenericTransform(
Imaging imOut, Imaging imIn, int x0, int y0, int x1, int y1,
2010-07-31 06:52:47 +04:00
ImagingTransformMap transform, void* transform_data,
2016-06-02 18:33:17 +03:00
int filterid, int fill)
2010-07-31 06:52:47 +04:00
{
/* slow generic transformation. use ImagingTransformAffine or
ImagingScaleAffine where possible. */
ImagingSectionCookie cookie;
int x, y;
char *out;
double xx, yy;
ImagingTransformFilter filter = getfilter(imIn, filterid);
if (!filter)
return (Imaging) ImagingError_ValueError("bad filter number");
2010-07-31 06:52:47 +04:00
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
ImagingSectionEnter(&cookie);
if (x0 < 0)
x0 = 0;
if (y0 < 0)
y0 = 0;
if (x1 > imOut->xsize)
x1 = imOut->xsize;
if (y1 > imOut->ysize)
y1 = imOut->ysize;
for (y = y0; y < y1; y++) {
2014-11-07 03:21:38 +03:00
out = imOut->image[y] + x0*imOut->pixelsize;
for (x = x0; x < x1; x++) {
2016-06-02 18:33:17 +03:00
if ( ! transform(&xx, &yy, x-x0, y-y0, transform_data) ||
! filter(out, imIn, xx, yy)) {
2010-07-31 06:52:47 +04:00
if (fill)
memset(out, 0, imOut->pixelsize);
}
out += imOut->pixelsize;
2014-11-07 03:21:38 +03:00
}
2010-07-31 06:52:47 +04:00
}
ImagingSectionLeave(&cookie);
return imOut;
}
static Imaging
ImagingScaleAffine(Imaging imOut, Imaging imIn,
int x0, int y0, int x1, int y1,
double a[6], int fill)
{
/* scale, nearest neighbour resampling */
ImagingSectionCookie cookie;
int x, y;
int xin;
double xo, yo;
int xmin, xmax;
int *xintab;
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
if (x0 < 0)
x0 = 0;
if (y0 < 0)
y0 = 0;
if (x1 > imOut->xsize)
x1 = imOut->xsize;
if (y1 > imOut->ysize)
y1 = imOut->ysize;
2016-09-03 05:23:42 +03:00
2016-03-16 14:47:18 +03:00
/* malloc check ok, uses calloc for overflow */
xintab = (int*) calloc(imOut->xsize, sizeof(int));
2010-07-31 06:52:47 +04:00
if (!xintab) {
2014-11-07 03:21:38 +03:00
ImagingDelete(imOut);
return (Imaging) ImagingError_MemoryError();
2010-07-31 06:52:47 +04:00
}
xo = a[2] + a[0] * 0.5;
yo = a[5] + a[4] * 0.5;
2010-07-31 06:52:47 +04:00
xmin = x1;
xmax = x0;
/* Pretabulate horizontal pixel positions */
for (x = x0; x < x1; x++) {
2014-11-07 03:21:38 +03:00
xin = COORD(xo);
if (xin >= 0 && xin < (int) imIn->xsize) {
xmax = x+1;
if (x < xmin)
xmin = x;
xintab[x] = xin;
}
xo += a[0];
2010-07-31 06:52:47 +04:00
}
2014-11-07 03:21:38 +03:00
#define AFFINE_SCALE(pixel, image)\
2010-07-31 06:52:47 +04:00
for (y = y0; y < y1; y++) {\
2014-11-07 03:21:38 +03:00
int yi = COORD(yo);\
pixel *in, *out;\
out = imOut->image[y];\
2010-07-31 06:52:47 +04:00
if (fill && x1 > x0)\
memset(out+x0, 0, (x1-x0)*sizeof(pixel));\
2014-11-07 03:21:38 +03:00
if (yi >= 0 && yi < imIn->ysize) {\
in = imIn->image[yi];\
for (x = xmin; x < xmax; x++)\
out[x] = in[xintab[x]];\
}\
yo += a[4];\
2010-07-31 06:52:47 +04:00
}
ImagingSectionEnter(&cookie);
if (imIn->image8) {
AFFINE_SCALE(UINT8, image8);
} else {
AFFINE_SCALE(INT32, image32);
}
ImagingSectionLeave(&cookie);
#undef AFFINE_SCALE
2010-07-31 06:52:47 +04:00
free(xintab);
return imOut;
}
static inline int
check_fixed(double a[6], int x, int y)
{
return (fabs(x*a[0] + y*a[1] + a[2]) < 32768.0 &&
fabs(x*a[3] + y*a[4] + a[5]) < 32768.0);
2010-07-31 06:52:47 +04:00
}
static inline Imaging
affine_fixed(Imaging imOut, Imaging imIn,
int x0, int y0, int x1, int y1,
double a[6], int filterid, int fill)
{
/* affine transform, nearest neighbour resampling, fixed point
arithmetics */
ImagingSectionCookie cookie;
2010-07-31 06:52:47 +04:00
int x, y;
int xin, yin;
int xsize, ysize;
int xx, yy;
int a0, a1, a2, a3, a4, a5;
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
xsize = (int) imIn->xsize;
ysize = (int) imIn->ysize;
/* use 16.16 fixed point arithmetics */
#define FIX(v) FLOOR((v)*65536.0 + 0.5)
2016-09-03 05:19:04 +03:00
a0 = FIX(a[0]); a1 = FIX(a[1]);
a3 = FIX(a[3]); a4 = FIX(a[4]);
a2 = FIX(a[2] + a[0] * 0.5 + a[1] * 0.5);
a5 = FIX(a[5] + a[3] * 0.5 + a[4] * 0.5);
2010-07-31 06:52:47 +04:00
#undef FIX
2014-11-07 03:21:38 +03:00
#define AFFINE_TRANSFORM_FIXED(pixel, image)\
2010-07-31 06:52:47 +04:00
for (y = y0; y < y1; y++) {\
2014-11-07 03:21:38 +03:00
pixel *out;\
xx = a2;\
yy = a5;\
2014-11-07 03:21:38 +03:00
out = imOut->image[y];\
2010-07-31 06:52:47 +04:00
if (fill && x1 > x0)\
memset(out+x0, 0, (x1-x0)*sizeof(pixel));\
for (x = x0; x < x1; x++, out++) {\
2014-11-07 03:21:38 +03:00
xin = xx >> 16;\
if (xin >= 0 && xin < xsize) {\
yin = yy >> 16;\
if (yin >= 0 && yin < ysize)\
2010-07-31 06:52:47 +04:00
*out = imIn->image[yin][xin];\
}\
xx += a0;\
yy += a3;\
2014-11-07 03:21:38 +03:00
}\
a2 += a1;\
a5 += a4;\
2010-07-31 06:52:47 +04:00
}
ImagingSectionEnter(&cookie);
2010-07-31 06:52:47 +04:00
if (imIn->image8)
2014-11-07 03:21:38 +03:00
AFFINE_TRANSFORM_FIXED(UINT8, image8)
2010-07-31 06:52:47 +04:00
else
2014-11-07 03:21:38 +03:00
AFFINE_TRANSFORM_FIXED(INT32, image32)
2010-07-31 06:52:47 +04:00
ImagingSectionLeave(&cookie);
#undef AFFINE_TRANSFORM_FIXED
2010-07-31 06:52:47 +04:00
return imOut;
}
Imaging
ImagingTransformAffine(Imaging imOut, Imaging imIn,
int x0, int y0, int x1, int y1,
double a[6], int filterid, int fill)
{
/* affine transform, nearest neighbour resampling, floating point
arithmetics*/
ImagingSectionCookie cookie;
int x, y;
int xin, yin;
int xsize, ysize;
double xx, yy;
double xo, yo;
if (filterid || imIn->type == IMAGING_TYPE_SPECIAL) {
return ImagingGenericTransform(
2010-07-31 06:52:47 +04:00
imOut, imIn,
x0, y0, x1, y1,
affine_transform, a,
2016-06-02 18:33:17 +03:00
filterid, fill);
2010-07-31 06:52:47 +04:00
}
if (a[1] == 0 && a[3] == 0) {
2014-11-07 03:21:38 +03:00
/* Scaling */
return ImagingScaleAffine(imOut, imIn, x0, y0, x1, y1, a, fill);
}
2010-07-31 06:52:47 +04:00
if (!imOut || !imIn || strcmp(imIn->mode, imOut->mode) != 0)
2014-11-07 03:21:38 +03:00
return (Imaging) ImagingError_ModeError();
2010-07-31 06:52:47 +04:00
if (x0 < 0)
x0 = 0;
if (y0 < 0)
y0 = 0;
if (x1 > imOut->xsize)
x1 = imOut->xsize;
if (y1 > imOut->ysize)
y1 = imOut->ysize;
/* translate all four corners to check if they are within the
range that can be represented by the fixed point arithmetics */
if (check_fixed(a, 0, 0) && check_fixed(a, x1-x0, y1-y0) &&
check_fixed(a, 0, y1-y0) && check_fixed(a, x1-x0, 0))
return affine_fixed(imOut, imIn, x0, y0, x1, y1, a, filterid, fill);
/* FIXME: cannot really think of any reasonable case when the
following code is used. maybe we should fall back on the slow
generic transform engine in this case? */
2016-09-03 05:23:42 +03:00
ImagingCopyPalette(imOut, imIn);
2010-07-31 06:52:47 +04:00
xsize = (int) imIn->xsize;
ysize = (int) imIn->ysize;
xo = a[2] + a[1] * 0.5 + a[0] * 0.5;
yo = a[5] + a[4] * 0.5 + a[3] * 0.5;
2010-07-31 06:52:47 +04:00
2014-11-07 03:21:38 +03:00
#define AFFINE_TRANSFORM(pixel, image)\
2010-07-31 06:52:47 +04:00
for (y = y0; y < y1; y++) {\
2014-11-07 03:21:38 +03:00
pixel *out;\
xx = xo;\
yy = yo;\
out = imOut->image[y];\
2010-07-31 06:52:47 +04:00
if (fill && x1 > x0)\
memset(out+x0, 0, (x1-x0)*sizeof(pixel));\
for (x = x0; x < x1; x++, out++) {\
2014-11-07 03:21:38 +03:00
xin = COORD(xx);\
if (xin >= 0 && xin < xsize) {\
yin = COORD(yy);\
if (yin >= 0 && yin < ysize)\
2010-07-31 06:52:47 +04:00
*out = imIn->image[yin][xin];\
}\
xx += a[0];\
yy += a[3];\
2014-11-07 03:21:38 +03:00
}\
xo += a[1];\
yo += a[4];\
2010-07-31 06:52:47 +04:00
}
ImagingSectionEnter(&cookie);
if (imIn->image8)
2014-11-07 03:21:38 +03:00
AFFINE_TRANSFORM(UINT8, image8)
2010-07-31 06:52:47 +04:00
else
2014-11-07 03:21:38 +03:00
AFFINE_TRANSFORM(INT32, image32)
2010-07-31 06:52:47 +04:00
ImagingSectionLeave(&cookie);
#undef AFFINE_TRANSFORM
2010-07-31 06:52:47 +04:00
return imOut;
}
Imaging
ImagingTransform(Imaging imOut, Imaging imIn, int method,
int x0, int y0, int x1, int y1,
double a[8], int filterid, int fill)
2010-07-31 06:52:47 +04:00
{
ImagingTransformMap transform;
2010-07-31 06:52:47 +04:00
switch(method) {
case IMAGING_TRANSFORM_AFFINE:
return ImagingTransformAffine(
imOut, imIn, x0, y0, x1, y1, a, filterid, fill);
break;
case IMAGING_TRANSFORM_PERSPECTIVE:
transform = perspective_transform;
break;
case IMAGING_TRANSFORM_QUAD:
transform = quad_transform;
break;
default:
return (Imaging) ImagingError_ValueError("bad transform method");
}
2010-07-31 06:52:47 +04:00
return ImagingGenericTransform(
2010-07-31 06:52:47 +04:00
imOut, imIn,
x0, y0, x1, y1,
2016-06-02 18:33:17 +03:00
transform, a, filterid, fill);
2010-07-31 06:52:47 +04:00
}