pyboost_cv_mat_converter.hpp

/****************************************************************************
 *  hipipe library
 *  Copyright (c) 2014, Gregory Kramida
 *  Modified by Filip Matzner, Adam Blazek
 *
 *  This file is distributed under the MIT License.
 *  See the accompanying file LICENSE.txt for the complete license agreement.
 ****************************************************************************/
 
#pragma once
#if defined HIPIPE_BUILD_PYTHON && defined HIPIPE_BUILD_PYTHON_OPENCV
 
// this header has to be included before the numpy header
#include <hipipe/core/python/initialize.hpp>
 
#include <Python.h>
#include <boost/python.hpp>
#include <numpy/ndarrayobject.h>
#include <opencv2/core/core.hpp>
 
#include <cstdio>
 
namespace hipipe::python::utility {
 
struct matToNDArrayBoostConverter {
    static PyObject* convert(cv::Mat const& m);
};
 
struct matFromNDArrayBoostConverter {
 
    matFromNDArrayBoostConverter();
 
    // check if PyObject is an array and can be converted to OpenCV matrix
    static void* convertible(PyObject* object);
 
    // construct a Mat from an NDArray object
    static void construct(PyObject* object,
                          boost::python::converter::rvalue_from_python_stage1_data* data);
};
 
}  // namespace hipipe::python::utility
 
#endif  // defined HIPIPE_BUILD_PYTHON && defined HIPIPE_BUILD_PYTHON_OPENCV