fix parser

Originally committed as revision 3689 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
Michael Niedermayer 2004-11-19 11:37:03 +00:00
parent acfd8f0fdb
commit e731697665
1 changed files with 19 additions and 40 deletions

View File

@ -851,30 +851,24 @@ static int h261_find_frame_end(ParseContext *pc, AVCodecContext* avctx, const ui
vop_found= pc->frame_start_found; vop_found= pc->frame_start_found;
state= pc->state; state= pc->state;
i=0; for(i=0; i<buf_size && !vop_found; i++){
if(!vop_found){ state= (state<<8) | buf[i];
for(i=0; i<buf_size; i++){ for(j=0; j<8; j++){
state= (state<<8) | buf[i]; if(((state>>j)&0xFFFFF) == 0x00010){
for(j=0; j<8; j++){ i++;
if(( ( (state<<j) | (buf[i]>>(8-j)) )>>(32-20) == 0x10 )&&(((state >> (17-j)) & 0x4000) == 0x0)){ vop_found=1;
i++; break;
vop_found=1;
break;
}
} }
if(vop_found)
break;
} }
} }
if(vop_found){ if(vop_found){
for(; i<buf_size; i++){ for(; i<buf_size; i++){
if(avctx->flags & CODEC_FLAG_TRUNCATED)//XXX ffplay workaround, someone a better solution? state= (state<<8) | buf[i];
state= (state<<8) | buf[i];
for(j=0; j<8; j++){ for(j=0; j<8; j++){
if(( ( (state<<j) | (buf[i]>>(8-j)) )>>(32-20) == 0x10 )&&(((state >> (17-j)) & 0x4000) == 0x0)){ if(((state>>j)&0xFFFFF) == 0x00010){
pc->frame_start_found=0; pc->frame_start_found=0;
pc->state=-1; pc->state= state>>(2*8);
return i-3; return i-1;
} }
} }
} }
@ -908,18 +902,11 @@ static int h261_parse(AVCodecParserContext *s,
* returns the number of bytes consumed for building the current frame * returns the number of bytes consumed for building the current frame
*/ */
static int get_consumed_bytes(MpegEncContext *s, int buf_size){ static int get_consumed_bytes(MpegEncContext *s, int buf_size){
if(s->flags&CODEC_FLAG_TRUNCATED){ int pos= get_bits_count(&s->gb)>>3;
int pos= (get_bits_count(&s->gb)+7)>>3; if(pos==0) pos=1; //avoid infinite loops (i doubt thats needed but ...)
pos -= s->parse_context.last_index; if(pos+10>buf_size) pos=buf_size; // oops ;)
if(pos<0) pos=0;// padding is not really read so this might be -1
return pos;
}else{
int pos= get_bits_count(&s->gb)>>3;
if(pos==0) pos=1; //avoid infinite loops (i doubt thats needed but ...)
if(pos+10>buf_size) pos=buf_size; // oops ;)
return pos; return pos;
}
} }
static int h261_decode_frame(AVCodecContext *avctx, static int h261_decode_frame(AVCodecContext *avctx,
@ -942,16 +929,8 @@ static int h261_decode_frame(AVCodecContext *avctx,
if (buf_size == 0) { if (buf_size == 0) {
return 0; return 0;
} }
if(s->flags&CODEC_FLAG_TRUNCATED){ h->gob_start_code_skipped=0;
int next;
next= h261_find_frame_end(&s->parse_context,avctx, buf, buf_size);
if( ff_combine_frame(&s->parse_context, next, &buf, &buf_size) < 0 )
return buf_size;
}
retry: retry:
@ -1013,7 +992,7 @@ retry:
assert(s->current_picture.pict_type == s->current_picture_ptr->pict_type); assert(s->current_picture.pict_type == s->current_picture_ptr->pict_type);
assert(s->current_picture.pict_type == s->pict_type); assert(s->current_picture.pict_type == s->pict_type);
*pict= *(AVFrame*)&s->current_picture; *pict= *(AVFrame*)s->current_picture_ptr;
ff_print_debug_info(s, pict); ff_print_debug_info(s, pict);
/* Return the Picture timestamp as the frame number */ /* Return the Picture timestamp as the frame number */
@ -1053,7 +1032,7 @@ AVCodec h261_decoder = {
NULL, NULL,
h261_decode_end, h261_decode_end,
h261_decode_frame, h261_decode_frame,
CODEC_CAP_TRUNCATED, CODEC_CAP_DR1,
}; };
AVCodecParser h261_parser = { AVCodecParser h261_parser = {