/*
 * Copyright (c) 2020 Amlogic, Inc. All rights reserved.
 *
 * This source code is subject to the terms and conditions defined in the
 * file 'LICENSE' which is part of this source code package.
 *
 * Description:
 */



#ifndef AM_VIDEO_DEC_BASE_H
#define AM_VIDEO_DEC_BASE_H

#include <stdint.h>
#include <sys/types.h>
#define AM_VIDEO_DEC_INIT_FLAG_DEFAULT              0
#define AM_VIDEO_DEC_INIT_FLAG_CODEC2               1
#define AM_VIDEO_DEC_INIT_FLAG_STREAMMODE           (1ul << 1)
#define AM_VIDEO_DEC_INIT_FLAG_DMXDATA_SOURCE       (1ul << 2)
#define AM_VIDEO_DEC_INIT_FLAG_TSPLAYER             (1ul << 3)
#define AM_VIDEO_DEC_INIT_FLAG_VIDEO_PASSTHROUGH    (1ul << 4)

#define AM_VIDEO_DEC_INIT_FLAG_USE_INPUT_BUFFER_POOL (1ul << 8)
#define AM_VIDEO_DEC_INIT_FLAG_USE_DYNAMIC_CAPTURE_BUFFER (1ul << 9)
#define AM_VIDEO_DEC_INIT_FLAG_USE_UVM               (1ul << 10)
#define AM_VIDEO_DEC_INIT_FLAG_USE_PTS_CALCULATOR    (1ul << 11)
#define AM_VIDEO_DEC_INIT_FLAG_USE_SECURE_VDEC_HAL   (1ul << 12)
#define AM_VIDEO_DEC_INIT_FLAG_USE_HI_PRIO           (1ul << 13)
#define AM_VIDEO_DEC_INIT_FLAG_USE_LOW_LATENCY_MODE  (1ul << 14)
#define AM_VIDEO_DEC_INIT_FLAG_ADD_CSD               (1ul << 15)
typedef struct {
    /* video */
    uint32_t    vpid;
    uint32_t    nVideoWidth;
    uint32_t    nVideoHeight;
    uint32_t    nFrameRate;
    uint32_t   vFmt;
    uint32_t    drmMode;
    /* audio */
    uint32_t	apid;
    uint32_t    nChannels;
    uint32_t    nSampleRate;
    uint32_t   aFmt;
    /* pcrid */
    uint32_t	pcrid;
    /* display */
    uint32_t dispMode;
    uint32_t    nSidebandType;
    uint32_t    nSidebandId;
    uint32_t    nAvsyncMode;

    int subtitleFlg;
    int32_t  mDemuxType;
    int32_t dmx_dev_id;
    int32_t dmx_player_id;
    unsigned int  stbuf_start;
    unsigned int  stbuf_size;
    uint32_t    nDecType;
    int32_t mPlayerInstansNo;
    int32_t nVideoRecoveryValue;
    int32_t mSourceType;
    int32_t mLowMem;
} init_param_t;

typedef struct {
    /*inherit old interface*/
    char mime[64];
    uint8_t* config;
    uint32_t configLen;
    bool secureMode;
    bool useV4l2;
    bool isTunnelMode;
    int32_t flags;
    /*decode info*/
    int32_t dispMode;
    int32_t pipelineMode;
    int32_t extDecInfo;//reserved
    int32_t resInfo1;
    int32_t resInfo2;
    /*resource manage info*/
    char resAppName[64];
    void (* resCallback)(void * resOpaque);
    void* resOpaque;
}video_dec_init_params;
class AmVideoDecCallback {
public:
    AmVideoDecCallback();
    virtual ~AmVideoDecCallback();
    virtual void onOutputFormatChanged(uint32_t requested_num_of_buffers,
                int32_t width, uint32_t height) = 0;
    virtual void onOutputBufferDone(int32_t pictureBufferId, int64_t bitstreamId,
                uint32_t width, uint32_t height) = 0;
    virtual void onInputBufferDone(int32_t bitstream_buffer_id) = 0;
    virtual void onUpdateDecInfo(const uint8_t* info, uint32_t isize) = 0;
    virtual void onFlushDone() = 0;
    virtual void onResetDone() = 0;
    virtual void onError(int32_t error) = 0;
    virtual void onUserdataReady(const uint8_t* userdata, uint32_t usize) = 0;
    virtual void onEvent(uint32_t event, void* param, uint32_t paramsize) = 0;
};

class AmVideoDecBase {

public:
    AmVideoDecBase(AmVideoDecCallback* callback);
    virtual ~AmVideoDecBase();

    virtual int32_t initialize(const char* mime, uint8_t* config, uint32_t configLen,
            bool secureMode, bool useV4l2 = 1) = 0;

    virtual int32_t setQueueCount(uint32_t queueCount) = 0;

    virtual int32_t queueInputBuffer(int32_t bitstreamId, int ashmemFd, off_t offset,
            uint32_t bytesUsed, uint64_t timestamp) = 0;

    virtual int32_t queueInputBuffer(int32_t bitstreamId, uint8_t* pbuf,
            off_t offset, uint32_t bytesUsed, uint64_t timestamp) = 0;

    virtual int32_t setupOutputBufferNum(uint32_t numOutputBuffers) = 0;

    virtual int32_t createOutputBuffer(uint32_t pictureBufferId,
                    int32_t dmabufFd, bool nv21 = 1, int32_t metaFd = -1) = 0;

    virtual int32_t createOutputBuffer(uint32_t pictureBufferId,
                    uint8_t* buf, size_t size, bool nv21 = 1) = 0;

    virtual int32_t queueOutputBuffer(int32_t pictureBufferId) = 0;
    virtual void flush() = 0;
    virtual void reset() = 0;
    virtual void destroy() = 0;
    virtual int32_t sendCommand(uint32_t index, void* param, uint32_t size) = 0;

    /* Ion output for non-bufferqueue */
    virtual int32_t allocIonBuffer(size_t size, void** mapaddr, int* fd = 0) = 0;
    virtual int32_t freeIonBuffer(void* mapaddr) = 0;
    virtual int32_t freeAllIonBuffer() = 0;

    /* uvm output for non-bufferqueue */
    virtual int32_t allocUvmBuffer(uint32_t width, uint32_t height, void** mapaddr, unsigned int i,
        int* fd = 0) = 0;
    virtual int32_t freeUvmBuffers() = 0;
    /*new interface for resman*/
    virtual int32_t initialize(video_dec_init_params* initParams) = 0;
};

extern "C" AmVideoDecBase* AmVideoDec_create(AmVideoDecCallback* callback);
extern "C" uint32_t AmVideoDec_getVersion(uint32_t* versionM, uint32_t* versionL);

#endif  // AM_VIDEO_DEC_BASE_H