nvbuf_utils.h

转到此文件的文档。

/*
 * 版权所有 (c) 2016-2021, NVIDIA CORPORATION。保留所有权利。
 *
 * NVIDIA 公司及其许可方保留本软件、相关文档中的所有知识产权
 * 和专有权利，以及对其的任何修改。未经 NVIDIA 公司明确
 * 许可协议，严禁使用、复制、披露或分发本软件及相关文档。
 * 分发本软件和相关文档。
 * 许可协议，严禁使用、复制、披露或分发本软件及相关文档。
 */

#ifndef _NVBUF_UTILS_H_
#define _NVBUF_UTILS_H_

#ifdef __cplusplus
extern "C"
{
#endif

#include <EGL/egl.h>
#include <EGL/eglext.h>
#include <errno.h>
#include <stdbool.h>

#define MAX_NUM_PLANES 4

#define MAX_COMPOSITE_FRAME 16

#define NVBUF_CHROMA_SUBSAMPLING_HORIZ_DEFAULT 0
#define NVBUF_CHROMA_SUBSAMPLING_VERT_DEFAULT 1

#define NVBUF_MAX_SYNCOBJ_PARAMS 5

#define NVBUFFER_SYNCPOINT_WAIT_INFINITE 0xFFFFFFFF

typedef enum
{
 NvBufferPayload_SurfArray,
 NvBufferPayload_MemHandle,
} NvBufferPayloadType;

typedef enum
{
 NvBufferDisplayScanFormat_Progressive = 0,
 NvBufferDisplayScanFormat_Interlaced,
} NvBufferDisplayScanFormat;

typedef enum
{
 NvBufferLayout_Pitch,
 NvBufferLayout_BlockLinear,
} NvBufferLayout;

typedef enum
{
 NvBufferMem_Read,
 NvBufferMem_Write,
 NvBufferMem_Read_Write,
} NvBufferMemFlags;

typedef enum
{
 NvBufferTag_NONE = 0x0,
 NvBufferTag_CAMERA = 0x200,
 NvBufferTag_JPEG = 0x1500,
 NvBufferTag_PROTECTED = 0x1504,
 NvBufferTag_VIDEO_ENC = 0x1200,
 NvBufferTag_VIDEO_DEC = 0x1400,
 NvBufferTag_VIDEO_CONVERT = 0xf01,
} NvBufferTag;

typedef enum
{
 NvBufferColorFormat_YUV420,
 NvBufferColorFormat_YVU420,
 NvBufferColorFormat_YUV422,
 NvBufferColorFormat_YUV420_ER,
 NvBufferColorFormat_YVU420_ER,
 NvBufferColorFormat_NV12,
 NvBufferColorFormat_NV12_ER,
 NvBufferColorFormat_NV21,
 NvBufferColorFormat_NV21_ER,
 NvBufferColorFormat_UYVY,
 NvBufferColorFormat_UYVY_ER,
 NvBufferColorFormat_VYUY,
 NvBufferColorFormat_VYUY_ER,
 NvBufferColorFormat_YUYV,
 NvBufferColorFormat_YUYV_ER,
 NvBufferColorFormat_YVYU,
 NvBufferColorFormat_YVYU_ER,
 NvBufferColorFormat_ABGR32,
 NvBufferColorFormat_XRGB32,
 NvBufferColorFormat_ARGB32,
 NvBufferColorFormat_NV12_10LE,
 NvBufferColorFormat_NV12_10LE_709,
 NvBufferColorFormat_NV12_10LE_709_ER,
 NvBufferColorFormat_NV12_10LE_2020,
 NvBufferColorFormat_NV21_10LE,
 NvBufferColorFormat_NV12_12LE,
 NvBufferColorFormat_NV12_12LE_2020,
 NvBufferColorFormat_NV21_12LE,
 NvBufferColorFormat_YUV420_709,
 NvBufferColorFormat_YUV420_709_ER,
 NvBufferColorFormat_NV12_709,
 NvBufferColorFormat_NV12_709_ER,
 NvBufferColorFormat_YUV420_2020,
 NvBufferColorFormat_NV12_2020,
 NvBufferColorFormat_YUV444,
 NvBufferColorFormat_SignedR16G16,
 NvBufferColorFormat_A32,
 NvBufferColorFormat_GRAY8,
 NvBufferColorFormat_NV16,
 NvBufferColorFormat_NV16_10LE,
 NvBufferColorFormat_NV24,
 NvBufferColorFormat_NV24_10LE,
 NvBufferColorFormat_NV16_ER,
 NvBufferColorFormat_NV24_ER,
 NvBufferColorFormat_NV16_709,
 NvBufferColorFormat_NV24_709,
 NvBufferColorFormat_NV16_709_ER,
 NvBufferColorFormat_NV24_709_ER,
 NvBufferColorFormat_NV24_10LE_709,
 NvBufferColorFormat_NV24_10LE_709_ER,
 NvBufferColorFormat_NV24_10LE_2020,
 NvBufferColorFormat_NV24_12LE_2020,
 NvBufferColorFormat_RGBA_10_10_10_2_709,
 NvBufferColorFormat_RGBA_10_10_10_2_2020,
 NvBufferColorFormat_BGRA_10_10_10_2_709,
 NvBufferColorFormat_BGRA_10_10_10_2_2020,
 NvBufferColorFormat_Invalid,
} NvBufferColorFormat;

typedef enum
{
 NvBufferTransform_None,
 NvBufferTransform_Rotate90,
 NvBufferTransform_Rotate180,
 NvBufferTransform_Rotate270,
 NvBufferTransform_FlipX,
 NvBufferTransform_FlipY,
 NvBufferTransform_Transpose,
 NvBufferTransform_InvTranspose,
} NvBufferTransform_Flip;

typedef enum
{
 NvBufferTransform_Filter_Nearest,
 NvBufferTransform_Filter_Bilinear,
 NvBufferTransform_Filter_5_Tap,
 NvBufferTransform_Filter_10_Tap,
 NvBufferTransform_Filter_Smart,
 NvBufferTransform_Filter_Nicest,
} NvBufferTransform_Filter;

typedef enum {
 NVBUFFER_TRANSFORM_CROP_SRC = 1,
 NVBUFFER_TRANSFORM_CROP_DST = 1 << 1,
 NVBUFFER_TRANSFORM_FILTER = 1 << 2,
 NVBUFFER_TRANSFORM_FLIP = 1 << 3,
} NvBufferTransform_Flag;

typedef enum {
 NVBUFFER_COMPOSITE = 1,
 NVBUFFER_BLEND = 1 << 1,
 NVBUFFER_COMPOSITE_FILTER = 1 << 2,
} NvBufferComposite_Flag;

typedef struct _NvBufferSyncObjParams
{
 uint32_t syncpointID;
 uint32_t value;
}NvBufferSyncObjParams;

typedef struct _NvBufferSyncObjRec
{
 NvBufferSyncObjParams insyncobj[NVBUF_MAX_SYNCOBJ_PARAMS];
 uint32_t num_insyncobj;
 NvBufferSyncObjParams outsyncobj;
 uint32_t use_outsyncobj;
}NvBufferSyncObj;

typedef struct
{
 float r;
 float g;
 float b;
}NvBufferCompositeBackground;

typedef struct
{
 uint32_t top;
 uint32_t left;
 uint32_t width;
 uint32_t height;
}NvBufferRect;

typedef struct _NvBufferSession * NvBufferSession;

typedef struct _NvBufferChromaSubSamplingParams
{
 uint8_t chromaLocHoriz;
 uint8_t chromaLocVert;
}NvBufferChromaSubsamplingParams;

#define NVBUF_CHROMA_SUBSAMPLING_PARAMS_DEFAULT \
 { \
 NVBUF_CHROMA_SUBSAMPLING_HORIZ_DEFAULT, \
 NVBUF_CHROMA_SUBSAMPLING_VERT_DEFAULT \
 }

typedef struct _NvBufferCreateParams
{
 int32_t width;
 int32_t height;
 NvBufferPayloadType payloadType;
 int32_t memsize;
 NvBufferLayout layout;
 NvBufferColorFormat colorFormat;
 NvBufferTag nvbuf_tag;
}NvBufferCreateParams;

typedef struct _NvBufferParams
{
 uint32_t dmabuf_fd;
 void *nv_buffer;
 NvBufferPayloadType payloadType;
 int32_t memsize;
 uint32_t nv_buffer_size;
 NvBufferColorFormat pixel_format;
 uint32_t num_planes;
 uint32_t width[MAX_NUM_PLANES];
 uint32_t height[MAX_NUM_PLANES];
 uint32_t pitch[MAX_NUM_PLANES];
 uint32_t offset[MAX_NUM_PLANES];
 uint32_t psize[MAX_NUM_PLANES];
 uint32_t layout[MAX_NUM_PLANES];
}NvBufferParams;

typedef struct _NvBufferParamsEx
{
 NvBufferParams params;
 int32_t startofvaliddata;
 int32_t sizeofvaliddatainbytes;
 NvBufferDisplayScanFormat scanformat[MAX_NUM_PLANES];
 uint32_t secondfieldoffset[MAX_NUM_PLANES];
 uint32_t blockheightlog2[MAX_NUM_PLANES];
 uint32_t physicaladdress[MAX_NUM_PLANES];
 uint64_t flags[MAX_NUM_PLANES];
 void *payloadmetaInfo;
 NvBufferChromaSubsamplingParams chromaSubsampling;
 bool is_protected;
 NvBufferSyncObj syncobj;
 void *reserved;
}NvBufferParamsEx;

typedef struct _NvBufferCompositeParams
{
 uint32_t composite_flag;
 uint32_t input_buf_count;
 NvBufferTransform_Filter composite_filter[MAX_COMPOSITE_FRAME];
 float dst_comp_rect_alpha[MAX_COMPOSITE_FRAME];
 NvBufferRect src_comp_rect[MAX_COMPOSITE_FRAME];
 NvBufferRect dst_comp_rect[MAX_COMPOSITE_FRAME];
 NvBufferCompositeBackground composite_bgcolor;
 NvBufferSession session;
}NvBufferCompositeParams;

typedef struct _NvBufferTransformParams
{
 uint32_t transform_flag;
 NvBufferTransform_Flip transform_flip;
 NvBufferTransform_Filter transform_filter;
 NvBufferRect src_rect;
 NvBufferRect dst_rect;
 NvBufferSession session;
}NvBufferTransformParams;


int NvBufferSyncObjWait (NvBufferSyncObjParams *syncobj_params, unsigned int timeout);

int NvBufferGetSize (void);

EGLImageKHR NvEGLImageFromFd (EGLDisplay display, int dmabuf_fd);

int NvDestroyEGLImage (EGLDisplay display, EGLImageKHR eglImage);

int NvBufferCreate (int *dmabuf_fd, int width, int height,
 NvBufferLayout layout, NvBufferColorFormat colorFormat);

int NvBufferCreateEx (int *dmabuf_fd, NvBufferCreateParams *input_params);

int NvBufferCreateInterlace (int *dmabuf_fd, NvBufferCreateParams *input_params);

int NvBufferCreateWithChromaLoc (int *dmabuf_fd, NvBufferCreateParams *input_params, NvBufferChromaSubsamplingParams *chromaSubsampling);

int NvBufferGetParams (int dmabuf_fd, NvBufferParams *params);

int NvBufferGetParamsEx (int dmabuf_fd, NvBufferParamsEx *exparams);

int NvBufferDestroy (int dmabuf_fd);

int ExtractFdFromNvBuffer (void *nvbuf, int *dmabuf_fd);

int NvReleaseFd (int dmabuf_fd);

int NvBufferMemSyncForCpu (int dmabuf_fd, unsigned int plane, void **pVirtAddr);

int NvBufferMemSyncForCpuEx (int dmabuf_fd, NvBufferParamsEx *exparams, unsigned int plane, void **pVirtAddr);

int NvBufferMemSyncForDevice (int dmabuf_fd, unsigned int plane, void **pVirtAddr);

int NvBufferMemSyncForDeviceEx (int dmabuf_fd, NvBufferParamsEx *exparams, unsigned int plane, void **pVirtAddr);

int NvBufferMemMap (int dmabuf_fd, unsigned int plane, NvBufferMemFlags memflag, void **pVirtAddr);

int NvBufferMemMapEx (int dmabuf_fd, NvBufferParamsEx *exparams, unsigned int plane, NvBufferMemFlags memflag, void **pVirtAddr);

int NvBufferMemUnMap (int dmabuf_fd, unsigned int plane, void **pVirtAddr);

int NvBufferMemUnMapEx (int dmabuf_fd, NvBufferParamsEx *exparams, unsigned int plane, void **pVirtAddr);

int NvBuffer2Raw (int dmabuf_fd, unsigned int plane, unsigned int out_width, unsigned int out_height, unsigned char *ptr);

int Raw2NvBuffer (unsigned char *ptr, unsigned int plane, unsigned int in_width, unsigned int in_height, int dmabuf_fd);

NvBufferSession NvBufferSessionCreate(void);

void NvBufferSessionDestroy(NvBufferSession session);

int NvBufferTransform (int src_dmabuf_fd, int dst_dmabuf_fd, NvBufferTransformParams *transform_params);

int NvBufferTransformEx (int src_dmabuf_fd, NvBufferParamsEx *input_params, int dst_dmabuf_fd, NvBufferParamsEx *output_params, NvBufferTransformParams *transform_params);

int NvBufferTransformAsync (int src_dmabuf_fd, int dst_dmabuf_fd, NvBufferTransformParams *transform_params, NvBufferSyncObj *syncobj);


int NvBufferComposite (int *src_dmabuf_fds, int dst_dmabuf_fd, NvBufferCompositeParams *composite_params);

#ifdef __cplusplus
}
#endif

#endif