1
mirror of https://git.videolan.org/git/ffmpeg.git synced 2024-08-02 01:09:59 +02:00

h264_parser: fully parse first_mb_in_slice to more reliably detect AU boundaries.

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2011-10-27 17:53:29 +02:00
parent 23040e2e3d
commit 4ffed61b49
2 changed files with 24 additions and 9 deletions

View File

@ -591,6 +591,10 @@ typedef struct H264Context{
int16_t slice_row[MAX_SLICES]; ///< to detect when MAX_SLICES is too low
int sync; ///< did we had a keyframe or recovery point
uint8_t parse_history[4];
int parse_history_count;
int parse_last_mb;
}H264Context;

View File

@ -93,17 +93,28 @@ static int ff_h264_find_frame_end(H264Context *h, const uint8_t *buf, int buf_si
goto found;
}
}else if(v==1 || v==2 || v==5){
if(pc->frame_start_found){
state+=8;
continue;
}else
pc->frame_start_found = 1;
state+=8;
continue;
}
state= 7;
}else{
if(buf[i] & 0x80)
goto found;
state= 7;
h->parse_history[h->parse_history_count++]= buf[i];
if(h->parse_history_count>3){
unsigned int mb, last_mb= h->parse_last_mb;
GetBitContext gb;
init_get_bits(&gb, h->parse_history, 8*h->parse_history_count);
h->parse_history_count=0;
mb= get_ue_golomb_long(&gb);
last_mb= h->parse_last_mb;
h->parse_last_mb= mb;
if(pc->frame_start_found){
if(mb <= last_mb)
goto found;
}else
pc->frame_start_found = 1;
state= 7;
}
}
}
pc->state= state;
@ -116,7 +127,7 @@ found:
pc->frame_start_found= 0;
if(h->is_avc)
return next_avc;
return i-(state&5);
return i-(state&5) - 3*(state>7);
}
/**