FFmpeg
vf_guided.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2021 Xuewei Meng
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 #include "libavutil/imgutils.h"
22 #include "libavutil/mem.h"
23 #include "libavutil/opt.h"
24 #include "libavutil/pixdesc.h"
25 #include "avfilter.h"
26 #include "filters.h"
27 #include "framesync.h"
28 #include "internal.h"
29 #include "video.h"
30 
31 enum FilterModes {
32  BASIC,
33  FAST,
34  NB_MODES,
35 };
36 
37 enum GuidanceModes {
38  OFF,
39  ON,
40  NB_GUIDANCE_MODES,
41 };
42 
43 typedef struct GuidedContext {
44  const AVClass *class;
45  FFFrameSync fs;
46 
47  int radius;
48  float eps;
49  int mode;
50  int sub;
51  int guidance;
52  int planes;
53 
54  int width;
55  int height;
56 
57  int nb_planes;
58  int depth;
59  int planewidth[4];
60  int planeheight[4];
61 
62  float *I;
63  float *II;
64  float *P;
65  float *IP;
66  float *meanI;
67  float *meanII;
68  float *meanP;
69  float *meanIP;
70 
71  float *A;
72  float *B;
73  float *meanA;
74  float *meanB;
75 
76  int (*box_slice)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs);
77 } GuidedContext;
78 
79 #define OFFSET(x) offsetof(GuidedContext, x)
80 #define TFLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
81 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
82 
83 static const AVOption guided_options[] = {
84  { "radius", "set the box radius", OFFSET(radius), AV_OPT_TYPE_INT, {.i64 = 3 }, 1, 20, TFLAGS },
85  { "eps", "set the regularization parameter (with square)", OFFSET(eps), AV_OPT_TYPE_FLOAT, {.dbl = 0.01 }, 0.0, 1, TFLAGS },
86  { "mode", "set filtering mode (0: basic mode; 1: fast mode)", OFFSET(mode), AV_OPT_TYPE_INT, {.i64 = BASIC}, BASIC, NB_MODES - 1, TFLAGS, .unit = "mode" },
87  { "basic", "basic guided filter", 0, AV_OPT_TYPE_CONST, {.i64 = BASIC}, 0, 0, TFLAGS, .unit = "mode" },
88  { "fast", "fast guided filter", 0, AV_OPT_TYPE_CONST, {.i64 = FAST }, 0, 0, TFLAGS, .unit = "mode" },
89  { "sub", "subsampling ratio for fast mode", OFFSET(sub), AV_OPT_TYPE_INT, {.i64 = 4 }, 2, 64, TFLAGS },
90  { "guidance", "set guidance mode (0: off mode; 1: on mode)", OFFSET(guidance), AV_OPT_TYPE_INT, {.i64 = OFF }, OFF, NB_GUIDANCE_MODES - 1, FLAGS, .unit = "guidance" },
91  { "off", "only one input is enabled", 0, AV_OPT_TYPE_CONST, {.i64 = OFF }, 0, 0, FLAGS, .unit = "guidance" },
92  { "on", "two inputs are required", 0, AV_OPT_TYPE_CONST, {.i64 = ON }, 0, 0, FLAGS, .unit = "guidance" },
93  { "planes", "set planes to filter", OFFSET(planes), AV_OPT_TYPE_INT, {.i64 = 1 }, 0, 0xF, TFLAGS },
94  { NULL }
95 };
96 
97 AVFILTER_DEFINE_CLASS(guided);
98 
99 typedef struct ThreadData {
100  int width;
101  int height;
102  float *src;
103  float *dst;
104  int srcStride;
105  int dstStride;
106 } ThreadData;
107 
108 static int box_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
109 {
110  GuidedContext *s = ctx->priv;
111  ThreadData *t = arg;
112 
113  const int width = t->width;
114  const int height = t->height;
115  const int src_stride = t->srcStride;
116  const int dst_stride = t->dstStride;
117  const int slice_start = (height * jobnr) / nb_jobs;
118  const int slice_end = (height * (jobnr + 1)) / nb_jobs;
119  const int radius = s->radius;
120  const float *src = t->src;
121  float *dst = t->dst;
122 
123  int w;
124  int numPix;
125  w = (radius << 1) + 1;
126  numPix = w * w;
127  for (int i = slice_start;i < slice_end;i++) {
128  for (int j = 0;j < width;j++) {
129  float temp = 0.0;
130  for (int row = -radius;row <= radius;row++) {
131  for (int col = -radius;col <= radius;col++) {
132  int x = i + row;
133  int y = j + col;
134  x = (x < 0) ? 0 : (x >= height ? height - 1 : x);
135  y = (y < 0) ? 0 : (y >= width ? width - 1 : y);
136  temp += src[x * src_stride + y];
137  }
138  }
139  dst[i * dst_stride + j] = temp / numPix;
140  }
141  }
142  return 0;
143 }
144 
145 static const enum AVPixelFormat pix_fmts[] = {
146  AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV440P,
147  AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ440P,
148  AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUVA420P, AV_PIX_FMT_YUV420P,
149  AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ420P,
150  AV_PIX_FMT_YUVJ411P, AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV410P,
151  AV_PIX_FMT_YUV420P9, AV_PIX_FMT_YUV422P9, AV_PIX_FMT_YUV444P9,
152  AV_PIX_FMT_YUV420P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV444P10,
153  AV_PIX_FMT_YUV420P12, AV_PIX_FMT_YUV422P12, AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV440P12,
154  AV_PIX_FMT_YUV420P14, AV_PIX_FMT_YUV422P14, AV_PIX_FMT_YUV444P14,
155  AV_PIX_FMT_YUV420P16, AV_PIX_FMT_YUV422P16, AV_PIX_FMT_YUV444P16,
156  AV_PIX_FMT_YUVA420P9, AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA444P9,
157  AV_PIX_FMT_YUVA420P10, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA444P10,
158  AV_PIX_FMT_YUVA420P16, AV_PIX_FMT_YUVA422P16, AV_PIX_FMT_YUVA444P16,
159  AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10,
160  AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRP14, AV_PIX_FMT_GBRP16,
161  AV_PIX_FMT_GBRAP, AV_PIX_FMT_GBRAP10, AV_PIX_FMT_GBRAP12, AV_PIX_FMT_GBRAP16,
162  AV_PIX_FMT_GRAY8, AV_PIX_FMT_GRAY9, AV_PIX_FMT_GRAY10, AV_PIX_FMT_GRAY12, AV_PIX_FMT_GRAY14, AV_PIX_FMT_GRAY16,
163  AV_PIX_FMT_NONE
164 };
165 
166 static int config_input(AVFilterLink *inlink)
167 {
168  AVFilterContext *ctx = inlink->dst;
169  GuidedContext *s = ctx->priv;
170  const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
171 
172  if (s->mode == BASIC) {
173  s->sub = 1;
174  } else if (s->mode == FAST) {
175  if (s->radius >= s->sub)
176  s->radius = s->radius / s->sub;
177  else {
178  s->radius = 1;
179  }
180  }
181 
182  s->depth = desc->comp[0].depth;
183  s->width = ctx->inputs[0]->w;
184  s->height = ctx->inputs[0]->h;
185 
186  s->planewidth[1] = s->planewidth[2] = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
187  s->planewidth[0] = s->planewidth[3] = inlink->w;
188  s->planeheight[1] = s->planeheight[2] = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
189  s->planeheight[0] = s->planeheight[3] = inlink->h;
190 
191  s->nb_planes = av_pix_fmt_count_planes(inlink->format);
192  s->box_slice = box_slice;
193  return 0;
194 }
195 
196 #define GUIDED(type, name) \
197 static int guided_##name(AVFilterContext *ctx, GuidedContext *s, \
198  const uint8_t *ssrc, const uint8_t *ssrcRef, \
199  uint8_t *ddst, int radius, float eps, int width, int height, \
200  int src_stride, int src_ref_stride, int dst_stride, \
201  float maxval) \
202 { \
203  int ret = 0; \
204  type *dst = (type *)ddst; \
205  const type *src = (const type *)ssrc; \
206  const type *srcRef = (const type *)ssrcRef; \
207  \
208  int sub = s->sub; \
209  int h = (height % sub) == 0 ? height / sub : height / sub + 1; \
210  int w = (width % sub) == 0 ? width / sub : width / sub + 1; \
211  \
212  ThreadData t; \
213  const int nb_threads = ff_filter_get_nb_threads(ctx); \
214  float *I = s->I; \
215  float *II = s->II; \
216  float *P = s->P; \
217  float *IP = s->IP; \
218  float *meanI = s->meanI; \
219  float *meanII = s->meanII; \
220  float *meanP = s->meanP; \
221  float *meanIP = s->meanIP; \
222  float *A = s->A; \
223  float *B = s->B; \
224  float *meanA = s->meanA; \
225  float *meanB = s->meanB; \
226  \
227  for (int i = 0;i < h;i++) { \
228  for (int j = 0;j < w;j++) { \
229  int x = i * w + j; \
230  I[x] = src[(i * src_stride + j) * sub] / maxval; \
231  II[x] = I[x] * I[x]; \
232  P[x] = srcRef[(i * src_ref_stride + j) * sub] / maxval; \
233  IP[x] = I[x] * P[x]; \
234  } \
235  } \
236  \
237  t.width = w; \
238  t.height = h; \
239  t.srcStride = w; \
240  t.dstStride = w; \
241  t.src = I; \
242  t.dst = meanI; \
243  ff_filter_execute(ctx, s->box_slice, &t, NULL, FFMIN(h, nb_threads)); \
244  t.src = II; \
245  t.dst = meanII; \
246  ff_filter_execute(ctx, s->box_slice, &t, NULL, FFMIN(h, nb_threads)); \
247  t.src = P; \
248  t.dst = meanP; \
249  ff_filter_execute(ctx, s->box_slice, &t, NULL, FFMIN(h, nb_threads)); \
250  t.src = IP; \
251  t.dst = meanIP; \
252  ff_filter_execute(ctx, s->box_slice, &t, NULL, FFMIN(h, nb_threads)); \
253  \
254  for (int i = 0;i < h;i++) { \
255  for (int j = 0;j < w;j++) { \
256  int x = i * w + j; \
257  float varI = meanII[x] - (meanI[x] * meanI[x]); \
258  float covIP = meanIP[x] - (meanI[x] * meanP[x]); \
259  A[x] = covIP / (varI + eps); \
260  B[x] = meanP[x] - A[x] * meanI[x]; \
261  } \
262  } \
263  \
264  t.src = A; \
265  t.dst = meanA; \
266  ff_filter_execute(ctx, s->box_slice, &t, NULL, FFMIN(h, nb_threads)); \
267  t.src = B; \
268  t.dst = meanB; \
269  ff_filter_execute(ctx, s->box_slice, &t, NULL, FFMIN(h, nb_threads)); \
270  \
271  for (int i = 0;i < height;i++) { \
272  for (int j = 0;j < width;j++) { \
273  int x = i / sub * w + j / sub; \
274  dst[i * dst_stride + j] = meanA[x] * src[i * src_stride + j] + \
275  meanB[x] * maxval; \
276  } \
277  } \
278  \
279  return ret; \
280 }
281 
282 GUIDED(uint8_t, byte)
283 GUIDED(uint16_t, word)
284 
285 static int filter_frame(AVFilterContext *ctx, AVFrame **out, AVFrame *in, AVFrame *ref)
286 {
287  GuidedContext *s = ctx->priv;
288  AVFilterLink *outlink = ctx->outputs[0];
289  *out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
290  if (!*out)
291  return AVERROR(ENOMEM);
292  av_frame_copy_props(*out, in);
293 
294  for (int plane = 0; plane < s->nb_planes; plane++) {
295  if (!(s->planes & (1 << plane))) {
296  av_image_copy_plane((*out)->data[plane], (*out)->linesize[plane],
297  in->data[plane], in->linesize[plane],
298  s->planewidth[plane] * ((s->depth + 7) / 8), s->planeheight[plane]);
299  continue;
300  }
301  if (s->depth <= 8)
302  guided_byte(ctx, s, in->data[plane], ref->data[plane], (*out)->data[plane], s->radius, s->eps,
303  s->planewidth[plane], s->planeheight[plane],
304  in->linesize[plane], ref->linesize[plane], (*out)->linesize[plane], (1 << s->depth) - 1.f);
305  else
306  guided_word(ctx, s, in->data[plane], ref->data[plane], (*out)->data[plane], s->radius, s->eps,
307  s->planewidth[plane], s->planeheight[plane],
308  in->linesize[plane] / 2, ref->linesize[plane] / 2, (*out)->linesize[plane] / 2, (1 << s->depth) - 1.f);
309  }
310 
311  return 0;
312 }
313 
314 static int process_frame(FFFrameSync *fs)
315 {
316  AVFilterContext *ctx = fs->parent;
317  AVFilterLink *outlink = ctx->outputs[0];
318  AVFrame *out_frame = NULL, *main_frame = NULL, *ref_frame = NULL;
319  int ret;
320  ret = ff_framesync_dualinput_get(fs, &main_frame, &ref_frame);
321  if (ret < 0)
322  return ret;
323 
324  if (ctx->is_disabled)
325  return ff_filter_frame(outlink, main_frame);
326 
327  ret = filter_frame(ctx, &out_frame, main_frame, ref_frame);
328  if (ret < 0)
329  return ret;
330  av_frame_free(&main_frame);
331 
332  return ff_filter_frame(outlink, out_frame);
333 }
334 
335 static int config_output(AVFilterLink *outlink)
336 {
337  AVFilterContext *ctx = outlink->src;
338  GuidedContext *s = ctx->priv;
339  AVFilterLink *mainlink = ctx->inputs[0];
340  FFFrameSyncIn *in;
341  int w, h, ret;
342 
343  if (s->guidance == ON) {
344  if (ctx->inputs[0]->w != ctx->inputs[1]->w ||
345  ctx->inputs[0]->h != ctx->inputs[1]->h) {
346  av_log(ctx, AV_LOG_ERROR, "Width and height of input videos must be same.\n");
347  return AVERROR(EINVAL);
348  }
349  }
350 
351  outlink->w = w = mainlink->w;
352  outlink->h = h = mainlink->h;
353  outlink->time_base = mainlink->time_base;
354  outlink->sample_aspect_ratio = mainlink->sample_aspect_ratio;
355  outlink->frame_rate = mainlink->frame_rate;
356 
357  s->I = av_calloc(w * h, sizeof(*s->I));
358  s->II = av_calloc(w * h, sizeof(*s->II));
359  s->P = av_calloc(w * h, sizeof(*s->P));
360  s->IP = av_calloc(w * h, sizeof(*s->IP));
361  s->meanI = av_calloc(w * h, sizeof(*s->meanI));
362  s->meanII = av_calloc(w * h, sizeof(*s->meanII));
363  s->meanP = av_calloc(w * h, sizeof(*s->meanP));
364  s->meanIP = av_calloc(w * h, sizeof(*s->meanIP));
365 
366  s->A = av_calloc(w * h, sizeof(*s->A));
367  s->B = av_calloc(w * h, sizeof(*s->B));
368  s->meanA = av_calloc(w * h, sizeof(*s->meanA));
369  s->meanB = av_calloc(w * h, sizeof(*s->meanA));
370 
371  if (!s->I || !s->II || !s->P || !s->IP || !s->meanI || !s->meanII || !s->meanP ||
372  !s->meanIP || !s->A || !s->B || !s->meanA || !s->meanB)
373  return AVERROR(ENOMEM);
374 
375  if (s->guidance == OFF)
376  return 0;
377 
378  if ((ret = ff_framesync_init(&s->fs, ctx, 2)) < 0)
379  return ret;
380 
381  outlink->time_base = s->fs.time_base;
382 
383  in = s->fs.in;
384  in[0].time_base = mainlink->time_base;
385  in[1].time_base = ctx->inputs[1]->time_base;
386  in[0].sync = 2;
387  in[0].before = EXT_INFINITY;
388  in[0].after = EXT_INFINITY;
389  in[1].sync = 1;
390  in[1].before = EXT_INFINITY;
391  in[1].after = EXT_INFINITY;
392  s->fs.opaque = s;
393  s->fs.on_event = process_frame;
394 
395  return ff_framesync_configure(&s->fs);
396 }
397 
398 static int activate(AVFilterContext *ctx)
399 {
400  GuidedContext *s = ctx->priv;
401  AVFilterLink *outlink = ctx->outputs[0];
402  AVFilterLink *inlink = ctx->inputs[0];
403  AVFrame *frame = NULL;
404  AVFrame *out = NULL;
405  int ret, status;
406  int64_t pts;
407  if (s->guidance)
408  return ff_framesync_activate(&s->fs);
409 
410  FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink);
411 
412  if ((ret = ff_inlink_consume_frame(inlink, &frame)) > 0) {
413  if (ctx->is_disabled)
414  return ff_filter_frame(outlink, frame);
415 
416  ret = filter_frame(ctx, &out, frame, frame);
417  av_frame_free(&frame);
418  if (ret < 0)
419  return ret;
420  ret = ff_filter_frame(outlink, out);
421  }
422  if (ret < 0)
423  return ret;
424  if (ff_inlink_acknowledge_status(inlink, &status, &pts)) {
425  ff_outlink_set_status(outlink, status, pts);
426  return 0;
427  }
428  if (ff_outlink_frame_wanted(outlink))
429  ff_inlink_request_frame(inlink);
430  return 0;
431 }
432 
433 static av_cold int init(AVFilterContext *ctx)
434 {
435  GuidedContext *s = ctx->priv;
436  AVFilterPad pad = { 0 };
437  int ret;
438 
439  pad.type = AVMEDIA_TYPE_VIDEO;
440  pad.name = "source";
441  pad.config_props = config_input;
442 
443  if ((ret = ff_append_inpad(ctx, &pad)) < 0)
444  return ret;
445 
446  if (s->guidance == ON) {
447  pad.type = AVMEDIA_TYPE_VIDEO;
448  pad.name = "guidance";
449  pad.config_props = NULL;
450 
451  if ((ret = ff_append_inpad(ctx, &pad)) < 0)
452  return ret;
453  }
454 
455  return 0;
456 }
457 
458 static av_cold void uninit(AVFilterContext *ctx)
459 {
460  GuidedContext *s = ctx->priv;
461  if (s->guidance == ON)
462  ff_framesync_uninit(&s->fs);
463 
464  av_freep(&s->I);
465  av_freep(&s->II);
466  av_freep(&s->P);
467  av_freep(&s->IP);
468  av_freep(&s->meanI);
469  av_freep(&s->meanII);
470  av_freep(&s->meanP);
471  av_freep(&s->meanIP);
472  av_freep(&s->A);
473  av_freep(&s->B);
474  av_freep(&s->meanA);
475  av_freep(&s->meanB);
476 
477  return;
478 }
479 
480 static const AVFilterPad guided_outputs[] = {
481  {
482  .name = "default",
483  .type = AVMEDIA_TYPE_VIDEO,
484  .config_props = config_output,
485  },
486 };
487 
488 const AVFilter ff_vf_guided = {
489  .name = "guided",
490  .description = NULL_IF_CONFIG_SMALL("Apply Guided filter."),
491  .init = init,
492  .uninit = uninit,
493  .priv_size = sizeof(GuidedContext),
494  .priv_class = &guided_class,
495  .activate = activate,
496  .inputs = NULL,
497  FILTER_OUTPUTS(guided_outputs),
498  FILTER_PIXFMTS_ARRAY(pix_fmts),
499  .flags = AVFILTER_FLAG_DYNAMIC_INPUTS | AVFILTER_FLAG_SLICE_THREADS |
500  AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL,
501  .process_command = ff_filter_process_command,
502 };
ff_get_video_buffer
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
Definition: video.c:112
AV_PIX_FMT_YUVA422P16
#define AV_PIX_FMT_YUVA422P16
Definition: pixfmt.h:522
AV_PIX_FMT_GBRAP16
#define AV_PIX_FMT_GBRAP16
Definition: pixfmt.h:501
FFFrameSyncIn::time_base
AVRational time_base
Time base for the incoming frames.
Definition: framesync.h:117
ff_framesync_configure
int ff_framesync_configure(FFFrameSync *fs)
Configure a frame sync structure.
Definition: framesync.c:134
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:71
config_input
static int config_input(AVFilterLink *inlink)
Definition: vf_guided.c:166
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
GuidedContext::meanI
float * meanI
Definition: vf_guided.c:66
ThreadData::dstStride
int dstStride
Definition: vf_guided.c:105
ff_framesync_uninit
void ff_framesync_uninit(FFFrameSync *fs)
Free all memory currently allocated.
Definition: framesync.c:304
out
FILE * out
Definition: movenc.c:55
filter_frame
static int filter_frame(AVFilterContext *ctx, AVFrame **out, AVFrame *in, AVFrame *ref)
Definition: vf_guided.c:285
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1015
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2965
GuidedContext::meanA
float * meanA
Definition: vf_guided.c:73
GuidedContext::depth
int depth
Definition: vf_guided.c:58
FILTER_PIXFMTS_ARRAY
#define FILTER_PIXFMTS_ARRAY(array)
Definition: internal.h:162
activate
static int activate(AVFilterContext *ctx)
Definition: vf_guided.c:398
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:160
AV_PIX_FMT_YUVA422P9
#define AV_PIX_FMT_YUVA422P9
Definition: pixfmt.h:514
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:374
pixdesc.h
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(guided)
AV_PIX_FMT_YUVA420P16
#define AV_PIX_FMT_YUVA420P16
Definition: pixfmt.h:521
w
uint8_t w
Definition: llviddspenc.c:38
GUIDED
#define GUIDED(type, name)
Definition: vf_guided.c:196
AV_PIX_FMT_YUVA420P10
#define AV_PIX_FMT_YUVA420P10
Definition: pixfmt.h:516
AVOption
AVOption.
Definition: opt.h:346
GuidedContext::meanII
float * meanII
Definition: vf_guided.c:67
AV_PIX_FMT_YUV420P10
#define AV_PIX_FMT_YUV420P10
Definition: pixfmt.h:478
AV_PIX_FMT_YUV440P
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:106
guided_outputs
static const AVFilterPad guided_outputs[]
Definition: vf_guided.c:480
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:170
FFFrameSync
Frame sync structure.
Definition: framesync.h:168
EXT_INFINITY
@ EXT_INFINITY
Extend the frame to infinity.
Definition: framesync.h:75
video.h
AV_PIX_FMT_YUVA422P10
#define AV_PIX_FMT_YUVA422P10
Definition: pixfmt.h:517
FF_FILTER_FORWARD_STATUS_BACK
#define FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink)
Forward the status on an output link to an input link.
Definition: filters.h:199
ThreadData::width
int width
Definition: vf_avgblur.c:65
AV_PIX_FMT_GRAY9
#define AV_PIX_FMT_GRAY9
Definition: pixfmt.h:458
GuidedContext::meanIP
float * meanIP
Definition: vf_guided.c:69
av_image_copy_plane
void av_image_copy_plane(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize, int bytewidth, int height)
Copy image plane from src to dst.
Definition: imgutils.c:374
ff_inlink_consume_frame
int ff_inlink_consume_frame(AVFilterLink *link, AVFrame **rframe)
Take a frame from the link's FIFO and update the link's stats.
Definition: avfilter.c:1442
av_pix_fmt_count_planes
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:3005
AV_PIX_FMT_YUVA420P9
#define AV_PIX_FMT_YUVA420P9
Definition: pixfmt.h:513
guided_options
static const AVOption guided_options[]
Definition: vf_guided.c:83
GuidedContext::IP
float * IP
Definition: vf_guided.c:65
AV_PIX_FMT_GBRP14
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:496
ff_append_inpad
int ff_append_inpad(AVFilterContext *f, AVFilterPad *p)
Append a new input/output pad to the filter's list of such pads.
Definition: avfilter.c:127
ref_frame
static int ref_frame(VVCFrame *dst, const VVCFrame *src)
Definition: dec.c:555
AV_PIX_FMT_GBRAP
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:212
GuidedContext::nb_planes
int nb_planes
Definition: vf_guided.c:57
AV_PIX_FMT_GBRP10
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:494
AV_PIX_FMT_YUVA444P16
#define AV_PIX_FMT_YUVA444P16
Definition: pixfmt.h:523
FFFrameSyncIn
Input stream structure.
Definition: framesync.h:102
AV_PIX_FMT_YUV422P9
#define AV_PIX_FMT_YUV422P9
Definition: pixfmt.h:476
pts
static int64_t pts
Definition: transcode_aac.c:644
AV_PIX_FMT_GRAY16
#define AV_PIX_FMT_GRAY16
Definition: pixfmt.h:462
FFFrameSyncIn::sync
unsigned sync
Synchronization level: frames on input at the highest sync level will generate output frame events.
Definition: framesync.h:160
AVFILTER_FLAG_DYNAMIC_INPUTS
#define AVFILTER_FLAG_DYNAMIC_INPUTS
The number of the filter inputs is not determined just by AVFilter.inputs.
Definition: avfilter.h:106
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:33
AV_PIX_FMT_YUV444P10
#define AV_PIX_FMT_YUV444P10
Definition: pixfmt.h:481
AV_PIX_FMT_YUVJ411P
@ AV_PIX_FMT_YUVJ411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
Definition: pixfmt.h:283
ff_vf_guided
const AVFilter ff_vf_guided
Definition: vf_guided.c:488
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
av_cold
#define av_cold
Definition: attributes.h:90
TFLAGS
#define TFLAGS
Definition: vf_guided.c:80
AV_PIX_FMT_YUV422P16
#define AV_PIX_FMT_YUV422P16
Definition: pixfmt.h:490
AV_PIX_FMT_YUVJ422P
@ AV_PIX_FMT_YUVJ422P
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:86
AV_PIX_FMT_GBRAP10
#define AV_PIX_FMT_GBRAP10
Definition: pixfmt.h:498
ff_outlink_set_status
static void ff_outlink_set_status(AVFilterLink *link, int status, int64_t pts)
Set the status field of a link from the source filter.
Definition: filters.h:189
ff_inlink_request_frame
void ff_inlink_request_frame(AVFilterLink *link)
Mark that a frame is wanted on the link.
Definition: avfilter.c:1568
width
#define width
OFFSET
#define OFFSET(x)
Definition: vf_guided.c:79
FilterModes
FilterModes
Definition: af_adynamicequalizer.c:37
s
#define s(width, name)
Definition: cbs_vp9.c:198
AV_PIX_FMT_GBRAP12
#define AV_PIX_FMT_GBRAP12
Definition: pixfmt.h:499
AV_PIX_FMT_YUVA420P
@ AV_PIX_FMT_YUVA420P
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
Definition: pixfmt.h:108
AV_PIX_FMT_YUV444P16
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:491
AV_CEIL_RSHIFT
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:59
ThreadData::height
int height
Definition: vf_avgblur.c:64
slice_end
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
Definition: mpeg12dec.c:1730
filters.h
AV_PIX_FMT_YUV420P9
#define AV_PIX_FMT_YUV420P9
Definition: pixfmt.h:475
AV_PIX_FMT_YUV420P16
#define AV_PIX_FMT_YUV420P16
Definition: pixfmt.h:489
ctx
AVFormatContext * ctx
Definition: movenc.c:49
AV_PIX_FMT_GRAY14
#define AV_PIX_FMT_GRAY14
Definition: pixfmt.h:461
GuidedContext::II
float * II
Definition: vf_guided.c:63
GuidedContext::planewidth
int planewidth[4]
Definition: vf_guided.c:59
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:73
AV_PIX_FMT_YUVJ444P
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:87
arg
const char * arg
Definition: jacosubdec.c:67
AV_PIX_FMT_GRAY10
#define AV_PIX_FMT_GRAY10
Definition: pixfmt.h:459
ThreadData::dst
AVFrame * dst
Definition: vf_blend.c:57
AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:497
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:66
NULL
#define NULL
Definition: coverity.c:32
av_frame_copy_props
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:709
fs
#define fs(width, name, subs,...)
Definition: cbs_vp9.c:200
ThreadData::src
const uint8_t * src
Definition: vf_bm3d.c:55
BASIC
@ BASIC
Definition: vf_guided.c:32
AV_PIX_FMT_YUVJ420P
@ AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:85
AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_YUV422P10
Definition: pixfmt.h:479
inputs
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 inputs
Definition: filter_design.txt:243
AV_PIX_FMT_GRAY8
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
Definition: pixfmt.h:81
AV_PIX_FMT_GBRP9
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:493
box_slice
static int box_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_guided.c:108
ff_inlink_acknowledge_status
int ff_inlink_acknowledge_status(AVFilterLink *link, int *rstatus, int64_t *rpts)
Test and acknowledge the change of status on the link.
Definition: avfilter.c:1389
GuidedContext::P
float * P
Definition: vf_guided.c:64
GuidedContext::height
int height
Definition: vf_guided.c:55
AVFilterPad::config_props
int(* config_props)(AVFilterLink *link)
Link configuration callback.
Definition: internal.h:113
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:94
ON
@ ON
Definition: vf_guided.c:39
AV_PIX_FMT_YUV422P12
#define AV_PIX_FMT_YUV422P12
Definition: pixfmt.h:483
AV_PIX_FMT_YUV444P12
#define AV_PIX_FMT_YUV444P12
Definition: pixfmt.h:485
OFF
@ OFF
Definition: vf_guided.c:38
ff_filter_process_command
int ff_filter_process_command(AVFilterContext *ctx, const char *cmd, const char *arg, char *res, int res_len, int flags)
Generic processing of user supplied commands that are set in the same way as the filter options.
Definition: avfilter.c:887
NB_MODES
@ NB_MODES
Definition: vf_guided.c:34
height
#define height
AV_PIX_FMT_YUVA444P
@ AV_PIX_FMT_YUVA444P
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
Definition: pixfmt.h:174
GuidedContext::width
int width
Definition: vf_guided.c:54
GuidedContext::mode
int mode
Definition: vf_guided.c:49
init
static av_cold int init(AVFilterContext *ctx)
Definition: vf_guided.c:433
AV_PIX_FMT_YUVA444P10
#define AV_PIX_FMT_YUVA444P10
Definition: pixfmt.h:518
FLAGS
#define FLAGS
Definition: vf_guided.c:81
internal.h
AV_OPT_TYPE_FLOAT
@ AV_OPT_TYPE_FLOAT
Definition: opt.h:238
GuidedContext::eps
float eps
Definition: vf_guided.c:48
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
FAST
@ FAST
Definition: vf_guided.c:33
AV_PIX_FMT_GBRP12
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:495
ThreadData
Used for passing data between threads.
Definition: dsddec.c:71
AV_PIX_FMT_YUVJ440P
@ AV_PIX_FMT_YUVJ440P
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range
Definition: pixfmt.h:107
GuidedContext::B
float * B
Definition: vf_guided.c:72
AVFilterPad::name
const char * name
Pad name.
Definition: internal.h:39
GuidedContext
Definition: vf_guided.c:43
av_calloc
void * av_calloc(size_t nmemb, size_t size)
Definition: mem.c:264
AV_PIX_FMT_YUV444P9
#define AV_PIX_FMT_YUV444P9
Definition: pixfmt.h:477
ThreadData::srcStride
int srcStride
Definition: vf_guided.c:104
AVFilter
Filter definition.
Definition: avfilter.h:166
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: vf_guided.c:145
GuidedContext::sub
int sub
Definition: vf_guided.c:50
ret
ret
Definition: filter_design.txt:187
AVFilterPad::type
enum AVMediaType type
AVFilterPad type.
Definition: internal.h:44
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
AV_PIX_FMT_YUVA444P9
#define AV_PIX_FMT_YUVA444P9
Definition: pixfmt.h:515
ff_framesync_init
int ff_framesync_init(FFFrameSync *fs, AVFilterContext *parent, unsigned nb_in)
Initialize a frame sync structure.
Definition: framesync.c:86
GuidedContext::meanP
float * meanP
Definition: vf_guided.c:68
AV_PIX_FMT_YUV420P12
#define AV_PIX_FMT_YUV420P12
Definition: pixfmt.h:482
GuidedContext::radius
int radius
Definition: vf_guided.c:47
AV_PIX_FMT_YUV422P14
#define AV_PIX_FMT_YUV422P14
Definition: pixfmt.h:487
FFFrameSyncIn::before
enum FFFrameSyncExtMode before
Extrapolation mode for timestamps before the first frame.
Definition: framesync.h:107
GuidedContext::planeheight
int planeheight[4]
Definition: vf_guided.c:60
status
ov_status_e status
Definition: dnn_backend_openvino.c:121
GuidedContext::box_slice
int(* box_slice)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_guided.c:76
framesync.h
mode
mode
Definition: ebur128.h:83
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:72
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:235
GuidedContext::fs
FFFrameSync fs
Definition: vf_guided.c:45
avfilter.h
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_guided.c:458
ref
static int ref[MAX_W *MAX_W]
Definition: jpeg2000dwt.c:112
temp
else temp
Definition: vf_mcdeint.c:263
slice_start
static int slice_start(SliceContext *sc, VVCContext *s, VVCFrameContext *fc, const CodedBitstreamUnit *unit, const int is_first_slice)
Definition: dec.c:688
GuidanceModes
GuidanceModes
Definition: vf_guided.c:37
planes
static const struct @400 planes[]
AV_PIX_FMT_YUV444P
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:78
AVFilterContext
An instance of a filter.
Definition: avfilter.h:407
AV_PIX_FMT_GBRP
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:165
GuidedContext::planes
int planes
Definition: vf_guided.c:52
AVFILTER_FLAG_SLICE_THREADS
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
Definition: avfilter.h:117
desc
const char * desc
Definition: libsvtav1.c:75
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AV_PIX_FMT_YUV422P
@ AV_PIX_FMT_YUV422P
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:77
GuidedContext::guidance
int guidance
Definition: vf_guided.c:51
mem.h
AVPixFmtDescriptor
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:69
NB_GUIDANCE_MODES
@ NB_GUIDANCE_MODES
Definition: vf_guided.c:40
GuidedContext::A
float * A
Definition: vf_guided.c:71
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:183
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:34
src
INIT_CLIP pixel * src
Definition: h264pred_template.c:418
AV_PIX_FMT_YUV411P
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:80
config_output
static int config_output(AVFilterLink *outlink)
Definition: vf_guided.c:335
AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
#define AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
Same as AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, except that the filter will have its filter_frame() c...
Definition: avfilter.h:155
imgutils.h
GuidedContext::meanB
float * meanB
Definition: vf_guided.c:74
AV_PIX_FMT_YUV410P
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
Definition: pixfmt.h:79
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
FFFrameSyncIn::after
enum FFFrameSyncExtMode after
Extrapolation mode for timestamps after the last frame.
Definition: framesync.h:112
AV_PIX_FMT_YUV440P12
#define AV_PIX_FMT_YUV440P12
Definition: pixfmt.h:484
h
h
Definition: vp9dsp_template.c:2038
ff_outlink_frame_wanted
the definition of that something depends on the semantic of the filter The callback must examine the status of the filter s links and proceed accordingly The status of output links is stored in the status_in and status_out fields and tested by the ff_outlink_frame_wanted() function. If this function returns true
AV_PIX_FMT_YUV444P14
#define AV_PIX_FMT_YUV444P14
Definition: pixfmt.h:488
ff_framesync_activate
int ff_framesync_activate(FFFrameSync *fs)
Examine the frames in the filter's input and try to produce output.
Definition: framesync.c:355
ff_framesync_dualinput_get
int ff_framesync_dualinput_get(FFFrameSync *fs, AVFrame **f0, AVFrame **f1)
Definition: framesync.c:393
AV_PIX_FMT_GRAY12
#define AV_PIX_FMT_GRAY12
Definition: pixfmt.h:460
int
int
Definition: ffmpeg_filter.c:424
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Definition: opt.h:244
process_frame
static int process_frame(FFFrameSync *fs)
Definition: vf_guided.c:314
GuidedContext::I
float * I
Definition: vf_guided.c:62
AV_PIX_FMT_YUVA422P
@ AV_PIX_FMT_YUVA422P
planar YUV 4:2:2 24bpp, (1 Cr & Cb sample per 2x1 Y & A samples)
Definition: pixfmt.h:173
AV_PIX_FMT_YUV420P14
#define AV_PIX_FMT_YUV420P14
Definition: pixfmt.h:486
Generated on Fri Apr 19 2024 19:22:29 for FFmpeg by   doxygen 1.8.17