| 1 | /* |
| 2 | * Chinese AVS video (AVS1-P2, JiZhun profile) decoder. |
| 3 | * Copyright (c) 2006 Stefan Gehrer <stefan.gehrer@gmx.de> |
| 4 | * |
| 5 | * This file is part of FFmpeg. |
| 6 | * |
| 7 | * FFmpeg is free software; you can redistribute it and/or |
| 8 | * modify it under the terms of the GNU Lesser General Public |
| 9 | * License as published by the Free Software Foundation; either |
| 10 | * version 2.1 of the License, or (at your option) any later version. |
| 11 | * |
| 12 | * FFmpeg is distributed in the hope that it will be useful, |
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 15 | * Lesser General Public License for more details. |
| 16 | * |
| 17 | * You should have received a copy of the GNU Lesser General Public |
| 18 | * License along with FFmpeg; if not, write to the Free Software |
| 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 20 | */ |
| 21 | |
| 22 | #ifndef AVCODEC_CAVS_H |
| 23 | #define AVCODEC_CAVS_H |
| 24 | |
| 25 | #include "cavsdsp.h" |
| 26 | #include "blockdsp.h" |
| 27 | #include "h264chroma.h" |
| 28 | #include "idctdsp.h" |
| 29 | #include "get_bits.h" |
| 30 | #include "videodsp.h" |
| 31 | |
| 32 | #define SLICE_MAX_START_CODE 0x000001af |
| 33 | #define EXT_START_CODE 0x000001b5 |
| 34 | #define USER_START_CODE 0x000001b2 |
| 35 | #define CAVS_START_CODE 0x000001b0 |
| 36 | #define PIC_I_START_CODE 0x000001b3 |
| 37 | #define PIC_PB_START_CODE 0x000001b6 |
| 38 | |
| 39 | #define A_AVAIL 1 |
| 40 | #define B_AVAIL 2 |
| 41 | #define C_AVAIL 4 |
| 42 | #define D_AVAIL 8 |
| 43 | #define NOT_AVAIL -1 |
| 44 | #define REF_INTRA -2 |
| 45 | #define REF_DIR -3 |
| 46 | |
| 47 | #define ESCAPE_CODE 59 |
| 48 | |
| 49 | #define FWD0 0x01 |
| 50 | #define FWD1 0x02 |
| 51 | #define BWD0 0x04 |
| 52 | #define BWD1 0x08 |
| 53 | #define SYM0 0x10 |
| 54 | #define SYM1 0x20 |
| 55 | #define SPLITH 0x40 |
| 56 | #define SPLITV 0x80 |
| 57 | |
| 58 | #define MV_BWD_OFFS 12 |
| 59 | #define MV_STRIDE 4 |
| 60 | |
| 61 | enum cavs_mb { |
| 62 | I_8X8 = 0, |
| 63 | P_SKIP, |
| 64 | P_16X16, |
| 65 | P_16X8, |
| 66 | P_8X16, |
| 67 | P_8X8, |
| 68 | B_SKIP, |
| 69 | B_DIRECT, |
| 70 | B_FWD_16X16, |
| 71 | B_BWD_16X16, |
| 72 | B_SYM_16X16, |
| 73 | B_8X8 = 29 |
| 74 | }; |
| 75 | |
| 76 | enum cavs_sub_mb { |
| 77 | B_SUB_DIRECT, |
| 78 | B_SUB_FWD, |
| 79 | B_SUB_BWD, |
| 80 | B_SUB_SYM |
| 81 | }; |
| 82 | |
| 83 | enum cavs_intra_luma { |
| 84 | INTRA_L_VERT, |
| 85 | INTRA_L_HORIZ, |
| 86 | INTRA_L_LP, |
| 87 | INTRA_L_DOWN_LEFT, |
| 88 | INTRA_L_DOWN_RIGHT, |
| 89 | INTRA_L_LP_LEFT, |
| 90 | INTRA_L_LP_TOP, |
| 91 | INTRA_L_DC_128 |
| 92 | }; |
| 93 | |
| 94 | enum cavs_intra_chroma { |
| 95 | INTRA_C_LP, |
| 96 | INTRA_C_HORIZ, |
| 97 | INTRA_C_VERT, |
| 98 | INTRA_C_PLANE, |
| 99 | INTRA_C_LP_LEFT, |
| 100 | INTRA_C_LP_TOP, |
| 101 | INTRA_C_DC_128, |
| 102 | }; |
| 103 | |
| 104 | enum cavs_mv_pred { |
| 105 | MV_PRED_MEDIAN, |
| 106 | MV_PRED_LEFT, |
| 107 | MV_PRED_TOP, |
| 108 | MV_PRED_TOPRIGHT, |
| 109 | MV_PRED_PSKIP, |
| 110 | MV_PRED_BSKIP |
| 111 | }; |
| 112 | |
| 113 | enum cavs_block { |
| 114 | BLK_16X16, |
| 115 | BLK_16X8, |
| 116 | BLK_8X16, |
| 117 | BLK_8X8 |
| 118 | }; |
| 119 | |
| 120 | enum cavs_mv_loc { |
| 121 | MV_FWD_D3 = 0, |
| 122 | MV_FWD_B2, |
| 123 | MV_FWD_B3, |
| 124 | MV_FWD_C2, |
| 125 | MV_FWD_A1, |
| 126 | MV_FWD_X0, |
| 127 | MV_FWD_X1, |
| 128 | MV_FWD_A3 = 8, |
| 129 | MV_FWD_X2, |
| 130 | MV_FWD_X3, |
| 131 | MV_BWD_D3 = MV_BWD_OFFS, |
| 132 | MV_BWD_B2, |
| 133 | MV_BWD_B3, |
| 134 | MV_BWD_C2, |
| 135 | MV_BWD_A1, |
| 136 | MV_BWD_X0, |
| 137 | MV_BWD_X1, |
| 138 | MV_BWD_A3 = MV_BWD_OFFS+8, |
| 139 | MV_BWD_X2, |
| 140 | MV_BWD_X3 |
| 141 | }; |
| 142 | |
| 143 | DECLARE_ALIGNED(8, typedef, struct) { |
| 144 | int16_t x; |
| 145 | int16_t y; |
| 146 | int16_t dist; |
| 147 | int16_t ref; |
| 148 | } cavs_vector; |
| 149 | |
| 150 | struct dec_2dvlc { |
| 151 | int8_t rltab[59][3]; |
| 152 | int8_t level_add[27]; |
| 153 | int8_t golomb_order; |
| 154 | int inc_limit; |
| 155 | int8_t max_run; |
| 156 | }; |
| 157 | |
| 158 | typedef struct AVSFrame { |
| 159 | AVFrame *f; |
| 160 | int poc; |
| 161 | } AVSFrame; |
| 162 | |
| 163 | typedef struct AVSContext { |
| 164 | AVCodecContext *avctx; |
| 165 | BlockDSPContext bdsp; |
| 166 | H264ChromaContext h264chroma; |
| 167 | IDCTDSPContext idsp; |
| 168 | VideoDSPContext vdsp; |
| 169 | CAVSDSPContext cdsp; |
| 170 | GetBitContext gb; |
| 171 | AVSFrame cur; ///< currently decoded frame |
| 172 | AVSFrame DPB[2]; ///< reference frames |
| 173 | int dist[2]; ///< temporal distances from current frame to ref frames |
| 174 | int low_delay; |
| 175 | int profile, level; |
| 176 | int aspect_ratio; |
| 177 | int mb_width, mb_height; |
| 178 | int width, height; |
| 179 | int stream_revision; ///<0 for samples from 2006, 1 for rm52j encoder |
| 180 | int progressive; |
| 181 | int pic_structure; |
| 182 | int skip_mode_flag; ///< select between skip_count or one skip_flag per MB |
| 183 | int loop_filter_disable; |
| 184 | int alpha_offset, beta_offset; |
| 185 | int ref_flag; |
| 186 | int mbx, mby, mbidx; ///< macroblock coordinates |
| 187 | int flags; ///< availability flags of neighbouring macroblocks |
| 188 | int stc; ///< last start code |
| 189 | uint8_t *cy, *cu, *cv; ///< current MB sample pointers |
| 190 | int left_qp; |
| 191 | uint8_t *top_qp; |
| 192 | |
| 193 | /** mv motion vector cache |
| 194 | 0: D3 B2 B3 C2 |
| 195 | 4: A1 X0 X1 - |
| 196 | 8: A3 X2 X3 - |
| 197 | |
| 198 | X are the vectors in the current macroblock (5,6,9,10) |
| 199 | A is the macroblock to the left (4,8) |
| 200 | B is the macroblock to the top (1,2) |
| 201 | C is the macroblock to the top-right (3) |
| 202 | D is the macroblock to the top-left (0) |
| 203 | |
| 204 | the same is repeated for backward motion vectors */ |
| 205 | cavs_vector mv[2*4*3]; |
| 206 | cavs_vector *top_mv[2]; |
| 207 | cavs_vector *col_mv; |
| 208 | |
| 209 | /** luma pred mode cache |
| 210 | 0: -- B2 B3 |
| 211 | 3: A1 X0 X1 |
| 212 | 6: A3 X2 X3 */ |
| 213 | int pred_mode_Y[3*3]; |
| 214 | int *top_pred_Y; |
| 215 | ptrdiff_t l_stride, c_stride; |
| 216 | int luma_scan[4]; |
| 217 | int qp; |
| 218 | int qp_fixed; |
| 219 | int pic_qp_fixed; |
| 220 | int cbp; |
| 221 | ScanTable scantable; |
| 222 | |
| 223 | /** intra prediction is done with un-deblocked samples |
| 224 | they are saved here before deblocking the MB */ |
| 225 | uint8_t *top_border_y, *top_border_u, *top_border_v; |
| 226 | uint8_t left_border_y[26], left_border_u[10], left_border_v[10]; |
| 227 | uint8_t intern_border_y[26]; |
| 228 | uint8_t topleft_border_y, topleft_border_u, topleft_border_v; |
| 229 | |
| 230 | void (*intra_pred_l[8])(uint8_t *d,uint8_t *top,uint8_t *left,int stride); |
| 231 | void (*intra_pred_c[7])(uint8_t *d,uint8_t *top,uint8_t *left,int stride); |
| 232 | uint8_t *col_type_base; |
| 233 | |
| 234 | /* scaling factors for MV prediction */ |
| 235 | int sym_factor; ///< for scaling in symmetrical B block |
| 236 | int direct_den[2]; ///< for scaling in direct B block |
| 237 | int scale_den[2]; ///< for scaling neighbouring MVs |
| 238 | |
| 239 | uint8_t *edge_emu_buffer; |
| 240 | |
| 241 | int got_keyframe; |
| 242 | int16_t *block; |
| 243 | } AVSContext; |
| 244 | |
| 245 | extern const uint8_t ff_cavs_chroma_qp[64]; |
| 246 | extern const uint8_t ff_cavs_partition_flags[30]; |
| 247 | extern const cavs_vector ff_cavs_intra_mv; |
| 248 | extern const cavs_vector ff_cavs_dir_mv; |
| 249 | |
| 250 | static inline void set_mvs(cavs_vector *mv, enum cavs_block size) { |
| 251 | switch(size) { |
| 252 | case BLK_16X16: |
| 253 | mv[MV_STRIDE ] = mv[0]; |
| 254 | mv[MV_STRIDE+1] = mv[0]; |
| 255 | case BLK_16X8: |
| 256 | mv[1] = mv[0]; |
| 257 | break; |
| 258 | case BLK_8X16: |
| 259 | mv[MV_STRIDE] = mv[0]; |
| 260 | break; |
| 261 | } |
| 262 | } |
| 263 | |
| 264 | void ff_cavs_filter(AVSContext *h, enum cavs_mb mb_type); |
| 265 | void ff_cavs_load_intra_pred_luma(AVSContext *h, uint8_t *top, uint8_t **left, |
| 266 | int block); |
| 267 | void ff_cavs_load_intra_pred_chroma(AVSContext *h); |
| 268 | void ff_cavs_modify_mb_i(AVSContext *h, int *pred_mode_uv); |
| 269 | void ff_cavs_inter(AVSContext *h, enum cavs_mb mb_type); |
| 270 | void ff_cavs_mv(AVSContext *h, enum cavs_mv_loc nP, enum cavs_mv_loc nC, |
| 271 | enum cavs_mv_pred mode, enum cavs_block size, int ref); |
| 272 | void ff_cavs_init_mb(AVSContext *h); |
| 273 | int ff_cavs_next_mb(AVSContext *h); |
| 274 | int ff_cavs_init_pic(AVSContext *h); |
| 275 | void ff_cavs_init_top_lines(AVSContext *h); |
| 276 | int ff_cavs_init(AVCodecContext *avctx); |
| 277 | int ff_cavs_end (AVCodecContext *avctx); |
| 278 | |
| 279 | #endif /* AVCODEC_CAVS_H */ |