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