RTK_mmp_enc.h 1.7 KB
#ifndef RTK_MMP_ENC_H
#define RTK_MMP_ENC_H

#include "BaseConfig.h"
#include "../RTK_mmp_LibInclude.h"

#ifdef RTK_MPP_STATUS_ON

#include "rockchip/rk_type.h"
#include "rockchip/mpp_frame.h"
#include "JZsdk_Base/JZsdk_Code/JZsdk_Code.h"


typedef struct T_rtk_mmp_enc_info
{
    //要求输入的参数
    MppFrameFormat Input_fmt; //输入视频的格式
    unsigned int width;   //输入视频的宽度
    unsigned int height;  //输入视频的高度
    unsigned int FrameNum; //帧数

    MppCodingType encType;   //编码的类型
    unsigned int IDR_gop; //i帧间隔数


    // 基础流上下文
    MppCtx ctx;
    MppApi *mpi;
    


    // 输入/输出
    MppBufferGroup BufferGop;   //数据缓冲池
    
    MppBuffer PacketBuffer; //输出数组缓冲区
    unsigned int Packet_data_size; //packet包的长度
    unsigned int Packet_eos;    //packet包的结束标志

    MppBuffer FrameBuffer; //输入数组缓冲区
    unsigned int Frame_data_size; //Frame包的长度
    unsigned int Frame_eos;     //Frame包的结束标志

    // 资源分配参数
    unsigned int hor_stride;
    unsigned int ver_stride;

    // 运行时的配置
    MppEncCfg cfg;
    MppEncHeaderMode    header_mode;
    int header_size;
    
    MppEncSeiMode sei_mode;

}T_rtk_mmp_enc_info ;

T_JZsdkReturnCode RTK_mmp_enc_Init(T_rtk_mmp_enc_info *enc_info);
T_JZsdkReturnCode RTK_mmp_enc_yuv_to_h264_byFrame(T_rtk_mmp_enc_info *enc_info, MppFrame input_data_frame, MppPacket *out_put_frame);
T_JZsdkReturnCode RTK_mmp_enc_data_to_h264(T_rtk_mmp_enc_info *enc_info, char *in_data, int in_data_length, MppPacket *Packet);
T_JZsdkReturnCode RTK_mmp_enc_SetNextFrame_IDR(T_rtk_mmp_enc_info *enc_info);

#endif

#endif