lavc/h264_ps: add ff_h264_ps_uninit and use it

This commit is contained in:
Matthieu Bouron 2016-06-20 16:02:10 +02:00
parent acfab2dce6
commit 0ea58059d6
4 changed files with 25 additions and 16 deletions

View File

@ -377,15 +377,7 @@ static av_cold int h264_decode_end(AVCodecContext *avctx)
h->nb_slice_ctx = 0;
ff_h264_sei_uninit(&h->sei);
for (i = 0; i < MAX_SPS_COUNT; i++)
av_buffer_unref(&h->ps.sps_list[i]);
for (i = 0; i < MAX_PPS_COUNT; i++)
av_buffer_unref(&h->ps.pps_list[i]);
av_buffer_unref(&h->ps.sps_ref);
av_buffer_unref(&h->ps.pps_ref);
ff_h264_ps_uninit(&h->ps);
ff_h2645_packet_uninit(&h->pkt);

View File

@ -694,6 +694,12 @@ typedef struct H264Context {
extern const uint16_t ff_h264_mb_sizes[4];
/**
* Uninit H264 param sets structure.
*/
void ff_h264_ps_uninit(H264ParamSets *ps);
/**
* Decode SPS
*/

View File

@ -649,17 +649,11 @@ static void h264_close(AVCodecParserContext *s)
{
H264ParseContext *p = s->priv_data;
ParseContext *pc = &p->pc;
int i;
av_freep(&pc->buffer);
ff_h264_sei_uninit(&p->sei);
for (i = 0; i < FF_ARRAY_ELEMS(p->ps.sps_list); i++)
av_buffer_unref(&p->ps.sps_list[i]);
for (i = 0; i < FF_ARRAY_ELEMS(p->ps.pps_list); i++)
av_buffer_unref(&p->ps.pps_list[i]);
ff_h264_ps_uninit(&p->ps);
}
static av_cold int init(AVCodecParserContext *s)

View File

@ -298,6 +298,23 @@ static void decode_scaling_matrices(GetBitContext *gb, SPS *sps,
}
}
void ff_h264_ps_uninit(H264ParamSets *ps)
{
int i;
for (i = 0; i < MAX_SPS_COUNT; i++)
av_buffer_unref(&ps->sps_list[i]);
for (i = 0; i < MAX_PPS_COUNT; i++)
av_buffer_unref(&ps->pps_list[i]);
av_buffer_unref(&ps->sps_ref);
av_buffer_unref(&ps->pps_ref);
ps->pps = NULL;
ps->sps = NULL;
}
int ff_h264_decode_seq_parameter_set(GetBitContext *gb, AVCodecContext *avctx,
H264ParamSets *ps, int ignore_truncation)
{