1/*
2 * H26L/H264/AVC/JVT/14496-10/... encoder/decoder
3 * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
4 *
5 * This file is part of Libav.
6 *
7 * Libav 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 * Libav 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 Libav; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22/**
23 * @file
24 * @brief
25 *     H264 / AVC / MPEG4 part10 codec data table
26 * @author Michael Niedermayer <michaelni@gmx.at>
27 */
28
29#ifndef AVCODEC_H264DATA_H
30#define AVCODEC_H264DATA_H
31
32#include <stdint.h>
33#include "libavutil/rational.h"
34#include "mpegvideo.h"
35#include "h264.h"
36
37
38static const uint8_t golomb_to_pict_type[5]=
39{AV_PICTURE_TYPE_P, AV_PICTURE_TYPE_B, AV_PICTURE_TYPE_I, AV_PICTURE_TYPE_SP, AV_PICTURE_TYPE_SI};
40
41static const uint8_t golomb_to_intra4x4_cbp[48]={
42 47, 31, 15,  0, 23, 27, 29, 30,  7, 11, 13, 14, 39, 43, 45, 46,
43 16,  3,  5, 10, 12, 19, 21, 26, 28, 35, 37, 42, 44,  1,  2,  4,
44  8, 17, 18, 20, 24,  6,  9, 22, 25, 32, 33, 34, 36, 40, 38, 41
45};
46
47static const uint8_t golomb_to_inter_cbp[48]={
48  0, 16,  1,  2,  4,  8, 32,  3,  5, 10, 12, 15, 47,  7, 11, 13,
49 14,  6,  9, 31, 35, 37, 42, 44, 33, 34, 36, 40, 39, 43, 45, 46,
50 17, 18, 20, 24, 19, 21, 26, 28, 23, 27, 29, 30, 22, 25, 38, 41
51};
52
53static const uint8_t zigzag_scan[16]={
54 0+0*4, 1+0*4, 0+1*4, 0+2*4,
55 1+1*4, 2+0*4, 3+0*4, 2+1*4,
56 1+2*4, 0+3*4, 1+3*4, 2+2*4,
57 3+1*4, 3+2*4, 2+3*4, 3+3*4,
58};
59
60static const uint8_t field_scan[16]={
61 0+0*4, 0+1*4, 1+0*4, 0+2*4,
62 0+3*4, 1+1*4, 1+2*4, 1+3*4,
63 2+0*4, 2+1*4, 2+2*4, 2+3*4,
64 3+0*4, 3+1*4, 3+2*4, 3+3*4,
65};
66
67static const uint8_t luma_dc_zigzag_scan[16]={
68 0*16 + 0*64, 1*16 + 0*64, 2*16 + 0*64, 0*16 + 2*64,
69 3*16 + 0*64, 0*16 + 1*64, 1*16 + 1*64, 2*16 + 1*64,
70 1*16 + 2*64, 2*16 + 2*64, 3*16 + 2*64, 0*16 + 3*64,
71 3*16 + 1*64, 1*16 + 3*64, 2*16 + 3*64, 3*16 + 3*64,
72};
73
74static const uint8_t luma_dc_field_scan[16]={
75 0*16 + 0*64, 2*16 + 0*64, 1*16 + 0*64, 0*16 + 2*64,
76 2*16 + 2*64, 3*16 + 0*64, 1*16 + 2*64, 3*16 + 2*64,
77 0*16 + 1*64, 2*16 + 1*64, 0*16 + 3*64, 2*16 + 3*64,
78 1*16 + 1*64, 3*16 + 1*64, 1*16 + 3*64, 3*16 + 3*64,
79};
80
81static const uint8_t chroma_dc_scan[4]={
82 (0+0*2)*16, (1+0*2)*16,
83 (0+1*2)*16, (1+1*2)*16,
84};
85
86static const uint8_t chroma422_dc_scan[8]={
87 (0+0*2)*16, (0+1*2)*16,
88 (1+0*2)*16, (0+2*2)*16,
89 (0+3*2)*16, (1+1*2)*16,
90 (1+2*2)*16, (1+3*2)*16,
91};
92
93// zigzag_scan8x8_cavlc[i] = zigzag_scan8x8[(i/4) + 16*(i%4)]
94static const uint8_t zigzag_scan8x8_cavlc[64]={
95 0+0*8, 1+1*8, 1+2*8, 2+2*8,
96 4+1*8, 0+5*8, 3+3*8, 7+0*8,
97 3+4*8, 1+7*8, 5+3*8, 6+3*8,
98 2+7*8, 6+4*8, 5+6*8, 7+5*8,
99 1+0*8, 2+0*8, 0+3*8, 3+1*8,
100 3+2*8, 0+6*8, 4+2*8, 6+1*8,
101 2+5*8, 2+6*8, 6+2*8, 5+4*8,
102 3+7*8, 7+3*8, 4+7*8, 7+6*8,
103 0+1*8, 3+0*8, 0+4*8, 4+0*8,
104 2+3*8, 1+5*8, 5+1*8, 5+2*8,
105 1+6*8, 3+5*8, 7+1*8, 4+5*8,
106 4+6*8, 7+4*8, 5+7*8, 6+7*8,
107 0+2*8, 2+1*8, 1+3*8, 5+0*8,
108 1+4*8, 2+4*8, 6+0*8, 4+3*8,
109 0+7*8, 4+4*8, 7+2*8, 3+6*8,
110 5+5*8, 6+5*8, 6+6*8, 7+7*8,
111};
112
113static const uint8_t field_scan8x8[64]={
114 0+0*8, 0+1*8, 0+2*8, 1+0*8,
115 1+1*8, 0+3*8, 0+4*8, 1+2*8,
116 2+0*8, 1+3*8, 0+5*8, 0+6*8,
117 0+7*8, 1+4*8, 2+1*8, 3+0*8,
118 2+2*8, 1+5*8, 1+6*8, 1+7*8,
119 2+3*8, 3+1*8, 4+0*8, 3+2*8,
120 2+4*8, 2+5*8, 2+6*8, 2+7*8,
121 3+3*8, 4+1*8, 5+0*8, 4+2*8,
122 3+4*8, 3+5*8, 3+6*8, 3+7*8,
123 4+3*8, 5+1*8, 6+0*8, 5+2*8,
124 4+4*8, 4+5*8, 4+6*8, 4+7*8,
125 5+3*8, 6+1*8, 6+2*8, 5+4*8,
126 5+5*8, 5+6*8, 5+7*8, 6+3*8,
127 7+0*8, 7+1*8, 6+4*8, 6+5*8,
128 6+6*8, 6+7*8, 7+2*8, 7+3*8,
129 7+4*8, 7+5*8, 7+6*8, 7+7*8,
130};
131
132static const uint8_t field_scan8x8_cavlc[64]={
133 0+0*8, 1+1*8, 2+0*8, 0+7*8,
134 2+2*8, 2+3*8, 2+4*8, 3+3*8,
135 3+4*8, 4+3*8, 4+4*8, 5+3*8,
136 5+5*8, 7+0*8, 6+6*8, 7+4*8,
137 0+1*8, 0+3*8, 1+3*8, 1+4*8,
138 1+5*8, 3+1*8, 2+5*8, 4+1*8,
139 3+5*8, 5+1*8, 4+5*8, 6+1*8,
140 5+6*8, 7+1*8, 6+7*8, 7+5*8,
141 0+2*8, 0+4*8, 0+5*8, 2+1*8,
142 1+6*8, 4+0*8, 2+6*8, 5+0*8,
143 3+6*8, 6+0*8, 4+6*8, 6+2*8,
144 5+7*8, 6+4*8, 7+2*8, 7+6*8,
145 1+0*8, 1+2*8, 0+6*8, 3+0*8,
146 1+7*8, 3+2*8, 2+7*8, 4+2*8,
147 3+7*8, 5+2*8, 4+7*8, 5+4*8,
148 6+3*8, 6+5*8, 7+3*8, 7+7*8,
149};
150
151typedef struct IMbInfo{
152    uint16_t type;
153    uint8_t pred_mode;
154    uint8_t cbp;
155} IMbInfo;
156
157static const IMbInfo i_mb_type_info[26]={
158{MB_TYPE_INTRA4x4  , -1, -1},
159{MB_TYPE_INTRA16x16,  2,  0},
160{MB_TYPE_INTRA16x16,  1,  0},
161{MB_TYPE_INTRA16x16,  0,  0},
162{MB_TYPE_INTRA16x16,  3,  0},
163{MB_TYPE_INTRA16x16,  2,  16},
164{MB_TYPE_INTRA16x16,  1,  16},
165{MB_TYPE_INTRA16x16,  0,  16},
166{MB_TYPE_INTRA16x16,  3,  16},
167{MB_TYPE_INTRA16x16,  2,  32},
168{MB_TYPE_INTRA16x16,  1,  32},
169{MB_TYPE_INTRA16x16,  0,  32},
170{MB_TYPE_INTRA16x16,  3,  32},
171{MB_TYPE_INTRA16x16,  2,  15+0},
172{MB_TYPE_INTRA16x16,  1,  15+0},
173{MB_TYPE_INTRA16x16,  0,  15+0},
174{MB_TYPE_INTRA16x16,  3,  15+0},
175{MB_TYPE_INTRA16x16,  2,  15+16},
176{MB_TYPE_INTRA16x16,  1,  15+16},
177{MB_TYPE_INTRA16x16,  0,  15+16},
178{MB_TYPE_INTRA16x16,  3,  15+16},
179{MB_TYPE_INTRA16x16,  2,  15+32},
180{MB_TYPE_INTRA16x16,  1,  15+32},
181{MB_TYPE_INTRA16x16,  0,  15+32},
182{MB_TYPE_INTRA16x16,  3,  15+32},
183{MB_TYPE_INTRA_PCM , -1, -1},
184};
185
186typedef struct PMbInfo{
187    uint16_t type;
188    uint8_t partition_count;
189} PMbInfo;
190
191static const PMbInfo p_mb_type_info[5]={
192{MB_TYPE_16x16|MB_TYPE_P0L0             , 1},
193{MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P1L0, 2},
194{MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P1L0, 2},
195{MB_TYPE_8x8  |MB_TYPE_P0L0|MB_TYPE_P1L0, 4},
196{MB_TYPE_8x8  |MB_TYPE_P0L0|MB_TYPE_P1L0|MB_TYPE_REF0, 4},
197};
198
199static const PMbInfo p_sub_mb_type_info[4]={
200{MB_TYPE_16x16|MB_TYPE_P0L0             , 1},
201{MB_TYPE_16x8 |MB_TYPE_P0L0             , 2},
202{MB_TYPE_8x16 |MB_TYPE_P0L0             , 2},
203{MB_TYPE_8x8  |MB_TYPE_P0L0             , 4},
204};
205
206static const PMbInfo b_mb_type_info[23]={
207{MB_TYPE_DIRECT2|MB_TYPE_L0L1                                      , 1, },
208{MB_TYPE_16x16|MB_TYPE_P0L0                                       , 1, },
209{MB_TYPE_16x16             |MB_TYPE_P0L1                          , 1, },
210{MB_TYPE_16x16|MB_TYPE_P0L0|MB_TYPE_P0L1                          , 1, },
211{MB_TYPE_16x8 |MB_TYPE_P0L0             |MB_TYPE_P1L0             , 2, },
212{MB_TYPE_8x16 |MB_TYPE_P0L0             |MB_TYPE_P1L0             , 2, },
213{MB_TYPE_16x8              |MB_TYPE_P0L1             |MB_TYPE_P1L1, 2, },
214{MB_TYPE_8x16              |MB_TYPE_P0L1             |MB_TYPE_P1L1, 2, },
215{MB_TYPE_16x8 |MB_TYPE_P0L0                          |MB_TYPE_P1L1, 2, },
216{MB_TYPE_8x16 |MB_TYPE_P0L0                          |MB_TYPE_P1L1, 2, },
217{MB_TYPE_16x8              |MB_TYPE_P0L1|MB_TYPE_P1L0             , 2, },
218{MB_TYPE_8x16              |MB_TYPE_P0L1|MB_TYPE_P1L0             , 2, },
219{MB_TYPE_16x8 |MB_TYPE_P0L0             |MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
220{MB_TYPE_8x16 |MB_TYPE_P0L0             |MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
221{MB_TYPE_16x8              |MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
222{MB_TYPE_8x16              |MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
223{MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0             , 2, },
224{MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0             , 2, },
225{MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1             |MB_TYPE_P1L1, 2, },
226{MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1             |MB_TYPE_P1L1, 2, },
227{MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
228{MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
229{MB_TYPE_8x8  |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 4, },
230};
231
232static const PMbInfo b_sub_mb_type_info[13]={
233{MB_TYPE_DIRECT2                                                   , 1, },
234{MB_TYPE_16x16|MB_TYPE_P0L0                                       , 1, },
235{MB_TYPE_16x16             |MB_TYPE_P0L1                          , 1, },
236{MB_TYPE_16x16|MB_TYPE_P0L0|MB_TYPE_P0L1                          , 1, },
237{MB_TYPE_16x8 |MB_TYPE_P0L0             |MB_TYPE_P1L0             , 2, },
238{MB_TYPE_8x16 |MB_TYPE_P0L0             |MB_TYPE_P1L0             , 2, },
239{MB_TYPE_16x8              |MB_TYPE_P0L1             |MB_TYPE_P1L1, 2, },
240{MB_TYPE_8x16              |MB_TYPE_P0L1             |MB_TYPE_P1L1, 2, },
241{MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
242{MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
243{MB_TYPE_8x8  |MB_TYPE_P0L0             |MB_TYPE_P1L0             , 4, },
244{MB_TYPE_8x8               |MB_TYPE_P0L1             |MB_TYPE_P1L1, 4, },
245{MB_TYPE_8x8  |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 4, },
246};
247
248static const uint8_t dequant4_coeff_init[6][3]={
249  {10,13,16},
250  {11,14,18},
251  {13,16,20},
252  {14,18,23},
253  {16,20,25},
254  {18,23,29},
255};
256
257static const uint8_t dequant8_coeff_init_scan[16] = {
258  0,3,4,3, 3,1,5,1, 4,5,2,5, 3,1,5,1
259};
260static const uint8_t dequant8_coeff_init[6][6]={
261  {20,18,32,19,25,24},
262  {22,19,35,21,28,26},
263  {26,23,42,24,33,31},
264  {28,25,45,26,35,33},
265  {32,28,51,30,40,38},
266  {36,32,58,34,46,43},
267};
268
269#endif /* AVCODEC_H264DATA_H */
270