3 * Copyright (C) 2000-2003 Michel Lespinasse <walken@zoy.org>
4 * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
6 * This file is part of mpeg2dec, a free MPEG-2 video stream decoder.
7 * See http://libmpeg2.sourceforge.net/ for updates.
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.
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.
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
24 #include "xxmc-config.h"
26 #include <string.h> /* memcmp/memset, try to remove */
31 #include "attributes.h"
32 #include "mpeg2_internal.h"
34 static int mpeg2_accels = 0;
36 #define BUFFER_SIZE (1194 * 1024)
38 const mpeg2_info_t *mpeg2_info( mpeg2dec_t * mpeg2dec )
40 return &(mpeg2dec->info);
43 static inline int skip_chunk( mpeg2dec_t * mpeg2dec, int bytes )
45 uint8_t *current = NULL;
47 uint8_t *limit = NULL;
53 current = mpeg2dec->buf_start;
54 shift = mpeg2dec->shift;
55 limit = current + bytes;
59 if (shift == 0x00000100)
63 mpeg2dec->shift = 0xffffff00;
64 skipped = current - mpeg2dec->buf_start;
65 mpeg2dec->buf_start = current;
68 shift = (shift | byte) << 8;
69 } while (current < limit);
71 mpeg2dec->shift = shift;
72 mpeg2dec->buf_start = current;
76 static inline int copy_chunk( mpeg2dec_t * mpeg2dec, int bytes )
78 uint8_t *current = NULL;
80 uint8_t *chunk_ptr = NULL;
81 uint8_t *limit = NULL;
87 current = mpeg2dec->buf_start;
88 shift = mpeg2dec->shift;
89 chunk_ptr = mpeg2dec->chunk_ptr;
90 limit = current + bytes;
94 if (shift == 0x00000100)
98 mpeg2dec->shift = 0xffffff00;
99 mpeg2dec->chunk_size = chunk_ptr - mpeg2dec->chunk_start - 3;
100 mpeg2dec->chunk_ptr = chunk_ptr + 1;
101 copied = current - mpeg2dec->buf_start;
102 mpeg2dec->buf_start = current;
105 shift = (shift | byte) << 8;
107 } while (current < limit);
109 mpeg2dec->shift = shift;
110 mpeg2dec->buf_start = current;
114 void mpeg2_buffer( mpeg2dec_t * mpeg2dec, uint8_t * start, uint8_t * end )
116 mpeg2dec->buf_start = start;
117 mpeg2dec->buf_end = end;
120 int mpeg2_getpos( mpeg2dec_t * mpeg2dec )
122 return mpeg2dec->buf_end - mpeg2dec->buf_start;
125 static inline mpeg2_state_t seek_chunk( mpeg2dec_t * mpeg2dec )
129 size = mpeg2dec->buf_end - mpeg2dec->buf_start;
130 skipped = skip_chunk (mpeg2dec, size);
133 mpeg2dec->bytes_since_tag += size;
136 mpeg2dec->bytes_since_tag += skipped;
137 mpeg2dec->code = mpeg2dec->buf_start[-1];
138 return (mpeg2_state_t)-1;
141 mpeg2_state_t mpeg2_seek_header( mpeg2dec_t * mpeg2dec )
143 while( (mpeg2dec->code != 0xb3) &&
144 (((mpeg2dec->code != 0xb7) &&
145 (mpeg2dec->code != 0xb8) &&
147 (mpeg2dec->sequence.width == (unsigned)-1)) )
149 if (seek_chunk (mpeg2dec) == STATE_BUFFER)
153 mpeg2dec->chunk_start = mpeg2dec->chunk_ptr = mpeg2dec->chunk_buffer;
154 mpeg2dec->user_data_len = 0;
155 return (mpeg2dec->code ? mpeg2_parse_header (mpeg2dec) :
156 mpeg2_header_picture_start (mpeg2dec));
159 #define RECEIVED(code,state) (((state) << 8) + (code))
161 mpeg2_state_t mpeg2_parse( mpeg2dec_t * mpeg2dec )
163 int size_buffer, size_chunk, copied;
165 if (mpeg2dec->action)
169 state = mpeg2dec->action (mpeg2dec);
176 while( (unsignedint) (mpeg2dec->code - mpeg2dec->first_decode_slice)
177 < mpeg2dec->nb_decode_slices )
179 size_buffer = mpeg2dec->buf_end - mpeg2dec->buf_start;
180 size_chunk = (mpeg2dec->chunk_buffer + BUFFER_SIZE -
181 mpeg2dec->chunk_ptr);
183 if (size_buffer <= size_chunk)
185 copied = copy_chunk (mpeg2dec, size_buffer);
188 mpeg2dec->bytes_since_tag += size_buffer;
189 mpeg2dec->chunk_ptr += size_buffer;
195 copied = copy_chunk (mpeg2dec, size_chunk);
199 /* filled the chunk buffer without finding a start code */
200 mpeg2dec->bytes_since_tag += size_chunk;
201 mpeg2dec->action = seek_chunk;
202 return STATE_INVALID;
205 mpeg2dec->bytes_since_tag += copied;
206 mpeg2_xxmc_slice( &(mpeg2dec->decoder), NULL,
207 mpeg2dec->code,mpeg2dec->chunk_start,
208 mpeg2dec->chunk_size);
209 mpeg2dec->prev_code = mpeg2dec->code;
210 mpeg2dec->code = mpeg2dec->buf_start[-1];
211 mpeg2dec->chunk_ptr = mpeg2dec->chunk_start;
214 if( (unsigned int) (mpeg2dec->code - 1) >= (0xb0 - 1) )
216 if( seek_chunk (mpeg2dec) == STATE_BUFFER )
220 switch( mpeg2dec->code )
223 mpeg2dec->action = mpeg2_header_picture_start;
224 return mpeg2dec->state;
226 mpeg2dec->action = mpeg2_header_end;
230 mpeg2dec->action = mpeg2_parse_header;
233 mpeg2dec->action = seek_chunk;
234 return STATE_INVALID;
236 return (mpeg2dec->state == STATE_SLICE) ? STATE_SLICE : STATE_INVALID;
239 mpeg2_state_t mpeg2_parse_header( mpeg2dec_t * mpeg2dec )
241 static int (* process_header[]) (mpeg2dec_t * mpeg2dec) =
243 mpeg2_header_picture,
244 mpeg2_header_extension,
245 mpeg2_header_user_data,
246 mpeg2_header_sequence,
253 int size_buffer, size_chunk, copied;
255 mpeg2dec->action = mpeg2_parse_header;
256 mpeg2dec->info.user_data = NULL;
257 mpeg2dec->info.user_data_len = 0;
261 size_buffer = mpeg2dec->buf_end - mpeg2dec->buf_start;
262 size_chunk = (mpeg2dec->chunk_buffer + BUFFER_SIZE -
263 mpeg2dec->chunk_ptr);
264 if( size_buffer <= size_chunk )
266 copied = copy_chunk (mpeg2dec, size_buffer);
269 mpeg2dec->bytes_since_tag += size_buffer;
270 mpeg2dec->chunk_ptr += size_buffer;
276 copied = copy_chunk (mpeg2dec, size_chunk);
279 /* filled the chunk buffer without finding a start code */
280 mpeg2dec->bytes_since_tag += size_chunk;
281 mpeg2dec->code = 0xb4;
282 mpeg2dec->action = mpeg2_seek_header;
283 return STATE_INVALID;
286 mpeg2dec->bytes_since_tag += copied;
288 if( process_header[mpeg2dec->code & 0x0b](mpeg2dec) )
290 mpeg2dec->code = mpeg2dec->buf_start[-1];
291 mpeg2dec->action = mpeg2_seek_header;
292 return STATE_INVALID;
294 mpeg2dec->code = mpeg2dec->buf_start[-1];
296 switch( RECEIVED(mpeg2dec->code, mpeg2dec->state) )
298 /* state transition after a sequence header */
299 case RECEIVED (0x00, STATE_SEQUENCE):
300 mpeg2dec->action = mpeg2_header_picture_start;
301 case RECEIVED (0xb8, STATE_SEQUENCE):
302 mpeg2_header_sequence_finalize( mpeg2dec );
304 /* other legal state transitions */
305 case RECEIVED (0x00, STATE_GOP):
306 mpeg2_header_gop_finalize( mpeg2dec );
307 mpeg2dec->action = mpeg2_header_picture_start;
309 case RECEIVED (0x01, STATE_PICTURE):
310 case RECEIVED (0x01, STATE_PICTURE_2ND):
311 mpeg2_header_picture_finalize( mpeg2dec, mpeg2_accels );
312 mpeg2dec->action = mpeg2_header_slice_start;
314 /* legal headers within a given state */
315 case RECEIVED (0xb2, STATE_SEQUENCE):
316 case RECEIVED (0xb2, STATE_GOP):
317 case RECEIVED (0xb2, STATE_PICTURE):
318 case RECEIVED (0xb2, STATE_PICTURE_2ND):
319 case RECEIVED (0xb5, STATE_SEQUENCE):
320 case RECEIVED (0xb5, STATE_PICTURE):
321 case RECEIVED (0xb5, STATE_PICTURE_2ND):
322 mpeg2dec->chunk_ptr = mpeg2dec->chunk_start;
325 mpeg2dec->action = mpeg2_seek_header;
326 return STATE_INVALID;
328 mpeg2dec->chunk_start = mpeg2dec->chunk_ptr = mpeg2dec->chunk_buffer;
329 mpeg2dec->user_data_len = 0;
330 return mpeg2dec->state;
334 int mpeg2_convert( mpeg2dec_t * mpeg2dec, mpeg2_convert_t convert, void * arg )
336 mpeg2_convert_init_t convert_init;
339 error = convert( MPEG2_CONVERT_SET, NULL, &(mpeg2dec->sequence), 0,
340 mpeg2_accels, arg, &convert_init );
343 mpeg2dec->convert = convert;
344 mpeg2dec->convert_arg = arg;
345 mpeg2dec->convert_id_size = convert_init.id_size;
346 mpeg2dec->convert_stride = 0;
351 int mpeg2_stride( mpeg2dec_t * mpeg2dec, int stride )
353 if (!mpeg2dec->convert)
355 if (stride < (int) mpeg2dec->sequence.width)
356 stride = mpeg2dec->sequence.width;
357 mpeg2dec->decoder.stride_frame = stride;
361 mpeg2_convert_init_t convert_init;
363 stride = mpeg2dec->convert( MPEG2_CONVERT_STRIDE, NULL,
364 &(mpeg2dec->sequence), stride,
365 mpeg2_accels, mpeg2dec->convert_arg,
367 mpeg2dec->convert_id_size = convert_init.id_size;
368 mpeg2dec->convert_stride = stride;
373 void mpeg2_set_buf( mpeg2dec_t * mpeg2dec, uint8_t * buf[3], void * id )
377 if (mpeg2dec->custom_fbuf)
379 if (mpeg2dec->state == STATE_SEQUENCE)
381 mpeg2dec->fbuf[2] = mpeg2dec->fbuf[1];
382 mpeg2dec->fbuf[1] = mpeg2dec->fbuf[0];
384 mpeg2_set_fbuf( mpeg2dec,
385 (mpeg2dec->decoder.coding_type == PIC_FLAG_CODING_TYPE_B) );
386 fbuf = mpeg2dec->fbuf[0];
390 fbuf = &(mpeg2dec->fbuf_alloc[mpeg2dec->alloc_index].fbuf);
391 mpeg2dec->alloc_index_user = ++mpeg2dec->alloc_index;
393 fbuf->buf[0] = buf[0];
394 fbuf->buf[1] = buf[1];
395 fbuf->buf[2] = buf[2];
399 void mpeg2_custom_fbuf( mpeg2dec_t * mpeg2dec, int custom_fbuf )
401 mpeg2dec->custom_fbuf = custom_fbuf;
404 void mpeg2_skip( mpeg2dec_t * mpeg2dec, int skip )
406 mpeg2dec->first_decode_slice = 1;
407 mpeg2dec->nb_decode_slices = skip ? 0 : (0xb0 - 1);
410 void mpeg2_slice_region( mpeg2dec_t * mpeg2dec, int start, int end )
412 start = (start < 1) ? 1 : (start > 0xb0) ? 0xb0 : start;
413 end = (end < start) ? start : (end > 0xb0) ? 0xb0 : end;
414 mpeg2dec->first_decode_slice = start;
415 mpeg2dec->nb_decode_slices = end - start;
418 void mpeg2_tag_picture( mpeg2dec_t * mpeg2dec, uint32_t tag, uint32_t tag2 )
420 mpeg2dec->tag_previous = mpeg2dec->tag_current;
421 mpeg2dec->tag2_previous = mpeg2dec->tag2_current;
422 mpeg2dec->tag_current = tag;
423 mpeg2dec->tag2_current = tag2;
424 mpeg2dec->num_tags++;
425 mpeg2dec->bytes_since_tag = 0;
428 uint32_t mpeg2_accel( uint32_t accel )
432 if( accel & MPEG2_ACCEL_DETECT )
433 accel |= mpeg2_detect_accel ();
434 mpeg2_accels = accel |= MPEG2_ACCEL_DETECT;
435 mpeg2_cpu_state_init (accel);
436 /* mpeg2_idct_init (accel); */
437 mpeg2_mc_init (accel);
439 return mpeg2_accels & ~MPEG2_ACCEL_DETECT;
442 void mpeg2_reset( mpeg2dec_t * mpeg2dec, int full_reset )
444 mpeg2dec->buf_start = mpeg2dec->buf_end = NULL;
445 mpeg2dec->num_tags = 0;
446 mpeg2dec->shift = 0xffffff00;
447 mpeg2dec->code = 0xb4;
448 mpeg2dec->action = mpeg2_seek_header;
449 mpeg2dec->state = STATE_INVALID;
451 mpeg2dec->ptr_forward_ref_picture = NULL;
452 mpeg2dec->ptr_backward_ref_picture = NULL;
454 mpeg2_reset_info(&(mpeg2dec->info));
455 mpeg2dec->info.gop = NULL;
456 mpeg2dec->info.user_data = NULL;
457 mpeg2dec->info.user_data_len = 0;
460 mpeg2dec->info.sequence = NULL;
461 mpeg2_header_state_init (mpeg2dec);
465 mpeg2dec_t *mpeg2_init( void )
467 mpeg2dec_t *mpeg2dec = NULL;
469 mpeg2_accel( MPEG2_ACCEL_DETECT );
470 mpeg2dec = (mpeg2dec_t *) mpeg2_malloc( sizeof (mpeg2dec_t),
471 MPEG2_ALLOC_MPEG2DEC );
472 if( mpeg2dec == NULL )
475 memset( mpeg2dec->decoder.DCTblock, 0, 64 * sizeof (int16_t) );
476 memset( mpeg2dec->quantizer_matrix, 0, 4 * 64 * sizeof (uint8_t) );
478 mpeg2dec->chunk_buffer = (uint8_t *) mpeg2_malloc( BUFFER_SIZE + 4,
480 mpeg2dec->sequence.width = (unsigned int)-1;
481 mpeg2_reset (mpeg2dec, 1);
485 void mpeg2_close( mpeg2dec_t * mpeg2dec )
487 mpeg2_header_state_init( mpeg2dec );
488 mpeg2_free( mpeg2dec->chunk_buffer );
489 mpeg2_free( mpeg2dec );