96
98
return i-(state&5);
102
* Parse NAL units of found picture and decode some basic information.
104
* @param s parser context.
105
* @param avctx codec context.
106
* @param buf buffer with field/frame data.
107
* @param buf_size size of the buffer.
109
static inline int parse_nal_units(AVCodecParserContext *s,
110
AVCodecContext *avctx,
111
const uint8_t *buf, int buf_size)
113
H264Context *h = s->priv_data;
114
const uint8_t *buf_end = buf + buf_size;
116
unsigned int slice_type;
120
/* set some sane default values */
121
s->pict_type = FF_I_TYPE;
125
h->sei_recovery_frame_cnt = -1;
126
h->sei_dpb_output_delay = 0;
127
h->sei_cpb_removal_delay = -1;
128
h->sei_buffering_period_present = 0;
131
int src_length, dst_length, consumed;
132
buf = ff_find_start_code(buf, buf_end, &state);
136
src_length = buf_end - buf;
137
switch (state & 0x1f) {
140
// Do not walk the whole buffer just to decode slice header
145
ptr= ff_h264_decode_nal(h, buf, &dst_length, &consumed, src_length);
146
if (ptr==NULL || dst_length < 0)
149
init_get_bits(&h->s.gb, ptr, 8*dst_length);
150
switch(h->nal_unit_type) {
152
ff_h264_decode_seq_parameter_set(h);
155
ff_h264_decode_picture_parameter_set(h, h->s.gb.size_in_bits);
158
ff_h264_decode_sei(h);
164
get_ue_golomb(&h->s.gb); // skip first_mb_in_slice
165
slice_type = get_ue_golomb_31(&h->s.gb);
166
s->pict_type = golomb_to_pict_type[slice_type % 5];
167
if (h->sei_recovery_frame_cnt >= 0) {
168
/* key frame, since recovery_frame_cnt is set */
171
pps_id= get_ue_golomb(&h->s.gb);
172
if(pps_id>=MAX_PPS_COUNT) {
173
av_log(h->s.avctx, AV_LOG_ERROR, "pps_id out of range\n");
176
if(!h->pps_buffers[pps_id]) {
177
av_log(h->s.avctx, AV_LOG_ERROR, "non-existing PPS referenced\n");
180
h->pps= *h->pps_buffers[pps_id];
181
if(!h->sps_buffers[h->pps.sps_id]) {
182
av_log(h->s.avctx, AV_LOG_ERROR, "non-existing SPS referenced\n");
185
h->sps = *h->sps_buffers[h->pps.sps_id];
186
h->frame_num = get_bits(&h->s.gb, h->sps.log2_max_frame_num);
188
if(h->sps.frame_mbs_only_flag){
189
h->s.picture_structure= PICT_FRAME;
191
if(get_bits1(&h->s.gb)) { //field_pic_flag
192
h->s.picture_structure= PICT_TOP_FIELD + get_bits1(&h->s.gb); //bottom_field_flag
194
h->s.picture_structure= PICT_FRAME;
198
if(h->sps.pic_struct_present_flag) {
199
switch (h->sei_pic_struct) {
200
case SEI_PIC_STRUCT_TOP_FIELD:
201
case SEI_PIC_STRUCT_BOTTOM_FIELD:
204
case SEI_PIC_STRUCT_FRAME:
205
case SEI_PIC_STRUCT_TOP_BOTTOM:
206
case SEI_PIC_STRUCT_BOTTOM_TOP:
209
case SEI_PIC_STRUCT_TOP_BOTTOM_TOP:
210
case SEI_PIC_STRUCT_BOTTOM_TOP_BOTTOM:
213
case SEI_PIC_STRUCT_FRAME_DOUBLING:
216
case SEI_PIC_STRUCT_FRAME_TRIPLING:
220
s->repeat_pict = h->s.picture_structure == PICT_FRAME ? 1 : 0;
224
s->repeat_pict = h->s.picture_structure == PICT_FRAME ? 1 : 0;
227
return 0; /* no need to evaluate the rest */
231
/* didn't find a picture! */
232
av_log(h->s.avctx, AV_LOG_ERROR, "missing picture in access unit\n");
99
236
static int h264_parse(AVCodecParserContext *s,
100
237
AVCodecContext *avctx,
101
238
const uint8_t **poutbuf, int *poutbuf_size,
120
257
assert(pc->last_index + next >= 0 );
121
258
ff_h264_find_frame_end(h, &pc->buffer[pc->last_index + next], -next); //update state
261
parse_nal_units(s, avctx, buf, buf_size);
263
if (h->sei_cpb_removal_delay >= 0) {
264
s->dts_sync_point = h->sei_buffering_period_present;
265
s->dts_ref_dts_delta = h->sei_cpb_removal_delay;
266
s->pts_dts_delta = h->sei_dpb_output_delay;
268
s->dts_sync_point = INT_MIN;
269
s->dts_ref_dts_delta = INT_MIN;
270
s->pts_dts_delta = INT_MIN;