#ifndef _JPU_API_H_
#define _JPU_API_H_

#include "jpuconfig.h"
#include "jdi.h"

#define CVI_JPG_DBG(msg, ...) \
	do {\
		if (1) {\
			printf("[DBG] %s = %d, " msg, __func__, __LINE__, ##__VA_ARGS__);\
		} \
	} while (0)

#define ALIGN_X(IN, ALIGN)     (((IN) + (ALIGN) - 1) / (ALIGN) * (ALIGN))
#define ALIGN_32

//------------------------------------------------------------------------------
// common struct and definition
//------------------------------------------------------------------------------

typedef enum {
	ENABLE_JPG_ROTATION,
	DISABLE_JPG_ROTATION,
	ENABLE_JPG_MIRRORING,
	DISABLE_JPG_MIRRORING,
	SET_JPG_MIRROR_DIRECTION,
	SET_JPG_ROTATION_ANGLE,
	SET_JPG_ROTATOR_OUTPUT,
	SET_JPG_ROTATOR_STRIDE,
	SET_JPG_SCALE_HOR,
	SET_JPG_SCALE_VER,
	SET_JPG_USE_PARTIAL_MODE,
	SET_JPG_PARTIAL_FRAME_NUM,
	SET_JPG_PARTIAL_LINE_NUM,
	SET_JPG_ENCODE_NEXT_LINE,
	SET_JPG_USE_STUFFING_BYTE_FF,
	ENC_JPG_GET_HEADER,
	ENABLE_LOGGING,
	DISABLE_LOGGING,
	JPG_CMD_END
} JpgCommand;

typedef enum {
	JPG_RET_SUCCESS,
	JPG_RET_FAILURE,
	JPG_RET_BIT_EMPTY,
	JPG_RET_EOS,
	JPG_RET_INVALID_HANDLE,
	JPG_RET_INVALID_PARAM,
	JPG_RET_INVALID_COMMAND,
	JPG_RET_ROTATOR_OUTPUT_NOT_SET,
	JPG_RET_ROTATOR_STRIDE_NOT_SET,
	JPG_RET_FRAME_NOT_COMPLETE,
	JPG_RET_INVALID_FRAME_BUFFER,
	JPG_RET_INSUFFICIENT_FRAME_BUFFERS,
	JPG_RET_INVALID_STRIDE,
	JPG_RET_WRONG_CALL_SEQUENCE,
	JPG_RET_CALLED_BEFORE,
	JPG_RET_NOT_INITIALIZED
} JpgRet;

typedef enum {
	MIRDIR_NONE,
	MIRDIR_VER,
	MIRDIR_HOR,
	MIRDIR_HOR_VER
} JpgMirrorDirection;

typedef enum {
	FORMAT_420 = 0,
	FORMAT_422 = 1,
	FORMAT_224 = 2,
	FORMAT_444 = 3,
	FORMAT_400 = 4
} FrameFormat;

typedef enum {
	CBCR_ORDER_NORMAL,
	CBCR_ORDER_REVERSED
} CbCrOrder;

typedef enum {
	CBCR_SEPARATED = 0,
	CBCR_INTERLEAVE
	,
	CRCB_INTERLEAVE
} CbCrInterLeave;

typedef enum {
	PACKED_FORMAT_NONE,
	PACKED_FORMAT_422_YUYV,
	PACKED_FORMAT_422_UYVY,
	PACKED_FORMAT_422_YVYU,
	PACKED_FORMAT_422_VYUY,
	PACKED_FORMAT_444
} PackedOutputFormat;

typedef enum {
	INT_JPU_DONE = 0,
	INT_JPU_ERROR = 1,
	INT_JPU_BIT_BUF_EMPTY = 2,
	INT_JPU_BIT_BUF_FULL = 2,
	INT_JPU_PARIAL_OVERFLOW = 3,
	INT_JPU_PARIAL_BUF0_EMPTY = 4,
	INT_JPU_PARIAL_BUF1_EMPTY,
	INT_JPU_PARIAL_BUF2_EMPTY,
	INT_JPU_PARIAL_BUF3_EMPTY,
	INT_JPU_BIT_BUF_STOP
} InterruptJpu;

typedef enum {
	JPG_TBL_NORMAL,
	JPG_TBL_MERGE
} JpgTableMode;

typedef enum {
	ENC_HEADER_MODE_NORMAL,
	ENC_HEADER_MODE_SOS_ONLY
} JpgEncHeaderMode;

typedef struct {
	PhysicalAddress bufY;
	PhysicalAddress bufCb;
	PhysicalAddress bufCr;
	int	stride;
} FrameBuffer;

struct JpgInst;

//------------------------------------------------------------------------------
// decode struct and definition
//------------------------------------------------------------------------------

typedef struct JpgInst JpgDecInst;
typedef JpgDecInst * JpgDecHandle;

typedef struct {
	PhysicalAddress bitstreamBuffer;
	int bitstreamBufferSize;
	BYTE *pBitStream;
	int streamEndian;
	int frameEndian;
	CbCrInterLeave chroma_interleave;
	int thumbNailEn;
	PackedOutputFormat packedFormat;
	int roiEnable;
	int roiOffsetX;
	int roiOffsetY;
	int roiWidth;
	int roiHeight;

} JpgDecOpenParam;

typedef struct {
	int picWidth;
	int picHeight;
	int minFrameBufferCount;
	int sourceFormat;
	int ecsPtr;
	int roiFrameWidth;
	int roiFrameHeight;
	int roiFrameOffsetX;
	int roiFrameOffsetY;
	int roiMCUSize;
	int colorComponents;
} JpgDecInitialInfo;

typedef struct {
	int scaleDownRatioWidth;
	int scaleDownRatioHeight;
} JpgDecParam;

typedef struct {
	int indexFrameDisplay;
	int numOfErrMBs;
	int decodingSuccess;
	int decPicHeight;
	int decPicWidth;
	int consumedByte;
	int bytePosFrameStart;
	int ecsPtr;
} JpgDecOutputInfo;

//------------------------------------------------------------------------------
// encode struct and definition
//------------------------------------------------------------------------------

typedef struct JpgInst JpgEncInst;
typedef JpgEncInst * JpgEncHandle;

typedef struct {
	PhysicalAddress bitstreamBuffer;
	Uint32 bitstreamBufferSize;
	int picWidth;
	int picHeight;
	int sourceFormat;
	int restartInterval;
	int streamEndian;
	int frameEndian;
	CbCrInterLeave chroma_interleave;
	BYTE huffVal[4][162];
	BYTE huffBits[4][256];
	BYTE qMatTab[4][64];
	PackedOutputFormat packedFormat;
} JpgEncOpenParam;

typedef struct {
	int minFrameBufferCount;
	int colorComponents;
} JpgEncInitialInfo;

typedef struct {
	FrameBuffer *sourceFrame;
} JpgEncParam;

typedef struct {
	PhysicalAddress bitstreamBuffer;
	Uint32 bitstreamSize;
} JpgEncOutputInfo;

typedef struct {
	PhysicalAddress paraSet;
	BYTE *pParaSet;
	int size;
	int headerMode;
	int quantMode;
	int huffMode;
	int disableAPPMarker;
} JpgEncParamSet;

#ifdef __cplusplus
extern "C" {
#endif

int JPU_IsBusy(void);
Uint32 JPU_GetStatus(void);
void JPU_ClrStatus(Uint32 val);
Uint32 JPU_IsInit(void);
Uint32 JPU_WaitInterrupt(int timeout);

JpgRet JPU_Init(void);
void JPU_DeInit(void);
int JPU_GetOpenInstanceNum(void);
JpgRet JPU_GetVersionInfo(Uint32 *versionInfo);

// function for decode
JpgRet JPU_DecOpen(JpgDecHandle *pHandle, JpgDecOpenParam *pop);
JpgRet JPU_DecClose(JpgDecHandle handle);
JpgRet JPU_DecGetInitialInfo(JpgDecHandle handle, JpgDecInitialInfo *info);

JpgRet JPU_DecSetRdPtr(JpgDecHandle handle, PhysicalAddress addr, int updateWrPtr);

JpgRet JPU_DecRegisterFrameBuffer(JpgDecHandle handle, FrameBuffer *bufArray, int num, int strideY, int strideC);
JpgRet JPU_DecGetBitstreamBuffer(JpgDecHandle handle, PhysicalAddress *prdPrt, PhysicalAddress *pwrPtr, int *size);
JpgRet JPU_DecUpdateBitstreamBuffer(JpgDecHandle handle, int size);
JpgRet JPU_HWReset(void);
JpgRet JPU_SWReset(void);
JpgRet JPU_DecStartOneFrame(JpgDecHandle handle, JpgDecParam *param);
JpgRet JPU_DecGetOutputInfo(JpgDecHandle handle, JpgDecOutputInfo *info);
JpgRet JPU_DecIssueStop(JpgDecHandle handle);
JpgRet JPU_DecCompleteStop(JpgDecHandle handle);
JpgRet JPU_DecGiveCommand(JpgDecHandle handle, JpgCommand cmd, void *parameter);
JpgRet JPU_EncGetBitstreamBuffer(JpgEncHandle handle, PhysicalAddress *prdPrt, PhysicalAddress *pwrPtr, int *size);
JpgRet JPU_EncUpdateBitstreamBuffer(JpgEncHandle handle, int size);

#ifdef __cplusplus
}
#endif

#endif //_JPU_API_H_