#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
namespace py = pybind11;

#include <opencv2/core.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/python/pyapi.hpp>

PyObject * draw_rect(py::array in) {
    
    cv::Mat mat;
    
    if (!PyOpenCV_NdarrayToMat(in.ptr(), mat, "in"))
        return NULL;
    
    cv::Point pt1, pt2;
    pt1.x = 10;
    pt1.y = 10;
    pt2.x = 100;
    pt2.y = 100;
    
    rectangle(mat, pt1, pt2, cv::Scalar(255, 0, 0));
    
    return PyOpenCV_MatToNdarray(mat);
}


PYBIND11_PLUGIN(test_package) {
    
    
    py::module m("test_package");
    
    m.def("draw_rect", &draw_rect);

#ifdef VERSION_INFO
    m.attr("__version__") = py::str(VERSION_INFO);
#else
    m.attr("__version__") = py::str("dev");
#endif

    import_opencv();
    return m.ptr();
}