summaryrefslogtreecommitdiff
path: root/apps/plugins/mpegplayer/decode.c
diff options
context:
space:
mode:
authorDave Chapman <dave@dchapman.com>2006-08-07 22:11:07 +0000
committerDave Chapman <dave@dchapman.com>2006-08-07 22:11:07 +0000
commitc9d66562afc15de210854b32f30976859bce2023 (patch)
tree4fc7c506797bf96e389bd2fb68dd0e36d2735750 /apps/plugins/mpegplayer/decode.c
parent754e173c252ab80a9e1e290bddfa126e6227ac1c (diff)
downloadrockbox-c9d66562afc15de210854b32f30976859bce2023.tar.gz
rockbox-c9d66562afc15de210854b32f30976859bce2023.zip
Initial commit of work-in-progress MPEG video player plugin based on libmpeg2. Works on all targets with colour LCDs, but most optimised for the ipod Color/Photo and Nano. It currently only plays raw MPEG-1 or MPEG-2 video streams (no audio). Also adds a new lcd_yuv_blit() function to the plugin API - currently only implemented for the ipod Color/Photo and Nano.
git-svn-id: svn://svn.rockbox.org/rockbox/trunk@10479 a1c6a512-1295-4272-9138-f99709370657
Diffstat (limited to 'apps/plugins/mpegplayer/decode.c')
-rw-r--r--apps/plugins/mpegplayer/decode.c447
1 files changed, 447 insertions, 0 deletions
diff --git a/apps/plugins/mpegplayer/decode.c b/apps/plugins/mpegplayer/decode.c
new file mode 100644
index 0000000000..613292699d
--- /dev/null
+++ b/apps/plugins/mpegplayer/decode.c
@@ -0,0 +1,447 @@
1/*
2 * decode.c
3 * Copyright (C) 2000-2003 Michel Lespinasse <walken@zoy.org>
4 * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
5 *
6 * This file is part of mpeg2dec, a free MPEG-2 video stream decoder.
7 * See http://libmpeg2.sourceforge.net/ for updates.
8 *
9 * mpeg2dec is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * mpeg2dec is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
18 *
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 */
23
24#include "plugin.h"
25
26#include "mpeg2dec_config.h"
27
28extern struct plugin_api* rb;
29
30#include "mpeg2.h"
31#include "attributes.h"
32#include "mpeg2_internal.h"
33
34static int mpeg2_accels = 0;
35
36#define BUFFER_SIZE (1194 * 1024)
37
38const mpeg2_info_t * mpeg2_info (mpeg2dec_t * mpeg2dec)
39{
40 return &(mpeg2dec->info);
41}
42
43static inline int skip_chunk (mpeg2dec_t * mpeg2dec, int bytes)
44{
45 uint8_t * current;
46 uint32_t shift;
47 uint8_t * limit;
48 uint8_t byte;
49
50 if (!bytes)
51 return 0;
52
53 current = mpeg2dec->buf_start;
54 shift = mpeg2dec->shift;
55 limit = current + bytes;
56
57 do {
58 byte = *current++;
59 if (shift == 0x00000100) {
60 int skipped;
61
62 mpeg2dec->shift = 0xffffff00;
63 skipped = current - mpeg2dec->buf_start;
64 mpeg2dec->buf_start = current;
65 return skipped;
66 }
67 shift = (shift | byte) << 8;
68 } while (current < limit);
69
70 mpeg2dec->shift = shift;
71 mpeg2dec->buf_start = current;
72 return 0;
73}
74
75static inline int copy_chunk (mpeg2dec_t * mpeg2dec, int bytes)
76{
77 uint8_t * current;
78 uint32_t shift;
79 uint8_t * chunk_ptr;
80 uint8_t * limit;
81 uint8_t byte;
82
83 if (!bytes)
84 return 0;
85
86 current = mpeg2dec->buf_start;
87 shift = mpeg2dec->shift;
88 chunk_ptr = mpeg2dec->chunk_ptr;
89 limit = current + bytes;
90
91 do {
92 byte = *current++;
93 if (shift == 0x00000100) {
94 int copied;
95
96 mpeg2dec->shift = 0xffffff00;
97 mpeg2dec->chunk_ptr = chunk_ptr + 1;
98 copied = current - mpeg2dec->buf_start;
99 mpeg2dec->buf_start = current;
100 return copied;
101 }
102 shift = (shift | byte) << 8;
103 *chunk_ptr++ = byte;
104 } while (current < limit);
105
106 mpeg2dec->shift = shift;
107 mpeg2dec->buf_start = current;
108 return 0;
109}
110
111void mpeg2_buffer (mpeg2dec_t * mpeg2dec, uint8_t * start, uint8_t * end)
112{
113 mpeg2dec->buf_start = start;
114 mpeg2dec->buf_end = end;
115}
116
117int mpeg2_getpos (mpeg2dec_t * mpeg2dec)
118{
119 return mpeg2dec->buf_end - mpeg2dec->buf_start;
120}
121
122static inline mpeg2_state_t seek_chunk (mpeg2dec_t * mpeg2dec)
123{
124 int size, skipped;
125
126 size = mpeg2dec->buf_end - mpeg2dec->buf_start;
127 skipped = skip_chunk (mpeg2dec, size);
128 if (!skipped) {
129 mpeg2dec->bytes_since_tag += size;
130 return STATE_BUFFER;
131 }
132 mpeg2dec->bytes_since_tag += skipped;
133 mpeg2dec->code = mpeg2dec->buf_start[-1];
134 return (mpeg2_state_t)-1;
135}
136
137mpeg2_state_t mpeg2_seek_header (mpeg2dec_t * mpeg2dec)
138{
139 while (mpeg2dec->code != 0xb3 &&
140 ((mpeg2dec->code != 0xb7 && mpeg2dec->code != 0xb8 &&
141 mpeg2dec->code) || mpeg2dec->sequence.width == (unsigned)-1))
142 if (seek_chunk (mpeg2dec) == STATE_BUFFER)
143 return STATE_BUFFER;
144 mpeg2dec->chunk_start = mpeg2dec->chunk_ptr = mpeg2dec->chunk_buffer;
145 mpeg2dec->user_data_len = 0;
146 return (mpeg2dec->code ? mpeg2_parse_header (mpeg2dec) :
147 mpeg2_header_picture_start (mpeg2dec));
148}
149
150#define RECEIVED(code,state) (((state) << 8) + (code))
151
152mpeg2_state_t mpeg2_parse (mpeg2dec_t * mpeg2dec)
153{
154 int size_buffer, size_chunk, copied;
155
156 if (mpeg2dec->action) {
157 mpeg2_state_t state;
158
159 state = mpeg2dec->action (mpeg2dec);
160 if ((int)state >= 0)
161 return state;
162 }
163
164 while (1) {
165 while ((unsigned) (mpeg2dec->code - mpeg2dec->first_decode_slice) <
166 mpeg2dec->nb_decode_slices) {
167 size_buffer = mpeg2dec->buf_end - mpeg2dec->buf_start;
168 size_chunk = (mpeg2dec->chunk_buffer + BUFFER_SIZE -
169 mpeg2dec->chunk_ptr);
170 if (size_buffer <= size_chunk) {
171 copied = copy_chunk (mpeg2dec, size_buffer);
172 if (!copied) {
173 mpeg2dec->bytes_since_tag += size_buffer;
174 mpeg2dec->chunk_ptr += size_buffer;
175 return STATE_BUFFER;
176 }
177 } else {
178 copied = copy_chunk (mpeg2dec, size_chunk);
179 if (!copied) {
180 /* filled the chunk buffer without finding a start code */
181 mpeg2dec->bytes_since_tag += size_chunk;
182 mpeg2dec->action = seek_chunk;
183 return STATE_INVALID;
184 }
185 }
186 mpeg2dec->bytes_since_tag += copied;
187
188 mpeg2_slice (&(mpeg2dec->decoder), mpeg2dec->code,
189 mpeg2dec->chunk_start);
190 mpeg2dec->code = mpeg2dec->buf_start[-1];
191 mpeg2dec->chunk_ptr = mpeg2dec->chunk_start;
192 }
193 if ((unsigned) (mpeg2dec->code - 1) >= 0xb0 - 1)
194 break;
195 if (seek_chunk (mpeg2dec) == STATE_BUFFER)
196 return STATE_BUFFER;
197 }
198
199 switch (mpeg2dec->code) {
200 case 0x00:
201 mpeg2dec->action = mpeg2_header_picture_start;
202 return mpeg2dec->state;
203 case 0xb7:
204 mpeg2dec->action = mpeg2_header_end;
205 break;
206 case 0xb3:
207 case 0xb8:
208 mpeg2dec->action = mpeg2_parse_header;
209 break;
210 default:
211 mpeg2dec->action = seek_chunk;
212 return STATE_INVALID;
213 }
214 return (mpeg2dec->state == STATE_SLICE) ? STATE_SLICE : STATE_INVALID;
215}
216
217mpeg2_state_t mpeg2_parse_header (mpeg2dec_t * mpeg2dec)
218{
219 static int (* process_header[]) (mpeg2dec_t * mpeg2dec) = {
220 mpeg2_header_picture, mpeg2_header_extension, mpeg2_header_user_data,
221 mpeg2_header_sequence, NULL, NULL, NULL, NULL, mpeg2_header_gop
222 };
223 int size_buffer, size_chunk, copied;
224
225 mpeg2dec->action = mpeg2_parse_header;
226 mpeg2dec->info.user_data = NULL; mpeg2dec->info.user_data_len = 0;
227 while (1) {
228 size_buffer = mpeg2dec->buf_end - mpeg2dec->buf_start;
229 size_chunk = (mpeg2dec->chunk_buffer + BUFFER_SIZE -
230 mpeg2dec->chunk_ptr);
231 if (size_buffer <= size_chunk) {
232 copied = copy_chunk (mpeg2dec, size_buffer);
233 if (!copied) {
234 mpeg2dec->bytes_since_tag += size_buffer;
235 mpeg2dec->chunk_ptr += size_buffer;
236 return STATE_BUFFER;
237 }
238 } else {
239 copied = copy_chunk (mpeg2dec, size_chunk);
240 if (!copied) {
241 /* filled the chunk buffer without finding a start code */
242 mpeg2dec->bytes_since_tag += size_chunk;
243 mpeg2dec->code = 0xb4;
244 mpeg2dec->action = mpeg2_seek_header;
245 return STATE_INVALID;
246 }
247 }
248 mpeg2dec->bytes_since_tag += copied;
249
250 if (process_header[mpeg2dec->code & 0x0b] (mpeg2dec)) {
251 mpeg2dec->code = mpeg2dec->buf_start[-1];
252 mpeg2dec->action = mpeg2_seek_header;
253 return STATE_INVALID;
254 }
255
256 mpeg2dec->code = mpeg2dec->buf_start[-1];
257 switch (RECEIVED (mpeg2dec->code, mpeg2dec->state)) {
258
259 /* state transition after a sequence header */
260 case RECEIVED (0x00, STATE_SEQUENCE):
261 mpeg2dec->action = mpeg2_header_picture_start;
262 case RECEIVED (0xb8, STATE_SEQUENCE):
263 mpeg2_header_sequence_finalize (mpeg2dec);
264 break;
265
266 /* other legal state transitions */
267 case RECEIVED (0x00, STATE_GOP):
268 mpeg2_header_gop_finalize (mpeg2dec);
269 mpeg2dec->action = mpeg2_header_picture_start;
270 break;
271 case RECEIVED (0x01, STATE_PICTURE):
272 case RECEIVED (0x01, STATE_PICTURE_2ND):
273 mpeg2_header_picture_finalize (mpeg2dec, mpeg2_accels);
274 mpeg2dec->action = mpeg2_header_slice_start;
275 break;
276
277 /* legal headers within a given state */
278 case RECEIVED (0xb2, STATE_SEQUENCE):
279 case RECEIVED (0xb2, STATE_GOP):
280 case RECEIVED (0xb2, STATE_PICTURE):
281 case RECEIVED (0xb2, STATE_PICTURE_2ND):
282 case RECEIVED (0xb5, STATE_SEQUENCE):
283 case RECEIVED (0xb5, STATE_PICTURE):
284 case RECEIVED (0xb5, STATE_PICTURE_2ND):
285 mpeg2dec->chunk_ptr = mpeg2dec->chunk_start;
286 continue;
287
288 default:
289 mpeg2dec->action = mpeg2_seek_header;
290 return STATE_INVALID;
291 }
292
293 mpeg2dec->chunk_start = mpeg2dec->chunk_ptr = mpeg2dec->chunk_buffer;
294 mpeg2dec->user_data_len = 0;
295 return mpeg2dec->state;
296 }
297}
298
299int mpeg2_convert (mpeg2dec_t * mpeg2dec, mpeg2_convert_t convert, void * arg)
300{
301 mpeg2_convert_init_t convert_init;
302 int error;
303
304 error = convert (MPEG2_CONVERT_SET, NULL, &(mpeg2dec->sequence), 0,
305 mpeg2_accels, arg, &convert_init);
306 if (!error) {
307 mpeg2dec->convert = convert;
308 mpeg2dec->convert_arg = arg;
309 mpeg2dec->convert_id_size = convert_init.id_size;
310 mpeg2dec->convert_stride = 0;
311 }
312 return error;
313}
314
315int mpeg2_stride (mpeg2dec_t * mpeg2dec, int stride)
316{
317 if (!mpeg2dec->convert) {
318 if (stride < (int) mpeg2dec->sequence.width)
319 stride = mpeg2dec->sequence.width;
320 mpeg2dec->decoder.stride_frame = stride;
321 } else {
322 mpeg2_convert_init_t convert_init;
323
324 stride = mpeg2dec->convert (MPEG2_CONVERT_STRIDE, NULL,
325 &(mpeg2dec->sequence), stride,
326 mpeg2_accels, mpeg2dec->convert_arg,
327 &convert_init);
328 mpeg2dec->convert_id_size = convert_init.id_size;
329 mpeg2dec->convert_stride = stride;
330 }
331 return stride;
332}
333
334void mpeg2_set_buf (mpeg2dec_t * mpeg2dec, uint8_t * buf[3], void * id)
335{
336 mpeg2_fbuf_t * fbuf;
337
338 if (mpeg2dec->custom_fbuf) {
339 if (mpeg2dec->state == STATE_SEQUENCE) {
340 mpeg2dec->fbuf[2] = mpeg2dec->fbuf[1];
341 mpeg2dec->fbuf[1] = mpeg2dec->fbuf[0];
342 }
343 mpeg2_set_fbuf (mpeg2dec, (mpeg2dec->decoder.coding_type ==
344 PIC_FLAG_CODING_TYPE_B));
345 fbuf = mpeg2dec->fbuf[0];
346 } else {
347 fbuf = &(mpeg2dec->fbuf_alloc[mpeg2dec->alloc_index].fbuf);
348 mpeg2dec->alloc_index_user = ++mpeg2dec->alloc_index;
349 }
350 fbuf->buf[0] = buf[0];
351 fbuf->buf[1] = buf[1];
352 fbuf->buf[2] = buf[2];
353 fbuf->id = id;
354}
355
356void mpeg2_custom_fbuf (mpeg2dec_t * mpeg2dec, int custom_fbuf)
357{
358 mpeg2dec->custom_fbuf = custom_fbuf;
359}
360
361void mpeg2_skip (mpeg2dec_t * mpeg2dec, int skip)
362{
363 mpeg2dec->first_decode_slice = 1;
364 mpeg2dec->nb_decode_slices = skip ? 0 : (0xb0 - 1);
365}
366
367void mpeg2_slice_region (mpeg2dec_t * mpeg2dec, int start, int end)
368{
369 start = (start < 1) ? 1 : (start > 0xb0) ? 0xb0 : start;
370 end = (end < start) ? start : (end > 0xb0) ? 0xb0 : end;
371 mpeg2dec->first_decode_slice = start;
372 mpeg2dec->nb_decode_slices = end - start;
373}
374
375void mpeg2_tag_picture (mpeg2dec_t * mpeg2dec, uint32_t tag, uint32_t tag2)
376{
377 mpeg2dec->tag_previous = mpeg2dec->tag_current;
378 mpeg2dec->tag2_previous = mpeg2dec->tag2_current;
379 mpeg2dec->tag_current = tag;
380 mpeg2dec->tag2_current = tag2;
381 mpeg2dec->num_tags++;
382 mpeg2dec->bytes_since_tag = 0;
383}
384
385uint32_t mpeg2_accel (uint32_t accel)
386{
387 if (!mpeg2_accels) {
388 if (accel & MPEG2_ACCEL_DETECT)
389 accel |= mpeg2_detect_accel ();
390 mpeg2_accels = accel |= MPEG2_ACCEL_DETECT;
391 mpeg2_cpu_state_init (accel);
392 mpeg2_idct_init (accel);
393 mpeg2_mc_init (accel);
394 }
395 return mpeg2_accels & ~MPEG2_ACCEL_DETECT;
396}
397
398void mpeg2_reset (mpeg2dec_t * mpeg2dec, int full_reset)
399{
400 mpeg2dec->buf_start = mpeg2dec->buf_end = NULL;
401 mpeg2dec->num_tags = 0;
402 mpeg2dec->shift = 0xffffff00;
403 mpeg2dec->code = 0xb4;
404 mpeg2dec->action = mpeg2_seek_header;
405 mpeg2dec->state = STATE_INVALID;
406 mpeg2dec->first = 1;
407
408 mpeg2_reset_info(&(mpeg2dec->info));
409 mpeg2dec->info.gop = NULL;
410 mpeg2dec->info.user_data = NULL;
411 mpeg2dec->info.user_data_len = 0;
412 if (full_reset) {
413 mpeg2dec->info.sequence = NULL;
414 mpeg2_header_state_init (mpeg2dec);
415 }
416
417}
418
419mpeg2dec_t * mpeg2_init (void)
420{
421 mpeg2dec_t * mpeg2dec;
422
423 mpeg2_accel (MPEG2_ACCEL_DETECT);
424
425 mpeg2dec = (mpeg2dec_t *) mpeg2_malloc (sizeof (mpeg2dec_t),
426 MPEG2_ALLOC_MPEG2DEC);
427 if (mpeg2dec == NULL)
428 return NULL;
429
430 rb->memset (mpeg2dec->decoder.DCTblock, 0, 64 * sizeof (int16_t));
431 rb->memset (mpeg2dec->quantizer_matrix, 0, 4 * 64 * sizeof (uint8_t));
432
433 mpeg2dec->chunk_buffer = (uint8_t *) mpeg2_malloc (BUFFER_SIZE + 4,
434 MPEG2_ALLOC_CHUNK);
435
436 mpeg2dec->sequence.width = (unsigned)-1;
437 mpeg2_reset (mpeg2dec, 1);
438
439 return mpeg2dec;
440}
441
442void mpeg2_close (mpeg2dec_t * mpeg2dec)
443{
444 mpeg2_header_state_init (mpeg2dec);
445 mpeg2_free (mpeg2dec->chunk_buffer);
446 mpeg2_free (mpeg2dec);
447}