FilesExpand file tree

 master

/

conversion.h

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

63 lines (49 loc) · 1.31 KB

 master

/

conversion.h

Copy path

Top

File metadata and controls

• Code
• Blame

63 lines (49 loc) · 1.31 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

#ifndef CONVERSION_H

#define CONVERSION_H

#include "opencv2/imgproc/imgproc.hpp"

#include "opencv2/highgui/highgui.hpp"

#include "opencv2/core/core.hpp"

#include "/usr/include/python3.5/Python.h"

#include "/usr/include/python3.5/numpy/ndarrayobject.h"

#define NUMPY_IMPORT_ARRAY_RETVAL

namespace NumpyAPI

{

static PyObject* opencv_error = 0;

static int failmsg(const char *fmt, ...);

class PyAllowThreads;

class PyEnsureGIL;

#define ERRWRAP2(expr) \

try \

{ \

PyAllowThreads allowThreads; \

expr; \

} \

catch (const cv::Exception &e) \

{ \

PyErr_SetString(opencv_error, e.what()); \

return 0; \

}

static PyObject* failmsgp(const char *fmt, ...);

static size_t REFCOUNT_OFFSET = (size_t)&(((PyObject*)0)->ob_refcnt) +

(0x12345678 != *(const size_t*)"\x78\x56\x34\x12\0\0\0\0\0")*sizeof(int);

static inline PyObject* pyObjectFromRefcount(const int* refcount)

{

return (PyObject*)((size_t)refcount - REFCOUNT_OFFSET);

}

static inline int* refcountFromPyObject(const PyObject* obj)

{

return (int*)((size_t)obj + REFCOUNT_OFFSET);

}

class NumpyAllocator;

enum { ARG_NONE = 0, ARG_MAT = 1, ARG_SCALAR = 2 };

class NDArrayConverter

{

private:

void init();

public:

NDArrayConverter();

cv::Mat toMat(const PyObject* o);

PyObject* toNDArray(const cv::Mat& mat);

};

}

#endif // CONVERSION_H