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