Method Export

python/mod_cvcuda/nvcv/Array.cpp:329–365

}

void Array::Export(py::module &m)
{
    using namespace py::literals;

    using CreateFromLengthPtr = std::shared_ptr<Array> (*)(int64_t, nvcv::DataType);
    using CreateFromShapePtr  = std::shared_ptr<Array> (*)(const Shape &, nvcv::DataType);

    using ResizeLengthPtr = std::shared_ptr<Array> (Array::*)(int64_t DataType);
    using ResizeShapePtr  = std::shared_ptr<Array> (Array::*)(Shape);

    using ResizeArrayLengthPtr = std::shared_ptr<Array> (*)(Array &, int64_t DataType);
    using ResizeArrayShapePtr  = std::shared_ptr<Array> (*)(Array &, Shape);

    py::class_<Array, std::shared_ptr<Array>, Container>(m, "Array", "Array")
        .def(py::init(static_cast<CreateFromLengthPtr>(&Array::Create)), "length"_a, "dtype"_a,
             "Create a Array object with the given length and data type.")
        .def(py::init(static_cast<CreateFromShapePtr>(&Array::Create)), "shape"_a, "dtype"_a,
             "Create a Array object with the given shape and data type.")
        .def_property_readonly("shape", &Array::shape, "The shape of the Array.")
        .def_property_readonly("dtype", &Array::dtype, "The data type of the Array.")
        // numpy and others use ndim, let's be consistent with them in python.
        // It's not a requirement to be consistent between NVCV Python and C/C++.
        // Each language use whatever is appropriate (and expected) in their environment.
        .def_property_readonly("ndim", &Array::rank, "The number of dimensions of the Array.")
        .def("cuda", &Array::cuda, "Reference to the Array on the CUDA device.")
        .def("resize", static_cast<ResizeLengthPtr>(&Array::Resize), "length"_a,
             "Produces an array pointing to the same data but with a new length.")
        .def("resize", static_cast<ResizeShapePtr>(&Array::Resize), "shape"_a,
             "Produces an array pointing to the same data but with a new shape.")
        .def("__repr__", &util::ToString<Array>, "Return the string representation of the Array object.");

    m.def("as_array", &Array::Wrap, "buffer"_a, "Wrap an existing buffer into a Array object with the given layout.");
    m.def("resize", static_cast<ResizeArrayLengthPtr>(&Array::ResizeArray), "array"_a, "length"_a,
          "Produces an array pointing to the same data but with a new length.");
    m.def("resize", static_cast<ResizeArrayShapePtr>(&Array::ResizeArray), "array"_a, "shape"_a,
          "Produces an array pointing to the same data but with a new shape.");
}

} // namespace nvcvpy::priv