FFmpeg: libavfilter/vf_geq.c Source File
Go to the documentation of this file. 1 /*
2 * Copyright (C) 2006 Michael Niedermayer <michaelni@gmx.at>
3 * Copyright (C) 2012 Clément Bœsch <u pkh me>
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /**
23 * @file
24 * Generic equation change filter
25 * Originally written by Michael Niedermayer for the MPlayer project, and
26 * ported by Clément Bœsch for FFmpeg.
27 */
28
35
39
40 #define MAX_NB_THREADS 32
42
47 };
48
49 static const char *
const var_names[] = {
"X",
"Y",
"W",
"H",
"N",
"SW",
"SH",
"T",
NULL };
51
55 char *
expr_str[4+3];
///< expression strings for each plane
57 uint8_t *
dst;
///< reference pointer to the 8bits output
58 uint16_t *
dst16;
///< reference pointer to the 16bits output
59 float *
dst32;
///< reference pointer to the 32bits output
66
70
72
73 #define OFFSET(x) offsetof(GEQContext, x)
74 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
75
98 };
99
101
102 static inline double getpix(
void *priv,
double x,
double y,
int plane)
103 {
107 const uint8_t *
src = picref->
data[plane];
108 int linesize = picref->
linesize[plane];
111
113 return 0;
114
116 int xn, yn;
117
121 yn =
FFMIN(yi + 1,
h - 1);
122
124 y -= yi;
125
126 if (geq->
bps > 8 && geq->
bps <= 16) {
127 const uint16_t *src16 = (
const uint16_t*)
src;
128 linesize /= 2;
129
130 return (1-y)*((1-x)*src16[
xi + yi * linesize] + x*src16[xn + yi * linesize])
131 + y *((1-x)*src16[
xi + yn * linesize] + x*src16[xn + yn * linesize]);
132 }
else if (geq->
bps == 32) {
133 const float *src32 = (
const float*)
src;
134 linesize /= 4;
135
136 return (1-y)*((1-x)*src32[
xi + yi * linesize] + x*src32[xn + yi * linesize])
137 + y *((1-x)*src32[
xi + yn * linesize] + x*src32[xn + yn * linesize]);
138 }
else if (geq->
bps == 8) {
139 return (1-y)*((1-x)*
src[
xi + yi * linesize] + x*
src[xn + yi * linesize])
140 + y *((1-x)*
src[
xi + yn * linesize] + x*
src[xn + yn * linesize]);
141 }
142 } else {
145
146 if (geq->
bps > 8 && geq->
bps <= 16) {
147 const uint16_t *src16 = (
const uint16_t*)
src;
148 linesize /= 2;
149
150 return src16[
xi + yi * linesize];
151 }
else if (geq->
bps == 32) {
152 const float *src32 = (
const float*)
src;
153 linesize /= 4;
154
155 return src32[
xi + yi * linesize];
156 }
else if (geq->
bps == 8) {
157 return src[
xi + yi * linesize];
158 }
159 }
160
161 return 0;
162 }
163
165 {
168 const uint8_t *
src = picref->
data[plane];
169 int linesize = picref->
linesize[plane];
170
176 linesize /= 4;
177 else if (geq->
bps > 8 && geq->
bps <= 16)
178 linesize /= 2;
179 for (yi = 0; yi <
h; yi ++) {
180 if (geq->
bps > 8 && geq->
bps <= 16) {
181 const uint16_t *src16 = (
const uint16_t*)
src;
182 double linesum = 0;
183
185 linesum += src16[
xi + yi * linesize];
187 }
188 }
else if (geq->
bps == 8) {
189 double linesum = 0;
190
192 linesum +=
src[
xi + yi * linesize];
194 }
195 }
else if (geq->
bps == 32) {
196 const float *src32 = (
const float*)
src;
197 double linesum = 0;
198
200 linesum += src32[
xi + yi * linesize];
202 }
203 }
204 if (yi)
207 }
208 }
209 return 0;
210 }
211
213 {
217 }
else if (y >
h - 1) {
220 } else if (x < 0) {
221 if (x == -1) return 0;
223 } else if (y < 0) {
224 if (y == -1) return 0;
226 }
227
229 }
230
234 const uint8_t *
src = picref->
data[plane];
237
239 return 0;
240
242 }
243
244 //TODO: cubic interpolate
245 //TODO: keep the last few frames
246 static double lum(
void *priv,
double x,
double y) {
return getpix(priv, x, y, 0); }
247 static double cb(
void *priv,
double x,
double y) {
return getpix(priv, x, y, 1); }
248 static double cr(
void *priv,
double x,
double y) {
return getpix(priv, x, y, 2); }
249 static double alpha(
void *priv,
double x,
double y) {
return getpix(priv, x, y, 3); }
250
255
257 {
260
264 goto end;
265 }
267
271 goto end;
272 }
273
275 /* No chroma at all: fallback on luma */
278 } else {
279 /* One chroma unspecified, fallback on the other */
282 }
283
288 }
295
298 :
301 goto end;
302 }
303
304 for (plane = 0; plane <
NB_PLANES; plane++) {
308 };
309 static const char *const func2_yuv_names[] = {
310 "lum" , "cb" , "cr" , "alpha" , "p",
311 "lumsum", "cbsum", "crsum", "alphasum", "psum",
313 static const char *const func2_rgb_names[] = {
314 "g" , "b" , "r" , "alpha" , "p",
315 "gsum", "bsum", "rsum", "alphasum", "psum",
317 const char *
const *
func2_names = geq->
is_rgb ? func2_rgb_names : func2_yuv_names;
322 int counter[10] = {0};
323
328 goto end;
329 }
330
332 geq->
needs_sum[plane] = counter[5] + counter[6] + counter[7] + counter[8] + counter[9];
333 }
334
335 end:
337 }
338
342 {
363 };
373 };
375
377 }
378
380 {
383
385
388 geq->
bps =
desc->comp[0].depth;
390 return 0;
391 }
392
399
401 {
406 const int plane = td->
plane;
410 int x, y;
411
419
424
425 for (x = 0; x <
width; x++) {
428 }
429 ptr += linesize;
430 }
431 }
else if (geq->
bps <= 16) {
435 for (x = 0; x <
width; x++) {
438 }
439 ptr16 += linesize/2;
440 }
441 } else {
445 for (x = 0; x <
width; x++) {
448 }
449 ptr32 += linesize/4;
450 }
451 }
452
453 return 0;
454 }
455
457 {
458 int plane;
465
468
474 }
476
477 for (plane = 0; plane < geq->
planes &&
out->data[plane]; plane++) {
480 const int linesize =
out->linesize[plane];
482
483 geq->
dst =
out->data[plane];
484 geq->
dst16 = (uint16_t*)
out->data[plane];
485 geq->
dst32 = (
float*)
out->data[plane];
486
491
496
499
502 }
503
506 }
507
509 {
512
518 }
519
521 {
526 },
527 };
528
532 .p.priv_class = &geq_class,
540 };
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
#define AV_PIX_FMT_YUVA422P16
#define AV_PIX_FMT_GBRAP16
AVPixelFormat
Pixel format.
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
static double getpix(void *priv, double x, double y, int plane)
static const char *const var_names[]
static double cb(void *priv, double x, double y)
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
static av_cold void geq_uninit(AVFilterContext *ctx)
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
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
char * av_asprintf(const char *fmt,...)
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
#define AV_PIX_FMT_YUVA422P9
#define FILTER_INPUTS(array)
This structure describes decoded (raw) audio or video data.
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
#define AV_PIX_FMT_YUVA420P16
#define AV_PIX_FMT_YUVA420P10
#define AV_PIX_FMT_YUV420P10
int av_expr_count_func(AVExpr *e, unsigned *counter, int size, int arg)
Track the presence of user provided functions and their number of occurrences in a parsed expression.
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
const char * name
Filter name.
int planes
number of planes
A link between two filters.
#define AV_PIX_FMT_YUVA422P10
static double lumsum(void *priv, double x, double y)
static double crsub(void *priv, double x, double y)
static const char *const func2_names[]
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
double * pixel_sums[NB_PLANES]
Link properties exposed to filter code, but not external callers.
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.
static int geq_config_props(AVFilterLink *inlink)
#define AV_PIX_FMT_YUVA420P9
#define AV_PIX_FMT_GBRP14
static int slice_end(AVCodecContext *avctx, AVFrame *pict, int *got_output)
Handle slice ends.
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
#define AV_PIX_FMT_GBRP10
#define AV_PIX_FMT_YUVA444P16
#define AV_PIX_FMT_YUV422P9
#define AV_PIX_FMT_GRAY16
void av_expr_free(AVExpr *e)
Free a parsed expression previously created with av_expr_parse().
A filter pad used for either input or output.
#define AV_PIX_FMT_YUV444P10
static double(*const func2[])(void *, double, double)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
#define FF_ARRAY_ELEMS(a)
#define AV_PIX_FMT_YUV422P16
static int geq_query_formats(const AVFilterContext *ctx, AVFilterFormatsConfig **cfg_in, AVFilterFormatsConfig **cfg_out)
const AVFilterPad ff_video_default_filterpad[1]
An AVFilterPad array whose only entry has name "default" and is of type AVMEDIA_TYPE_VIDEO.
#define AV_PIX_FMT_GBRAP10
#define AV_PIX_FMT_GBRAP12
@ AV_PIX_FMT_YUVA420P
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
#define AV_PIX_FMT_YUV444P16
#define AV_CEIL_RSHIFT(a, b)
static double av_q2d(AVRational a)
Convert an AVRational to a double.
int vsub
chroma subsampling
#define av_assert0(cond)
assert() equivalent, that is always enabled.
static enum AVPixelFormat pix_fmts[]
#define AV_PIX_FMT_YUV420P9
#define AV_PIX_FMT_YUV420P16
#define xi(width, name, var, range_min, range_max, subs,...)
double av_expr_eval(AVExpr *e, const double *const_values, void *opaque)
Evaluate a previously parsed expression.
static const AVOption geq_options[]
#define AV_PIX_FMT_GRAY14
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
#define FILTER_OUTPUTS(array)
#define AV_PIX_FMT_GRAYF32
static double alpha(void *priv, double x, double y)
#define AV_PIX_FMT_GRAY10
#define AV_PIX_FMT_GBRP16
Describe the class of an AVClass context structure.
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
static double getpix_integrate(void *priv, double x, double y, int plane)
AVFrame * picref
current input buffer
#define AV_PIX_FMT_YUV440P10
static int calculate_sums(GEQContext *geq, int plane, int w, int h)
#define AV_PIX_FMT_YUV422P10
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
static double cbsum(void *priv, double x, double y)
static FilterLink * ff_filter_link(AVFilterLink *link)
AVExpr * e[NB_PLANES][MAX_NB_THREADS]
expressions for each plane and thread
int(* init)(AVBSFContext *ctx)
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
#define AV_PIX_FMT_GBRPF32
#define AV_PIX_FMT_YUV422P12
double values[VAR_VARS_NB]
expression values
#define AV_NOPTS_VALUE
Undefined timestamp value.
#define AV_PIX_FMT_YUV444P12
static av_cold int geq_init(AVFilterContext *ctx)
@ AV_PIX_FMT_YUVA444P
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
#define AV_PIX_FMT_YUVA444P10
static int interpolation(DeclickChannel *c, const double *src, int ar_order, double *acoefficients, int *index, int nb_errors, double *auxiliary, double *interpolated)
#define AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
Some filters support a generic "enable" expression option that can be used to enable or disable a fil...
static void uninit(AVBSFContext *ctx)
#define i(width, name, range_min, range_max)
int w
agreed upon image width
static double lum(void *priv, double x, double y)
#define AV_PIX_FMT_GBRP12
#define av_malloc_array(a, b)
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Used for passing data between threads.
#define FILTER_QUERY_FUNC2(func)
const char * name
Pad name.
char * expr_str[4+3]
expression strings for each plane
uint8_t * dst
reference pointer to the 8bits output
uint16_t * dst16
reference pointer to the 16bits output
#define AV_PIX_FMT_YUV444P9
static int slice_start(SliceContext *sc, VVCContext *s, VVCFrameContext *fc, const CodedBitstreamUnit *unit, const int is_first_slice)
#define AV_PIX_FMT_YUVA444P9
#define AV_PIX_FMT_YUV420P12
#define AV_PIX_FMT_YUV422P14
int h
agreed upon image height
int ff_filter_execute(AVFilterContext *ctx, avfilter_action_func *func, void *arg, int *ret, int nb_jobs)
static int geq_filter_frame(AVFilterLink *inlink, AVFrame *in)
@ AV_OPT_TYPE_INT
Underlying C type is int.
#define AV_PIX_FMT_GBRAPF32
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 values
IDirect3DDxgiInterfaceAccess _COM_Outptr_ void ** p
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
AVFILTER_DEFINE_CLASS(geq)
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
char * av_strdup(const char *s)
Duplicate a string.
static double alphasum(void *priv, double x, double y)
AVFilter p
The public AVFilter.
@ AV_PIX_FMT_YUV422P
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
static enum AVPixelFormat yuv_pix_fmts[]
static double cr(void *priv, double x, double y)
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
int linesize[AV_NUM_DATA_POINTERS]
For video, a positive or negative value, which is typically indicating the size in bytes of each pict...
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
float * dst32
reference pointer to the 32bits output
#define AV_PIX_FMT_YUV444P14
static enum AVPixelFormat rgb_pix_fmts[]
@ AV_OPT_TYPE_STRING
Underlying C type is a uint8_t* that is either NULL or points to a C string allocated with the av_mal...
#define AV_PIX_FMT_GRAY12
@ AV_OPT_TYPE_CONST
Special option type for declaring named constants.
static int slice_geq_filter(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static const AVFilterPad geq_inputs[]
@ AV_PIX_FMT_YUVA422P
planar YUV 4:2:2 24bpp, (1 Cr & Cb sample per 2x1 Y & A samples)
#define AV_PIX_FMT_YUV420P14
static double getpix_integrate_internal(GEQContext *geq, int x, int y, int plane, int w, int h)
Generated on Tue Nov 18 2025 19:23:12 for FFmpeg by
doxygen
1.8.17