diff options
Diffstat (limited to 'libavcodec/noise_bsf.c')
-rw-r--r-- | libavcodec/noise_bsf.c | 45 |
1 files changed, 25 insertions, 20 deletions
diff --git a/libavcodec/noise_bsf.c b/libavcodec/noise_bsf.c index 3b41dbf..d79f63b 100644 --- a/libavcodec/noise_bsf.c +++ b/libavcodec/noise_bsf.c @@ -1,20 +1,20 @@ /* * copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at> * - * This file is part of Libav. + * This file is part of FFmpeg. * - * Libav is free software; you can redistribute it and/or + * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * - * Libav is distributed in the hope that it will be useful, + * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public - * License along with Libav; if not, write to the Free Software + * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ @@ -31,45 +31,50 @@ typedef struct NoiseContext { const AVClass *class; int amount; + int dropamount; unsigned int state; } NoiseContext; -static int noise(AVBSFContext *ctx, AVPacket *out) +static int noise(AVBSFContext *ctx, AVPacket *pkt) { NoiseContext *s = ctx->priv_data; - AVPacket *in; int amount = s->amount > 0 ? s->amount : (s->state % 10001 + 1); int i, ret = 0; - ret = ff_bsf_get_packet(ctx, &in); + if (amount <= 0) + return AVERROR(EINVAL); + + ret = ff_bsf_get_packet_ref(ctx, pkt); if (ret < 0) return ret; - ret = av_new_packet(out, in->size); - if (ret < 0) - goto fail; + if (s->dropamount > 0 && s->state % s->dropamount == 0) { + s->state++; + av_packet_unref(pkt); + return AVERROR(EAGAIN); + } - ret = av_packet_copy_props(out, in); + ret = av_packet_make_writable(pkt); if (ret < 0) goto fail; - memcpy(out->data, in->data, in->size); - - for (i = 0; i < out->size; i++) { - s->state += out->data[i] + 1; + for (i = 0; i < pkt->size; i++) { + s->state += pkt->data[i] + 1; if (s->state % amount == 0) - out->data[i] = s->state; + pkt->data[i] = s->state; } fail: if (ret < 0) - av_packet_unref(out); - av_packet_free(&in); + av_packet_unref(pkt); + return ret; } #define OFFSET(x) offsetof(NoiseContext, x) +#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_BSF_PARAM) static const AVOption options[] = { - { "amount", NULL, OFFSET(amount), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX }, + { "amount", NULL, OFFSET(amount), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, FLAGS }, + { "dropamount", NULL, OFFSET(dropamount), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, FLAGS }, { NULL }, }; @@ -82,7 +87,7 @@ static const AVClass noise_class = { const AVBitStreamFilter ff_noise_bsf = { .name = "noise", - .priv_data_size = sizeof(int), + .priv_data_size = sizeof(NoiseContext), .priv_class = &noise_class, .filter = noise, }; |