/*
|
* Copyright 2015 Rockchip Electronics Co. LTD
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
|
#ifndef __H264E_DPB_H__
|
#define __H264E_DPB_H__
|
|
#include "h264e_syntax.h"
|
#include "h264e_sps.h"
|
#include "mpp_enc_ref.h"
|
|
/*
|
* H.264 encoder dpb structure info
|
*
|
* +----------+ DPB +-> build list +-> REF_LIST
|
* | + +
|
* v v v
|
* FRM_BUF_GRP DPB_FRM +------+------> get_reorder
|
* + + |
|
* | v |
|
* +--------> FRM_BUF +------> get_mmco
|
*
|
* H264eDpb is overall structure contain the whole dpb info.
|
* It is composed H264eDpbFrm and H264eDpbList.
|
*
|
*/
|
|
#define MAX_GOP_SIZE 8
|
#define MAX_GOP_FMT_CNT (MAX_GOP_SIZE+1)
|
#define MAX_GOP_FMT_SIZE 5
|
#define MAX_GOP_FMT_BUF_STUFF 16
|
#define MAX_GOP_FMT_BUF_SIZE (MAX_GOP_FMT_CNT * MAX_GOP_FMT_SIZE + MAX_GOP_FMT_BUF_STUFF)
|
|
#define H264E_ST_GOP_FLAG (0x00000001)
|
#define H264E_ST_GOP_WITH_LT_REF (0x00000002)
|
#define H264E_LT_GOP_FLAG (0x00000010)
|
|
#define REF_BY_RECN(idx) (0x00000001 << idx)
|
#define REF_BY_REFR(idx) (0x00000001 << idx)
|
|
typedef struct H264eDpbFrm_t {
|
RK_S32 slot_idx;
|
// frame index in frames
|
RK_S32 seq_idx;
|
|
union {
|
RK_U32 on_used;
|
struct {
|
RK_U32 dpb_used : 8;
|
RK_U32 hal_used : 8;
|
};
|
};
|
|
/* frame status */
|
EncFrmStatus status;
|
|
/* frame number from H264eSlice */
|
RK_S32 frame_num;
|
RK_S32 lt_idx;
|
/* poc from H264eSlice */
|
RK_S32 poc;
|
/* pts from input MppFrame */
|
RK_S64 pts;
|
} H264eDpbFrm;
|
|
/* runtime status record */
|
typedef struct H264eDpbRt_t {
|
RK_S32 last_seq_idx;
|
RK_S32 last_is_ref;
|
RK_S32 last_frm_num;
|
RK_S32 last_poc_lsb;
|
RK_S32 last_poc_msb;
|
} H264eDpbRt;
|
|
/*
|
* dpb frame arrangement
|
*
|
* If dpb size is 3 then dpb frame will be total 4 frames.
|
* Then the frame 3 is always current frame and frame 0~2 is reference frame
|
* in the gop structure.
|
*
|
* When one frame is encoded all it info will be moved to its gop position for
|
* next frame encoding.
|
*/
|
typedef struct H264eDpb_t {
|
H264eReorderInfo *reorder;
|
H264eMarkingInfo *marking;
|
|
MppEncCpbInfo info;
|
RK_S32 next_max_lt_idx;
|
RK_S32 curr_max_lt_idx;
|
RK_S32 st_size;
|
RK_S32 lt_size;
|
RK_S32 used_size;
|
|
RK_S32 dpb_size;
|
RK_S32 total_cnt;
|
|
/* status on dpb rebuild is needed */
|
RK_S32 max_frm_num;
|
RK_S32 max_poc_lsb;
|
RK_S32 poc_type;
|
|
RK_S32 last_frm_num;
|
RK_S32 last_poc_lsb;
|
RK_S32 last_poc_msb;
|
|
H264eDpbFrm *curr;
|
H264eDpbFrm *refr;
|
H264eDpbFrm *list[H264E_MAX_REFS_CNT];
|
H264eDpbFrm *stref[H264E_MAX_REFS_CNT];
|
H264eDpbFrm *ltref[H264E_MAX_REFS_CNT];
|
H264eDpbFrm *map[H264E_MAX_REFS_CNT + 1];
|
|
// frame storage
|
H264eDpbRt rt;
|
H264eDpbRt rt_bak;
|
H264eDpbFrm frames[H264E_MAX_REFS_CNT + 1];
|
H264eDpbFrm frm_bak[H264E_MAX_REFS_CNT + 1];
|
} H264eDpb;
|
|
#ifdef __cplusplus
|
extern "C" {
|
#endif
|
|
MPP_RET h264e_dpb_init(H264eDpb *dpb, H264eReorderInfo *reorder, H264eMarkingInfo *marking);
|
MPP_RET h264e_dpb_deinit(H264eDpb *dpb);
|
|
MPP_RET h264e_dpb_setup(H264eDpb *dpb, MppEncCfgSet* cfg, H264eSps *sps);
|
|
/*
|
* Setup current frame config using flags
|
* This config function will be called before each frame is encoded:
|
*
|
* idr - current frame is force to IDR or not
|
* lt_ref - current frame is marked as longterm reference
|
*/
|
MPP_RET h264e_dpb_proc(H264eDpb *dpb, EncCpbStatus *cpb);
|
|
/*
|
* hal usage flag mark / unmark function
|
*/
|
MPP_RET h264e_dpb_hal_start(H264eDpb *dpb, RK_S32 slot_idx);
|
MPP_RET h264e_dpb_hal_end(H264eDpb *dpb, RK_S32 slot_idx);
|
|
void h264e_dpb_check(H264eDpb *dpb, EncCpbStatus *cpb);
|
|
#define h264e_dpb_dump_frms(dpb) h264e_dpb_dump_frm(dpb, __FUNCTION__, __LINE__)
|
|
void h264e_dpb_dump_frm(H264eDpb *dpb, const char *caller, RK_S32 line);
|
|
#ifdef __cplusplus
|
}
|
#endif
|
|
#endif /* __H264E_DPB_H__ */
|
