static int decode_extradata_ps(const uint8_t *data, int size, H264ParamSets *ps,
int is_avc, void *logctx)
{
H2645Packet pkt = { 0 };
int i, ret = 0;
ret = ff_h2645_packet_split(&pkt, data, size, logctx, is_avc, 2, AV_CODEC_ID_H264);
if (ret < 0)
goto fail;
for (i = 0; i < pkt.nb_nals; i++) {
H2645NAL *nal = &pkt.nals[i];
switch (nal->type) {
case NAL_SPS:
ret = ff_h264_decode_seq_parameter_set(&nal->gb, logctx, ps);
if (ret < 0)
goto fail;
break;
case NAL_PPS:
ret = ff_h264_decode_picture_parameter_set(&nal->gb, logctx, ps,
nal->size_bits);
if (ret < 0)
goto fail;
break;
default:
av_log(logctx, AV_LOG_VERBOSE, "Ignoring NAL type %d in extradata\n",
nal->type);
break;
}
}
fail:
ff_h2645_packet_uninit(&pkt);
return ret;
}
/**
* ffdshow custom stuff (based on decode_nal_units)
* Copyright (c) 2008-2012 Haruhiko Yamagata (except the parts copied from decode_nal_units)
*
* @param[out] recovery_frame_cnt. Valid only if GDR.
* @return 0: no recovery point, 1:I-frame 2:Recovery Point SEI (GDR), 3:IDR, -1:error
*/
int avcodec_h264_search_recovery_point(AVCodecContext *avctx,
const uint8_t *buf, int buf_size, int *recovery_frame_cnt)
{
H264Context *h;
MpegEncContext *s;
int buf_index = 0;
int found = 0; // 0: no recovery point, 1:Recovery Point SEI (GDR), 2:IDR
AVCodecContext *users_MpegEncContext_avctx;
avctx = get_thread0_avctx(avctx); // Next frame will start on thread 0, and we want to store SPS and PPS in the context of thread 0.
h = avctx->priv_data;
h->is_avc = avcodec_h264_decode_init_is_avc(avctx);
s = &h->s;
users_MpegEncContext_avctx = s->avctx; // save it and write back before return.
if (s->avctx == NULL)
s->avctx = avctx; // Hack, this function can be used before decoding, so we can't expect everything initialized.
h->nal_length_size = avctx->nal_length_size ? avctx->nal_length_size : 4;
for(;;){
int consumed;
int dst_length;
int bit_length;
const uint8_t *ptr;
int i, nalsize = 0;
if(h->is_avc) {
if(buf_index >= buf_size) break;
nalsize = 0;
for(i = 0; i < h->nal_length_size; i++)
nalsize = (nalsize << 8) | buf[buf_index++];
if(nalsize <= 1 || (nalsize+buf_index > buf_size)){
if(nalsize == 1){
buf_index++;
continue;
}else{
av_log(h->s.avctx, AV_LOG_ERROR, "AVC: nal size %d\n", nalsize);
break;
}
}
} else {
// start code prefix search
for(; buf_index + 3 < buf_size; buf_index++){
// This should always succeed in the first iteration.
if(buf[buf_index] == 0 && buf[buf_index+1] == 0 && buf[buf_index+2] == 1)
break;
}
if(buf_index+3 >= buf_size) break;
buf_index+=3;
}
ptr= ff_h264_decode_nal(h, buf + buf_index, &dst_length, &consumed, h->is_avc ? nalsize : buf_size - buf_index);
if (ptr==NULL || dst_length < 0){
found = -1;
goto end;
}
while(ptr[dst_length - 1] == 0 && dst_length > 0)
dst_length--;
bit_length= !dst_length ? 0 : (8*dst_length - decode_rbsp_trailing(h, ptr + dst_length - 1));
if (h->is_avc && (nalsize != consumed)){
av_log(h->s.avctx, AV_LOG_ERROR, "AVC: Consumed only %d bytes instead of %d\n", consumed, nalsize);
consumed= nalsize;
}
buf_index += consumed;
switch(h->nal_unit_type){
case NAL_SEI:
if (ptr[0] == 6/* Recovery Point SEI */){
init_get_bits(&s->gb, ptr, bit_length);
ff_h264_decode_sei(h);
if (found < 2)
found = 2;
break;
}
break;
case NAL_SPS:
init_get_bits(&s->gb, ptr, bit_length);
ff_h264_decode_seq_parameter_set(h);
if(s->flags& CODEC_FLAG_LOW_DELAY)
s->low_delay=1;
if(avctx->has_b_frames < 2)
avctx->has_b_frames= !s->low_delay;
break;
case NAL_PPS:
init_get_bits(&s->gb, ptr, bit_length);
ff_h264_decode_picture_parameter_set(h, bit_length);
break;
case NAL_AUD:
{
int primary_pic_type;
init_get_bits(&s->gb, ptr, bit_length);
primary_pic_type = get_bits(&s->gb, 3);
if (found == 0 && (primary_pic_type == 0 || primary_pic_type == 3)) // I-frame (all I/SI slices)
found = 1;
break;
}
case NAL_IDR_SLICE:
case NAL_SLICE:
case NAL_DPA:
// decode part of slice header and find I frame
init_get_bits(&s->gb, ptr, bit_length);
int first_mb_in_slice = get_ue_golomb(&s->gb);
unsigned int slice_type= get_ue_golomb(&s->gb);
if (!is_I_slice(slice_type))
goto end;
// at least one slice must start at the left top corner.
// ASO should work, while FMO may not. FMO is not supported by current Libav anyway.
if (first_mb_in_slice == 0) {
if (h->nal_unit_type == NAL_IDR_SLICE) {
found = 3;
goto end;
} else {
if (found == 0) {
found = 1;
goto end;
}
}
}
break;
case NAL_DPB:
case NAL_DPC:
case NAL_END_SEQUENCE:
case NAL_END_STREAM:
case NAL_FILLER_DATA:
case NAL_SPS_EXT:
case NAL_AUXILIARY_SLICE:
break;
default:
av_log(avctx, AV_LOG_DEBUG, "Unknown NAL code: %d (%d bits)\n", h->nal_unit_type, bit_length);
}
}
*recovery_frame_cnt = h->sei_recovery_frame_cnt;
end:
s->avctx = users_MpegEncContext_avctx;
return found;
}
