avcodec/remove_extradata: move the reference in the bsf internal buffer

There's no need to allocate a new packet for it.

Reviewed-by: Mark Thompson <sw@jkqxz.net>
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
James Almer 2018-03-11 13:45:19 -03:00
parent 11bef2fe72
commit a1a0859ad5
1 changed files with 7 additions and 11 deletions

View File

@ -38,30 +38,26 @@ typedef struct RemoveExtradataContext {
AVCodecContext *avctx;
} RemoveExtradataContext;
static int remove_extradata(AVBSFContext *ctx, AVPacket *out)
static int remove_extradata(AVBSFContext *ctx, AVPacket *pkt)
{
RemoveExtradataContext *s = ctx->priv_data;
AVPacket *in;
int ret;
ret = ff_bsf_get_packet(ctx, &in);
ret = ff_bsf_get_packet_ref(ctx, pkt);
if (ret < 0)
return ret;
if (s->parser && s->parser->parser->split) {
if (s->freq == REMOVE_FREQ_ALL ||
(s->freq == REMOVE_FREQ_NONKEYFRAME && !(in->flags & AV_PKT_FLAG_KEY)) ||
(s->freq == REMOVE_FREQ_KEYFRAME && in->flags & AV_PKT_FLAG_KEY)) {
int i = s->parser->parser->split(s->avctx, in->data, in->size);
in->data += i;
in->size -= i;
(s->freq == REMOVE_FREQ_NONKEYFRAME && !(pkt->flags & AV_PKT_FLAG_KEY)) ||
(s->freq == REMOVE_FREQ_KEYFRAME && pkt->flags & AV_PKT_FLAG_KEY)) {
int i = s->parser->parser->split(s->avctx, pkt->data, pkt->size);
pkt->data += i;
pkt->size -= i;
}
}
av_packet_move_ref(out, in);
av_packet_free(&in);
return 0;
}