5 #include <opencv/cxcore.h>
7 #include <opencv/cvwimage.h>
8 #include <opencv/highgui.h>
10 #define MODULESTR "cv"
12 static PyObject *opencv_error;
35 struct cvhistogram_t {
41 struct cvmemstorage_t {
49 PyObject *container; // Containing cvmemstorage_t
55 PyObject *container; // Containing cvmemstorage_t
62 PyObject *container; // Containing cvmemstorage_t
65 struct cvsubdiv2dpoint_t {
68 PyObject *container; // Containing cvmemstorage_t
71 struct cvsubdiv2dedge_t {
74 PyObject *container; // Containing cvmemstorage_t
77 struct cvlineiterator_t {
84 struct iplconvkernel_t {
94 struct cvvideowriter_t {
99 typedef IplImage ROIplImage;
111 struct cvhaarclassifiercascade_t {
113 CvHaarClassifierCascade *a;
116 struct cvcontourtree_t {
121 struct cvpositobject_t {
131 static int is_iplimage(PyObject *o);
132 static int is_cvmat(PyObject *o);
133 static int is_cvmatnd(PyObject *o);
134 static int convert_to_CvArr(PyObject *o, CvArr **dst, const char *name = "no_name");
135 static int convert_to_IplImage(PyObject *o, IplImage **dst, const char *name = "no_name");
136 static int convert_to_CvMat(PyObject *o, CvMat **dst, const char *name = "no_name");
137 static int convert_to_CvMatND(PyObject *o, CvMatND **dst, const char *name = "no_name");
138 static PyObject *what_data(PyObject *o);
140 static void translate_error_to_exception(void)
142 PyErr_SetString(opencv_error, cvErrorStr(cvGetErrStatus()));
146 #define ERRCHK do { if (cvGetErrStatus() != 0) { translate_error_to_exception(); return NULL; } } while (0)
153 catch (const cv::Exception &e) \
155 PyErr_SetString(opencv_error, e.err.c_str()); \
161 /************************************************************************/
163 static int failmsg(const char *fmt, ...)
169 vsnprintf(str, sizeof(str), fmt, ap);
172 PyErr_SetString(PyExc_TypeError, str);
176 /************************************************************************/
178 /* These get/setters are polymorphic, used in both iplimage and cvmat */
180 static PyObject *PyObject_FromCvScalar(CvScalar s, int type)
182 int i, spe = CV_MAT_CN(type);
185 r = PyTuple_New(spe);
186 for (i = 0; i < spe; i++)
187 PyTuple_SET_ITEM(r, i, PyFloat_FromDouble(s.val[i]));
189 r = PyFloat_FromDouble(s.val[0]);
194 static PyObject *cvarr_GetItem(PyObject *o, PyObject *key);
195 static int cvarr_SetItem(PyObject *o, PyObject *key, PyObject *v);
197 // o is a Python string or buffer object. Return its size.
199 static Py_ssize_t what_size(PyObject *o)
202 Py_ssize_t buffer_len;
204 if (PyString_Check(o)) {
205 return PyString_Size(o);
206 } else if (PyObject_AsWriteBuffer(o, &buffer, &buffer_len) == 0) {
209 assert(0); // argument must be string or buffer.
215 /************************************************************************/
219 static void iplimage_dealloc(PyObject *self)
221 iplimage_t *pc = (iplimage_t*)self;
222 cvReleaseImageHeader((IplImage**)&pc->a);
227 static PyObject *iplimage_repr(PyObject *self)
229 iplimage_t *cva = (iplimage_t*)self;
230 IplImage* ipl = (IplImage*)(cva->a);
232 sprintf(str, "<iplimage(");
233 char *d = str + strlen(str);
234 sprintf(d, "nChannels=%d ", ipl->nChannels);
236 sprintf(d, "width=%d ", ipl->width);
238 sprintf(d, "height=%d ", ipl->height);
240 sprintf(d, "widthStep=%d ", ipl->widthStep);
243 return PyString_FromString(str);
246 static PyObject *iplimage_tostring(PyObject *self, PyObject *args)
248 iplimage_t *pc = (iplimage_t*)self;
250 if (!convert_to_IplImage(self, &i, "self"))
272 return (PyObject*)failmsg("Unrecognised depth %d", i->depth);
274 int bpl = i->width * i->nChannels * bps;
275 if (bpl == i->widthStep && pc->offset == 0 && ((bpl * i->height) == what_size(pc->data))) {
279 int l = bpl * i->height;
280 char *s = new char[l];
282 for (y = 0; y < i->height; y++) {
283 memcpy(s + y * bpl, i->imageData + y * i->widthStep, bpl);
285 PyObject *r = PyString_FromStringAndSize(s, l);
291 static struct PyMethodDef iplimage_methods[] =
293 {"tostring", iplimage_tostring, METH_VARARGS},
297 static PyObject *iplimage_getnChannels(iplimage_t *cva)
299 return PyInt_FromLong(((IplImage*)(cva->a))->nChannels);
301 static PyObject *iplimage_getwidth(iplimage_t *cva)
303 return PyInt_FromLong(((IplImage*)(cva->a))->width);
305 static PyObject *iplimage_getheight(iplimage_t *cva)
307 return PyInt_FromLong(((IplImage*)(cva->a))->height);
309 static PyObject *iplimage_getdepth(iplimage_t *cva)
311 return PyInt_FromLong(((IplImage*)(cva->a))->depth);
313 static PyObject *iplimage_getorigin(iplimage_t *cva)
315 return PyInt_FromLong(((IplImage*)(cva->a))->origin);
317 static void iplimage_setorigin(iplimage_t *cva, PyObject *v)
319 ((IplImage*)(cva->a))->origin = PyInt_AsLong(v);
322 static PyGetSetDef iplimage_getseters[] = {
323 {(char*)"nChannels", (getter)iplimage_getnChannels, (setter)NULL, (char*)"nChannels", NULL},
324 {(char*)"width", (getter)iplimage_getwidth, (setter)NULL, (char*)"width", NULL},
325 {(char*)"height", (getter)iplimage_getheight, (setter)NULL, (char*)"height", NULL},
326 {(char*)"depth", (getter)iplimage_getdepth, (setter)NULL, (char*)"depth", NULL},
327 {(char*)"origin", (getter)iplimage_getorigin, (setter)iplimage_setorigin, (char*)"origin", NULL},
328 {NULL} /* Sentinel */
331 static PyMappingMethods iplimage_as_map = {
337 static PyTypeObject iplimage_Type = {
338 PyObject_HEAD_INIT(&PyType_Type)
340 MODULESTR".iplimage", /*name*/
341 sizeof(iplimage_t), /*basicsize*/
344 static void iplimage_specials(void)
346 iplimage_Type.tp_dealloc = iplimage_dealloc;
347 iplimage_Type.tp_as_mapping = &iplimage_as_map;
348 iplimage_Type.tp_repr = iplimage_repr;
349 iplimage_Type.tp_methods = iplimage_methods;
350 iplimage_Type.tp_getset = iplimage_getseters;
353 static int is_iplimage(PyObject *o)
355 return PyType_IsSubtype(o->ob_type, &iplimage_Type);
358 /************************************************************************/
362 static void cvmat_dealloc(PyObject *self)
364 cvmat_t *pc = (cvmat_t*)self;
370 static PyObject *cvmat_repr(PyObject *self)
372 CvMat *m = ((cvmat_t*)self)->a;
374 sprintf(str, "<cvmat(");
375 char *d = str + strlen(str);
376 sprintf(d, "type=%08x ", m->type);
378 sprintf(d, "rows=%d ", m->rows);
380 sprintf(d, "cols=%d ", m->cols);
382 sprintf(d, "step=%d ", m->step);
385 return PyString_FromString(str);
388 static PyObject *cvmat_tostring(PyObject *self, PyObject *args)
391 if (!convert_to_CvMat(self, &m, "self"))
394 int bps; // bytes per sample
396 switch (CV_MAT_DEPTH(m->type)) {
399 bps = CV_MAT_CN(m->type) * 1;
403 bps = CV_MAT_CN(m->type) * 2;
407 bps = CV_MAT_CN(m->type) * 4;
410 bps = CV_MAT_CN(m->type) * 8;
413 return (PyObject*)failmsg("Unrecognised depth %d", CV_MAT_DEPTH(m->type));
416 int bpl = m->cols * bps; // bytes per line
417 cvmat_t *pc = (cvmat_t*)self;
418 if (bpl == m->step && pc->offset == 0 && ((bpl * m->rows) == what_size(pc->data))) {
422 int l = bpl * m->rows;
423 char *s = new char[l];
425 for (y = 0; y < m->rows; y++) {
426 memcpy(s + y * bpl, m->data.ptr + y * m->step, bpl);
428 PyObject *r = PyString_FromStringAndSize(s, l);
434 static struct PyMethodDef cvmat_methods[] =
436 {"tostring", cvmat_tostring, METH_VARARGS},
440 static PyObject *cvmat_gettype(cvmat_t *cva)
442 return PyInt_FromLong(cva->a->type);
445 static PyObject *cvmat_getstep(cvmat_t *cva)
447 return PyInt_FromLong(cva->a->step);
450 static PyObject *cvmat_getrows(cvmat_t *cva)
452 return PyInt_FromLong(cva->a->rows);
455 static PyObject *cvmat_getcols(cvmat_t *cva)
457 return PyInt_FromLong(cva->a->cols);
460 static PyGetSetDef cvmat_getseters[] = {
461 {(char*)"type", (getter)cvmat_gettype, (setter)NULL, (char*)"type", NULL},
462 {(char*)"step", (getter)cvmat_getstep, (setter)NULL, (char*)"step", NULL},
463 {(char*)"rows", (getter)cvmat_getrows, (setter)NULL, (char*)"rows", NULL},
464 {(char*)"cols", (getter)cvmat_getcols, (setter)NULL, (char*)"cols", NULL},
465 {(char*)"width", (getter)cvmat_getcols, (setter)NULL, (char*)"width", NULL},
466 {(char*)"height", (getter)cvmat_getrows, (setter)NULL, (char*)"height", NULL},
467 {NULL} /* Sentinel */
470 static PyMappingMethods cvmat_as_map = {
476 static PyTypeObject cvmat_Type = {
477 PyObject_HEAD_INIT(&PyType_Type)
479 MODULESTR".cvmat", /*name*/
480 sizeof(cvmat_t), /*basicsize*/
483 static void cvmat_specials(void)
485 cvmat_Type.tp_dealloc = cvmat_dealloc;
486 cvmat_Type.tp_as_mapping = &cvmat_as_map;
487 cvmat_Type.tp_repr = cvmat_repr;
488 cvmat_Type.tp_methods = cvmat_methods;
489 cvmat_Type.tp_getset = cvmat_getseters;
492 static int is_cvmat(PyObject *o)
494 return PyType_IsSubtype(o->ob_type, &cvmat_Type);
497 /************************************************************************/
501 static void cvmatnd_dealloc(PyObject *self)
503 cvmatnd_t *pc = (cvmatnd_t*)self;
509 static PyObject *cvmatnd_repr(PyObject *self)
511 CvMatND *m = ((cvmatnd_t*)self)->a;
513 sprintf(str, "<cvmatnd(");
514 char *d = str + strlen(str);
515 sprintf(d, "type=%08x ", m->type);
518 return PyString_FromString(str);
521 static PyObject *cvmatnd_tostring(PyObject *self, PyObject *args)
524 if (!convert_to_CvMatND(self, &m, "self"))
528 switch (CV_MAT_DEPTH(m->type)) {
531 bps = CV_MAT_CN(m->type) * 1;
535 bps = CV_MAT_CN(m->type) * 2;
539 bps = CV_MAT_CN(m->type) * 4;
542 bps = CV_MAT_CN(m->type) * 8;
545 return (PyObject*)failmsg("Unrecognised depth %d", CV_MAT_DEPTH(m->type));
549 for (int d = 0; d < m->dims; d++) {
554 for (d = 0; d < m->dims; d++) {
557 int rowsize = m->dim[m->dims-1].size * bps;
558 char *s = new char[l];
563 memcpy(ps, cvPtrND(m, i), rowsize);
565 for (d = m->dims - 2; 0 <= d; d--) {
566 if (++i[d] < cvGetDimSize(m, d)) {
576 return PyString_FromStringAndSize(s, ps - s);
579 static struct PyMethodDef cvmatnd_methods[] =
581 {"tostring", cvmatnd_tostring, METH_VARARGS},
585 static PyMappingMethods cvmatnd_as_map = {
591 static PyTypeObject cvmatnd_Type = {
592 PyObject_HEAD_INIT(&PyType_Type)
594 MODULESTR".cvmatnd", /*name*/
595 sizeof(cvmatnd_t), /*basicsize*/
598 static void cvmatnd_specials(void)
600 cvmatnd_Type.tp_dealloc = cvmatnd_dealloc;
601 cvmatnd_Type.tp_as_mapping = &cvmatnd_as_map;
602 cvmatnd_Type.tp_repr = cvmatnd_repr;
603 cvmatnd_Type.tp_methods = cvmatnd_methods;
606 static int is_cvmatnd(PyObject *o)
608 return PyType_IsSubtype(o->ob_type, &cvmatnd_Type);
611 /************************************************************************/
615 static void cvhistogram_dealloc(PyObject *self)
617 cvhistogram_t *cvh = (cvhistogram_t*)self;
618 Py_DECREF(cvh->bins);
622 static PyTypeObject cvhistogram_Type = {
623 PyObject_HEAD_INIT(&PyType_Type)
625 MODULESTR".cvhistogram", /*name*/
626 sizeof(cvhistogram_t), /*basicsize*/
629 static PyObject *cvhistogram_getbins(cvhistogram_t *cvh)
631 Py_INCREF(cvh->bins);
635 static PyGetSetDef cvhistogram_getseters[] = {
636 {(char*)"bins", (getter)cvhistogram_getbins, (setter)NULL, (char*)"bins", NULL},
637 {NULL} /* Sentinel */
640 static void cvhistogram_specials(void)
642 cvhistogram_Type.tp_dealloc = cvhistogram_dealloc;
643 cvhistogram_Type.tp_getset = cvhistogram_getseters;
646 /************************************************************************/
650 static PyObject *cvlineiterator_iter(PyObject *o)
656 static PyObject *cvlineiterator_next(PyObject *o)
658 cvlineiterator_t *pi = (cvlineiterator_t*)o;
664 cvRawDataToScalar( (void*)(pi->iter.ptr), pi->type, &r);
665 PyObject *pr = PyObject_FromCvScalar(r, pi->type);
667 CV_NEXT_LINE_POINT(pi->iter);
675 static PyTypeObject cvlineiterator_Type = {
676 PyObject_HEAD_INIT(&PyType_Type)
678 MODULESTR".cvlineiterator", /*name*/
679 sizeof(cvlineiterator_t), /*basicsize*/
682 static void cvlineiterator_specials(void)
684 cvlineiterator_Type.tp_iter = cvlineiterator_iter;
685 cvlineiterator_Type.tp_iternext = cvlineiterator_next;
688 /************************************************************************/
692 static void iplconvkernel_dealloc(PyObject *self)
694 iplconvkernel_t *pi = (iplconvkernel_t*)self;
695 cvReleaseStructuringElement(&(pi->a));
699 static PyTypeObject iplconvkernel_Type = {
700 PyObject_HEAD_INIT(&PyType_Type)
702 MODULESTR".iplconvkernel", /*name*/
703 sizeof(iplconvkernel_t), /*basicsize*/
706 static void iplconvkernel_specials(void)
708 iplconvkernel_Type.tp_dealloc = iplconvkernel_dealloc;
711 /************************************************************************/
715 static void cvcapture_dealloc(PyObject *self)
717 cvcapture_t *pi = (cvcapture_t*)self;
718 cvReleaseCapture(&(pi->a));
722 static PyTypeObject cvcapture_Type = {
723 PyObject_HEAD_INIT(&PyType_Type)
725 MODULESTR".cvcapture", /*name*/
726 sizeof(cvcapture_t), /*basicsize*/
729 static void cvcapture_specials(void)
731 cvcapture_Type.tp_dealloc = cvcapture_dealloc;
734 /************************************************************************/
738 static void cvvideowriter_dealloc(PyObject *self)
740 cvvideowriter_t *pi = (cvvideowriter_t*)self;
741 cvReleaseVideoWriter(&(pi->a));
745 static PyTypeObject cvvideowriter_Type = {
746 PyObject_HEAD_INIT(&PyType_Type)
748 MODULESTR".cvvideowriter", /*name*/
749 sizeof(cvvideowriter_t), /*basicsize*/
752 static void cvvideowriter_specials(void)
754 cvvideowriter_Type.tp_dealloc = cvvideowriter_dealloc;
759 /************************************************************************/
763 static PyTypeObject cvmoments_Type = {
764 PyObject_HEAD_INIT(&PyType_Type)
766 MODULESTR".cvmoments", /*name*/
767 sizeof(cvmoments_t), /*basicsize*/
770 static void cvmoments_specials(void)
774 /************************************************************************/
778 static void cvmemstorage_dealloc(PyObject *self)
780 cvmemstorage_t *ps = (cvmemstorage_t*)self;
781 cvReleaseMemStorage(&(ps->a));
785 static PyTypeObject cvmemstorage_Type = {
786 PyObject_HEAD_INIT(&PyType_Type)
788 MODULESTR".cvmemstorage", /*name*/
789 sizeof(cvmemstorage_t), /*basicsize*/
792 static void cvmemstorage_specials(void)
794 cvmemstorage_Type.tp_dealloc = cvmemstorage_dealloc;
797 /************************************************************************/
801 static PyTypeObject cvfont_Type = {
802 PyObject_HEAD_INIT(&PyType_Type)
804 MODULESTR".cvfont", /*name*/
805 sizeof(cvfont_t), /*basicsize*/
808 static void cvfont_specials(void) { }
810 /************************************************************************/
814 static void cvpositobject_dealloc(PyObject *self)
816 cvpositobject_t *pi = (cvpositobject_t*)self;
817 cvReleasePOSITObject(&(pi->a));
821 static PyTypeObject cvpositobject_Type = {
822 PyObject_HEAD_INIT(&PyType_Type)
824 MODULESTR".cvpositobject", /*name*/
825 sizeof(cvpositobject_t), /*basicsize*/
828 static void cvpositobject_specials(void)
830 cvpositobject_Type.tp_dealloc = cvpositobject_dealloc;
833 /************************************************************************/
837 static PyTypeObject cvrng_Type = {
838 PyObject_HEAD_INIT(&PyType_Type)
840 MODULESTR".cvrng", /*name*/
841 sizeof(cvrng_t), /*basicsize*/
844 static void cvrng_specials(void)
848 /************************************************************************/
850 /* cvhaarclassifiercascade */
852 static PyTypeObject cvhaarclassifiercascade_Type = {
853 PyObject_HEAD_INIT(&PyType_Type)
855 MODULESTR".cvhaarclassifiercascade", /*name*/
856 sizeof(cvhaarclassifiercascade_t), /*basicsize*/
859 static void cvhaarclassifiercascade_specials(void) { }
861 /************************************************************************/
865 static PyTypeObject cvcontourtree_Type = {
866 PyObject_HEAD_INIT(&PyType_Type)
868 MODULESTR".cvcontourtree", /*name*/
869 sizeof(cvcontourtree_t), /*basicsize*/
872 static void cvcontourtree_specials(void) { }
875 /************************************************************************/
879 static PyTypeObject cvsubdiv2dedge_Type = {
880 PyObject_HEAD_INIT(&PyType_Type)
882 MODULESTR".cvsubdiv2dedge", /*name*/
883 sizeof(cvsubdiv2dedge_t), /*basicsize*/
886 static int cvsubdiv2dedge_compare(PyObject *o1, PyObject *o2)
888 cvsubdiv2dedge_t *e1 = (cvsubdiv2dedge_t*)o1;
889 cvsubdiv2dedge_t *e2 = (cvsubdiv2dedge_t*)o2;
892 else if (e1->a > e2->a)
898 static PyObject *cvquadedge_repr(PyObject *self)
900 CvSubdiv2DEdge m = ((cvsubdiv2dedge_t*)self)->a;
902 sprintf(str, "<cvsubdiv2dedge(");
903 char *d = str + strlen(str);
904 sprintf(d, "%zx.%d", m & ~3, (int)(m & 3));
907 return PyString_FromString(str);
910 static void cvsubdiv2dedge_specials(void) {
911 cvsubdiv2dedge_Type.tp_compare = cvsubdiv2dedge_compare;
912 cvsubdiv2dedge_Type.tp_repr = cvquadedge_repr;
915 /************************************************************************/
919 static void cvseq_dealloc(PyObject *self)
921 cvseq_t *ps = (cvseq_t*)self;
922 Py_DECREF(ps->container);
926 static PyObject *cvseq_h_next(PyObject *self, PyObject *args);
927 static PyObject *cvseq_h_prev(PyObject *self, PyObject *args);
928 static PyObject *cvseq_v_next(PyObject *self, PyObject *args);
929 static PyObject *cvseq_v_prev(PyObject *self, PyObject *args);
931 static struct PyMethodDef cvseq_methods[] =
933 {"h_next", cvseq_h_next, METH_VARARGS},
934 {"h_prev", cvseq_h_prev, METH_VARARGS},
935 {"v_next", cvseq_v_next, METH_VARARGS},
936 {"v_prev", cvseq_v_prev, METH_VARARGS},
940 static Py_ssize_t cvseq_seq_length(PyObject *o)
942 cvseq_t *ps = (cvseq_t*)o;
943 return (Py_ssize_t)(ps->a->total);
946 static PyObject* cvseq_seq_getitem(PyObject *o, Py_ssize_t i)
948 cvseq_t *ps = (cvseq_t*)o;
956 if (i < (Py_ssize_t)(ps->a->total)) {
957 switch (CV_SEQ_ELTYPE(ps->a)) {
959 case CV_SEQ_ELTYPE_POINT:
960 pt = CV_GET_SEQ_ELEM(CvPoint, ps->a, i);
961 return Py_BuildValue("ii", pt->x, pt->y);
963 case CV_SEQ_ELTYPE_GENERIC:
964 switch (ps->a->elem_size) {
965 case sizeof(CvQuadEdge2D):
967 cvsubdiv2dedge_t *r = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
968 r->a = (CvSubdiv2DEdge)CV_GET_SEQ_ELEM(CvQuadEdge2D, ps->a, i);
969 r->container = ps->container;
970 Py_INCREF(r->container);
974 printf("seq elem size is %d\n", ps->a->elem_size);
975 printf("KIND %d\n", CV_SEQ_KIND(ps->a));
978 return PyInt_FromLong(*CV_GET_SEQ_ELEM(unsigned char, ps->a, i));
980 case CV_SEQ_ELTYPE_PTR:
981 case CV_SEQ_ELTYPE_INDEX:
982 return PyInt_FromLong(*CV_GET_SEQ_ELEM(int, ps->a, i));
985 pp = CV_GET_SEQ_ELEM(pointpair, ps->a, i);
986 return Py_BuildValue("(ii),(ii)", pp->a.x, pp->a.y, pp->b.x, pp->b.y);
989 pt2 = CV_GET_SEQ_ELEM(CvPoint2D32f, ps->a, i);
990 return Py_BuildValue("ff", pt2->x, pt2->y);
992 case CV_SEQ_ELTYPE_POINT3D:
993 pt3 = CV_GET_SEQ_ELEM(CvPoint3D32f, ps->a, i);
994 return Py_BuildValue("fff", pt3->x, pt3->y, pt3->z);
997 printf("Unknown element type %08x\n", CV_SEQ_ELTYPE(ps->a));
1005 static PyObject* cvseq_map_getitem(PyObject *o, PyObject *item)
1007 cvseq_t *ps = (cvseq_t*)o;
1008 if (PyInt_Check(item)) {
1009 long i = PyInt_AS_LONG(item);
1012 return cvseq_seq_getitem(o, i);
1013 } else if (PySlice_Check(item)) {
1014 Py_ssize_t start, stop, step, slicelength, cur, i;
1017 if (PySlice_GetIndicesEx((PySliceObject*)item, ps->a->total,
1018 &start, &stop, &step, &slicelength) < 0) {
1022 if (slicelength <= 0) {
1023 return PyList_New(0);
1025 result = PyList_New(slicelength);
1026 if (!result) return NULL;
1028 for (cur = start, i = 0; i < slicelength;
1030 PyList_SET_ITEM(result, i, cvseq_seq_getitem(o, cur));
1036 PyErr_SetString(PyExc_TypeError, "CvSeq indices must be integers");
1042 PySequenceMethods cvseq_sequence = {
1049 static PyMappingMethods cvseq_mapping = {
1055 static PyTypeObject cvseq_Type = {
1056 PyObject_HEAD_INIT(&PyType_Type)
1058 MODULESTR".cvseq", /*name*/
1059 sizeof(cvseq_t), /*basicsize*/
1062 static void cvseq_specials(void)
1064 cvseq_Type.tp_dealloc = cvseq_dealloc;
1065 cvseq_Type.tp_as_sequence = &cvseq_sequence;
1066 cvseq_Type.tp_as_mapping = &cvseq_mapping;
1067 cvseq_Type.tp_methods = cvseq_methods;
1070 #define MK_ACCESSOR(FIELD) \
1071 static PyObject *cvseq_##FIELD(PyObject *self, PyObject *args) \
1073 cvseq_t *ps = (cvseq_t*)self; \
1075 if (s->FIELD == NULL) { \
1078 cvseq_t *r = PyObject_NEW(cvseq_t, &cvseq_Type); \
1080 r->container = ps->container; \
1081 Py_INCREF(r->container); \
1082 return (PyObject*)r; \
1092 /************************************************************************/
1096 static void cvset_dealloc(PyObject *self)
1098 cvset_t *ps = (cvset_t*)self;
1099 Py_DECREF(ps->container);
1103 static PyTypeObject cvset_Type = {
1104 PyObject_HEAD_INIT(&PyType_Type)
1106 MODULESTR".cvset", /*name*/
1107 sizeof(cvset_t), /*basicsize*/
1110 static PyObject *cvset_iter(PyObject *o)
1113 cvset_t *ps = (cvset_t*)o;
1118 static PyObject *cvset_next(PyObject *o)
1120 cvset_t *ps = (cvset_t*)o;
1122 while (ps->i < ps->a->total) {
1123 CvSetElem *e = cvGetSetElem(ps->a, ps->i);
1124 int prev_i = ps->i++;
1126 return cvseq_seq_getitem(o, prev_i);
1132 static void cvset_specials(void)
1134 cvset_Type.tp_dealloc = cvset_dealloc;
1135 cvset_Type.tp_iter = cvset_iter;
1136 cvset_Type.tp_iternext = cvset_next;
1139 /************************************************************************/
1143 static PyTypeObject cvsubdiv2d_Type = {
1144 PyObject_HEAD_INIT(&PyType_Type)
1146 MODULESTR".cvsubdiv2d", /*name*/
1147 sizeof(cvsubdiv2d_t), /*basicsize*/
1150 static PyObject *cvsubdiv2d_getattro(PyObject *o, PyObject *name)
1152 cvsubdiv2d_t *p = (cvsubdiv2d_t*)o;
1153 if (strcmp(PyString_AsString(name), "edges") == 0) {
1154 cvset_t *r = PyObject_NEW(cvset_t, &cvset_Type);
1156 r->container = p->container;
1157 Py_INCREF(r->container);
1158 return (PyObject*)r;
1160 PyErr_SetString(PyExc_TypeError, "cvsubdiv2d has no such attribute");
1165 static void cvsubdiv2d_specials(void)
1167 cvsubdiv2d_Type.tp_getattro = cvsubdiv2d_getattro;
1170 /************************************************************************/
1172 /* cvsubdiv2dpoint */
1174 static PyTypeObject cvsubdiv2dpoint_Type = {
1175 PyObject_HEAD_INIT(&PyType_Type)
1177 MODULESTR".cvsubdiv2dpoint", /*name*/
1178 sizeof(cvsubdiv2dpoint_t), /*basicsize*/
1181 static PyObject *cvsubdiv2dpoint_getattro(PyObject *o, PyObject *name)
1183 cvsubdiv2dpoint_t *p = (cvsubdiv2dpoint_t*)o;
1184 if (strcmp(PyString_AsString(name), "first") == 0) {
1185 cvsubdiv2dedge_t *r = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
1187 r->container = p->container;
1188 Py_INCREF(r->container);
1189 return (PyObject*)r;
1190 } else if (strcmp(PyString_AsString(name), "pt") == 0) {
1191 return Py_BuildValue("(ff)", p->a->pt.x, p->a->pt.y);
1193 PyErr_SetString(PyExc_TypeError, "cvsubdiv2dpoint has no such attribute");
1198 static void cvsubdiv2dpoint_specials(void)
1200 cvsubdiv2dpoint_Type.tp_getattro = cvsubdiv2dpoint_getattro;
1203 /************************************************************************/
1204 /* convert_to_X: used after PyArg_ParseTuple in the generated code */
1206 static int convert_to_char(PyObject *o, char *dst, const char *name = "no_name")
1208 if (PyString_Check(o) && PyString_Size(o) == 1) {
1209 *dst = PyString_AsString(o)[0];
1213 return failmsg("Expected single character string for argument '%s'", name);
1217 static int convert_to_CvMemStorage(PyObject *o, CvMemStorage **dst, const char *name = "no_name")
1219 if (PyType_IsSubtype(o->ob_type, &cvmemstorage_Type)) {
1220 (*dst) = (((cvmemstorage_t*)o)->a);
1223 (*dst) = (CvMemStorage*)NULL;
1224 return failmsg("Expected CvMemStorage for argument '%s'", name);
1228 static int convert_to_CvSeq(PyObject *o, CvSeq **dst, const char *name = "no_name")
1230 if (PyType_IsSubtype(o->ob_type, &cvseq_Type)) {
1231 (*dst) = (((cvseq_t*)o)->a);
1234 (*dst) = (CvSeq*)NULL;
1235 return failmsg("Expected CvSeq for argument '%s'", name);
1239 static int convert_to_CvSize(PyObject *o, CvSize *dst, const char *name = "no_name")
1241 if (!PyArg_ParseTuple(o, "ii", &dst->width, &dst->height))
1242 return failmsg("CvSize argument '%s' expects two integers", name);
1247 static int convert_to_CvScalar(PyObject *o, CvScalar *s, const char *name = "no_name")
1249 if (PySequence_Check(o)) {
1250 PyObject *fi = PySequence_Fast(o, name);
1253 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1254 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1255 if (PyNumber_Check(item))
1256 s->val[i] = PyFloat_AsDouble(item);
1258 return failmsg("CvScalar value for argument '%s' is not numeric", name);
1262 if (PyNumber_Check(o)) {
1263 s->val[0] = PyFloat_AsDouble(o);
1265 return failmsg("CvScalar value for argument '%s' is not numeric", name);
1271 static int convert_to_CvPointPTR(PyObject *o, CvPoint **p, const char *name = "no_name")
1273 if (!PySequence_Check(o))
1274 return failmsg("Expected sequence for point list argument '%s'", name);
1275 PyObject *fi = PySequence_Fast(o, name);
1278 *p = new CvPoint[PySequence_Fast_GET_SIZE(fi)];
1279 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1280 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1281 if (!PyTuple_Check(item))
1282 return failmsg("Expected tuple for element in point list argument '%s'", name);
1283 if (!PyArg_ParseTuple(item, "ii", &((*p)[i].x), &((*p)[i].y))) {
1291 static int convert_to_CvPoint2D32fPTR(PyObject *o, CvPoint2D32f **p, const char *name = "no_name")
1293 PyObject *fi = PySequence_Fast(o, name);
1296 *p = new CvPoint2D32f[PySequence_Fast_GET_SIZE(fi)];
1297 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1298 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1299 if (!PyTuple_Check(item))
1300 return failmsg("Expected tuple for CvPoint2D32f argument '%s'", name);
1301 if (!PyArg_ParseTuple(item, "ff", &((*p)[i].x), &((*p)[i].y))) {
1309 static int convert_to_CvPoint3D32fPTR(PyObject *o, CvPoint3D32f **p, const char *name = "no_name")
1311 PyObject *fi = PySequence_Fast(o, name);
1314 *p = new CvPoint3D32f[PySequence_Fast_GET_SIZE(fi)];
1315 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1316 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1317 if (!PyTuple_Check(item))
1318 return failmsg("Expected tuple for CvPoint3D32f argument '%s'", name);
1319 if (!PyArg_ParseTuple(item, "fff", &((*p)[i].x), &((*p)[i].y), &((*p)[i].z))) {
1327 static int convert_to_CvStarDetectorParams(PyObject *o, CvStarDetectorParams *dst, const char *name = "no_name")
1329 if (!PyArg_ParseTuple(o,
1332 &dst->responseThreshold,
1333 &dst->lineThresholdProjected,
1334 &dst->lineThresholdBinarized,
1335 &dst->suppressNonmaxSize))
1336 return failmsg("CvRect argument '%s' expects four integers", name);
1341 static int convert_to_CvRect(PyObject *o, CvRect *dst, const char *name = "no_name")
1343 if (!PyArg_ParseTuple(o, "iiii", &dst->x, &dst->y, &dst->width, &dst->height))
1344 return failmsg("CvRect argument '%s' expects four integers", name);
1349 static int convert_to_CvRectPTR(PyObject *o, CvRect **dst, const char *name = "no_name")
1352 if (!PyArg_ParseTuple(o, "iiii", &(*dst)->x, &(*dst)->y, &(*dst)->width, &(*dst)->height))
1353 return failmsg("CvRect argument '%s' expects four integers", name);
1358 static int convert_to_CvSlice(PyObject *o, CvSlice *dst, const char *name = "no_name")
1360 if (!PyArg_ParseTuple(o, "ii", &dst->start_index, &dst->end_index))
1361 return failmsg("CvSlice argument '%s' expects two integers", name);
1366 static int convert_to_CvPoint(PyObject *o, CvPoint *dst, const char *name = "no_name")
1368 if (!PyArg_ParseTuple(o, "ii", &dst->x, &dst->y))
1369 return failmsg("CvPoint argument '%s' expects two integers", name);
1374 static int convert_to_CvPoint2D32f(PyObject *o, CvPoint2D32f *dst, const char *name = "no_name")
1376 if (!PyArg_ParseTuple(o, "ff", &dst->x, &dst->y))
1377 return failmsg("CvPoint2D32f argument '%s' expects two floats", name);
1382 static int convert_to_CvPoint3D32f(PyObject *o, CvPoint3D32f *dst, const char *name = "no_name")
1384 if (!PyArg_ParseTuple(o, "fff", &dst->x, &dst->y, &dst->z))
1385 return failmsg("CvPoint3D32f argument '%s' expects three floats", name);
1390 static int convert_to_IplImage(PyObject *o, IplImage **dst, const char *name)
1392 iplimage_t *ipl = (iplimage_t*)o;
1394 Py_ssize_t buffer_len;
1396 if (!is_iplimage(o)) {
1397 return failmsg("Argument '%s' must be IplImage", name);
1398 } else if (PyString_Check(ipl->data)) {
1399 cvSetData(ipl->a, PyString_AsString(ipl->data) + ipl->offset, ipl->a->widthStep);
1400 assert(cvGetErrStatus() == 0);
1403 } else if (ipl->data && PyObject_AsWriteBuffer(ipl->data, &buffer, &buffer_len) == 0) {
1404 cvSetData(ipl->a, (void*)((char*)buffer + ipl->offset), ipl->a->widthStep);
1405 assert(cvGetErrStatus() == 0);
1409 return failmsg("IplImage argument '%s' has no data", name);
1413 static int convert_to_CvMat(PyObject *o, CvMat **dst, const char *name)
1415 cvmat_t *m = (cvmat_t*)o;
1417 Py_ssize_t buffer_len;
1420 return failmsg("Argument '%s' must be CvMat", name);
1421 } else if (m->data && PyString_Check(m->data)) {
1422 assert(cvGetErrStatus() == 0);
1423 cvSetData(m->a, PyString_AsString(m->data) + m->offset, m->a->step);
1424 assert(cvGetErrStatus() == 0);
1427 } else if (m->data && PyObject_AsWriteBuffer(m->data, &buffer, &buffer_len) == 0) {
1428 cvSetData(m->a, (void*)((char*)buffer + m->offset), m->a->step);
1429 assert(cvGetErrStatus() == 0);
1433 return failmsg("CvMat argument '%s' has no data", name);
1437 static int convert_to_CvMatND(PyObject *o, CvMatND **dst, const char *name)
1439 cvmatnd_t *m = (cvmatnd_t*)o;
1441 Py_ssize_t buffer_len;
1443 if (!is_cvmatnd(o)) {
1444 return failmsg("Argument '%s' must be CvMatND", name);
1445 } else if (m->data && PyString_Check(m->data)) {
1446 m->a->data.ptr = ((uchar*)PyString_AsString(m->data)) + m->offset;
1449 } else if (m->data && PyObject_AsWriteBuffer(m->data, &buffer, &buffer_len) == 0) {
1450 m->a->data.ptr = ((uchar*)buffer + m->offset);
1454 return failmsg("CvMatND argument '%s' has no data", name);
1458 static int convert_to_CvArr(PyObject *o, CvArr **dst, const char *name)
1463 } else if (is_iplimage(o)) {
1464 return convert_to_IplImage(o, (IplImage**)dst, name);
1465 } else if (is_cvmat(o)) {
1466 return convert_to_CvMat(o, (CvMat**)dst, name);
1467 } else if (is_cvmatnd(o)) {
1468 return convert_to_CvMatND(o, (CvMatND**)dst, name);
1470 return failmsg("CvArr argument '%s' must be IplImage, CvMat or CvMatND", name);
1474 static int convert_to_CvHistogram(PyObject *o, CvHistogram **dst, const char *name = "no_name")
1476 if (PyType_IsSubtype(o->ob_type, &cvhistogram_Type)) {
1477 cvhistogram_t *ht = (cvhistogram_t*)o;
1479 return convert_to_CvArr(ht->bins, &(ht->h.bins), "bins");
1481 *dst = (CvHistogram *)NULL;
1482 return failmsg("Expected CvHistogram for argument '%s'", name);
1486 // Used by FillPoly, FillConvexPoly, PolyLine
1487 struct pts_npts_contours {
1493 static int convert_to_pts_npts_contours(PyObject *o, pts_npts_contours *dst, const char *name = "no_name")
1495 PyObject *fi = PySequence_Fast(o, name);
1498 dst->contours = PySequence_Fast_GET_SIZE(fi);
1499 dst->pts = new CvPoint*[dst->contours];
1500 dst->npts = new int[dst->contours];
1501 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1502 if (!convert_to_CvPointPTR(PySequence_Fast_GET_ITEM(fi, i), &dst->pts[i], name))
1504 dst->npts[i] = PySequence_Size(PySequence_Fast_GET_ITEM(fi, i)); // safe because convert_ just succeeded
1514 static int convert_to_cvarrseq(PyObject *o, cvarrseq *dst, const char *name = "no_name")
1516 if (PyType_IsSubtype(o->ob_type, &cvseq_Type)) {
1517 return convert_to_CvSeq(o, (CvSeq**)&(dst->v), name);
1518 } else if (PySequence_Check(o)) {
1519 PyObject *fi = PySequence_Fast(o, name);
1522 Py_ssize_t size = -1;
1523 // Make a pass through the sequence, checking that each element is
1524 // a sequence and that they are all the same size
1525 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1526 PyObject *e = PySequence_Fast_GET_ITEM(fi, i);
1528 if (!PySequence_Check(e))
1529 return failmsg("Sequence '%s' must contain sequences", name);
1531 size = (int)PySequence_Size(e);
1532 else if (size != PySequence_Size(e))
1533 return failmsg("All elements of sequence '%s' must be same size", name);
1536 CvMat *mt = cvCreateMat((int)PySequence_Fast_GET_SIZE(fi), 1, CV_32SC(size));
1537 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1538 PyObject *e = PySequence_Fast_GET_ITEM(fi, i);
1539 PyObject *fe = PySequence_Fast(e, name);
1541 int *pdst = (int*)cvPtr2D(mt, i, 0);
1542 for (Py_ssize_t j = 0; j < size; j++) {
1543 *pdst++ = PyInt_AsLong(PySequence_Fast_GET_ITEM(fe, j));
1551 return convert_to_CvArr(o, (CvArr**)&(dst->v), name);
1555 struct cvarr_count {
1560 static int convert_to_cvarr_count(PyObject *o, cvarr_count *dst, const char *name = "no_name")
1562 PyObject *fi = PySequence_Fast(o, name);
1565 dst->count = PySequence_Fast_GET_SIZE(fi);
1566 dst->cvarr = new CvArr*[dst->count];
1567 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1568 if (!convert_to_CvArr(PySequence_Fast_GET_ITEM(fi, i), &dst->cvarr[i], name))
1581 static int convert_to_intpair(PyObject *o, intpair *dst, const char *name = "no_name")
1583 PyObject *fi = PySequence_Fast(o, name);
1586 dst->count = PySequence_Fast_GET_SIZE(fi);
1587 dst->pairs = new int[2 * dst->count];
1588 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1589 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1590 if (!PyArg_ParseTuple(item, "ii", &dst->pairs[2 * i], &dst->pairs[2 * i + 1])) {
1598 struct cvpoint2d32f_count {
1599 CvPoint2D32f* points;
1603 static int convert_to_cvpoint2d32f_count(PyObject *o, cvpoint2d32f_count *dst, const char *name = "no_name")
1605 if (PyInt_Check(o)) {
1606 dst->count = PyInt_AsLong(o);
1607 dst->points = new CvPoint2D32f[dst->count];
1610 return failmsg("Expected integer for CvPoint2D32f count");
1618 static int convert_to_floats(PyObject *o, floats *dst, const char *name = "no_name")
1620 PyObject *fi = PySequence_Fast(o, name);
1623 dst->count = PySequence_Fast_GET_SIZE(fi);
1624 dst->f = new float[dst->count];
1625 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1626 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1627 dst->f[i] = (float)PyFloat_AsDouble(item);
1637 /// convert_to_chars not used
1643 static int convert_to_CvPoints(PyObject *o, CvPoints *dst, const char *name = "no_name")
1645 PyObject *fi = PySequence_Fast(o, name);
1648 dst->count = PySequence_Fast_GET_SIZE(fi);
1649 dst->p = new CvPoint[dst->count];
1650 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1651 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1652 convert_to_CvPoint(item, &dst->p[i], name);
1658 struct CvPoint3D32fs {
1662 static int convert_to_CvPoint3D32fs(PyObject *o, CvPoint3D32fs *dst, const char *name = "no_name")
1664 PyObject *fi = PySequence_Fast(o, name);
1667 dst->count = PySequence_Fast_GET_SIZE(fi);
1668 dst->p = new CvPoint3D32f[dst->count];
1669 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1670 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1671 convert_to_CvPoint3D32f(item, &dst->p[i], name);
1677 struct CvPoint2D32fs {
1681 static int convert_to_CvPoint2D32fs(PyObject *o, CvPoint2D32fs *dst, const char *name = "no_name")
1683 PyObject *fi = PySequence_Fast(o, name);
1686 dst->count = PySequence_Fast_GET_SIZE(fi);
1687 dst->p = new CvPoint2D32f[dst->count];
1688 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1689 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1690 convert_to_CvPoint2D32f(item, &dst->p[i], name);
1700 static int convert_to_ints(PyObject *o, ints *dst, const char *name = "no_name")
1702 PyObject *fi = PySequence_Fast(o, name);
1705 dst->count = PySequence_Fast_GET_SIZE(fi);
1706 dst->i = new int[dst->count];
1707 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1708 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1709 dst->i[i] = PyInt_AsLong(item);
1719 static int convert_to_ints0(PyObject *o, ints0 *dst, const char *name = "no_name")
1721 PyObject *fi = PySequence_Fast(o, name);
1724 dst->count = PySequence_Fast_GET_SIZE(fi);
1725 dst->i = new int[dst->count + 1];
1726 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1727 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1728 dst->i[i] = PyInt_AsLong(item);
1730 dst->i[dst->count] = 0;
1739 int step[CV_MAX_DIM];
1740 int length[CV_MAX_DIM];
1743 static int convert_to_dim(PyObject *item, int i, dims *dst, CvArr *cva, const char *name = "no_name")
1745 if (PySlice_Check(item)) {
1746 Py_ssize_t start, stop, step, slicelength;
1747 PySlice_GetIndicesEx((PySliceObject*)item, cvGetDimSize(cva, i), &start, &stop, &step, &slicelength);
1749 dst->step[i] = step;
1750 dst->length[i] = slicelength;
1752 int index = PyInt_AsLong(item);
1756 dst->i[i] = cvGetDimSize(cva, i) + index;
1763 static int convert_to_dims(PyObject *o, dims *dst, CvArr *cva, const char *name = "no_name")
1765 if (!PyTuple_Check(o)) {
1767 return convert_to_dim(o, 0, dst, cva, name);
1769 PyObject *fi = PySequence_Fast(o, name);
1771 PyErr_SetString(PyExc_TypeError, "Expected tuple for index");
1774 dst->count = PySequence_Fast_GET_SIZE(fi);
1775 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1776 if (i >= cvGetDims(cva)) {
1777 return failmsg("Access specifies %d dimensions, but array only has %d", PySequence_Fast_GET_SIZE(fi), cvGetDims(cva));
1779 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1780 if (!convert_to_dim(item, i, dst, cva, name))
1792 static int convert_to_IplImages(PyObject *o, IplImages *dst, const char *name = "no_name")
1794 PyObject *fi = PySequence_Fast(o, name);
1797 dst->count = PySequence_Fast_GET_SIZE(fi);
1798 dst->ims = new IplImage*[dst->count];
1799 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1800 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1801 if (!convert_to_IplImage(item, &dst->ims[i]))
1812 static int convert_to_CvArrs(PyObject *o, CvArrs *dst, const char *name = "no_name")
1814 PyObject *fi = PySequence_Fast(o, name);
1817 dst->count = PySequence_Fast_GET_SIZE(fi);
1818 dst->ims = new CvArr*[dst->count];
1819 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1820 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1821 if (!convert_to_CvArr(item, &dst->ims[i]))
1828 static int convert_to_floatPTRPTR(PyObject *o, float*** dst, const char *name = "no_name")
1830 PyObject *fi = PySequence_Fast(o, name);
1833 Py_ssize_t sz = PySequence_Fast_GET_SIZE(fi);
1834 float **r = new float*[sz];
1835 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1836 PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1838 if (!convert_to_floats(item, &ff))
1846 static int convert_to_IplConvKernelPTR(PyObject *o, IplConvKernel** dst, const char *name = "no_name")
1848 if (PyType_IsSubtype(o->ob_type, &iplconvkernel_Type)) {
1849 *dst = ((iplconvkernel_t*)o)->a;
1852 (*dst) = (IplConvKernel*)NULL;
1853 return failmsg("Expected IplConvKernel for argument '%s'", name);
1857 static int convert_to_CvCapturePTR(PyObject *o, CvCapture** dst, const char *name = "no_name")
1859 if (PyType_IsSubtype(o->ob_type, &cvcapture_Type)) {
1860 *dst = ((cvcapture_t*)o)->a;
1863 (*dst) = (CvCapture*)NULL;
1864 return failmsg("Expected CvCapture for argument '%s'", name);
1868 static int convert_to_CvVideoWriterPTR(PyObject *o, CvVideoWriter** dst, const char *name = "no_name")
1870 if (PyType_IsSubtype(o->ob_type, &cvvideowriter_Type)) {
1871 *dst = ((cvvideowriter_t*)o)->a;
1874 (*dst) = (CvVideoWriter*)NULL;
1875 return failmsg("Expected CvVideoWriter for argument '%s'", name);
1879 static int convert_to_CvMomentsPTR(PyObject *o, CvMoments** dst, const char *name = "no_name")
1881 if (PyType_IsSubtype(o->ob_type, &cvmoments_Type)) {
1882 (*dst) = &(((cvmoments_t*)o)->a);
1885 (*dst) = (CvMoments*)NULL;
1886 return failmsg("Expected CvMoments for argument '%s'", name);
1890 static int convert_to_CvFontPTR(PyObject *o, CvFont** dst, const char *name = "no_name")
1892 if (PyType_IsSubtype(o->ob_type, &cvfont_Type)) {
1893 (*dst) = &(((cvfont_t*)o)->a);
1896 (*dst) = (CvFont*)NULL;
1897 return failmsg("Expected CvFont for argument '%s'", name);
1901 static int convert_to_CvHaarClassifierCascadePTR(PyObject *o, CvHaarClassifierCascade** dst, const char *name = "no_name")
1903 if (PyType_IsSubtype(o->ob_type, &cvhaarclassifiercascade_Type)) {
1904 (*dst) = ((cvhaarclassifiercascade_t*)o)->a;
1907 (*dst) = (CvHaarClassifierCascade*)NULL;
1908 return failmsg("Expected CvHaarClassifierCascade for argument '%s'", name);
1912 static int convert_to_CvContourTreePTR(PyObject *o, CvContourTree** dst, const char *name = "no_name")
1914 if (PyType_IsSubtype(o->ob_type, &cvcontourtree_Type)) {
1915 (*dst) = ((cvcontourtree_t*)o)->a;
1919 return failmsg("Expected CvContourTree for argument '%s'", name);
1923 static int convert_to_CvPOSITObjectPTR(PyObject *o, CvPOSITObject** dst, const char *name = "no_name")
1925 if (PyType_IsSubtype(o->ob_type, &cvpositobject_Type)) {
1926 (*dst) = ((cvpositobject_t*)o)->a;
1929 (*dst) = (CvPOSITObject*)NULL;
1930 return failmsg("Expected CvPOSITObject for argument '%s'", name);
1934 static int convert_to_CvRNGPTR(PyObject *o, CvRNG** dst, const char *name = "no_name")
1936 if (PyType_IsSubtype(o->ob_type, &cvrng_Type)) {
1937 (*dst) = &(((cvrng_t*)o)->a);
1940 (*dst) = (CvRNG*)NULL;
1941 return failmsg("Expected CvRNG for argument '%s'", name);
1945 typedef void* generic;
1946 static int convert_to_generic(PyObject *o, generic *dst, const char *name = "no_name")
1948 if (PyType_IsSubtype(o->ob_type, &iplimage_Type))
1949 return convert_to_IplImage(o, (IplImage**)dst, name);
1950 else if (PyType_IsSubtype(o->ob_type, &cvmat_Type))
1951 return convert_to_CvMat(o, (CvMat**)dst, name);
1952 else if (PyType_IsSubtype(o->ob_type, &cvmatnd_Type))
1953 return convert_to_CvMatND(o, (CvMatND**)dst, name);
1955 return failmsg("Cannot identify type of '%s'", name);
1959 static int convert_to_CvTermCriteria(PyObject *o, CvTermCriteria* dst, const char *name = "no_name")
1961 if (!PyArg_ParseTuple(o, "iid", &dst->type, &dst->max_iter, &dst->epsilon))
1966 static int convert_to_CvBox2D(PyObject *o, CvBox2D* dst, const char *name = "no_name")
1968 if (!PyArg_ParseTuple(o, "(ff)(ff)f", &dst->center.x, &dst->center.y, &dst->size.width, &dst->size.height, &dst->angle))
1973 static int convert_to_CvSubdiv2DPTR(PyObject *o, CvSubdiv2D** dst, const char *name = "no_name")
1975 if (PyType_IsSubtype(o->ob_type, &cvsubdiv2d_Type)) {
1976 (*dst) = (((cvsubdiv2d_t*)o)->a);
1979 (*dst) = (CvSubdiv2D*)NULL;
1980 return failmsg("Expected CvSubdiv2D for argument '%s'", name);
1984 static int convert_to_CvNextEdgeType(PyObject *o, CvNextEdgeType *dst, const char *name = "no_name")
1986 if (!PyNumber_Check(o)) {
1987 *dst = (CvNextEdgeType)NULL;
1988 return failmsg("Expected number for CvNextEdgeType argument '%s'", name);
1990 *dst = (CvNextEdgeType)PyInt_AsLong(o);
1995 static int convert_to_CvSubdiv2DEdge(PyObject *o, CvSubdiv2DEdge *dst, const char *name = "no_name")
1997 if (PyType_IsSubtype(o->ob_type, &cvsubdiv2dedge_Type)) {
1998 (*dst) = (((cvsubdiv2dedge_t*)o)->a);
2002 return failmsg("Expected CvSubdiv2DEdge for argument '%s'", name);
2006 /************************************************************************/
2008 static PyObject *pythonize_CvMat(cvmat_t *m)
2010 // Need to make this CvMat look like any other, with a Python
2011 // string as its data.
2012 // So copy the image data into a Python string object, then release
2015 assert(mat->step != 0);
2016 PyObject *data = PyString_FromStringAndSize((char*)(mat->data.ptr), mat->rows * mat->step);
2019 cvDecRefData(mat); // Ref count should be zero here, so this is a release
2020 return (PyObject*)m;
2023 static PyObject *pythonize_IplImage(iplimage_t *cva)
2025 // Need to make this iplimage look like any other, with a Python
2026 // string as its data.
2027 // So copy the image data into a Python string object, then release
2030 IplImage *ipl = (IplImage*)(cva->a);
2031 PyObject *data = PyString_FromStringAndSize(ipl->imageData, ipl->imageSize);
2036 return (PyObject*)cva;
2039 static PyObject *pythonize_CvMatND(cvmatnd_t *m)
2042 // Need to make this CvMatND look like any other, with a Python
2043 // string as its data.
2044 // So copy the image data into a Python string object, then release
2048 CvMatND *mat = m->a;
2049 assert(mat->dim[0].step != 0);
2050 PyObject *data = PyString_FromStringAndSize((char*)(mat->data.ptr), mat->dim[0].size * mat->dim[0].step);
2053 cvDecRefData(mat); // Ref count should be zero here, so this is a release
2055 return (PyObject*)m;
2058 /************************************************************************/
2059 /* FROM_xxx: C -> Python converters.
2061 * Turn various OpenCV types (and some aggregate types above)
2062 * into Python objects. Used by the generated code.
2064 * All these functions and macros return a new reference.
2067 #define FROM_double(r) PyFloat_FromDouble(r)
2068 #define FROM_float(r) PyFloat_FromDouble(r)
2069 #define FROM_int(r) PyInt_FromLong(r)
2070 #define FROM_unsigned(r) PyLong_FromUnsignedLong(r)
2071 #define FROM_CvBox2D(r) Py_BuildValue("(ff)(ff)f", r.center.x, r.center.y, r.size.width, r.size.height, r.angle)
2072 #define FROM_CvScalar(r) Py_BuildValue("(ffff)", r.val[0], r.val[1], r.val[2], r.val[3])
2073 #define FROM_CvPoint(r) Py_BuildValue("(ii)", r.x, r.y)
2074 #define FROM_CvConnectedComp(r) Py_BuildValue("(fNN)", r.area, FROM_CvScalar(r.value), FROM_CvRect(r.rect))
2075 #define FROM_CvPoint2D32f(r) Py_BuildValue("(ff)", r.x, r.y)
2076 #define FROM_CvSize(r) Py_BuildValue("(ii)", r.width, r.height)
2077 #define FROM_CvRect(r) Py_BuildValue("(iiii)", r.x, r.y, r.width, r.height)
2078 #define FROM_CvSeqPTR(r) _FROM_CvSeqPTR(r, pyobj_storage)
2079 #define FROM_CvSubdiv2DPTR(r) _FROM_CvSubdiv2DPTR(r, pyobj_storage)
2080 #define FROM_CvPoint2D64f(r) Py_BuildValue("(ff)", r.x, r.y)
2082 static PyObject *_FROM_CvSeqPTR(CvSeq *s, PyObject *storage)
2084 cvseq_t *ps = PyObject_NEW(cvseq_t, &cvseq_Type);
2086 ps->container = storage;
2087 Py_INCREF(ps->container);
2088 return (PyObject*)ps;
2091 static PyObject *_FROM_CvSubdiv2DPTR(CvSubdiv2D *s, PyObject *storage)
2093 cvsubdiv2d_t *ps = PyObject_NEW(cvsubdiv2d_t, &cvsubdiv2d_Type);
2095 ps->container = storage;
2096 Py_INCREF(ps->container);
2097 return (PyObject*)ps;
2100 static PyObject *FROM_floats(floats r)
2104 pr = PyList_New(r.count);
2105 for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2106 PyList_SetItem(pr, i, PyFloat_FromDouble(r.f[i]));
2111 static PyObject *FROM_chars(chars r)
2115 pr = PyList_New(r.count);
2116 for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2117 PyList_SetItem(pr, i, PyInt_FromLong(r.f[i]));
2122 static PyObject *FROM_cvpoint2d32f_count(cvpoint2d32f_count r)
2126 pr = PyList_New(r.count);
2127 for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2128 PyList_SetItem(pr, i, FROM_CvPoint2D32f(r.points[i]));
2133 static PyObject *FROM_CvPoint2D32fs(CvPoint2D32fs r)
2137 pr = PyList_New(r.count);
2138 for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2139 PyList_SetItem(pr, i, FROM_CvPoint2D32f(r.p[i]));
2144 typedef CvSeq CvSeqOfCvConvexityDefect;
2145 static PyObject *FROM_CvSeqOfCvConvexityDefectPTR(CvSeqOfCvConvexityDefect *r)
2148 pr = PyList_New(r->total);
2149 for (int i = 0; i < r->total; i++) {
2150 CvConvexityDefect *pd = CV_GET_SEQ_ELEM(CvConvexityDefect, r, i);
2151 PyList_SetItem(pr, i, Py_BuildValue("(ii)(ii)(ii)f",
2152 pd->start->x, pd->start->y,
2153 pd->end->x, pd->end->y,
2154 pd->depth_point->x, pd->depth_point->y,
2157 // This function has copied the CvSeq data into a list. Hence the
2158 // CvSeq is not being returned to the caller. Hence, no reference
2159 // count increase for the storage, unlike _FROM_CvSeqPTR.
2163 typedef CvSeq CvSeqOfCvAvgComp;
2164 static PyObject *FROM_CvSeqOfCvAvgCompPTR(CvSeqOfCvAvgComp *r)
2167 pr = PyList_New(r->total);
2168 for (int i = 0; i < r->total; i++) {
2169 CvAvgComp *pd = CV_GET_SEQ_ELEM(CvAvgComp, r, i);
2170 PyList_SetItem(pr, i, Py_BuildValue("(iiii)i",
2171 pd->rect.x, pd->rect.y,
2172 pd->rect.width, pd->rect.height,
2175 // This function has copied the CvSeq data into a list. Hence the
2176 // CvSeq is not being returned to the caller. Hence, no reference
2177 // count increase for the storage, unlike _FROM_CvSeqPTR.
2181 typedef CvSeq CvSeqOfCvStarKeypoint;
2182 static PyObject *FROM_CvSeqOfCvStarKeypointPTR(CvSeqOfCvStarKeypoint *r)
2185 pr = PyList_New(r->total);
2186 for (int i = 0; i < r->total; i++) {
2187 CvStarKeypoint *pd = CV_GET_SEQ_ELEM(CvStarKeypoint, r, i);
2188 PyList_SetItem(pr, i, Py_BuildValue("(ii)if",
2193 // This function has copied the CvSeq data into a list. Hence the
2194 // CvSeq is not being returned to the caller. Hence, no reference
2195 // count increase for the storage, unlike _FROM_CvSeqPTR.
2199 typedef CvSeq CvSeqOfCvSURFPoint;
2200 static PyObject *FROM_CvSeqOfCvSURFPointPTR(CvSeqOfCvSURFPoint *r)
2203 pr = PyList_New(r->total);
2204 for (int i = 0; i < r->total; i++) {
2205 CvSURFPoint *pd = CV_GET_SEQ_ELEM(CvSURFPoint, r, i);
2206 PyList_SetItem(pr, i, Py_BuildValue("(ff)iiff",
2213 // This function has copied the CvSeq data into a list. Hence the
2214 // CvSeq is not being returned to the caller. Hence, no reference
2215 // count increase for the storage, unlike _FROM_CvSeqPTR.
2219 typedef CvSeq CvSeqOfCvSURFDescriptor;
2220 static PyObject *FROM_CvSeqOfCvSURFDescriptorPTR(CvSeqOfCvSURFDescriptor *r)
2223 pr = PyList_New(r->total);
2224 for (int i = 0; i < r->total; i++) {
2225 float *pd = (float*)cvGetSeqElem(r, i);
2226 int count = r->elem_size / sizeof(float);
2227 PyObject *oi = PyList_New(count);
2228 for (int j = 0; j < count; j++) {
2229 PyList_SetItem(oi, j, PyFloat_FromDouble(pd[j]));
2231 PyList_SetItem(pr, i, oi);
2233 // This function has copied the CvSeq data into a list. Hence the
2234 // CvSeq is not being returned to the caller. Hence, no reference
2235 // count increase for the storage, unlike _FROM_CvSeqPTR.
2239 static PyObject *FROM_IplConvKernelPTR(IplConvKernel *r)
2241 iplconvkernel_t *ick = PyObject_NEW(iplconvkernel_t, &iplconvkernel_Type);
2243 return (PyObject*)ick;
2246 static PyObject *FROM_CvCapturePTR(CvCapture *r)
2248 cvcapture_t *c = PyObject_NEW(cvcapture_t, &cvcapture_Type);
2250 return (PyObject*)c;
2253 static PyObject *FROM_CvVideoWriterPTR(CvVideoWriter *r)
2255 cvvideowriter_t *c = PyObject_NEW(cvvideowriter_t, &cvvideowriter_Type);
2257 return (PyObject*)c;
2260 typedef CvPoint2D32f CvPoint2D32f_4[4];
2261 static PyObject *FROM_CvPoint2D32f_4(CvPoint2D32f* r)
2263 return Py_BuildValue("(ff)(ff)(ff)(ff)",
2270 typedef float CvMatr32f_i[9];
2272 static PyObject *FROM_CvMatr32f_i(CvMatr32f_i r)
2274 return Py_BuildValue("(fff)(fff)(fff)",
2280 typedef float CvVect32f_i[3];
2281 static PyObject *FROM_CvVect32f_i(CvVect32f_i r)
2283 return Py_BuildValue("fff",
2287 static PyObject *FROM_CvFont(CvFont r)
2289 cvfont_t *cf = PyObject_NEW(cvfont_t, &cvfont_Type);
2291 return (PyObject*)cf;
2294 static PyObject *FROM_CvSubdiv2DPointPTR(CvSubdiv2DPoint* r)
2297 cvsubdiv2dpoint_t *cf = PyObject_NEW(cvsubdiv2dpoint_t, &cvsubdiv2dpoint_Type);
2299 return (PyObject*)cf;
2306 static PyObject *FROM_IplImagePTR(IplImage *r)
2308 iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2310 return pythonize_IplImage(cva);
2313 static PyObject *FROM_ROIplImagePTR(ROIplImage *r)
2316 iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2317 cva->a = cvCreateImageHeader(cvSize(100,100), 8, 1);
2319 cva->data = PyBuffer_FromReadWriteMemory(r->imageData, r->height * r->widthStep);
2321 return (PyObject*)cva;
2327 static PyObject *FROM_CvMatPTR(CvMat *r)
2329 cvmat_t *cvm = PyObject_NEW(cvmat_t, &cvmat_Type);
2332 return pythonize_CvMat(cvm);
2335 static PyObject *FROM_CvMat(CvMat *r)
2337 cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2339 return pythonize_CvMat(m);
2342 static PyObject *FROM_CvMatNDPTR(CvMatND *r)
2344 cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2346 return pythonize_CvMatND(m);
2349 static PyObject *FROM_CvPOSITObjectPTR(CvPOSITObject *r)
2351 cvpositobject_t *m = PyObject_NEW(cvpositobject_t, &cvpositobject_Type);
2353 return (PyObject*)m;
2356 static PyObject *FROM_CvRNG(CvRNG r)
2358 cvrng_t *m = PyObject_NEW(cvrng_t, &cvrng_Type);
2360 return (PyObject*)m;
2363 static PyObject *FROM_CvHaarClassifierCascade(CvHaarClassifierCascade *r)
2365 cvhaarclassifiercascade_t *m = PyObject_NEW(cvhaarclassifiercascade_t, &cvhaarclassifiercascade_Type);
2367 return (PyObject*)m;
2370 static PyObject *FROM_CvMoments(CvMoments r)
2372 cvmoments_t *m = PyObject_NEW(cvmoments_t, &cvmoments_Type);
2374 return (PyObject*)m;
2377 static PyObject *FROM_CvContourTreePTR(CvContourTree *r)
2379 cvcontourtree_t *m = PyObject_NEW(cvcontourtree_t, &cvcontourtree_Type);
2381 return (PyObject*)m;
2384 static PyObject *FROM_generic(generic r)
2386 CvTypeInfo* t = cvTypeOf(r);
2388 failmsg("OpenCV returned NULL");
2390 } if (strcmp(t->type_name, "opencv-image") == 0)
2391 return FROM_IplImagePTR((IplImage*)r);
2392 else if (strcmp(t->type_name, "opencv-matrix") == 0)
2393 return FROM_CvMat((CvMat*)r);
2394 else if (strcmp(t->type_name, "opencv-haar-classifier") == 0)
2395 return FROM_CvHaarClassifierCascade((CvHaarClassifierCascade*)r);
2397 failmsg("Unknown OpenCV type '%s'", t->type_name);
2402 static PyObject *FROM_CvSubdiv2DEdge(CvSubdiv2DEdge r)
2404 cvsubdiv2dedge_t *m = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
2406 m->container = Py_None; // XXX
2407 Py_INCREF(m->container);
2408 return (PyObject*)m;
2411 /************************************************************************/
2413 /* A few functions are too odd to be generated,
2414 * so are handwritten here */
2416 static PyObject *pycvWaitKey(PyObject *self, PyObject *args, PyObject *kw)
2420 const char *keywords[] = { "delay", NULL };
2421 if (!PyArg_ParseTupleAndKeywords(args, kw, "|i", (char**)keywords, &delay))
2424 Py_BEGIN_ALLOW_THREADS
2425 r = cvWaitKey(delay);
2426 Py_END_ALLOW_THREADS
2430 static PyObject *pycvLoadImage(PyObject *self, PyObject *args, PyObject *kw)
2432 const char *keywords[] = { "filename", "iscolor", NULL };
2434 int iscolor = CV_LOAD_IMAGE_COLOR;
2436 if (!PyArg_ParseTupleAndKeywords(args, kw, "s|i", (char**)keywords, &filename, &iscolor))
2439 // Inside ALLOW_THREADS, must not reference 'filename' because it might move.
2440 // So make a local copy 'filename_copy'.
2441 char filename_copy[2048];
2442 strncpy(filename_copy, filename, sizeof(filename_copy));
2445 Py_BEGIN_ALLOW_THREADS
2446 r = cvLoadImage(filename_copy, iscolor);
2447 Py_END_ALLOW_THREADS
2450 PyErr_SetFromErrnoWithFilename(PyExc_IOError, filename);
2453 return FROM_IplImagePTR(r);
2457 static PyObject *pycvCreateImageHeader(PyObject *self, PyObject *args)
2459 int w, h, depth, channels;
2460 if (!PyArg_ParseTuple(args, "(ii)Ii", &w, &h, &depth, &channels))
2462 iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2463 cva->a = cvCreateImageHeader(cvSize(w, h), depth, channels);
2464 if (cva->a == NULL) {
2465 PyErr_SetString(PyExc_TypeError, "CreateImage failed");
2468 cva->data = Py_None;
2469 Py_INCREF(cva->data);
2472 return (PyObject*)cva;
2476 static PyObject *pycvCreateImage(PyObject *self, PyObject *args)
2478 int w, h, depth, channels;
2479 if (!PyArg_ParseTuple(args, "(ii)Ii:CreateImage", &w, &h, &depth, &channels))
2481 iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2482 cva->a = cvCreateImage(cvSize(w, h), depth, channels);
2483 if (cva->a == NULL) {
2484 PyErr_SetString(PyExc_TypeError, "CreateImage failed");
2487 return pythonize_IplImage(cva);
2491 static PyObject *pycvCreateMatHeader(PyObject *self, PyObject *args)
2493 int rows, cols, type;
2494 if (!PyArg_ParseTuple(args, "iii", &rows, &cols, &type))
2496 cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2497 m->a = cvCreateMatHeader(rows, cols, type);
2499 PyErr_SetString(PyExc_TypeError, "CreateMat failed");
2505 return (PyObject*)m;
2509 static PyObject *pycvCreateMat(PyObject *self, PyObject *args)
2511 int rows, cols, type;
2512 if (!PyArg_ParseTuple(args, "iii", &rows, &cols, &type))
2514 cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2515 m->a = cvCreateMat(rows, cols, type);
2517 PyErr_SetString(PyExc_TypeError, "CreateMat failed");
2520 return pythonize_CvMat(m);
2524 static PyObject *pycvCreateMatNDHeader(PyObject *self, PyObject *args)
2529 if (!PyArg_ParseTuple(args, "O&i", convert_to_ints, (void*)&dims, &type))
2531 cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2532 ERRWRAP(m->a = cvCreateMatNDHeader(dims.count, dims.i, type));
2536 return (PyObject*)m;
2540 static PyObject *pycvCreateMatND(PyObject *self, PyObject *args)
2545 if (!PyArg_ParseTuple(args, "O&i", convert_to_ints, (void*)&dims, &type))
2547 cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2548 ERRWRAP(m->a = cvCreateMatND(dims.count, dims.i, type));
2549 return pythonize_CvMatND(m);
2552 static PyObject *pycvCreateHist(PyObject *self, PyObject *args)
2556 float **ranges = NULL;
2559 if (!PyArg_ParseTuple(args, "Oi|O&i", &dims, &type, convert_to_floatPTRPTR, (void*)&ranges, &uniform)) {
2562 cvhistogram_t *h = PyObject_NEW(cvhistogram_t, &cvhistogram_Type);
2563 args = Py_BuildValue("Ni", dims, CV_32FC1);
2564 h->bins = pycvCreateMatND(self, args);
2566 if (h->bins == NULL) {
2569 h->h.type = CV_HIST_MAGIC_VAL;
2570 if (!convert_to_CvArr(h->bins, &(h->h.bins), "bins"))
2573 ERRWRAP(cvSetHistBinRanges(&(h->h), ranges, uniform));
2575 return (PyObject*)h;
2578 static PyObject *pycvInitLineIterator(PyObject *self, PyObject *args)
2583 int connectivity = 8;
2584 int left_to_right = 0;
2586 if (!PyArg_ParseTuple(args, "O&O&O&|ii",
2587 convert_to_CvArr, &image,
2588 convert_to_CvPoint, &pt1,
2589 convert_to_CvPoint, &pt2,
2594 cvlineiterator_t *pi = PyObject_NEW(cvlineiterator_t, &cvlineiterator_Type);
2595 pi->count = cvInitLineIterator(image, pt1, pt2, &pi->iter, connectivity, left_to_right);
2596 ERRWRAP(pi->type = cvGetElemType(image));
2597 return (PyObject*)pi;
2600 static PyObject *pycvCreateMemStorage(PyObject *self, PyObject *args)
2603 if (!PyArg_ParseTuple(args, "|i", &block_size))
2605 cvmemstorage_t *pm = PyObject_NEW(cvmemstorage_t, &cvmemstorage_Type);
2606 pm->a = cvCreateMemStorage(block_size);
2607 return (PyObject*)pm;
2610 // single index: return row
2611 // 2 indices: row, column
2612 // both row and column can be slices. column slice must have a step of 1.
2614 // returns a scalar when all dimensions are specified and all are integers. Otherwise returns a CvMat.
2616 static PyObject *cvarr_GetItem(PyObject *o, PyObject *key)
2621 if (!convert_to_CvArr(o, &cva, "src"))
2624 if (!convert_to_dims(key, &dd, cva, "key")) {
2628 // Figure out if all supplied indices have a stride of zero - means they are not slices
2629 // and if all indices are positive
2631 for (int i = 0; i < dd.count; i++) {
2632 all0 &= (dd.step[i] == 0) && (0 <= dd.i[i]);
2635 // if every dimension supplied, and none are slices, return the scalar
2636 if ((cvGetDims(cva) == dd.count) && all0) {
2638 ERRWRAP(s = cvGetND(cva, dd.i));
2639 return PyObject_FromCvScalar(s, cvGetElemType(cva));
2641 // pad missing dimensions
2642 for (int i = dd.count; i < cvGetDims(cva); i++) {
2645 dd.length[i] = cvGetDimSize(cva, i);
2647 dd.count = cvGetDims(cva);
2649 // negative steps are illegal for OpenCV
2650 for (int i = 0; i < dd.count; i++) {
2652 return (PyObject*)failmsg("Negative step is illegal");
2655 // zero length illegal for OpenCV
2656 for (int i = 0; i < dd.count; i++) {
2657 if (dd.length[i] == 0)
2658 return (PyObject*)failmsg("Zero sized dimension is illegal");
2661 // column step can only be 0 or 1
2662 if ((dd.step[dd.count-1] != 0) && (dd.step[dd.count-1] != 1))
2663 return (PyObject*)failmsg("Column step is illegal");
2665 if (is_cvmat(o) || is_iplimage(o)) {
2666 cvmat_t *sub = PyObject_NEW(cvmat_t, &cvmat_Type);
2667 sub->a = cvCreateMatHeader(dd.length[0], dd.length[1], cvGetElemType(cva));
2668 uchar *old0; // pointer to first element in old mat
2670 cvGetRawData(cva, &old0, &oldstep);
2671 uchar *new0; // pointer to first element in new mat
2672 ERRWRAP(new0 = cvPtrND(cva, dd.i));
2674 sub->a->step = oldstep * dd.step[0];
2675 sub->data = what_data(o);
2676 Py_INCREF(sub->data);
2677 sub->offset = new0 - old0;
2678 return (PyObject*)sub;
2680 cvmatnd_t *sub = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2681 sub->a = cvCreateMatNDHeader(dd.count, dd.length, cvGetElemType(cva));
2682 uchar *old0; // pointer to first element in old mat
2683 cvGetRawData(cva, &old0);
2684 uchar *new0; // pointer to first element in new mat
2685 ERRWRAP(new0 = cvPtrND(cva, dd.i));
2687 for (int d = 0; d < dd.count; d++) {
2688 int stp = dd.step[d];
2689 sub->a->dim[d].step = ((CvMatND*)cva)->dim[d].step * ((stp == 0) ? 1 : stp);
2690 sub->a->dim[d].size = dd.length[d];
2692 sub->data = what_data(o);
2693 Py_INCREF(sub->data);
2694 sub->offset = new0 - old0;
2695 return (PyObject*)sub;
2700 static int cvarr_SetItem(PyObject *o, PyObject *key, PyObject *v)
2705 if (!convert_to_CvArr(o, &cva, "src"))
2708 if (!convert_to_dims(key, &dd, cva, "key")) {
2712 if (cvGetDims(cva) != dd.count) {
2713 PyErr_SetString(PyExc_TypeError, "key length does not match array dimension");
2718 if (PySequence_Check(v)) {
2719 PyObject *fi = PySequence_Fast(v, "v");
2722 if (PySequence_Fast_GET_SIZE(fi) != CV_MAT_CN(cvGetElemType(cva))) {
2723 PyErr_SetString(PyExc_TypeError, "sequence size must be same as channel count");
2726 for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++)
2727 s.val[i] = PyFloat_AsDouble(PySequence_Fast_GET_ITEM(fi, i));
2730 if (1 != CV_MAT_CN(cvGetElemType(cva))) {
2731 PyErr_SetString(PyExc_TypeError, "scalar supplied but channel count does not equal 1");
2734 s.val[0] = PyFloat_AsDouble(v);
2738 cvSet1D(cva, dd.i[0], s);
2741 cvSet2D(cva, dd.i[0], dd.i[1], s);
2744 cvSet3D(cva, dd.i[0], dd.i[1], dd.i[2], s);
2747 cvSetND(cva, dd.i, s);
2748 // XXX - OpenCV bug? - seems as if an error in cvSetND does not set error status?
2751 if (cvGetErrStatus() != 0) {
2752 translate_error_to_exception();
2760 static PyObject *pycvSetData(PyObject *self, PyObject *args)
2765 if (!PyArg_ParseTuple(args, "OOi", &o, &s, &step))
2767 if (is_iplimage(o)) {
2768 iplimage_t *ipl = (iplimage_t*)o;
2769 ipl->a->widthStep = step;
2770 Py_DECREF(ipl->data);
2772 Py_INCREF(ipl->data);
2773 } else if (is_cvmat(o)) {
2774 cvmat_t *m = (cvmat_t*)o;
2779 } else if (is_cvmatnd(o)) {
2780 cvmatnd_t *m = (cvmatnd_t*)o;
2785 PyErr_SetString(PyExc_TypeError, "SetData argument must be either IplImage, CvMat or CvMatND");
2792 static PyObject *what_data(PyObject *o)
2794 if (is_iplimage(o)) {
2795 iplimage_t *ipl = (iplimage_t*)o;
2797 } else if (is_cvmat(o)) {
2798 cvmat_t *m = (cvmat_t*)o;
2800 } else if (is_cvmatnd(o)) {
2801 cvmatnd_t *m = (cvmatnd_t*)o;
2809 static PyObject *pycvCreateData(PyObject *self, PyObject *args)
2813 if (!PyArg_ParseTuple(args, "O", &o))
2817 if (!convert_to_CvArr(o, &a, "arr"))
2819 ERRWRAP(cvCreateData(a));
2821 Py_DECREF(what_data(o));
2822 if (is_iplimage(o)) {
2823 iplimage_t *ipl = (iplimage_t*)o;
2824 pythonize_IplImage(ipl);
2825 } else if (is_cvmat(o)) {
2826 cvmat_t *m = (cvmat_t*)o;
2828 } else if (is_cvmatnd(o)) {
2829 cvmatnd_t *m = (cvmatnd_t*)o;
2830 pythonize_CvMatND(m);
2832 PyErr_SetString(PyExc_TypeError, "CreateData argument must be either IplImage, CvMat or CvMatND");
2839 static PyObject *pycvGetDims(PyObject *self, PyObject *args)
2843 if (!PyArg_ParseTuple(args, "O", &o))
2846 if (!convert_to_CvArr(o, &cva, "src"))
2850 ERRWRAP(nd = cvGetDims(cva));
2851 PyObject *r = PyTuple_New(nd);
2852 for (i = 0; i < nd; i++)
2853 PyTuple_SetItem(r, i, PyInt_FromLong(cvGetDimSize(cva, i)));
2857 static PyObject *pycvGetImage(PyObject *self, PyObject *args)
2861 if (!PyArg_ParseTuple(args, "O", &o))
2863 if (is_iplimage(o)) {
2867 IplImage *ipl = cvCreateImageHeader(cvSize(100,100), 8, 1); // these args do not matter, because overwritten
2869 if (!convert_to_CvArr(o, &cva, "src"))
2871 ERRWRAP(cvGetImage(cva, ipl));
2873 iplimage_t *oipl = PyObject_NEW(iplimage_t, &iplimage_Type);
2875 oipl->data = what_data(o);
2876 Py_INCREF(oipl->data);
2879 r = (PyObject*)oipl;
2884 static PyObject *pycvGetMat(PyObject *self, PyObject *args)
2888 if (!PyArg_ParseTuple(args, "O", &o))
2894 CvMat *m = cvCreateMatHeader(100,100, 1); // these args do not matter, because overwritten
2896 if (!convert_to_CvArr(o, &cva, "src"))
2898 ERRWRAP(cvGetMat(cva, m));
2900 cvmat_t *om = PyObject_NEW(cvmat_t, &cvmat_Type);
2902 om->data = what_data(o);
2903 Py_INCREF(om->data);
2911 static PyObject *pycvReshape(PyObject *self, PyObject *args)
2917 if (!PyArg_ParseTuple(args, "Oi|i", &o, &new_cn, &new_rows))
2920 CvMat *m = cvCreateMatHeader(100,100, 1); // these args do not matter, because overwritten
2922 if (!convert_to_CvArr(o, &cva, "src"))
2924 ERRWRAP(cvReshape(cva, m, new_cn, new_rows));
2926 cvmat_t *om = PyObject_NEW(cvmat_t, &cvmat_Type);
2928 om->data = what_data(o);
2929 Py_INCREF(om->data);
2932 return (PyObject*)om;
2935 static void OnMouse(int event, int x, int y, int flags, void* param)
2937 PyGILState_STATE gstate;
2938 gstate = PyGILState_Ensure();
2940 PyObject *o = (PyObject*)param;
2941 PyObject *args = Py_BuildValue("iiiiO", event, x, y, flags, PyTuple_GetItem(o, 1));
2943 PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
2949 PyGILState_Release(gstate);
2952 static PyObject *pycvSetMouseCallback(PyObject *self, PyObject *args, PyObject *kw)
2954 const char *keywords[] = { "window_name", "on_mouse", "param", NULL };
2957 PyObject *param = NULL;
2959 if (!PyArg_ParseTupleAndKeywords(args, kw, "sO|O", (char**)keywords, &name, &on_mouse, ¶m))
2961 if (!PyCallable_Check(on_mouse)) {
2962 PyErr_SetString(PyExc_TypeError, "on_mouse must be callable");
2965 if (param == NULL) {
2968 ERRWRAP(cvSetMouseCallback(name, OnMouse, Py_BuildValue("OO", on_mouse, param)));
2972 void OnChange(int pos, void *param)
2974 PyGILState_STATE gstate;
2975 gstate = PyGILState_Ensure();
2977 PyObject *o = (PyObject*)param;
2978 PyObject *args = Py_BuildValue("(i)", pos);
2979 PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
2983 PyGILState_Release(gstate);
2986 static PyObject *pycvCreateTrackbar(PyObject *self, PyObject *args, PyObject *kw)
2988 const char *keywords[] = { "trackbar_name", "window_name", "value", "count", "on_change", NULL };
2989 PyObject *on_change;
2990 char* trackbar_name;
2992 int *value = new int;
2995 if (!PyArg_ParseTupleAndKeywords(args, kw, "ssiiO", (char**)keywords, &trackbar_name, &window_name, value, &count, &on_change))
2997 if (!PyCallable_Check(on_change)) {
2998 PyErr_SetString(PyExc_TypeError, "on_change must be callable");
3001 ERRWRAP(cvCreateTrackbar2(trackbar_name, window_name, value, count, OnChange, Py_BuildValue("OO", on_change, Py_None)));
3005 static PyObject *pycvFindContours(PyObject *self, PyObject *args, PyObject *kw)
3008 PyObject *pyobj_image = NULL;
3009 CvMemStorage* storage;
3010 PyObject *pyobj_storage = NULL;
3011 CvSeq* first_contour;
3012 int header_size = sizeof(CvContour);
3013 int mode = CV_RETR_LIST;
3014 int method = CV_CHAIN_APPROX_SIMPLE;
3015 CvPoint offset = cvPoint(0,0);
3016 PyObject *pyobj_offset = NULL;
3018 const char *keywords[] = { "image", "storage", "mode", "method", "offset", NULL };
3019 if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|iiO", (char**)keywords, &pyobj_image, &pyobj_storage, &mode, &method, &pyobj_offset))
3021 if (!convert_to_CvArr(pyobj_image, &image, "image")) return NULL;
3022 if (!convert_to_CvMemStorage(pyobj_storage, &storage, "storage")) return NULL;
3023 if ((pyobj_offset != NULL) && !convert_to_CvPoint(pyobj_offset, &offset, "offset")) return NULL;
3024 ERRWRAP(cvFindContours(image, storage, &first_contour, header_size, mode, method, offset));
3025 cvseq_t *ps = PyObject_NEW(cvseq_t, &cvseq_Type);
3026 ps->a = first_contour;
3027 ps->container = PyTuple_GetItem(args, 1); // storage
3028 Py_INCREF(ps->container);
3029 return (PyObject*)ps;
3032 static PyObject *pycvApproxPoly(PyObject *self, PyObject *args, PyObject *kw)
3035 PyObject *pyobj_src_seq = NULL;
3036 int header_size = sizeof(CvContour);
3037 CvMemStorage* storage;
3038 PyObject *pyobj_storage = NULL;
3040 double parameter = 0;
3043 const char *keywords[] = { "src_seq", "storage", "method", "parameter", "parameter2", NULL };
3044 if (!PyArg_ParseTupleAndKeywords(args, kw, "OOi|di", (char**)keywords, &pyobj_src_seq, &pyobj_storage, &method, ¶meter, ¶meter2))
3046 if (!convert_to_cvarrseq(pyobj_src_seq, &src_seq, "src_seq")) return NULL;
3047 if (!convert_to_CvMemStorage(pyobj_storage, &storage, "storage")) return NULL;
3049 ERRWRAP(r = cvApproxPoly(src_seq.v, header_size, storage, method, parameter, parameter2));
3050 return FROM_CvSeqPTR(r);
3053 static float distance_function_glue( const float* a, const float* b, void* user_param )
3055 PyObject *o = (PyObject*)user_param;
3056 PyObject *args = Py_BuildValue("(ff)(ff)O", a[0], a[1], b[0], b[1], PyTuple_GetItem(o, 1));
3057 PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
3059 return (float)PyFloat_AsDouble(r);
3062 static PyObject *pycvCalcEMD2(PyObject *self, PyObject *args, PyObject *kw)
3064 const char *keywords[] = { "signature1", "signature2", "distance_type", "distance_func", "cost_matrix", "flow", "lower_bound", "userdata", NULL };
3066 PyObject *pyobj_signature1;
3068 PyObject *pyobj_signature2;
3070 PyObject *distance_func = NULL;
3071 CvArr* cost_matrix=NULL;
3072 PyObject *pyobj_cost_matrix = NULL;
3074 PyObject *pyobj_flow = NULL;
3075 float lower_bound = 0.0;
3076 PyObject *userdata = NULL;
3078 if (!PyArg_ParseTupleAndKeywords(args, kw, "OOi|OOOfO", (char**)keywords,
3088 if (!convert_to_CvArr(pyobj_signature1, &signature1, "signature1")) return NULL;
3089 if (!convert_to_CvArr(pyobj_signature2, &signature2, "signature2")) return NULL;
3090 if (pyobj_cost_matrix && !convert_to_CvArr(pyobj_cost_matrix, &cost_matrix, "cost_matrix")) return NULL;
3091 if (pyobj_flow && !convert_to_CvArr(pyobj_flow, &flow, "flow")) return NULL;
3093 if (distance_func == NULL) {
3094 distance_func = Py_None;
3096 if (userdata == NULL) {
3100 PyObject *ud = Py_BuildValue("OO", distance_func, userdata);
3102 ERRWRAP(r = cvCalcEMD2(signature1, signature2, distance_type, distance_function_glue, cost_matrix, flow, &lower_bound, (void*)ud));
3105 return PyFloat_FromDouble(r);
3108 static PyObject *pycvSubdiv2DLocate(PyObject *self, PyObject *args)
3110 PyObject *pyobj_subdiv;
3114 CvSubdiv2DEdge edge;
3115 CvSubdiv2DPoint* vertex;
3117 if (!PyArg_ParseTuple(args, "OO", &pyobj_subdiv, &pyobj_pt))
3119 if (!convert_to_CvSubdiv2DPTR(pyobj_subdiv, &subdiv, "subdiv"))
3121 if (!convert_to_CvPoint2D32f(pyobj_pt, &pt, "pt"))
3124 CvSubdiv2DPointLocation loc = cvSubdiv2DLocate(subdiv, pt, &edge, &vertex);
3127 case CV_PTLOC_INSIDE:
3128 case CV_PTLOC_ON_EDGE:
3129 r = FROM_CvSubdiv2DEdge(edge);
3131 case CV_PTLOC_VERTEX:
3132 r = FROM_CvSubdiv2DPointPTR(vertex);
3134 case CV_PTLOC_OUTSIDE_RECT:
3139 return (PyObject*)failmsg("Unexpected loc from cvSubdiv2DLocate");
3141 return Py_BuildValue("iO", (int)loc, r);;
3144 static PyObject *pycvCalcOpticalFlowPyrLK(PyObject *self, PyObject *args)
3147 PyObject *pyobj_prev = NULL;
3149 PyObject *pyobj_curr = NULL;
3151 PyObject *pyobj_prev_pyr = NULL;
3153 PyObject *pyobj_curr_pyr = NULL;
3154 CvPoint2D32f* prev_features;
3155 PyObject *pyobj_prev_features = NULL;
3156 PyObject *pyobj_curr_features = NULL;
3157 CvPoint2D32f* curr_features;
3160 CvTermCriteria criteria;
3163 if (!PyArg_ParseTuple(args, "OOOOO(ii)i(iif)i|O",
3164 &pyobj_prev, &pyobj_curr, &pyobj_prev_pyr, &pyobj_curr_pyr,
3165 &pyobj_prev_features,
3166 &win_size.width, &win_size.height, &level,
3167 &criteria.type, &criteria.max_iter, &criteria.epsilon,
3169 &pyobj_curr_features))
3171 if (!convert_to_CvArr(pyobj_prev, &prev, "prev")) return NULL;
3172 if (!convert_to_CvArr(pyobj_curr, &curr, "curr")) return NULL;
3173 if (!convert_to_CvArr(pyobj_prev_pyr, &prev_pyr, "prev_pyr")) return NULL;
3174 if (!convert_to_CvArr(pyobj_curr_pyr, &curr_pyr, "curr_pyr")) return NULL;
3175 if (!convert_to_CvPoint2D32fPTR(pyobj_prev_features, &prev_features, "prev_features")) return NULL;
3176 int count = (int)PySequence_Length(pyobj_prev_features);
3177 if (flags & CV_LKFLOW_INITIAL_GUESSES) {
3178 failmsg("flag CV_LKFLOW_INITIAL_GUESSES is determined automatically from function arguments - it is not required");
3181 if (!pyobj_curr_features) {
3182 curr_features = new CvPoint2D32f[count];
3184 if (PySequence_Length(pyobj_curr_features) != count) {
3185 failmsg("curr_features must have same length as prev_features");
3188 if (!convert_to_CvPoint2D32fPTR(pyobj_curr_features, &curr_features, "curr_features")) return NULL;
3189 flags |= CV_LKFLOW_INITIAL_GUESSES;
3191 float *track_error = new float[count];
3192 char* status = new char[count];
3193 ERRWRAP(cvCalcOpticalFlowPyrLK(prev, curr, prev_pyr, curr_pyr, prev_features, curr_features, count, win_size, level, status, track_error, criteria, flags));
3195 cvpoint2d32f_count r0;
3196 r0.points = curr_features;
3207 return Py_BuildValue("NNN", FROM_cvpoint2d32f_count(r0), FROM_chars(r1), FROM_floats(r2));
3210 static PyObject *pycvClipLine(PyObject *self, PyObject *args, PyObject *kw)
3213 PyObject *pyobj_img_size = NULL;
3215 PyObject *pyobj_pt1 = NULL;
3217 PyObject *pyobj_pt2 = NULL;
3219 if (!PyArg_ParseTuple(args, "OOO", &pyobj_img_size, &pyobj_pt1, &pyobj_pt2))
3221 if (!convert_to_CvSize(pyobj_img_size, &img_size, "img_size")) return NULL;
3222 if (!convert_to_CvPoint(pyobj_pt1, &pt1, "pt1")) return NULL;
3223 if (!convert_to_CvPoint(pyobj_pt2, &pt2, "pt2")) return NULL;
3225 ERRWRAP(r = cvClipLine(img_size, &pt1, &pt2));
3229 return Py_BuildValue("NN", FROM_CvPoint(pt1), FROM_CvPoint(pt2));
3233 static PyObject *temp_test(PyObject *self, PyObject *args)
3236 CvArr *im = cvLoadImage("../samples/c/lena.jpg", 0);
3237 printf("im=%p\n", im);
3238 CvMat *m = cvEncodeImage(".jpeg", im);
3241 CvArr *im = cvLoadImage("lena.jpg", 0);
3242 float r0[] = { 0, 255 };
3243 float *ranges[] = { r0 };
3244 int hist_size[] = { 256 };
3245 CvHistogram *hist = cvCreateHist(1, hist_size, CV_HIST_ARRAY, ranges, 1);
3246 cvCalcHist(im, hist, 0, 0);
3250 CvMat* mat = cvCreateMat( 3, 3, CV_32F );
3251 CvMat row_header, *row;
3252 row = cvReshape( mat, &row_header, 0, 1 );
3253 printf("%d,%d\n", row_header.rows, row_header.cols);
3254 printf("ge %08x\n", cvGetElemType(mat));
3258 CvMat *m = cvCreateMat(1, 10, CV_8UC1);
3259 printf("CvMat stride ===> %d\n", m->step);
3263 CvPoint2D32f src[3] = { { 0,0 }, { 1,0 }, { 0,1 } };
3264 CvPoint2D32f dst[3] = { { 0,0 }, { 17,0 }, { 0,17 } };
3266 CvMat* mapping = cvCreateMat(2, 3, CV_32FC1);
3267 cvGetAffineTransform(src, dst, mapping);
3268 printf("===> %f\n", cvGetReal2D(mapping, 0, 0));
3272 CvArr *im = cvLoadImage("checker77.png");
3273 CvPoint2D32f corners[49];
3275 cvFindChessboardCorners(im, cvSize(7,7), corners, &count, 0);
3276 printf("count=%d\n", count);
3279 return PyFloat_FromDouble(0.0);
3282 static PyObject *pycvFindChessboardCorners(PyObject *self, PyObject *args, PyObject *kw)
3285 PyObject *pyobj_image = NULL;
3286 CvSize pattern_size;
3287 PyObject *pyobj_pattern_size = NULL;
3288 cvpoint2d32f_count corners;
3289 int flags = CV_CALIB_CB_ADAPTIVE_THRESH;
3291 const char *keywords[] = { "image", "pattern_size", "flags", NULL };
3292 if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|i", (char**)keywords, &pyobj_image, &pyobj_pattern_size, &flags))
3294 if (!convert_to_CvArr(pyobj_image, &image, "image")) return NULL;
3295 if (!convert_to_CvSize(pyobj_pattern_size, &pattern_size, "pattern_size")) return NULL;
3297 corners.points = new CvPoint2D32f[pattern_size.width * pattern_size.height];
3298 ERRWRAP(r = cvFindChessboardCorners(image, pattern_size, corners.points,&corners.count, flags));
3299 return Py_BuildValue("NN", FROM_int(r), FROM_cvpoint2d32f_count(corners));
3302 // For functions GetSubRect, GetRow, GetCol.
3303 // recipient has a view into donor's data, and needs to share it.
3304 // make recipient use the donor's data, compute the offset,
3305 // and manage reference counts.
3307 static void preShareData(CvArr *donor, CvMat **recipient)
3309 *recipient = cvCreateMatHeader(4, 4, cvGetElemType(donor));
3312 static PyObject *shareData(PyObject *donor, CvArr *pdonor, CvMat *precipient)
3314 PyObject *recipient = (PyObject*)PyObject_NEW(cvmat_t, &cvmat_Type);
3315 ((cvmat_t*)recipient)->a = precipient;
3316 ((cvmat_t*)recipient)->offset = cvPtr1D(precipient, 0) - cvPtr1D(pdonor, 0);
3319 if (is_cvmat(donor)) {
3320 arr_data = ((cvmat_t*)donor)->data;
3321 } else if (is_iplimage(donor)) {
3322 arr_data = ((iplimage_t*)donor)->data;
3324 return (PyObject*)failmsg("Argument 'mat' must be either IplImage or CvMat");
3326 ((cvmat_t*)recipient)->data = arr_data;
3327 Py_INCREF(arr_data);
3331 static PyObject *pycvGetHuMoments(PyObject *self, PyObject *args, PyObject *kw)
3334 PyObject *pyobj_moments = NULL;
3336 if (!PyArg_ParseTuple(args, "O", &pyobj_moments))
3338 if (!convert_to_CvMomentsPTR(pyobj_moments, &moments, "moments")) return NULL;
3340 ERRWRAP(cvGetHuMoments(moments, &r));
3341 return Py_BuildValue("ddddddd", r.hu1, r.hu2, r.hu3, r.hu4, r.hu5, r.hu6, r.hu7);
3344 static PyObject *pycvFitLine(PyObject *self, PyObject *args, PyObject *kw)
3347 PyObject *pyobj_points = NULL;
3354 if (!PyArg_ParseTuple(args, "Oifff", &pyobj_points, &dist_type, ¶m, &reps, &aeps))
3356 if (!convert_to_cvarrseq(pyobj_points, &points, "points")) return NULL;
3357 ERRWRAP(cvFitLine(points.v, dist_type, param, reps, aeps, r));
3359 if (strcmp("opencv-matrix", cvTypeOf(points.v)->type_name) == 0)
3360 dimension = CV_MAT_CN(cvGetElemType(points.v));
3362 // sequence case... don't think there is a sequence of 3d points,
3367 return Py_BuildValue("dddd", r[0], r[1], r[2], r[3]);
3369 return Py_BuildValue("dddddd", r[0], r[1], r[2], r[3], r[4], r[5]);
3372 static PyObject *pycvGetMinMaxHistValue(PyObject *self, PyObject *args, PyObject *kw)
3375 PyObject *pyobj_hist = NULL;
3378 int min_loc[CV_MAX_DIM];
3379 int max_loc[CV_MAX_DIM];
3381 if (!PyArg_ParseTuple(args, "O", &pyobj_hist))
3383 if (!convert_to_CvHistogram(pyobj_hist, &hist, "hist")) return NULL;
3384 ERRWRAP(cvGetMinMaxHistValue(hist, &min_val, &max_val, min_loc, max_loc));
3385 int d = cvGetDims(hist->bins);
3386 PyObject *pminloc = PyTuple_New(d), *pmaxloc = PyTuple_New(d);
3387 for (int i = 0; i < d; i++) {
3388 PyTuple_SetItem(pminloc, i, PyInt_FromLong(min_loc[i]));
3389 PyTuple_SetItem(pmaxloc, i, PyInt_FromLong(max_loc[i]));
3391 return Py_BuildValue("ffNN", min_val, max_val, pminloc, pmaxloc);
3394 static int zero = 0;
3396 /************************************************************************/
3397 /* Generated functions */
3399 #include "generated0.i"
3401 static PyMethodDef methods[] = {
3403 {"CreateHist", pycvCreateHist, METH_VARARGS, "CreateHist(dims, type, ranges, uniform = 1) -> hist"},
3404 {"CreateImage", pycvCreateImage, METH_VARARGS, "CreateImage(size, depth, channels) -> image"},
3405 {"CreateImageHeader", pycvCreateImageHeader, METH_VARARGS, "CreateImageHeader(size, depth, channels) -> image"},
3406 {"CreateMat", pycvCreateMat, METH_VARARGS, "CreateMat(row, cols, type) -> mat"},
3407 {"CreateMatHeader", pycvCreateMatHeader, METH_VARARGS, "CreateMatHeader(rows, cols, type) -> mat"},
3408 {"CreateMatND", pycvCreateMatND, METH_VARARGS, "CreateMatND(dims, type) -> matnd"},
3409 {"CreateMatNDHeader", pycvCreateMatNDHeader, METH_VARARGS, "CreateMatNDHeader(dims, type) -> matnd"},
3410 {"CreateMemStorage", pycvCreateMemStorage, METH_VARARGS, "CreateMemStorage(block_size) -> memstorage"},
3411 {"FindContours", (PyCFunction)pycvFindContours, METH_KEYWORDS, "FindContours(image, storage, mode=CV_RETR_LIST, method=CV_CHAIN_APPROX_SIMPLE, offset=(0, 0)) -> cvseq"},
3412 {"ApproxPoly", (PyCFunction)pycvApproxPoly, METH_KEYWORDS, "ApproxPoly(src_seq, storage, method, parameter=0, parameter2=0) -> None"},
3413 {"CreateData", pycvCreateData, METH_VARARGS, "CreateData(arr) -> None"},
3414 {"GetDims", pycvGetDims, METH_VARARGS, "GetDims(arr) -> dims"},
3415 {"GetImage", pycvGetImage, METH_VARARGS, "GetImage(cvmat) -> image"},
3416 {"GetMat", pycvGetMat, METH_VARARGS, "GetMat(image) -> cvmat"},
3417 {"Reshape", pycvReshape, METH_VARARGS, "Reshape(arr, new_cn, new_rows=0) -> cvmat"},
3418 {"InitLineIterator", pycvInitLineIterator, METH_VARARGS, "InitLineIterator(image, pt1, pt2, connectivity=8, left_to_right=0) -> None"},
3419 {"LoadImage", (PyCFunction)pycvLoadImage, METH_KEYWORDS, "LoadImage(filename, iscolor=CV_LOAD_IMAGE_COLOR)"},
3420 {"SetData", pycvSetData, METH_VARARGS, "SetData(arr, data, step)"},
3421 {"SetMouseCallback", (PyCFunction)pycvSetMouseCallback, METH_KEYWORDS, "SetMouseCallback(window_name, on_mouse, param) -> None"},
3422 {"CreateTrackbar", (PyCFunction)pycvCreateTrackbar, METH_KEYWORDS, "CreateTrackbar(trackbar_name, window_name, value, count, on_change) -> None"},
3423 {"CalcEMD2", (PyCFunction)pycvCalcEMD2, METH_KEYWORDS, "CalcEMD2(signature1, signature2, distance_type, distance_func = None, cost_matrix=None, flow=None, lower_bound=None, userdata = None) -> float"},
3424 {"FindChessboardCorners", (PyCFunction)pycvFindChessboardCorners, METH_KEYWORDS, "FindChessboardCorners(image, pattern_size, flags=CV_CALIB_CB_ADAPTIVE_THRESH) -> success,corners"},
3425 {"FitLine", (PyCFunction)pycvFitLine, METH_KEYWORDS, "FitLine(points, dist_type, param, reps, aeps) -> line"},
3426 {"Subdiv2DLocate", pycvSubdiv2DLocate, METH_VARARGS, "Subdiv2DLocate(subdiv, pt) -> (loc, where)"},
3427 {"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)"},
3428 {"ClipLine", (PyCFunction)pycvClipLine, METH_KEYWORDS, "ClipLine(img, pt1, pt2) -> (clipped_pt1, clipped_pt2)"},
3429 {"GetHuMoments", (PyCFunction)pycvGetHuMoments, METH_KEYWORDS, "GetHuMoments(cvmoments) -> (h1, h2, h3, h4, h5, h5, h7)"},
3430 {"GetMinMaxHistValue", (PyCFunction)pycvGetMinMaxHistValue, METH_KEYWORDS, "GetMinMaxHistValue(hist) -> min_val,max_val,min_loc,max_loc"},
3431 {"WaitKey", (PyCFunction)pycvWaitKey, METH_KEYWORDS, "WaitKey(delay=0) -> int"},
3433 {"temp_test", temp_test, METH_VARARGS},
3435 #include "generated1.i"
3440 /************************************************************************/
3443 static int to_ok(PyTypeObject *to)
3445 to->tp_alloc = PyType_GenericAlloc;
3446 to->tp_new = PyType_GenericNew;
3447 to->tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE;
3448 return (PyType_Ready(to) == 0);
3451 #define MKTYPE(NAME) do { NAME##_specials(); if (!to_ok(&NAME##_Type)) return; } while (0)
3454 #if defined WIN32 || defined _WIN32
3455 __declspec(dllexport)
3461 cvSetErrMode(CV_ErrModeParent);
3464 MKTYPE(cvcontourtree);
3466 MKTYPE(cvhaarclassifiercascade);
3467 MKTYPE(cvhistogram);
3468 MKTYPE(cvlineiterator);
3471 MKTYPE(cvmemstorage);
3473 MKTYPE(cvpositobject);
3474 MKTYPE(cvsubdiv2dedge);
3479 MKTYPE(cvsubdiv2dpoint);
3480 MKTYPE(cvvideowriter);
3481 MKTYPE(iplconvkernel);
3484 m = Py_InitModule(MODULESTR"", methods);
3485 d = PyModule_GetDict(m);
3487 opencv_error = PyErr_NewException((char*)MODULESTR".error", NULL, NULL);
3488 PyDict_SetItemString(d, "error", opencv_error);
3490 PyDict_SetItemString(d, "iplimage", (PyObject*)&iplimage_Type);
3491 PyDict_SetItemString(d, "cvmat", (PyObject*)&cvmat_Type);
3493 #define PUBLISH(I) PyDict_SetItemString(d, #I, PyInt_FromLong(I))
3495 PUBLISH(IPL_DEPTH_8U);
3496 PUBLISH(IPL_DEPTH_8S);
3497 PUBLISH(IPL_DEPTH_16U);
3498 PUBLISH(IPL_DEPTH_16S);
3499 PUBLISH(IPL_DEPTH_32S);
3500 PUBLISH(IPL_DEPTH_32F);
3501 PUBLISH(IPL_DEPTH_64F);
3502 PUBLISH(CV_LOAD_IMAGE_COLOR);
3503 PUBLISH(CV_LOAD_IMAGE_GRAYSCALE);
3504 PUBLISH(CV_LOAD_IMAGE_UNCHANGED);
3505 PUBLISH(CV_HIST_ARRAY);
3506 PUBLISH(CV_HIST_SPARSE);
3535 PUBLISH(CV_NEXT_AROUND_ORG);
3536 PUBLISH(CV_NEXT_AROUND_DST);
3537 PUBLISH(CV_PREV_AROUND_ORG);
3538 PUBLISH(CV_PREV_AROUND_DST);
3539 PUBLISH(CV_NEXT_AROUND_LEFT);
3540 PUBLISH(CV_NEXT_AROUND_RIGHT);
3541 PUBLISH(CV_PREV_AROUND_LEFT);
3542 PUBLISH(CV_PREV_AROUND_RIGHT);
3544 PUBLISH(CV_WINDOW_AUTOSIZE);
3546 PUBLISH(CV_PTLOC_INSIDE);
3547 PUBLISH(CV_PTLOC_ON_EDGE);
3548 PUBLISH(CV_PTLOC_VERTEX);
3549 PUBLISH(CV_PTLOC_OUTSIDE_RECT);
3551 #include "generated2.i"
3555 int sizes[] = { 10 } ;
3556 float ranges[] = { 0.0, 1.0 };
3557 // CvHistogram*h = cvCreateHist(1, sizes, CV_HIST_ARRAY);
3560 CvHistogram*h = cvMakeHistHeaderForArray(1, sizes, &H, data);
3561 printf("h->type = %08x\n", h->type);
3562 printf("h->bins = %p\n", h->bins);
3563 printf("h->mat = %p\n", &(h->mat));