diff options
author | Dave Chapman <dave@dchapman.com> | 2006-08-07 22:11:07 +0000 |
---|---|---|
committer | Dave Chapman <dave@dchapman.com> | 2006-08-07 22:11:07 +0000 |
commit | c9d66562afc15de210854b32f30976859bce2023 (patch) | |
tree | 4fc7c506797bf96e389bd2fb68dd0e36d2735750 /apps/plugins/mpegplayer/decode.c | |
parent | 754e173c252ab80a9e1e290bddfa126e6227ac1c (diff) | |
download | rockbox-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.c | 447 |
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 | |||
28 | extern struct plugin_api* rb; | ||
29 | |||
30 | #include "mpeg2.h" | ||
31 | #include "attributes.h" | ||
32 | #include "mpeg2_internal.h" | ||
33 | |||
34 | static int mpeg2_accels = 0; | ||
35 | |||
36 | #define BUFFER_SIZE (1194 * 1024) | ||
37 | |||
38 | const mpeg2_info_t * mpeg2_info (mpeg2dec_t * mpeg2dec) | ||
39 | { | ||
40 | return &(mpeg2dec->info); | ||
41 | } | ||
42 | |||
43 | static 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 | |||
75 | static 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 | |||
111 | void mpeg2_buffer (mpeg2dec_t * mpeg2dec, uint8_t * start, uint8_t * end) | ||
112 | { | ||
113 | mpeg2dec->buf_start = start; | ||
114 | mpeg2dec->buf_end = end; | ||
115 | } | ||
116 | |||
117 | int mpeg2_getpos (mpeg2dec_t * mpeg2dec) | ||
118 | { | ||
119 | return mpeg2dec->buf_end - mpeg2dec->buf_start; | ||
120 | } | ||
121 | |||
122 | static 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 | |||
137 | mpeg2_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 | |||
152 | mpeg2_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 | |||
217 | mpeg2_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 | |||
299 | int 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 | |||
315 | int 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 | |||
334 | void 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 | |||
356 | void mpeg2_custom_fbuf (mpeg2dec_t * mpeg2dec, int custom_fbuf) | ||
357 | { | ||
358 | mpeg2dec->custom_fbuf = custom_fbuf; | ||
359 | } | ||
360 | |||
361 | void 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 | |||
367 | void 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 | |||
375 | void 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 | |||
385 | uint32_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 | |||
398 | void 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 | |||
419 | mpeg2dec_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 | |||
442 | void mpeg2_close (mpeg2dec_t * mpeg2dec) | ||
443 | { | ||
444 | mpeg2_header_state_init (mpeg2dec); | ||
445 | mpeg2_free (mpeg2dec->chunk_buffer); | ||
446 | mpeg2_free (mpeg2dec); | ||
447 | } | ||