#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__*/