mirror of
https://github.com/python/cpython.git
synced 2024-11-28 16:45:42 +01:00
954 lines
21 KiB
C
954 lines
21 KiB
C
/***********************************************************
|
|
Copyright 1991, 1992, 1993, 1994 by Stichting Mathematisch Centrum,
|
|
Amsterdam, The Netherlands.
|
|
|
|
All Rights Reserved
|
|
|
|
Permission to use, copy, modify, and distribute this software and its
|
|
documentation for any purpose and without fee is hereby granted,
|
|
provided that the above copyright notice appear in all copies and that
|
|
both that copyright notice and this permission notice appear in
|
|
supporting documentation, and that the names of Stichting Mathematisch
|
|
Centrum or CWI not be used in advertising or publicity pertaining to
|
|
distribution of the software without specific, written prior permission.
|
|
|
|
STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
|
|
THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
|
|
FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
|
|
FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
|
|
OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
******************************************************************/
|
|
|
|
|
|
/* Cl objects */
|
|
|
|
#define CLDEBUG
|
|
|
|
#include <stdarg.h>
|
|
#include <cl.h>
|
|
#include "allobjects.h"
|
|
#include "modsupport.h" /* For getargs() etc. */
|
|
#include "ceval.h" /* For call_object() */
|
|
|
|
typedef struct {
|
|
OB_HEAD
|
|
int ob_isCompressor; /* Compressor or Decompressor */
|
|
CL_Handle ob_compressorHdl;
|
|
int *ob_paramtypes;
|
|
int ob_nparams;
|
|
} clobject;
|
|
|
|
static object *ClError; /* exception cl.error */
|
|
|
|
static int error_handler_called = 0;
|
|
|
|
/*
|
|
* We want to use the function prototypes that are available in the C
|
|
* compiler on the SGI. Because of that, we need to declare the first
|
|
* argument of the compressor and decompressor methods as "object *",
|
|
* even though they are really "clobject *". Therefore we cast the
|
|
* argument to the proper type using this macro.
|
|
*/
|
|
#define SELF ((clobject *) self)
|
|
|
|
/********************************************************************
|
|
Utility routines.
|
|
********************************************************************/
|
|
static void
|
|
cl_ErrorHandler(CL_Handle handle, int code, const char *fmt, ...)
|
|
{
|
|
va_list ap;
|
|
char errbuf[BUFSIZ]; /* hopefully big enough */
|
|
char *p;
|
|
|
|
if (err_occurred()) /* don't change existing error */
|
|
return;
|
|
error_handler_called = 1;
|
|
va_start(ap, fmt);
|
|
vsprintf(errbuf, fmt, ap);
|
|
va_end(ap);
|
|
p = &errbuf[strlen(errbuf) - 1]; /* swat the line feed */
|
|
if (*p == '\n')
|
|
*p = 0;
|
|
err_setstr(ClError, errbuf);
|
|
}
|
|
|
|
/*
|
|
* This assumes that params are always in the range 0 to some maximum.
|
|
*/
|
|
static int
|
|
param_type_is_float(clobject *self, int param)
|
|
{
|
|
int bufferlength;
|
|
int ret;
|
|
|
|
if (self->ob_paramtypes == NULL) {
|
|
error_handler_called = 0;
|
|
bufferlength = clQueryParams(self->ob_compressorHdl, 0, 0);
|
|
if (error_handler_called)
|
|
return -1;
|
|
|
|
self->ob_paramtypes = NEW(int, bufferlength);
|
|
if (self->ob_paramtypes == NULL)
|
|
return -1;
|
|
self->ob_nparams = bufferlength / 2;
|
|
|
|
(void) clQueryParams(self->ob_compressorHdl, self->ob_paramtypes, bufferlength);
|
|
if (error_handler_called) {
|
|
DEL(self->ob_paramtypes);
|
|
self->ob_paramtypes = NULL;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if (param < 0 || param >= self->ob_nparams)
|
|
return -1;
|
|
|
|
if (self->ob_paramtypes[param*2 + 1] == CL_FLOATING_ENUM_VALUE ||
|
|
self->ob_paramtypes[param*2 + 1] == CL_FLOATING_RANGE_VALUE)
|
|
return 1;
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
/********************************************************************
|
|
Single image compression/decompression.
|
|
********************************************************************/
|
|
static object *
|
|
cl_CompressImage(object *self, object *args)
|
|
{
|
|
int compressionScheme, width, height, originalFormat;
|
|
float compressionRatio;
|
|
int frameBufferSize, compressedBufferSize;
|
|
char *frameBuffer;
|
|
object *compressedBuffer;
|
|
|
|
if (!getargs(args, "(iiiifs#)", &compressionScheme, &width, &height,
|
|
&originalFormat, &compressionRatio, &frameBuffer,
|
|
&frameBufferSize))
|
|
return NULL;
|
|
|
|
retry:
|
|
compressedBuffer = newsizedstringobject(NULL, frameBufferSize);
|
|
if (compressedBuffer == NULL)
|
|
return NULL;
|
|
|
|
compressedBufferSize = frameBufferSize;
|
|
error_handler_called = 0;
|
|
if (clCompressImage(compressionScheme, width, height, originalFormat,
|
|
compressionRatio, (void *) frameBuffer,
|
|
&compressedBufferSize,
|
|
(void *) getstringvalue(compressedBuffer))
|
|
== FAILURE) {
|
|
DECREF(compressedBuffer);
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "clCompressImage failed");
|
|
return NULL;
|
|
}
|
|
|
|
if (compressedBufferSize > frameBufferSize) {
|
|
frameBufferSize = compressedBufferSize;
|
|
DECREF(compressedBuffer);
|
|
goto retry;
|
|
}
|
|
|
|
if (compressedBufferSize < frameBufferSize)
|
|
if (resizestring(&compressedBuffer, compressedBufferSize))
|
|
return NULL;
|
|
|
|
return compressedBuffer;
|
|
}
|
|
|
|
static object *
|
|
cl_DecompressImage(object *self, object *args)
|
|
{
|
|
int compressionScheme, width, height, originalFormat;
|
|
char *compressedBuffer;
|
|
int compressedBufferSize, frameBufferSize;
|
|
object *frameBuffer;
|
|
|
|
if (!getargs(args, "(iiiis#)", &compressionScheme, &width, &height,
|
|
&originalFormat, &compressedBuffer,
|
|
&compressedBufferSize))
|
|
return NULL;
|
|
|
|
frameBufferSize = width * height * CL_BytesPerPixel(originalFormat);
|
|
|
|
frameBuffer = newsizedstringobject(NULL, frameBufferSize);
|
|
if (frameBuffer == NULL)
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
if (clDecompressImage(compressionScheme, width, height, originalFormat,
|
|
compressedBufferSize, compressedBuffer,
|
|
(void *) getstringvalue(frameBuffer)) == FAILURE) {
|
|
DECREF(frameBuffer);
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "clDecompressImage failed");
|
|
return NULL;
|
|
}
|
|
|
|
return frameBuffer;
|
|
}
|
|
|
|
/********************************************************************
|
|
Sequential compression/decompression.
|
|
********************************************************************/
|
|
#define CheckCompressor(self) if ((self)->ob_compressorHdl == NULL) { \
|
|
err_setstr(RuntimeError, "(de)compressor not active"); \
|
|
return NULL; \
|
|
}
|
|
|
|
static object *
|
|
doClose(clobject *self, object *args, int (*close_func)(CL_Handle))
|
|
{
|
|
CheckCompressor(self);
|
|
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
if ((*close_func)(self->ob_compressorHdl) == FAILURE) {
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "close failed");
|
|
return NULL;
|
|
}
|
|
|
|
self->ob_compressorHdl = NULL;
|
|
|
|
if (self->ob_paramtypes)
|
|
DEL(self->ob_paramtypes);
|
|
self->ob_paramtypes = NULL;
|
|
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
clm_CloseCompressor(object *self, object *args)
|
|
{
|
|
return doClose(SELF, args, clCloseCompressor);
|
|
}
|
|
|
|
static object *
|
|
clm_CloseDecompressor(object *self, object *args)
|
|
{
|
|
return doClose(SELF, args, clCloseDecompressor);
|
|
}
|
|
|
|
static object *
|
|
clm_Compress(object *self, object *args)
|
|
{
|
|
int numberOfFrames;
|
|
int frameBufferSize, compressedBufferSize, size;
|
|
char *frameBuffer;
|
|
object *data;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getargs(args, "(is#)", &numberOfFrames, &frameBuffer, &frameBufferSize))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
size = clGetParam(SELF->ob_compressorHdl, CL_COMPRESSED_BUFFER_SIZE);
|
|
compressedBufferSize = size;
|
|
if (error_handler_called)
|
|
return NULL;
|
|
|
|
data = newsizedstringobject(NULL, size);
|
|
if (data == NULL)
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
if (clCompress(SELF->ob_compressorHdl, numberOfFrames,
|
|
(void *) frameBuffer, &compressedBufferSize,
|
|
(void *) getstringvalue(data)) == FAILURE) {
|
|
DECREF(data);
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "compress failed");
|
|
return NULL;
|
|
}
|
|
|
|
if (compressedBufferSize < size)
|
|
if (resizestring(&data, compressedBufferSize))
|
|
return NULL;
|
|
|
|
if (compressedBufferSize > size) {
|
|
/* we didn't get all "compressed" data */
|
|
DECREF(data);
|
|
err_setstr(ClError, "compressed data is more than fitted");
|
|
return NULL;
|
|
}
|
|
|
|
return data;
|
|
}
|
|
|
|
static object *
|
|
clm_Decompress(object *self, object *args)
|
|
{
|
|
object *data;
|
|
int numberOfFrames;
|
|
char *compressedData;
|
|
int compressedDataSize, dataSize;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getargs(args, "(is#)", &numberOfFrames, &compressedData,
|
|
&compressedDataSize))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
dataSize = clGetParam(SELF->ob_compressorHdl, CL_FRAME_BUFFER_SIZE);
|
|
if (error_handler_called)
|
|
return NULL;
|
|
|
|
data = newsizedstringobject(NULL, dataSize);
|
|
if (data == NULL)
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
if (clDecompress(SELF->ob_compressorHdl, numberOfFrames,
|
|
compressedDataSize, (void *) compressedData,
|
|
(void *) getstringvalue(data)) == FAILURE) {
|
|
DECREF(data);
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "decompress failed");
|
|
return NULL;
|
|
}
|
|
|
|
return data;
|
|
}
|
|
|
|
static object *
|
|
doParams(clobject *self, object *args, int (*func)(CL_Handle, int *, int),
|
|
int modified)
|
|
{
|
|
object *list, *v;
|
|
int *PVbuffer;
|
|
int length;
|
|
int i;
|
|
float number;
|
|
|
|
CheckCompressor(self);
|
|
|
|
if (!getargs(args, "O", &list))
|
|
return NULL;
|
|
if (!is_listobject(list)) {
|
|
err_badarg();
|
|
return NULL;
|
|
}
|
|
length = getlistsize(list);
|
|
PVbuffer = NEW(int, length);
|
|
if (PVbuffer == NULL)
|
|
return err_nomem();
|
|
for (i = 0; i < length; i++) {
|
|
v = getlistitem(list, i);
|
|
if (is_floatobject(v)) {
|
|
number = getfloatvalue(v);
|
|
PVbuffer[i] = CL_TypeIsInt(number);
|
|
} else if (is_intobject(v)) {
|
|
PVbuffer[i] = getintvalue(v);
|
|
if ((i & 1) &&
|
|
param_type_is_float(self, PVbuffer[i-1]) > 0) {
|
|
number = PVbuffer[i];
|
|
PVbuffer[i] = CL_TypeIsInt(number);
|
|
}
|
|
} else {
|
|
DEL(PVbuffer);
|
|
err_badarg();
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
error_handler_called = 0;
|
|
(*func)(self->ob_compressorHdl, PVbuffer, length);
|
|
if (error_handler_called) {
|
|
DEL(PVbuffer);
|
|
return NULL;
|
|
}
|
|
|
|
if (modified) {
|
|
for (i = 0; i < length; i++) {
|
|
if ((i & 1) &&
|
|
param_type_is_float(self, PVbuffer[i-1]) > 0) {
|
|
number = CL_TypeIsFloat(PVbuffer[i]);
|
|
v = newfloatobject(number);
|
|
} else
|
|
v = newintobject(PVbuffer[i]);
|
|
setlistitem(list, i, v);
|
|
}
|
|
}
|
|
|
|
DEL(PVbuffer);
|
|
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
clm_GetParams(object *self, object *args)
|
|
{
|
|
return doParams(SELF, args, clGetParams, 1);
|
|
}
|
|
|
|
static object *
|
|
clm_SetParams(object *self, object *args)
|
|
{
|
|
return doParams(SELF, args, clSetParams, 0);
|
|
}
|
|
|
|
static object *
|
|
do_get(clobject *self, object *args, int (*func)(CL_Handle, int))
|
|
{
|
|
int paramID, value;
|
|
float fvalue;
|
|
|
|
CheckCompressor(self);
|
|
|
|
if (!getargs(args, "i", ¶mID))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
value = (*func)(self->ob_compressorHdl, paramID);
|
|
if (error_handler_called)
|
|
return NULL;
|
|
|
|
if (param_type_is_float(self, paramID) > 0) {
|
|
fvalue = CL_TypeIsFloat(value);
|
|
return newfloatobject(fvalue);
|
|
}
|
|
|
|
return newintobject(value);
|
|
}
|
|
|
|
static object *
|
|
clm_GetParam(object *self, object *args)
|
|
{
|
|
return do_get(SELF, args, clGetParam);
|
|
}
|
|
|
|
static object *
|
|
clm_GetDefault(object *self, object *args)
|
|
{
|
|
return do_get(SELF, args, clGetDefault);
|
|
}
|
|
|
|
static object *
|
|
clm_SetParam(object *self, object *args)
|
|
{
|
|
int paramID, value;
|
|
float fvalue;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getargs(args, "(ii)", ¶mID, &value)) {
|
|
err_clear();
|
|
if (!getargs(args, "(if)", ¶mID, &fvalue)) {
|
|
err_clear();
|
|
err_setstr(TypeError, "bad argument list (format '(ii)' or '(if)')");
|
|
return NULL;
|
|
}
|
|
value = CL_TypeIsInt(fvalue);
|
|
} else {
|
|
if (param_type_is_float(SELF, paramID) > 0) {
|
|
fvalue = value;
|
|
value = CL_TypeIsInt(fvalue);
|
|
}
|
|
}
|
|
|
|
error_handler_called = 0;
|
|
value = clSetParam(SELF->ob_compressorHdl, paramID, value);
|
|
if (error_handler_called)
|
|
return NULL;
|
|
|
|
if (param_type_is_float(SELF, paramID) > 0)
|
|
return newfloatobject(CL_TypeIsFloat(value));
|
|
else
|
|
return newintobject(value);
|
|
}
|
|
|
|
static object *
|
|
clm_GetParamID(object *self, object *args)
|
|
{
|
|
char *name;
|
|
int value;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getargs(args, "s", &name))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
value = clGetParamID(SELF->ob_compressorHdl, name);
|
|
if (value == FAILURE) {
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "getparamid failed");
|
|
return NULL;
|
|
}
|
|
|
|
return newintobject(value);
|
|
}
|
|
|
|
static object *
|
|
clm_QueryParams(object *self, object *args)
|
|
{
|
|
int bufferlength;
|
|
int *PVbuffer;
|
|
object *list;
|
|
int i;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
bufferlength = clQueryParams(SELF->ob_compressorHdl, 0, 0);
|
|
if (error_handler_called)
|
|
return NULL;
|
|
|
|
PVbuffer = NEW(int, bufferlength);
|
|
if (PVbuffer == NULL)
|
|
return err_nomem();
|
|
|
|
bufferlength = clQueryParams(SELF->ob_compressorHdl, PVbuffer,
|
|
bufferlength);
|
|
if (error_handler_called) {
|
|
DEL(PVbuffer);
|
|
return NULL;
|
|
}
|
|
|
|
list = newlistobject(bufferlength);
|
|
if (list == NULL) {
|
|
DEL(PVbuffer);
|
|
return NULL;
|
|
}
|
|
|
|
for (i = 0; i < bufferlength; i++) {
|
|
if (i & 1)
|
|
setlistitem(list, i, newintobject(PVbuffer[i]));
|
|
else if (PVbuffer[i] == 0) {
|
|
INCREF(None);
|
|
setlistitem(list, i, None);
|
|
} else
|
|
setlistitem(list, i, newstringobject((char *) PVbuffer[i]));
|
|
}
|
|
|
|
DEL(PVbuffer);
|
|
|
|
return list;
|
|
}
|
|
|
|
static object *
|
|
clm_GetMinMax(object *self, object *args)
|
|
{
|
|
int param, min, max;
|
|
float fmin, fmax;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getargs(args, "i", ¶m))
|
|
return NULL;
|
|
|
|
clGetMinMax(SELF->ob_compressorHdl, param, &min, &max);
|
|
|
|
if (param_type_is_float(SELF, param) > 0) {
|
|
fmin = CL_TypeIsFloat(min);
|
|
fmax = CL_TypeIsFloat(max);
|
|
return mkvalue("(ff)", fmin, fmax);
|
|
}
|
|
|
|
return mkvalue("(ii)", min, max);
|
|
}
|
|
|
|
static object *
|
|
clm_GetName(object *self, object *args)
|
|
{
|
|
int param;
|
|
char *name;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getargs(args, "i", ¶m))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
name = clGetName(SELF->ob_compressorHdl, param);
|
|
if (name == NULL || error_handler_called) {
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "getname failed");
|
|
return NULL;
|
|
}
|
|
|
|
return newstringobject(name);
|
|
}
|
|
|
|
static object *
|
|
clm_QuerySchemeFromHandle(object *self, object *args)
|
|
{
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
|
|
return newintobject(clQuerySchemeFromHandle(SELF->ob_compressorHdl));
|
|
}
|
|
|
|
static object *
|
|
clm_ReadHeader(object *self, object *args)
|
|
{
|
|
char *header;
|
|
int headerSize;
|
|
|
|
CheckCompressor(SELF);
|
|
|
|
if (!getargs(args, "s#", &header, &headerSize))
|
|
return NULL;
|
|
|
|
return newintobject(clReadHeader(SELF->ob_compressorHdl,
|
|
headerSize, header));
|
|
}
|
|
|
|
static struct methodlist compressor_methods[] = {
|
|
{"close", clm_CloseCompressor}, /* alias */
|
|
{"CloseCompressor", clm_CloseCompressor},
|
|
{"Compress", clm_Compress},
|
|
{"GetDefault", clm_GetDefault},
|
|
{"GetMinMax", clm_GetMinMax},
|
|
{"GetName", clm_GetName},
|
|
{"GetParam", clm_GetParam},
|
|
{"GetParamID", clm_GetParamID},
|
|
{"GetParams", clm_GetParams},
|
|
{"QueryParams", clm_QueryParams},
|
|
{"QuerySchemeFromHandle",clm_QuerySchemeFromHandle},
|
|
{"SetParam", clm_SetParam},
|
|
{"SetParams", clm_SetParams},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
static struct methodlist decompressor_methods[] = {
|
|
{"close", clm_CloseDecompressor}, /* alias */
|
|
{"CloseDecompressor", clm_CloseDecompressor},
|
|
{"Decompress", clm_Decompress},
|
|
{"GetDefault", clm_GetDefault},
|
|
{"GetMinMax", clm_GetMinMax},
|
|
{"GetName", clm_GetName},
|
|
{"GetParam", clm_GetParam},
|
|
{"GetParamID", clm_GetParamID},
|
|
{"GetParams", clm_GetParams},
|
|
{"ReadHeader", clm_ReadHeader},
|
|
{"QueryParams", clm_QueryParams},
|
|
{"QuerySchemeFromHandle",clm_QuerySchemeFromHandle},
|
|
{"SetParam", clm_SetParam},
|
|
{"SetParams", clm_SetParams},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
static void
|
|
cl_dealloc(object *self)
|
|
{
|
|
if (SELF->ob_compressorHdl) {
|
|
if (SELF->ob_isCompressor)
|
|
clCloseCompressor(SELF->ob_compressorHdl);
|
|
else
|
|
clCloseDecompressor(SELF->ob_compressorHdl);
|
|
}
|
|
DEL(self);
|
|
}
|
|
|
|
static object *
|
|
cl_getattr(object *self, char *name)
|
|
{
|
|
if (SELF->ob_isCompressor)
|
|
return findmethod(compressor_methods, self, name);
|
|
else
|
|
return findmethod(decompressor_methods, self, name);
|
|
}
|
|
|
|
static typeobject Cltype = {
|
|
OB_HEAD_INIT(&Typetype)
|
|
0, /*ob_size*/
|
|
"cl", /*tp_name*/
|
|
sizeof(clobject), /*tp_size*/
|
|
0, /*tp_itemsize*/
|
|
/* methods */
|
|
(destructor)cl_dealloc, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
(getattrfunc)cl_getattr, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
0, /*tp_compare*/
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
};
|
|
|
|
static object *
|
|
doOpen(object *self, object *args, int (*open_func)(int, CL_Handle *),
|
|
int iscompressor)
|
|
{
|
|
int scheme;
|
|
clobject *new;
|
|
|
|
if (!getargs(args, "i", &scheme))
|
|
return NULL;
|
|
|
|
new = NEWOBJ(clobject, &Cltype);
|
|
if (new == NULL)
|
|
return NULL;
|
|
|
|
new->ob_compressorHdl = NULL;
|
|
new->ob_isCompressor = iscompressor;
|
|
new->ob_paramtypes = NULL;
|
|
|
|
error_handler_called = 0;
|
|
if ((*open_func)(scheme, &new->ob_compressorHdl) == FAILURE) {
|
|
DECREF(new);
|
|
if (!error_handler_called)
|
|
err_setstr(ClError, "Open(De)Compressor failed");
|
|
return NULL;
|
|
}
|
|
return (object *)new;
|
|
}
|
|
|
|
static object *
|
|
cl_OpenCompressor(object *self, object *args)
|
|
{
|
|
return doOpen(self, args, clOpenCompressor, 1);
|
|
}
|
|
|
|
static object *
|
|
cl_OpenDecompressor(object *self, object *args)
|
|
{
|
|
return doOpen(self, args, clOpenDecompressor, 0);
|
|
}
|
|
|
|
static object *
|
|
cl_QueryScheme(object *self, object *args)
|
|
{
|
|
char *header;
|
|
int headerlen;
|
|
int scheme;
|
|
|
|
if (!getargs(args, "s#", &header, &headerlen))
|
|
return NULL;
|
|
|
|
scheme = clQueryScheme(header);
|
|
if (scheme < 0) {
|
|
err_setstr(ClError, "unknown compression scheme");
|
|
return NULL;
|
|
}
|
|
|
|
return newintobject(scheme);
|
|
}
|
|
|
|
static object *
|
|
cl_QueryMaxHeaderSize(object *self, object *args)
|
|
{
|
|
int scheme;
|
|
|
|
if (!getargs(args, "i", &scheme))
|
|
return NULL;
|
|
|
|
return newintobject(clQueryMaxHeaderSize(scheme));
|
|
}
|
|
|
|
static object *
|
|
cl_QueryAlgorithms(object *self, object *args)
|
|
{
|
|
int algorithmMediaType;
|
|
int bufferlength;
|
|
int *PVbuffer;
|
|
object *list;
|
|
int i;
|
|
|
|
if (!getargs(args, "i", &algorithmMediaType))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
bufferlength = clQueryAlgorithms(algorithmMediaType, 0, 0);
|
|
if (error_handler_called)
|
|
return NULL;
|
|
|
|
PVbuffer = NEW(int, bufferlength);
|
|
if (PVbuffer == NULL)
|
|
return err_nomem();
|
|
|
|
bufferlength = clQueryAlgorithms(algorithmMediaType, PVbuffer,
|
|
bufferlength);
|
|
if (error_handler_called) {
|
|
DEL(PVbuffer);
|
|
return NULL;
|
|
}
|
|
|
|
list = newlistobject(bufferlength);
|
|
if (list == NULL) {
|
|
DEL(PVbuffer);
|
|
return NULL;
|
|
}
|
|
|
|
for (i = 0; i < bufferlength; i++) {
|
|
if (i & 1)
|
|
setlistitem(list, i, newintobject(PVbuffer[i]));
|
|
else if (PVbuffer[i] == 0) {
|
|
INCREF(None);
|
|
setlistitem(list, i, None);
|
|
} else
|
|
setlistitem(list, i, newstringobject((char *) PVbuffer[i]));
|
|
}
|
|
|
|
DEL(PVbuffer);
|
|
|
|
return list;
|
|
}
|
|
|
|
static object *
|
|
cl_QuerySchemeFromName(object *self, object *args)
|
|
{
|
|
int algorithmMediaType;
|
|
char *name;
|
|
int scheme;
|
|
|
|
if (!getargs(args, "(is)", &algorithmMediaType, &name))
|
|
return NULL;
|
|
|
|
error_handler_called = 0;
|
|
scheme = clQuerySchemeFromName(algorithmMediaType, name);
|
|
if (error_handler_called) {
|
|
err_setstr(ClError, "unknown compression scheme");
|
|
return NULL;
|
|
}
|
|
|
|
return newintobject(scheme);
|
|
}
|
|
|
|
static object *
|
|
cl_GetAlgorithmName(object *self, object *args)
|
|
{
|
|
int scheme;
|
|
char *name;
|
|
|
|
if (!getargs(args, "i", &scheme))
|
|
return NULL;
|
|
|
|
name = clGetAlgorithmName(scheme);
|
|
if (name == 0) {
|
|
err_setstr(ClError, "unknown compression scheme");
|
|
return NULL;
|
|
}
|
|
|
|
return newstringobject(name);
|
|
}
|
|
|
|
static object *
|
|
do_set(object *self, object *args, int (*func)(int, int, int))
|
|
{
|
|
int scheme, paramID, value;
|
|
float fvalue;
|
|
int is_float = 0;
|
|
|
|
if (!getargs(args, "(iii)", &scheme, ¶mID, &value)) {
|
|
err_clear();
|
|
if (!getargs(args, "(iif)", &scheme, ¶mID, &fvalue)) {
|
|
err_clear();
|
|
err_setstr(TypeError, "bad argument list (format '(iii)' or '(iif)')");
|
|
return NULL;
|
|
}
|
|
value = CL_TypeIsInt(fvalue);
|
|
is_float = 1;
|
|
} else {
|
|
/* check some parameters which we know to be floats */
|
|
switch (scheme) {
|
|
case CL_COMPRESSION_RATIO:
|
|
case CL_SPEED:
|
|
fvalue = value;
|
|
value = CL_TypeIsInt(fvalue);
|
|
is_float = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
error_handler_called = 0;
|
|
value = (*func)(scheme, paramID, value);
|
|
if (error_handler_called)
|
|
return NULL;
|
|
|
|
if (is_float)
|
|
return newfloatobject(CL_TypeIsFloat(value));
|
|
else
|
|
return newintobject(value);
|
|
}
|
|
|
|
static object *
|
|
cl_SetDefault(object *self, object *args)
|
|
{
|
|
return do_set(self, args, clSetDefault);
|
|
}
|
|
|
|
static object *
|
|
cl_SetMin(object *self, object *args)
|
|
{
|
|
return do_set(self, args, clSetMin);
|
|
}
|
|
|
|
static object *
|
|
cl_SetMax(object *self, object *args)
|
|
{
|
|
return do_set(self, args, clSetMax);
|
|
}
|
|
|
|
#ifdef CLDEBUG
|
|
static object *
|
|
cvt_type(object *self, object *args)
|
|
{
|
|
int number;
|
|
float fnumber;
|
|
|
|
if (getargs(args, "i", &number))
|
|
return newfloatobject(CL_TypeIsFloat(number));
|
|
else {
|
|
err_clear();
|
|
if (getargs(args, "f", &fnumber))
|
|
return newintobject(CL_TypeIsInt(fnumber));
|
|
return NULL;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static struct methodlist cl_methods[] = {
|
|
{"CompressImage", cl_CompressImage},
|
|
{"DecompressImage", cl_DecompressImage},
|
|
{"GetAlgorithmName", cl_GetAlgorithmName},
|
|
{"OpenCompressor", cl_OpenCompressor},
|
|
{"OpenDecompressor", cl_OpenDecompressor},
|
|
{"QueryAlgorithms", cl_QueryAlgorithms},
|
|
{"QueryMaxHeaderSize", cl_QueryMaxHeaderSize},
|
|
{"QueryScheme", cl_QueryScheme},
|
|
{"QuerySchemeFromName", cl_QuerySchemeFromName},
|
|
{"SetDefault", cl_SetDefault},
|
|
{"SetMax", cl_SetMax},
|
|
{"SetMin", cl_SetMin},
|
|
#ifdef CLDEBUG
|
|
{"cvt_type", cvt_type},
|
|
#endif
|
|
{NULL, NULL} /* Sentinel */
|
|
};
|
|
|
|
void
|
|
initcl()
|
|
{
|
|
object *m, *d;
|
|
|
|
m = initmodule("cl", cl_methods);
|
|
d = getmoduledict(m);
|
|
|
|
ClError = newstringobject("cl.error");
|
|
if (ClError == NULL || dictinsert(d, "error", ClError) != 0)
|
|
fatal("can't define cl.error");
|
|
|
|
(void) clSetErrorHandler(cl_ErrorHandler);
|
|
}
|