opencv/interfaces/python/cv.cpp

   1 #include <Python.h>
   2
   3 #include <assert.h>
   4
   5 #include <opencv/cxcore.h>
   6 #include <opencv/cv.h>
   7 #include <opencv/cvaux.h>
   8 #include <opencv/cvwimage.h>
   9 #include <opencv/highgui.h>
  10
  11 #define MODULESTR "cv"
  12
  13 static PyObject *opencv_error;
  14
  15 struct memtrack_t {
  16   PyObject_HEAD
  17   void *ptr;
  18   Py_ssize_t size;
  19 };
  20
  21 struct iplimage_t {
  22   PyObject_HEAD
  23   IplImage *a;
  24   PyObject *data;
  25   size_t offset;
  26 };
  27
  28 struct cvmat_t {
  29   PyObject_HEAD
  30   CvMat *a;
  31   PyObject *data;
  32   size_t offset;
  33 };
  34
  35 struct cvmatnd_t {
  36   PyObject_HEAD
  37   CvMatND *a;
  38   PyObject *data;
  39   size_t offset;
  40 };
  41
  42 struct cvhistogram_t {
  43   PyObject_HEAD
  44   CvHistogram h;
  45   PyObject *bins;
  46 };
  47
  48 struct cvmemstorage_t {
  49   PyObject_HEAD
  50   CvMemStorage *a;
  51 };
  52
  53 struct cvseq_t {
  54   PyObject_HEAD
  55   CvSeq *a;
  56   PyObject *container;  // Containing cvmemstorage_t
  57 };
  58
  59 struct cvset_t {
  60   PyObject_HEAD
  61   CvSet *a;
  62   PyObject *container;  // Containing cvmemstorage_t
  63   int i;
  64 };
  65
  66 struct cvsubdiv2d_t {
  67   PyObject_HEAD
  68   CvSubdiv2D *a;
  69   PyObject *container;  // Containing cvmemstorage_t
  70 };
  71
  72 struct cvsubdiv2dpoint_t {
  73   PyObject_HEAD
  74   CvSubdiv2DPoint *a;
  75   PyObject *container;  // Containing cvmemstorage_t
  76 };
  77
  78 struct cvsubdiv2dedge_t {
  79   PyObject_HEAD
  80   CvSubdiv2DEdge a;
  81   PyObject *container;  // Containing cvmemstorage_t
  82 };
  83
  84 struct cvlineiterator_t {
  85   PyObject_HEAD
  86   CvLineIterator iter;
  87   int count;
  88   int type;
  89 };
  90
  91 struct iplconvkernel_t {
  92   PyObject_HEAD
  93   IplConvKernel *a;
  94 };
  95
  96 struct cvcapture_t {
  97   PyObject_HEAD
  98   CvCapture *a;
  99 };
 100
 101 struct cvvideowriter_t {
 102   PyObject_HEAD
 103   CvVideoWriter *a;
 104 };
 105
 106 typedef IplImage ROIplImage;
 107
 108 struct cvmoments_t {
 109   PyObject_HEAD
 110   CvMoments a;
 111 };
 112
 113 struct cvfont_t {
 114   PyObject_HEAD
 115   CvFont a;
 116 };
 117
 118 struct cvhaarclassifiercascade_t {
 119   PyObject_HEAD
 120   CvHaarClassifierCascade *a;
 121 };
 122
 123 struct cvcontourtree_t {
 124   PyObject_HEAD
 125   CvContourTree *a;
 126 };
 127
 128 struct cvpositobject_t {
 129   PyObject_HEAD
 130   CvPOSITObject *a;
 131 };
 132
 133 struct cvrng_t {
 134   PyObject_HEAD
 135   CvRNG a;
 136 };
 137
 138 static int is_iplimage(PyObject *o);
 139 static int is_cvmat(PyObject *o);
 140 static int is_cvmatnd(PyObject *o);
 141 static int convert_to_CvArr(PyObject *o, CvArr **dst, const char *name = "no_name");
 142 static int convert_to_IplImage(PyObject *o, IplImage **dst, const char *name = "no_name");
 143 static int convert_to_CvMat(PyObject *o, CvMat **dst, const char *name = "no_name");
 144 static int convert_to_CvMatND(PyObject *o, CvMatND **dst, const char *name = "no_name");
 145 static PyObject *what_data(PyObject *o);
 146
 147 static void translate_error_to_exception(void)
 148 {
 149   PyErr_SetString(opencv_error, cvErrorStr(cvGetErrStatus()));
 150   cvSetErrStatus(0);
 151 }
 152
 153 #define ERRCHK do { if (cvGetErrStatus() != 0) { translate_error_to_exception(); return NULL; } } while (0)
 154 #define ERRWRAP(F) \
 155     do { \
 156         try \
 157         { \
 158             F; \
 159         } \
 160         catch (const cv::Exception &e) \
 161         { \
 162            PyErr_SetString(opencv_error, e.err.c_str()); \
 163            return NULL; \
 164         } \
 165         ERRCHK; \
 166     } while(0)
 167
 168 /************************************************************************/
 169
 170 static int failmsg(const char *fmt, ...)
 171 {
 172   char str[1000];
 173
 174   va_list ap;
 175   va_start(ap, fmt);
 176   vsnprintf(str, sizeof(str), fmt, ap);
 177   va_end(ap);
 178
 179   PyErr_SetString(PyExc_TypeError, str);
 180   return 0;
 181 }
 182
 183 /************************************************************************/
 184
 185 /* These get/setters are polymorphic, used in both iplimage and cvmat */
 186
 187 static PyObject *PyObject_FromCvScalar(CvScalar s, int type)
 188 {
 189   int i, spe = CV_MAT_CN(type);
 190   PyObject *r;
 191   if (spe > 1) {
 192     r = PyTuple_New(spe);
 193     for (i = 0; i < spe; i++)
 194       PyTuple_SET_ITEM(r, i, PyFloat_FromDouble(s.val[i]));
 195   } else {
 196     r = PyFloat_FromDouble(s.val[0]);
 197   }
 198   return r;
 199 }
 200
 201 static PyObject *cvarr_GetItem(PyObject *o, PyObject *key);
 202 static int cvarr_SetItem(PyObject *o, PyObject *key, PyObject *v);
 203
 204 // o is a Python string or buffer object.  Return its size.
 205
 206 static Py_ssize_t what_size(PyObject *o)
 207 {
 208   void *buffer;
 209   Py_ssize_t buffer_len;
 210
 211   if (PyString_Check(o)) {
 212     return PyString_Size(o);
 213   } else if (PyObject_AsWriteBuffer(o, &buffer, &buffer_len) == 0) {
 214     return buffer_len;
 215   } else {
 216     assert(0);  // argument must be string or buffer.
 217     return 0;
 218   }
 219 }
 220
 221
 222 /************************************************************************/
 223
 224 /* iplimage */
 225
 226 static void iplimage_dealloc(PyObject *self)
 227 {
 228   iplimage_t *pc = (iplimage_t*)self;
 229   cvReleaseImageHeader((IplImage**)&pc->a);
 230   Py_DECREF(pc->data);
 231   PyObject_Del(self);
 232 }
 233
 234 static PyObject *iplimage_repr(PyObject *self)
 235 {
 236   iplimage_t *cva = (iplimage_t*)self;
 237   IplImage* ipl = (IplImage*)(cva->a);
 238   char str[1000];
 239   sprintf(str, "<iplimage(");
 240   char *d = str + strlen(str);
 241   sprintf(d, "nChannels=%d ", ipl->nChannels);
 242   d += strlen(d);
 243   sprintf(d, "width=%d ", ipl->width);
 244   d += strlen(d);
 245   sprintf(d, "height=%d ", ipl->height);
 246   d += strlen(d);
 247   sprintf(d, "widthStep=%d ", ipl->widthStep);
 248   d += strlen(d);
 249   sprintf(d, ")>");
 250   return PyString_FromString(str);
 251 }
 252
 253 static PyObject *iplimage_tostring(PyObject *self, PyObject *args)
 254 {
 255   iplimage_t *pc = (iplimage_t*)self;
 256   IplImage *i;
 257   if (!convert_to_IplImage(self, &i, "self"))
 258     return NULL;
 259   if (i == NULL)
 260     return NULL;
 261   int bps;
 262   switch (i->depth) {
 263   case IPL_DEPTH_8U:
 264   case IPL_DEPTH_8S:
 265     bps = 1;
 266     break;
 267   case IPL_DEPTH_16U:
 268   case IPL_DEPTH_16S:
 269     bps = 2;
 270     break;
 271   case IPL_DEPTH_32S:
 272   case IPL_DEPTH_32F:
 273     bps = 4;
 274     break;
 275   case IPL_DEPTH_64F:
 276     bps = 8;
 277     break;
 278   default:
 279     return (PyObject*)failmsg("Unrecognised depth %d", i->depth);
 280   }
 281   int bpl = i->width * i->nChannels * bps;
 282   if (PyString_Check(pc->data) && bpl == i->widthStep && pc->offset == 0 && ((bpl * i->height) == what_size(pc->data))) {
 283     Py_INCREF(pc->data);
 284     return pc->data;
 285   } else {
 286     int l = bpl * i->height;
 287     char *s = new char[l];
 288     int y;
 289     for (y = 0; y < i->height; y++) {
 290       memcpy(s + y * bpl, i->imageData + y * i->widthStep, bpl);
 291     }
 292     PyObject *r = PyString_FromStringAndSize(s, l);
 293     delete s;
 294     return r;
 295   }
 296 }
 297
 298 static struct PyMethodDef iplimage_methods[] =
 299 {
 300   {"tostring", iplimage_tostring, METH_VARARGS},
 301   {NULL,          NULL}
 302 };
 303
 304 static PyObject *iplimage_getnChannels(iplimage_t *cva)
 305 {
 306   return PyInt_FromLong(((IplImage*)(cva->a))->nChannels);
 307 }
 308 static PyObject *iplimage_getwidth(iplimage_t *cva)
 309 {
 310   return PyInt_FromLong(((IplImage*)(cva->a))->width);
 311 }
 312 static PyObject *iplimage_getheight(iplimage_t *cva)
 313 {
 314   return PyInt_FromLong(((IplImage*)(cva->a))->height);
 315 }
 316 static PyObject *iplimage_getdepth(iplimage_t *cva)
 317 {
 318   return PyLong_FromUnsignedLong((unsigned)((IplImage*)(cva->a))->depth);
 319 }
 320 static PyObject *iplimage_getorigin(iplimage_t *cva)
 321 {
 322   return PyInt_FromLong(((IplImage*)(cva->a))->origin);
 323 }
 324 static void iplimage_setorigin(iplimage_t *cva, PyObject *v)
 325 {
 326   ((IplImage*)(cva->a))->origin = PyInt_AsLong(v);
 327 }
 328
 329 static PyGetSetDef iplimage_getseters[] = {
 330   {(char*)"nChannels", (getter)iplimage_getnChannels, (setter)NULL, (char*)"nChannels", NULL},
 331   {(char*)"width", (getter)iplimage_getwidth, (setter)NULL, (char*)"width", NULL},
 332   {(char*)"height", (getter)iplimage_getheight, (setter)NULL, (char*)"height", NULL},
 333   {(char*)"depth", (getter)iplimage_getdepth, (setter)NULL, (char*)"depth", NULL},
 334   {(char*)"origin", (getter)iplimage_getorigin, (setter)iplimage_setorigin, (char*)"origin", NULL},
 335   {NULL}  /* Sentinel */
 336 };
 337
 338 static PyMappingMethods iplimage_as_map = {
 339   NULL,
 340   &cvarr_GetItem,
 341   &cvarr_SetItem,
 342 };
 343
 344 static PyTypeObject iplimage_Type = {
 345   PyObject_HEAD_INIT(&PyType_Type)
 346   0,                                      /*size*/
 347   MODULESTR".iplimage",                          /*name*/
 348   sizeof(iplimage_t),                        /*basicsize*/
 349 };
 350
 351 static void iplimage_specials(void)
 352 {
 353   iplimage_Type.tp_dealloc = iplimage_dealloc;
 354   iplimage_Type.tp_as_mapping = &iplimage_as_map;
 355   iplimage_Type.tp_repr = iplimage_repr;
 356   iplimage_Type.tp_methods = iplimage_methods;
 357   iplimage_Type.tp_getset = iplimage_getseters;
 358 }
 359
 360 static int is_iplimage(PyObject *o)
 361 {
 362   return PyType_IsSubtype(o->ob_type, &iplimage_Type);
 363 }
 364
 365 /************************************************************************/
 366
 367 /* cvmat */
 368
 369 static void cvmat_dealloc(PyObject *self)
 370 {
 371   cvmat_t *pc = (cvmat_t*)self;
 372   Py_DECREF(pc->data);
 373   cvFree((void**)&pc->a);
 374   PyObject_Del(self);
 375 }
 376
 377 static PyObject *cvmat_repr(PyObject *self)
 378 {
 379   CvMat *m = ((cvmat_t*)self)->a;
 380   char str[1000];
 381   sprintf(str, "<cvmat(");
 382   char *d = str + strlen(str);
 383   sprintf(d, "type=%08x ", m->type);
 384   d += strlen(d);
 385   sprintf(d, "rows=%d ", m->rows);
 386   d += strlen(d);
 387   sprintf(d, "cols=%d ", m->cols);
 388   d += strlen(d);
 389   sprintf(d, "step=%d ", m->step);
 390   d += strlen(d);
 391   sprintf(d, ")>");
 392   return PyString_FromString(str);
 393 }
 394
 395 static PyObject *cvmat_tostring(PyObject *self, PyObject *args)
 396 {
 397   CvMat *m;
 398   if (!convert_to_CvMat(self, &m, "self"))
 399     return NULL;
 400
 401   int bps;                     // bytes per sample
 402
 403   switch (CV_MAT_DEPTH(m->type)) {
 404   case CV_8U:
 405   case CV_8S:
 406     bps = CV_MAT_CN(m->type) * 1;
 407     break;
 408   case CV_16U:
 409   case CV_16S:
 410     bps = CV_MAT_CN(m->type) * 2;
 411     break;
 412   case CV_32S:
 413   case CV_32F:
 414     bps = CV_MAT_CN(m->type) * 4;
 415     break;
 416   case CV_64F:
 417     bps = CV_MAT_CN(m->type) * 8;
 418     break;
 419   default:
 420     return (PyObject*)failmsg("Unrecognised depth %d", CV_MAT_DEPTH(m->type));
 421   }
 422
 423   int bpl = m->cols * bps; // bytes per line
 424   cvmat_t *pc = (cvmat_t*)self;
 425   if (PyString_Check(pc->data) && bpl == m->step && pc->offset == 0 && ((bpl * m->rows) == what_size(pc->data))) {
 426     Py_INCREF(pc->data);
 427     return pc->data;
 428   } else {
 429     int l = bpl * m->rows;
 430     char *s = new char[l];
 431     int y;
 432     for (y = 0; y < m->rows; y++) {
 433       memcpy(s + y * bpl, m->data.ptr + y * m->step, bpl);
 434     }
 435     PyObject *r = PyString_FromStringAndSize(s, l);
 436     delete s;
 437     return r;
 438   }
 439 }
 440
 441 static struct PyMethodDef cvmat_methods[] =
 442 {
 443   {"tostring", cvmat_tostring, METH_VARARGS},
 444   {NULL,          NULL}
 445 };
 446
 447 static PyObject *cvmat_gettype(cvmat_t *cva)
 448 {
 449   return PyInt_FromLong(cvGetElemType(cva->a));
 450 }
 451
 452 static PyObject *cvmat_getstep(cvmat_t *cva)
 453 {
 454   return PyInt_FromLong(cva->a->step);
 455 }
 456
 457 static PyObject *cvmat_getrows(cvmat_t *cva)
 458 {
 459   return PyInt_FromLong(cva->a->rows);
 460 }
 461
 462 static PyObject *cvmat_getcols(cvmat_t *cva)
 463 {
 464   return PyInt_FromLong(cva->a->cols);
 465 }
 466
 467 static PyGetSetDef cvmat_getseters[] = {
 468   {(char*)"type",   (getter)cvmat_gettype, (setter)NULL, (char*)"type",   NULL},
 469   {(char*)"step",   (getter)cvmat_getstep, (setter)NULL, (char*)"step",   NULL},
 470   {(char*)"rows",   (getter)cvmat_getrows, (setter)NULL, (char*)"rows",   NULL},
 471   {(char*)"cols",   (getter)cvmat_getcols, (setter)NULL, (char*)"cols",   NULL},
 472   {(char*)"width",  (getter)cvmat_getcols, (setter)NULL, (char*)"width",  NULL},
 473   {(char*)"height", (getter)cvmat_getrows, (setter)NULL, (char*)"height", NULL},
 474   {NULL}  /* Sentinel */
 475 };
 476
 477 static PyMappingMethods cvmat_as_map = {
 478   NULL,
 479   &cvarr_GetItem,
 480   &cvarr_SetItem,
 481 };
 482
 483 static PyTypeObject cvmat_Type = {
 484   PyObject_HEAD_INIT(&PyType_Type)
 485   0,                                      /*size*/
 486   MODULESTR".cvmat",                      /*name*/
 487   sizeof(cvmat_t),                        /*basicsize*/
 488 };
 489
 490 static void cvmat_specials(void)
 491 {
 492   cvmat_Type.tp_dealloc = cvmat_dealloc;
 493   cvmat_Type.tp_as_mapping = &cvmat_as_map;
 494   cvmat_Type.tp_repr = cvmat_repr;
 495   cvmat_Type.tp_methods = cvmat_methods;
 496   cvmat_Type.tp_getset = cvmat_getseters;
 497 }
 498
 499 static int is_cvmat(PyObject *o)
 500 {
 501   return PyType_IsSubtype(o->ob_type, &cvmat_Type);
 502 }
 503
 504 /************************************************************************/
 505
 506 /* cvmatnd */
 507
 508 static void cvmatnd_dealloc(PyObject *self)
 509 {
 510   cvmatnd_t *pc = (cvmatnd_t*)self;
 511   Py_DECREF(pc->data);
 512   PyObject_Del(self);
 513 }
 514
 515 static PyObject *cvmatnd_repr(PyObject *self)
 516 {
 517   CvMatND *m = ((cvmatnd_t*)self)->a;
 518   char str[1000];
 519   sprintf(str, "<cvmatnd(");
 520   char *d = str + strlen(str);
 521   sprintf(d, "type=%08x ", m->type);
 522   d += strlen(d);
 523   sprintf(d, ")>");
 524   return PyString_FromString(str);
 525 }
 526
 527 static size_t cvmatnd_size(CvMatND *m)
 528 {
 529   int bps;
 530   switch (CV_MAT_DEPTH(m->type)) {
 531   case CV_8U:
 532   case CV_8S:
 533     bps = CV_MAT_CN(m->type) * 1;
 534     break;
 535   case CV_16U:
 536   case CV_16S:
 537     bps = CV_MAT_CN(m->type) * 2;
 538     break;
 539   case CV_32S:
 540   case CV_32F:
 541     bps = CV_MAT_CN(m->type) * 4;
 542     break;
 543   case CV_64F:
 544     bps = CV_MAT_CN(m->type) * 8;
 545     break;
 546   default:
 547     assert(0);
 548   }
 549   size_t l = bps;
 550   for (int d = 0; d < m->dims; d++) {
 551     l *= m->dim[d].size;
 552   }
 553   return l;
 554 }
 555
 556 static PyObject *cvmatnd_tostring(PyObject *self, PyObject *args)
 557 {
 558   CvMatND *m;
 559   if (!convert_to_CvMatND(self, &m, "self"))
 560     return NULL;
 561
 562   int bps;
 563   switch (CV_MAT_DEPTH(m->type)) {
 564   case CV_8U:
 565   case CV_8S:
 566     bps = CV_MAT_CN(m->type) * 1;
 567     break;
 568   case CV_16U:
 569   case CV_16S:
 570     bps = CV_MAT_CN(m->type) * 2;
 571     break;
 572   case CV_32S:
 573   case CV_32F:
 574     bps = CV_MAT_CN(m->type) * 4;
 575     break;
 576   case CV_64F:
 577     bps = CV_MAT_CN(m->type) * 8;
 578     break;
 579   default:
 580     return (PyObject*)failmsg("Unrecognised depth %d", CV_MAT_DEPTH(m->type));
 581   }
 582
 583   int l = bps;
 584   for (int d = 0; d < m->dims; d++) {
 585     l *= m->dim[d].size;
 586   }
 587   int i[CV_MAX_DIM];
 588   int d;
 589   for (d = 0; d < m->dims; d++) {
 590     i[d] = 0;
 591   }
 592   int rowsize = m->dim[m->dims-1].size * bps;
 593   char *s = new char[l];
 594   char *ps = s;
 595
 596   int finished = 0;
 597   while (!finished) {
 598     memcpy(ps, cvPtrND(m, i), rowsize);
 599     ps += rowsize;
 600     for (d = m->dims - 2; 0 <= d; d--) {
 601       if (++i[d] < cvGetDimSize(m, d)) {
 602         break;
 603       } else {
 604         i[d] = 0;
 605       }
 606     }
 607     if (d < 0)
 608       finished = 1;
 609   }
 610
 611   return PyString_FromStringAndSize(s, ps - s);
 612 }
 613
 614 static struct PyMethodDef cvmatnd_methods[] =
 615 {
 616   {"tostring", cvmatnd_tostring, METH_VARARGS},
 617   {NULL,          NULL}
 618 };
 619
 620 static PyMappingMethods cvmatnd_as_map = {
 621   NULL,
 622   &cvarr_GetItem,
 623   &cvarr_SetItem,
 624 };
 625
 626 static PyTypeObject cvmatnd_Type = {
 627   PyObject_HEAD_INIT(&PyType_Type)
 628   0,                                      /*size*/
 629   MODULESTR".cvmatnd",                          /*name*/
 630   sizeof(cvmatnd_t),                        /*basicsize*/
 631 };
 632
 633 static void cvmatnd_specials(void)
 634 {
 635   cvmatnd_Type.tp_dealloc = cvmatnd_dealloc;
 636   cvmatnd_Type.tp_as_mapping = &cvmatnd_as_map;
 637   cvmatnd_Type.tp_repr = cvmatnd_repr;
 638   cvmatnd_Type.tp_methods = cvmatnd_methods;
 639 }
 640
 641 static int is_cvmatnd(PyObject *o)
 642 {
 643   return PyType_IsSubtype(o->ob_type, &cvmatnd_Type);
 644 }
 645
 646 /************************************************************************/
 647
 648 /* cvhistogram */
 649
 650 static void cvhistogram_dealloc(PyObject *self)
 651 {
 652   cvhistogram_t *cvh = (cvhistogram_t*)self;
 653   Py_DECREF(cvh->bins);
 654   PyObject_Del(self);
 655 }
 656
 657 static PyTypeObject cvhistogram_Type = {
 658   PyObject_HEAD_INIT(&PyType_Type)
 659   0,                                      /*size*/
 660   MODULESTR".cvhistogram",                /*name*/
 661   sizeof(cvhistogram_t),                  /*basicsize*/
 662 };
 663
 664 static PyObject *cvhistogram_getbins(cvhistogram_t *cvh)
 665 {
 666   Py_INCREF(cvh->bins);
 667   return cvh->bins;
 668 }
 669
 670 static PyGetSetDef cvhistogram_getseters[] = {
 671   {(char*)"bins", (getter)cvhistogram_getbins, (setter)NULL, (char*)"bins", NULL},
 672   {NULL}  /* Sentinel */
 673 };
 674
 675 static void cvhistogram_specials(void)
 676 {
 677   cvhistogram_Type.tp_dealloc = cvhistogram_dealloc;
 678   cvhistogram_Type.tp_getset = cvhistogram_getseters;
 679 }
 680
 681 /************************************************************************/
 682
 683 /* cvlineiterator */
 684
 685 static PyObject *cvlineiterator_iter(PyObject *o)
 686 {
 687   Py_INCREF(o);
 688   return o;
 689 }
 690
 691 static PyObject *cvlineiterator_next(PyObject *o)
 692 {
 693   cvlineiterator_t *pi = (cvlineiterator_t*)o;
 694
 695   if (pi->count) {
 696       pi->count--;
 697
 698       CvScalar r;
 699       cvRawDataToScalar( (void*)(pi->iter.ptr), pi->type, &r);
 700       PyObject *pr = PyObject_FromCvScalar(r, pi->type);
 701
 702       CV_NEXT_LINE_POINT(pi->iter);
 703
 704       return pr;
 705   } else {
 706     return NULL;
 707   }
 708 }
 709
 710 static PyTypeObject cvlineiterator_Type = {
 711   PyObject_HEAD_INIT(&PyType_Type)
 712   0,                                      /*size*/
 713   MODULESTR".cvlineiterator",             /*name*/
 714   sizeof(cvlineiterator_t),               /*basicsize*/
 715 };
 716
 717 static void cvlineiterator_specials(void)
 718 {
 719   cvlineiterator_Type.tp_iter = cvlineiterator_iter;
 720   cvlineiterator_Type.tp_iternext = cvlineiterator_next;
 721 }
 722
 723 /************************************************************************/
 724
 725 /* memtrack */
 726
 727 static void memtrack_dealloc(PyObject *self)
 728 {
 729   memtrack_t *pi = (memtrack_t*)self;
 730   // printf("===> memtrack_dealloc %p!\n", pi->ptr);
 731   cvFree(&pi->ptr);
 732   PyObject_Del(self);
 733 }
 734
 735 static PyTypeObject memtrack_Type = {
 736   PyObject_HEAD_INIT(&PyType_Type)
 737   0,                                      /*size*/
 738   MODULESTR".memtrack",                          /*name*/
 739   sizeof(memtrack_t),                        /*basicsize*/
 740 };
 741
 742 Py_ssize_t memtrack_getreadbuffer(PyObject *self, Py_ssize_t segment, void **ptrptr)
 743 {
 744   *ptrptr = &((memtrack_t*)self)->ptr;
 745   return ((memtrack_t*)self)->size;
 746 }
 747
 748 Py_ssize_t memtrack_getwritebuffer(PyObject *self, Py_ssize_t segment, void **ptrptr)
 749 {
 750   *ptrptr = ((memtrack_t*)self)->ptr;
 751   return ((memtrack_t*)self)->size;
 752 }
 753
 754 Py_ssize_t memtrack_getsegcount(PyObject *self, Py_ssize_t *lenp)
 755 {
 756   return (Py_ssize_t)1;
 757 }
 758
 759 PyBufferProcs memtrack_as_buffer = {
 760   memtrack_getreadbuffer,
 761   memtrack_getwritebuffer,
 762   memtrack_getsegcount
 763 };
 764
 765 static void memtrack_specials(void)
 766 {
 767   memtrack_Type.tp_dealloc = memtrack_dealloc;
 768   memtrack_Type.tp_as_buffer = &memtrack_as_buffer;
 769 }
 770
 771 /************************************************************************/
 772
 773 /* iplconvkernel */
 774
 775 static void iplconvkernel_dealloc(PyObject *self)
 776 {
 777   iplconvkernel_t *pi = (iplconvkernel_t*)self;
 778   cvReleaseStructuringElement(&(pi->a));
 779   PyObject_Del(self);
 780 }
 781
 782 static PyTypeObject iplconvkernel_Type = {
 783   PyObject_HEAD_INIT(&PyType_Type)
 784   0,                                      /*size*/
 785   MODULESTR".iplconvkernel",                          /*name*/
 786   sizeof(iplconvkernel_t),                        /*basicsize*/
 787 };
 788
 789 static void iplconvkernel_specials(void)
 790 {
 791   iplconvkernel_Type.tp_dealloc = iplconvkernel_dealloc;
 792 }
 793
 794 /************************************************************************/
 795
 796 /* cvcapture */
 797
 798 static void cvcapture_dealloc(PyObject *self)
 799 {
 800   cvcapture_t *pi = (cvcapture_t*)self;
 801   cvReleaseCapture(&(pi->a));
 802   PyObject_Del(self);
 803 }
 804
 805 static PyTypeObject cvcapture_Type = {
 806   PyObject_HEAD_INIT(&PyType_Type)
 807   0,                                      /*size*/
 808   MODULESTR".cvcapture",                  /*name*/
 809   sizeof(cvcapture_t),                    /*basicsize*/
 810 };
 811
 812 static void cvcapture_specials(void)
 813 {
 814   cvcapture_Type.tp_dealloc = cvcapture_dealloc;
 815 }
 816
 817 /************************************************************************/
 818
 819 /* cvvideowriter */
 820
 821 static void cvvideowriter_dealloc(PyObject *self)
 822 {
 823   cvvideowriter_t *pi = (cvvideowriter_t*)self;
 824   cvReleaseVideoWriter(&(pi->a));
 825   PyObject_Del(self);
 826 }
 827
 828 static PyTypeObject cvvideowriter_Type = {
 829   PyObject_HEAD_INIT(&PyType_Type)
 830   0,                                      /*size*/
 831   MODULESTR".cvvideowriter",              /*name*/
 832   sizeof(cvvideowriter_t),                /*basicsize*/
 833 };
 834
 835 static void cvvideowriter_specials(void)
 836 {
 837   cvvideowriter_Type.tp_dealloc = cvvideowriter_dealloc;
 838 }
 839
 840
 841
 842 /************************************************************************/
 843
 844 /* cvmoments */
 845
 846 static PyTypeObject cvmoments_Type = {
 847   PyObject_HEAD_INIT(&PyType_Type)
 848   0,                                      /*size*/
 849   MODULESTR".cvmoments",                          /*name*/
 850   sizeof(cvmoments_t),                        /*basicsize*/
 851 };
 852
 853 static void cvmoments_specials(void)
 854 {
 855 }
 856
 857 /************************************************************************/
 858
 859 /* cvmemstorage */
 860
 861 static void cvmemstorage_dealloc(PyObject *self)
 862 {
 863   cvmemstorage_t *ps = (cvmemstorage_t*)self;
 864   cvReleaseMemStorage(&(ps->a));
 865   PyObject_Del(self);
 866 }
 867
 868 static PyTypeObject cvmemstorage_Type = {
 869   PyObject_HEAD_INIT(&PyType_Type)
 870   0,                                      /*size*/
 871   MODULESTR".cvmemstorage",               /*name*/
 872   sizeof(cvmemstorage_t),                 /*basicsize*/
 873 };
 874
 875 static void cvmemstorage_specials(void)
 876 {
 877   cvmemstorage_Type.tp_dealloc = cvmemstorage_dealloc;
 878 }
 879
 880 /************************************************************************/
 881
 882 /* cvfont */
 883
 884 static PyTypeObject cvfont_Type = {
 885   PyObject_HEAD_INIT(&PyType_Type)
 886   0,                                      /*size*/
 887   MODULESTR".cvfont",                     /*name*/
 888   sizeof(cvfont_t),                       /*basicsize*/
 889 };
 890
 891 static void cvfont_specials(void) { }
 892
 893 /************************************************************************/
 894
 895 /* cvpositobject */
 896
 897 static void cvpositobject_dealloc(PyObject *self)
 898 {
 899   cvpositobject_t *pi = (cvpositobject_t*)self;
 900   cvReleasePOSITObject(&(pi->a));
 901   PyObject_Del(self);
 902 }
 903
 904 static PyTypeObject cvpositobject_Type = {
 905   PyObject_HEAD_INIT(&PyType_Type)
 906   0,                                      /*size*/
 907   MODULESTR".cvpositobject",                     /*name*/
 908   sizeof(cvpositobject_t),                       /*basicsize*/
 909 };
 910
 911 static void cvpositobject_specials(void)
 912 {
 913   cvpositobject_Type.tp_dealloc = cvpositobject_dealloc;
 914 }
 915
 916 /************************************************************************/
 917
 918 /* cvrng */
 919
 920 static PyTypeObject cvrng_Type = {
 921   PyObject_HEAD_INIT(&PyType_Type)
 922   0,                                      /*size*/
 923   MODULESTR".cvrng",                     /*name*/
 924   sizeof(cvrng_t),                       /*basicsize*/
 925 };
 926
 927 static void cvrng_specials(void)
 928 {
 929 }
 930
 931 /************************************************************************/
 932
 933 /* cvhaarclassifiercascade */
 934
 935 static PyTypeObject cvhaarclassifiercascade_Type = {
 936   PyObject_HEAD_INIT(&PyType_Type)
 937   0,                                      /*size*/
 938   MODULESTR".cvhaarclassifiercascade",                     /*name*/
 939   sizeof(cvhaarclassifiercascade_t),                       /*basicsize*/
 940 };
 941
 942 static void cvhaarclassifiercascade_specials(void) { }
 943
 944 /************************************************************************/
 945
 946 /* cvcontourtree */
 947
 948 static PyTypeObject cvcontourtree_Type = {
 949   PyObject_HEAD_INIT(&PyType_Type)
 950   0,                                      /*size*/
 951   MODULESTR".cvcontourtree",                     /*name*/
 952   sizeof(cvcontourtree_t),                       /*basicsize*/
 953 };
 954
 955 static void cvcontourtree_specials(void) { }
 956
 957
 958 /************************************************************************/
 959
 960 /* cvsubdiv2dedge */
 961
 962 static PyTypeObject cvsubdiv2dedge_Type = {
 963   PyObject_HEAD_INIT(&PyType_Type)
 964   0,                                      /*size*/
 965   MODULESTR".cvsubdiv2dedge",                     /*name*/
 966   sizeof(cvsubdiv2dedge_t),                       /*basicsize*/
 967 };
 968
 969 static int cvsubdiv2dedge_compare(PyObject *o1, PyObject *o2)
 970 {
 971   cvsubdiv2dedge_t *e1 = (cvsubdiv2dedge_t*)o1;
 972   cvsubdiv2dedge_t *e2 = (cvsubdiv2dedge_t*)o2;
 973   if (e1->a < e2->a)
 974     return -1;
 975   else if (e1->a > e2->a)
 976     return 1;
 977   else
 978     return 0;
 979 }
 980
 981 static PyObject *cvquadedge_repr(PyObject *self)
 982 {
 983   CvSubdiv2DEdge m = ((cvsubdiv2dedge_t*)self)->a;
 984   char str[1000];
 985   sprintf(str, "<cvsubdiv2dedge(");
 986   char *d = str + strlen(str);
 987   sprintf(d, "%zx.%d", m & ~3, (int)(m & 3));
 988   d += strlen(d);
 989   sprintf(d, ")>");
 990   return PyString_FromString(str);
 991 }
 992
 993 static void cvsubdiv2dedge_specials(void) {
 994   cvsubdiv2dedge_Type.tp_compare = cvsubdiv2dedge_compare;
 995   cvsubdiv2dedge_Type.tp_repr = cvquadedge_repr;
 996 }
 997
 998 /************************************************************************/
 999
1000 /* cvseq */
1001
1002 static void cvseq_dealloc(PyObject *self)
1003 {
1004   cvseq_t *ps = (cvseq_t*)self;
1005   Py_DECREF(ps->container);
1006   PyObject_Del(self);
1007 }
1008
1009 static PyObject *cvseq_h_next(PyObject *self, PyObject *args);
1010 static PyObject *cvseq_h_prev(PyObject *self, PyObject *args);
1011 static PyObject *cvseq_v_next(PyObject *self, PyObject *args);
1012 static PyObject *cvseq_v_prev(PyObject *self, PyObject *args);
1013
1014 static struct PyMethodDef cvseq_methods[] =
1015 {
1016   {"h_next", cvseq_h_next, METH_VARARGS},
1017   {"h_prev", cvseq_h_prev, METH_VARARGS},
1018   {"v_next", cvseq_v_next, METH_VARARGS},
1019   {"v_prev", cvseq_v_prev, METH_VARARGS},
1020   {NULL,          NULL}
1021 };
1022
1023 static Py_ssize_t cvseq_seq_length(PyObject *o)
1024 {
1025   cvseq_t *ps = (cvseq_t*)o;
1026   return (Py_ssize_t)(ps->a->total);
1027 }
1028
1029 static PyObject* cvseq_seq_getitem(PyObject *o, Py_ssize_t i)
1030 {
1031   cvseq_t *ps = (cvseq_t*)o;
1032   CvPoint *pt;
1033   struct pointpair{
1034     CvPoint a, b;
1035   } *pp;
1036   CvPoint2D32f *pt2;
1037   CvPoint3D32f *pt3;
1038
1039   if (i < (Py_ssize_t)(ps->a->total)) {
1040     switch (CV_SEQ_ELTYPE(ps->a)) {
1041
1042     case CV_SEQ_ELTYPE_POINT:
1043       pt = CV_GET_SEQ_ELEM(CvPoint, ps->a, i);
1044       return Py_BuildValue("ii", pt->x, pt->y);
1045
1046     case CV_SEQ_ELTYPE_GENERIC:
1047       switch (ps->a->elem_size) {
1048       case sizeof(CvQuadEdge2D):
1049         {
1050           cvsubdiv2dedge_t *r = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
1051           r->a = (CvSubdiv2DEdge)CV_GET_SEQ_ELEM(CvQuadEdge2D, ps->a, i);
1052           r->container = ps->container;
1053           Py_INCREF(r->container);
1054           return (PyObject*)r;
1055         }
1056       default:
1057         printf("seq elem size is %d\n", ps->a->elem_size);
1058         printf("KIND %d\n", CV_SEQ_KIND(ps->a));
1059         assert(0);
1060       }
1061       return PyInt_FromLong(*CV_GET_SEQ_ELEM(unsigned char, ps->a, i));
1062
1063     case CV_SEQ_ELTYPE_PTR:
1064     case CV_SEQ_ELTYPE_INDEX:
1065       return PyInt_FromLong(*CV_GET_SEQ_ELEM(int, ps->a, i));
1066
1067     case CV_32SC4:
1068       pp = CV_GET_SEQ_ELEM(pointpair, ps->a, i);
1069       return Py_BuildValue("(ii),(ii)", pp->a.x, pp->a.y, pp->b.x, pp->b.y);
1070
1071     case CV_32FC2:
1072       pt2 = CV_GET_SEQ_ELEM(CvPoint2D32f, ps->a, i);
1073       return Py_BuildValue("ff", pt2->x, pt2->y);
1074
1075     case CV_SEQ_ELTYPE_POINT3D:
1076       pt3 = CV_GET_SEQ_ELEM(CvPoint3D32f, ps->a, i);
1077       return Py_BuildValue("fff", pt3->x, pt3->y, pt3->z);
1078
1079     default:
1080       printf("Unknown element type %08x\n", CV_SEQ_ELTYPE(ps->a));
1081       assert(0);
1082       return NULL;
1083     }
1084   } else
1085     return NULL;
1086 }
1087
1088 static PyObject* cvseq_map_getitem(PyObject *o, PyObject *item)
1089 {
1090   cvseq_t *ps = (cvseq_t*)o;
1091   if (PyInt_Check(item)) {
1092     long i = PyInt_AS_LONG(item);
1093     if (i < 0)
1094       i += ps->a->total;
1095     return cvseq_seq_getitem(o, i);
1096   } else if (PySlice_Check(item)) {
1097     Py_ssize_t start, stop, step, slicelength, cur, i;
1098     PyObject* result;
1099
1100     if (PySlice_GetIndicesEx((PySliceObject*)item, ps->a->total,
1101          &start, &stop, &step, &slicelength) < 0) {
1102       return NULL;
1103     }
1104
1105     if (slicelength <= 0) {
1106       return PyList_New(0);
1107     } else {
1108       result = PyList_New(slicelength);
1109       if (!result) return NULL;
1110
1111       for (cur = start, i = 0; i < slicelength;
1112            cur += step, i++) {
1113         PyList_SET_ITEM(result, i, cvseq_seq_getitem(o, cur));
1114       }
1115
1116       return result;
1117     }
1118   } else {
1119     PyErr_SetString(PyExc_TypeError, "CvSeq indices must be integers");
1120     return NULL;
1121   }
1122 }
1123
1124 static
1125 PySequenceMethods cvseq_sequence = {
1126   cvseq_seq_length,
1127   NULL,
1128   NULL,
1129   cvseq_seq_getitem
1130 };
1131
1132 static PyMappingMethods cvseq_mapping = {
1133   cvseq_seq_length,
1134   cvseq_map_getitem,
1135   NULL,
1136 };
1137
1138 static PyTypeObject cvseq_Type = {
1139   PyObject_HEAD_INIT(&PyType_Type)
1140   0,                                      /*size*/
1141   MODULESTR".cvseq",                          /*name*/
1142   sizeof(cvseq_t),                        /*basicsize*/
1143 };
1144
1145 static void cvseq_specials(void)
1146 {
1147   cvseq_Type.tp_dealloc = cvseq_dealloc;
1148   cvseq_Type.tp_as_sequence = &cvseq_sequence;
1149   cvseq_Type.tp_as_mapping = &cvseq_mapping;
1150   cvseq_Type.tp_methods = cvseq_methods;
1151 }
1152
1153 #define MK_ACCESSOR(FIELD) \
1154 static PyObject *cvseq_##FIELD(PyObject *self, PyObject *args) \
1155 { \
1156   cvseq_t *ps = (cvseq_t*)self; \
1157   CvSeq *s = ps->a; \
1158   if (s->FIELD == NULL) { \
1159     Py_RETURN_NONE; \
1160   } else { \
1161     cvseq_t *r = PyObject_NEW(cvseq_t, &cvseq_Type); \
1162     r->a = s->FIELD; \
1163     r->container = ps->container; \
1164     Py_INCREF(r->container); \
1165     return (PyObject*)r; \
1166   } \
1167 }
1168
1169 MK_ACCESSOR(h_next)
1170 MK_ACCESSOR(h_prev)
1171 MK_ACCESSOR(v_next)
1172 MK_ACCESSOR(v_prev)
1173 #undef MK_ACCESSOR
1174
1175 /************************************************************************/
1176
1177 /* cvset */
1178
1179 static void cvset_dealloc(PyObject *self)
1180 {
1181   cvset_t *ps = (cvset_t*)self;
1182   Py_DECREF(ps->container);
1183   PyObject_Del(self);
1184 }
1185
1186 static PyTypeObject cvset_Type = {
1187   PyObject_HEAD_INIT(&PyType_Type)
1188   0,                                      /*size*/
1189   MODULESTR".cvset",                          /*name*/
1190   sizeof(cvset_t),                        /*basicsize*/
1191 };
1192
1193 static PyObject *cvset_iter(PyObject *o)
1194 {
1195   Py_INCREF(o);
1196   cvset_t *ps = (cvset_t*)o;
1197   ps->i = 0;
1198   return o;
1199 }
1200
1201 static PyObject *cvset_next(PyObject *o)
1202 {
1203   cvset_t *ps = (cvset_t*)o;
1204
1205   while (ps->i < ps->a->total) {
1206     CvSetElem *e = cvGetSetElem(ps->a, ps->i);
1207     int prev_i = ps->i++;
1208     if (e != NULL) {
1209       return cvseq_seq_getitem(o, prev_i);
1210     }
1211   }
1212   return NULL;
1213 }
1214
1215 static void cvset_specials(void)
1216 {
1217   cvset_Type.tp_dealloc = cvset_dealloc;
1218   cvset_Type.tp_iter = cvset_iter;
1219   cvset_Type.tp_iternext = cvset_next;
1220 }
1221
1222 /************************************************************************/
1223
1224 /* cvsubdiv2d */
1225
1226 static PyTypeObject cvsubdiv2d_Type = {
1227   PyObject_HEAD_INIT(&PyType_Type)
1228   0,                                      /*size*/
1229   MODULESTR".cvsubdiv2d",                     /*name*/
1230   sizeof(cvsubdiv2d_t),                       /*basicsize*/
1231 };
1232
1233 static PyObject *cvsubdiv2d_getattro(PyObject *o, PyObject *name)
1234 {
1235   cvsubdiv2d_t *p = (cvsubdiv2d_t*)o;
1236   if (strcmp(PyString_AsString(name), "edges") == 0) {
1237     cvset_t *r = PyObject_NEW(cvset_t, &cvset_Type);
1238     r->a = p->a->edges;
1239     r->container = p->container;
1240     Py_INCREF(r->container);
1241     return (PyObject*)r;
1242   } else {
1243     PyErr_SetString(PyExc_TypeError, "cvsubdiv2d has no such attribute");
1244     return NULL;
1245   }
1246 }
1247
1248 static void cvsubdiv2d_specials(void)
1249 {
1250   cvsubdiv2d_Type.tp_getattro = cvsubdiv2d_getattro;
1251 }
1252
1253 /************************************************************************/
1254
1255 /* cvsubdiv2dpoint */
1256
1257 static PyTypeObject cvsubdiv2dpoint_Type = {
1258   PyObject_HEAD_INIT(&PyType_Type)
1259   0,                                      /*size*/
1260   MODULESTR".cvsubdiv2dpoint",                     /*name*/
1261   sizeof(cvsubdiv2dpoint_t),                       /*basicsize*/
1262 };
1263
1264 static PyObject *cvsubdiv2dpoint_getattro(PyObject *o, PyObject *name)
1265 {
1266   cvsubdiv2dpoint_t *p = (cvsubdiv2dpoint_t*)o;
1267   if (strcmp(PyString_AsString(name), "first") == 0) {
1268     cvsubdiv2dedge_t *r = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
1269     r->a = p->a->first;
1270     r->container = p->container;
1271     Py_INCREF(r->container);
1272     return (PyObject*)r;
1273   } else if (strcmp(PyString_AsString(name), "pt") == 0) {
1274     return Py_BuildValue("(ff)", p->a->pt.x, p->a->pt.y);
1275   } else {
1276     PyErr_SetString(PyExc_TypeError, "cvsubdiv2dpoint has no such attribute");
1277     return NULL;
1278   }
1279 }
1280
1281 static void cvsubdiv2dpoint_specials(void)
1282 {
1283   cvsubdiv2dpoint_Type.tp_getattro = cvsubdiv2dpoint_getattro;
1284 }
1285
1286 /************************************************************************/
1287 /* convert_to_X: used after PyArg_ParseTuple in the generated code  */
1288
1289 static int convert_to_char(PyObject *o, char *dst, const char *name = "no_name")
1290 {
1291   if (PyString_Check(o) && PyString_Size(o) == 1) {
1292     *dst = PyString_AsString(o)[0];
1293     return 1;
1294   } else {
1295     (*dst) = 0;
1296     return failmsg("Expected single character string for argument '%s'", name);
1297   }
1298 }
1299
1300 static int convert_to_CvMemStorage(PyObject *o, CvMemStorage **dst, const char *name = "no_name")
1301 {
1302   if (PyType_IsSubtype(o->ob_type, &cvmemstorage_Type)) {
1303     (*dst) = (((cvmemstorage_t*)o)->a);
1304     return 1;
1305   } else {
1306     (*dst) = (CvMemStorage*)NULL;
1307     return failmsg("Expected CvMemStorage for argument '%s'", name);
1308   }
1309 }
1310
1311 static int convert_to_CvSeq(PyObject *o, CvSeq **dst, const char *name = "no_name")
1312 {
1313   if (PyType_IsSubtype(o->ob_type, &cvseq_Type)) {
1314     (*dst) = (((cvseq_t*)o)->a);
1315     return 1;
1316   } else {
1317     (*dst) = (CvSeq*)NULL;
1318     return failmsg("Expected CvSeq for argument '%s'", name);
1319   }
1320 }
1321
1322 static int convert_to_CvSize(PyObject *o, CvSize *dst, const char *name = "no_name")
1323 {
1324   if (!PyArg_ParseTuple(o, "ii", &dst->width, &dst->height))
1325     return failmsg("CvSize argument '%s' expects two integers", name);
1326   else
1327     return 1;
1328 }
1329
1330 static int convert_to_CvScalar(PyObject *o, CvScalar *s, const char *name = "no_name")
1331 {
1332   if (PySequence_Check(o)) {
1333     PyObject *fi = PySequence_Fast(o, name);
1334     if (fi == NULL)
1335       return 0;
1336     for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1337       PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1338       if (PyNumber_Check(item))
1339         s->val[i] = PyFloat_AsDouble(item);
1340       else
1341         return failmsg("CvScalar value for argument '%s' is not numeric", name);
1342     }
1343     Py_DECREF(fi);
1344   } else {
1345     if (PyNumber_Check(o)) {
1346       s->val[0] = PyFloat_AsDouble(o);
1347     } else {
1348       return failmsg("CvScalar value for argument '%s' is not numeric", name);
1349     }
1350   }
1351   return 1;
1352 }
1353
1354 static int convert_to_CvPointPTR(PyObject *o, CvPoint **p, const char *name = "no_name")
1355 {
1356   if (!PySequence_Check(o))
1357     return failmsg("Expected sequence for point list argument '%s'", name);
1358   PyObject *fi = PySequence_Fast(o, name);
1359   if (fi == NULL)
1360     return 0;
1361   *p = new CvPoint[PySequence_Fast_GET_SIZE(fi)];
1362   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1363     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1364     if (!PyTuple_Check(item))
1365       return failmsg("Expected tuple for element in point list argument '%s'", name);
1366     if (!PyArg_ParseTuple(item, "ii", &((*p)[i].x), &((*p)[i].y))) {
1367       return 0;
1368     }
1369   }
1370   Py_DECREF(fi);
1371   return 1;
1372 }
1373
1374 static int convert_to_CvPoint2D32fPTR(PyObject *o, CvPoint2D32f **p, const char *name = "no_name")
1375 {
1376   PyObject *fi = PySequence_Fast(o, name);
1377   if (fi == NULL)
1378     return 0;
1379   *p = new CvPoint2D32f[PySequence_Fast_GET_SIZE(fi)];
1380   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1381     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1382     if (!PyTuple_Check(item))
1383       return failmsg("Expected tuple for CvPoint2D32f argument '%s'", name);
1384     if (!PyArg_ParseTuple(item, "ff", &((*p)[i].x), &((*p)[i].y))) {
1385       return 0;
1386     }
1387   }
1388   Py_DECREF(fi);
1389   return 1;
1390 }
1391
1392 static int convert_to_CvPoint3D32fPTR(PyObject *o, CvPoint3D32f **p, const char *name = "no_name")
1393 {
1394   PyObject *fi = PySequence_Fast(o, name);
1395   if (fi == NULL)
1396     return 0;
1397   *p = new CvPoint3D32f[PySequence_Fast_GET_SIZE(fi)];
1398   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1399     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1400     if (!PyTuple_Check(item))
1401       return failmsg("Expected tuple for CvPoint3D32f argument '%s'", name);
1402     if (!PyArg_ParseTuple(item, "fff", &((*p)[i].x), &((*p)[i].y), &((*p)[i].z))) {
1403       return 0;
1404     }
1405   }
1406   Py_DECREF(fi);
1407   return 1;
1408 }
1409
1410 static int convert_to_CvStarDetectorParams(PyObject *o, CvStarDetectorParams *dst, const char *name = "no_name")
1411 {
1412   if (!PyArg_ParseTuple(o,
1413                         "iiiii",
1414                         &dst->maxSize,
1415                         &dst->responseThreshold,
1416                         &dst->lineThresholdProjected,
1417                         &dst->lineThresholdBinarized,
1418                         &dst->suppressNonmaxSize))
1419     return failmsg("CvRect argument '%s' expects four integers", name);
1420   else
1421     return 1;
1422 }
1423
1424 static int convert_to_CvRect(PyObject *o, CvRect *dst, const char *name = "no_name")
1425 {
1426   if (!PyArg_ParseTuple(o, "iiii", &dst->x, &dst->y, &dst->width, &dst->height))
1427     return failmsg("CvRect argument '%s' expects four integers", name);
1428   else
1429     return 1;
1430 }
1431
1432 static int convert_to_CvRectPTR(PyObject *o, CvRect **dst, const char *name = "no_name")
1433 {
1434   *dst = new CvRect;
1435   if (!PyArg_ParseTuple(o, "iiii", &(*dst)->x, &(*dst)->y, &(*dst)->width, &(*dst)->height))
1436     return failmsg("CvRect argument '%s' expects four integers", name);
1437   else
1438     return 1;
1439 }
1440
1441 static int convert_to_CvSlice(PyObject *o, CvSlice *dst, const char *name = "no_name")
1442 {
1443   if (!PyArg_ParseTuple(o, "ii", &dst->start_index, &dst->end_index))
1444     return failmsg("CvSlice argument '%s' expects two integers", name);
1445   else
1446     return 1;
1447 }
1448
1449 static int convert_to_CvPoint(PyObject *o, CvPoint *dst, const char *name = "no_name")
1450 {
1451   if (!PyArg_ParseTuple(o, "ii", &dst->x, &dst->y))
1452     return failmsg("CvPoint argument '%s' expects two integers", name);
1453   else
1454     return 1;
1455 }
1456
1457 static int convert_to_CvPoint2D32f(PyObject *o, CvPoint2D32f *dst, const char *name = "no_name")
1458 {
1459   if (!PyArg_ParseTuple(o, "ff", &dst->x, &dst->y))
1460     return failmsg("CvPoint2D32f argument '%s' expects two floats", name);
1461   else
1462     return 1;
1463 }
1464
1465 static int convert_to_CvPoint3D32f(PyObject *o, CvPoint3D32f *dst, const char *name = "no_name")
1466 {
1467   if (!PyArg_ParseTuple(o, "fff", &dst->x, &dst->y, &dst->z))
1468     return failmsg("CvPoint3D32f argument '%s' expects three floats", name);
1469   else
1470     return 1;
1471 }
1472
1473 static int convert_to_IplImage(PyObject *o, IplImage **dst, const char *name)
1474 {
1475   iplimage_t *ipl = (iplimage_t*)o;
1476   void *buffer;
1477   Py_ssize_t buffer_len;
1478
1479   if (!is_iplimage(o)) {
1480     return failmsg("Argument '%s' must be IplImage", name);
1481   } else if (PyString_Check(ipl->data)) {
1482     cvSetData(ipl->a, PyString_AsString(ipl->data) + ipl->offset, ipl->a->widthStep);
1483     assert(cvGetErrStatus() == 0);
1484     *dst = ipl->a;
1485     return 1;
1486   } else if (ipl->data && PyObject_AsWriteBuffer(ipl->data, &buffer, &buffer_len) == 0) {
1487     cvSetData(ipl->a, (void*)((char*)buffer + ipl->offset), ipl->a->widthStep);
1488     assert(cvGetErrStatus() == 0);
1489     *dst = ipl->a;
1490     return 1;
1491   } else {
1492     return failmsg("IplImage argument '%s' has no data", name);
1493   }
1494 }
1495
1496 static int convert_to_CvMat(PyObject *o, CvMat **dst, const char *name)
1497 {
1498   cvmat_t *m = (cvmat_t*)o;
1499   void *buffer;
1500   Py_ssize_t buffer_len;
1501
1502   if (!is_cvmat(o)) {
1503     return failmsg("Argument '%s' must be CvMat", name);
1504   } else {
1505     m->a->refcount = NULL;
1506     if (m->data && PyString_Check(m->data)) {
1507       assert(cvGetErrStatus() == 0);
1508       char *ptr = PyString_AsString(m->data) + m->offset;
1509       cvSetData(m->a, ptr, m->a->step);
1510       assert(cvGetErrStatus() == 0);
1511       *dst = m->a;
1512       return 1;
1513     } else if (m->data && PyObject_AsWriteBuffer(m->data, &buffer, &buffer_len) == 0) {
1514       cvSetData(m->a, (void*)((char*)buffer + m->offset), m->a->step);
1515       assert(cvGetErrStatus() == 0);
1516       *dst = m->a;
1517       return 1;
1518     } else {
1519       return failmsg("CvMat argument '%s' has no data", name);
1520     }
1521   }
1522 }
1523
1524 static int convert_to_CvMatND(PyObject *o, CvMatND **dst, const char *name)
1525 {
1526   cvmatnd_t *m = (cvmatnd_t*)o;
1527   void *buffer;
1528   Py_ssize_t buffer_len;
1529
1530   if (!is_cvmatnd(o)) {
1531     return failmsg("Argument '%s' must be CvMatND", name);
1532   } else if (m->data && PyString_Check(m->data)) {
1533     m->a->data.ptr = ((uchar*)PyString_AsString(m->data)) + m->offset;
1534     *dst = m->a;
1535     return 1;
1536   } else if (m->data && PyObject_AsWriteBuffer(m->data, &buffer, &buffer_len) == 0) {
1537     m->a->data.ptr = ((uchar*)buffer + m->offset);
1538     *dst = m->a;
1539     return 1;
1540   } else {
1541     return failmsg("CvMatND argument '%s' has no data", name);
1542   }
1543 }
1544
1545 static int convert_to_CvArr(PyObject *o, CvArr **dst, const char *name)
1546 {
1547   if (o == Py_None) {
1548     *dst = (void*)NULL;
1549     return 1;
1550   } else if (is_iplimage(o)) {
1551     return convert_to_IplImage(o, (IplImage**)dst, name);
1552   } else if (is_cvmat(o)) {
1553     return convert_to_CvMat(o, (CvMat**)dst, name);
1554   } else if (is_cvmatnd(o)) {
1555     return convert_to_CvMatND(o, (CvMatND**)dst, name);
1556   } else {
1557     return failmsg("CvArr argument '%s' must be IplImage, CvMat or CvMatND", name);
1558   }
1559 }
1560
1561 static int convert_to_CvHistogram(PyObject *o, CvHistogram **dst, const char *name = "no_name")
1562 {
1563   if (PyType_IsSubtype(o->ob_type, &cvhistogram_Type)) {
1564     cvhistogram_t *ht = (cvhistogram_t*)o;
1565     *dst = &ht->h;
1566     return convert_to_CvArr(ht->bins, &(ht->h.bins), "bins");
1567   } else {
1568     *dst = (CvHistogram *)NULL;
1569     return failmsg("Expected CvHistogram for argument '%s'", name);
1570   }
1571 }
1572
1573 // Used by FillPoly, FillConvexPoly, PolyLine
1574 struct pts_npts_contours {
1575   CvPoint** pts;
1576   int* npts;
1577   int contours;
1578 };
1579
1580 static int convert_to_pts_npts_contours(PyObject *o, pts_npts_contours *dst, const char *name = "no_name")
1581 {
1582   PyObject *fi = PySequence_Fast(o, name);
1583   if (fi == NULL)
1584     return 0;
1585   dst->contours = PySequence_Fast_GET_SIZE(fi);
1586   dst->pts = new CvPoint*[dst->contours];
1587   dst->npts = new int[dst->contours];
1588   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1589     if (!convert_to_CvPointPTR(PySequence_Fast_GET_ITEM(fi, i), &dst->pts[i], name))
1590       return 0;
1591     dst->npts[i] = PySequence_Size(PySequence_Fast_GET_ITEM(fi, i)); // safe because convert_ just succeeded
1592   }
1593   Py_DECREF(fi);
1594   return 1;
1595 }
1596
1597 struct cvarrseq {
1598   void *v;
1599 };
1600
1601 static int convert_to_cvarrseq(PyObject *o, cvarrseq *dst, const char *name = "no_name")
1602 {
1603   if (PyType_IsSubtype(o->ob_type, &cvseq_Type)) {
1604     return convert_to_CvSeq(o, (CvSeq**)&(dst->v), name);
1605   } else if (PySequence_Check(o)) {
1606     PyObject *fi = PySequence_Fast(o, name);
1607     if (fi == NULL)
1608       return 0;
1609     Py_ssize_t size = -1;
1610     // Make a pass through the sequence, checking that each element is
1611     // a sequence and that they are all the same size
1612     for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1613       PyObject *e = PySequence_Fast_GET_ITEM(fi, i);
1614
1615       if (!PySequence_Check(e))
1616         return failmsg("Sequence '%s' must contain sequences", name);
1617       if (i == 0)
1618         size = (int)PySequence_Size(e);
1619       else if (size != PySequence_Size(e))
1620         return failmsg("All elements of sequence '%s' must be same size", name);
1621     }
1622     assert(size != -1);
1623     CvMat *mt = cvCreateMat((int)PySequence_Fast_GET_SIZE(fi), 1, CV_32SC(size));
1624     for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1625       PyObject *e = PySequence_Fast_GET_ITEM(fi, i);
1626       PyObject *fe = PySequence_Fast(e, name);
1627       assert(fe != NULL);
1628       int *pdst = (int*)cvPtr2D(mt, i, 0);
1629       for (Py_ssize_t j = 0; j < size; j++) {
1630         *pdst++ = PyInt_AsLong(PySequence_Fast_GET_ITEM(fe, j));
1631       }
1632       Py_DECREF(fe);
1633     }
1634     Py_DECREF(fi);
1635     dst->v = mt;
1636     return 1;
1637   } else {
1638     return convert_to_CvArr(o, (CvArr**)&(dst->v), name);
1639   }
1640 }
1641
1642 struct cvarr_count {
1643   CvArr **cvarr;
1644   int count;
1645 };
1646
1647 static int convert_to_cvarr_count(PyObject *o, cvarr_count *dst, const char *name = "no_name")
1648 {
1649   PyObject *fi = PySequence_Fast(o, name);
1650   if (fi == NULL)
1651     return 0;
1652   dst->count = PySequence_Fast_GET_SIZE(fi);
1653   dst->cvarr = new CvArr*[dst->count];
1654   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1655     if (!convert_to_CvArr(PySequence_Fast_GET_ITEM(fi, i), &dst->cvarr[i], name))
1656       return 0;
1657   }
1658   Py_DECREF(fi);
1659   return 1;
1660 }
1661
1662 struct intpair
1663 {
1664   int *pairs;
1665   int count;
1666 };
1667
1668 static int convert_to_intpair(PyObject *o, intpair *dst, const char *name = "no_name")
1669 {
1670   PyObject *fi = PySequence_Fast(o, name);
1671   if (fi == NULL)
1672     return 0;
1673   dst->count = PySequence_Fast_GET_SIZE(fi);
1674   dst->pairs = new int[2 * dst->count];
1675   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1676     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1677     if (!PyArg_ParseTuple(item, "ii", &dst->pairs[2 * i], &dst->pairs[2 * i + 1])) {
1678       return 0;
1679     }
1680   }
1681   Py_DECREF(fi);
1682   return 1;
1683 }
1684
1685 struct cvpoint2d32f_count {
1686   CvPoint2D32f* points;
1687   int count;
1688 };
1689
1690 static int convert_to_cvpoint2d32f_count(PyObject *o, cvpoint2d32f_count *dst, const char *name = "no_name")
1691 {
1692   if (PyInt_Check(o)) {
1693     dst->count = PyInt_AsLong(o);
1694     dst->points = new CvPoint2D32f[dst->count];
1695     return 1;
1696   } else {
1697     return failmsg("Expected integer for CvPoint2D32f count");
1698   }
1699 }
1700
1701 struct floats {
1702   float *f;
1703   int count;
1704 };
1705 static int convert_to_floats(PyObject *o, floats *dst, const char *name = "no_name")
1706 {
1707   PyObject *fi = PySequence_Fast(o, name);
1708   if (fi == NULL)
1709     return 0;
1710   dst->count = PySequence_Fast_GET_SIZE(fi);
1711   dst->f = new float[dst->count];
1712   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1713     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1714     dst->f[i] = (float)PyFloat_AsDouble(item);
1715   }
1716   Py_DECREF(fi);
1717   return 1;
1718 }
1719
1720 struct chars {
1721   char *f;
1722   int count;
1723 };
1724 /// convert_to_chars not used
1725
1726 struct CvPoints {
1727   CvPoint *p;
1728   int count;
1729 };
1730 static int convert_to_CvPoints(PyObject *o, CvPoints *dst, const char *name = "no_name")
1731 {
1732   PyObject *fi = PySequence_Fast(o, name);
1733   if (fi == NULL)
1734     return 0;
1735   dst->count = PySequence_Fast_GET_SIZE(fi);
1736   dst->p = new CvPoint[dst->count];
1737   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1738     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1739     convert_to_CvPoint(item, &dst->p[i], name);
1740   }
1741   Py_DECREF(fi);
1742   return 1;
1743 }
1744
1745 struct CvPoint3D32fs {
1746   CvPoint3D32f *p;
1747   int count;
1748 };
1749 static int convert_to_CvPoint3D32fs(PyObject *o, CvPoint3D32fs *dst, const char *name = "no_name")
1750 {
1751   PyObject *fi = PySequence_Fast(o, name);
1752   if (fi == NULL)
1753     return 0;
1754   dst->count = PySequence_Fast_GET_SIZE(fi);
1755   dst->p = new CvPoint3D32f[dst->count];
1756   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1757     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1758     convert_to_CvPoint3D32f(item, &dst->p[i], name);
1759   }
1760   Py_DECREF(fi);
1761   return 1;
1762 }
1763
1764 struct CvPoint2D32fs {
1765   CvPoint2D32f *p;
1766   int count;
1767 };
1768 static int convert_to_CvPoint2D32fs(PyObject *o, CvPoint2D32fs *dst, const char *name = "no_name")
1769 {
1770   PyObject *fi = PySequence_Fast(o, name);
1771   if (fi == NULL)
1772     return 0;
1773   dst->count = PySequence_Fast_GET_SIZE(fi);
1774   dst->p = new CvPoint2D32f[dst->count];
1775   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1776     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1777     convert_to_CvPoint2D32f(item, &dst->p[i], name);
1778   }
1779   Py_DECREF(fi);
1780   return 1;
1781 }
1782
1783 struct ints {
1784   int *i;
1785   int count;
1786 };
1787 static int convert_to_ints(PyObject *o, ints *dst, const char *name = "no_name")
1788 {
1789   PyObject *fi = PySequence_Fast(o, name);
1790   if (fi == NULL)
1791     return 0;
1792   dst->count = PySequence_Fast_GET_SIZE(fi);
1793   dst->i = new int[dst->count];
1794   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1795     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1796     dst->i[i] = PyInt_AsLong(item);
1797   }
1798   Py_DECREF(fi);
1799   return 1;
1800 }
1801
1802 struct ints0 {
1803   int *i;
1804   int count;
1805 };
1806 static int convert_to_ints0(PyObject *o, ints0 *dst, const char *name = "no_name")
1807 {
1808   PyObject *fi = PySequence_Fast(o, name);
1809   if (fi == NULL)
1810     return 0;
1811   dst->count = PySequence_Fast_GET_SIZE(fi);
1812   dst->i = new int[dst->count + 1];
1813   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1814     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1815     dst->i[i] = PyInt_AsLong(item);
1816   }
1817   dst->i[dst->count] = 0;
1818   Py_DECREF(fi);
1819   return 1;
1820 }
1821
1822 struct dims
1823 {
1824   int count;
1825   int i[CV_MAX_DIM];
1826   int step[CV_MAX_DIM];
1827   int length[CV_MAX_DIM];
1828 };
1829
1830 static int convert_to_dim(PyObject *item, int i, dims *dst, CvArr *cva, const char *name = "no_name")
1831 {
1832   if (PySlice_Check(item)) {
1833     Py_ssize_t start, stop, step, slicelength;
1834     PySlice_GetIndicesEx((PySliceObject*)item, cvGetDimSize(cva, i), &start, &stop, &step, &slicelength);
1835     dst->i[i] = start;
1836     dst->step[i] = step;
1837     dst->length[i] = slicelength;
1838   } else {
1839     int index = PyInt_AsLong(item);
1840     if (0 <= index)
1841       dst->i[i] = index;
1842     else
1843       dst->i[i] = cvGetDimSize(cva, i) + index;
1844     dst->step[i] = 0;
1845     dst->length[i] = 1;
1846   }
1847   return 1;
1848 }
1849
1850 static int convert_to_dims(PyObject *o, dims *dst, CvArr *cva, const char *name = "no_name")
1851 {
1852   if (!PyTuple_Check(o)) {
1853     dst->count = 1;
1854     return convert_to_dim(o, 0, dst, cva, name);
1855   } else {
1856     PyObject *fi = PySequence_Fast(o, name);
1857     if (fi == NULL) {
1858       PyErr_SetString(PyExc_TypeError, "Expected tuple for index");
1859       return 0;
1860     }
1861     dst->count = PySequence_Fast_GET_SIZE(fi);
1862     for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1863       if (i >= cvGetDims(cva)) {
1864         return failmsg("Access specifies %d dimensions, but array only has %d", PySequence_Fast_GET_SIZE(fi), cvGetDims(cva));
1865       }
1866       PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1867       if (!convert_to_dim(item, i, dst, cva, name))
1868         return 0;
1869     }
1870     Py_DECREF(fi);
1871     return 1;
1872   }
1873 }
1874
1875 struct IplImages {
1876   IplImage **ims;
1877   int count;
1878 };
1879 static int convert_to_IplImages(PyObject *o, IplImages *dst, const char *name = "no_name")
1880 {
1881   PyObject *fi = PySequence_Fast(o, name);
1882   if (fi == NULL)
1883     return 0;
1884   dst->count = PySequence_Fast_GET_SIZE(fi);
1885   dst->ims = new IplImage*[dst->count];
1886   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1887     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1888     if (!convert_to_IplImage(item, &dst->ims[i]))
1889       return 0;
1890   }
1891   Py_DECREF(fi);
1892   return 1;
1893 }
1894
1895 struct CvArrs {
1896   CvArr **ims;
1897   int count;
1898 };
1899 static int convert_to_CvArrs(PyObject *o, CvArrs *dst, const char *name = "no_name")
1900 {
1901   PyObject *fi = PySequence_Fast(o, name);
1902   if (fi == NULL)
1903     return 0;
1904   dst->count = PySequence_Fast_GET_SIZE(fi);
1905   dst->ims = new CvArr*[dst->count];
1906   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1907     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1908     if (!convert_to_CvArr(item, &dst->ims[i]))
1909       return 0;
1910   }
1911   Py_DECREF(fi);
1912   return 1;
1913 }
1914
1915 static int convert_to_floatPTRPTR(PyObject *o, float*** dst, const char *name = "no_name")
1916 {
1917   PyObject *fi = PySequence_Fast(o, name);
1918   if (fi == NULL)
1919     return 0;
1920   Py_ssize_t sz = PySequence_Fast_GET_SIZE(fi);
1921   float **r = new float*[sz];
1922   for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1923     PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1924     floats ff;
1925     if (!convert_to_floats(item, &ff))
1926       return 0;
1927     r[i] = ff.f;
1928   }
1929   *dst = r;
1930   return 1;
1931 }
1932
1933 static int convert_to_IplConvKernelPTR(PyObject *o, IplConvKernel** dst, const char *name = "no_name")
1934 {
1935   if (PyType_IsSubtype(o->ob_type, &iplconvkernel_Type)) {
1936     *dst = ((iplconvkernel_t*)o)->a;
1937     return 1;
1938   } else {
1939     (*dst) = (IplConvKernel*)NULL;
1940     return failmsg("Expected IplConvKernel for argument '%s'", name);
1941   }
1942 }
1943
1944 static int convert_to_CvCapturePTR(PyObject *o, CvCapture** dst, const char *name = "no_name")
1945 {
1946   if (PyType_IsSubtype(o->ob_type, &cvcapture_Type)) {
1947     *dst = ((cvcapture_t*)o)->a;
1948     return 1;
1949   } else {
1950     (*dst) = (CvCapture*)NULL;
1951     return failmsg("Expected CvCapture for argument '%s'", name);
1952   }
1953 }
1954
1955 static int convert_to_CvVideoWriterPTR(PyObject *o, CvVideoWriter** dst, const char *name = "no_name")
1956 {
1957   if (PyType_IsSubtype(o->ob_type, &cvvideowriter_Type)) {
1958     *dst = ((cvvideowriter_t*)o)->a;
1959     return 1;
1960   } else {
1961     (*dst) = (CvVideoWriter*)NULL;
1962     return failmsg("Expected CvVideoWriter for argument '%s'", name);
1963   }
1964 }
1965
1966 static int convert_to_CvMomentsPTR(PyObject *o, CvMoments** dst, const char *name = "no_name")
1967 {
1968   if (PyType_IsSubtype(o->ob_type, &cvmoments_Type)) {
1969     (*dst) = &(((cvmoments_t*)o)->a);
1970     return 1;
1971   } else {
1972     (*dst) = (CvMoments*)NULL;
1973     return failmsg("Expected CvMoments for argument '%s'", name);
1974   }
1975 }
1976
1977 static int convert_to_CvFontPTR(PyObject *o, CvFont** dst, const char *name = "no_name")
1978 {
1979   if (PyType_IsSubtype(o->ob_type, &cvfont_Type)) {
1980     (*dst) = &(((cvfont_t*)o)->a);
1981     return 1;
1982   } else {
1983     (*dst) = (CvFont*)NULL;
1984     return failmsg("Expected CvFont for argument '%s'", name);
1985   }
1986 }
1987
1988 static int convert_to_CvHaarClassifierCascadePTR(PyObject *o, CvHaarClassifierCascade** dst, const char *name = "no_name")
1989 {
1990   if (PyType_IsSubtype(o->ob_type, &cvhaarclassifiercascade_Type)) {
1991     (*dst) = ((cvhaarclassifiercascade_t*)o)->a;
1992     return 1;
1993   } else {
1994     (*dst) = (CvHaarClassifierCascade*)NULL;
1995     return failmsg("Expected CvHaarClassifierCascade for argument '%s'", name);
1996   }
1997 }
1998
1999 static int convert_to_CvContourTreePTR(PyObject *o, CvContourTree** dst, const char *name = "no_name")
2000 {
2001   if (PyType_IsSubtype(o->ob_type, &cvcontourtree_Type)) {
2002     (*dst) = ((cvcontourtree_t*)o)->a;
2003     return 1;
2004   } else {
2005     (*dst) = NULL;
2006     return failmsg("Expected CvContourTree for argument '%s'", name);
2007   }
2008 }
2009
2010 static int convert_to_CvPOSITObjectPTR(PyObject *o, CvPOSITObject** dst, const char *name = "no_name")
2011 {
2012   if (PyType_IsSubtype(o->ob_type, &cvpositobject_Type)) {
2013     (*dst) = ((cvpositobject_t*)o)->a;
2014     return 1;
2015   } else {
2016     (*dst) = (CvPOSITObject*)NULL;
2017     return failmsg("Expected CvPOSITObject for argument '%s'", name);
2018   }
2019 }
2020
2021 static int convert_to_CvRNGPTR(PyObject *o, CvRNG** dst, const char *name = "no_name")
2022 {
2023   if (PyType_IsSubtype(o->ob_type, &cvrng_Type)) {
2024     (*dst) = &(((cvrng_t*)o)->a);
2025     return 1;
2026   } else {
2027     (*dst) = (CvRNG*)NULL;
2028     return failmsg("Expected CvRNG for argument '%s'", name);
2029   }
2030 }
2031
2032 typedef void* generic;
2033 static int convert_to_generic(PyObject *o, generic *dst, const char *name = "no_name")
2034 {
2035   if (PyType_IsSubtype(o->ob_type, &iplimage_Type))
2036     return convert_to_IplImage(o, (IplImage**)dst, name);
2037   else if (PyType_IsSubtype(o->ob_type, &cvmat_Type))
2038     return convert_to_CvMat(o, (CvMat**)dst, name);
2039   else if (PyType_IsSubtype(o->ob_type, &cvmatnd_Type))
2040     return convert_to_CvMatND(o, (CvMatND**)dst, name);
2041   else {
2042     return failmsg("Cannot identify type of '%s'", name);
2043   }
2044 }
2045
2046 static int convert_to_CvTermCriteria(PyObject *o, CvTermCriteria* dst, const char *name = "no_name")
2047 {
2048   if (!PyArg_ParseTuple(o, "iid", &dst->type, &dst->max_iter, &dst->epsilon))
2049     return 0;
2050   return 1;
2051 }
2052
2053 static int convert_to_CvBox2D(PyObject *o, CvBox2D* dst, const char *name = "no_name")
2054 {
2055   if (!PyArg_ParseTuple(o, "(ff)(ff)f", &dst->center.x, &dst->center.y, &dst->size.width, &dst->size.height, &dst->angle))
2056     return 0;
2057   return 1;
2058 }
2059
2060 static int convert_to_CvSubdiv2DPTR(PyObject *o, CvSubdiv2D** dst, const char *name = "no_name")
2061 {
2062   if (PyType_IsSubtype(o->ob_type, &cvsubdiv2d_Type)) {
2063     (*dst) = (((cvsubdiv2d_t*)o)->a);
2064     return 1;
2065   } else {
2066     (*dst) = (CvSubdiv2D*)NULL;
2067     return failmsg("Expected CvSubdiv2D for argument '%s'", name);
2068   }
2069 }
2070
2071 static int convert_to_CvNextEdgeType(PyObject *o, CvNextEdgeType *dst, const char *name = "no_name")
2072 {
2073   if (!PyNumber_Check(o)) {
2074     *dst = (CvNextEdgeType)NULL;
2075     return failmsg("Expected number for CvNextEdgeType argument '%s'", name);
2076   } else {
2077     *dst = (CvNextEdgeType)PyInt_AsLong(o);
2078     return 1;
2079   }
2080 }
2081
2082 static int convert_to_CvSubdiv2DEdge(PyObject *o, CvSubdiv2DEdge *dst, const char *name = "no_name")
2083 {
2084   if (PyType_IsSubtype(o->ob_type, &cvsubdiv2dedge_Type)) {
2085     (*dst) = (((cvsubdiv2dedge_t*)o)->a);
2086     return 1;
2087   } else {
2088     *dst = 0L;
2089     return failmsg("Expected CvSubdiv2DEdge for argument '%s'", name);
2090   }
2091 }
2092
2093 /************************************************************************/
2094
2095 static PyObject *pythonize_CvMat(cvmat_t *m)
2096 {
2097   // Need to make this CvMat look like any other, with a Python
2098   // buffer object as its data.
2099   // So copy the image data into a Python string object, then release
2100   // the original.
2101   CvMat *mat = m->a;
2102   assert(mat->step != 0);
2103 #if 0
2104   PyObject *data = PyString_FromStringAndSize((char*)(mat->data.ptr), mat->rows * mat->step);
2105 #else
2106   memtrack_t *o = PyObject_NEW(memtrack_t, &memtrack_Type);
2107   size_t gap = mat->data.ptr - (uchar*)mat->refcount;
2108   o->ptr = mat->refcount;
2109   o->size = gap + mat->rows * mat->step;
2110   PyObject *data = PyBuffer_FromReadWriteObject((PyObject*)o, (size_t)gap, mat->rows * mat->step);
2111   if (data == NULL)
2112     return NULL;
2113 #endif
2114   m->data = data;
2115   m->offset = 0;
2116   Py_DECREF(o);
2117
2118   // Now m has a reference to data, which has a reference to o.
2119
2120   // cvDecRefData(mat); // Ref count should be zero here, so this is a release
2121   return (PyObject*)m;
2122 }
2123
2124 static PyObject *pythonize_IplImage(iplimage_t *cva)
2125 {
2126   // Need to make this iplimage look like any other, with a Python
2127   // string as its data.
2128   // So copy the image data into a Python string object, then release
2129   // it.
2130
2131   IplImage *ipl = (IplImage*)(cva->a);
2132   // PyObject *data = PyString_FromStringAndSize(ipl->imageData, ipl->imageSize);
2133
2134   memtrack_t *o = PyObject_NEW(memtrack_t, &memtrack_Type);
2135   assert(ipl->imageDataOrigin == ipl->imageData);
2136   o->ptr = ipl->imageDataOrigin;
2137   o->size = ipl->height * ipl->widthStep;
2138   PyObject *data = PyBuffer_FromReadWriteObject((PyObject*)o, (size_t)0, o->size);
2139   if (data == NULL)
2140     return NULL;
2141   Py_DECREF(o);
2142   cva->data = data;
2143   cva->offset = 0;
2144
2145   return (PyObject*)cva;
2146 }
2147
2148 static PyObject *pythonize_CvMatND(cvmatnd_t *m)
2149 {
2150   //
2151   // Need to make this CvMatND look like any other, with a Python
2152   // string as its data.
2153   // So copy the image data into a Python string object, then release
2154   // it.
2155   //
2156
2157   CvMatND *mat = m->a;
2158   assert(mat->dim[0].step != 0);
2159 #if 1
2160   PyObject *data = PyString_FromStringAndSize((char*)(mat->data.ptr), mat->dim[0].size * mat->dim[0].step);
2161 #else
2162   memtrack_t *o = PyObject_NEW(memtrack_t, &memtrack_Type);
2163   size_t gap = mat->data.ptr - (uchar*)mat->refcount;
2164   o->ptr = mat->refcount;
2165   o->size = gap + mat->rows * mat->step;
2166   PyObject *data = PyBuffer_FromReadWriteObject((PyObject*)o, (size_t)gap, mat->rows * mat->step);
2167   if (data == NULL)
2168     return NULL;
2169 #endif
2170   m->data = data;
2171   m->offset = 0;
2172   cvDecRefData(mat); // Ref count should be zero here, so this is a release
2173
2174   return (PyObject*)m;
2175 }
2176
2177 /************************************************************************/
2178 /* FROM_xxx:   C -> Python converters.
2179  *
2180  * Turn various OpenCV types (and some aggregate types above)
2181  * into Python objects.  Used by the generated code.
2182  *
2183  * All these functions and macros return a new reference.
2184  */
2185
2186 #define FROM_double(r)  PyFloat_FromDouble(r)
2187 #define FROM_float(r)  PyFloat_FromDouble(r)
2188 #define FROM_int(r)  PyInt_FromLong(r)
2189 #define FROM_unsigned(r)  PyLong_FromUnsignedLong(r)
2190 #define FROM_CvBox2D(r) Py_BuildValue("(ff)(ff)f", r.center.x, r.center.y, r.size.width, r.size.height, r.angle)
2191 #define FROM_CvScalar(r)  Py_BuildValue("(ffff)", r.val[0], r.val[1], r.val[2], r.val[3])
2192 #define FROM_CvPoint(r)  Py_BuildValue("(ii)", r.x, r.y)
2193 #define FROM_CvConnectedComp(r) Py_BuildValue("(fNN)", r.area, FROM_CvScalar(r.value), FROM_CvRect(r.rect))
2194 #define FROM_CvPoint2D32f(r) Py_BuildValue("(ff)", r.x, r.y)
2195 #define FROM_CvSize(r) Py_BuildValue("(ii)", r.width, r.height)
2196 #define FROM_CvRect(r) Py_BuildValue("(iiii)", r.x, r.y, r.width, r.height)
2197 #define FROM_CvSeqPTR(r) _FROM_CvSeqPTR(r, pyobj_storage)
2198 #define FROM_CvSubdiv2DPTR(r) _FROM_CvSubdiv2DPTR(r, pyobj_storage)
2199 #define FROM_CvPoint2D64f(r) Py_BuildValue("(ff)", r.x, r.y)
2200
2201 static PyObject *_FROM_CvSeqPTR(CvSeq *s, PyObject *storage)
2202 {
2203   cvseq_t *ps = PyObject_NEW(cvseq_t, &cvseq_Type);
2204   ps->a = s;
2205   ps->container = storage;
2206   Py_INCREF(ps->container);
2207   return (PyObject*)ps;
2208 }
2209
2210 static PyObject *_FROM_CvSubdiv2DPTR(CvSubdiv2D *s, PyObject *storage)
2211 {
2212   cvsubdiv2d_t *ps = PyObject_NEW(cvsubdiv2d_t, &cvsubdiv2d_Type);
2213   ps->a = s;
2214   ps->container = storage;
2215   Py_INCREF(ps->container);
2216   return (PyObject*)ps;
2217 }
2218
2219 static PyObject *FROM_floats(floats r)
2220 {
2221   PyObject *pr;
2222
2223   pr = PyList_New(r.count);
2224   for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2225     PyList_SetItem(pr, i, PyFloat_FromDouble(r.f[i]));
2226   }
2227   return pr;
2228 }
2229
2230 static PyObject *FROM_chars(chars r)
2231 {
2232   PyObject *pr;
2233
2234   pr = PyList_New(r.count);
2235   for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2236     PyList_SetItem(pr, i, PyInt_FromLong(r.f[i]));
2237   }
2238   return pr;
2239 }
2240
2241 static PyObject *FROM_cvpoint2d32f_count(cvpoint2d32f_count r)
2242 {
2243   PyObject *pr;
2244
2245   pr = PyList_New(r.count);
2246   for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2247     PyList_SetItem(pr, i, FROM_CvPoint2D32f(r.points[i]));
2248   }
2249   return pr;
2250 }
2251
2252 static PyObject *FROM_CvPoint2D32fs(CvPoint2D32fs r)
2253 {
2254   PyObject *pr;
2255
2256   pr = PyList_New(r.count);
2257   for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2258     PyList_SetItem(pr, i, FROM_CvPoint2D32f(r.p[i]));
2259   }
2260   return pr;
2261 }
2262
2263 typedef CvSeq CvSeqOfCvConvexityDefect;
2264 static PyObject *FROM_CvSeqOfCvConvexityDefectPTR(CvSeqOfCvConvexityDefect *r)
2265 {
2266   PyObject *pr;
2267   pr = PyList_New(r->total);
2268   for (int i = 0; i < r->total; i++) {
2269     CvConvexityDefect *pd = CV_GET_SEQ_ELEM(CvConvexityDefect, r, i);
2270     PyList_SetItem(pr, i, Py_BuildValue("(ii)(ii)(ii)f",
2271                                         pd->start->x, pd->start->y,
2272                                         pd->end->x, pd->end->y,
2273                                         pd->depth_point->x, pd->depth_point->y,
2274                                         pd->depth));
2275   }
2276   // This function has copied the CvSeq data into a list.  Hence the
2277   // CvSeq is not being returned to the caller.  Hence, no reference
2278   // count increase for the storage, unlike _FROM_CvSeqPTR.
2279   return pr;
2280 }
2281
2282 typedef CvSeq CvSeqOfCvAvgComp;
2283 static PyObject *FROM_CvSeqOfCvAvgCompPTR(CvSeqOfCvAvgComp *r)
2284 {
2285   PyObject *pr;
2286   pr = PyList_New(r->total);
2287   for (int i = 0; i < r->total; i++) {
2288     CvAvgComp *pd = CV_GET_SEQ_ELEM(CvAvgComp, r, i);
2289     PyList_SetItem(pr, i, Py_BuildValue("(iiii)i",
2290                                         pd->rect.x, pd->rect.y,
2291                                         pd->rect.width, pd->rect.height,
2292                                         pd->neighbors));
2293   }
2294   // This function has copied the CvSeq data into a list.  Hence the
2295   // CvSeq is not being returned to the caller.  Hence, no reference
2296   // count increase for the storage, unlike _FROM_CvSeqPTR.
2297   return pr;
2298 }
2299
2300 typedef CvSeq CvSeqOfCvStarKeypoint;
2301 static PyObject *FROM_CvSeqOfCvStarKeypointPTR(CvSeqOfCvStarKeypoint *r)
2302 {
2303   PyObject *pr;
2304   pr = PyList_New(r->total);
2305   for (int i = 0; i < r->total; i++) {
2306     CvStarKeypoint *pd = CV_GET_SEQ_ELEM(CvStarKeypoint, r, i);
2307     PyList_SetItem(pr, i, Py_BuildValue("(ii)if",
2308                                         pd->pt.x, pd->pt.y,
2309                                         pd->size,
2310                                         pd->response));
2311   }
2312   // This function has copied the CvSeq data into a list.  Hence the
2313   // CvSeq is not being returned to the caller.  Hence, no reference
2314   // count increase for the storage, unlike _FROM_CvSeqPTR.
2315   return pr;
2316 }
2317
2318 typedef CvSeq CvSeqOfCvSURFPoint;
2319 static PyObject *FROM_CvSeqOfCvSURFPointPTR(CvSeqOfCvSURFPoint *r)
2320 {
2321   PyObject *pr;
2322   pr = PyList_New(r->total);
2323   for (int i = 0; i < r->total; i++) {
2324     CvSURFPoint *pd = CV_GET_SEQ_ELEM(CvSURFPoint, r, i);
2325     PyList_SetItem(pr, i, Py_BuildValue("(ff)iiff",
2326                                         pd->pt.x, pd->pt.y,
2327                                         pd->laplacian,
2328                                         pd->size,
2329                                         pd->dir,
2330                                         pd->hessian));
2331   }
2332   // This function has copied the CvSeq data into a list.  Hence the
2333   // CvSeq is not being returned to the caller.  Hence, no reference
2334   // count increase for the storage, unlike _FROM_CvSeqPTR.
2335   return pr;
2336 }
2337
2338 typedef CvSeq CvSeqOfCvSURFDescriptor;
2339 static PyObject *FROM_CvSeqOfCvSURFDescriptorPTR(CvSeqOfCvSURFDescriptor *r)
2340 {
2341   PyObject *pr;
2342   pr = PyList_New(r->total);
2343   for (int i = 0; i < r->total; i++) {
2344     float *pd = (float*)cvGetSeqElem(r, i);
2345     int count = r->elem_size / sizeof(float);
2346     PyObject *oi = PyList_New(count);
2347     for (int j = 0; j < count; j++) {
2348       PyList_SetItem(oi, j, PyFloat_FromDouble(pd[j]));
2349     }
2350     PyList_SetItem(pr, i, oi);
2351   }
2352   // This function has copied the CvSeq data into a list.  Hence the
2353   // CvSeq is not being returned to the caller.  Hence, no reference
2354   // count increase for the storage, unlike _FROM_CvSeqPTR.
2355   return pr;
2356 }
2357
2358 static PyObject *FROM_IplConvKernelPTR(IplConvKernel *r)
2359 {
2360   iplconvkernel_t *ick = PyObject_NEW(iplconvkernel_t, &iplconvkernel_Type);
2361   ick->a = r;
2362   return (PyObject*)ick;
2363 }
2364
2365 static PyObject *FROM_CvCapturePTR(CvCapture *r)
2366 {
2367   cvcapture_t *c = PyObject_NEW(cvcapture_t, &cvcapture_Type);
2368   c->a = r;
2369   return (PyObject*)c;
2370 }
2371
2372 static PyObject *FROM_CvVideoWriterPTR(CvVideoWriter *r)
2373 {
2374   cvvideowriter_t *c = PyObject_NEW(cvvideowriter_t, &cvvideowriter_Type);
2375   c->a = r;
2376   return (PyObject*)c;
2377 }
2378
2379 typedef CvPoint2D32f CvPoint2D32f_4[4];
2380 static PyObject *FROM_CvPoint2D32f_4(CvPoint2D32f* r)
2381 {
2382   return Py_BuildValue("(ff)(ff)(ff)(ff)",
2383                        r[0].x, r[0].y,
2384                        r[1].x, r[1].y,
2385                        r[2].x, r[2].y,
2386                        r[3].x, r[3].y);
2387 }
2388
2389 typedef float CvMatr32f_i[9];
2390
2391 static PyObject *FROM_CvMatr32f_i(CvMatr32f_i r)
2392 {
2393   return Py_BuildValue("(fff)(fff)(fff)",
2394     r[0], r[1], r[2],
2395     r[3], r[4], r[5],
2396     r[6], r[7], r[8]);
2397 }
2398
2399 typedef float CvVect32f_i[3];
2400 static PyObject *FROM_CvVect32f_i(CvVect32f_i r)
2401 {
2402   return Py_BuildValue("fff",
2403     r[0], r[1], r[2]);
2404 }
2405
2406 static PyObject *FROM_CvFont(CvFont r)
2407 {
2408   cvfont_t *cf = PyObject_NEW(cvfont_t, &cvfont_Type);
2409   cf->a = r;
2410   return (PyObject*)cf;
2411 }
2412
2413 static PyObject *FROM_CvSubdiv2DPointPTR(CvSubdiv2DPoint* r)
2414 {
2415   if (r != NULL) {
2416     cvsubdiv2dpoint_t *cf = PyObject_NEW(cvsubdiv2dpoint_t, &cvsubdiv2dpoint_Type);
2417     cf->a = r;
2418     return (PyObject*)cf;
2419   } else {
2420     Py_INCREF(Py_None);
2421     return Py_None;
2422   }
2423 }
2424
2425 static PyObject *FROM_IplImagePTR(IplImage *r)
2426 {
2427   iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2428   cva->a = r;
2429   return pythonize_IplImage(cva);
2430 }
2431
2432 static PyObject *FROM_ROIplImagePTR(ROIplImage *r)
2433 {
2434   if (r != NULL) {
2435     iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2436     cva->a = cvCreateImageHeader(cvSize(100,100), 8, 1);
2437     *(cva->a) = *r;
2438     cva->data = PyBuffer_FromReadWriteMemory(r->imageData, r->height * r->widthStep);
2439     cva->offset = 0;
2440     return (PyObject*)cva;
2441   } else {
2442     Py_RETURN_NONE;
2443   }
2444 }
2445
2446 static PyObject *FROM_CvMatPTR(CvMat *r)
2447 {
2448   cvmat_t *cvm = PyObject_NEW(cvmat_t, &cvmat_Type);
2449   cvm->a = r;
2450
2451   return pythonize_CvMat(cvm);
2452 }
2453
2454 static PyObject *FROM_CvMat(CvMat *r)
2455 {
2456   cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2457   m->a = r;
2458   return pythonize_CvMat(m);
2459 }
2460
2461 static PyObject *FROM_CvMatNDPTR(CvMatND *r)
2462 {
2463   cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2464   m->a = r;
2465   return pythonize_CvMatND(m);
2466 }
2467
2468 static PyObject *FROM_CvPOSITObjectPTR(CvPOSITObject *r)
2469 {
2470   cvpositobject_t *m = PyObject_NEW(cvpositobject_t, &cvpositobject_Type);
2471   m->a = r;
2472   return (PyObject*)m;
2473 }
2474
2475 static PyObject *FROM_CvRNG(CvRNG r)
2476 {
2477   cvrng_t *m = PyObject_NEW(cvrng_t, &cvrng_Type);
2478   m->a = r;
2479   return (PyObject*)m;
2480 }
2481
2482 static PyObject *FROM_CvHaarClassifierCascade(CvHaarClassifierCascade *r)
2483 {
2484   cvhaarclassifiercascade_t *m = PyObject_NEW(cvhaarclassifiercascade_t, &cvhaarclassifiercascade_Type);
2485   m->a = r;
2486   return (PyObject*)m;
2487 }
2488
2489 static PyObject *FROM_CvMoments(CvMoments r)
2490 {
2491   cvmoments_t *m = PyObject_NEW(cvmoments_t, &cvmoments_Type);
2492   m->a = r;
2493   return (PyObject*)m;
2494 }
2495
2496 static PyObject *FROM_CvContourTreePTR(CvContourTree *r)
2497 {
2498   cvcontourtree_t *m = PyObject_NEW(cvcontourtree_t, &cvcontourtree_Type);
2499   m->a = r;
2500   return (PyObject*)m;
2501 }
2502
2503 static PyObject *FROM_generic(generic r)
2504 {
2505   CvTypeInfo* t = cvTypeOf(r);
2506   if (r == NULL) {
2507     failmsg("OpenCV returned NULL");
2508     return NULL;
2509   } if (strcmp(t->type_name, "opencv-image") == 0)
2510     return FROM_IplImagePTR((IplImage*)r);
2511   else if (strcmp(t->type_name, "opencv-matrix") == 0)
2512     return FROM_CvMat((CvMat*)r);
2513   else if (strcmp(t->type_name, "opencv-haar-classifier") == 0)
2514     return FROM_CvHaarClassifierCascade((CvHaarClassifierCascade*)r);
2515   else {
2516     failmsg("Unknown OpenCV type '%s'", t->type_name);
2517     return NULL;
2518   }
2519 }
2520
2521 static PyObject *FROM_CvSubdiv2DEdge(CvSubdiv2DEdge r)
2522 {
2523   cvsubdiv2dedge_t *m = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
2524   m->a = r;
2525   m->container = Py_None; // XXX
2526   Py_INCREF(m->container);
2527   return (PyObject*)m;
2528 }
2529
2530 /************************************************************************/
2531
2532 /* A few functions are too odd to be generated,
2533  * so are handwritten here */
2534
2535 static PyObject *pycvWaitKey(PyObject *self, PyObject *args, PyObject *kw)
2536 {
2537   int delay = 0;
2538
2539   const char *keywords[] = { "delay", NULL };
2540   if (!PyArg_ParseTupleAndKeywords(args, kw, "|i", (char**)keywords, &delay))
2541     return NULL;
2542   int r;
2543   Py_BEGIN_ALLOW_THREADS
2544   r = cvWaitKey(delay);
2545   Py_END_ALLOW_THREADS
2546   return FROM_int(r);
2547 }
2548
2549 static PyObject *pycvLoadImage(PyObject *self, PyObject *args, PyObject *kw)
2550 {
2551   const char *keywords[] = { "filename", "iscolor", NULL };
2552   char *filename;
2553   int iscolor = CV_LOAD_IMAGE_COLOR;
2554
2555   if (!PyArg_ParseTupleAndKeywords(args, kw, "s|i", (char**)keywords, &filename, &iscolor))
2556     return NULL;
2557
2558   // Inside ALLOW_THREADS, must not reference 'filename' because it might move.
2559   // So make a local copy 'filename_copy'.
2560   char filename_copy[2048];
2561   strncpy(filename_copy, filename, sizeof(filename_copy));
2562
2563   IplImage *r;
2564   Py_BEGIN_ALLOW_THREADS
2565   r = cvLoadImage(filename_copy, iscolor);
2566   Py_END_ALLOW_THREADS
2567
2568   if (r == NULL) {
2569     PyErr_SetFromErrnoWithFilename(PyExc_IOError, filename);
2570     return NULL;
2571   } else {
2572     return FROM_IplImagePTR(r);
2573   }
2574 }
2575
2576 static PyObject *pycvCreateImageHeader(PyObject *self, PyObject *args)
2577 {
2578   int w, h, depth, channels;
2579   if (!PyArg_ParseTuple(args, "(ii)Ii", &w, &h, &depth, &channels))
2580     return NULL;
2581   iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2582   cva->a = cvCreateImageHeader(cvSize(w, h), depth, channels);
2583   if (cva->a == NULL) {
2584     PyErr_SetString(PyExc_TypeError, "CreateImage failed");
2585     return NULL;
2586   } else {
2587     cva->data = Py_None;
2588     Py_INCREF(cva->data);
2589     cva->offset = 0;
2590
2591     return (PyObject*)cva;
2592   }
2593 }
2594
2595 static PyObject *pycvCreateImage(PyObject *self, PyObject *args)
2596 {
2597   int w, h, depth, channels;
2598   if (!PyArg_ParseTuple(args, "(ii)Ii:CreateImage", &w, &h, &depth, &channels))
2599     return NULL;
2600   iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2601   cva->a = cvCreateImage(cvSize(w, h), depth, channels);
2602   if (cva->a == NULL) {
2603     PyErr_SetString(PyExc_TypeError, "CreateImage failed");
2604     return NULL;
2605   } else {
2606     return pythonize_IplImage(cva);
2607   }
2608 }
2609
2610 static PyObject *pycvCreateMatHeader(PyObject *self, PyObject *args)
2611 {
2612   int rows, cols, type;
2613   if (!PyArg_ParseTuple(args, "iii", &rows, &cols, &type))
2614     return NULL;
2615   cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2616   m->a = cvCreateMatHeader(rows, cols, type);
2617   if (m->a == NULL) {
2618     PyErr_SetString(PyExc_TypeError, "CreateMat failed");
2619     return NULL;
2620   } else {
2621     m->data = Py_None;
2622     Py_INCREF(m->data);
2623     m->offset = 0;
2624     return (PyObject*)m;
2625   }
2626 }
2627
2628 static PyObject *pycvCreateMat(PyObject *self, PyObject *args)
2629 {
2630   int rows, cols, type;
2631   if (!PyArg_ParseTuple(args, "iii", &rows, &cols, &type))
2632     return NULL;
2633   cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2634   m->a = cvCreateMat(rows, cols, type);
2635   if (m->a == NULL) {
2636     PyErr_SetString(PyExc_TypeError, "CreateMat failed");
2637     return NULL;
2638   } else {
2639     return pythonize_CvMat(m);
2640   }
2641 }
2642
2643 static PyObject *pycvCreateMatNDHeader(PyObject *self, PyObject *args)
2644 {
2645   ints dims;
2646   int type;
2647
2648   if (!PyArg_ParseTuple(args, "O&i", convert_to_ints, (void*)&dims, &type))
2649     return NULL;
2650   cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2651   ERRWRAP(m->a = cvCreateMatNDHeader(dims.count, dims.i, type));
2652
2653   m->data = Py_None;
2654   Py_INCREF(m->data);
2655   return (PyObject*)m;
2656 }
2657
2658
2659 static PyObject *pycvCreateMatND(PyObject *self, PyObject *args)
2660 {
2661   ints dims;
2662   int type;
2663
2664   if (!PyArg_ParseTuple(args, "O&i", convert_to_ints, (void*)&dims, &type))
2665     return NULL;
2666   cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2667   ERRWRAP(m->a = cvCreateMatND(dims.count, dims.i, type));
2668   return pythonize_CvMatND(m);
2669 }
2670
2671 static PyObject *pycvCreateHist(PyObject *self, PyObject *args)
2672 {
2673   PyObject *dims;
2674   int type;
2675   float **ranges = NULL;
2676   int uniform = 1;
2677
2678   if (!PyArg_ParseTuple(args, "Oi|O&i", &dims, &type, convert_to_floatPTRPTR, (void*)&ranges, &uniform)) {
2679     return NULL;
2680   }
2681   cvhistogram_t *h = PyObject_NEW(cvhistogram_t, &cvhistogram_Type);
2682   args = Py_BuildValue("Oi", dims, CV_32FC1);
2683   h->bins = pycvCreateMatND(self, args);
2684   Py_DECREF(args);
2685   if (h->bins == NULL) {
2686     return NULL;
2687   }
2688   h->h.type = CV_HIST_MAGIC_VAL;
2689   if (!convert_to_CvArr(h->bins, &(h->h.bins), "bins"))
2690     return NULL;
2691
2692   ERRWRAP(cvSetHistBinRanges(&(h->h), ranges, uniform));
2693
2694   return (PyObject*)h;
2695 }
2696
2697 static PyObject *pycvInitLineIterator(PyObject *self, PyObject *args)
2698 {
2699   CvArr *image;
2700   CvPoint pt1;
2701   CvPoint pt2;
2702   int connectivity = 8;
2703   int left_to_right = 0;
2704
2705   if (!PyArg_ParseTuple(args, "O&O&O&|ii",
2706                         convert_to_CvArr, &image,
2707                         convert_to_CvPoint, &pt1,
2708                         convert_to_CvPoint, &pt2,
2709                         &connectivity,
2710                         &left_to_right))
2711     return NULL;
2712
2713   cvlineiterator_t *pi = PyObject_NEW(cvlineiterator_t, &cvlineiterator_Type);
2714   pi->count = cvInitLineIterator(image, pt1, pt2, &pi->iter, connectivity, left_to_right);
2715   ERRWRAP(pi->type = cvGetElemType(image));
2716   return (PyObject*)pi;
2717 }
2718
2719 static PyObject *pycvCreateMemStorage(PyObject *self, PyObject *args)
2720 {
2721   int block_size = 0;
2722   if (!PyArg_ParseTuple(args, "|i", &block_size))
2723     return NULL;
2724   cvmemstorage_t *pm = PyObject_NEW(cvmemstorage_t, &cvmemstorage_Type);
2725   pm->a = cvCreateMemStorage(block_size);
2726   return (PyObject*)pm;
2727 }
2728
2729 // single index: return row
2730 // 2 indices: row, column
2731 // both row and column can be slices.  column slice must have a step of 1.
2732 //
2733 // returns a scalar when all dimensions are specified and all are integers.  Otherwise returns a CvMat.
2734 //
2735 static PyObject *cvarr_GetItem(PyObject *o, PyObject *key)
2736 {
2737   dims dd;
2738
2739   CvArr *cva;
2740   if (!convert_to_CvArr(o, &cva, "src"))
2741     return NULL;
2742
2743   if (!convert_to_dims(key, &dd, cva, "key")) {
2744     return NULL;
2745   }
2746
2747   // Figure out if all supplied indices have a stride of zero - means they are not slices
2748   // and if all indices are positive
2749   int all0 = 1;
2750   for (int i = 0; i < dd.count; i++) {
2751     all0 &= (dd.step[i] == 0) && (0 <= dd.i[i]);
2752   }
2753
2754   // if every dimension supplied, and none are slices, return the scalar
2755   if ((cvGetDims(cva) == dd.count) && all0) {
2756     CvScalar s;
2757     ERRWRAP(s = cvGetND(cva, dd.i));
2758     return PyObject_FromCvScalar(s, cvGetElemType(cva));
2759   } else {
2760     // pad missing dimensions
2761     for (int i = dd.count; i < cvGetDims(cva); i++) {
2762       dd.i[i] = 0;
2763       dd.step[i] = 1;
2764       dd.length[i] = cvGetDimSize(cva, i);
2765     }
2766     dd.count = cvGetDims(cva);
2767
2768     // negative steps are illegal for OpenCV
2769     for (int i = 0; i < dd.count; i++) {
2770       if (dd.step[i] < 0)
2771         return (PyObject*)failmsg("Negative step is illegal");
2772     }
2773
2774     // zero length illegal for OpenCV
2775     for (int i = 0; i < dd.count; i++) {
2776       if (dd.length[i] == 0)
2777         return (PyObject*)failmsg("Zero sized dimension is illegal");
2778     }
2779
2780     // column step can only be 0 or 1
2781     if ((dd.step[dd.count-1] != 0) && (dd.step[dd.count-1] != 1))
2782         return (PyObject*)failmsg("Column step is illegal");
2783
2784     if (is_cvmat(o) || is_iplimage(o)) {
2785       cvmat_t *sub = PyObject_NEW(cvmat_t, &cvmat_Type);
2786       sub->a = cvCreateMatHeader(dd.length[0], dd.length[1], cvGetElemType(cva));
2787       uchar *old0;  // pointer to first element in old mat
2788       int oldstep;
2789       cvGetRawData(cva, &old0, &oldstep);
2790       uchar *new0;  // pointer to first element in new mat
2791       ERRWRAP(new0 = cvPtrND(cva, dd.i));
2792
2793       sub->a->step = oldstep * dd.step[0];
2794       sub->data = what_data(o);
2795       Py_INCREF(sub->data);
2796       sub->offset = new0 - old0;
2797       return (PyObject*)sub;
2798     } else {
2799       cvmatnd_t *sub = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2800       sub->a = cvCreateMatNDHeader(dd.count, dd.length, cvGetElemType(cva));
2801       uchar *old0;  // pointer to first element in old mat
2802       cvGetRawData(cva, &old0);
2803       uchar *new0;  // pointer to first element in new mat
2804       ERRWRAP(new0 = cvPtrND(cva, dd.i));
2805
2806       for (int d = 0; d < dd.count; d++) {
2807         int stp = dd.step[d];
2808         sub->a->dim[d].step = ((CvMatND*)cva)->dim[d].step * ((stp == 0) ? 1 : stp);
2809         sub->a->dim[d].size = dd.length[d];
2810       }
2811       sub->data = what_data(o);
2812       Py_INCREF(sub->data);
2813       sub->offset = new0 - old0;
2814       return (PyObject*)sub;
2815     }
2816   }
2817 }
2818
2819 static int cvarr_SetItem(PyObject *o, PyObject *key, PyObject *v)
2820 {
2821   dims dd;
2822
2823   CvArr *cva;
2824   if (!convert_to_CvArr(o, &cva, "src"))
2825     return -1;
2826
2827   if (!convert_to_dims(key, &dd, cva, "key")) {
2828     return -1;
2829   }
2830
2831   if (cvGetDims(cva) != dd.count) {
2832     PyErr_SetString(PyExc_TypeError, "key length does not match array dimension");
2833     return -1;
2834   }
2835
2836   CvScalar s;
2837   if (PySequence_Check(v)) {
2838     PyObject *fi = PySequence_Fast(v, "v");
2839     if (fi == NULL)
2840       return -1;
2841     if (PySequence_Fast_GET_SIZE(fi) != CV_MAT_CN(cvGetElemType(cva))) {
2842       PyErr_SetString(PyExc_TypeError, "sequence size must be same as channel count");
2843       return -1;
2844     }
2845     for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++)
2846       s.val[i] = PyFloat_AsDouble(PySequence_Fast_GET_ITEM(fi, i));
2847     Py_DECREF(fi);
2848   } else {
2849     if (1 != CV_MAT_CN(cvGetElemType(cva))) {
2850       PyErr_SetString(PyExc_TypeError, "scalar supplied but channel count does not equal 1");
2851       return -1;
2852     }
2853     s.val[0] = PyFloat_AsDouble(v);
2854   }
2855   switch (dd.count) {
2856   case 1:
2857     cvSet1D(cva, dd.i[0], s);
2858     break;
2859   case 2:
2860     cvSet2D(cva, dd.i[0], dd.i[1], s);
2861     break;
2862   case 3:
2863     cvSet3D(cva, dd.i[0], dd.i[1], dd.i[2], s);
2864     break;
2865   default:
2866     cvSetND(cva, dd.i, s);
2867     // XXX - OpenCV bug? - seems as if an error in cvSetND does not set error status?
2868     break;
2869   }
2870   if (cvGetErrStatus() != 0) {
2871     translate_error_to_exception();
2872     return -1;
2873   }
2874
2875   return 0;
2876 }
2877
2878
2879 static PyObject *pycvSetData(PyObject *self, PyObject *args)
2880 {
2881   PyObject *o, *s;
2882   int step = -1;
2883
2884   if (!PyArg_ParseTuple(args, "OOi", &o, &s, &step))
2885     return NULL;
2886   if (is_iplimage(o)) {
2887     iplimage_t *ipl = (iplimage_t*)o;
2888     ipl->a->widthStep = step;
2889     Py_DECREF(ipl->data);
2890     ipl->data = s;
2891     Py_INCREF(ipl->data);
2892   } else if (is_cvmat(o)) {
2893     cvmat_t *m = (cvmat_t*)o;
2894     m->a->step = step;
2895     Py_DECREF(m->data);
2896     m->data = s;
2897     Py_INCREF(m->data);
2898   } else if (is_cvmatnd(o)) {
2899     cvmatnd_t *m = (cvmatnd_t*)o;
2900     Py_DECREF(m->data);
2901     m->data = s;
2902     Py_INCREF(m->data);
2903   } else {
2904     PyErr_SetString(PyExc_TypeError, "SetData argument must be either IplImage, CvMat or CvMatND");
2905     return NULL;
2906   }
2907
2908   Py_RETURN_NONE;
2909 }
2910
2911 static PyObject *what_data(PyObject *o)
2912 {
2913   if (is_iplimage(o)) {
2914     iplimage_t *ipl = (iplimage_t*)o;
2915     return ipl->data;
2916   } else if (is_cvmat(o)) {
2917     cvmat_t *m = (cvmat_t*)o;
2918     return m->data;
2919   } else if (is_cvmatnd(o)) {
2920     cvmatnd_t *m = (cvmatnd_t*)o;
2921     return m->data;
2922   } else {
2923     assert(0);
2924     return NULL;
2925   }
2926 }
2927
2928 static PyObject *pycvCreateData(PyObject *self, PyObject *args)
2929 {
2930   PyObject *o;
2931
2932   if (!PyArg_ParseTuple(args, "O", &o))
2933     return NULL;
2934
2935   CvArr *a;
2936   if (!convert_to_CvArr(o, &a, "arr"))
2937     return NULL;
2938   ERRWRAP(cvCreateData(a));
2939
2940   Py_DECREF(what_data(o));
2941   if (is_iplimage(o)) {
2942     iplimage_t *ipl = (iplimage_t*)o;
2943     pythonize_IplImage(ipl);
2944   } else if (is_cvmat(o)) {
2945     cvmat_t *m = (cvmat_t*)o;
2946     pythonize_CvMat(m);
2947   } else if (is_cvmatnd(o)) {
2948     cvmatnd_t *m = (cvmatnd_t*)o;
2949     pythonize_CvMatND(m);
2950   } else {
2951     PyErr_SetString(PyExc_TypeError, "CreateData argument must be either IplImage, CvMat or CvMatND");
2952     return NULL;
2953   }
2954
2955   Py_RETURN_NONE;
2956 }
2957
2958 static PyObject *pycvGetDims(PyObject *self, PyObject *args)
2959 {
2960   PyObject *o;
2961
2962   if (!PyArg_ParseTuple(args, "O", &o))
2963     return NULL;
2964   CvArr *cva;
2965   if (!convert_to_CvArr(o, &cva, "src"))
2966     return NULL;
2967
2968   int i, nd;
2969   ERRWRAP(nd = cvGetDims(cva));
2970   PyObject *r = PyTuple_New(nd);
2971   for (i = 0; i < nd; i++)
2972     PyTuple_SetItem(r, i, PyInt_FromLong(cvGetDimSize(cva, i)));
2973   return r;
2974 }
2975
2976 static PyObject *pycvGetImage(PyObject *self, PyObject *args)
2977 {
2978   PyObject *o, *r;
2979
2980   if (!PyArg_ParseTuple(args, "O", &o))
2981     return NULL;
2982   if (is_iplimage(o)) {
2983     r = o;
2984     Py_INCREF(o);
2985   } else {
2986     IplImage *ipl = cvCreateImageHeader(cvSize(100,100), 8, 1); // these args do not matter, because overwritten
2987     CvArr *cva;
2988     if (!convert_to_CvArr(o, &cva, "src"))
2989       return NULL;
2990     ERRWRAP(cvGetImage(cva, ipl));
2991
2992     iplimage_t *oipl = PyObject_NEW(iplimage_t, &iplimage_Type);
2993     oipl->a = ipl;
2994     oipl->data = what_data(o);
2995     Py_INCREF(oipl->data);
2996     oipl->offset = 0;
2997
2998     r = (PyObject*)oipl;
2999   }
3000   return r;
3001 }
3002
3003 static PyObject *pycvGetMat(PyObject *self, PyObject *args)
3004 {
3005   PyObject *o, *r;
3006
3007   if (!PyArg_ParseTuple(args, "O", &o))
3008     return NULL;
3009   if (is_cvmat(o)) {
3010     r = o;
3011     Py_INCREF(o);
3012   } else {
3013     CvMat *m = cvCreateMatHeader(100,100, 1); // these args do not matter, because overwritten
3014     CvArr *cva;
3015     if (!convert_to_CvArr(o, &cva, "src"))
3016       return NULL;
3017     ERRWRAP(cvGetMat(cva, m));
3018
3019     cvmat_t *om = PyObject_NEW(cvmat_t, &cvmat_Type);
3020     om->a = m;
3021     om->data = what_data(o);
3022     Py_INCREF(om->data);
3023     om->offset = 0;
3024
3025     r = (PyObject*)om;
3026   }
3027   return r;
3028 }
3029
3030 static PyObject *pycvReshape(PyObject *self, PyObject *args)
3031 {
3032   PyObject *o;
3033   int new_cn;
3034   int new_rows = 0;
3035
3036   if (!PyArg_ParseTuple(args, "Oi|i", &o, &new_cn, &new_rows))
3037     return NULL;
3038
3039   CvMat *m = cvCreateMatHeader(100,100, 1); // these args do not matter, because overwritten
3040   CvArr *cva;
3041   if (!convert_to_CvArr(o, &cva, "src"))
3042     return NULL;
3043   ERRWRAP(cvReshape(cva, m, new_cn, new_rows));
3044
3045   cvmat_t *om = PyObject_NEW(cvmat_t, &cvmat_Type);
3046   om->a = m;
3047   om->data = what_data(o);
3048   Py_INCREF(om->data);
3049   om->offset = 0;
3050
3051   return (PyObject*)om;
3052 }
3053
3054 static void OnMouse(int event, int x, int y, int flags, void* param)
3055 {
3056   PyGILState_STATE gstate;
3057   gstate = PyGILState_Ensure();
3058
3059   PyObject *o = (PyObject*)param;
3060   PyObject *args = Py_BuildValue("iiiiO", event, x, y, flags, PyTuple_GetItem(o, 1));
3061
3062   PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
3063   if (r == NULL)
3064     PyErr_Print();
3065   else
3066     Py_DECREF(r);
3067   Py_DECREF(args);
3068   PyGILState_Release(gstate);
3069 }
3070
3071 static PyObject *pycvSetMouseCallback(PyObject *self, PyObject *args, PyObject *kw)
3072 {
3073   const char *keywords[] = { "window_name", "on_mouse", "param", NULL };
3074   char* name;
3075   PyObject *on_mouse;
3076   PyObject *param = NULL;
3077
3078   if (!PyArg_ParseTupleAndKeywords(args, kw, "sO|O", (char**)keywords, &name, &on_mouse, &param))
3079     return NULL;
3080   if (!PyCallable_Check(on_mouse)) {
3081     PyErr_SetString(PyExc_TypeError, "on_mouse must be callable");
3082     return NULL;
3083   }
3084   if (param == NULL) {
3085     param = Py_None;
3086   }
3087   ERRWRAP(cvSetMouseCallback(name, OnMouse, Py_BuildValue("OO", on_mouse, param)));
3088   Py_RETURN_NONE;
3089 }
3090
3091 void OnChange(int pos, void *param)
3092 {
3093   PyGILState_STATE gstate;
3094   gstate = PyGILState_Ensure();
3095
3096   PyObject *o = (PyObject*)param;
3097   PyObject *args = Py_BuildValue("(i)", pos);
3098   PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
3099   if (r == NULL)
3100     PyErr_Print();
3101   Py_DECREF(args);
3102   PyGILState_Release(gstate);
3103 }
3104
3105 static PyObject *pycvCreateTrackbar(PyObject *self, PyObject *args, PyObject *kw)
3106 {
3107   const char *keywords[] = { "trackbar_name", "window_name", "value", "count", "on_change", NULL };
3108   PyObject *on_change;
3109   char* trackbar_name;
3110   char* window_name;
3111   int *value = new int;
3112   int count;
3113
3114   if (!PyArg_ParseTupleAndKeywords(args, kw, "ssiiO", (char**)keywords, &trackbar_name, &window_name, value, &count, &on_change))
3115     return NULL;
3116   if (!PyCallable_Check(on_change)) {
3117     PyErr_SetString(PyExc_TypeError, "on_change must be callable");
3118     return NULL;
3119   }
3120   ERRWRAP(cvCreateTrackbar2(trackbar_name, window_name, value, count, OnChange, Py_BuildValue("OO", on_change, Py_None)));
3121   Py_RETURN_NONE;
3122 }
3123
3124 static PyObject *pycvFindContours(PyObject *self, PyObject *args, PyObject *kw)
3125 {
3126   CvArr* image;
3127   PyObject *pyobj_image = NULL;
3128   CvMemStorage* storage;
3129   PyObject *pyobj_storage = NULL;
3130   CvSeq* first_contour;
3131   int header_size = sizeof(CvContour);
3132   int mode = CV_RETR_LIST;
3133   int method = CV_CHAIN_APPROX_SIMPLE;
3134   CvPoint offset = cvPoint(0,0);
3135   PyObject *pyobj_offset = NULL;
3136
3137   const char *keywords[] = { "image", "storage", "mode", "method", "offset", NULL };
3138   if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|iiO", (char**)keywords, &pyobj_image, &pyobj_storage, &mode, &method, &pyobj_offset))
3139     return NULL;
3140   if (!convert_to_CvArr(pyobj_image, &image, "image")) return NULL;
3141   if (!convert_to_CvMemStorage(pyobj_storage, &storage, "storage")) return NULL;
3142   if ((pyobj_offset != NULL) && !convert_to_CvPoint(pyobj_offset, &offset, "offset")) return NULL;
3143   ERRWRAP(cvFindContours(image, storage, &first_contour, header_size, mode, method, offset));
3144   cvseq_t *ps = PyObject_NEW(cvseq_t, &cvseq_Type);
3145   ps->a = first_contour;
3146   ps->container = PyTuple_GetItem(args, 1); // storage
3147   Py_INCREF(ps->container);
3148   return (PyObject*)ps;
3149 }
3150
3151 static PyObject *pycvApproxPoly(PyObject *self, PyObject *args, PyObject *kw)
3152 {
3153   cvarrseq src_seq;
3154   PyObject *pyobj_src_seq = NULL;
3155   int header_size = sizeof(CvContour);
3156   CvMemStorage* storage;
3157   PyObject *pyobj_storage = NULL;
3158   int method;
3159   double parameter = 0;
3160   int parameter2 = 0;
3161
3162   const char *keywords[] = { "src_seq", "storage", "method", "parameter", "parameter2", NULL };
3163   if (!PyArg_ParseTupleAndKeywords(args, kw, "OOi|di", (char**)keywords, &pyobj_src_seq, &pyobj_storage, &method, &parameter, &parameter2))
3164     return NULL;
3165   if (!convert_to_cvarrseq(pyobj_src_seq, &src_seq, "src_seq")) return NULL;
3166   if (!convert_to_CvMemStorage(pyobj_storage, &storage, "storage")) return NULL;
3167   CvSeq* r;
3168   ERRWRAP(r = cvApproxPoly(src_seq.v, header_size, storage, method, parameter, parameter2));
3169   return FROM_CvSeqPTR(r);
3170 }
3171
3172 static float distance_function_glue( const float* a, const float* b, void* user_param )
3173 {
3174   PyObject *o = (PyObject*)user_param;
3175   PyObject *args = Py_BuildValue("(ff)(ff)O", a[0], a[1], b[0], b[1], PyTuple_GetItem(o, 1));
3176   PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
3177   Py_DECREF(args);
3178   return (float)PyFloat_AsDouble(r);
3179 }
3180
3181 static PyObject *pycvCalcEMD2(PyObject *self, PyObject *args, PyObject *kw)
3182 {
3183   const char *keywords[] = { "signature1", "signature2", "distance_type", "distance_func", "cost_matrix", "flow", "lower_bound", "userdata", NULL };
3184   CvArr* signature1;
3185   PyObject *pyobj_signature1;
3186   CvArr* signature2;
3187   PyObject *pyobj_signature2;
3188   int distance_type;
3189   PyObject *distance_func = NULL;
3190   CvArr* cost_matrix=NULL;
3191   PyObject *pyobj_cost_matrix = NULL;
3192   CvArr* flow=NULL;
3193   PyObject *pyobj_flow = NULL;
3194   float lower_bound = 0.0;
3195   PyObject *userdata = NULL;
3196
3197   if (!PyArg_ParseTupleAndKeywords(args, kw, "OOi|OOOfO", (char**)keywords,
3198                                    &pyobj_signature1,
3199                                    &pyobj_signature2,
3200                                    &distance_type,
3201                                    &distance_func,
3202                                    &pyobj_cost_matrix,
3203                                    &pyobj_flow,
3204                                    &lower_bound,
3205                                    &userdata))
3206     return NULL;
3207   if (!convert_to_CvArr(pyobj_signature1, &signature1, "signature1")) return NULL;
3208   if (!convert_to_CvArr(pyobj_signature2, &signature2, "signature2")) return NULL;
3209   if (pyobj_cost_matrix && !convert_to_CvArr(pyobj_cost_matrix, &cost_matrix, "cost_matrix")) return NULL;
3210   if (pyobj_flow && !convert_to_CvArr(pyobj_flow, &flow, "flow")) return NULL;
3211
3212   if (distance_func == NULL) {
3213     distance_func = Py_None;
3214   }
3215   if (userdata == NULL) {
3216     userdata = Py_None;
3217   }
3218
3219   PyObject *ud = Py_BuildValue("OO", distance_func, userdata);
3220   float r;
3221   ERRWRAP(r = cvCalcEMD2(signature1, signature2, distance_type, distance_function_glue, cost_matrix, flow, &lower_bound, (void*)ud));
3222   Py_DECREF(ud);
3223
3224   return PyFloat_FromDouble(r);
3225 }
3226
3227 static PyObject *pycvSubdiv2DLocate(PyObject *self, PyObject *args)
3228 {
3229   PyObject *pyobj_subdiv;
3230   PyObject *pyobj_pt;
3231   CvSubdiv2D *subdiv;
3232   CvPoint2D32f pt;
3233   CvSubdiv2DEdge edge;
3234   CvSubdiv2DPoint* vertex;
3235
3236   if (!PyArg_ParseTuple(args, "OO", &pyobj_subdiv, &pyobj_pt))
3237     return NULL;
3238   if (!convert_to_CvSubdiv2DPTR(pyobj_subdiv, &subdiv, "subdiv"))
3239     return NULL;
3240   if (!convert_to_CvPoint2D32f(pyobj_pt, &pt, "pt"))
3241     return NULL;
3242
3243   CvSubdiv2DPointLocation loc = cvSubdiv2DLocate(subdiv, pt, &edge, &vertex);
3244   PyObject *r;
3245   switch (loc) {
3246   case CV_PTLOC_INSIDE:
3247   case CV_PTLOC_ON_EDGE:
3248     r = FROM_CvSubdiv2DEdge(edge);
3249     break;
3250   case CV_PTLOC_VERTEX:
3251     r = FROM_CvSubdiv2DPointPTR(vertex);
3252     break;
3253   case CV_PTLOC_OUTSIDE_RECT:
3254     r = Py_None;
3255     Py_INCREF(Py_None);
3256     break;
3257   default:
3258     return (PyObject*)failmsg("Unexpected loc from cvSubdiv2DLocate");
3259   }
3260   return Py_BuildValue("iO", (int)loc, r);;
3261 }
3262
3263 static PyObject *pycvCalcOpticalFlowPyrLK(PyObject *self, PyObject *args)
3264 {
3265   CvArr* prev;
3266   PyObject *pyobj_prev = NULL;
3267   CvArr* curr;
3268   PyObject *pyobj_curr = NULL;
3269   CvArr* prev_pyr;
3270   PyObject *pyobj_prev_pyr = NULL;
3271   CvArr* curr_pyr;
3272   PyObject *pyobj_curr_pyr = NULL;
3273   CvPoint2D32f* prev_features;
3274   PyObject *pyobj_prev_features = NULL;
3275   PyObject *pyobj_curr_features = NULL;
3276   CvPoint2D32f* curr_features;
3277   CvSize win_size;
3278   int level;
3279   CvTermCriteria criteria;
3280   int flags;
3281
3282   if (!PyArg_ParseTuple(args, "OOOOO(ii)i(iif)i|O",
3283     &pyobj_prev, &pyobj_curr, &pyobj_prev_pyr, &pyobj_curr_pyr,
3284     &pyobj_prev_features,
3285     &win_size.width, &win_size.height, &level,
3286     &criteria.type, &criteria.max_iter, &criteria.epsilon,
3287     &flags,
3288     &pyobj_curr_features))
3289     return NULL;
3290   if (!convert_to_CvArr(pyobj_prev, &prev, "prev")) return NULL;
3291   if (!convert_to_CvArr(pyobj_curr, &curr, "curr")) return NULL;
3292   if (!convert_to_CvArr(pyobj_prev_pyr, &prev_pyr, "prev_pyr")) return NULL;
3293   if (!convert_to_CvArr(pyobj_curr_pyr, &curr_pyr, "curr_pyr")) return NULL;
3294   if (!convert_to_CvPoint2D32fPTR(pyobj_prev_features, &prev_features, "prev_features")) return NULL;
3295   int count = (int)PySequence_Length(pyobj_prev_features);
3296   if (flags & CV_LKFLOW_INITIAL_GUESSES) {
3297     failmsg("flag CV_LKFLOW_INITIAL_GUESSES is determined automatically from function arguments - it is not required");
3298     return NULL;
3299   }
3300   if (!pyobj_curr_features) {
3301     curr_features = new CvPoint2D32f[count];
3302   } else {
3303     if (PySequence_Length(pyobj_curr_features) != count) {
3304       failmsg("curr_features must have same length as prev_features");
3305       return NULL;
3306     }
3307     if (!convert_to_CvPoint2D32fPTR(pyobj_curr_features, &curr_features, "curr_features")) return NULL;
3308     flags |= CV_LKFLOW_INITIAL_GUESSES;
3309   }
3310   float *track_error = new float[count];
3311   char* status = new char[count];
3312   ERRWRAP(cvCalcOpticalFlowPyrLK(prev, curr, prev_pyr, curr_pyr, prev_features, curr_features, count, win_size, level, status, track_error, criteria, flags));
3313
3314   cvpoint2d32f_count r0;
3315   r0.points = curr_features;
3316   r0.count = count;
3317
3318   chars r1;
3319   r1.f = status;
3320   r1.count = count;
3321
3322   floats r2;
3323   r2.f = track_error;
3324   r2.count = count;
3325
3326   return Py_BuildValue("NNN", FROM_cvpoint2d32f_count(r0), FROM_chars(r1), FROM_floats(r2));
3327 }
3328
3329 static PyObject *pycvClipLine(PyObject *self, PyObject *args, PyObject *kw)
3330 {
3331   CvSize img_size;
3332   PyObject *pyobj_img_size = NULL;
3333   CvPoint pt1;
3334   PyObject *pyobj_pt1 = NULL;
3335   CvPoint pt2;
3336   PyObject *pyobj_pt2 = NULL;
3337
3338   if (!PyArg_ParseTuple(args, "OOO", &pyobj_img_size, &pyobj_pt1, &pyobj_pt2))
3339     return NULL;
3340   if (!convert_to_CvSize(pyobj_img_size, &img_size, "img_size")) return NULL;
3341   if (!convert_to_CvPoint(pyobj_pt1, &pt1, "pt1")) return NULL;
3342   if (!convert_to_CvPoint(pyobj_pt2, &pt2, "pt2")) return NULL;
3343   int r;
3344   ERRWRAP(r = cvClipLine(img_size, &pt1, &pt2));
3345   if (r == 0) {
3346     Py_RETURN_NONE;
3347   } else {
3348     return Py_BuildValue("NN", FROM_CvPoint(pt1), FROM_CvPoint(pt2));
3349   }
3350 }
3351
3352 static PyObject *temp_test(PyObject *self, PyObject *args)
3353 {
3354 #if 0
3355   CvArr *im = cvLoadImage("../samples/c/lena.jpg", 0);
3356   printf("im=%p\n", im);
3357   CvMat *m = cvEncodeImage(".jpeg", im);
3358 #endif
3359 #if 0
3360   CvArr *im = cvLoadImage("lena.jpg", 0);
3361   float r0[] = { 0, 255 };
3362   float *ranges[] = { r0 };
3363   int hist_size[] = { 256 };
3364   CvHistogram *hist = cvCreateHist(1, hist_size, CV_HIST_ARRAY, ranges, 1);
3365   cvCalcHist(im, hist, 0, 0);
3366 #endif
3367
3368 #if 0
3369   CvMat* mat = cvCreateMat( 3, 3, CV_32F );
3370   CvMat row_header, *row;
3371   row = cvReshape( mat, &row_header, 0, 1 );
3372   printf("%d,%d\n", row_header.rows, row_header.cols);
3373   printf("ge %08x\n", cvGetElemType(mat));
3374 #endif
3375
3376 #if 0
3377   CvMat *m = cvCreateMat(1, 10, CV_8UC1);
3378   printf("CvMat stride ===> %d\n", m->step);
3379 #endif
3380
3381 #if 0
3382   CvPoint2D32f src[3] = { { 0,0 }, { 1,0 }, { 0,1 } };
3383   CvPoint2D32f dst[3] = { { 0,0 }, { 17,0 }, { 0,17 } };
3384
3385   CvMat* mapping = cvCreateMat(2, 3, CV_32FC1);
3386   cvGetAffineTransform(src, dst, mapping);
3387   printf("===> %f\n", cvGetReal2D(mapping, 0, 0));
3388 #endif
3389
3390 #if 0
3391   CvArr *im = cvLoadImage("checker77.png");
3392   CvPoint2D32f corners[49];
3393   int count;
3394   cvFindChessboardCorners(im, cvSize(7,7), corners, &count, 0);
3395   printf("count=%d\n", count);
3396 #endif
3397
3398   return PyFloat_FromDouble(0.0);
3399 }
3400
3401 static PyObject *pycvFindChessboardCorners(PyObject *self, PyObject *args, PyObject *kw)
3402 {
3403   CvArr* image;
3404   PyObject *pyobj_image = NULL;
3405   CvSize pattern_size;
3406   PyObject *pyobj_pattern_size = NULL;
3407   cvpoint2d32f_count corners;
3408   int flags = CV_CALIB_CB_ADAPTIVE_THRESH;
3409
3410   const char *keywords[] = { "image", "pattern_size", "flags", NULL };
3411   if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|i", (char**)keywords, &pyobj_image, &pyobj_pattern_size, &flags))
3412     return NULL;
3413   if (!convert_to_CvArr(pyobj_image, &image, "image")) return NULL;
3414   if (!convert_to_CvSize(pyobj_pattern_size, &pattern_size, "pattern_size")) return NULL;
3415   int r;
3416   corners.points = new CvPoint2D32f[pattern_size.width * pattern_size.height];
3417   ERRWRAP(r = cvFindChessboardCorners(image, pattern_size, corners.points,&corners.count, flags));
3418   return Py_BuildValue("NN", FROM_int(r), FROM_cvpoint2d32f_count(corners));
3419 }
3420
3421 // For functions GetSubRect, GetRow, GetCol.
3422 // recipient has a view into donor's data, and needs to share it.
3423 // make recipient use the donor's data, compute the offset,
3424 // and manage reference counts.
3425
3426 static void preShareData(CvArr *donor, CvMat **recipient)
3427 {
3428   *recipient = cvCreateMatHeader(4, 4, cvGetElemType(donor));
3429 }
3430
3431 static PyObject *shareData(PyObject *donor, CvArr *pdonor, CvMat *precipient)
3432 {
3433   PyObject *recipient = (PyObject*)PyObject_NEW(cvmat_t, &cvmat_Type);
3434   ((cvmat_t*)recipient)->a = precipient;
3435   ((cvmat_t*)recipient)->offset = cvPtr1D(precipient, 0) - cvPtr1D(pdonor, 0);
3436
3437   PyObject *arr_data;
3438   if (is_cvmat(donor)) {
3439     arr_data = ((cvmat_t*)donor)->data;
3440   } else if (is_iplimage(donor)) {
3441     arr_data = ((iplimage_t*)donor)->data;
3442   } else {
3443     return (PyObject*)failmsg("Argument 'mat' must be either IplImage or CvMat");
3444   }
3445   ((cvmat_t*)recipient)->data = arr_data;
3446   Py_INCREF(arr_data);
3447   return recipient;
3448 }
3449
3450 static PyObject *pycvGetHuMoments(PyObject *self, PyObject *args, PyObject *kw)
3451 {
3452   CvMoments* moments;
3453   PyObject *pyobj_moments = NULL;
3454
3455   if (!PyArg_ParseTuple(args, "O", &pyobj_moments))
3456     return NULL;
3457   if (!convert_to_CvMomentsPTR(pyobj_moments, &moments, "moments")) return NULL;
3458   CvHuMoments r;
3459   ERRWRAP(cvGetHuMoments(moments, &r));
3460   return Py_BuildValue("ddddddd", r.hu1, r.hu2, r.hu3, r.hu4, r.hu5, r.hu6, r.hu7);
3461 }
3462
3463 static PyObject *pycvFitLine(PyObject *self, PyObject *args, PyObject *kw)
3464 {
3465   cvarrseq points;
3466   PyObject *pyobj_points = NULL;
3467   int dist_type;
3468   float param;
3469   float reps;
3470   float aeps;
3471   float r[6];
3472
3473   if (!PyArg_ParseTuple(args, "Oifff", &pyobj_points, &dist_type, &param, &reps, &aeps))
3474     return NULL;
3475   if (!convert_to_cvarrseq(pyobj_points, &points, "points")) return NULL;
3476   ERRWRAP(cvFitLine(points.v, dist_type, param, reps, aeps, r));
3477   int dimension;
3478   if (strcmp("opencv-matrix", cvTypeOf(points.v)->type_name) == 0)
3479     dimension = CV_MAT_CN(cvGetElemType(points.v));
3480   else {
3481     // sequence case... don't think there is a sequence of 3d points,
3482     // so assume 2D
3483     dimension = 2;
3484   }
3485   if (dimension == 2)
3486     return Py_BuildValue("dddd", r[0], r[1], r[2], r[3]);
3487   else
3488     return Py_BuildValue("dddddd", r[0], r[1], r[2], r[3], r[4], r[5]);
3489 }
3490
3491 static PyObject *pycvGetMinMaxHistValue(PyObject *self, PyObject *args, PyObject *kw)
3492 {
3493   CvHistogram* hist;
3494   PyObject *pyobj_hist = NULL;
3495   float min_val;
3496   float max_val;
3497   int min_loc[CV_MAX_DIM];
3498   int max_loc[CV_MAX_DIM];
3499
3500   if (!PyArg_ParseTuple(args, "O", &pyobj_hist))
3501     return NULL;
3502   if (!convert_to_CvHistogram(pyobj_hist, &hist, "hist")) return NULL;
3503   ERRWRAP(cvGetMinMaxHistValue(hist, &min_val, &max_val, min_loc, max_loc));
3504   int d = cvGetDims(hist->bins);
3505   PyObject *pminloc = PyTuple_New(d), *pmaxloc = PyTuple_New(d);
3506   for (int i = 0; i < d; i++) {
3507     PyTuple_SetItem(pminloc, i, PyInt_FromLong(min_loc[i]));
3508     PyTuple_SetItem(pmaxloc, i, PyInt_FromLong(max_loc[i]));
3509   }
3510   return Py_BuildValue("ffNN", min_val, max_val, pminloc, pmaxloc);
3511 }
3512
3513 static CvSeq* cvHOGDetectMultiScale( const CvArr* image, CvMemStorage* storage,
3514   const CvArr* svm_classifier=NULL, CvSize win_stride=cvSize(0,0),
3515   double hit_threshold=0, double scale=1.05,
3516   int group_threshold=2, CvSize padding=cvSize(0,0),
3517   CvSize win_size=cvSize(64,128), CvSize block_size=cvSize(16,16),
3518   CvSize block_stride=cvSize(8,8), CvSize cell_size=cvSize(8,8),
3519   int nbins=9, int gammaCorrection=1 )
3520 {
3521     cv::HOGDescriptor hog(win_size, block_size, block_stride, cell_size, nbins, 1, -1, cv::HOGDescriptor::L2Hys, 0.2, gammaCorrection!=0);
3522     if(win_stride.width == 0 && win_stride.height == 0)
3523         win_stride = block_stride;
3524     cv::Mat img = cv::cvarrToMat(image);
3525     std::vector<cv::Rect> found;
3526     if(svm_classifier)
3527     {
3528         CvMat stub, *m = cvGetMat(svm_classifier, &stub);
3529         int sz = m->cols*m->rows;
3530         CV_Assert(CV_IS_MAT_CONT(m->type) && (m->cols == 1 || m->rows == 1) && CV_MAT_TYPE(m->type) == CV_32FC1);
3531         std::vector<float> w(sz);
3532         std::copy(m->data.fl, m->data.fl + sz, w.begin());
3533         hog.setSVMDetector(w);
3534     }
3535     else
3536         hog.setSVMDetector(cv::HOGDescriptor::getDefaultPeopleDetector());
3537     hog.detectMultiScale(img, found, hit_threshold, win_stride, padding, scale, group_threshold);
3538     CvSeq* seq = cvCreateSeq(cv::DataType<cv::Rect>::type, sizeof(CvSeq), sizeof(cv::Rect), storage);
3539     if(found.size())
3540         cvSeqPushMulti(seq, &found[0], (int)found.size());
3541     return seq;
3542 }
3543
3544 static int zero = 0;
3545
3546 /************************************************************************/
3547 /* Custom Validators */
3548
3549 #define CVPY_VALIDATE_DrawChessboardCorners() do { \
3550   if ((pattern_size.width * pattern_size.height) != corners.count) \
3551     return (PyObject*)failmsg("Size is %dx%d, but corner list is length %d", pattern_size.width, pattern_size.height, corners.count); \
3552   } while (0)
3553
3554 #define cvGetRotationMatrix2D cv2DRotationMatrix
3555
3556 /************************************************************************/
3557 /* Generated functions */
3558
3559 #include "generated0.i"
3560
3561 static PyMethodDef methods[] = {
3562
3563   {"CreateHist", pycvCreateHist, METH_VARARGS, "CreateHist(dims, type, ranges, uniform = 1) -> hist"},
3564   {"CreateImage", pycvCreateImage, METH_VARARGS, "CreateImage(size, depth, channels) -> image"},
3565   {"CreateImageHeader", pycvCreateImageHeader, METH_VARARGS, "CreateImageHeader(size, depth, channels) -> image"},
3566   {"CreateMat", pycvCreateMat, METH_VARARGS, "CreateMat(row, cols, type) -> mat"},
3567   {"CreateMatHeader", pycvCreateMatHeader, METH_VARARGS, "CreateMatHeader(rows, cols, type) -> mat"},
3568   {"CreateMatND", pycvCreateMatND, METH_VARARGS, "CreateMatND(dims, type) -> matnd"},
3569   {"CreateMatNDHeader", pycvCreateMatNDHeader, METH_VARARGS, "CreateMatNDHeader(dims, type) -> matnd"},
3570   {"CreateMemStorage", pycvCreateMemStorage, METH_VARARGS, "CreateMemStorage(block_size) -> memstorage"},
3571   {"FindContours", (PyCFunction)pycvFindContours, METH_KEYWORDS, "FindContours(image, storage, mode=CV_RETR_LIST, method=CV_CHAIN_APPROX_SIMPLE, offset=(0, 0)) -> cvseq"},
3572   {"ApproxPoly", (PyCFunction)pycvApproxPoly, METH_KEYWORDS, "ApproxPoly(src_seq, storage, method, parameter=0, parameter2=0) -> None"},
3573   {"CreateData", pycvCreateData, METH_VARARGS, "CreateData(arr) -> None"},
3574   {"GetDims", pycvGetDims, METH_VARARGS, "GetDims(arr) -> dims"},
3575   {"GetImage", pycvGetImage, METH_VARARGS, "GetImage(cvmat) -> image"},
3576   {"GetMat", pycvGetMat, METH_VARARGS, "GetMat(image) -> cvmat"},
3577   {"Reshape", pycvReshape, METH_VARARGS, "Reshape(arr, new_cn, new_rows=0) -> cvmat"},
3578   {"InitLineIterator", pycvInitLineIterator, METH_VARARGS, "InitLineIterator(image, pt1, pt2, connectivity=8, left_to_right=0) -> None"},
3579   {"LoadImage", (PyCFunction)pycvLoadImage, METH_KEYWORDS, "LoadImage(filename, iscolor=CV_LOAD_IMAGE_COLOR)"},
3580   {"SetData", pycvSetData, METH_VARARGS, "SetData(arr, data, step)"},
3581   {"SetMouseCallback", (PyCFunction)pycvSetMouseCallback, METH_KEYWORDS, "SetMouseCallback(window_name, on_mouse, param) -> None"},
3582   {"CreateTrackbar", (PyCFunction)pycvCreateTrackbar, METH_KEYWORDS, "CreateTrackbar(trackbar_name, window_name, value, count, on_change) -> None"},
3583   {"CalcEMD2", (PyCFunction)pycvCalcEMD2, METH_KEYWORDS, "CalcEMD2(signature1, signature2, distance_type, distance_func = None, cost_matrix=None, flow=None, lower_bound=None, userdata = None) -> float"},
3584   {"FindChessboardCorners", (PyCFunction)pycvFindChessboardCorners, METH_KEYWORDS, "FindChessboardCorners(image, pattern_size, flags=CV_CALIB_CB_ADAPTIVE_THRESH) -> success,corners"},
3585   {"FitLine", (PyCFunction)pycvFitLine, METH_KEYWORDS, "FitLine(points, dist_type, param, reps, aeps) -> line"},
3586   {"Subdiv2DLocate", pycvSubdiv2DLocate, METH_VARARGS, "Subdiv2DLocate(subdiv, pt) -> (loc, where)"},
3587   {"CalcOpticalFlowPyrLK", pycvCalcOpticalFlowPyrLK, METH_VARARGS, "CalcOpticalFlowPyrLK(prev, curr, prev_pyr, curr_pyr, prev_features, CvSize win_size, int level, criteria, flags, guesses = None) -> (curr_features, status, track_error)"},
3588   {"ClipLine", (PyCFunction)pycvClipLine, METH_KEYWORDS, "ClipLine(img, pt1, pt2) -> (clipped_pt1, clipped_pt2)"},
3589   {"GetHuMoments", (PyCFunction)pycvGetHuMoments, METH_KEYWORDS, "GetHuMoments(cvmoments) -> (h1, h2, h3, h4, h5, h5, h7)"},
3590   {"GetMinMaxHistValue", (PyCFunction)pycvGetMinMaxHistValue, METH_KEYWORDS, "GetMinMaxHistValue(hist) -> min_val,max_val,min_loc,max_loc"},
3591   {"WaitKey", (PyCFunction)pycvWaitKey, METH_KEYWORDS, "WaitKey(delay=0) -> int"},
3592   //{"CalcOpticalFlowFarneback", (PyCFunction)pycvCalcOpticalFlowFarneback, METH_KEYWORDS, "CalcOpticalFlowFarneback(prev, next, flow, pyr_scale=0.5, levels=3, win_size=15, iterations=3, poly_n=7, poly_sigma=1.5, flags=0) -> None"},
3593   //{"_HOGComputeDescriptors", (PyCFunction)pycvHOGComputeDescriptors, METH_KEYWORDS, "_HOGComputeDescriptors(image, win_stride=block_stride, locations=None, padding=(0,0), win_size=(64,128), block_size=(16,16), block_stride=(8,8), cell_size=(8,8), nbins=9, gammaCorrection=true) -> list_of_descriptors"},
3594   //{"_HOGDetect", (PyCFunction)pycvHOGDetect, METH_KEYWORDS, "_HOGDetect(image, svm_classifier, win_stride=block_stride, locations=None, padding=(0,0), win_size=(64,128), block_size=(16,16), block_stride=(8,8), cell_size=(8,8), nbins=9, gammaCorrection=true) -> list_of_points"},
3595   //{"_HOGDetectMultiScale", (PyCFunction)pycvHOGDetectMultiScale, METH_KEYWORDS, "_HOGDetectMultiScale(image, svm_classifier, win_stride=block_stride, scale=1.05, group_threshold=2, padding=(0,0), win_size=(64,128), block_size=(16,16), block_stride=(8,8), cell_size=(8,8), nbins=9, gammaCorrection=true) -> list_of_points"},
3596
3597   {"temp_test", temp_test, METH_VARARGS},
3598
3599 #include "generated1.i"
3600
3601   {NULL, NULL},
3602 };
3603
3604 /************************************************************************/
3605 /* Module init */
3606
3607 static int to_ok(PyTypeObject *to)
3608 {
3609   to->tp_alloc = PyType_GenericAlloc;
3610   to->tp_new = PyType_GenericNew;
3611   to->tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE;
3612   return (PyType_Ready(to) == 0);
3613 }
3614
3615 #define MKTYPE(NAME)  do { NAME##_specials(); if (!to_ok(&NAME##_Type)) return; } while (0)
3616
3617 extern "C"
3618 #if defined WIN32 || defined _WIN32
3619 __declspec(dllexport)
3620 #endif
3621 void initcv()
3622 {
3623   PyObject *m, *d;
3624
3625   cvSetErrMode(CV_ErrModeParent);
3626
3627   MKTYPE(cvcapture);
3628   MKTYPE(cvcontourtree);
3629   MKTYPE(cvfont);
3630   MKTYPE(cvhaarclassifiercascade);
3631   MKTYPE(cvhistogram);
3632   MKTYPE(cvlineiterator);
3633   MKTYPE(cvmat);
3634   MKTYPE(cvmatnd);
3635   MKTYPE(cvmemstorage);
3636   MKTYPE(cvmoments);
3637   MKTYPE(cvpositobject);
3638   MKTYPE(cvsubdiv2dedge);
3639   MKTYPE(cvrng);
3640   MKTYPE(cvseq);
3641   MKTYPE(cvset);
3642   MKTYPE(cvsubdiv2d);
3643   MKTYPE(cvsubdiv2dpoint);
3644   MKTYPE(cvvideowriter);
3645   MKTYPE(iplconvkernel);
3646   MKTYPE(iplimage);
3647   MKTYPE(memtrack);
3648
3649   m = Py_InitModule(MODULESTR"", methods);
3650   d = PyModule_GetDict(m);
3651
3652   opencv_error = PyErr_NewException((char*)MODULESTR".error", NULL, NULL);
3653   PyDict_SetItemString(d, "error", opencv_error);
3654
3655   PyDict_SetItemString(d, "iplimage", (PyObject*)&iplimage_Type);
3656   PyDict_SetItemString(d, "cvmat", (PyObject*)&cvmat_Type);
3657
3658 #define PUBLISH(I) PyDict_SetItemString(d, #I, PyInt_FromLong(I))
3659 #define PUBLISHU(I) PyDict_SetItemString(d, #I, PyLong_FromUnsignedLong(I))
3660
3661   PUBLISHU(IPL_DEPTH_8U);
3662   PUBLISHU(IPL_DEPTH_8S);
3663   PUBLISHU(IPL_DEPTH_16U);
3664   PUBLISHU(IPL_DEPTH_16S);
3665   PUBLISHU(IPL_DEPTH_32S);
3666   PUBLISHU(IPL_DEPTH_32F);
3667   PUBLISHU(IPL_DEPTH_64F);
3668
3669   PUBLISH(CV_LOAD_IMAGE_COLOR);
3670   PUBLISH(CV_LOAD_IMAGE_GRAYSCALE);
3671   PUBLISH(CV_LOAD_IMAGE_UNCHANGED);
3672   PUBLISH(CV_HIST_ARRAY);
3673   PUBLISH(CV_HIST_SPARSE);
3674   PUBLISH(CV_8UC1);
3675   PUBLISH(CV_8UC2);
3676   PUBLISH(CV_8UC3);
3677   PUBLISH(CV_8UC4);
3678   PUBLISH(CV_8SC1);
3679   PUBLISH(CV_8SC2);
3680   PUBLISH(CV_8SC3);
3681   PUBLISH(CV_8SC4);
3682   PUBLISH(CV_16UC1);
3683   PUBLISH(CV_16UC2);
3684   PUBLISH(CV_16UC3);
3685   PUBLISH(CV_16UC4);
3686   PUBLISH(CV_16SC1);
3687   PUBLISH(CV_16SC2);
3688   PUBLISH(CV_16SC3);
3689   PUBLISH(CV_16SC4);
3690   PUBLISH(CV_32SC1);
3691   PUBLISH(CV_32SC2);
3692   PUBLISH(CV_32SC3);
3693   PUBLISH(CV_32SC4);
3694   PUBLISH(CV_32FC1);
3695   PUBLISH(CV_32FC2);
3696   PUBLISH(CV_32FC3);
3697   PUBLISH(CV_32FC4);
3698   PUBLISH(CV_64FC1);
3699   PUBLISH(CV_64FC2);
3700   PUBLISH(CV_64FC3);
3701   PUBLISH(CV_64FC4);
3702   PUBLISH(CV_NEXT_AROUND_ORG);
3703   PUBLISH(CV_NEXT_AROUND_DST);
3704   PUBLISH(CV_PREV_AROUND_ORG);
3705   PUBLISH(CV_PREV_AROUND_DST);
3706   PUBLISH(CV_NEXT_AROUND_LEFT);
3707   PUBLISH(CV_NEXT_AROUND_RIGHT);
3708   PUBLISH(CV_PREV_AROUND_LEFT);
3709   PUBLISH(CV_PREV_AROUND_RIGHT);
3710
3711   PUBLISH(CV_WINDOW_AUTOSIZE);
3712
3713   PUBLISH(CV_PTLOC_INSIDE);
3714   PUBLISH(CV_PTLOC_ON_EDGE);
3715   PUBLISH(CV_PTLOC_VERTEX);
3716   PUBLISH(CV_PTLOC_OUTSIDE_RECT);
3717
3718 #include "generated2.i"
3719
3720 #if 0
3721   {
3722     int sizes[] = { 10 } ;
3723     float ranges[] = { 0.0, 1.0 };
3724     // CvHistogram*h = cvCreateHist(1, sizes, CV_HIST_ARRAY);
3725     CvHistogram H;
3726     float data[10];
3727     CvHistogram*h = cvMakeHistHeaderForArray(1, sizes, &H, data);
3728     printf("h->type = %08x\n", h->type);
3729     printf("h->bins = %p\n", h->bins);
3730     printf("h->mat = %p\n", &(h->mat));
3731   }
3732 #endif
3733 }
3734