/*
 * 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.
 */

/*
 * @file        m2vd_parser.h
 * @brief
 * @author      lks(lks@rock-chips.com)

 * @version     1.0.0
 * @history
 *   2015.7.15 : Create
 */

#ifndef __M2VD_PARSER_H__
#define __M2VD_PARSER_H__

#include "mpp_mem.h"
#include "mpp_bitread.h"

#include "parser_api.h"
#include "m2vd_syntax.h"
#include "m2vd_com.h"

#define M2VD_SKIP_ERROR_FRAME_EN  1
#define M2VD_DEC_OK                   0
#define M2VD_IFNO_CHG                 0x10
#define M2VD_DEC_UNSURPORT            -1
#define M2VD_DEC_MEMORY_FAIL      -2
#define M2VD_DEC_FILE_END         -3
#define M2VD_HW_DEC_UNKNOW_ERROR     -4
#define M2VD_DEC_PICHEAD_OK       1
#define M2VD_DEC_STREAM_END       2

#define TEMP_DBG_FM     0


#define PICTURE_START_CODE      0x100
#define SLICE_START_CODE_MIN    0x101
#define SLICE_START_CODE_MAX    0x1AF
#define USER_DATA_START_CODE    0x1B2
#define SEQUENCE_HEADER_CODE    0x1B3
#define SEQUENCE_ERROR_CODE     0x1B4
#define EXTENSION_START_CODE    0x1B5
#define SEQUENCE_END_CODE       0x1B7
#define GROUP_START_CODE        0x1B8
#define SYSTEM_START_CODE_MIN   0x1B9
#define SYSTEM_START_CODE_MAX   0x1FF

#define ISO_END_CODE            0x1B9
#define PACK_START_CODE         0x1BA
#define SYSTEM_START_CODE       0x1BB

#define VIDEO_ELEMENTARY_STREAM 0x1e0
#define NO_MORE_STREAM      0xffffffff

#define SEQUENCE_EXTENSION_ID                    1
#define SEQUENCE_DISPLAY_EXTENSION_ID            2
#define QUANT_MATRIX_EXTENSION_ID                3
#define COPYRIGHT_EXTENSION_ID                   4
#define SEQUENCE_SCALABLE_EXTENSION_ID           5
#define PICTURE_DISPLAY_EXTENSION_ID             7
#define PICTURE_CODING_EXTENSION_ID              8
#define PICTURE_SPATIAL_SCALABLE_EXTENSION_ID    9
#define PICTURE_TEMPORAL_SCALABLE_EXTENSION_ID  10


typedef enum M2VDPicCodingType_e {
    M2VD_CODING_TYPE_I = 1,
    M2VD_CODING_TYPE_P = 2,
    M2VD_CODING_TYPE_B = 3,
    M2VD_CODING_TYPE_D = 4
} M2VDPicCodingType;

typedef enum M2VDPicStruct_e {
    M2VD_PIC_STRUCT_TOP_FIELD    = 1,
    M2VD_PIC_STRUCT_BOTTOM_FIELD = 2,
    M2VD_PIC_STRUCT_FRAME        = 3
} M2VDPicStruct;

#define     M2VD_DBG_FILE_NUM               1
#define     M2VD_DBG_FILE_W                 1
#define     M2VD_BUF_SIZE_BITMEM            (512 * 1024)
#define     M2VD_BUF_SIZE_QPTAB            (256)
#define     M2V_OUT_FLAG                   0x1;

typedef enum M2VDBufGrpIdx_t {
    M2VD_BUF_GRP_BITMEM,
    M2VD_BUF_GRP_QPTAB,
    M2VD_BUF_GRP_BUTT,
} M2VDBufGrpIdx;

typedef struct M2VFrameHead_t {
    RK_U32          frameNumber;
    RK_U32          tr;
    RK_U32          picCodingType;
    RK_U32          totalMbInFrame;
    RK_U32          frameWidth; /* in macro blocks */
    RK_U32          frameHeight;    /* in macro blocks */
    RK_U32          mb_width;
    RK_U32          mb_height;
    RK_U32          vlcSet;
    RK_U32          qp;
    RK_U32          frame_codelen;
    // VPU_FRAME       *frame_space;
    MppFrame        f;
    RK_U32          flags;
    RK_S32          slot_index;
    //RK_U32          error_info;
} M2VDFrameHead;


/* ISO/IEC 13818-2 section 6.2.2.1:  sequence_header() */
typedef struct M2VDHeadSeq_t {
    RK_U32          decode_width; //horizontal_size_value
    RK_U32          decode_height; //vertical_size_value
    RK_S32          aspect_ratio_information;
    RK_S32          frame_rate_code;
    RK_S32          bit_rate_value;
    RK_S32          vbv_buffer_size;
    RK_S32          constrained_parameters_flag;
    RK_U32          load_intra_quantizer_matrix; //[TEMP]
    RK_U32          load_non_intra_quantizer_matrix; //[TEMP]
    RK_U8           *pIntra_table; //intra_quantiser_matrix[64]
    RK_U8           *pInter_table; //non_intra_quantiser_matrix[64]
} M2VDHeadSeq;

/* ISO/IEC 13818-2 section 6.2.2.3:  sequence_extension() */
typedef struct M2VDHeadSeqExt_t {
    RK_U32             horizontal_size_extension; //[TEMP]
    RK_U32             vertical_size_extension; //[TEMP]
    RK_U32             bit_rate_extension; //[TEMP]
    RK_U32             vbv_buffer_size_extension; //[TEMP]
    RK_S32             profile_and_level_indication;
    RK_S32             progressive_sequence;
    RK_S32             chroma_format;
    RK_S32             low_delay;
    RK_S32             frame_rate_extension_n;
    RK_S32             frame_rate_extension_d;
} M2VDHeadSeqExt;

/* ISO/IEC 13818-2 section 6.2.2.6: group_of_pictures_header()  */
typedef struct M2VDHeadGop_t {
    RK_S32             drop_flag;
    RK_S32             hour;
    RK_S32             minute;
    RK_S32             sec;
    RK_S32             frame;
    RK_S32             closed_gop;
    RK_S32             broken_link;
} M2VDHeadGop;


/* ISO/IEC 13818-2 section 6.2.3: picture_header() */
typedef struct M2VDHeadPic_t {
    RK_S32             temporal_reference;
    RK_S32             picture_coding_type;
    RK_S32             pre_picture_coding_type;
    RK_S32             vbv_delay;
    RK_S32             full_pel_forward_vector;
    RK_S32             forward_f_code;
    RK_S32             full_pel_backward_vector;
    RK_S32             backward_f_code;
    RK_S32             pre_temporal_reference;
} M2VDHeadPic;


/* ISO/IEC 13818-2 section 6.2.2.4:  sequence_display_extension() */
typedef struct M2VDHeadSeqDispExt_t {
    RK_S32             video_format;
    RK_S32             color_description;
    RK_S32             color_primaries;
    RK_S32             transfer_characteristics;
    RK_S32             matrix_coefficients;
} M2VDHeadSeqDispExt;

/* ISO/IEC 13818-2 section 6.2.3.1: picture_coding_extension() header */
typedef struct M2VDHeadPicCodeExt_t {
    RK_S32             f_code[2][2];
    RK_S32             intra_dc_precision;
    RK_S32             picture_structure;
    RK_S32             top_field_first;
    RK_S32             frame_pred_frame_dct;
    RK_S32             concealment_motion_vectors;
    RK_S32             q_scale_type;
    RK_S32             intra_vlc_format;
    RK_S32             alternate_scan;
    RK_S32             repeat_first_field;
    RK_S32             chroma_420_type;
    RK_S32             progressive_frame;
    RK_S32             composite_display_flag;
    RK_S32             v_axis;
    RK_S32             field_sequence;
    RK_S32             sub_carrier;
    RK_S32             burst_amplitude;
    RK_S32             sub_carrier_phase;
} M2VDHeadPicCodeExt;


/* ISO/IEC 13818-2 section 6.2.3.3: picture_display_extension() header */
typedef struct M2VDHeadPicDispExt_t {
    RK_S32             frame_center_horizontal_offset[3];
    RK_S32             frame_center_vertical_offset[3];
} M2VDHeadPicDispExt;

typedef struct M2VDCombMem_t {
    MppBuffer hw_buf;
    RK_U8*    sw_buf;
    RK_U8     buf_size;
    RK_U8*    sw_pos;
    RK_U32    length;
} M2VDCombMem;

typedef struct M2VDParserContext_t {
    MppDecCfgSet    *cfg;
    M2VDDxvaParam   *dxva_ctx;
    BitReadCtx_t    *bitread_ctx;
    RK_U8           *bitstream_sw_buf;
    RK_U8           *qp_tab_sw_buf;
    RK_U32          max_stream_size;
    RK_U32          left_length;
    RK_U32          state;
    RK_U32          vop_header_found;

    RK_U32          frame_size;

    RK_U32          display_width;
    RK_U32          display_height;
    RK_U32          frame_width;
    RK_U32          frame_height;
    RK_U32          mb_width;
    RK_U32          mb_height;
    RK_U32          MPEG2_Flag;

    RK_U32          info_changed;
    M2VDHeadSeq         seq_head;
    M2VDHeadSeqExt      seq_ext_head;
    M2VDHeadGop         gop_head;
    M2VDHeadPic         pic_head;
    M2VDHeadSeqDispExt  seq_disp_ext_head;
    M2VDHeadPicCodeExt  pic_code_ext_head;
    M2VDHeadPicDispExt  pic_disp_ext_head;

    RK_S32             resetFlag;

    RK_U64          pre_pts_27M;
    RK_S64          group_start_time_27M;
    RK_U32          group_frm_cnt;
    RK_U32          pretemporal_reference;
    RK_U32          pretime_temporal;
    RK_U32          max_temporal_reference;
    RK_U32          prechange_temporal_ref;
    RK_S32          frame_period;
    RK_S32          preframe_period;
    RK_U32          maxFrame_inGOP;
    RK_U32          pts_is_90K;
    RK_U32          ref_frame_cnt;
    long long       top_first_cnt;
    long long       bottom_first_cnt;
    RK_S32          mHeaderDecFlag;
    RK_S32          mExtraHeaderDecFlag;
    M2VDFrameHead   Framehead[3];
    M2VDFrameHead   *frame_ref0;
    M2VDFrameHead   *frame_ref1;
    M2VDFrameHead   *frame_cur;
#if M2VD_SKIP_ERROR_FRAME_EN
    RK_S32           mHwDecStatus;
#endif
    RK_S32           flush_dpb_eos;

    MppPacket       input_packet;
    RK_U32          eos;

    RK_S32          initFlag;
    RK_S32          decoder_err;

    MppBufSlots     packet_slots;
    MppBufSlots     frame_slots;
    RK_U32          cur_slot_index;
    MppDecCfgSet    *dec_cfg;

    RK_U64 pts;
    RK_U64 dts;

    FILE *fp_dbg_file[M2VD_DBG_FILE_NUM];
    FILE *fp_dbg_yuv;
} M2VDParserContext;

MPP_RET  m2vd_parser_init   (void *ctx, ParserCfg *cfg);
MPP_RET  m2vd_parser_deinit (void *ctx);
MPP_RET  m2vd_parser_reset  (void *ctx);
MPP_RET  m2vd_parser_flush  (void *ctx);
MPP_RET  m2vd_parser_control(void *ctx, MpiCmd cmd_type, void *param);
MPP_RET  m2vd_parser_prepare(void *ctx, MppPacket pkt, HalDecTask *task);
MPP_RET  m2vd_parser_parse  (void *ctx, HalDecTask *task);
MPP_RET  m2vd_parser_callback(void *ctx, void *err_info);

#endif

