[deb_ffmpeg.git] / ffmpeg / libavcodec / cavs.h

/*
 * Chinese AVS video (AVS1-P2, JiZhun profile) decoder.
 * Copyright (c) 2006  Stefan Gehrer <stefan.gehrer@gmx.de>
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg 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.
 *
 * FFmpeg 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 FFmpeg; 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 pic_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_chroma_qp[64];
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);
int 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 */
Commit	Line	Data
2ba45a60 DM	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[243];
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 */