mpeg12decdata.h: Move all tables to the only place they are used
[libav.git] / libavcodec / mpeg12.c
CommitLineData
de6d9b64 1/*
054480a5 2 * MPEG-1/2 decoder
6192b6f3 3 * Copyright (c) 2000, 2001 Fabrice Bellard
115329f1 4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
de6d9b64 5 *
2912e87a 6 * This file is part of Libav.
b78e7197 7 *
2912e87a 8 * Libav is free software; you can redistribute it and/or
ff4ec49e
FB
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
b78e7197 11 * version 2.1 of the License, or (at your option) any later version.
de6d9b64 12 *
2912e87a 13 * Libav is distributed in the hope that it will be useful,
de6d9b64 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
ff4ec49e
FB
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
de6d9b64 17 *
ff4ec49e 18 * You should have received a copy of the GNU Lesser General Public
2912e87a 19 * License along with Libav; if not, write to the Free Software
5509bffa 20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
de6d9b64 21 */
115329f1 22
983e3246 23/**
ba87f080 24 * @file
054480a5 25 * MPEG-1/2 decoder
983e3246 26 */
115329f1 27
6fee1b90 28#include "libavutil/attributes.h"
c269cf68 29#include "internal.h"
de6d9b64
FB
30#include "avcodec.h"
31#include "dsputil.h"
32#include "mpegvideo.h"
5f401b7b 33#include "error_resilience.h"
a6bc5731 34#include "mpeg12.h"
de6d9b64 35#include "mpeg12data.h"
92d6b7fd 36#include "bytestream.h"
4440bd0d 37#include "xvmc_internal.h"
6a9c8594 38#include "thread.h"
de6d9b64 39
a05ea77c
DB
40uint8_t ff_mpeg12_static_rl_table_store[2][2][2*MAX_RUN + MAX_LEVEL + 3];
41
38f64c03
DB
42static const uint8_t table_mb_ptype[7][2] = {
43 { 3, 5 }, // 0x01 MB_INTRA
44 { 1, 2 }, // 0x02 MB_PAT
45 { 1, 3 }, // 0x08 MB_FOR
46 { 1, 1 }, // 0x0A MB_FOR|MB_PAT
47 { 1, 6 }, // 0x11 MB_QUANT|MB_INTRA
48 { 1, 5 }, // 0x12 MB_QUANT|MB_PAT
49 { 2, 5 }, // 0x1A MB_QUANT|MB_FOR|MB_PAT
50};
51
52static const uint8_t table_mb_btype[11][2] = {
53 { 3, 5 }, // 0x01 MB_INTRA
54 { 2, 3 }, // 0x04 MB_BACK
55 { 3, 3 }, // 0x06 MB_BACK|MB_PAT
56 { 2, 4 }, // 0x08 MB_FOR
57 { 3, 4 }, // 0x0A MB_FOR|MB_PAT
58 { 2, 2 }, // 0x0C MB_FOR|MB_BACK
59 { 3, 2 }, // 0x0E MB_FOR|MB_BACK|MB_PAT
60 { 1, 6 }, // 0x11 MB_QUANT|MB_INTRA
61 { 2, 6 }, // 0x16 MB_QUANT|MB_BACK|MB_PAT
62 { 3, 6 }, // 0x1A MB_QUANT|MB_FOR|MB_PAT
63 { 2, 5 }, // 0x1E MB_QUANT|MB_FOR|MB_BACK|MB_PAT
64};
65
a05ea77c
DB
66#define INIT_2D_VLC_RL(rl, static_size)\
67{\
68 static RL_VLC_ELEM rl_vlc_table[static_size];\
69 INIT_VLC_STATIC(&rl.vlc, TEX_VLC_BITS, rl.n + 2,\
70 &rl.table_vlc[0][1], 4, 2,\
71 &rl.table_vlc[0][0], 4, 2, static_size);\
72\
73 rl.rl_vlc[0] = rl_vlc_table;\
74 init_2d_vlc_rl(&rl);\
75}
76
6fee1b90 77static av_cold void init_2d_vlc_rl(RLTable *rl)
a05ea77c
DB
78{
79 int i;
80
81 for (i = 0; i < rl->vlc.table_size; i++) {
82 int code = rl->vlc.table[i][0];
83 int len = rl->vlc.table[i][1];
84 int level, run;
85
86 if (len == 0) { // illegal code
87 run = 65;
88 level = MAX_LEVEL;
89 } else if (len<0) { //more bits needed
90 run = 0;
91 level = code;
6192b6f3 92 } else {
a05ea77c
DB
93 if (code == rl->n) { //esc
94 run = 65;
95 level = 0;
96 } else if (code == rl->n+1) { //eob
97 run = 0;
98 level = 127;
99 } else {
100 run = rl->table_run [code] + 1;
101 level = rl->table_level[code];
7bc9090a 102 }
a05ea77c
DB
103 }
104 rl->rl_vlc[0][i].len = len;
105 rl->rl_vlc[0][i].level = level;
106 rl->rl_vlc[0][i].run = run;
107 }
108}
7bc9090a 109
6fee1b90 110av_cold void ff_mpeg12_common_init(MpegEncContext *s)
a05ea77c 111{
356ab1da 112
a05ea77c
DB
113 s->y_dc_scale_table =
114 s->c_dc_scale_table = ff_mpeg2_dc_scale_table[s->intra_dc_precision];
356ab1da 115
a05ea77c 116}
356ab1da 117
a05ea77c
DB
118void ff_mpeg1_clean_buffers(MpegEncContext *s)
119{
120 s->last_dc[0] = 1 << (7 + s->intra_dc_precision);
121 s->last_dc[1] = s->last_dc[0];
122 s->last_dc[2] = s->last_dc[0];
123 memset(s->last_mv, 0, sizeof(s->last_mv));
124}
356ab1da 125
356ab1da 126
a05ea77c
DB
127/******************************************/
128/* decoding */
356ab1da 129
1b6d6674
DB
130VLC ff_mv_vlc;
131
a05ea77c
DB
132VLC ff_dc_lum_vlc;
133VLC ff_dc_chroma_vlc;
356ab1da 134
1b6d6674
DB
135VLC ff_mbincr_vlc;
136VLC ff_mb_ptype_vlc;
137VLC ff_mb_btype_vlc;
138VLC ff_mb_pat_vlc;
115329f1 139
a05ea77c
DB
140av_cold void ff_mpeg12_init_vlcs(void)
141{
142 static int done = 0;
115329f1 143
a05ea77c
DB
144 if (!done) {
145 done = 1;
7bc9090a 146
a05ea77c
DB
147 INIT_VLC_STATIC(&ff_dc_lum_vlc, DC_VLC_BITS, 12,
148 ff_mpeg12_vlc_dc_lum_bits, 1, 1,
149 ff_mpeg12_vlc_dc_lum_code, 2, 2, 512);
150 INIT_VLC_STATIC(&ff_dc_chroma_vlc, DC_VLC_BITS, 12,
151 ff_mpeg12_vlc_dc_chroma_bits, 1, 1,
152 ff_mpeg12_vlc_dc_chroma_code, 2, 2, 514);
1b6d6674 153 INIT_VLC_STATIC(&ff_mv_vlc, MV_VLC_BITS, 17,
a05ea77c
DB
154 &ff_mpeg12_mbMotionVectorTable[0][1], 2, 1,
155 &ff_mpeg12_mbMotionVectorTable[0][0], 2, 1, 518);
1b6d6674 156 INIT_VLC_STATIC(&ff_mbincr_vlc, MBINCR_VLC_BITS, 36,
a05ea77c
DB
157 &ff_mpeg12_mbAddrIncrTable[0][1], 2, 1,
158 &ff_mpeg12_mbAddrIncrTable[0][0], 2, 1, 538);
1b6d6674 159 INIT_VLC_STATIC(&ff_mb_pat_vlc, MB_PAT_VLC_BITS, 64,
a05ea77c
DB
160 &ff_mpeg12_mbPatTable[0][1], 2, 1,
161 &ff_mpeg12_mbPatTable[0][0], 2, 1, 512);
a579db0c 162
1b6d6674 163 INIT_VLC_STATIC(&ff_mb_ptype_vlc, MB_PTYPE_VLC_BITS, 7,
a05ea77c
DB
164 &table_mb_ptype[0][1], 2, 1,
165 &table_mb_ptype[0][0], 2, 1, 64);
1b6d6674 166 INIT_VLC_STATIC(&ff_mb_btype_vlc, MB_BTYPE_VLC_BITS, 11,
a05ea77c
DB
167 &table_mb_btype[0][1], 2, 1,
168 &table_mb_btype[0][0], 2, 1, 64);
e96b4a53
MS
169 ff_init_rl(&ff_rl_mpeg1, ff_mpeg12_static_rl_table_store[0]);
170 ff_init_rl(&ff_rl_mpeg2, ff_mpeg12_static_rl_table_store[1]);
115329f1 171
a05ea77c
DB
172 INIT_2D_VLC_RL(ff_rl_mpeg1, 680);
173 INIT_2D_VLC_RL(ff_rl_mpeg2, 674);
de6d9b64 174 }
de6d9b64
FB
175}
176
1b6d6674
DB
177/**
178 * Find the end of the current frame in the bitstream.
179 * @return the position of the first byte of the next frame, or -1
180 */
181int ff_mpeg1_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_size, AVCodecParserContext *s)
a05ea77c 182{
1b6d6674
DB
183 int i;
184 uint32_t state = pc->state;
115329f1 185
1b6d6674
DB
186 /* EOF considered as end of frame */
187 if (buf_size == 0)
a05ea77c 188 return 0;
8db1a1dd 189
1b6d6674
DB
190/*
191 0 frame start -> 1/4
192 1 first_SEQEXT -> 0/2
193 2 first field start -> 3/0
194 3 second_SEQEXT -> 2/0
195 4 searching end
196*/
115329f1 197
1b6d6674
DB
198 for (i = 0; i < buf_size; i++) {
199 assert(pc->frame_start_found >= 0 && pc->frame_start_found <= 4);
200 if (pc->frame_start_found & 1) {
201 if (state == EXT_START_CODE && (buf[i] & 0xF0) != 0x80)
202 pc->frame_start_found--;
203 else if (state == EXT_START_CODE + 2) {
204 if ((buf[i] & 3) == 3)
205 pc->frame_start_found = 0;
206 else
207 pc->frame_start_found = (pc->frame_start_found + 1) & 3;
a05ea77c 208 }
1b6d6674 209 state++;
a05ea77c 210 } else {
1b6d6674
DB
211 i = avpriv_find_start_code(buf + i, buf + buf_size, &state) - buf - 1;
212 if (pc->frame_start_found == 0 && state >= SLICE_MIN_START_CODE && state <= SLICE_MAX_START_CODE) {
213 i++;
214 pc->frame_start_found = 4;
a05ea77c 215 }
1b6d6674
DB
216 if (state == SEQ_END_CODE) {
217 pc->frame_start_found = 0;
218 pc->state=-1;
219 return i+1;
a05ea77c 220 }
1b6d6674
DB
221 if (pc->frame_start_found == 2 && state == SEQ_START_CODE)
222 pc->frame_start_found = 0;
223 if (pc->frame_start_found < 4 && state == EXT_START_CODE)
224 pc->frame_start_found++;
225 if (pc->frame_start_found == 4 && (state & 0xFFFFFF00) == 0x100) {
226 if (state < SLICE_MIN_START_CODE || state > SLICE_MAX_START_CODE) {
227 pc->frame_start_found = 0;
228 pc->state = -1;
229 return i - 3;
a05ea77c
DB
230 }
231 }
1b6d6674
DB
232 if (pc->frame_start_found == 0 && s && state == PICTURE_START_CODE) {
233 ff_fetch_timestamp(s, i - 3, 1);
dff0f035 234 }
dff0f035 235 }
dff0f035 236 }
1b6d6674
DB
237 pc->state = state;
238 return END_NOT_FOUND;
de6d9b64 239}