nvbuf_utils.h

Go to the documentation of this file.

/*
 * Copyright (c) 2016-2019, NVIDIA CORPORATION.  All rights reserved.
 *
 * NVIDIA Corporation and its licensors retain all intellectual property
 * and proprietary rights in and to this software, related documentation
 * and any modifications thereto.  Any use, reproduction, disclosure or
 * distribution of this software and related documentation without an express
 * license agreement from NVIDIA Corporation is strictly prohibited.
 */

#ifndef _NVBUF_UTILS_H_
#define _NVBUF_UTILS_H_

#ifdef __cplusplus
extern "C"
{
#endif

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

#define MAX_NUM_PLANES 4

#define MAX_COMPOSITE_FRAME 16

typedef enum
{
  NvBufferPayload_SurfArray,
  NvBufferPayload_MemHandle,
} NvBufferPayloadType;

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_VIDEO_ENC       = 0x1200,
  NvBufferTag_VIDEO_DEC       = 0x1400,
  NvBufferTag_VIDEO_CONVERT   = 0xf01,
} NvBufferTag;

typedef enum
{
  NvBufferColorFormat_YUV420,
  NvBufferColorFormat_YVU420,
  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_NV21_10LE,
  NvBufferColorFormat_NV12_12LE,
  NvBufferColorFormat_NV21_12LE,
  NvBufferColorFormat_YUV420_709,
  NvBufferColorFormat_YUV420_709_ER,
  NvBufferColorFormat_NV12_709,
  NvBufferColorFormat_NV12_709_ER,
  NvBufferColorFormat_YUV420_2020,
  NvBufferColorFormat_NV12_2020,
  NvBufferColorFormat_SignedR16G16,
  NvBufferColorFormat_A32,
  NvBufferColorFormat_YUV444,
  NvBufferColorFormat_GRAY8,
  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,
} NvBufferComposite_Flag;

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 _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 _NvBufferCompositeParams
{
  uint32_t composite_flag;
  uint32_t input_buf_count;
  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 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 NvBufferGetParams (int dmabuf_fd, NvBufferParams *params);

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 NvBufferMemSyncForDevice (int dmabuf_fd, unsigned int plane, void **pVirtAddr);

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

int NvBufferMemUnMap (int dmabuf_fd, 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 NvBufferComposite (int *src_dmabuf_fds, int dst_dmabuf_fd, NvBufferCompositeParams *composite_params);

#ifdef __cplusplus
}
#endif

#endif