/*
 * 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_SLICE_H__
#define __H264E_SLICE_H__

#include "mpp_enc_cfg.h"

#include "h264e_syntax.h"
#include "h264e_sps.h"
#include "h264e_pps.h"

/*
 * For H.264 encoder slice header process.
 * Remove some syntax that encoder not supported.
 * Field, mbaff, B slice are not supported yet.
 *
 * The reorder and mmco syntax will be create by dpb and attach to slice.
 * Then slice syntax will be passed to hal to config hardware or modify the
 * hardware stream.
 */

/* reference picture list modification operation */
typedef struct H264eRplmo_t {
    RK_S32      modification_of_pic_nums_idc;
    RK_S32      abs_diff_pic_num_minus1;
    RK_S32      long_term_pic_idx;
    RK_S32      abs_diff_view_idx_minus1;
} H264eRplmo;

typedef struct H264eReorderInfo_t {
    RK_S32      rd_cnt;
    RK_S32      wr_cnt;
    RK_S32      size;
    H264eRplmo  ops[H264E_MAX_REFS_CNT];
} H264eReorderInfo;

/*
 * mmco (Memory management control operation) value
 * 0 - End memory_management_control_operation syntax element loop
 * 1 - Mark a short-term reference picture as "unused for reference"
 * 2 - Mark a long-term reference picture as "unused for reference"
 * 3 - Mark a short-term reference picture as "used for long-term
 *     reference" and assign a long-term frame index to it
 * 4 - Specify the maximum long-term frame index and mark all long-term
 *     reference pictures having long-term frame indices greater than
 *     the maximum value as "unused for reference"
 * 5 - Mark all reference pictures as "unused for reference" and set the
 *     MaxLongTermFrameIdx variable to "no long-term frame indices"
 * 6 - Mark the current picture as "used for long-term reference" and
 *     assign a long-term frame index to it
 */
typedef struct H264eMmco_t {
    RK_S32      mmco;
    RK_S32      difference_of_pic_nums_minus1;   // for MMCO 1 & 3
    RK_S32      long_term_pic_num;               // for MMCO 2
    RK_S32      long_term_frame_idx;             // for MMCO 3 & 6
    RK_S32      max_long_term_frame_idx_plus1;   // for MMCO 4
} H264eMmco;

#define MAX_H264E_MMCO_CNT      8

typedef struct H264eMarkingInfo_t {
    RK_S32      idr_flag;
    /* idr marking flag */
    RK_S32      no_output_of_prior_pics;
    RK_S32      long_term_reference_flag;
    /* non-idr marking flag */
    RK_S32      adaptive_ref_pic_buffering;
    RK_S32      rd_cnt;
    RK_S32      wr_cnt;
    RK_S32      size;
    H264eMmco   ops[MAX_H264E_MMCO_CNT];
} H264eMarkingInfo;

typedef struct H264eSlice_t {
    /* Copy of sps/pps parameter */
    RK_S32      mb_w;
    RK_S32      mb_h;
    RK_U32      max_num_ref_frames;
    RK_U32      entropy_coding_mode;
    RK_S32      log2_max_frame_num;
    RK_S32      log2_max_poc_lsb;
    RK_S32      pic_order_cnt_type;
    RK_S32      qp_init;

    /* Nal parameters */
    RK_S32      nal_reference_idc;
    RK_S32      nalu_type;

    /* Unchanged parameters  */
    RK_U32      first_mb_in_slice;
    RK_U32      slice_type;
    RK_U32      pic_parameter_set_id;
    RK_S32      frame_num;
    RK_S32      num_ref_idx_override;
    RK_S32      qp_delta;
    RK_U32      cabac_init_idc;
    RK_U32      disable_deblocking_filter_idc;
    RK_S32      slice_alpha_c0_offset_div2;
    RK_S32      slice_beta_offset_div2;

    /* reorder parameter */
    RK_S32      ref_pic_list_modification_flag;
    H264eReorderInfo    *reorder;
    H264eMarkingInfo    *marking;

    RK_S32      idr_flag;
    RK_U32      idr_pic_id;
    RK_U32      next_idr_pic_id;
    /* for poc mode 0 */
    RK_U32      pic_order_cnt_lsb;
    RK_S32      num_ref_idx_active;
    /* idr mmco flag */
    RK_S32      no_output_of_prior_pics;
    RK_S32      long_term_reference_flag;

    /* Changable parameters  */
    RK_S32      adaptive_ref_pic_buffering;

    /* for multi-slice writing */
    RK_S32      is_multi_slice;
} H264eSlice;

#ifdef  __cplusplus
extern "C" {
#endif

/*
 * reorder context for both dpb and slice
 */
MPP_RET h264e_reorder_init(H264eReorderInfo *reorder);
MPP_RET h264e_reorder_wr_rewind(H264eReorderInfo *info);
MPP_RET h264e_reorder_rd_rewind(H264eReorderInfo *info);
MPP_RET h264e_reorder_wr_op(H264eReorderInfo *info, H264eRplmo *op);
MPP_RET h264e_reorder_rd_op(H264eReorderInfo *info, H264eRplmo *op);

/* mmco context for both dpb and slice */
MPP_RET h264e_marking_init(H264eMarkingInfo *marking);
RK_S32 h264e_marking_is_empty(H264eMarkingInfo *info);
MPP_RET h264e_marking_wr_rewind(H264eMarkingInfo *marking);
MPP_RET h264e_marking_rd_rewind(H264eMarkingInfo *marking);
MPP_RET h264e_marking_wr_op(H264eMarkingInfo *info, H264eMmco *op);
MPP_RET h264e_marking_rd_op(H264eMarkingInfo *info, H264eMmco *op);

/*
 * h264e_slice_update is called only on cfg is update.
 * When cfg has no changes just use slice next to setup
 */
void h264e_slice_init(H264eSlice *slice, H264eReorderInfo *reorder,
                      H264eMarkingInfo *marking);
RK_S32 h264e_slice_update(H264eSlice *slice, MppEncCfgSet *cfg,
                          H264eSps *sps, H264ePps *pps,
                          H264eDpbFrm *frm);

RK_S32 h264e_slice_read(H264eSlice *slice, void *p, RK_S32 size);
RK_S32 h264e_slice_write(H264eSlice *slice, void *p, RK_U32 size);
RK_S32 h264e_slice_write_pskip(H264eSlice *slice, void *p, RK_U32 size);
RK_S32 h264e_slice_move(RK_U8 *dst, RK_U8 *src, RK_S32 dst_bit, RK_S32 src_bit,
                        RK_S32 src_size);

RK_S32 h264e_slice_write_prefix_nal_unit_svc(H264ePrefixNal *nal, void *p, RK_S32 size);

#ifdef __cplusplus
}
#endif

#endif /* __H264E_SLICE_H__ */