mirror of https://github.com/FFmpeg/FFmpeg.git
This is a similar filter to f_metadata, only it works on side data. Since adding side data from a user provided arbitrary binary string is unsafe, because current code assumes that a side data of a certain kind has the proper size, this filter only implements selection and deletion. Also, no value matching support is implemented yet, because there is no uniform way to specify a side data textually. Signed-off-by: Marton Balint <cus@passwd.hu>pull/237/merge
Marton Balint
9 years ago
parent
7a65aef00d
commit
7845c13881
6 changed files with 217 additions and 1 deletions
@ -0,0 +1,181 @@ |
||||
/*
|
||||
* 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 side data |
||||
*/ |
||||
|
||||
#include "libavutil/avassert.h" |
||||
#include "libavutil/internal.h" |
||||
#include "libavutil/frame.h" |
||||
#include "libavutil/opt.h" |
||||
#include "avfilter.h" |
||||
#include "formats.h" |
||||
#include "internal.h" |
||||
|
||||
enum SideDataMode { |
||||
SIDEDATA_SELECT, |
||||
SIDEDATA_DELETE, |
||||
SIDEDATA_NB |
||||
}; |
||||
|
||||
typedef struct SideDataContext { |
||||
const AVClass *class; |
||||
|
||||
int mode; |
||||
enum AVFrameSideDataType type; |
||||
} SideDataContext; |
||||
|
||||
#define OFFSET(x) offsetof(SideDataContext, x) |
||||
#define DEFINE_OPTIONS(filt_name, FLAGS) \ |
||||
static const AVOption filt_name##_options[] = { \
|
||||
{ "mode", "set a mode of operation", OFFSET(mode), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, SIDEDATA_NB-1, FLAGS, "mode" }, \
|
||||
{ "select", "select frame", 0, AV_OPT_TYPE_CONST, {.i64 = SIDEDATA_SELECT }, 0, 0, FLAGS, "mode" }, \
|
||||
{ "delete", "delete side data", 0, AV_OPT_TYPE_CONST, {.i64 = SIDEDATA_DELETE }, 0, 0, FLAGS, "mode" }, \
|
||||
{ "type", "set side data type", OFFSET(type), AV_OPT_TYPE_INT, {.i64 = -1 }, -1, INT_MAX, FLAGS, "type" }, \
|
||||
{ "PANSCAN", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_PANSCAN }, 0, 0, FLAGS, "type" }, \
|
||||
{ "A53_CC", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_A53_CC }, 0, 0, FLAGS, "type" }, \
|
||||
{ "STEREO3D", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_STEREO3D }, 0, 0, FLAGS, "type" }, \
|
||||
{ "MATRIXENCODING", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_MATRIXENCODING }, 0, 0, FLAGS, "type" }, \
|
||||
{ "DOWNMIX_INFO", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_DOWNMIX_INFO }, 0, 0, FLAGS, "type" }, \
|
||||
{ "REPLAYGAIN", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_REPLAYGAIN }, 0, 0, FLAGS, "type" }, \
|
||||
{ "DISPLAYMATRIX", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_DISPLAYMATRIX }, 0, 0, FLAGS, "type" }, \
|
||||
{ "AFD", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_AFD }, 0, 0, FLAGS, "type" }, \
|
||||
{ "MOTION_VECTORS", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_MOTION_VECTORS }, 0, 0, FLAGS, "type" }, \
|
||||
{ "SKIP_SAMPLES", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_SKIP_SAMPLES }, 0, 0, FLAGS, "type" }, \
|
||||
{ "AUDIO_SERVICE_TYPE", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_AUDIO_SERVICE_TYPE }, 0, 0, FLAGS, "type" }, \
|
||||
{ "MASTERING_DISPLAY_METADATA", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_MASTERING_DISPLAY_METADATA }, 0, 0, FLAGS, "type" }, \
|
||||
{ "GOP_TIMECODE", "", 0, AV_OPT_TYPE_CONST, {.i64 = AV_FRAME_DATA_GOP_TIMECODE }, 0, 0, FLAGS, "type" }, \
|
||||
{ NULL } \
|
||||
} |
||||
|
||||
static av_cold int init(AVFilterContext *ctx) |
||||
{ |
||||
SideDataContext *s = ctx->priv; |
||||
|
||||
if (s->type == -1 && s->mode != SIDEDATA_DELETE) { |
||||
av_log(ctx, AV_LOG_ERROR, "Side data type must be set\n"); |
||||
return AVERROR(EINVAL); |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static int filter_frame(AVFilterLink *inlink, AVFrame *frame) |
||||
{ |
||||
AVFilterContext *ctx = inlink->dst; |
||||
AVFilterLink *outlink = ctx->outputs[0]; |
||||
SideDataContext *s = ctx->priv; |
||||
AVFrameSideData *sd = NULL; |
||||
|
||||
if (s->type != -1) |
||||
sd = av_frame_get_side_data(frame, s->type); |
||||
|
||||
switch (s->mode) { |
||||
case SIDEDATA_SELECT: |
||||
if (sd) { |
||||
return ff_filter_frame(outlink, frame); |
||||
} |
||||
break; |
||||
case SIDEDATA_DELETE: |
||||
if (s->type == -1) { |
||||
while (frame->nb_side_data) |
||||
av_frame_remove_side_data(frame, frame->side_data[0]->type); |
||||
} else if (sd) { |
||||
av_frame_remove_side_data(frame, s->type); |
||||
} |
||||
return ff_filter_frame(outlink, frame); |
||||
break; |
||||
default: |
||||
av_assert0(0); |
||||
}; |
||||
|
||||
av_frame_free(&frame); |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
#if CONFIG_ASIDEDATA_FILTER |
||||
|
||||
DEFINE_OPTIONS(asidedata, AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM); |
||||
AVFILTER_DEFINE_CLASS(asidedata); |
||||
|
||||
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_asidedata = { |
||||
.name = "asidedata", |
||||
.description = NULL_IF_CONFIG_SMALL("Manipulate audio frame side data."), |
||||
.priv_size = sizeof(SideDataContext), |
||||
.priv_class = &asidedata_class, |
||||
.init = init, |
||||
.query_formats = ff_query_formats_all, |
||||
.inputs = ainputs, |
||||
.outputs = aoutputs, |
||||
.flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, |
||||
}; |
||||
#endif /* CONFIG_ASIDEDATA_FILTER */ |
||||
|
||||
#if CONFIG_SIDEDATA_FILTER |
||||
|
||||
DEFINE_OPTIONS(sidedata, AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM); |
||||
AVFILTER_DEFINE_CLASS(sidedata); |
||||
|
||||
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_sidedata = { |
||||
.name = "sidedata", |
||||
.description = NULL_IF_CONFIG_SMALL("Manipulate video frame side data."), |
||||
.priv_size = sizeof(SideDataContext), |
||||
.priv_class = &sidedata_class, |
||||
.init = init, |
||||
.inputs = inputs, |
||||
.outputs = outputs, |
||||
.flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, |
||||
}; |
||||
#endif /* CONFIG_SIDEDATA_FILTER */ |
||||
Loading…
Reference in new issue