diff options
Diffstat (limited to 'Modules/imageop.c')
| -rw-r--r-- | Modules/imageop.c | 1240 |
1 files changed, 625 insertions, 615 deletions
diff --git a/Modules/imageop.c b/Modules/imageop.c index 686d0c6..0f8e0ed 100644 --- a/Modules/imageop.c +++ b/Modules/imageop.c @@ -35,715 +35,725 @@ PERFORMANCE OF THIS SOFTWARE. #define signed #endif -#include "allobjects.h" -#include "modsupport.h" +#include "Python.h" #define CHARP(cp, xmax, x, y) ((char *)(cp+y*xmax+x)) #define SHORTP(cp, xmax, x, y) ((short *)(cp+2*(y*xmax+x))) #define LONGP(cp, xmax, x, y) ((long *)(cp+4*(y*xmax+x))) -static object *ImageopError; +static PyObject *ImageopError; -static object * +static PyObject * imageop_crop(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - char *cp, *ncp; - short *nsp; - long *nlp; - int len, size, x, y, newx1, newx2, newy1, newy2; - int ix, iy, xstep, ystep; - object *rv; - - if ( !getargs(args, "(s#iiiiiii)", &cp, &len, &size, &x, &y, - &newx1, &newy1, &newx2, &newy2) ) - return 0; - - if ( size != 1 && size != 2 && size != 4 ) { - err_setstr(ImageopError, "Size should be 1, 2 or 4"); - return 0; - } - if ( len != size*x*y ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - xstep = (newx1 < newx2)? 1 : -1; - ystep = (newy1 < newy2)? 1 : -1; - - rv = newsizedstringobject(NULL, - (abs(newx2-newx1)+1)*(abs(newy2-newy1)+1)*size); - if ( rv == 0 ) - return 0; - ncp = (char *)getstringvalue(rv); - nsp = (short *)ncp; - nlp = (long *)ncp; - newy2 += ystep; - newx2 += xstep; - for( iy = newy1; iy != newy2; iy+=ystep ) { - for ( ix = newx1; ix != newx2; ix+=xstep ) { - if ( iy < 0 || iy >= y || ix < 0 || ix >= x ) { - if ( size == 1 ) *ncp++ = 0; - else *nlp++ = 0; - } else { - if ( size == 1 ) *ncp++ = *CHARP(cp, x, ix, iy); - else if ( size == 2 ) *nsp++ = *SHORTP(cp, x, ix, iy); - else *nlp++ = *LONGP(cp, x, ix, iy); - } - } - } - return rv; + char *cp, *ncp; + short *nsp; + long *nlp; + int len, size, x, y, newx1, newx2, newy1, newy2; + int ix, iy, xstep, ystep; + PyObject *rv; + + if ( !PyArg_Parse(args, "(s#iiiiiii)", &cp, &len, &size, &x, &y, + &newx1, &newy1, &newx2, &newy2) ) + return 0; + + if ( size != 1 && size != 2 && size != 4 ) { + PyErr_SetString(ImageopError, "Size should be 1, 2 or 4"); + return 0; + } + if ( len != size*x*y ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + xstep = (newx1 < newx2)? 1 : -1; + ystep = (newy1 < newy2)? 1 : -1; + + rv = PyString_FromStringAndSize(NULL, + (abs(newx2-newx1)+1)*(abs(newy2-newy1)+1)*size); + if ( rv == 0 ) + return 0; + ncp = (char *)PyString_AsString(rv); + nsp = (short *)ncp; + nlp = (long *)ncp; + newy2 += ystep; + newx2 += xstep; + for( iy = newy1; iy != newy2; iy+=ystep ) { + for ( ix = newx1; ix != newx2; ix+=xstep ) { + if ( iy < 0 || iy >= y || ix < 0 || ix >= x ) { + if ( size == 1 ) + *ncp++ = 0; + else + *nlp++ = 0; + } else { + if ( size == 1 ) + *ncp++ = *CHARP(cp, x, ix, iy); + else if ( size == 2 ) + *nsp++ = *SHORTP(cp, x, ix, iy); + else + *nlp++ = *LONGP(cp, x, ix, iy); + } + } + } + return rv; } -static object * +static PyObject * imageop_scale(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - char *cp, *ncp; - short *nsp; - long *nlp; - int len, size, x, y, newx, newy; - int ix, iy; - int oix, oiy; - object *rv; - - if ( !getargs(args, "(s#iiiii)", &cp, &len, &size, &x, &y, &newx, &newy) ) - return 0; - - if ( size != 1 && size != 2 && size != 4 ) { - err_setstr(ImageopError, "Size should be 1, 2 or 4"); - return 0; - } - if ( len != size*x*y ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, newx*newy*size); - if ( rv == 0 ) - return 0; - ncp = (char *)getstringvalue(rv); - nsp = (short *)ncp; - nlp = (long *)ncp; - for( iy = 0; iy < newy; iy++ ) { - for ( ix = 0; ix < newx; ix++ ) { - oix = ix * x / newx; - oiy = iy * y / newy; - if ( size == 1 ) *ncp++ = *CHARP(cp, x, oix, oiy); - else if ( size == 2 ) *nsp++ = *SHORTP(cp, x, oix, oiy); - else *nlp++ = *LONGP(cp, x, oix, oiy); - } - } - return rv; + char *cp, *ncp; + short *nsp; + long *nlp; + int len, size, x, y, newx, newy; + int ix, iy; + int oix, oiy; + PyObject *rv; + + if ( !PyArg_Parse(args, "(s#iiiii)", + &cp, &len, &size, &x, &y, &newx, &newy) ) + return 0; + + if ( size != 1 && size != 2 && size != 4 ) { + PyErr_SetString(ImageopError, "Size should be 1, 2 or 4"); + return 0; + } + if ( len != size*x*y ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, newx*newy*size); + if ( rv == 0 ) + return 0; + ncp = (char *)PyString_AsString(rv); + nsp = (short *)ncp; + nlp = (long *)ncp; + for( iy = 0; iy < newy; iy++ ) { + for ( ix = 0; ix < newx; ix++ ) { + oix = ix * x / newx; + oiy = iy * y / newy; + if ( size == 1 ) + *ncp++ = *CHARP(cp, x, oix, oiy); + else if ( size == 2 ) + *nsp++ = *SHORTP(cp, x, oix, oiy); + else + *nlp++ = *LONGP(cp, x, oix, oiy); + } + } + return rv; } /* Note: this routine can use a bit of optimizing */ -static object * +static PyObject * imageop_tovideo(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int maxx, maxy, x, y, len; - int i; - unsigned char *cp, *ncp; - int width; - object *rv; + int maxx, maxy, x, y, len; + int i; + unsigned char *cp, *ncp; + int width; + PyObject *rv; - if ( !getargs(args, "(s#iii)", &cp, &len, &width, &maxx, &maxy) ) - return 0; - - if ( width != 1 && width != 4 ) { - err_setstr(ImageopError, "Size should be 1 or 4"); - return 0; - } - if ( maxx*maxy*width != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, len); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - if ( width == 1 ) { - memcpy(ncp, cp, maxx); /* Copy first line */ - ncp += maxx; - for (y=1; y<maxy; y++) { /* Interpolate other lines */ - for(x=0; x<maxx; x++) { - i = y*maxx + x; - *ncp++ = ((int)cp[i] + (int)cp[i-maxx]) >> 1; - } - } - } else { - memcpy(ncp, cp, maxx*4); /* Copy first line */ - ncp += maxx*4; - for (y=1; y<maxy; y++) { /* Interpolate other lines */ - for(x=0; x<maxx; x++) { - i = (y*maxx + x)*4 + 1; - *ncp++ = 0; /* Skip alfa comp */ - *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1; - i++; - *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1; - i++; - *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1; - } - } - } - return rv; + if ( !PyArg_Parse(args, "(s#iii)", &cp, &len, &width, &maxx, &maxy) ) + return 0; + + if ( width != 1 && width != 4 ) { + PyErr_SetString(ImageopError, "Size should be 1 or 4"); + return 0; + } + if ( maxx*maxy*width != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, len); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + if ( width == 1 ) { + memcpy(ncp, cp, maxx); /* Copy first line */ + ncp += maxx; + for (y=1; y<maxy; y++) { /* Interpolate other lines */ + for(x=0; x<maxx; x++) { + i = y*maxx + x; + *ncp++ = ((int)cp[i] + (int)cp[i-maxx]) >> 1; + } + } + } else { + memcpy(ncp, cp, maxx*4); /* Copy first line */ + ncp += maxx*4; + for (y=1; y<maxy; y++) { /* Interpolate other lines */ + for(x=0; x<maxx; x++) { + i = (y*maxx + x)*4 + 1; + *ncp++ = 0; /* Skip alfa comp */ + *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1; + i++; + *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1; + i++; + *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1; + } + } + } + return rv; } -static object * +static PyObject * imageop_grey2mono(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int tres, x, y, len; - unsigned char *cp, *ncp; - unsigned char ovalue; - object *rv; - int i, bit; + int tres, x, y, len; + unsigned char *cp, *ncp; + unsigned char ovalue; + PyObject *rv; + int i, bit; - if ( !getargs(args, "(s#iii)", &cp, &len, &x, &y, &tres) ) - return 0; - - if ( x*y != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, (len+7)/8); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - bit = 0x80; - ovalue = 0; - for ( i=0; i < len; i++ ) { - if ( (int)cp[i] > tres ) - ovalue |= bit; - bit >>= 1; - if ( bit == 0 ) { - *ncp++ = ovalue; - bit = 0x80; - ovalue = 0; - } - } - if ( bit != 0x80 ) - *ncp++ = ovalue; - return rv; + if ( !PyArg_Parse(args, "(s#iii)", &cp, &len, &x, &y, &tres) ) + return 0; + + if ( x*y != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, (len+7)/8); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + bit = 0x80; + ovalue = 0; + for ( i=0; i < len; i++ ) { + if ( (int)cp[i] > tres ) + ovalue |= bit; + bit >>= 1; + if ( bit == 0 ) { + *ncp++ = ovalue; + bit = 0x80; + ovalue = 0; + } + } + if ( bit != 0x80 ) + *ncp++ = ovalue; + return rv; } -static object * +static PyObject * imageop_grey2grey4(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len; - unsigned char *cp, *ncp; - unsigned char ovalue; - object *rv; - int i; - int pos; + int x, y, len; + unsigned char *cp, *ncp; + unsigned char ovalue; + PyObject *rv; + int i; + int pos; - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - if ( x*y != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, (len+1)/2); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - pos = 0; - ovalue = 0; - for ( i=0; i < len; i++ ) { - ovalue |= ((int)cp[i] & 0xf0) >> pos; - pos += 4; - if ( pos == 8 ) { - *ncp++ = ovalue; - ovalue = 0; - pos = 0; - } - } - if ( pos != 0 ) - *ncp++ = ovalue; - return rv; + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + if ( x*y != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, (len+1)/2); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + pos = 0; + ovalue = 0; + for ( i=0; i < len; i++ ) { + ovalue |= ((int)cp[i] & 0xf0) >> pos; + pos += 4; + if ( pos == 8 ) { + *ncp++ = ovalue; + ovalue = 0; + pos = 0; + } + } + if ( pos != 0 ) + *ncp++ = ovalue; + return rv; } -static object * +static PyObject * imageop_grey2grey2(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len; - unsigned char *cp, *ncp; - unsigned char ovalue; - object *rv; - int i; - int pos; + int x, y, len; + unsigned char *cp, *ncp; + unsigned char ovalue; + PyObject *rv; + int i; + int pos; - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - if ( x*y != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, (len+3)/4); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - pos = 0; - ovalue = 0; - for ( i=0; i < len; i++ ) { - ovalue |= ((int)cp[i] & 0xc0) >> pos; - pos += 2; - if ( pos == 8 ) { - *ncp++ = ovalue; - ovalue = 0; - pos = 0; - } - } - if ( pos != 0 ) - *ncp++ = ovalue; - return rv; + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + if ( x*y != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, (len+3)/4); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + pos = 0; + ovalue = 0; + for ( i=0; i < len; i++ ) { + ovalue |= ((int)cp[i] & 0xc0) >> pos; + pos += 2; + if ( pos == 8 ) { + *ncp++ = ovalue; + ovalue = 0; + pos = 0; + } + } + if ( pos != 0 ) + *ncp++ = ovalue; + return rv; } -static object * +static PyObject * imageop_dither2mono(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int sum, x, y, len; - unsigned char *cp, *ncp; - unsigned char ovalue; - object *rv; - int i, bit; + int sum, x, y, len; + unsigned char *cp, *ncp; + unsigned char ovalue; + PyObject *rv; + int i, bit; - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - if ( x*y != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, (len+7)/8); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - bit = 0x80; - ovalue = 0; - sum = 0; - for ( i=0; i < len; i++ ) { - sum += cp[i]; - if ( sum >= 256 ) { - sum -= 256; - ovalue |= bit; - } - bit >>= 1; - if ( bit == 0 ) { - *ncp++ = ovalue; - bit = 0x80; - ovalue = 0; - } - } - if ( bit != 0x80 ) - *ncp++ = ovalue; - return rv; + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + if ( x*y != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, (len+7)/8); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + bit = 0x80; + ovalue = 0; + sum = 0; + for ( i=0; i < len; i++ ) { + sum += cp[i]; + if ( sum >= 256 ) { + sum -= 256; + ovalue |= bit; + } + bit >>= 1; + if ( bit == 0 ) { + *ncp++ = ovalue; + bit = 0x80; + ovalue = 0; + } + } + if ( bit != 0x80 ) + *ncp++ = ovalue; + return rv; } -static object * +static PyObject * imageop_dither2grey2(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len; - unsigned char *cp, *ncp; - unsigned char ovalue; - object *rv; - int i; - int pos; - int sum = 0, nvalue; + int x, y, len; + unsigned char *cp, *ncp; + unsigned char ovalue; + PyObject *rv; + int i; + int pos; + int sum = 0, nvalue; - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - if ( x*y != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, (len+3)/4); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - pos = 1; - ovalue = 0; - for ( i=0; i < len; i++ ) { - sum += cp[i]; - nvalue = sum & 0x180; - sum -= nvalue; - ovalue |= nvalue >> pos; - pos += 2; - if ( pos == 9 ) { - *ncp++ = ovalue; - ovalue = 0; - pos = 1; - } - } - if ( pos != 0 ) - *ncp++ = ovalue; - return rv; + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + if ( x*y != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, (len+3)/4); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + pos = 1; + ovalue = 0; + for ( i=0; i < len; i++ ) { + sum += cp[i]; + nvalue = sum & 0x180; + sum -= nvalue; + ovalue |= nvalue >> pos; + pos += 2; + if ( pos == 9 ) { + *ncp++ = ovalue; + ovalue = 0; + pos = 1; + } + } + if ( pos != 0 ) + *ncp++ = ovalue; + return rv; } -static object * +static PyObject * imageop_mono2grey(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int v0, v1, x, y, len, nlen; - unsigned char *cp, *ncp; - object *rv; - int i, bit; - - if ( !getargs(args, "(s#iiii)", &cp, &len, &x, &y, &v0, &v1) ) - return 0; - - nlen = x*y; - if ( (nlen+7)/8 != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, nlen); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - bit = 0x80; - for ( i=0; i < nlen; i++ ) { - if ( *cp & bit ) - *ncp++ = v1; - else - *ncp++ = v0; - bit >>= 1; - if ( bit == 0 ) { - bit = 0x80; - cp++; - } - } - return rv; + int v0, v1, x, y, len, nlen; + unsigned char *cp, *ncp; + PyObject *rv; + int i, bit; + + if ( !PyArg_Parse(args, "(s#iiii)", &cp, &len, &x, &y, &v0, &v1) ) + return 0; + + nlen = x*y; + if ( (nlen+7)/8 != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, nlen); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + bit = 0x80; + for ( i=0; i < nlen; i++ ) { + if ( *cp & bit ) + *ncp++ = v1; + else + *ncp++ = v0; + bit >>= 1; + if ( bit == 0 ) { + bit = 0x80; + cp++; + } + } + return rv; } -static object * +static PyObject * imageop_grey22grey(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len, nlen; - unsigned char *cp, *ncp; - object *rv; - int i, pos, value = 0, nvalue; - - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - nlen = x*y; - if ( (nlen+3)/4 != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, nlen); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - pos = 0; - for ( i=0; i < nlen; i++ ) { - if ( pos == 0 ) { - value = *cp++; - pos = 8; - } - pos -= 2; - nvalue = (value >> pos) & 0x03; - *ncp++ = nvalue | (nvalue << 2) | (nvalue << 4) | (nvalue << 6); - } - return rv; + int x, y, len, nlen; + unsigned char *cp, *ncp; + PyObject *rv; + int i, pos, value = 0, nvalue; + + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + nlen = x*y; + if ( (nlen+3)/4 != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, nlen); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + pos = 0; + for ( i=0; i < nlen; i++ ) { + if ( pos == 0 ) { + value = *cp++; + pos = 8; + } + pos -= 2; + nvalue = (value >> pos) & 0x03; + *ncp++ = nvalue | (nvalue << 2) | + (nvalue << 4) | (nvalue << 6); + } + return rv; } -static object * +static PyObject * imageop_grey42grey(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len, nlen; - unsigned char *cp, *ncp; - object *rv; - int i, pos, value = 0, nvalue; - - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - nlen = x*y; - if ( (nlen+1)/2 != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, nlen); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - pos = 0; - for ( i=0; i < nlen; i++ ) { - if ( pos == 0 ) { - value = *cp++; - pos = 8; - } - pos -= 4; - nvalue = (value >> pos) & 0x0f; - *ncp++ = nvalue | (nvalue << 4); - } - return rv; + int x, y, len, nlen; + unsigned char *cp, *ncp; + PyObject *rv; + int i, pos, value = 0, nvalue; + + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + nlen = x*y; + if ( (nlen+1)/2 != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, nlen); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + pos = 0; + for ( i=0; i < nlen; i++ ) { + if ( pos == 0 ) { + value = *cp++; + pos = 8; + } + pos -= 4; + nvalue = (value >> pos) & 0x0f; + *ncp++ = nvalue | (nvalue << 4); + } + return rv; } -static object * +static PyObject * imageop_rgb2rgb8(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len, nlen; - unsigned long *cp; - unsigned char *ncp; - object *rv; - int i, r, g, b; - unsigned long value, nvalue; - - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - nlen = x*y; - if ( nlen*4 != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, nlen); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - for ( i=0; i < nlen; i++ ) { - /* Bits in source: aaaaaaaa BBbbbbbb GGGggggg RRRrrrrr */ - value = *cp++; + int x, y, len, nlen; + unsigned long *cp; + unsigned char *ncp; + PyObject *rv; + int i, r, g, b; + unsigned long value, nvalue; + + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + nlen = x*y; + if ( nlen*4 != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, nlen); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + for ( i=0; i < nlen; i++ ) { + /* Bits in source: aaaaaaaa BBbbbbbb GGGggggg RRRrrrrr */ + value = *cp++; #if 0 - r = (value >> 5) & 7; - g = (value >> 13) & 7; - b = (value >> 22) & 3; + r = (value >> 5) & 7; + g = (value >> 13) & 7; + b = (value >> 22) & 3; #else - r = (int) ((value & 0xff) / 255. * 7. + .5); - g = (int) (((value >> 8) & 0xff) / 255. * 7. + .5); - b = (int) (((value >> 16) & 0xff) / 255. * 3. + .5); + r = (int) ((value & 0xff) / 255. * 7. + .5); + g = (int) (((value >> 8) & 0xff) / 255. * 7. + .5); + b = (int) (((value >> 16) & 0xff) / 255. * 3. + .5); #endif - nvalue = (r<<5) | (b<<3) | g; - *ncp++ = nvalue; - } - return rv; + nvalue = (r<<5) | (b<<3) | g; + *ncp++ = nvalue; + } + return rv; } -static object * +static PyObject * imageop_rgb82rgb(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len, nlen; - unsigned char *cp; - unsigned long *ncp; - object *rv; - int i, r, g, b; - unsigned long value, nvalue; - - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - nlen = x*y; - if ( nlen != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, nlen*4); - if ( rv == 0 ) - return 0; - ncp = (unsigned long *)getstringvalue(rv); - - for ( i=0; i < nlen; i++ ) { - /* Bits in source: RRRBBGGG - ** Red and Green are multiplied by 36.5, Blue by 85 - */ - value = *cp++; - r = (value >> 5) & 7; - g = (value ) & 7; - b = (value >> 3) & 3; - r = (r<<5) | (r<<3) | (r>>1); - g = (g<<5) | (g<<3) | (g>>1); - b = (b<<6) | (b<<4) | (b<<2) | b; - nvalue = r | (g<<8) | (b<<16); - *ncp++ = nvalue; - } - return rv; + int x, y, len, nlen; + unsigned char *cp; + unsigned long *ncp; + PyObject *rv; + int i, r, g, b; + unsigned long value, nvalue; + + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + nlen = x*y; + if ( nlen != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, nlen*4); + if ( rv == 0 ) + return 0; + ncp = (unsigned long *)PyString_AsString(rv); + + for ( i=0; i < nlen; i++ ) { + /* Bits in source: RRRBBGGG + ** Red and Green are multiplied by 36.5, Blue by 85 + */ + value = *cp++; + r = (value >> 5) & 7; + g = (value ) & 7; + b = (value >> 3) & 3; + r = (r<<5) | (r<<3) | (r>>1); + g = (g<<5) | (g<<3) | (g>>1); + b = (b<<6) | (b<<4) | (b<<2) | b; + nvalue = r | (g<<8) | (b<<16); + *ncp++ = nvalue; + } + return rv; } -static object * +static PyObject * imageop_rgb2grey(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len, nlen; - unsigned long *cp; - unsigned char *ncp; - object *rv; - int i, r, g, b; - unsigned long value, nvalue; - - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - nlen = x*y; - if ( nlen*4 != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, nlen); - if ( rv == 0 ) - return 0; - ncp = (unsigned char *)getstringvalue(rv); - - for ( i=0; i < nlen; i++ ) { - value = *cp++; - r = (value ) & 0xff; - g = (value >> 8) & 0xff; - b = (value >> 16) & 0xff; - nvalue = (int)(0.30*r + 0.59*g + 0.11*b); - if ( nvalue > 255 ) nvalue = 255; - *ncp++ = nvalue; - } - return rv; + int x, y, len, nlen; + unsigned long *cp; + unsigned char *ncp; + PyObject *rv; + int i, r, g, b; + unsigned long value, nvalue; + + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + nlen = x*y; + if ( nlen*4 != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, nlen); + if ( rv == 0 ) + return 0; + ncp = (unsigned char *)PyString_AsString(rv); + + for ( i=0; i < nlen; i++ ) { + value = *cp++; + r = (value ) & 0xff; + g = (value >> 8) & 0xff; + b = (value >> 16) & 0xff; + nvalue = (int)(0.30*r + 0.59*g + 0.11*b); + if ( nvalue > 255 ) nvalue = 255; + *ncp++ = nvalue; + } + return rv; } -static object * +static PyObject * imageop_grey2rgb(self, args) - object *self; - object *args; + PyObject *self; +PyObject *args; { - int x, y, len, nlen; - unsigned char *cp; - unsigned long *ncp; - object *rv; - int i; - unsigned long value; - - if ( !getargs(args, "(s#ii)", &cp, &len, &x, &y) ) - return 0; - - nlen = x*y; - if ( nlen != len ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, nlen*4); - if ( rv == 0 ) - return 0; - ncp = (unsigned long *)getstringvalue(rv); - - for ( i=0; i < nlen; i++ ) { - value = *cp++; - *ncp++ = value | (value << 8 ) | (value << 16); - } - return rv; + int x, y, len, nlen; + unsigned char *cp; + unsigned long *ncp; + PyObject *rv; + int i; + unsigned long value; + + if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &x, &y) ) + return 0; + + nlen = x*y; + if ( nlen != len ) { + PyErr_SetString(ImageopError, "String has incorrect length"); + return 0; + } + + rv = PyString_FromStringAndSize(NULL, nlen*4); + if ( rv == 0 ) + return 0; + ncp = (unsigned long *)PyString_AsString(rv); + + for ( i=0; i < nlen; i++ ) { + value = *cp++; + *ncp++ = value | (value << 8 ) | (value << 16); + } + return rv; } /* static object * imageop_mul(self, args) - object *self; - object *args; + object *self; +object *args; { - char *cp, *ncp; - int len, size, x, y; - object *rv; - int i; - - if ( !getargs(args, "(s#iii)", &cp, &len, &size, &x, &y) ) - return 0; - - if ( size != 1 && size != 4 ) { - err_setstr(ImageopError, "Size should be 1 or 4"); - return 0; - } - if ( len != size*x*y ) { - err_setstr(ImageopError, "String has incorrect length"); - return 0; - } - - rv = newsizedstringobject(NULL, XXXX); - if ( rv == 0 ) - return 0; - ncp = (char *)getstringvalue(rv); - - - for ( i=0; i < len; i += size ) { - } - return rv; + char *cp, *ncp; + int len, size, x, y; + object *rv; + int i; + + if ( !getargs(args, "(s#iii)", &cp, &len, &size, &x, &y) ) + return 0; + + if ( size != 1 && size != 4 ) { + err_setstr(ImageopError, "Size should be 1 or 4"); + return 0; + } + if ( len != size*x*y ) { + err_setstr(ImageopError, "String has incorrect length"); + return 0; + } + + rv = newsizedstringobject(NULL, XXXX); + if ( rv == 0 ) + return 0; + ncp = (char *)getstringvalue(rv); + + + for ( i=0; i < len; i += size ) { + } + return rv; } */ -static struct methodlist imageop_methods[] = { - { "crop", imageop_crop }, - { "scale", imageop_scale }, - { "grey2mono", imageop_grey2mono }, - { "grey2grey2", imageop_grey2grey2 }, - { "grey2grey4", imageop_grey2grey4 }, - { "dither2mono", imageop_dither2mono }, - { "dither2grey2", imageop_dither2grey2 }, - { "mono2grey", imageop_mono2grey }, - { "grey22grey", imageop_grey22grey }, - { "grey42grey", imageop_grey42grey }, - { "tovideo", imageop_tovideo }, - { "rgb2rgb8", imageop_rgb2rgb8 }, - { "rgb82rgb", imageop_rgb82rgb }, - { "rgb2grey", imageop_rgb2grey }, - { "grey2rgb", imageop_grey2rgb }, - { 0, 0 } +static PyMethodDef imageop_methods[] = { + { "crop", imageop_crop }, + { "scale", imageop_scale }, + { "grey2mono", imageop_grey2mono }, + { "grey2grey2", imageop_grey2grey2 }, + { "grey2grey4", imageop_grey2grey4 }, + { "dither2mono", imageop_dither2mono }, + { "dither2grey2", imageop_dither2grey2 }, + { "mono2grey", imageop_mono2grey }, + { "grey22grey", imageop_grey22grey }, + { "grey42grey", imageop_grey42grey }, + { "tovideo", imageop_tovideo }, + { "rgb2rgb8", imageop_rgb2rgb8 }, + { "rgb82rgb", imageop_rgb82rgb }, + { "rgb2grey", imageop_rgb2grey }, + { "grey2rgb", imageop_grey2rgb }, + { 0, 0 } }; void initimageop() { - object *m, *d; - m = initmodule("imageop", imageop_methods); - d = getmoduledict(m); - ImageopError = newstringobject("imageop.error"); - if ( ImageopError == NULL || dictinsert(d,"error",ImageopError) ) - fatal("can't define imageop.error"); + PyObject *m, *d; + m = Py_InitModule("imageop", imageop_methods); + d = PyModule_GetDict(m); + ImageopError = PyString_FromString("imageop.error"); + if ( ImageopError == NULL || + PyDict_SetItemString(d,"error",ImageopError) ) + Py_FatalError("can't define imageop.error"); } |