mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-11 12:17:14 +03:00
f79379ed7c
The file `libImaging/Imaging.h` gets installed flat on Ubuntus into the public Python includes. When building a newer version of Pillow from source and hinting the Python includes "too early" in includes, e.g. in package managers, this can confuse the two files and pick up the external file over the internal one. With different versions, this mismatch can lead to build errors, e.g. undefined macros. The most robust way to avoid this is to pre-fix the internal include accordingly, so that the relative path to the including file has to match as well.
971 lines
24 KiB
C
971 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 "Gif.h"
|
|
#include "Raw.h"
|
|
#include "Bit.h"
|
|
#include "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) PyErr_NoMemory();
|
|
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 PyErr_NoMemory();
|
|
}
|
|
state->bytes = (state->bits * state->xsize+7)/8;
|
|
}
|
|
/* malloc check ok, overflow checked above */
|
|
state->buffer = (UINT8*) malloc(state->bytes);
|
|
if (!state->buffer) {
|
|
return PyErr_NoMemory();
|
|
}
|
|
}
|
|
|
|
/* 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 "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 "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 "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 "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 */
|
|
|