| 1 |
|
|---|
| 2 | /* imageopmodule - Various operations on pictures */
|
|---|
| 3 |
|
|---|
| 4 | #ifdef sun
|
|---|
| 5 | #define signed
|
|---|
| 6 | #endif
|
|---|
| 7 |
|
|---|
| 8 | #include "Python.h"
|
|---|
| 9 |
|
|---|
| 10 | #if SIZEOF_INT == 4
|
|---|
| 11 | typedef int Py_Int32;
|
|---|
| 12 | typedef unsigned int Py_UInt32;
|
|---|
| 13 | #else
|
|---|
| 14 | #if SIZEOF_LONG == 4
|
|---|
| 15 | typedef long Py_Int32;
|
|---|
| 16 | typedef unsigned long Py_UInt32;
|
|---|
| 17 | #else
|
|---|
| 18 | #error "No 4-byte integral type"
|
|---|
| 19 | #endif
|
|---|
| 20 | #endif
|
|---|
| 21 |
|
|---|
| 22 | #define CHARP(cp, xmax, x, y) ((char *)(cp+y*xmax+x))
|
|---|
| 23 | #define SHORTP(cp, xmax, x, y) ((short *)(cp+2*(y*xmax+x)))
|
|---|
| 24 | #define LONGP(cp, xmax, x, y) ((Py_Int32 *)(cp+4*(y*xmax+x)))
|
|---|
| 25 |
|
|---|
| 26 | static PyObject *ImageopError;
|
|---|
| 27 | static PyObject *ImageopDict;
|
|---|
| 28 |
|
|---|
| 29 | /**
|
|---|
| 30 | * Check a coordonnate, make sure that (0 < value).
|
|---|
| 31 | * Return 0 on error.
|
|---|
| 32 | */
|
|---|
| 33 | static int
|
|---|
| 34 | check_coordonnate(int value, const char* name)
|
|---|
| 35 | {
|
|---|
| 36 | if ( 0 < value)
|
|---|
| 37 | return 1;
|
|---|
| 38 | PyErr_Format(PyExc_ValueError, "%s value is negative or nul", name);
|
|---|
| 39 | return 0;
|
|---|
| 40 | }
|
|---|
| 41 |
|
|---|
| 42 | /**
|
|---|
| 43 | * Check integer overflow to make sure that product == x*y*size.
|
|---|
| 44 | * Return 0 on error.
|
|---|
| 45 | */
|
|---|
| 46 | static int
|
|---|
| 47 | check_multiply_size(int product, int x, const char* xname, int y, const char* yname, int size)
|
|---|
| 48 | {
|
|---|
| 49 | if ( !check_coordonnate(x, xname) )
|
|---|
| 50 | return 0;
|
|---|
| 51 | if ( !check_coordonnate(y, yname) )
|
|---|
| 52 | return 0;
|
|---|
| 53 | if ( size == (product / y) / x )
|
|---|
| 54 | return 1;
|
|---|
| 55 | PyErr_SetString(ImageopError, "String has incorrect length");
|
|---|
| 56 | return 0;
|
|---|
| 57 | }
|
|---|
| 58 |
|
|---|
| 59 | /**
|
|---|
| 60 | * Check integer overflow to make sure that product == x*y.
|
|---|
| 61 | * Return 0 on error.
|
|---|
| 62 | */
|
|---|
| 63 | static int
|
|---|
| 64 | check_multiply(int product, int x, int y)
|
|---|
| 65 | {
|
|---|
| 66 | return check_multiply_size(product, x, "x", y, "y", 1);
|
|---|
| 67 | }
|
|---|
| 68 |
|
|---|
| 69 | /* If this function returns true (the default if anything goes wrong), we're
|
|---|
| 70 | behaving in a backward-compatible way with respect to how multi-byte pixels
|
|---|
| 71 | are stored in the strings. The code in this module was originally written
|
|---|
| 72 | for an SGI which is a big-endian system, and so the old code assumed that
|
|---|
| 73 | 4-byte integers hold the R, G, and B values in a particular order.
|
|---|
| 74 | However, on little-endian systems the order is reversed, and so not
|
|---|
| 75 | actually compatible with what gl.lrectwrite and imgfile expect.
|
|---|
| 76 | (gl.lrectwrite and imgfile are also SGI-specific, however, it is
|
|---|
| 77 | conceivable that the data handled here comes from or goes to an SGI or that
|
|---|
| 78 | it is otherwise used in the expectation that the byte order in the strings
|
|---|
| 79 | is as specified.)
|
|---|
| 80 |
|
|---|
| 81 | The function returns the value of the module variable
|
|---|
| 82 | "backward_compatible", or 1 if the variable does not exist or is not an
|
|---|
| 83 | int.
|
|---|
| 84 | */
|
|---|
| 85 |
|
|---|
| 86 | static int
|
|---|
| 87 | imageop_backward_compatible(void)
|
|---|
| 88 | {
|
|---|
| 89 | static PyObject *bcos;
|
|---|
| 90 | PyObject *bco;
|
|---|
| 91 | long rc;
|
|---|
| 92 |
|
|---|
| 93 | if (ImageopDict == NULL) /* "cannot happen" */
|
|---|
| 94 | return 1;
|
|---|
| 95 | if (bcos == NULL) {
|
|---|
| 96 | /* cache string object for future use */
|
|---|
| 97 | bcos = PyString_FromString("backward_compatible");
|
|---|
| 98 | if (bcos == NULL)
|
|---|
| 99 | return 1;
|
|---|
| 100 | }
|
|---|
| 101 | bco = PyDict_GetItem(ImageopDict, bcos);
|
|---|
| 102 | if (bco == NULL)
|
|---|
| 103 | return 1;
|
|---|
| 104 | if (!PyInt_Check(bco))
|
|---|
| 105 | return 1;
|
|---|
| 106 | rc = PyInt_AsLong(bco);
|
|---|
| 107 | if (PyErr_Occurred()) {
|
|---|
| 108 | /* not an integer, or too large, or something */
|
|---|
| 109 | PyErr_Clear();
|
|---|
| 110 | rc = 1;
|
|---|
| 111 | }
|
|---|
| 112 | return rc != 0; /* convert to values 0, 1 */
|
|---|
| 113 | }
|
|---|
| 114 |
|
|---|
| 115 | static PyObject *
|
|---|
| 116 | imageop_crop(PyObject *self, PyObject *args)
|
|---|
| 117 | {
|
|---|
| 118 | char *cp, *ncp;
|
|---|
| 119 | short *nsp;
|
|---|
| 120 | Py_Int32 *nlp;
|
|---|
| 121 | int len, size, x, y, newx1, newx2, newy1, newy2, nlen;
|
|---|
| 122 | int ix, iy, xstep, ystep;
|
|---|
| 123 | PyObject *rv;
|
|---|
| 124 |
|
|---|
| 125 | if ( !PyArg_ParseTuple(args, "s#iiiiiii", &cp, &len, &size, &x, &y,
|
|---|
| 126 | &newx1, &newy1, &newx2, &newy2) )
|
|---|
| 127 | return 0;
|
|---|
| 128 |
|
|---|
| 129 | if ( size != 1 && size != 2 && size != 4 ) {
|
|---|
| 130 | PyErr_SetString(ImageopError, "Size should be 1, 2 or 4");
|
|---|
| 131 | return 0;
|
|---|
| 132 | }
|
|---|
| 133 | if ( !check_multiply_size(len, x, "x", y, "y", size) )
|
|---|
| 134 | return 0;
|
|---|
| 135 |
|
|---|
| 136 | xstep = (newx1 < newx2)? 1 : -1;
|
|---|
| 137 | ystep = (newy1 < newy2)? 1 : -1;
|
|---|
| 138 |
|
|---|
| 139 | nlen = (abs(newx2-newx1)+1)*(abs(newy2-newy1)+1)*size;
|
|---|
| 140 | if ( !check_multiply_size(nlen, abs(newx2-newx1)+1, "abs(newx2-newx1)+1", abs(newy2-newy1)+1, "abs(newy2-newy1)+1", size) )
|
|---|
| 141 | return 0;
|
|---|
| 142 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 143 | if ( rv == 0 )
|
|---|
| 144 | return 0;
|
|---|
| 145 | ncp = (char *)PyString_AsString(rv);
|
|---|
| 146 | nsp = (short *)ncp;
|
|---|
| 147 | nlp = (Py_Int32 *)ncp;
|
|---|
| 148 | newy2 += ystep;
|
|---|
| 149 | newx2 += xstep;
|
|---|
| 150 | for( iy = newy1; iy != newy2; iy+=ystep ) {
|
|---|
| 151 | for ( ix = newx1; ix != newx2; ix+=xstep ) {
|
|---|
| 152 | if ( iy < 0 || iy >= y || ix < 0 || ix >= x ) {
|
|---|
| 153 | if ( size == 1 )
|
|---|
| 154 | *ncp++ = 0;
|
|---|
| 155 | else
|
|---|
| 156 | *nlp++ = 0;
|
|---|
| 157 | } else {
|
|---|
| 158 | if ( size == 1 )
|
|---|
| 159 | *ncp++ = *CHARP(cp, x, ix, iy);
|
|---|
| 160 | else if ( size == 2 )
|
|---|
| 161 | *nsp++ = *SHORTP(cp, x, ix, iy);
|
|---|
| 162 | else
|
|---|
| 163 | *nlp++ = *LONGP(cp, x, ix, iy);
|
|---|
| 164 | }
|
|---|
| 165 | }
|
|---|
| 166 | }
|
|---|
| 167 | return rv;
|
|---|
| 168 | }
|
|---|
| 169 |
|
|---|
| 170 | static PyObject *
|
|---|
| 171 | imageop_scale(PyObject *self, PyObject *args)
|
|---|
| 172 | {
|
|---|
| 173 | char *cp, *ncp;
|
|---|
| 174 | short *nsp;
|
|---|
| 175 | Py_Int32 *nlp;
|
|---|
| 176 | int len, size, x, y, newx, newy, nlen;
|
|---|
| 177 | int ix, iy;
|
|---|
| 178 | int oix, oiy;
|
|---|
| 179 | PyObject *rv;
|
|---|
| 180 |
|
|---|
| 181 | if ( !PyArg_ParseTuple(args, "s#iiiii",
|
|---|
| 182 | &cp, &len, &size, &x, &y, &newx, &newy) )
|
|---|
| 183 | return 0;
|
|---|
| 184 |
|
|---|
| 185 | if ( size != 1 && size != 2 && size != 4 ) {
|
|---|
| 186 | PyErr_SetString(ImageopError, "Size should be 1, 2 or 4");
|
|---|
| 187 | return 0;
|
|---|
| 188 | }
|
|---|
| 189 | if ( !check_multiply_size(len, x, "x", y, "y", size) )
|
|---|
| 190 | return 0;
|
|---|
| 191 | nlen = newx*newy*size;
|
|---|
| 192 | if ( !check_multiply_size(nlen, newx, "newx", newy, "newy", size) )
|
|---|
| 193 | return 0;
|
|---|
| 194 |
|
|---|
| 195 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 196 | if ( rv == 0 )
|
|---|
| 197 | return 0;
|
|---|
| 198 | ncp = (char *)PyString_AsString(rv);
|
|---|
| 199 | nsp = (short *)ncp;
|
|---|
| 200 | nlp = (Py_Int32 *)ncp;
|
|---|
| 201 | for( iy = 0; iy < newy; iy++ ) {
|
|---|
| 202 | for ( ix = 0; ix < newx; ix++ ) {
|
|---|
| 203 | oix = ix * x / newx;
|
|---|
| 204 | oiy = iy * y / newy;
|
|---|
| 205 | if ( size == 1 )
|
|---|
| 206 | *ncp++ = *CHARP(cp, x, oix, oiy);
|
|---|
| 207 | else if ( size == 2 )
|
|---|
| 208 | *nsp++ = *SHORTP(cp, x, oix, oiy);
|
|---|
| 209 | else
|
|---|
| 210 | *nlp++ = *LONGP(cp, x, oix, oiy);
|
|---|
| 211 | }
|
|---|
| 212 | }
|
|---|
| 213 | return rv;
|
|---|
| 214 | }
|
|---|
| 215 |
|
|---|
| 216 | /* Note: this routine can use a bit of optimizing */
|
|---|
| 217 |
|
|---|
| 218 | static PyObject *
|
|---|
| 219 | imageop_tovideo(PyObject *self, PyObject *args)
|
|---|
| 220 | {
|
|---|
| 221 | int maxx, maxy, x, y, len;
|
|---|
| 222 | int i;
|
|---|
| 223 | unsigned char *cp, *ncp;
|
|---|
| 224 | int width;
|
|---|
| 225 | PyObject *rv;
|
|---|
| 226 |
|
|---|
| 227 |
|
|---|
| 228 | if ( !PyArg_ParseTuple(args, "s#iii", &cp, &len, &width, &maxx, &maxy) )
|
|---|
| 229 | return 0;
|
|---|
| 230 |
|
|---|
| 231 | if ( width != 1 && width != 4 ) {
|
|---|
| 232 | PyErr_SetString(ImageopError, "Size should be 1 or 4");
|
|---|
| 233 | return 0;
|
|---|
| 234 | }
|
|---|
| 235 | if ( !check_multiply_size(len, maxx, "max", maxy, "maxy", width) )
|
|---|
| 236 | return 0;
|
|---|
| 237 |
|
|---|
| 238 | rv = PyString_FromStringAndSize(NULL, len);
|
|---|
| 239 | if ( rv == 0 )
|
|---|
| 240 | return 0;
|
|---|
| 241 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 242 |
|
|---|
| 243 | if ( width == 1 ) {
|
|---|
| 244 | memcpy(ncp, cp, maxx); /* Copy first line */
|
|---|
| 245 | ncp += maxx;
|
|---|
| 246 | for (y=1; y<maxy; y++) { /* Interpolate other lines */
|
|---|
| 247 | for(x=0; x<maxx; x++) {
|
|---|
| 248 | i = y*maxx + x;
|
|---|
| 249 | *ncp++ = ((int)cp[i] + (int)cp[i-maxx]) >> 1;
|
|---|
| 250 | }
|
|---|
| 251 | }
|
|---|
| 252 | } else {
|
|---|
| 253 | memcpy(ncp, cp, maxx*4); /* Copy first line */
|
|---|
| 254 | ncp += maxx*4;
|
|---|
| 255 | for (y=1; y<maxy; y++) { /* Interpolate other lines */
|
|---|
| 256 | for(x=0; x<maxx; x++) {
|
|---|
| 257 | i = (y*maxx + x)*4 + 1;
|
|---|
| 258 | *ncp++ = 0; /* Skip alfa comp */
|
|---|
| 259 | *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1;
|
|---|
| 260 | i++;
|
|---|
| 261 | *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1;
|
|---|
| 262 | i++;
|
|---|
| 263 | *ncp++ = ((int)cp[i] + (int)cp[i-4*maxx]) >> 1;
|
|---|
| 264 | }
|
|---|
| 265 | }
|
|---|
| 266 | }
|
|---|
| 267 | return rv;
|
|---|
| 268 | }
|
|---|
| 269 |
|
|---|
| 270 | static PyObject *
|
|---|
| 271 | imageop_grey2mono(PyObject *self, PyObject *args)
|
|---|
| 272 | {
|
|---|
| 273 | int tres, x, y, len;
|
|---|
| 274 | unsigned char *cp, *ncp;
|
|---|
| 275 | unsigned char ovalue;
|
|---|
| 276 | PyObject *rv;
|
|---|
| 277 | int i, bit;
|
|---|
| 278 |
|
|---|
| 279 |
|
|---|
| 280 | if ( !PyArg_ParseTuple(args, "s#iii", &cp, &len, &x, &y, &tres) )
|
|---|
| 281 | return 0;
|
|---|
| 282 |
|
|---|
| 283 | if ( !check_multiply(len, x, y) )
|
|---|
| 284 | return 0;
|
|---|
| 285 |
|
|---|
| 286 | rv = PyString_FromStringAndSize(NULL, (len+7)/8);
|
|---|
| 287 | if ( rv == 0 )
|
|---|
| 288 | return 0;
|
|---|
| 289 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 290 |
|
|---|
| 291 | bit = 0x80;
|
|---|
| 292 | ovalue = 0;
|
|---|
| 293 | for ( i=0; i < len; i++ ) {
|
|---|
| 294 | if ( (int)cp[i] > tres )
|
|---|
| 295 | ovalue |= bit;
|
|---|
| 296 | bit >>= 1;
|
|---|
| 297 | if ( bit == 0 ) {
|
|---|
| 298 | *ncp++ = ovalue;
|
|---|
| 299 | bit = 0x80;
|
|---|
| 300 | ovalue = 0;
|
|---|
| 301 | }
|
|---|
| 302 | }
|
|---|
| 303 | if ( bit != 0x80 )
|
|---|
| 304 | *ncp++ = ovalue;
|
|---|
| 305 | return rv;
|
|---|
| 306 | }
|
|---|
| 307 |
|
|---|
| 308 | static PyObject *
|
|---|
| 309 | imageop_grey2grey4(PyObject *self, PyObject *args)
|
|---|
| 310 | {
|
|---|
| 311 | int x, y, len;
|
|---|
| 312 | unsigned char *cp, *ncp;
|
|---|
| 313 | unsigned char ovalue;
|
|---|
| 314 | PyObject *rv;
|
|---|
| 315 | int i;
|
|---|
| 316 | int pos;
|
|---|
| 317 |
|
|---|
| 318 |
|
|---|
| 319 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 320 | return 0;
|
|---|
| 321 |
|
|---|
| 322 | if ( !check_multiply(len, x, y) )
|
|---|
| 323 | return 0;
|
|---|
| 324 |
|
|---|
| 325 | rv = PyString_FromStringAndSize(NULL, (len+1)/2);
|
|---|
| 326 | if ( rv == 0 )
|
|---|
| 327 | return 0;
|
|---|
| 328 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 329 | pos = 0;
|
|---|
| 330 | ovalue = 0;
|
|---|
| 331 | for ( i=0; i < len; i++ ) {
|
|---|
| 332 | ovalue |= ((int)cp[i] & 0xf0) >> pos;
|
|---|
| 333 | pos += 4;
|
|---|
| 334 | if ( pos == 8 ) {
|
|---|
| 335 | *ncp++ = ovalue;
|
|---|
| 336 | ovalue = 0;
|
|---|
| 337 | pos = 0;
|
|---|
| 338 | }
|
|---|
| 339 | }
|
|---|
| 340 | if ( pos != 0 )
|
|---|
| 341 | *ncp++ = ovalue;
|
|---|
| 342 | return rv;
|
|---|
| 343 | }
|
|---|
| 344 |
|
|---|
| 345 | static PyObject *
|
|---|
| 346 | imageop_grey2grey2(PyObject *self, PyObject *args)
|
|---|
| 347 | {
|
|---|
| 348 | int x, y, len;
|
|---|
| 349 | unsigned char *cp, *ncp;
|
|---|
| 350 | unsigned char ovalue;
|
|---|
| 351 | PyObject *rv;
|
|---|
| 352 | int i;
|
|---|
| 353 | int pos;
|
|---|
| 354 |
|
|---|
| 355 |
|
|---|
| 356 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 357 | return 0;
|
|---|
| 358 |
|
|---|
| 359 | if ( !check_multiply(len, x, y) )
|
|---|
| 360 | return 0;
|
|---|
| 361 |
|
|---|
| 362 | rv = PyString_FromStringAndSize(NULL, (len+3)/4);
|
|---|
| 363 | if ( rv == 0 )
|
|---|
| 364 | return 0;
|
|---|
| 365 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 366 | pos = 0;
|
|---|
| 367 | ovalue = 0;
|
|---|
| 368 | for ( i=0; i < len; i++ ) {
|
|---|
| 369 | ovalue |= ((int)cp[i] & 0xc0) >> pos;
|
|---|
| 370 | pos += 2;
|
|---|
| 371 | if ( pos == 8 ) {
|
|---|
| 372 | *ncp++ = ovalue;
|
|---|
| 373 | ovalue = 0;
|
|---|
| 374 | pos = 0;
|
|---|
| 375 | }
|
|---|
| 376 | }
|
|---|
| 377 | if ( pos != 0 )
|
|---|
| 378 | *ncp++ = ovalue;
|
|---|
| 379 | return rv;
|
|---|
| 380 | }
|
|---|
| 381 |
|
|---|
| 382 | static PyObject *
|
|---|
| 383 | imageop_dither2mono(PyObject *self, PyObject *args)
|
|---|
| 384 | {
|
|---|
| 385 | int sum, x, y, len;
|
|---|
| 386 | unsigned char *cp, *ncp;
|
|---|
| 387 | unsigned char ovalue;
|
|---|
| 388 | PyObject *rv;
|
|---|
| 389 | int i, bit;
|
|---|
| 390 |
|
|---|
| 391 |
|
|---|
| 392 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 393 | return 0;
|
|---|
| 394 |
|
|---|
| 395 | if ( !check_multiply(len, x, y) )
|
|---|
| 396 | return 0;
|
|---|
| 397 |
|
|---|
| 398 | rv = PyString_FromStringAndSize(NULL, (len+7)/8);
|
|---|
| 399 | if ( rv == 0 )
|
|---|
| 400 | return 0;
|
|---|
| 401 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 402 |
|
|---|
| 403 | bit = 0x80;
|
|---|
| 404 | ovalue = 0;
|
|---|
| 405 | sum = 0;
|
|---|
| 406 | for ( i=0; i < len; i++ ) {
|
|---|
| 407 | sum += cp[i];
|
|---|
| 408 | if ( sum >= 256 ) {
|
|---|
| 409 | sum -= 256;
|
|---|
| 410 | ovalue |= bit;
|
|---|
| 411 | }
|
|---|
| 412 | bit >>= 1;
|
|---|
| 413 | if ( bit == 0 ) {
|
|---|
| 414 | *ncp++ = ovalue;
|
|---|
| 415 | bit = 0x80;
|
|---|
| 416 | ovalue = 0;
|
|---|
| 417 | }
|
|---|
| 418 | }
|
|---|
| 419 | if ( bit != 0x80 )
|
|---|
| 420 | *ncp++ = ovalue;
|
|---|
| 421 | return rv;
|
|---|
| 422 | }
|
|---|
| 423 |
|
|---|
| 424 | static PyObject *
|
|---|
| 425 | imageop_dither2grey2(PyObject *self, PyObject *args)
|
|---|
| 426 | {
|
|---|
| 427 | int x, y, len;
|
|---|
| 428 | unsigned char *cp, *ncp;
|
|---|
| 429 | unsigned char ovalue;
|
|---|
| 430 | PyObject *rv;
|
|---|
| 431 | int i;
|
|---|
| 432 | int pos;
|
|---|
| 433 | int sum = 0, nvalue;
|
|---|
| 434 |
|
|---|
| 435 |
|
|---|
| 436 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 437 | return 0;
|
|---|
| 438 |
|
|---|
| 439 | if ( !check_multiply(len, x, y) )
|
|---|
| 440 | return 0;
|
|---|
| 441 |
|
|---|
| 442 | rv = PyString_FromStringAndSize(NULL, (len+3)/4);
|
|---|
| 443 | if ( rv == 0 )
|
|---|
| 444 | return 0;
|
|---|
| 445 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 446 | pos = 1;
|
|---|
| 447 | ovalue = 0;
|
|---|
| 448 | for ( i=0; i < len; i++ ) {
|
|---|
| 449 | sum += cp[i];
|
|---|
| 450 | nvalue = sum & 0x180;
|
|---|
| 451 | sum -= nvalue;
|
|---|
| 452 | ovalue |= nvalue >> pos;
|
|---|
| 453 | pos += 2;
|
|---|
| 454 | if ( pos == 9 ) {
|
|---|
| 455 | *ncp++ = ovalue;
|
|---|
| 456 | ovalue = 0;
|
|---|
| 457 | pos = 1;
|
|---|
| 458 | }
|
|---|
| 459 | }
|
|---|
| 460 | if ( pos != 0 )
|
|---|
| 461 | *ncp++ = ovalue;
|
|---|
| 462 | return rv;
|
|---|
| 463 | }
|
|---|
| 464 |
|
|---|
| 465 | static PyObject *
|
|---|
| 466 | imageop_mono2grey(PyObject *self, PyObject *args)
|
|---|
| 467 | {
|
|---|
| 468 | int v0, v1, x, y, len, nlen;
|
|---|
| 469 | unsigned char *cp, *ncp;
|
|---|
| 470 | PyObject *rv;
|
|---|
| 471 | int i, bit;
|
|---|
| 472 |
|
|---|
| 473 | if ( !PyArg_ParseTuple(args, "s#iiii", &cp, &len, &x, &y, &v0, &v1) )
|
|---|
| 474 | return 0;
|
|---|
| 475 |
|
|---|
| 476 | nlen = x*y;
|
|---|
| 477 | if ( !check_multiply(nlen, x, y) )
|
|---|
| 478 | return 0;
|
|---|
| 479 | if ( (nlen+7)/8 != len ) {
|
|---|
| 480 | PyErr_SetString(ImageopError, "String has incorrect length");
|
|---|
| 481 | return 0;
|
|---|
| 482 | }
|
|---|
| 483 |
|
|---|
| 484 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 485 | if ( rv == 0 )
|
|---|
| 486 | return 0;
|
|---|
| 487 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 488 |
|
|---|
| 489 | bit = 0x80;
|
|---|
| 490 | for ( i=0; i < nlen; i++ ) {
|
|---|
| 491 | if ( *cp & bit )
|
|---|
| 492 | *ncp++ = v1;
|
|---|
| 493 | else
|
|---|
| 494 | *ncp++ = v0;
|
|---|
| 495 | bit >>= 1;
|
|---|
| 496 | if ( bit == 0 ) {
|
|---|
| 497 | bit = 0x80;
|
|---|
| 498 | cp++;
|
|---|
| 499 | }
|
|---|
| 500 | }
|
|---|
| 501 | return rv;
|
|---|
| 502 | }
|
|---|
| 503 |
|
|---|
| 504 | static PyObject *
|
|---|
| 505 | imageop_grey22grey(PyObject *self, PyObject *args)
|
|---|
| 506 | {
|
|---|
| 507 | int x, y, len, nlen;
|
|---|
| 508 | unsigned char *cp, *ncp;
|
|---|
| 509 | PyObject *rv;
|
|---|
| 510 | int i, pos, value = 0, nvalue;
|
|---|
| 511 |
|
|---|
| 512 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 513 | return 0;
|
|---|
| 514 |
|
|---|
| 515 | nlen = x*y;
|
|---|
| 516 | if ( !check_multiply(nlen, x, y) ) {
|
|---|
| 517 | return 0;
|
|---|
| 518 | }
|
|---|
| 519 | if ( (nlen+3)/4 != len ) {
|
|---|
| 520 | PyErr_SetString(ImageopError, "String has incorrect length");
|
|---|
| 521 | return 0;
|
|---|
| 522 | }
|
|---|
| 523 |
|
|---|
| 524 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 525 | if ( rv == 0 )
|
|---|
| 526 | return 0;
|
|---|
| 527 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 528 |
|
|---|
| 529 | pos = 0;
|
|---|
| 530 | for ( i=0; i < nlen; i++ ) {
|
|---|
| 531 | if ( pos == 0 ) {
|
|---|
| 532 | value = *cp++;
|
|---|
| 533 | pos = 8;
|
|---|
| 534 | }
|
|---|
| 535 | pos -= 2;
|
|---|
| 536 | nvalue = (value >> pos) & 0x03;
|
|---|
| 537 | *ncp++ = nvalue | (nvalue << 2) |
|
|---|
| 538 | (nvalue << 4) | (nvalue << 6);
|
|---|
| 539 | }
|
|---|
| 540 | return rv;
|
|---|
| 541 | }
|
|---|
| 542 |
|
|---|
| 543 | static PyObject *
|
|---|
| 544 | imageop_grey42grey(PyObject *self, PyObject *args)
|
|---|
| 545 | {
|
|---|
| 546 | int x, y, len, nlen;
|
|---|
| 547 | unsigned char *cp, *ncp;
|
|---|
| 548 | PyObject *rv;
|
|---|
| 549 | int i, pos, value = 0, nvalue;
|
|---|
| 550 |
|
|---|
| 551 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 552 | return 0;
|
|---|
| 553 |
|
|---|
| 554 | nlen = x*y;
|
|---|
| 555 | if ( !check_multiply(nlen, x, y) )
|
|---|
| 556 | return 0;
|
|---|
| 557 | if ( (nlen+1)/2 != len ) {
|
|---|
| 558 | PyErr_SetString(ImageopError, "String has incorrect length");
|
|---|
| 559 | return 0;
|
|---|
| 560 | }
|
|---|
| 561 |
|
|---|
| 562 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 563 | if ( rv == 0 )
|
|---|
| 564 | return 0;
|
|---|
| 565 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 566 |
|
|---|
| 567 | pos = 0;
|
|---|
| 568 | for ( i=0; i < nlen; i++ ) {
|
|---|
| 569 | if ( pos == 0 ) {
|
|---|
| 570 | value = *cp++;
|
|---|
| 571 | pos = 8;
|
|---|
| 572 | }
|
|---|
| 573 | pos -= 4;
|
|---|
| 574 | nvalue = (value >> pos) & 0x0f;
|
|---|
| 575 | *ncp++ = nvalue | (nvalue << 4);
|
|---|
| 576 | }
|
|---|
| 577 | return rv;
|
|---|
| 578 | }
|
|---|
| 579 |
|
|---|
| 580 | static PyObject *
|
|---|
| 581 | imageop_rgb2rgb8(PyObject *self, PyObject *args)
|
|---|
| 582 | {
|
|---|
| 583 | int x, y, len, nlen;
|
|---|
| 584 | unsigned char *cp;
|
|---|
| 585 | unsigned char *ncp;
|
|---|
| 586 | PyObject *rv;
|
|---|
| 587 | int i, r, g, b;
|
|---|
| 588 | int backward_compatible = imageop_backward_compatible();
|
|---|
| 589 |
|
|---|
| 590 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 591 | return 0;
|
|---|
| 592 |
|
|---|
| 593 | if ( !check_multiply_size(len, x, "x", y, "y", 4) )
|
|---|
| 594 | return 0;
|
|---|
| 595 | nlen = x*y;
|
|---|
| 596 | if ( !check_multiply(nlen, x, y) )
|
|---|
| 597 | return 0;
|
|---|
| 598 |
|
|---|
| 599 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 600 | if ( rv == 0 )
|
|---|
| 601 | return 0;
|
|---|
| 602 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 603 |
|
|---|
| 604 | for ( i=0; i < nlen; i++ ) {
|
|---|
| 605 | /* Bits in source: aaaaaaaa BBbbbbbb GGGggggg RRRrrrrr */
|
|---|
| 606 | if (backward_compatible) {
|
|---|
| 607 | Py_UInt32 value = * (Py_UInt32 *) cp;
|
|---|
| 608 | cp += 4;
|
|---|
| 609 | r = (int) ((value & 0xff) / 255. * 7. + .5);
|
|---|
| 610 | g = (int) (((value >> 8) & 0xff) / 255. * 7. + .5);
|
|---|
| 611 | b = (int) (((value >> 16) & 0xff) / 255. * 3. + .5);
|
|---|
| 612 | } else {
|
|---|
| 613 | cp++; /* skip alpha channel */
|
|---|
| 614 | b = (int) (*cp++ / 255. * 3. + .5);
|
|---|
| 615 | g = (int) (*cp++ / 255. * 7. + .5);
|
|---|
| 616 | r = (int) (*cp++ / 255. * 7. + .5);
|
|---|
| 617 | }
|
|---|
| 618 | *ncp++ = (unsigned char)((r<<5) | (b<<3) | g);
|
|---|
| 619 | }
|
|---|
| 620 | return rv;
|
|---|
| 621 | }
|
|---|
| 622 |
|
|---|
| 623 | static PyObject *
|
|---|
| 624 | imageop_rgb82rgb(PyObject *self, PyObject *args)
|
|---|
| 625 | {
|
|---|
| 626 | int x, y, len, nlen;
|
|---|
| 627 | unsigned char *cp;
|
|---|
| 628 | unsigned char *ncp;
|
|---|
| 629 | PyObject *rv;
|
|---|
| 630 | int i, r, g, b;
|
|---|
| 631 | unsigned char value;
|
|---|
| 632 | int backward_compatible = imageop_backward_compatible();
|
|---|
| 633 |
|
|---|
| 634 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 635 | return 0;
|
|---|
| 636 |
|
|---|
| 637 | if ( !check_multiply(len, x, y) )
|
|---|
| 638 | return 0;
|
|---|
| 639 | nlen = x*y*4;
|
|---|
| 640 | if ( !check_multiply_size(nlen, x, "x", y, "y", 4) )
|
|---|
| 641 | return 0;
|
|---|
| 642 |
|
|---|
| 643 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 644 | if ( rv == 0 )
|
|---|
| 645 | return 0;
|
|---|
| 646 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 647 |
|
|---|
| 648 | for ( i=0; i < len; i++ ) {
|
|---|
| 649 | /* Bits in source: RRRBBGGG
|
|---|
| 650 | ** Red and Green are multiplied by 36.5, Blue by 85
|
|---|
| 651 | */
|
|---|
| 652 | value = *cp++;
|
|---|
| 653 | r = (value >> 5) & 7;
|
|---|
| 654 | g = (value ) & 7;
|
|---|
| 655 | b = (value >> 3) & 3;
|
|---|
| 656 | r = (r<<5) | (r<<3) | (r>>1);
|
|---|
| 657 | g = (g<<5) | (g<<3) | (g>>1);
|
|---|
| 658 | b = (b<<6) | (b<<4) | (b<<2) | b;
|
|---|
| 659 | if (backward_compatible) {
|
|---|
| 660 | Py_UInt32 nvalue = r | (g<<8) | (b<<16);
|
|---|
| 661 | * (Py_UInt32 *) ncp = nvalue;
|
|---|
| 662 | ncp += 4;
|
|---|
| 663 | } else {
|
|---|
| 664 | *ncp++ = 0;
|
|---|
| 665 | *ncp++ = b;
|
|---|
| 666 | *ncp++ = g;
|
|---|
| 667 | *ncp++ = r;
|
|---|
| 668 | }
|
|---|
| 669 | }
|
|---|
| 670 | return rv;
|
|---|
| 671 | }
|
|---|
| 672 |
|
|---|
| 673 | static PyObject *
|
|---|
| 674 | imageop_rgb2grey(PyObject *self, PyObject *args)
|
|---|
| 675 | {
|
|---|
| 676 | int x, y, len, nlen;
|
|---|
| 677 | unsigned char *cp;
|
|---|
| 678 | unsigned char *ncp;
|
|---|
| 679 | PyObject *rv;
|
|---|
| 680 | int i, r, g, b;
|
|---|
| 681 | int nvalue;
|
|---|
| 682 | int backward_compatible = imageop_backward_compatible();
|
|---|
| 683 |
|
|---|
| 684 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 685 | return 0;
|
|---|
| 686 |
|
|---|
| 687 | if ( !check_multiply_size(len, x, "x", y, "y", 4) )
|
|---|
| 688 | return 0;
|
|---|
| 689 | nlen = x*y;
|
|---|
| 690 | if ( !check_multiply(nlen, x, y) )
|
|---|
| 691 | return 0;
|
|---|
| 692 |
|
|---|
| 693 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 694 | if ( rv == 0 )
|
|---|
| 695 | return 0;
|
|---|
| 696 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 697 |
|
|---|
| 698 | for ( i=0; i < nlen; i++ ) {
|
|---|
| 699 | if (backward_compatible) {
|
|---|
| 700 | Py_UInt32 value = * (Py_UInt32 *) cp;
|
|---|
| 701 | cp += 4;
|
|---|
| 702 | r = (int) ((value & 0xff) / 255. * 7. + .5);
|
|---|
| 703 | g = (int) (((value >> 8) & 0xff) / 255. * 7. + .5);
|
|---|
| 704 | b = (int) (((value >> 16) & 0xff) / 255. * 3. + .5);
|
|---|
| 705 | } else {
|
|---|
| 706 | cp++; /* skip alpha channel */
|
|---|
| 707 | b = *cp++;
|
|---|
| 708 | g = *cp++;
|
|---|
| 709 | r = *cp++;
|
|---|
| 710 | }
|
|---|
| 711 | nvalue = (int)(0.30*r + 0.59*g + 0.11*b);
|
|---|
| 712 | if ( nvalue > 255 ) nvalue = 255;
|
|---|
| 713 | *ncp++ = (unsigned char)nvalue;
|
|---|
| 714 | }
|
|---|
| 715 | return rv;
|
|---|
| 716 | }
|
|---|
| 717 |
|
|---|
| 718 | static PyObject *
|
|---|
| 719 | imageop_grey2rgb(PyObject *self, PyObject *args)
|
|---|
| 720 | {
|
|---|
| 721 | int x, y, len, nlen;
|
|---|
| 722 | unsigned char *cp;
|
|---|
| 723 | unsigned char *ncp;
|
|---|
| 724 | PyObject *rv;
|
|---|
| 725 | int i;
|
|---|
| 726 | unsigned char value;
|
|---|
| 727 | int backward_compatible = imageop_backward_compatible();
|
|---|
| 728 |
|
|---|
| 729 | if ( !PyArg_ParseTuple(args, "s#ii", &cp, &len, &x, &y) )
|
|---|
| 730 | return 0;
|
|---|
| 731 |
|
|---|
| 732 | if ( !check_multiply(len, x, y) )
|
|---|
| 733 | return 0;
|
|---|
| 734 | nlen = x*y*4;
|
|---|
| 735 | if ( !check_multiply_size(nlen, x, "x", y, "y", 4) )
|
|---|
| 736 | return 0;
|
|---|
| 737 |
|
|---|
| 738 | rv = PyString_FromStringAndSize(NULL, nlen);
|
|---|
| 739 | if ( rv == 0 )
|
|---|
| 740 | return 0;
|
|---|
| 741 | ncp = (unsigned char *)PyString_AsString(rv);
|
|---|
| 742 |
|
|---|
| 743 | for ( i=0; i < len; i++ ) {
|
|---|
| 744 | value = *cp++;
|
|---|
| 745 | if (backward_compatible) {
|
|---|
| 746 | * (Py_UInt32 *) ncp = (Py_UInt32) value | ((Py_UInt32) value << 8 ) | ((Py_UInt32) value << 16);
|
|---|
| 747 | ncp += 4;
|
|---|
| 748 | } else {
|
|---|
| 749 | *ncp++ = 0;
|
|---|
| 750 | *ncp++ = value;
|
|---|
| 751 | *ncp++ = value;
|
|---|
| 752 | *ncp++ = value;
|
|---|
| 753 | }
|
|---|
| 754 | }
|
|---|
| 755 | return rv;
|
|---|
| 756 | }
|
|---|
| 757 |
|
|---|
| 758 | static PyMethodDef imageop_methods[] = {
|
|---|
| 759 | { "crop", imageop_crop, METH_VARARGS },
|
|---|
| 760 | { "scale", imageop_scale, METH_VARARGS },
|
|---|
| 761 | { "grey2mono", imageop_grey2mono, METH_VARARGS },
|
|---|
| 762 | { "grey2grey2", imageop_grey2grey2, METH_VARARGS },
|
|---|
| 763 | { "grey2grey4", imageop_grey2grey4, METH_VARARGS },
|
|---|
| 764 | { "dither2mono", imageop_dither2mono, METH_VARARGS },
|
|---|
| 765 | { "dither2grey2", imageop_dither2grey2, METH_VARARGS },
|
|---|
| 766 | { "mono2grey", imageop_mono2grey, METH_VARARGS },
|
|---|
| 767 | { "grey22grey", imageop_grey22grey, METH_VARARGS },
|
|---|
| 768 | { "grey42grey", imageop_grey42grey, METH_VARARGS },
|
|---|
| 769 | { "tovideo", imageop_tovideo, METH_VARARGS },
|
|---|
| 770 | { "rgb2rgb8", imageop_rgb2rgb8, METH_VARARGS },
|
|---|
| 771 | { "rgb82rgb", imageop_rgb82rgb, METH_VARARGS },
|
|---|
| 772 | { "rgb2grey", imageop_rgb2grey, METH_VARARGS },
|
|---|
| 773 | { "grey2rgb", imageop_grey2rgb, METH_VARARGS },
|
|---|
| 774 | { 0, 0 }
|
|---|
| 775 | };
|
|---|
| 776 |
|
|---|
| 777 |
|
|---|
| 778 | PyMODINIT_FUNC
|
|---|
| 779 | initimageop(void)
|
|---|
| 780 | {
|
|---|
| 781 | PyObject *m;
|
|---|
| 782 |
|
|---|
| 783 | if (PyErr_WarnPy3k("the imageop module has been removed in "
|
|---|
| 784 | "Python 3.0", 2) < 0)
|
|---|
| 785 | return;
|
|---|
| 786 |
|
|---|
| 787 | m = Py_InitModule("imageop", imageop_methods);
|
|---|
| 788 | if (m == NULL)
|
|---|
| 789 | return;
|
|---|
| 790 | ImageopDict = PyModule_GetDict(m);
|
|---|
| 791 | ImageopError = PyErr_NewException("imageop.error", NULL, NULL);
|
|---|
| 792 | if (ImageopError != NULL)
|
|---|
| 793 | PyDict_SetItemString(ImageopDict, "error", ImageopError);
|
|---|
| 794 | }
|
|---|