Hi, patch attached.
From c719b95ce84bb1c3db3e6b751d80a840f35726b0 Mon Sep 17 00:00:00 2001 From: Paul B Mahol <one...@gmail.com> Date: Sat, 6 Feb 2016 11:19:45 +0100 Subject: [PATCH] avfilter: add metadata filters
Signed-off-by: Paul B Mahol <one...@gmail.com> --- doc/filters.texi | 45 ++++++++++ libavfilter/Makefile | 2 + libavfilter/allfilters.c | 2 + libavfilter/f_metadata.c | 211 +++++++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 260 insertions(+) create mode 100644 libavfilter/f_metadata.c diff --git a/doc/filters.texi b/doc/filters.texi index 664ebe8..54ea63e 100644 --- a/doc/filters.texi +++ b/doc/filters.texi @@ -8444,6 +8444,51 @@ format=rgb24,mergeplanes=0x000102:yuv444p @end example @end itemize +@section metadata, ametadata + +Manipulate frames metadata. + +This filter accepts the following options: + +@table @option +@item m +Set mode of operation of the filter. + +Can be one of the following: + +@table @samp +@item select +If both @code{value} and @code{key} is set, select frame +which have such metadata. If only @code{key} is set, select +every frame that have such key in metadata. + +@item add +Add new metadata @code{key} and @code{value}. If key is already available +do nothing. + +@item modify +Modify value of already present key. + +@item delete +If @code{value} is set, delete only keys that have such value. +Otherwise, delete key. + +@item print +Print key and value if metadata was found. +@end table + +@item key +Set key used with all modes, Must be set. + +@item value +Set metadata value which will be used. This option is mandatory for modify +and add mode. + +@item length +Set length of how much characters of two metadata values need to match to be +considered same. Default is all available characters. +@end table + @section mpdecimate Drop frames that do not differ greatly from the previous frame in diff --git a/libavfilter/Makefile b/libavfilter/Makefile index 8ffd693..664c773 100644 --- a/libavfilter/Makefile +++ b/libavfilter/Makefile @@ -39,6 +39,7 @@ OBJS-$(CONFIG_AINTERLEAVE_FILTER) += f_interleave.o OBJS-$(CONFIG_ALIMITER_FILTER) += af_alimiter.o OBJS-$(CONFIG_ALLPASS_FILTER) += af_biquads.o OBJS-$(CONFIG_AMERGE_FILTER) += af_amerge.o +OBJS-$(CONFIG_AMETADATA_FILTER) += f_metadata.o OBJS-$(CONFIG_AMIX_FILTER) += af_amix.o OBJS-$(CONFIG_ANULL_FILTER) += af_anull.o OBJS-$(CONFIG_APAD_FILTER) += af_apad.o @@ -186,6 +187,7 @@ OBJS-$(CONFIG_LUTYUV_FILTER) += vf_lut.o OBJS-$(CONFIG_MASKEDMERGE_FILTER) += vf_maskedmerge.o framesync.o OBJS-$(CONFIG_MCDEINT_FILTER) += vf_mcdeint.o OBJS-$(CONFIG_MERGEPLANES_FILTER) += vf_mergeplanes.o framesync.o +OBJS-$(CONFIG_METADATA_FILTER) += f_metadata.o OBJS-$(CONFIG_MPDECIMATE_FILTER) += vf_mpdecimate.o OBJS-$(CONFIG_NEGATE_FILTER) += vf_lut.o OBJS-$(CONFIG_NNEDI_FILTER) += vf_nnedi.o diff --git a/libavfilter/allfilters.c b/libavfilter/allfilters.c index d4d37c9..f56b390 100644 --- a/libavfilter/allfilters.c +++ b/libavfilter/allfilters.c @@ -59,6 +59,7 @@ void avfilter_register_all(void) REGISTER_FILTER(ALIMITER, alimiter, af); REGISTER_FILTER(ALLPASS, allpass, af); REGISTER_FILTER(AMERGE, amerge, af); + REGISTER_FILTER(AMETADATA, ametadata, af); REGISTER_FILTER(AMIX, amix, af); REGISTER_FILTER(ANEQUALIZER, anequalizer, af); REGISTER_FILTER(ANULL, anull, af); @@ -207,6 +208,7 @@ void avfilter_register_all(void) REGISTER_FILTER(MASKEDMERGE, maskedmerge, vf); REGISTER_FILTER(MCDEINT, mcdeint, vf); REGISTER_FILTER(MERGEPLANES, mergeplanes, vf); + REGISTER_FILTER(METADATA, metadata, vf); REGISTER_FILTER(MPDECIMATE, mpdecimate, vf); REGISTER_FILTER(NEGATE, negate, vf); REGISTER_FILTER(NNEDI, nnedi, vf); diff --git a/libavfilter/f_metadata.c b/libavfilter/f_metadata.c new file mode 100644 index 0000000..e248322 --- /dev/null +++ b/libavfilter/f_metadata.c @@ -0,0 +1,211 @@ +/* + * Copyright (c) 2016 Paul B Mahol + * + * This file is part of FFmpeg. + * + * 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. + * + * 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 FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +/** + * @file + * filter for manipulating frame metadata + */ + +#include "libavutil/avassert.h" +#include "libavutil/avstring.h" +#include "libavutil/eval.h" +#include "libavutil/fifo.h" +#include "libavutil/internal.h" +#include "libavutil/opt.h" +#include "libavutil/pixelutils.h" +#include "avfilter.h" +#include "audio.h" +#include "formats.h" +#include "internal.h" +#include "video.h" + +enum MetadataMode { + METADATA_SELECT, + METADATA_ADD, + METADATA_MODIFY, + METADATA_DELETE, + METADATA_PRINT, + METADATA_NB +}; + +typedef struct MetadataContext { + const AVClass *class; + + int mode; + char *key; + char *value; + int length; +} MetadataContext; + +#define OFFSET(x) offsetof(MetadataContext, x) +#define DEFINE_OPTIONS(filt_name, FLAGS) \ +static const AVOption filt_name##_options[] = { \ + { "m", "set a mode of operation", OFFSET(mode), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, METADATA_NB-1, FLAGS, "mode" }, \ + { "select", "select frame", 0, AV_OPT_TYPE_CONST, {.i64 = METADATA_SELECT }, 0, 0, FLAGS, "mode" }, \ + { "add", "add new metadata", 0, AV_OPT_TYPE_CONST, {.i64 = METADATA_ADD }, 0, 0, FLAGS, "mode" }, \ + { "modify", "modify metadata", 0, AV_OPT_TYPE_CONST, {.i64 = METADATA_MODIFY }, 0, 0, FLAGS, "mode" }, \ + { "delete", "delete metadata", 0, AV_OPT_TYPE_CONST, {.i64 = METADATA_DELETE }, 0, 0, FLAGS, "mode" }, \ + { "print", "print metadata", 0, AV_OPT_TYPE_CONST, {.i64 = METADATA_PRINT }, 0, 0, FLAGS, "mode" }, \ + { "key", "set metadata key", OFFSET(key), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, FLAGS }, \ + { "value", "set metadata value", OFFSET(value), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, FLAGS }, \ + { "length", "compare up to N chars", OFFSET(length), AV_OPT_TYPE_INT, {.i64 = INT_MAX }, 1, INT_MAX, FLAGS }, \ + { NULL } \ +} + +static av_cold int init(AVFilterContext *ctx) +{ + MetadataContext *s = ctx->priv; + + if (!s->key) + return AVERROR(EINVAL); + + if ((s->mode == METADATA_MODIFY || + s->mode == METADATA_ADD) && !s->value) + return AVERROR(EINVAL); + + if (s->mode == METADATA_PRINT && s->value) + av_log(ctx, AV_LOG_WARNING, "Ignoring value parameter\n"); + + return 0; +} + +static int filter_frame(AVFilterLink *inlink, AVFrame *frame) +{ + AVFilterContext *ctx = inlink->dst; + AVFilterLink *outlink = ctx->outputs[0]; + MetadataContext *s = ctx->priv; + AVDictionary *metadata = av_frame_get_metadata(frame); + AVDictionaryEntry *e; + + if (!metadata) + return ff_filter_frame(outlink, frame); + + e = av_dict_get(metadata, s->key, NULL, 0); + + switch (s->mode) { + case METADATA_SELECT: + if (!s->value && e && e->value) { + return ff_filter_frame(outlink, frame); + } else if (s->value && e && e->value && + !strncmp(s->value, e->value, s->length)) { + return ff_filter_frame(outlink, frame); + } + break; + case METADATA_ADD: + if (e && e->value) { + ; + } else { + av_dict_set(&metadata, s->key, s->value, 0); + } + return ff_filter_frame(outlink, frame); + break; + case METADATA_MODIFY: + if (e && e->value) { + av_dict_set(&metadata, s->key, s->value, 0); + } + return ff_filter_frame(outlink, frame); + break; + case METADATA_PRINT: + if (e && e->value) + av_log(ctx, AV_LOG_INFO, "%s=%s\n", s->key, e->value); + return ff_filter_frame(outlink, frame); + break; + case METADATA_DELETE: + if (e && e->value && s->value && !strncmp(s->value, e->value, s->length)) { + av_dict_set(&metadata, s->key, NULL, 0); + } else if (e && e->value) { + av_dict_set(&metadata, s->key, NULL, 0); + } + return ff_filter_frame(outlink, frame); + break; + default: + av_assert0(0); + }; + + av_frame_free(&frame); + + return 0; +} + +#if CONFIG_AMETADATA_FILTER + +DEFINE_OPTIONS(ametadata, AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM); +AVFILTER_DEFINE_CLASS(ametadata); + +static const AVFilterPad ainputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_AUDIO, + .filter_frame = filter_frame, + }, + { NULL } +}; + +static const AVFilterPad aoutputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_AUDIO, + }, + { NULL } +}; + +AVFilter ff_af_ametadata = { + .name = "ametadata", + .description = NULL_IF_CONFIG_SMALL("Manipulate audio frames metadata."), + .priv_size = sizeof(MetadataContext), + .priv_class = &ametadata_class, + .init = init, + .inputs = ainputs, + .outputs = aoutputs, +}; +#endif /* CONFIG_AMETADATA_FILTER */ + +#if CONFIG_METADATA_FILTER + +DEFINE_OPTIONS(metadata, AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM); +AVFILTER_DEFINE_CLASS(metadata); + +static const AVFilterPad inputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_VIDEO, + .filter_frame = filter_frame, + }, + { NULL } +}; + +static const AVFilterPad outputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_VIDEO, + }, + { NULL } +}; + +AVFilter ff_vf_metadata = { + .name = "metadata", + .description = NULL_IF_CONFIG_SMALL("Manipulate video frames metadata."), + .priv_size = sizeof(MetadataContext), + .priv_class = &metadata_class, + .init = init, + .inputs = inputs, + .outputs = outputs, +}; +#endif /* CONFIG_METADATA_FILTER */ -- 1.9.1
_______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org http://ffmpeg.org/mailman/listinfo/ffmpeg-devel