mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-22 11:58:55 +03:00
924 lines
24 KiB
C
924 lines
24 KiB
C
/*
|
|
* The Python Imaging Library.
|
|
*
|
|
* standard decoder interfaces for the Imaging library
|
|
*
|
|
* history:
|
|
* 1996-03-28 fl Moved from _imagingmodule.c
|
|
* 1996-04-15 fl Support subregions in setimage
|
|
* 1996-04-19 fl Allocate decoder buffer (where appropriate)
|
|
* 1996-05-02 fl Added jpeg decoder
|
|
* 1996-05-12 fl Compile cleanly as C++
|
|
* 1996-05-16 fl Added hex decoder
|
|
* 1996-05-26 fl Added jpeg configuration parameters
|
|
* 1996-12-14 fl Added zip decoder
|
|
* 1996-12-30 fl Plugged potential memory leak for tiled images
|
|
* 1997-01-03 fl Added fli and msp decoders
|
|
* 1997-01-04 fl Added sun_rle and tga_rle decoders
|
|
* 1997-05-31 fl Added bitfield decoder
|
|
* 1998-09-11 fl Added orientation and pixelsize fields to tga_rle decoder
|
|
* 1998-12-29 fl Added mode/rawmode argument to decoders
|
|
* 1998-12-30 fl Added mode argument to *all* decoders
|
|
* 2002-06-09 fl Added stride argument to pcx decoder
|
|
*
|
|
* Copyright (c) 1997-2002 by Secret Labs AB.
|
|
* Copyright (c) 1995-2002 by Fredrik Lundh.
|
|
*
|
|
* See the README file for information on usage and redistribution.
|
|
*/
|
|
|
|
/* FIXME: make these pluggable! */
|
|
|
|
#define PY_SSIZE_T_CLEAN
|
|
#include "Python.h"
|
|
|
|
#include "libImaging/Imaging.h"
|
|
|
|
#include "libImaging/Gif.h"
|
|
#include "libImaging/Raw.h"
|
|
#include "libImaging/Bit.h"
|
|
#include "libImaging/Sgi.h"
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* Common */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
typedef struct {
|
|
PyObject_HEAD int (*decode)(
|
|
Imaging im, ImagingCodecState state, UINT8 *buffer, Py_ssize_t bytes);
|
|
int (*cleanup)(ImagingCodecState state);
|
|
struct ImagingCodecStateInstance state;
|
|
Imaging im;
|
|
PyObject *lock;
|
|
int pulls_fd;
|
|
} ImagingDecoderObject;
|
|
|
|
static PyTypeObject ImagingDecoderType;
|
|
|
|
static ImagingDecoderObject *
|
|
PyImaging_DecoderNew(int contextsize) {
|
|
ImagingDecoderObject *decoder;
|
|
void *context;
|
|
|
|
if (PyType_Ready(&ImagingDecoderType) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyObject_New(ImagingDecoderObject, &ImagingDecoderType);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
/* Clear the decoder state */
|
|
memset(&decoder->state, 0, sizeof(decoder->state));
|
|
|
|
/* Allocate decoder context */
|
|
if (contextsize > 0) {
|
|
context = (void *)calloc(1, contextsize);
|
|
if (!context) {
|
|
Py_DECREF(decoder);
|
|
(void)ImagingError_MemoryError();
|
|
return NULL;
|
|
}
|
|
} else {
|
|
context = 0;
|
|
}
|
|
|
|
/* Initialize decoder context */
|
|
decoder->state.context = context;
|
|
|
|
/* Target image */
|
|
decoder->lock = NULL;
|
|
decoder->im = NULL;
|
|
|
|
/* Initialize the cleanup function pointer */
|
|
decoder->cleanup = NULL;
|
|
|
|
/* set if the decoder needs to pull data from the fd, instead of
|
|
having it pushed */
|
|
decoder->pulls_fd = 0;
|
|
|
|
return decoder;
|
|
}
|
|
|
|
static void
|
|
_dealloc(ImagingDecoderObject *decoder) {
|
|
if (decoder->cleanup) {
|
|
decoder->cleanup(&decoder->state);
|
|
}
|
|
free(decoder->state.buffer);
|
|
free(decoder->state.context);
|
|
Py_XDECREF(decoder->lock);
|
|
Py_XDECREF(decoder->state.fd);
|
|
PyObject_Del(decoder);
|
|
}
|
|
|
|
static PyObject *
|
|
_decode(ImagingDecoderObject *decoder, PyObject *args) {
|
|
UINT8 *buffer;
|
|
Py_ssize_t bufsize;
|
|
int status;
|
|
ImagingSectionCookie cookie;
|
|
|
|
if (!PyArg_ParseTuple(args, "y#", &buffer, &bufsize)) {
|
|
return NULL;
|
|
}
|
|
|
|
if (!decoder->pulls_fd) {
|
|
ImagingSectionEnter(&cookie);
|
|
}
|
|
|
|
status = decoder->decode(decoder->im, &decoder->state, buffer, bufsize);
|
|
|
|
if (!decoder->pulls_fd) {
|
|
ImagingSectionLeave(&cookie);
|
|
}
|
|
|
|
return Py_BuildValue("ii", status, decoder->state.errcode);
|
|
}
|
|
|
|
static PyObject *
|
|
_decode_cleanup(ImagingDecoderObject *decoder, PyObject *args) {
|
|
int status = 0;
|
|
|
|
if (decoder->cleanup) {
|
|
status = decoder->cleanup(&decoder->state);
|
|
}
|
|
|
|
return Py_BuildValue("i", status);
|
|
}
|
|
|
|
extern Imaging
|
|
PyImaging_AsImaging(PyObject *op);
|
|
|
|
static PyObject *
|
|
_setimage(ImagingDecoderObject *decoder, PyObject *args) {
|
|
PyObject *op;
|
|
Imaging im;
|
|
ImagingCodecState state;
|
|
int x0, y0, x1, y1;
|
|
|
|
x0 = y0 = x1 = y1 = 0;
|
|
|
|
/* FIXME: should publish the ImagingType descriptor */
|
|
if (!PyArg_ParseTuple(args, "O|(iiii)", &op, &x0, &y0, &x1, &y1)) {
|
|
return NULL;
|
|
}
|
|
im = PyImaging_AsImaging(op);
|
|
if (!im) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->im = im;
|
|
|
|
state = &decoder->state;
|
|
|
|
/* Setup decoding tile extent */
|
|
if (x0 == 0 && x1 == 0) {
|
|
state->xsize = im->xsize;
|
|
state->ysize = im->ysize;
|
|
} else {
|
|
state->xoff = x0;
|
|
state->yoff = y0;
|
|
state->xsize = x1 - x0;
|
|
state->ysize = y1 - y0;
|
|
}
|
|
|
|
if (state->xsize <= 0 || state->xsize + state->xoff > (int)im->xsize ||
|
|
state->ysize <= 0 || state->ysize + state->yoff > (int)im->ysize) {
|
|
PyErr_SetString(PyExc_ValueError, "tile cannot extend outside image");
|
|
return NULL;
|
|
}
|
|
|
|
/* Allocate memory buffer (if bits field is set) */
|
|
if (state->bits > 0) {
|
|
if (!state->bytes) {
|
|
if (state->xsize > ((INT_MAX / state->bits) - 7)) {
|
|
return ImagingError_MemoryError();
|
|
}
|
|
state->bytes = (state->bits * state->xsize + 7) / 8;
|
|
}
|
|
/* malloc check ok, overflow checked above */
|
|
state->buffer = (UINT8 *)malloc(state->bytes);
|
|
if (!state->buffer) {
|
|
return ImagingError_MemoryError();
|
|
}
|
|
}
|
|
|
|
/* Keep a reference to the image object, to make sure it doesn't
|
|
go away before we do */
|
|
Py_INCREF(op);
|
|
Py_XDECREF(decoder->lock);
|
|
decoder->lock = op;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *
|
|
_setfd(ImagingDecoderObject *decoder, PyObject *args) {
|
|
PyObject *fd;
|
|
ImagingCodecState state;
|
|
|
|
if (!PyArg_ParseTuple(args, "O", &fd)) {
|
|
return NULL;
|
|
}
|
|
|
|
state = &decoder->state;
|
|
|
|
Py_XINCREF(fd);
|
|
state->fd = fd;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *
|
|
_get_pulls_fd(ImagingDecoderObject *decoder) {
|
|
return PyBool_FromLong(decoder->pulls_fd);
|
|
}
|
|
|
|
static struct PyMethodDef methods[] = {
|
|
{"decode", (PyCFunction)_decode, 1},
|
|
{"cleanup", (PyCFunction)_decode_cleanup, 1},
|
|
{"setimage", (PyCFunction)_setimage, 1},
|
|
{"setfd", (PyCFunction)_setfd, 1},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
static struct PyGetSetDef getseters[] = {
|
|
{"pulls_fd",
|
|
(getter)_get_pulls_fd,
|
|
NULL,
|
|
"True if this decoder expects to pull from self.fd itself.",
|
|
NULL},
|
|
{NULL, NULL, NULL, NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
static PyTypeObject ImagingDecoderType = {
|
|
PyVarObject_HEAD_INIT(NULL, 0) "ImagingDecoder", /*tp_name*/
|
|
sizeof(ImagingDecoderObject), /*tp_size*/
|
|
0, /*tp_itemsize*/
|
|
/* methods */
|
|
(destructor)_dealloc, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
0, /*tp_compare*/
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number */
|
|
0, /*tp_as_sequence */
|
|
0, /*tp_as_mapping */
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT, /*tp_flags*/
|
|
0, /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
methods, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
getseters, /*tp_getset*/
|
|
};
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
int
|
|
get_unpacker(ImagingDecoderObject *decoder, const char *mode, const char *rawmode) {
|
|
int bits;
|
|
ImagingShuffler unpack;
|
|
|
|
unpack = ImagingFindUnpacker(mode, rawmode, &bits);
|
|
if (!unpack) {
|
|
Py_DECREF(decoder);
|
|
PyErr_SetString(PyExc_ValueError, "unknown raw mode");
|
|
return -1;
|
|
}
|
|
|
|
decoder->state.shuffle = unpack;
|
|
decoder->state.bits = bits;
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* BIT (packed fields) */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_BitDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
int bits = 8;
|
|
int pad = 8;
|
|
int fill = 0;
|
|
int sign = 0;
|
|
int ystep = 1;
|
|
if (!PyArg_ParseTuple(args, "s|iiiii", &mode, &bits, &pad, &fill, &sign, &ystep)) {
|
|
return NULL;
|
|
}
|
|
|
|
if (strcmp(mode, "F") != 0) {
|
|
PyErr_SetString(PyExc_ValueError, "bad image mode");
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(BITSTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingBitDecode;
|
|
|
|
decoder->state.ystep = ystep;
|
|
|
|
((BITSTATE *)decoder->state.context)->bits = bits;
|
|
((BITSTATE *)decoder->state.context)->pad = pad;
|
|
((BITSTATE *)decoder->state.context)->fill = fill;
|
|
((BITSTATE *)decoder->state.context)->sign = sign;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* BCn: GPU block-compressed texture formats */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_BcnDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *actual;
|
|
int n = 0;
|
|
int ystep = 1;
|
|
if (!PyArg_ParseTuple(args, "s|ii", &mode, &n, &ystep)) {
|
|
return NULL;
|
|
}
|
|
|
|
switch (n) {
|
|
case 1: /* BC1: 565 color, 1-bit alpha */
|
|
case 2: /* BC2: 565 color, 4-bit alpha */
|
|
case 3: /* BC3: 565 color, 2-endpoint 8-bit interpolated alpha */
|
|
case 5: /* BC5: 2-channel 8-bit via 2 BC3 alpha blocks */
|
|
case 7: /* BC7: 4-channel 8-bit via everything */
|
|
actual = "RGBA";
|
|
break;
|
|
case 4: /* BC4: 1-channel 8-bit via 1 BC3 alpha block */
|
|
actual = "L";
|
|
break;
|
|
case 6: /* BC6: 3-channel 16-bit float */
|
|
/* TODO: support 4-channel floating point images */
|
|
actual = "RGBAF";
|
|
break;
|
|
default:
|
|
PyErr_SetString(PyExc_ValueError, "block compression type unknown");
|
|
return NULL;
|
|
}
|
|
|
|
if (strcmp(mode, actual) != 0) {
|
|
PyErr_SetString(PyExc_ValueError, "bad image mode");
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingBcnDecode;
|
|
decoder->state.state = n;
|
|
decoder->state.ystep = ystep;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* FLI */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_FliDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingFliDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* GIF */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_GifDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
int bits = 8;
|
|
int interlace = 0;
|
|
if (!PyArg_ParseTuple(args, "s|ii", &mode, &bits, &interlace)) {
|
|
return NULL;
|
|
}
|
|
|
|
if (strcmp(mode, "L") != 0 && strcmp(mode, "P") != 0) {
|
|
PyErr_SetString(PyExc_ValueError, "bad image mode");
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(GIFDECODERSTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingGifDecode;
|
|
|
|
((GIFDECODERSTATE *)decoder->state.context)->bits = bits;
|
|
((GIFDECODERSTATE *)decoder->state.context)->interlace = interlace;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* HEX */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_HexDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
if (!PyArg_ParseTuple(args, "ss", &mode, &rawmode)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingHexDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* LibTiff */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
#ifdef HAVE_LIBTIFF
|
|
|
|
#include "libImaging/TiffDecode.h"
|
|
|
|
#include <string.h>
|
|
|
|
PyObject *
|
|
PyImaging_LibTiffDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
char *mode;
|
|
char *rawmode;
|
|
char *compname;
|
|
int fp;
|
|
uint32 ifdoffset;
|
|
|
|
if (!PyArg_ParseTuple(args, "sssiI", &mode, &rawmode, &compname, &fp, &ifdoffset)) {
|
|
return NULL;
|
|
}
|
|
|
|
TRACE(("new tiff decoder %s\n", compname));
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(TIFFSTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
if (!ImagingLibTiffInit(&decoder->state, fp, ifdoffset)) {
|
|
Py_DECREF(decoder);
|
|
PyErr_SetString(PyExc_RuntimeError, "tiff codec initialization failed");
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingLibTiffDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
#endif
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* PackBits */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_PackbitsDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
if (!PyArg_ParseTuple(args, "ss", &mode, &rawmode)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingPackbitsDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* PCD */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_PcdDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
/* Unpack from PhotoYCC to RGB */
|
|
if (get_unpacker(decoder, "RGB", "YCC;P") < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingPcdDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* PCX */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_PcxDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
int stride;
|
|
if (!PyArg_ParseTuple(args, "ssi", &mode, &rawmode, &stride)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->state.bytes = stride;
|
|
|
|
decoder->decode = ImagingPcxDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* RAW */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_RawDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
int stride = 0;
|
|
int ystep = 1;
|
|
if (!PyArg_ParseTuple(args, "ss|ii", &mode, &rawmode, &stride, &ystep)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(RAWSTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingRawDecode;
|
|
|
|
decoder->state.ystep = ystep;
|
|
|
|
((RAWSTATE *)decoder->state.context)->stride = stride;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* SGI RLE */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_SgiRleDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
int ystep = 1;
|
|
int bpc = 1;
|
|
if (!PyArg_ParseTuple(args, "ss|ii", &mode, &rawmode, &ystep, &bpc)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(SGISTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->pulls_fd = 1;
|
|
decoder->decode = ImagingSgiRleDecode;
|
|
decoder->state.ystep = ystep;
|
|
|
|
((SGISTATE *)decoder->state.context)->bpc = bpc;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* SUN RLE */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_SunRleDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
if (!PyArg_ParseTuple(args, "ss", &mode, &rawmode)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingSunRleDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* TGA RLE */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_TgaRleDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
int ystep = 1;
|
|
int depth = 8;
|
|
if (!PyArg_ParseTuple(args, "ss|ii", &mode, &rawmode, &ystep, &depth)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingTgaRleDecode;
|
|
|
|
decoder->state.ystep = ystep;
|
|
decoder->state.count = depth / 8;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* XBM */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
PyObject *
|
|
PyImaging_XbmDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
decoder = PyImaging_DecoderNew(0);
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, "1", "1;R") < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingXbmDecode;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* ZIP */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
#ifdef HAVE_LIBZ
|
|
|
|
#include "libImaging/ZipCodecs.h"
|
|
|
|
PyObject *
|
|
PyImaging_ZipDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode;
|
|
int interlaced = 0;
|
|
if (!PyArg_ParseTuple(args, "ss|i", &mode, &rawmode, &interlaced)) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(ZIPSTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingZipDecode;
|
|
decoder->cleanup = ImagingZipDecodeCleanup;
|
|
|
|
((ZIPSTATE *)decoder->state.context)->interlaced = interlaced;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
#endif
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* JPEG */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
#ifdef HAVE_LIBJPEG
|
|
|
|
/* We better define this decoder last in this file, so the following
|
|
undef's won't mess things up for the Imaging library proper. */
|
|
|
|
#undef HAVE_PROTOTYPES
|
|
#undef HAVE_STDDEF_H
|
|
#undef HAVE_STDLIB_H
|
|
#undef UINT8
|
|
#undef UINT16
|
|
#undef UINT32
|
|
#undef INT8
|
|
#undef INT16
|
|
#undef INT32
|
|
|
|
#include "libImaging/Jpeg.h"
|
|
|
|
PyObject *
|
|
PyImaging_JpegDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
|
|
char *mode;
|
|
char *rawmode; /* what we want from the decoder */
|
|
char *jpegmode; /* what's in the file */
|
|
int scale = 1;
|
|
int draft = 0;
|
|
|
|
if (!PyArg_ParseTuple(args, "ssz|ii", &mode, &rawmode, &jpegmode, &scale, &draft)) {
|
|
return NULL;
|
|
}
|
|
|
|
if (!jpegmode) {
|
|
jpegmode = "";
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(JPEGSTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
// libjpeg-turbo supports different output formats.
|
|
// We are choosing Pillow's native format (3 color bytes + 1 padding)
|
|
// to avoid extra conversion in Unpack.c.
|
|
if (ImagingJpegUseJCSExtensions() && strcmp(rawmode, "RGB") == 0) {
|
|
rawmode = "RGBX";
|
|
}
|
|
|
|
if (get_unpacker(decoder, mode, rawmode) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->decode = ImagingJpegDecode;
|
|
decoder->cleanup = ImagingJpegDecodeCleanup;
|
|
|
|
strncpy(((JPEGSTATE *)decoder->state.context)->rawmode, rawmode, 8);
|
|
strncpy(((JPEGSTATE *)decoder->state.context)->jpegmode, jpegmode, 8);
|
|
|
|
((JPEGSTATE *)decoder->state.context)->scale = scale;
|
|
((JPEGSTATE *)decoder->state.context)->draft = draft;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
#endif
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/* JPEG 2000 */
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
#ifdef HAVE_OPENJPEG
|
|
|
|
#include "libImaging/Jpeg2K.h"
|
|
|
|
PyObject *
|
|
PyImaging_Jpeg2KDecoderNew(PyObject *self, PyObject *args) {
|
|
ImagingDecoderObject *decoder;
|
|
JPEG2KDECODESTATE *context;
|
|
|
|
char *mode;
|
|
char *format;
|
|
OPJ_CODEC_FORMAT codec_format;
|
|
int reduce = 0;
|
|
int layers = 0;
|
|
int fd = -1;
|
|
PY_LONG_LONG length = -1;
|
|
|
|
if (!PyArg_ParseTuple(
|
|
args, "ss|iiiL", &mode, &format, &reduce, &layers, &fd, &length)) {
|
|
return NULL;
|
|
}
|
|
|
|
if (strcmp(format, "j2k") == 0) {
|
|
codec_format = OPJ_CODEC_J2K;
|
|
} else if (strcmp(format, "jpt") == 0) {
|
|
codec_format = OPJ_CODEC_JPT;
|
|
} else if (strcmp(format, "jp2") == 0) {
|
|
codec_format = OPJ_CODEC_JP2;
|
|
} else {
|
|
return NULL;
|
|
}
|
|
|
|
decoder = PyImaging_DecoderNew(sizeof(JPEG2KDECODESTATE));
|
|
if (decoder == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
decoder->pulls_fd = 1;
|
|
decoder->decode = ImagingJpeg2KDecode;
|
|
decoder->cleanup = ImagingJpeg2KDecodeCleanup;
|
|
|
|
context = (JPEG2KDECODESTATE *)decoder->state.context;
|
|
|
|
context->fd = fd;
|
|
context->length = (off_t)length;
|
|
context->format = codec_format;
|
|
context->reduce = reduce;
|
|
context->layers = layers;
|
|
|
|
return (PyObject *)decoder;
|
|
}
|
|
#endif /* HAVE_OPENJPEG */
|