Method WrapExternalBufferMany

python/mod_cvcuda/nvcv/Image.cpp:588–632

}

std::vector<std::shared_ptr<Image>> Image::WrapExternalBufferMany(std::vector<std::shared_ptr<ExternalBuffer>> &buffers,
                                                                  nvcv::ImageFormat                             fmt)
{
    // This is the key of an image wrapper.
    // All image wrappers have the same key.
    Image::Key key;

    std::vector<std::shared_ptr<CacheItem>> items = Cache::Instance().fetch(key);

    std::vector<std::shared_ptr<Image>> out;
    out.reserve(buffers.size());

    for (size_t i = 0; i < buffers.size(); ++i)
    {
        std::vector<std::shared_ptr<ExternalBuffer>> spBuffers;
        spBuffers.push_back(buffers[i]);

        if (!spBuffers.back())
            throw std::runtime_error("Input buffer doesn't provide cuda_array_interface or DLPack interfaces");

        std::vector<DLPackTensor> bufinfos;
        bufinfos.emplace_back(spBuffers[0]->dlTensor());
        nvcv::ImageDataStridedCuda imgData = CreateNVCVImageDataCuda(bufinfos, fmt);

        // None found?
        if (items.empty())
        {
            // Need to add wrappers into cache so that they don't get destroyed by
            // the cuda stream when they're last used, and python script isn't
            // holding a reference to them. If we don't do it, things might break.
            std::shared_ptr<Image> img(new Image(std::move(spBuffers), imgData));
            Cache::Instance().add(*img);
            out.push_back(img);
        }
        else
        {
            std::shared_ptr<Image> img = std::static_pointer_cast<Image>(items.back());
            items.pop_back();
            img->setWrapData(std::move(spBuffers), imgData);
            out.push_back(img);
        }
    }

    return out;
}

std::shared_ptr<Image> Image::WrapExternalBufferVector(std::vector<py::object> buffers, nvcv::ImageFormat fmt)
{