/*********************************************************** Copyright 1991, 1992, 1993 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 #include #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) 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 */ cl_dealloc, /*tp_dealloc*/ 0, /*tp_print*/ 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 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); }