111 lines
2.7 KiB
C
111 lines
2.7 KiB
C
|
#ifndef __NVMPI_H__
|
||
|
|
#define __NVMPI_H__
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdbool.h>
|
||
|
|
|
||
|
|
//Maximum size of the encoded buffers on the capture plane in bytes
|
||
|
|
#define NVMPI_ENC_CHUNK_SIZE 2*1024*1024
|
||
|
|
|
||
|
|
typedef struct nvmpictx nvmpictx;
|
||
|
|
|
||
|
|
typedef enum {
|
||
|
|
NV_PIX_NV12,
|
||
|
|
NV_PIX_YUV420
|
||
|
|
}nvPixFormat;
|
||
|
|
|
||
|
|
typedef enum {
|
||
|
|
NV_VIDEO_CodingUnused,
|
||
|
|
NV_VIDEO_CodingH264, /**< H.264 */
|
||
|
|
NV_VIDEO_CodingMPEG4, /**< MPEG-4 */
|
||
|
|
NV_VIDEO_CodingMPEG2, /**< MPEG-2 */
|
||
|
|
NV_VIDEO_CodingVP8, /**< VP8 */
|
||
|
|
NV_VIDEO_CodingVP9, /**< VP9 */
|
||
|
|
NV_VIDEO_CodingHEVC, /**< H.265/HEVC */
|
||
|
|
} nvCodingType;
|
||
|
|
|
||
|
|
typedef struct _NVSIZE{
|
||
|
|
unsigned int width;
|
||
|
|
unsigned int height;
|
||
|
|
}nvSize;
|
||
|
|
|
||
|
|
typedef struct _NVENCPARAM{
|
||
|
|
unsigned int width;
|
||
|
|
unsigned int height;
|
||
|
|
unsigned int profile;
|
||
|
|
unsigned int level;
|
||
|
|
unsigned int bitrate;
|
||
|
|
unsigned int peak_bitrate;
|
||
|
|
char enableLossless;
|
||
|
|
char mode_vbr;
|
||
|
|
char insert_spspps_idr;
|
||
|
|
unsigned int iframe_interval;
|
||
|
|
unsigned int idr_interval;
|
||
|
|
unsigned int fps_n;
|
||
|
|
unsigned int fps_d;
|
||
|
|
int capture_num;
|
||
|
|
unsigned int max_b_frames;
|
||
|
|
unsigned int refs;
|
||
|
|
unsigned int qmax;
|
||
|
|
unsigned int qmin;
|
||
|
|
unsigned int hw_preset_type;
|
||
|
|
unsigned int vbv_buffer_size; //virtual buffer size of the encoder
|
||
|
|
nvCodingType codingType;
|
||
|
|
} nvEncParam;
|
||
|
|
|
||
|
|
typedef struct _NVDECPARAM{
|
||
|
|
int frame_pool_size;
|
||
|
|
nvCodingType codingType;
|
||
|
|
nvPixFormat pixFormat;
|
||
|
|
nvSize resized;
|
||
|
|
} nvDecParam;
|
||
|
|
|
||
|
|
typedef struct _NVPACKET{
|
||
|
|
unsigned long flags;
|
||
|
|
unsigned long payload_size;
|
||
|
|
unsigned char *payload;
|
||
|
|
unsigned long pts;
|
||
|
|
//NVMPI_pkt pointer. used by encoder
|
||
|
|
void* privData;
|
||
|
|
} nvPacket;
|
||
|
|
|
||
|
|
typedef struct _NVFRAME{
|
||
|
|
unsigned long flags;
|
||
|
|
unsigned long payload_size[3];
|
||
|
|
unsigned char *payload[3];
|
||
|
|
unsigned int linesize[3];
|
||
|
|
nvPixFormat type;
|
||
|
|
unsigned int width;
|
||
|
|
unsigned int height;
|
||
|
|
time_t timestamp;
|
||
|
|
}nvFrame;
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
extern "C" {
|
||
|
|
#endif
|
||
|
|
|
||
|
|
nvmpictx* nvmpi_create_decoder(nvDecParam* param);
|
||
|
|
|
||
|
|
int nvmpi_decoder_put_packet(nvmpictx* ctx, nvPacket* packet);
|
||
|
|
|
||
|
|
int nvmpi_decoder_get_frame(nvmpictx* ctx, nvFrame* frame,bool wait);
|
||
|
|
|
||
|
|
int nvmpi_decoder_close(nvmpictx* ctx);
|
||
|
|
|
||
|
|
nvmpictx* nvmpi_create_encoder(nvEncParam* param);
|
||
|
|
//add frame to encoder
|
||
|
|
int nvmpi_encoder_put_frame(nvmpictx* ctx, nvFrame* frame);
|
||
|
|
//get filled packet from encoder
|
||
|
|
int nvmpi_encoder_get_packet(nvmpictx* ctx, nvPacket** packet);
|
||
|
|
//get empty packet with allocated buffer from encoder packet pool
|
||
|
|
int nvmpi_encoder_dqEmptyPacket(nvmpictx* ctx, nvPacket** packet);
|
||
|
|
//add empty packet with allocated buffer to encoder packet pool
|
||
|
|
void nvmpi_encoder_qEmptyPacket(nvmpictx* ctx, nvPacket* packet);
|
||
|
|
//close encoder
|
||
|
|
int nvmpi_encoder_close(nvmpictx* ctx);
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#endif /*__NVMPI_H__*/
|