FFmpeg
vf_addroi.c
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #include "libavutil/avassert.h"
20 #include "libavutil/eval.h"
21 #include "libavutil/opt.h"
22 #include "avfilter.h"
23 #include "internal.h"
24 
25 enum {
26  X, Y, W, H,
28 };
29 static const char addroi_param_names[] = {
30  'x', 'y', 'w', 'h',
31 };
32 
33 enum {
37 };
38 static const char *const addroi_var_names[] = {
39  "iw",
40  "ih",
41 };
42 
43 typedef struct AddROIContext {
44  const AVClass *class;
45 
48 
51 
52  int clear;
54 
56 {
57  AVFilterContext *avctx = inlink->dst;
58  AddROIContext *ctx = avctx->priv;
59  int i;
60  double vars[NB_VARS];
61  double val;
62 
63  vars[VAR_IW] = inlink->w;
64  vars[VAR_IH] = inlink->h;
65 
66  for (i = 0; i < NB_PARAMS; i++) {
67  int max_value;
68  switch (i) {
69  case X: max_value = inlink->w; break;
70  case Y: max_value = inlink->h; break;
71  case W: max_value = inlink->w - ctx->region[X]; break;
72  case H: max_value = inlink->h - ctx->region[Y]; break;
73  }
74 
75  val = av_expr_eval(ctx->region_expr[i], vars, NULL);
76  if (val < 0.0) {
77  av_log(avctx, AV_LOG_WARNING, "Calculated value %g for %c is "
78  "less than zero - using zero instead.\n", val,
80  val = 0.0;
81  } else if (val > max_value) {
82  av_log(avctx, AV_LOG_WARNING, "Calculated value %g for %c is "
83  "greater than maximum allowed value %d - "
84  "using %d instead.\n", val, addroi_param_names[i],
85  max_value, max_value);
86  val = max_value;
87  }
88  ctx->region[i] = val;
89  }
90 
91  return 0;
92 }
93 
95 {
96  AVFilterContext *avctx = inlink->dst;
97  AVFilterLink *outlink = avctx->outputs[0];
98  AddROIContext *ctx = avctx->priv;
99  AVRegionOfInterest *roi;
100  AVFrameSideData *sd;
101  int err;
102 
103  if (ctx->clear) {
105  sd = NULL;
106  } else {
108  }
109  if (sd) {
110  const AVRegionOfInterest *old_roi;
111  uint32_t old_roi_size;
112  AVBufferRef *roi_ref;
113  int nb_roi, i;
114 
115  old_roi = (const AVRegionOfInterest*)sd->data;
116  old_roi_size = old_roi->self_size;
117  av_assert0(old_roi_size && sd->size % old_roi_size == 0);
118  nb_roi = sd->size / old_roi_size + 1;
119 
120  roi_ref = av_buffer_alloc(sizeof(*roi) * nb_roi);
121  if (!roi_ref) {
122  err = AVERROR(ENOMEM);
123  goto fail;
124  }
125  roi = (AVRegionOfInterest*)roi_ref->data;
126 
127  for (i = 0; i < nb_roi - 1; i++) {
128  old_roi = (const AVRegionOfInterest*)
129  (sd->data + old_roi_size * i);
130 
131  roi[i] = (AVRegionOfInterest) {
132  .self_size = sizeof(*roi),
133  .top = old_roi->top,
134  .bottom = old_roi->bottom,
135  .left = old_roi->left,
136  .right = old_roi->right,
137  .qoffset = old_roi->qoffset,
138  };
139  }
140 
141  roi[nb_roi - 1] = (AVRegionOfInterest) {
142  .self_size = sizeof(*roi),
143  .top = ctx->region[Y],
144  .bottom = ctx->region[Y] + ctx->region[H],
145  .left = ctx->region[X],
146  .right = ctx->region[X] + ctx->region[W],
147  .qoffset = ctx->qoffset,
148  };
149 
151 
154  roi_ref);
155  if (!sd) {
156  av_buffer_unref(&roi_ref);
157  err = AVERROR(ENOMEM);
158  goto fail;
159  }
160 
161  } else {
163  sizeof(AVRegionOfInterest));
164  if (!sd) {
165  err = AVERROR(ENOMEM);
166  goto fail;
167  }
168  roi = (AVRegionOfInterest*)sd->data;
169  *roi = (AVRegionOfInterest) {
170  .self_size = sizeof(*roi),
171  .top = ctx->region[Y],
172  .bottom = ctx->region[Y] + ctx->region[H],
173  .left = ctx->region[X],
174  .right = ctx->region[X] + ctx->region[W],
175  .qoffset = ctx->qoffset,
176  };
177  }
178 
179  return ff_filter_frame(outlink, frame);
180 
181 fail:
183  return err;
184 }
185 
187 {
188  AddROIContext *ctx = avctx->priv;
189  int i, err;
190 
191  for (i = 0; i < NB_PARAMS; i++) {
192  err = av_expr_parse(&ctx->region_expr[i], ctx->region_str[i],
194  0, avctx);
195  if (err < 0) {
197  "Error parsing %c expression '%s'.\n",
198  addroi_param_names[i], ctx->region_str[i]);
199  return err;
200  }
201  }
202 
203  return 0;
204 }
205 
207 {
208  AddROIContext *ctx = avctx->priv;
209  int i;
210 
211  for (i = 0; i < NB_PARAMS; i++) {
212  av_expr_free(ctx->region_expr[i]);
213  ctx->region_expr[i] = NULL;
214  }
215 }
216 
217 #define OFFSET(x) offsetof(AddROIContext, x)
218 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
219 static const AVOption addroi_options[] = {
220  { "x", "Region distance from left edge of frame.",
221  OFFSET(region_str[X]), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS },
222  { "y", "Region distance from top edge of frame.",
223  OFFSET(region_str[Y]), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS },
224  { "w", "Region width.",
225  OFFSET(region_str[W]), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS },
226  { "h", "Region height.",
227  OFFSET(region_str[H]), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS },
228 
229  { "qoffset", "Quantisation offset to apply in the region.",
230  OFFSET(qoffset), AV_OPT_TYPE_RATIONAL, { .dbl = -0.1 }, -1, +1, FLAGS },
231 
232  { "clear", "Remove any existing regions of interest before adding the new one.",
233  OFFSET(clear), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, FLAGS },
234 
235  { NULL }
236 };
237 
238 AVFILTER_DEFINE_CLASS(addroi);
239 
240 static const AVFilterPad addroi_inputs[] = {
241  {
242  .name = "default",
243  .type = AVMEDIA_TYPE_VIDEO,
244  .config_props = addroi_config_input,
245  .filter_frame = addroi_filter_frame,
246  },
247 };
248 
249 static const AVFilterPad addroi_outputs[] = {
250  {
251  .name = "default",
252  .type = AVMEDIA_TYPE_VIDEO,
253  },
254 };
255 
257  .name = "addroi",
258  .description = NULL_IF_CONFIG_SMALL("Add region of interest to frame."),
259  .init = addroi_init,
260  .uninit = addroi_uninit,
261 
262  .priv_size = sizeof(AddROIContext),
263  .priv_class = &addroi_class,
264 
266 
269 };
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:186
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
FLAGS
#define FLAGS
Definition: vf_addroi.c:218
av_frame_get_side_data
AVFrameSideData * av_frame_get_side_data(const AVFrame *frame, enum AVFrameSideDataType type)
Definition: frame.c:682
av_frame_new_side_data
AVFrameSideData * av_frame_new_side_data(AVFrame *frame, enum AVFrameSideDataType type, size_t size)
Add a new side data to a frame.
Definition: frame.c:670
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:969
AVBufferRef::data
uint8_t * data
The data buffer.
Definition: buffer.h:90
inlink
The exact code depends on how similar the blocks are and how related they are to the and needs to apply these operations to the correct inlink or outlink if there are several Macros are available to factor that when no extra processing is inlink
Definition: filter_design.txt:212
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:99
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:330
AVOption
AVOption.
Definition: opt.h:251
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:165
ff_vf_addroi
const AVFilter ff_vf_addroi
Definition: vf_addroi.c:256
AV_OPT_TYPE_RATIONAL
@ AV_OPT_TYPE_RATIONAL
Definition: opt.h:230
AddROIContext::region_str
char * region_str[NB_PARAMS]
Definition: vf_addroi.c:46
H
@ H
Definition: vf_addroi.c:26
av_expr_parse
int av_expr_parse(AVExpr **expr, const char *s, const char *const *const_names, const char *const *func1_names, double(*const *funcs1)(void *, double), const char *const *func2_names, double(*const *funcs2)(void *, double, double), int log_offset, void *log_ctx)
Parse an expression.
Definition: eval.c:685
AVFilterContext::priv
void * priv
private data for use by the filter
Definition: avfilter.h:407
addroi_outputs
static const AVFilterPad addroi_outputs[]
Definition: vf_addroi.c:249
fail
#define fail()
Definition: checkasm.h:134
AddROIContext::clear
int clear
Definition: vf_addroi.c:52
OFFSET
#define OFFSET(x)
Definition: vf_addroi.c:217
val
static double val(void *priv, double ch)
Definition: aeval.c:77
VAR_IH
@ VAR_IH
Definition: vf_addroi.c:35
av_expr_free
void av_expr_free(AVExpr *e)
Free a parsed expression previously created with av_expr_parse().
Definition: eval.c:336
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:49
avassert.h
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
AVFrameSideData::size
size_t size
Definition: frame.h:239
av_cold
#define av_cold
Definition: attributes.h:90
AVRegionOfInterest
Structure describing a single Region Of Interest.
Definition: frame.h:255
addroi_param_names
static const char addroi_param_names[]
Definition: vf_addroi.c:29
AVRegionOfInterest::bottom
int bottom
Definition: frame.h:271
av_assert0
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
ctx
AVFormatContext * ctx
Definition: movenc.c:48
av_expr_eval
double av_expr_eval(AVExpr *e, const double *const_values, void *opaque)
Evaluate a previously parsed expression.
Definition: eval.c:766
AVExpr
Definition: eval.c:157
addroi_init
static av_cold int addroi_init(AVFilterContext *avctx)
Definition: vf_addroi.c:186
NB_VARS
@ NB_VARS
Definition: vf_addroi.c:36
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: internal.h:194
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:66
NULL
#define NULL
Definition: coverity.c:32
av_buffer_unref
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it.
Definition: buffer.c:139
vars
static const uint8_t vars[2][12]
Definition: camellia.c:183
AVRational
Rational number (pair of numerator and denominator).
Definition: rational.h:58
AVRegionOfInterest::self_size
uint32_t self_size
Must be set to the size of this data structure (that is, sizeof(AVRegionOfInterest)).
Definition: frame.h:260
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(addroi)
av_frame_new_side_data_from_buf
AVFrameSideData * av_frame_new_side_data_from_buf(AVFrame *frame, enum AVFrameSideDataType type, AVBufferRef *buf)
Add a new side data to a frame from an existing AVBufferRef.
Definition: frame.c:638
Y
@ Y
Definition: vf_addroi.c:26
eval.h
addroi_options
static const AVOption addroi_options[]
Definition: vf_addroi.c:219
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:115
for
for(k=2;k<=8;++k)
Definition: h264pred_template.c:425
AddROIContext::region
int region[NB_PARAMS]
Definition: vf_addroi.c:49
AVFrameSideData::data
uint8_t * data
Definition: frame.h:238
av_frame_remove_side_data
void av_frame_remove_side_data(AVFrame *frame, enum AVFrameSideDataType type)
Remove and free all side data instances of the given type.
Definition: frame.c:782
VAR_IW
@ VAR_IW
Definition: vf_addroi.c:34
av_buffer_alloc
AVBufferRef * av_buffer_alloc(size_t size)
Allocate an AVBuffer of the given size using av_malloc().
Definition: buffer.c:77
addroi_config_input
static int addroi_config_input(AVFilterLink *inlink)
Definition: vf_addroi.c:55
AVRegionOfInterest::right
int right
Definition: frame.h:273
internal.h
AVRegionOfInterest::left
int left
Definition: frame.h:272
addroi_inputs
static const AVFilterPad addroi_inputs[]
Definition: vf_addroi.c:240
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:269
AVRegionOfInterest::top
int top
Distance in pixels from the top edge of the frame to the top and bottom edges and from the left edge ...
Definition: frame.h:270
addroi_filter_frame
static int addroi_filter_frame(AVFilterLink *inlink, AVFrame *frame)
Definition: vf_addroi.c:94
AVFilterPad::name
const char * name
Pad name.
Definition: internal.h:55
AVFilter
Filter definition.
Definition: avfilter.h:161
frame
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return or at least make progress towards producing a frame
Definition: filter_design.txt:264
AddROIContext
Definition: vf_addroi.c:43
addroi_uninit
static av_cold void addroi_uninit(AVFilterContext *avctx)
Definition: vf_addroi.c:206
avfilter.h
AVFILTER_FLAG_METADATA_ONLY
#define AVFILTER_FLAG_METADATA_ONLY
The filter is a "metadata" filter - it does not modify the frame data in any way.
Definition: avfilter.h:133
W
@ W
Definition: vf_addroi.c:26
AddROIContext::region_expr
AVExpr * region_expr[NB_PARAMS]
Definition: vf_addroi.c:47
AVFilterContext
An instance of a filter.
Definition: avfilter.h:392
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AVBufferRef
A reference to a data buffer.
Definition: buffer.h:82
AVFrameSideData
Structure to hold side data for an AVFrame.
Definition: frame.h:236
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Definition: opt.h:244
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:195
AV_FRAME_DATA_REGIONS_OF_INTEREST
@ AV_FRAME_DATA_REGIONS_OF_INTEREST
Regions Of Interest, the data is an array of AVRegionOfInterest type, the number of array element is ...
Definition: frame.h:165
flags
#define flags(name, subs,...)
Definition: cbs_av1.c:561
X
@ X
Definition: vf_addroi.c:26
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
AV_OPT_TYPE_STRING
@ AV_OPT_TYPE_STRING
Definition: opt.h:229
AddROIContext::qoffset
AVRational qoffset
Definition: vf_addroi.c:50
NB_PARAMS
@ NB_PARAMS
Definition: vf_addroi.c:27
AVRegionOfInterest::qoffset
AVRational qoffset
Quantisation offset.
Definition: frame.h:297
addroi_var_names
static const char *const addroi_var_names[]
Definition: vf_addroi.c:38
AVFilterContext::outputs
AVFilterLink ** outputs
array of pointers to output links
Definition: avfilter.h:404