/* * Chinese AVS video (AVS1-P2, JiZhun profile) decoder. * Copyright (c) 2006 Stefan Gehrer <stefan.gehrer@gmx.de> * * This file is part of Libav. * * Libav is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * Libav is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with Libav; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #ifndef AVCODEC_CAVS_H #define AVCODEC_CAVS_H #include "cavsdsp.h" #include "blockdsp.h" #include "h264chroma.h" #include "idctdsp.h" #include "get_bits.h" #include "videodsp.h" #define SLICE_MAX_START_CODE 0x000001af #define EXT_START_CODE 0x000001b5 #define USER_START_CODE 0x000001b2 #define CAVS_START_CODE 0x000001b0 #define PIC_I_START_CODE 0x000001b3 #define PIC_PB_START_CODE 0x000001b6 #define A_AVAIL 1 #define B_AVAIL 2 #define C_AVAIL 4 #define D_AVAIL 8 #define NOT_AVAIL -1 #define REF_INTRA -2 #define REF_DIR -3 #define ESCAPE_CODE 59 #define FWD0 0x01 #define FWD1 0x02 #define BWD0 0x04 #define BWD1 0x08 #define SYM0 0x10 #define SYM1 0x20 #define SPLITH 0x40 #define SPLITV 0x80 #define MV_BWD_OFFS 12 #define MV_STRIDE 4 enum cavs_mb { I_8X8 = 0, P_SKIP, P_16X16, P_16X8, P_8X16, P_8X8, B_SKIP, B_DIRECT, B_FWD_16X16, B_BWD_16X16, B_SYM_16X16, B_8X8 = 29 }; enum cavs_sub_mb { B_SUB_DIRECT, B_SUB_FWD, B_SUB_BWD, B_SUB_SYM }; enum cavs_intra_luma { INTRA_L_VERT, INTRA_L_HORIZ, INTRA_L_LP, INTRA_L_DOWN_LEFT, INTRA_L_DOWN_RIGHT, INTRA_L_LP_LEFT, INTRA_L_LP_TOP, INTRA_L_DC_128 }; enum cavs_intra_chroma { INTRA_C_LP, INTRA_C_HORIZ, INTRA_C_VERT, INTRA_C_PLANE, INTRA_C_LP_LEFT, INTRA_C_LP_TOP, INTRA_C_DC_128, }; enum cavs_mv_pred { MV_PRED_MEDIAN, MV_PRED_LEFT, MV_PRED_TOP, MV_PRED_TOPRIGHT, MV_PRED_PSKIP, MV_PRED_BSKIP }; enum cavs_block { BLK_16X16, BLK_16X8, BLK_8X16, BLK_8X8 }; enum cavs_mv_loc { MV_FWD_D3 = 0, MV_FWD_B2, MV_FWD_B3, MV_FWD_C2, MV_FWD_A1, MV_FWD_X0, MV_FWD_X1, MV_FWD_A3 = 8, MV_FWD_X2, MV_FWD_X3, MV_BWD_D3 = MV_BWD_OFFS, MV_BWD_B2, MV_BWD_B3, MV_BWD_C2, MV_BWD_A1, MV_BWD_X0, MV_BWD_X1, MV_BWD_A3 = MV_BWD_OFFS+8, MV_BWD_X2, MV_BWD_X3 }; DECLARE_ALIGNED(8, typedef, struct) { int16_t x; int16_t y; int16_t dist; int16_t ref; } cavs_vector; struct dec_2dvlc { int8_t rltab[59][3]; int8_t level_add[27]; int8_t golomb_order; int inc_limit; int8_t max_run; }; typedef struct AVSFrame { AVFrame *f; int poc; } AVSFrame; typedef struct AVSContext { AVCodecContext *avctx; BlockDSPContext bdsp; H264ChromaContext h264chroma; IDCTDSPContext idsp; VideoDSPContext vdsp; CAVSDSPContext cdsp; GetBitContext gb; AVSFrame cur; ///< currently decoded frame AVSFrame DPB[2]; ///< reference frames int dist[2]; ///< temporal distances from current frame to ref frames int low_delay; int profile, level; int aspect_ratio; int mb_width, mb_height; int width, height; int stream_revision; ///<0 for samples from 2006, 1 for rm52j encoder int progressive; int pic_structure; int skip_mode_flag; ///< select between skip_count or one skip_flag per MB int loop_filter_disable; int alpha_offset, beta_offset; int ref_flag; int mbx, mby, mbidx; ///< macroblock coordinates int flags; ///< availability flags of neighbouring macroblocks int stc; ///< last start code uint8_t *cy, *cu, *cv; ///< current MB sample pointers int left_qp; uint8_t *top_qp; /** mv motion vector cache 0: D3 B2 B3 C2 4: A1 X0 X1 - 8: A3 X2 X3 - X are the vectors in the current macroblock (5,6,9,10) A is the macroblock to the left (4,8) B is the macroblock to the top (1,2) C is the macroblock to the top-right (3) D is the macroblock to the top-left (0) the same is repeated for backward motion vectors */ cavs_vector mv[2*4*3]; cavs_vector *top_mv[2]; cavs_vector *col_mv; /** luma pred mode cache 0: -- B2 B3 3: A1 X0 X1 6: A3 X2 X3 */ int pred_mode_Y[3*3]; int *top_pred_Y; ptrdiff_t l_stride, c_stride; int luma_scan[4]; int qp; int qp_fixed; int cbp; ScanTable scantable; /** intra prediction is done with un-deblocked samples they are saved here before deblocking the MB */ uint8_t *top_border_y, *top_border_u, *top_border_v; uint8_t left_border_y[26], left_border_u[10], left_border_v[10]; uint8_t intern_border_y[26]; uint8_t topleft_border_y, topleft_border_u, topleft_border_v; void (*intra_pred_l[8])(uint8_t *d,uint8_t *top,uint8_t *left,int stride); void (*intra_pred_c[7])(uint8_t *d,uint8_t *top,uint8_t *left,int stride); uint8_t *col_type_base; /* scaling factors for MV prediction */ int sym_factor; ///< for scaling in symmetrical B block int direct_den[2]; ///< for scaling in direct B block int scale_den[2]; ///< for scaling neighbouring MVs uint8_t *edge_emu_buffer; int got_keyframe; int16_t *block; } AVSContext; extern const uint8_t ff_cavs_partition_flags[30]; extern const cavs_vector ff_cavs_intra_mv; extern const cavs_vector ff_cavs_dir_mv; static inline void set_mvs(cavs_vector *mv, enum cavs_block size) { switch(size) { case BLK_16X16: mv[MV_STRIDE ] = mv[0]; mv[MV_STRIDE+1] = mv[0]; case BLK_16X8: mv[1] = mv[0]; break; case BLK_8X16: mv[MV_STRIDE] = mv[0]; break; } } void ff_cavs_filter(AVSContext *h, enum cavs_mb mb_type); void ff_cavs_load_intra_pred_luma(AVSContext *h, uint8_t *top, uint8_t **left, int block); void ff_cavs_load_intra_pred_chroma(AVSContext *h); void ff_cavs_modify_mb_i(AVSContext *h, int *pred_mode_uv); void ff_cavs_inter(AVSContext *h, enum cavs_mb mb_type); void ff_cavs_mv(AVSContext *h, enum cavs_mv_loc nP, enum cavs_mv_loc nC, enum cavs_mv_pred mode, enum cavs_block size, int ref); void ff_cavs_init_mb(AVSContext *h); int ff_cavs_next_mb(AVSContext *h); void ff_cavs_init_pic(AVSContext *h); void ff_cavs_init_top_lines(AVSContext *h); int ff_cavs_init(AVCodecContext *avctx); int ff_cavs_end (AVCodecContext *avctx); #endif /* AVCODEC_CAVS_H */