Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Port lzx plugin to build on python2/python3

Browse Source
This commit is contained in:
Eli Schwartz 2018-12-17 12:38:20 -05:00
parent fdcd6d746c
commit e20566c7fc
2 changed files with 92 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

103
src/calibre/utils/lzx/compressor.c
View File

@ -67,7 +67,7 @@ static void
Compressor_dealloc(Compressor *self)
{
Compressor_clear(self);
if (self->stream) {
lzxc_finish(self->stream, NULL);
self->stream = NULL;
@ -81,7 +81,7 @@ Compressor_dealloc(Compressor *self)
self->output.data = NULL;
}
self->ob_type->tp_free((PyObject *)self);
Py_TYPE(self)->tp_free((PyObject *)self);
}
static PyObject *
@ -99,17 +99,17 @@ Compressor_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
self->wbits = 0;
self->blocksize = 0;
self->flushing = 0;
BUFFER_INIT(self->residue);
BUFFER_INIT(self->input);
BUFFER_INIT(self->output);
}
return (PyObject *)self;
}
static int
get_bytes(void *context, int nbytes, void *buf)
get_bytes(void *context, int nbytes, void *buf)
{
Compressor *self = (Compressor *)context;
unsigned char *data = (unsigned char *)buf;
@ -130,7 +130,7 @@ get_bytes(void *context, int nbytes, void *buf)
nbytes -= resrem;
}
}
if (inrem == 0) {
return resrem;
} else if (nbytes > inrem) {
@ -143,7 +143,7 @@ get_bytes(void *context, int nbytes, void *buf)
}
static int
at_eof(void *context)
at_eof(void *context)
{
Compressor *self = (Compressor *)context;
return (self->flushing && (COMPRESSOR_REMAINING(self) == 0));
@ -190,10 +190,10 @@ Compressor_init(Compressor *self, PyObject *args, PyObject *kwds)
int retval = 0;
self->reset = 1;
if (!PyArg_ParseTupleAndKeywords(
args, kwds, "I|b", kwlist, &wbits, &self->reset)) {
return -1;
return -1;
}
/* TODO: check window size. */
@ -216,7 +216,7 @@ Compressor_init(Compressor *self, PyObject *args, PyObject *kwds)
PyErr_SetString(LZXError, "Failed to create compression stream");
return -1;
}
return 0;
}
@ -234,12 +234,12 @@ Compressor_compress__(
PyObject *cdata = NULL;
PyObject *rtable = NULL;
PyObject *result = NULL;
self->flushing = flush;
input->data = data;
input->size = inlen;
input->offset = 0;
outlen = inlen;
remainder = outlen % blocksize;
if (remainder != 0) {
@ -288,7 +288,7 @@ Compressor_compress__(
self->rtable = rtable;
return NULL;
}
cdata = PyString_FromStringAndSize(output->data, output->offset);
cdata = PyBytes_FromStringAndSize(output->data, output->offset);
if (cdata == NULL) {
Py_DECREF(rtable);
return NULL;
@ -333,43 +333,42 @@ static PyMethodDef Compressor_methods[] = {
};
PyTypeObject CompressorType = {
PyObject_HEAD_INIT(NULL)
0, /*ob_size*/
"lzx.Compressor", /*tp_name*/
sizeof(Compressor), /*tp_basicsize*/
0, /*tp_itemsize*/
(destructor)Compressor_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 | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, /*tp_flags*/
"Compressor objects", /* tp_doc */
(traverseproc)Compressor_traverse, /* tp_traverse */
(inquiry)Compressor_clear, /* tp_clear */
0, /* tp_richcompare */
0, /* tp_weaklistoffset */
0, /* tp_iter */
0, /* tp_iternext */
Compressor_methods, /* tp_methods */
Compressor_members, /* tp_members */
0, /* tp_getset */
0, /* tp_base */
0, /* tp_dict */
0, /* tp_descr_get */
0, /* tp_descr_set */
0, /* tp_dictoffset */
(initproc)Compressor_init, /* tp_init */
0, /* tp_alloc */
Compressor_new, /* tp_new */
PyVarObject_HEAD_INIT(NULL, 0)
/* tp_name */ "lzx.Compressor",
/* tp_basicsize */ sizeof(Compressor),
/* tp_itemsize */ 0,
/* tp_dealloc */ (destructor)Compressor_dealloc,
/* 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 */ 0,
/* tp_flags */ Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,
/* tp_doc */ "Compressor objects",
/* tp_traverse */ (traverseproc)Compressor_traverse,
/* tp_clear */ (inquiry)Compressor_clear,
/* tp_richcompare */ 0,
/* tp_weaklistoffset */ 0,
/* tp_iter */ 0,
/* tp_iternext */ 0,
/* tp_methods */ Compressor_methods,
/* tp_members */ Compressor_members,
/* tp_getset */ 0,
/* tp_base */ 0,
/* tp_dict */ 0,
/* tp_descr_get */ 0,
/* tp_descr_set */ 0,
/* tp_dictoffset */ 0,
/* tp_init */ (initproc)Compressor_init,
/* tp_alloc */ 0,
/* tp_new */ Compressor_new,
};

63
src/calibre/utils/lzx/lzxmodule.c
View File

@ -11,7 +11,7 @@
#include <lzxmodule.h>
static char lzx_doc[] =
static char lzx_doc[] =
"Provide basic LZX compression and decompression using the code from\n"
"liblzxcomp and libmspack respectively.";
@ -68,13 +68,13 @@ glue_read(struct mspack_file *file, void *buffer, int bytes)
mem = (memory_file *)file;
if (mem->magic != 0xB5) return -1;
remaining = mem->total_bytes - mem->current_bytes;
if (!remaining) return 0;
if (bytes > remaining) bytes = remaining;
memcpy(buffer, (unsigned char *)mem->buffer + mem->current_bytes, bytes);
mem->current_bytes += bytes;
return bytes;
}
@ -86,7 +86,7 @@ glue_write(struct mspack_file *file, void *buffer, int bytes)
mem = (memory_file *)file;
if (mem->magic != 0xB5) return -1;
remaining = mem->total_bytes - mem->current_bytes;
if (bytes > remaining) {
PyErr_SetString(LZXError,
@ -99,7 +99,7 @@ glue_write(struct mspack_file *file, void *buffer, int bytes)
}
struct mspack_system lzxglue_system = {
glue_open,
glue_open,
glue_close,
glue_read, /* Read */
glue_write, /* Write */
@ -125,7 +125,7 @@ init(PyObject *self, PyObject *args)
if (!PyArg_ParseTuple(args, "i", &window)) {
return NULL;
}
LZXwindow = window;
lzx_stream = NULL;
@ -143,7 +143,7 @@ reset(PyObject *self, PyObject *args)
Py_RETURN_NONE;
}
//int LZXdecompress(unsigned char *inbuf, unsigned char *outbuf,
//int LZXdecompress(unsigned char *inbuf, unsigned char *outbuf,
// unsigned int inlen, unsigned int outlen)
static PyObject *
decompress(PyObject *self, PyObject *args)
@ -161,12 +161,12 @@ decompress(PyObject *self, PyObject *args)
return NULL;
}
retval = PyString_FromStringAndSize(NULL, outlen);
retval = PyBytes_FromStringAndSize(NULL, outlen);
if (retval == NULL) {
return NULL;
}
outbuf = (unsigned char *)PyString_AS_STRING(retval);
outbuf = (unsigned char *)PyBytes_AS_STRING(retval);
source.magic = 0xB5;
source.buffer = inbuf;
source.current_bytes = 0;
@ -176,9 +176,9 @@ decompress(PyObject *self, PyObject *args)
dest.buffer = outbuf;
dest.current_bytes = 0;
dest.total_bytes = outlen;
lzx_stream = lzxd_init(&lzxglue_system, (struct mspack_file *)&source,
(struct mspack_file *)&dest, LZXwindow,
(struct mspack_file *)&dest, LZXwindow,
0x7fff /* Never reset, I do it */, 4096, outlen);
err = -1;
if (lzx_stream) err = lzxd_decompress(lzx_stream, outlen);
@ -191,7 +191,7 @@ decompress(PyObject *self, PyObject *args)
retval = NULL;
PyErr_SetString(LZXError, "LZX decompression failed");
}
return retval;
}
@ -202,20 +202,37 @@ static PyMethodDef lzx_methods[] = {
{ NULL }
};
CALIBRE_MODINIT_FUNC
initlzx(void)
{
PyObject *m;
#if PY_MAJOR_VERSION >= 3
#define INITERROR return NULL
#define INITMODULE PyModule_Create(&lzx_module)
static struct PyModuleDef lzx_module = {
/* m_base */ PyModuleDef_HEAD_INIT,
/* m_name */ "lzx",
/* m_doc */ lzx_doc,
/* m_size */ -1,
/* m_methods */ lzx_methods,
/* m_slots */ 0,
/* m_traverse */ 0,
/* m_clear */ 0,
/* m_free */ 0,
};
CALIBRE_MODINIT_FUNC PyInit_lzx(void) {
#else
#define INITERROR return
#define INITMODULE Py_InitModule3("lzx", lzx_methods, lzx_doc);
CALIBRE_MODINIT_FUNC initlzx(void) {
#endif
if (PyType_Ready(&CompressorType) < 0) {
return;
INITERROR;
}
m = Py_InitModule3("lzx", lzx_methods, lzx_doc);
PyObject *m = INITMODULE;
if (m == NULL) {
return;
INITERROR;
}
LZXError = PyErr_NewException("lzx.LZXError", NULL, NULL);
Py_INCREF(LZXError);
PyModule_AddObject(m, "LZXError", LZXError);
@ -223,5 +240,7 @@ initlzx(void)
Py_INCREF(&CompressorType);
PyModule_AddObject(m, "Compressor", (PyObject *)&CompressorType);
return;
#if PY_MAJOR_VERSION >= 3
return m;
#endif
}