OpenCVInterop.hpp

前往此文件的文档。

/*
 * 版权所有 2020 NVIDIA Corporation。保留所有权利。
 *
 * 给被许可方的通知：
 *
 * 此源代码和/或文档（“许可交付物”）受美国和
 * 国际版权法项下的 NVIDIA 知识产权保护。
 *
 * 此处包含的许可交付物是 NVIDIA 的专有和
 * 机密信息，并根据 NVIDIA 与被许可方之间（“许可协议”）的 NVIDIA 软件许可协议条款和
 * 条件或被许可方电子接受的协议提供。尽管许可协议中有任何相反的条款或条件，
 * 未经 NVIDIA 明确书面同意，禁止向任何第三方复制或披露
 * 许可交付物。
 *
 * 尽管许可协议中有任何相反的条款或条件，
 * NVIDIA 对这些许可交付物用于任何目的的适用性不做任何陈述。
 * 它们按“原样”提供，不附带任何明示或暗示的担保。
 * NVIDIA 声明对这些许可交付物不作任何担保，
 * 包括所有关于适销性、非侵权性以及适用于特定用途的暗示担保。
 * 尽管许可协议中有任何相反的条款或条件，
 * 在任何情况下，NVIDIA 均不对任何
 * 特殊、间接、附带或后果性损害，或任何
 * 因使用、数据或利润损失而导致的任何损害负责，
 * 无论是在合同诉讼、过失或其他侵权行为中，
 * 因使用或执行这些许可交付物而引起或与之相关的。
 *
 * 美国政府最终用户。这些许可交付物是
 * “商业项目”，该术语在 48 C.F.R. 2.101（OCT
 * 1995）中定义，包括“商业计算机软件”和“商业
 * 计算机软件文档”，这些术语在 48
 * C.F.R. 12.212（SEPT 1995）中使用，并且仅作为商业最终项目提供给美国政府。
 * 与 48 C.F.R.12.212 和
 * 48 C.F.R. 227.7202-1 至 227.7202-4（1995 年 6 月）一致，所有
 * 美国政府最终用户均获得许可交付物，
 * 仅具有此处规定的权利。
 *
 * 在个人和商业软件中对许可交付物的任何使用，都必须在用户文档和代码的内部
 * 注释中包含上述免责声明和美国政府最终用户声明。
 * 用户声明。
 */

#ifndef NV_VPI_OPENCV_INTEROP_HPP
#define NV_VPI_OPENCV_INTEROP_HPP

#include "detail/OpenCVUtils.hpp"

#include "Image.h"

#include <opencv2/core/core.hpp>

#include <vector>

inline VPIStatus vpiImageCreateWrapperOpenCVMat(const cv::Mat &mat, VPIImageFormat fmt, uint64_t flags, VPIImage *img)
{
 VPIImageData imgData = {};

 VPIStatus status = ::nv::vpi::detail::FillImageData(mat, fmt, &imgData);
 if (status != VPI_SUCCESS)
 {
 return status;
 }

 return vpiImageCreateWrapper(&imgData, nullptr, flags, img);
}

inline VPIStatus vpiImageCreateWrapperOpenCVMat(const cv::Mat &mat, uint64_t flags, VPIImage *img)
{
 VPIImageData imgData = {};
 VPIStatus status = ::nv::vpi::detail::FillImageData(mat, &imgData);
 if (status != VPI_SUCCESS)
 {
 return status;
 }

 return vpiImageCreateWrapper(&imgData, nullptr, flags, img);
}

inline VPIStatus vpiImageSetWrappedOpenCVMat(VPIImage img, const cv::Mat &mat)
{
 VPIImageFormat fmt;
 VPIStatus status = vpiImageGetFormat(img, &fmt);
 if (status != VPI_SUCCESS)
 {
 return status;
 }

 VPIImageData imgData = {};
 status = ::nv::vpi::detail::FillImageData(mat, fmt, &imgData);
 if (status != VPI_SUCCESS)
 {
 return status;
 }

 return vpiImageSetWrapper(img, &imgData);
}

inline VPIStatus vpiImageCreateWrapperOpenCVMatPlanes(const std::vector<cv::Mat> &matPlanes, VPIImageFormat fmt,
 uint64_t flags, VPIImage *img)
{
 VPIImageData imgData = {};
 VPIStatus status = ::nv::vpi::detail::FillImageData(matPlanes, fmt, &imgData);
 if (status != VPI_SUCCESS)
 {
 return status;
 }

 return vpiImageCreateWrapper(&imgData, nullptr, flags, img);
}

inline VPIStatus vpiImageSetWrappedOpenCVMatPlanes(VPIImage img, const std::vector<cv::Mat> &matPlanes)
{
 VPIImageFormat fmt;
 VPIStatus status = vpiImageGetFormat(img, &fmt);
 if (status != VPI_SUCCESS)
 {
 return status;
 }

 VPIImageData imgData = {};
 status = ::nv::vpi::detail::FillImageData(matPlanes, fmt, &imgData);
 if (status != VPI_SUCCESS)
 {
 return status;
 }

 return vpiImageSetWrapper(img, &imgData);
}

inline VPIStatus vpiImageDataExportOpenCVMat(const VPIImageData &imgData, cv::Mat *mat)
{
 if (mat == NULL)
 {
 return VPI_ERROR_INVALID_ARGUMENT;
 }

 return ::nv::vpi::detail::ExportOpenCVMat(imgData, *mat);
}

inline VPIStatus vpiImageDataExportOpenCVMatPlanes(const VPIImageData &imgData, std::vector<cv::Mat> *matPlanes)
{
 if (matPlanes == NULL)
 {
 return VPI_ERROR_INVALID_ARGUMENT;
 }

 return ::nv::vpi::detail::ExportOpenCVMatPlanes(imgData, *matPlanes);
}

inline VPIStatus vpiImageDataImportOpenCVMat(const cv::Mat &mat, VPIImageData *imgData)
{
 if (imgData == NULL)
 {
 return VPI_ERROR_INVALID_ARGUMENT;
 }

 return ::nv::vpi::detail::FillImageData(mat, imgData);
}

inline VPIStatus vpiImageDataImportOpenCVMat(const cv::Mat &mat, VPIImageFormat fmt, VPIImageData *imgData)
{
 if (imgData == NULL)
 {
 return VPI_ERROR_INVALID_ARGUMENT;
 }

 return ::nv::vpi::detail::FillImageData(mat, fmt, imgData);
}

inline VPIStatus vpiImageDataImportOpenCVMatPlanes(const std::vector<cv::Mat> &matPlanes, VPIImageFormat fmt,
 VPIImageData *imgData)
{
 if (imgData == NULL)
 {
 return VPI_ERROR_INVALID_ARGUMENT;
 }

 return ::nv::vpi::detail::FillImageData(matPlanes, fmt, imgData);
}

#endif // NV_VPI_OPENCV_INTEROP_HPP
Image.h
用于处理 VPI 图像的函数和结构。
VPIImageFormat
uint64_t VPIImageFormat
预定义的图像格式。
定义: ImageFormat.h:94
VPIImage
struct VPIImageImpl * VPIImage