Function PYBIND11_MODULE

python/mod_cvcuda/Main.cpp:60–182

namespace py = pybind11;

PYBIND11_MODULE(_cvcuda, m)
{
    m.attr("__name__")    = "cvcuda";
    m.attr("__version__") = CVCUDA_VERSION_STRING;

    {
        using namespace nvcvpy::priv;

        // Submodule used for additional functionality needed only by tests
        // so that some level of white-box testing is possible.
        //
        // This guarantees a clear separation of public and private APIs.
        // Users are restricted to the public API, allowing us to change the
        // private APIs as needed, without worring in breaking user's code.
        //
        // To retrieve it from inside the Export call, include "Definitions.hpp"
        // and call:
        //    py::module_ internal = m.attr(INTERNAL_SUBMODULE_NAME);
        // Functions and other properties can be then exposed as usual, e.g.
        //    internal.def("foo", &Foo");
        // and accessed in python as you'd expect:
        //    cvcuda.internal.foo()
        m.def_submodule(INTERNAL_SUBMODULE_NAME);

        // Export NVCV core types (Tensor, Image, Format, etc.)
        // Supporting objects
        ExportColorSpec(m);
        ExportImageFormat(m);
        ExportDataType(m);
        ExportRect(m);
        ExportThreadScope(m);

        // Core Entities
        ExportCAPI(m);
        Resource::Export(m);
        Cache::Export(m);
        Container::Export(m);
        ExternalBuffer::Export(m);

        // Objects
        Tensor::Export(m);
        TensorBatch::Export(m);
        Image::Export(m);
        ImageBatchVarShape::Export(m);
        Stream::Export(m);
    }

    {
        using namespace cvcudapy;

        // doctag: Non-Operators
        // Operators' auxiliary entities
        ExportAdaptiveThresholdType(m);
        ExportBorderType(m);
        ExportBoxBlur(m);
        ExportChannelManipType(m);
        ExportOSD(m);
        ExportColorConversionCode(m);