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
61enum 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
76enum cavs_sub_mb {
77  B_SUB_DIRECT,
78  B_SUB_FWD,
79  B_SUB_BWD,
80  B_SUB_SYM
81};
82
83enum 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
94enum 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
104enum 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
113enum cavs_block {
114  BLK_16X16,
115  BLK_16X8,
116  BLK_8X16,
117  BLK_8X8
118};
119
120enum 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
143DECLARE_ALIGNED(8, typedef, struct) {
144    int16_t x;
145    int16_t y;
146    int16_t dist;
147    int16_t ref;
148} cavs_vector;
149
150struct 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
158typedef struct AVSFrame {
159    AVFrame *f;
160    int poc;
161} AVSFrame;
162
163typedef 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
245extern const uint8_t     ff_cavs_chroma_qp[64];
246extern const uint8_t     ff_cavs_partition_flags[30];
247extern const cavs_vector ff_cavs_intra_mv;
248extern const cavs_vector ff_cavs_dir_mv;
249
250static 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
264void ff_cavs_filter(AVSContext *h, enum cavs_mb mb_type);
265void ff_cavs_load_intra_pred_luma(AVSContext *h, uint8_t *top, uint8_t **left,
266                                  int block);
267void ff_cavs_load_intra_pred_chroma(AVSContext *h);
268void ff_cavs_modify_mb_i(AVSContext *h, int *pred_mode_uv);
269void ff_cavs_inter(AVSContext *h, enum cavs_mb mb_type);
270void 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);
272void ff_cavs_init_mb(AVSContext *h);
273int  ff_cavs_next_mb(AVSContext *h);
274int ff_cavs_init_pic(AVSContext *h);
275void ff_cavs_init_top_lines(AVSContext *h);
276int ff_cavs_init(AVCodecContext *avctx);
277int ff_cavs_end (AVCodecContext *avctx);
278
279#endif /* AVCODEC_CAVS_H */
280