Method WrapExternalBufferVector

python/mod_cvcuda/nvcv/Image.cpp:634–676

}

std::shared_ptr<Image> Image::WrapExternalBufferVector(std::vector<py::object> buffers, nvcv::ImageFormat fmt)
{
    std::vector<std::shared_ptr<ExternalBuffer>> spBuffers;
    for (auto &obj : buffers)
    {
        std::shared_ptr<ExternalBuffer> buffer = cast_py_object_as<ExternalBuffer>(obj);
        if (!buffer)
            throw std::runtime_error("Input buffer doesn't provide cuda_array_interface or DLPack interfaces");
        spBuffers.push_back(std::move(buffer));
    }

    std::vector<DLPackTensor> bufinfos;

    for (size_t i = 0; i < spBuffers.size(); ++i)
    {
        bufinfos.emplace_back(spBuffers[i]->dlTensor());
    }

    nvcv::ImageDataStridedCuda imgData = CreateNVCVImageDataCuda(std::move(bufinfos), fmt);

    // This is the key of an image wrapper.
    // All image wrappers have the same key.
    Image::Key key;

    std::shared_ptr<CacheItem> item = Cache::Instance().fetchOne(key);

    // None found?
    if (!item)
    {
        // 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);
        return img;
    }
    else
    {
        std::shared_ptr<Image> img = std::static_pointer_cast<Image>(item);
        img->setWrapData(std::move(spBuffers), imgData);
        return img;
    }
}

void Image::setWrapData(std::vector<std::shared_ptr<ExternalBuffer>> bufs, const nvcv::ImageDataStridedCuda &imgData)
{