?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
cosmetics after last commit
[ffmpeg-lucabe.git] / libavcodec / rv10.c
blobd552bf9501c35bb3cf00059b135a3dc0fc026df9
1 /*
2 * RV10/RV20 decoder
3 * Copyright (c) 2000,2001 Fabrice Bellard
4 * Copyright (c) 2002-2004 Michael Niedermayer
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 /**
24 * @file libavcodec/rv10.c
25 * RV10/RV20 decoder
26 */
27
28 #include "avcodec.h"
29 #include "dsputil.h"
30 #include "mpegvideo.h"
31
32 //#define DEBUG
33
34 #define DC_VLC_BITS 14 //FIXME find a better solution
35
36 static const uint16_t rv_lum_code[256] =
37 {
38 0x3e7f, 0x0f00, 0x0f01, 0x0f02, 0x0f03, 0x0f04, 0x0f05, 0x0f06,
39 0x0f07, 0x0f08, 0x0f09, 0x0f0a, 0x0f0b, 0x0f0c, 0x0f0d, 0x0f0e,
40 0x0f0f, 0x0f10, 0x0f11, 0x0f12, 0x0f13, 0x0f14, 0x0f15, 0x0f16,
41 0x0f17, 0x0f18, 0x0f19, 0x0f1a, 0x0f1b, 0x0f1c, 0x0f1d, 0x0f1e,
42 0x0f1f, 0x0f20, 0x0f21, 0x0f22, 0x0f23, 0x0f24, 0x0f25, 0x0f26,
43 0x0f27, 0x0f28, 0x0f29, 0x0f2a, 0x0f2b, 0x0f2c, 0x0f2d, 0x0f2e,
44 0x0f2f, 0x0f30, 0x0f31, 0x0f32, 0x0f33, 0x0f34, 0x0f35, 0x0f36,
45 0x0f37, 0x0f38, 0x0f39, 0x0f3a, 0x0f3b, 0x0f3c, 0x0f3d, 0x0f3e,
46 0x0f3f, 0x0380, 0x0381, 0x0382, 0x0383, 0x0384, 0x0385, 0x0386,
47 0x0387, 0x0388, 0x0389, 0x038a, 0x038b, 0x038c, 0x038d, 0x038e,
48 0x038f, 0x0390, 0x0391, 0x0392, 0x0393, 0x0394, 0x0395, 0x0396,
49 0x0397, 0x0398, 0x0399, 0x039a, 0x039b, 0x039c, 0x039d, 0x039e,
50 0x039f, 0x00c0, 0x00c1, 0x00c2, 0x00c3, 0x00c4, 0x00c5, 0x00c6,
51 0x00c7, 0x00c8, 0x00c9, 0x00ca, 0x00cb, 0x00cc, 0x00cd, 0x00ce,
52 0x00cf, 0x0050, 0x0051, 0x0052, 0x0053, 0x0054, 0x0055, 0x0056,
53 0x0057, 0x0020, 0x0021, 0x0022, 0x0023, 0x000c, 0x000d, 0x0004,
54 0x0000, 0x0005, 0x000e, 0x000f, 0x0024, 0x0025, 0x0026, 0x0027,
55 0x0058, 0x0059, 0x005a, 0x005b, 0x005c, 0x005d, 0x005e, 0x005f,
56 0x00d0, 0x00d1, 0x00d2, 0x00d3, 0x00d4, 0x00d5, 0x00d6, 0x00d7,
57 0x00d8, 0x00d9, 0x00da, 0x00db, 0x00dc, 0x00dd, 0x00de, 0x00df,
58 0x03a0, 0x03a1, 0x03a2, 0x03a3, 0x03a4, 0x03a5, 0x03a6, 0x03a7,
59 0x03a8, 0x03a9, 0x03aa, 0x03ab, 0x03ac, 0x03ad, 0x03ae, 0x03af,
60 0x03b0, 0x03b1, 0x03b2, 0x03b3, 0x03b4, 0x03b5, 0x03b6, 0x03b7,
61 0x03b8, 0x03b9, 0x03ba, 0x03bb, 0x03bc, 0x03bd, 0x03be, 0x03bf,
62 0x0f40, 0x0f41, 0x0f42, 0x0f43, 0x0f44, 0x0f45, 0x0f46, 0x0f47,
63 0x0f48, 0x0f49, 0x0f4a, 0x0f4b, 0x0f4c, 0x0f4d, 0x0f4e, 0x0f4f,
64 0x0f50, 0x0f51, 0x0f52, 0x0f53, 0x0f54, 0x0f55, 0x0f56, 0x0f57,
65 0x0f58, 0x0f59, 0x0f5a, 0x0f5b, 0x0f5c, 0x0f5d, 0x0f5e, 0x0f5f,
66 0x0f60, 0x0f61, 0x0f62, 0x0f63, 0x0f64, 0x0f65, 0x0f66, 0x0f67,
67 0x0f68, 0x0f69, 0x0f6a, 0x0f6b, 0x0f6c, 0x0f6d, 0x0f6e, 0x0f6f,
68 0x0f70, 0x0f71, 0x0f72, 0x0f73, 0x0f74, 0x0f75, 0x0f76, 0x0f77,
69 0x0f78, 0x0f79, 0x0f7a, 0x0f7b, 0x0f7c, 0x0f7d, 0x0f7e, 0x0f7f,
70 };
71
72 static const uint8_t rv_lum_bits[256] =
73 {
74 14, 12, 12, 12, 12, 12, 12, 12,
75 12, 12, 12, 12, 12, 12, 12, 12,
76 12, 12, 12, 12, 12, 12, 12, 12,
77 12, 12, 12, 12, 12, 12, 12, 12,
78 12, 12, 12, 12, 12, 12, 12, 12,
79 12, 12, 12, 12, 12, 12, 12, 12,
80 12, 12, 12, 12, 12, 12, 12, 12,
81 12, 12, 12, 12, 12, 12, 12, 12,
82 12, 10, 10, 10, 10, 10, 10, 10,
83 10, 10, 10, 10, 10, 10, 10, 10,
84 10, 10, 10, 10, 10, 10, 10, 10,
85 10, 10, 10, 10, 10, 10, 10, 10,
86 10, 8, 8, 8, 8, 8, 8, 8,
87 8, 8, 8, 8, 8, 8, 8, 8,
88 8, 7, 7, 7, 7, 7, 7, 7,
89 7, 6, 6, 6, 6, 5, 5, 4,
90 2, 4, 5, 5, 6, 6, 6, 6,
91 7, 7, 7, 7, 7, 7, 7, 7,
92 8, 8, 8, 8, 8, 8, 8, 8,
93 8, 8, 8, 8, 8, 8, 8, 8,
94 10, 10, 10, 10, 10, 10, 10, 10,
95 10, 10, 10, 10, 10, 10, 10, 10,
96 10, 10, 10, 10, 10, 10, 10, 10,
97 10, 10, 10, 10, 10, 10, 10, 10,
98 12, 12, 12, 12, 12, 12, 12, 12,
99 12, 12, 12, 12, 12, 12, 12, 12,
100 12, 12, 12, 12, 12, 12, 12, 12,
101 12, 12, 12, 12, 12, 12, 12, 12,
102 12, 12, 12, 12, 12, 12, 12, 12,
103 12, 12, 12, 12, 12, 12, 12, 12,
104 12, 12, 12, 12, 12, 12, 12, 12,
105 12, 12, 12, 12, 12, 12, 12, 12,
106 };
107
108 static const uint16_t rv_chrom_code[256] =
109 {
110 0xfe7f, 0x3f00, 0x3f01, 0x3f02, 0x3f03, 0x3f04, 0x3f05, 0x3f06,
111 0x3f07, 0x3f08, 0x3f09, 0x3f0a, 0x3f0b, 0x3f0c, 0x3f0d, 0x3f0e,
112 0x3f0f, 0x3f10, 0x3f11, 0x3f12, 0x3f13, 0x3f14, 0x3f15, 0x3f16,
113 0x3f17, 0x3f18, 0x3f19, 0x3f1a, 0x3f1b, 0x3f1c, 0x3f1d, 0x3f1e,
114 0x3f1f, 0x3f20, 0x3f21, 0x3f22, 0x3f23, 0x3f24, 0x3f25, 0x3f26,
115 0x3f27, 0x3f28, 0x3f29, 0x3f2a, 0x3f2b, 0x3f2c, 0x3f2d, 0x3f2e,
116 0x3f2f, 0x3f30, 0x3f31, 0x3f32, 0x3f33, 0x3f34, 0x3f35, 0x3f36,
117 0x3f37, 0x3f38, 0x3f39, 0x3f3a, 0x3f3b, 0x3f3c, 0x3f3d, 0x3f3e,
118 0x3f3f, 0x0f80, 0x0f81, 0x0f82, 0x0f83, 0x0f84, 0x0f85, 0x0f86,
119 0x0f87, 0x0f88, 0x0f89, 0x0f8a, 0x0f8b, 0x0f8c, 0x0f8d, 0x0f8e,
120 0x0f8f, 0x0f90, 0x0f91, 0x0f92, 0x0f93, 0x0f94, 0x0f95, 0x0f96,
121 0x0f97, 0x0f98, 0x0f99, 0x0f9a, 0x0f9b, 0x0f9c, 0x0f9d, 0x0f9e,
122 0x0f9f, 0x03c0, 0x03c1, 0x03c2, 0x03c3, 0x03c4, 0x03c5, 0x03c6,
123 0x03c7, 0x03c8, 0x03c9, 0x03ca, 0x03cb, 0x03cc, 0x03cd, 0x03ce,
124 0x03cf, 0x00e0, 0x00e1, 0x00e2, 0x00e3, 0x00e4, 0x00e5, 0x00e6,
125 0x00e7, 0x0030, 0x0031, 0x0032, 0x0033, 0x0008, 0x0009, 0x0002,
126 0x0000, 0x0003, 0x000a, 0x000b, 0x0034, 0x0035, 0x0036, 0x0037,
127 0x00e8, 0x00e9, 0x00ea, 0x00eb, 0x00ec, 0x00ed, 0x00ee, 0x00ef,
128 0x03d0, 0x03d1, 0x03d2, 0x03d3, 0x03d4, 0x03d5, 0x03d6, 0x03d7,
129 0x03d8, 0x03d9, 0x03da, 0x03db, 0x03dc, 0x03dd, 0x03de, 0x03df,
130 0x0fa0, 0x0fa1, 0x0fa2, 0x0fa3, 0x0fa4, 0x0fa5, 0x0fa6, 0x0fa7,
131 0x0fa8, 0x0fa9, 0x0faa, 0x0fab, 0x0fac, 0x0fad, 0x0fae, 0x0faf,
132 0x0fb0, 0x0fb1, 0x0fb2, 0x0fb3, 0x0fb4, 0x0fb5, 0x0fb6, 0x0fb7,
133 0x0fb8, 0x0fb9, 0x0fba, 0x0fbb, 0x0fbc, 0x0fbd, 0x0fbe, 0x0fbf,
134 0x3f40, 0x3f41, 0x3f42, 0x3f43, 0x3f44, 0x3f45, 0x3f46, 0x3f47,
135 0x3f48, 0x3f49, 0x3f4a, 0x3f4b, 0x3f4c, 0x3f4d, 0x3f4e, 0x3f4f,
136 0x3f50, 0x3f51, 0x3f52, 0x3f53, 0x3f54, 0x3f55, 0x3f56, 0x3f57,
137 0x3f58, 0x3f59, 0x3f5a, 0x3f5b, 0x3f5c, 0x3f5d, 0x3f5e, 0x3f5f,
138 0x3f60, 0x3f61, 0x3f62, 0x3f63, 0x3f64, 0x3f65, 0x3f66, 0x3f67,
139 0x3f68, 0x3f69, 0x3f6a, 0x3f6b, 0x3f6c, 0x3f6d, 0x3f6e, 0x3f6f,
140 0x3f70, 0x3f71, 0x3f72, 0x3f73, 0x3f74, 0x3f75, 0x3f76, 0x3f77,
141 0x3f78, 0x3f79, 0x3f7a, 0x3f7b, 0x3f7c, 0x3f7d, 0x3f7e, 0x3f7f,
142 };
143
144 static const uint8_t rv_chrom_bits[256] =
145 {
146 16, 14, 14, 14, 14, 14, 14, 14,
147 14, 14, 14, 14, 14, 14, 14, 14,
148 14, 14, 14, 14, 14, 14, 14, 14,
149 14, 14, 14, 14, 14, 14, 14, 14,
150 14, 14, 14, 14, 14, 14, 14, 14,
151 14, 14, 14, 14, 14, 14, 14, 14,
152 14, 14, 14, 14, 14, 14, 14, 14,
153 14, 14, 14, 14, 14, 14, 14, 14,
154 14, 12, 12, 12, 12, 12, 12, 12,
155 12, 12, 12, 12, 12, 12, 12, 12,
156 12, 12, 12, 12, 12, 12, 12, 12,
157 12, 12, 12, 12, 12, 12, 12, 12,
158 12, 10, 10, 10, 10, 10, 10, 10,
159 10, 10, 10, 10, 10, 10, 10, 10,
160 10, 8, 8, 8, 8, 8, 8, 8,
161 8, 6, 6, 6, 6, 4, 4, 3,
162 2, 3, 4, 4, 6, 6, 6, 6,
163 8, 8, 8, 8, 8, 8, 8, 8,
164 10, 10, 10, 10, 10, 10, 10, 10,
165 10, 10, 10, 10, 10, 10, 10, 10,
166 12, 12, 12, 12, 12, 12, 12, 12,
167 12, 12, 12, 12, 12, 12, 12, 12,
168 12, 12, 12, 12, 12, 12, 12, 12,
169 12, 12, 12, 12, 12, 12, 12, 12,
170 14, 14, 14, 14, 14, 14, 14, 14,
171 14, 14, 14, 14, 14, 14, 14, 14,
172 14, 14, 14, 14, 14, 14, 14, 14,
173 14, 14, 14, 14, 14, 14, 14, 14,
174 14, 14, 14, 14, 14, 14, 14, 14,
175 14, 14, 14, 14, 14, 14, 14, 14,
176 14, 14, 14, 14, 14, 14, 14, 14,
177 14, 14, 14, 14, 14, 14, 14, 14,
178 };
179
180 static VLC rv_dc_lum, rv_dc_chrom;
181
182 int rv_decode_dc(MpegEncContext *s, int n)
183 {
184 int code;
185
186 if (n < 4) {
187 code = get_vlc2(&s->gb, rv_dc_lum.table, DC_VLC_BITS, 2);
188 if (code < 0) {
189 /* XXX: I don't understand why they use LONGER codes than
190 necessary. The following code would be completely useless
191 if they had thought about it !!! */
192 code = get_bits(&s->gb, 7);
193 if (code == 0x7c) {
194 code = (int8_t)(get_bits(&s->gb, 7) + 1);
195 } else if (code == 0x7d) {
196 code = -128 + get_bits(&s->gb, 7);
197 } else if (code == 0x7e) {
198 if (get_bits1(&s->gb) == 0)
199 code = (int8_t)(get_bits(&s->gb, 8) + 1);
200 else
201 code = (int8_t)(get_bits(&s->gb, 8));
202 } else if (code == 0x7f) {
203 skip_bits(&s->gb, 11);
204 code = 1;
205 }
206 } else {
207 code -= 128;
208 }
209 } else {
210 code = get_vlc2(&s->gb, rv_dc_chrom.table, DC_VLC_BITS, 2);
211 /* same remark */
212 if (code < 0) {
213 code = get_bits(&s->gb, 9);
214 if (code == 0x1fc) {
215 code = (int8_t)(get_bits(&s->gb, 7) + 1);
216 } else if (code == 0x1fd) {
217 code = -128 + get_bits(&s->gb, 7);
218 } else if (code == 0x1fe) {
219 skip_bits(&s->gb, 9);
220 code = 1;
221 } else {
222 av_log(s->avctx, AV_LOG_ERROR, "chroma dc error\n");
223 return 0xffff;
224 }
225 } else {
226 code -= 128;
227 }
228 }
229 return -code;
230 }
231
232 /* read RV 1.0 compatible frame header */
233 static int rv10_decode_picture_header(MpegEncContext *s)
234 {
235 int mb_count, pb_frame, marker, unk, mb_xy;
236
237 marker = get_bits1(&s->gb);
238
239 if (get_bits1(&s->gb))
240 s->pict_type = FF_P_TYPE;
241 else
242 s->pict_type = FF_I_TYPE;
243 if(!marker) av_log(s->avctx, AV_LOG_ERROR, "marker missing\n");
244 pb_frame = get_bits1(&s->gb);
245
246 dprintf(s->avctx, "pict_type=%d pb_frame=%d\n", s->pict_type, pb_frame);
247
248 if (pb_frame){
249 av_log(s->avctx, AV_LOG_ERROR, "pb frame not supported\n");
250 return -1;
251 }
252
253 s->qscale = get_bits(&s->gb, 5);
254 if(s->qscale==0){
255 av_log(s->avctx, AV_LOG_ERROR, "error, qscale:0\n");
256 return -1;
257 }
258
259 if (s->pict_type == FF_I_TYPE) {
260 if (s->rv10_version == 3) {
261 /* specific MPEG like DC coding not used */
262 s->last_dc[0] = get_bits(&s->gb, 8);
263 s->last_dc[1] = get_bits(&s->gb, 8);
264 s->last_dc[2] = get_bits(&s->gb, 8);
265 dprintf(s->avctx, "DC:%d %d %d\n", s->last_dc[0],
266 s->last_dc[1], s->last_dc[2]);
267 }
268 }
269 /* if multiple packets per frame are sent, the position at which
270 to display the macroblocks is coded here */
271
272 mb_xy= s->mb_x + s->mb_y*s->mb_width;
273 if(show_bits(&s->gb, 12)==0 || (mb_xy && mb_xy < s->mb_num)){
274 s->mb_x = get_bits(&s->gb, 6); /* mb_x */
275 s->mb_y = get_bits(&s->gb, 6); /* mb_y */
276 mb_count = get_bits(&s->gb, 12);
277 } else {
278 s->mb_x = 0;
279 s->mb_y = 0;
280 mb_count = s->mb_width * s->mb_height;
281 }
282 unk= get_bits(&s->gb, 3); /* ignored */
283 s->f_code = 1;
284 s->unrestricted_mv = 1;
285
286 return mb_count;
287 }
288
289 static int rv20_decode_picture_header(MpegEncContext *s)
290 {
291 int seq, mb_pos, i;
292
293 #if 0
294 GetBitContext gb= s->gb;
295 for(i=0; i<64; i++){
296 av_log(s->avctx, AV_LOG_DEBUG, "%d", get_bits1(&gb));
297 if(i%4==3) av_log(s->avctx, AV_LOG_DEBUG, " ");
298 }
299 av_log(s->avctx, AV_LOG_DEBUG, "\n");
300 #endif
301 #if 0
302 av_log(s->avctx, AV_LOG_DEBUG, "%3dx%03d/%02Xx%02X ", s->width, s->height, s->width/4, s->height/4);
303 for(i=0; i<s->avctx->extradata_size; i++){
304 av_log(s->avctx, AV_LOG_DEBUG, "%02X ", ((uint8_t*)s->avctx->extradata)[i]);
305 if(i%4==3) av_log(s->avctx, AV_LOG_DEBUG, " ");
306 }
307 av_log(s->avctx, AV_LOG_DEBUG, "\n");
308 #endif
309
310 if(s->avctx->sub_id == 0x30202002 || s->avctx->sub_id == 0x30203002){
311 if (get_bits(&s->gb, 3)){
312 av_log(s->avctx, AV_LOG_ERROR, "unknown triplet set\n");
313 return -1;
314 }
315 }
316
317 i= get_bits(&s->gb, 2);
318 switch(i){
319 case 0: s->pict_type= FF_I_TYPE; break;
320 case 1: s->pict_type= FF_I_TYPE; break; //hmm ...
321 case 2: s->pict_type= FF_P_TYPE; break;
322 case 3: s->pict_type= FF_B_TYPE; break;
323 default:
324 av_log(s->avctx, AV_LOG_ERROR, "unknown frame type\n");
325 return -1;
326 }
327
328 if(s->last_picture_ptr==NULL && s->pict_type==FF_B_TYPE){
329 av_log(s->avctx, AV_LOG_ERROR, "early B pix\n");
330 return -1;
331 }
332
333 if (get_bits1(&s->gb)){
334 av_log(s->avctx, AV_LOG_ERROR, "unknown bit set\n");
335 return -1;
336 }
337
338 s->qscale = get_bits(&s->gb, 5);
339 if(s->qscale==0){
340 av_log(s->avctx, AV_LOG_ERROR, "error, qscale:0\n");
341 return -1;
342 }
343 if(s->avctx->sub_id == 0x30203002){
344 if (get_bits1(&s->gb)){
345 av_log(s->avctx, AV_LOG_ERROR, "unknown bit2 set\n");
346 return -1;
347 }
348 }
349
350 if(s->avctx->has_b_frames){
351 int f, new_w, new_h;
352 int v= s->avctx->extradata_size >= 4 ? 7&((uint8_t*)s->avctx->extradata)[1] : 0;
353
354 if (get_bits1(&s->gb)){
355 av_log(s->avctx, AV_LOG_ERROR, "unknown bit3 set\n");
356 }
357 seq= get_bits(&s->gb, 13)<<2;
358
359 f= get_bits(&s->gb, av_log2(v)+1);
360
361 if(f){
362 new_w= 4*((uint8_t*)s->avctx->extradata)[6+2*f];
363 new_h= 4*((uint8_t*)s->avctx->extradata)[7+2*f];
364 }else{
365 new_w= s->width; //FIXME wrong we of course must save the original in the context
366 new_h= s->height;
367 }
368 if(new_w != s->width || new_h != s->height){
369 av_log(s->avctx, AV_LOG_DEBUG, "attempting to change resolution to %dx%d\n", new_w, new_h);
370 if (avcodec_check_dimensions(s->avctx, new_h, new_w) < 0)
371 return -1;
372 MPV_common_end(s);
373 s->width = s->avctx->width = new_w;
374 s->height = s->avctx->height= new_h;
375 if (MPV_common_init(s) < 0)
376 return -1;
377 }
378
379 if(s->avctx->debug & FF_DEBUG_PICT_INFO){
380 av_log(s->avctx, AV_LOG_DEBUG, "F %d/%d\n", f, v);
381 }
382 }else{
383 seq= get_bits(&s->gb, 8)*128;
384 }
385
386 // if(s->avctx->sub_id <= 0x20201002){ //0x20201002 definitely needs this
387 mb_pos= ff_h263_decode_mba(s);
388 /* }else{
389 mb_pos= get_bits(&s->gb, av_log2(s->mb_num-1)+1);
390 s->mb_x= mb_pos % s->mb_width;
391 s->mb_y= mb_pos / s->mb_width;
392 }*/
393 //av_log(s->avctx, AV_LOG_DEBUG, "%d\n", seq);
394 seq |= s->time &~0x7FFF;
395 if(seq - s->time > 0x4000) seq -= 0x8000;
396 if(seq - s->time < -0x4000) seq += 0x8000;
397 if(seq != s->time){
398 if(s->pict_type!=FF_B_TYPE){
399 s->time= seq;
400 s->pp_time= s->time - s->last_non_b_time;
401 s->last_non_b_time= s->time;
402 }else{
403 s->time= seq;
404 s->pb_time= s->pp_time - (s->last_non_b_time - s->time);
405 if(s->pp_time <=s->pb_time || s->pp_time <= s->pp_time - s->pb_time || s->pp_time<=0){
406 av_log(s->avctx, AV_LOG_DEBUG, "messed up order, possible from seeking? skipping current b frame\n");
407 return FRAME_SKIPPED;
408 }
409 ff_mpeg4_init_direct_mv(s);
410 }
411 }
412 // printf("%d %d %d %d %d\n", seq, (int)s->time, (int)s->last_non_b_time, s->pp_time, s->pb_time);
413 /*for(i=0; i<32; i++){
414 av_log(s->avctx, AV_LOG_DEBUG, "%d", get_bits1(&s->gb));
415 }
416 av_log(s->avctx, AV_LOG_DEBUG, "\n");*/
417 s->no_rounding= get_bits1(&s->gb);
418
419 s->f_code = 1;
420 s->unrestricted_mv = 1;
421 s->h263_aic= s->pict_type == FF_I_TYPE;
422 // s->alt_inter_vlc=1;
423 // s->obmc=1;
424 // s->umvplus=1;
425 s->modified_quant=1;
426 if(!s->avctx->lowres)
427 s->loop_filter=1;
428
429 if(s->avctx->debug & FF_DEBUG_PICT_INFO){
430 av_log(s->avctx, AV_LOG_INFO, "num:%5d x:%2d y:%2d type:%d qscale:%2d rnd:%d\n",
431 seq, s->mb_x, s->mb_y, s->pict_type, s->qscale, s->no_rounding);
432 }
433
434 assert(s->pict_type != FF_B_TYPE || !s->low_delay);
435
436 return s->mb_width*s->mb_height - mb_pos;
437 }
438
439 static av_cold int rv10_decode_init(AVCodecContext *avctx)
440 {
441 MpegEncContext *s = avctx->priv_data;
442 static int done=0;
443
444 if (avctx->extradata_size < 8) {
445 av_log(avctx, AV_LOG_ERROR, "Extradata is too small.\n");
446 return -1;
447 }
448
449 MPV_decode_defaults(s);
450
451 s->avctx= avctx;
452 s->out_format = FMT_H263;
453 s->codec_id= avctx->codec_id;
454
455 s->width = avctx->coded_width;
456 s->height = avctx->coded_height;
457
458 s->h263_long_vectors= ((uint8_t*)avctx->extradata)[3] & 1;
459 avctx->sub_id= AV_RB32((uint8_t*)avctx->extradata + 4);
460
461 if (avctx->sub_id == 0x10000000) {
462 s->rv10_version= 0;
463 s->low_delay=1;
464 } else if (avctx->sub_id == 0x10001000) {
465 s->rv10_version= 3;
466 s->low_delay=1;
467 } else if (avctx->sub_id == 0x10002000) {
468 s->rv10_version= 3;
469 s->low_delay=1;
470 s->obmc=1;
471 } else if (avctx->sub_id == 0x10003000) {
472 s->rv10_version= 3;
473 s->low_delay=1;
474 } else if (avctx->sub_id == 0x10003001) {
475 s->rv10_version= 3;
476 s->low_delay=1;
477 } else if ( avctx->sub_id == 0x20001000
478 || (avctx->sub_id >= 0x20100000 && avctx->sub_id < 0x201a0000)) {
479 s->low_delay=1;
480 } else if ( avctx->sub_id == 0x30202002
481 || avctx->sub_id == 0x30203002
482 || (avctx->sub_id >= 0x20200002 && avctx->sub_id < 0x20300000)) {
483 s->low_delay=0;
484 s->avctx->has_b_frames=1;
485 } else
486 av_log(s->avctx, AV_LOG_ERROR, "unknown header %X\n", avctx->sub_id);
487
488 if(avctx->debug & FF_DEBUG_PICT_INFO){
489 av_log(avctx, AV_LOG_DEBUG, "ver:%X ver0:%X\n", avctx->sub_id, avctx->extradata_size >= 4 ? ((uint32_t*)avctx->extradata)[0] : -1);
490 }
491
492 avctx->pix_fmt = PIX_FMT_YUV420P;
493
494 if (MPV_common_init(s) < 0)
495 return -1;
496
497 h263_decode_init_vlc(s);
498
499 /* init rv vlc */
500 if (!done) {
501 INIT_VLC_STATIC(&rv_dc_lum, DC_VLC_BITS, 256,
502 rv_lum_bits, 1, 1,
503 rv_lum_code, 2, 2, 16384);
504 INIT_VLC_STATIC(&rv_dc_chrom, DC_VLC_BITS, 256,
505 rv_chrom_bits, 1, 1,
506 rv_chrom_code, 2, 2, 16388);
507 done = 1;
508 }
509
510 return 0;
511 }
512
513 static av_cold int rv10_decode_end(AVCodecContext *avctx)
514 {
515 MpegEncContext *s = avctx->priv_data;
516
517 MPV_common_end(s);
518 return 0;
519 }
520
521 static int rv10_decode_packet(AVCodecContext *avctx,
522 const uint8_t *buf, int buf_size)
523 {
524 MpegEncContext *s = avctx->priv_data;
525 int mb_count, mb_pos, left, start_mb_x;
526
527 init_get_bits(&s->gb, buf, buf_size*8);
528 if(s->codec_id ==CODEC_ID_RV10)
529 mb_count = rv10_decode_picture_header(s);
530 else
531 mb_count = rv20_decode_picture_header(s);
532 if (mb_count < 0) {
533 av_log(s->avctx, AV_LOG_ERROR, "HEADER ERROR\n");
534 return -1;
535 }
536
537 if (s->mb_x >= s->mb_width ||
538 s->mb_y >= s->mb_height) {
539 av_log(s->avctx, AV_LOG_ERROR, "POS ERROR %d %d\n", s->mb_x, s->mb_y);
540 return -1;
541 }
542 mb_pos = s->mb_y * s->mb_width + s->mb_x;
543 left = s->mb_width * s->mb_height - mb_pos;
544 if (mb_count > left) {
545 av_log(s->avctx, AV_LOG_ERROR, "COUNT ERROR\n");
546 return -1;
547 }
548
549 if ((s->mb_x == 0 && s->mb_y == 0) || s->current_picture_ptr==NULL) {
550 if(s->current_picture_ptr){ //FIXME write parser so we always have complete frames?
551 ff_er_frame_end(s);
552 MPV_frame_end(s);
553 s->mb_x= s->mb_y = s->resync_mb_x = s->resync_mb_y= 0;
554 }
555 if(MPV_frame_start(s, avctx) < 0)
556 return -1;
557 ff_er_frame_start(s);
558 }
559
560 dprintf(avctx, "qscale=%d\n", s->qscale);
561
562 /* default quantization values */
563 if(s->codec_id== CODEC_ID_RV10){
564 if(s->mb_y==0) s->first_slice_line=1;
565 }else{
566 s->first_slice_line=1;
567 s->resync_mb_x= s->mb_x;
568 }
569 start_mb_x= s->mb_x;
570 s->resync_mb_y= s->mb_y;
571 if(s->h263_aic){
572 s->y_dc_scale_table=
573 s->c_dc_scale_table= ff_aic_dc_scale_table;
574 }else{
575 s->y_dc_scale_table=
576 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
577 }
578
579 if(s->modified_quant)
580 s->chroma_qscale_table= ff_h263_chroma_qscale_table;
581
582 ff_set_qscale(s, s->qscale);
583
584 s->rv10_first_dc_coded[0] = 0;
585 s->rv10_first_dc_coded[1] = 0;
586 s->rv10_first_dc_coded[2] = 0;
587 s->block_wrap[0]=
588 s->block_wrap[1]=
589 s->block_wrap[2]=
590 s->block_wrap[3]= s->b8_stride;
591 s->block_wrap[4]=
592 s->block_wrap[5]= s->mb_stride;
593 ff_init_block_index(s);
594 /* decode each macroblock */
595
596 for(s->mb_num_left= mb_count; s->mb_num_left>0; s->mb_num_left--) {
597 int ret;
598 ff_update_block_index(s);
599 dprintf(avctx, "**mb x=%d y=%d\n", s->mb_x, s->mb_y);
600
601 s->mv_dir = MV_DIR_FORWARD;
602 s->mv_type = MV_TYPE_16X16;
603 ret=ff_h263_decode_mb(s, s->block);
604
605 if (ret == SLICE_ERROR || s->gb.size_in_bits < get_bits_count(&s->gb)) {
606 av_log(s->avctx, AV_LOG_ERROR, "ERROR at MB %d %d\n", s->mb_x, s->mb_y);
607 return -1;
608 }
609 if(s->pict_type != FF_B_TYPE)
610 ff_h263_update_motion_val(s);
611 MPV_decode_mb(s, s->block);
612 if(s->loop_filter)
613 ff_h263_loop_filter(s);
614
615 if (++s->mb_x == s->mb_width) {
616 s->mb_x = 0;
617 s->mb_y++;
618 ff_init_block_index(s);
619 }
620 if(s->mb_x == s->resync_mb_x)
621 s->first_slice_line=0;
622 if(ret == SLICE_END) break;
623 }
624
625 ff_er_add_slice(s, start_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, AC_END|DC_END|MV_END);
626
627 return buf_size;
628 }
629
630 static int get_slice_offset(AVCodecContext *avctx, const uint8_t *buf, int n)
631 {
632 if(avctx->slice_count) return avctx->slice_offset[n];
633 else return AV_RL32(buf + n*8);
634 }
635
636 static int rv10_decode_frame(AVCodecContext *avctx,
637 void *data, int *data_size,
638 AVPacket *avpkt)
639 {
640 const uint8_t *buf = avpkt->data;
641 int buf_size = avpkt->size;
642 MpegEncContext *s = avctx->priv_data;
643 int i;
644 AVFrame *pict = data;
645 int slice_count;
646 const uint8_t *slices_hdr = NULL;
647
648 dprintf(avctx, "*****frame %d size=%d\n", avctx->frame_number, buf_size);
649
650 /* no supplementary picture */
651 if (buf_size == 0) {
652 return 0;
653 }
654
655 if(!avctx->slice_count){
656 slice_count = (*buf++) + 1;
657 slices_hdr = buf + 4;
658 buf += 8 * slice_count;
659 }else
660 slice_count = avctx->slice_count;
661
662 for(i=0; i<slice_count; i++){
663 int offset= get_slice_offset(avctx, slices_hdr, i);
664 int size;
665
666 if(i+1 == slice_count)
667 size= buf_size - offset;
668 else
669 size= get_slice_offset(avctx, slices_hdr, i+1) - offset;
670
671 rv10_decode_packet(avctx, buf+offset, size);
672 }
673
674 if(s->current_picture_ptr != NULL && s->mb_y>=s->mb_height){
675 ff_er_frame_end(s);
676 MPV_frame_end(s);
677
678 if (s->pict_type == FF_B_TYPE || s->low_delay) {
679 *pict= *(AVFrame*)s->current_picture_ptr;
680 } else if (s->last_picture_ptr != NULL) {
681 *pict= *(AVFrame*)s->last_picture_ptr;
682 }
683
684 if(s->last_picture_ptr || s->low_delay){
685 *data_size = sizeof(AVFrame);
686 ff_print_debug_info(s, pict);
687 }
688 s->current_picture_ptr= NULL; //so we can detect if frame_end wasnt called (find some nicer solution...)
689 }
690
691 return buf_size;
692 }
693
694 AVCodec rv10_decoder = {
695 "rv10",
696 CODEC_TYPE_VIDEO,
697 CODEC_ID_RV10,
698 sizeof(MpegEncContext),
699 rv10_decode_init,
700 NULL,
701 rv10_decode_end,
702 rv10_decode_frame,
703 CODEC_CAP_DR1,
704 .long_name = NULL_IF_CONFIG_SMALL("RealVideo 1.0"),
705 .pix_fmts= ff_pixfmt_list_420,
706 };
707
708 AVCodec rv20_decoder = {
709 "rv20",
710 CODEC_TYPE_VIDEO,
711 CODEC_ID_RV20,
712 sizeof(MpegEncContext),
713 rv10_decode_init,
714 NULL,
715 rv10_decode_end,
716 rv10_decode_frame,
717 CODEC_CAP_DR1 | CODEC_CAP_DELAY,
718 .flush= ff_mpeg_flush,
719 .long_name = NULL_IF_CONFIG_SMALL("RealVideo 2.0"),
720 .pix_fmts= ff_pixfmt_list_420,
721 };
lucabe's unfinished and experimental patches for ffmpeg