FFmpeg  4.4.8
vf_lut.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2011 Stefano Sabatini
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 /**
22  * @file
23  * Compute a look-up table for binding the input value to the output
24  * value, and apply it to input video.
25  */
26 
27 #include "libavutil/attributes.h"
28 #include "libavutil/bswap.h"
29 #include "libavutil/common.h"
30 #include "libavutil/eval.h"
31 #include "libavutil/opt.h"
32 #include "libavutil/pixdesc.h"
33 #include "avfilter.h"
34 #include "filters.h"
35 #include "drawutils.h"
36 #include "formats.h"
37 #include "internal.h"
38 #include "video.h"
39 
40 static const char *const var_names[] = {
41  "w", ///< width of the input video
42  "h", ///< height of the input video
43  "val", ///< input value for the pixel
44  "maxval", ///< max value for the pixel
45  "minval", ///< min value for the pixel
46  "negval", ///< negated value
47  "clipval",
48  NULL
49 };
50 
51 enum var_name {
60 };
61 
62 typedef struct LutContext {
63  const AVClass *class;
64  uint16_t lut[4][256 * 256]; ///< lookup table for each component
65  char *comp_expr_str[4];
67  int hsub, vsub;
69  int is_rgb, is_yuv;
70  int is_planar;
71  int is_16bit;
72  int step;
73  int negate_alpha; /* only used by negate */
74 } LutContext;
75 
76 #define Y 0
77 #define U 1
78 #define V 2
79 #define R 0
80 #define G 1
81 #define B 2
82 #define A 3
83 
84 #define OFFSET(x) offsetof(LutContext, x)
85 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
86 
87 static const AVOption options[] = {
88  { "c0", "set component #0 expression", OFFSET(comp_expr_str[0]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
89  { "c1", "set component #1 expression", OFFSET(comp_expr_str[1]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
90  { "c2", "set component #2 expression", OFFSET(comp_expr_str[2]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
91  { "c3", "set component #3 expression", OFFSET(comp_expr_str[3]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
92  { "y", "set Y expression", OFFSET(comp_expr_str[Y]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
93  { "u", "set U expression", OFFSET(comp_expr_str[U]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
94  { "v", "set V expression", OFFSET(comp_expr_str[V]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
95  { "r", "set R expression", OFFSET(comp_expr_str[R]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
96  { "g", "set G expression", OFFSET(comp_expr_str[G]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
97  { "b", "set B expression", OFFSET(comp_expr_str[B]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
98  { "a", "set A expression", OFFSET(comp_expr_str[A]), AV_OPT_TYPE_STRING, { .str = "clipval" }, .flags = FLAGS },
99  { NULL }
100 };
101 
103 {
104  LutContext *s = ctx->priv;
105  int i;
106 
107  for (i = 0; i < 4; i++) {
108  av_expr_free(s->comp_expr[i]);
109  s->comp_expr[i] = NULL;
110  av_freep(&s->comp_expr_str[i]);
111  }
112 }
113 
114 #define YUV_FORMATS \
115  AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV420P, \
116  AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV440P, \
117  AV_PIX_FMT_YUVA420P, AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUVA444P, \
118  AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ420P, \
119  AV_PIX_FMT_YUVJ440P, \
120  AV_PIX_FMT_YUV444P9LE, AV_PIX_FMT_YUV422P9LE, AV_PIX_FMT_YUV420P9LE, \
121  AV_PIX_FMT_YUV444P10LE, AV_PIX_FMT_YUV422P10LE, AV_PIX_FMT_YUV420P10LE, AV_PIX_FMT_YUV440P10LE, \
122  AV_PIX_FMT_YUV444P12LE, AV_PIX_FMT_YUV422P12LE, AV_PIX_FMT_YUV420P12LE, AV_PIX_FMT_YUV440P12LE, \
123  AV_PIX_FMT_YUV444P14LE, AV_PIX_FMT_YUV422P14LE, AV_PIX_FMT_YUV420P14LE, \
124  AV_PIX_FMT_YUV444P16LE, AV_PIX_FMT_YUV422P16LE, AV_PIX_FMT_YUV420P16LE, \
125  AV_PIX_FMT_YUVA444P16LE, AV_PIX_FMT_YUVA422P16LE, AV_PIX_FMT_YUVA420P16LE
126 
127 #define RGB_FORMATS \
128  AV_PIX_FMT_ARGB, AV_PIX_FMT_RGBA, \
129  AV_PIX_FMT_ABGR, AV_PIX_FMT_BGRA, \
130  AV_PIX_FMT_RGB24, AV_PIX_FMT_BGR24, \
131  AV_PIX_FMT_RGB48LE, AV_PIX_FMT_RGBA64LE, \
132  AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRAP, \
133  AV_PIX_FMT_GBRP9LE, AV_PIX_FMT_GBRP10LE, \
134  AV_PIX_FMT_GBRAP10LE, \
135  AV_PIX_FMT_GBRP12LE, AV_PIX_FMT_GBRP14LE, \
136  AV_PIX_FMT_GBRP16LE, AV_PIX_FMT_GBRAP12LE, \
137  AV_PIX_FMT_GBRAP16LE
138 
139 #define GRAY_FORMATS \
140  AV_PIX_FMT_GRAY8, AV_PIX_FMT_GRAY9LE, AV_PIX_FMT_GRAY10LE, \
141  AV_PIX_FMT_GRAY12LE, AV_PIX_FMT_GRAY14LE, AV_PIX_FMT_GRAY16LE
142 
143 static const enum AVPixelFormat yuv_pix_fmts[] = { YUV_FORMATS, AV_PIX_FMT_NONE };
144 static const enum AVPixelFormat rgb_pix_fmts[] = { RGB_FORMATS, AV_PIX_FMT_NONE };
146 
148 {
149  LutContext *s = ctx->priv;
150 
151  const enum AVPixelFormat *pix_fmts = s->is_rgb ? rgb_pix_fmts :
152  s->is_yuv ? yuv_pix_fmts :
153  all_pix_fmts;
155  if (!fmts_list)
156  return AVERROR(ENOMEM);
157  return ff_set_common_formats(ctx, fmts_list);
158 }
159 
160 /**
161  * Clip value val in the minval - maxval range.
162  */
163 static double clip(void *opaque, double val)
164 {
165  LutContext *s = opaque;
166  double minval = s->var_values[VAR_MINVAL];
167  double maxval = s->var_values[VAR_MAXVAL];
168 
169  return av_clip(val, minval, maxval);
170 }
171 
172 /**
173  * Compute gamma correction for value val, assuming the minval-maxval
174  * range, val is clipped to a value contained in the same interval.
175  */
176 static double compute_gammaval(void *opaque, double gamma)
177 {
178  LutContext *s = opaque;
179  double val = s->var_values[VAR_CLIPVAL];
180  double minval = s->var_values[VAR_MINVAL];
181  double maxval = s->var_values[VAR_MAXVAL];
182 
183  return pow((val-minval)/(maxval-minval), gamma) * (maxval-minval)+minval;
184 }
185 
186 /**
187  * Compute ITU Rec.709 gamma correction of value val.
188  */
189 static double compute_gammaval709(void *opaque, double gamma)
190 {
191  LutContext *s = opaque;
192  double val = s->var_values[VAR_CLIPVAL];
193  double minval = s->var_values[VAR_MINVAL];
194  double maxval = s->var_values[VAR_MAXVAL];
195  double level = (val - minval) / (maxval - minval);
196  level = level < 0.018 ? 4.5 * level
197  : 1.099 * pow(level, 1.0 / gamma) - 0.099;
198  return level * (maxval - minval) + minval;
199 }
200 
201 static double (* const funcs1[])(void *, double) = {
202  clip,
205  NULL
206 };
207 
208 static const char * const funcs1_names[] = {
209  "clip",
210  "gammaval",
211  "gammaval709",
212  NULL
213 };
214 
215 static int config_props(AVFilterLink *inlink)
216 {
217  AVFilterContext *ctx = inlink->dst;
218  LutContext *s = ctx->priv;
220  uint8_t rgba_map[4]; /* component index -> RGBA color index map */
221  int min[4], max[4];
222  int val, color, ret;
223 
224  s->hsub = desc->log2_chroma_w;
225  s->vsub = desc->log2_chroma_h;
226 
227  s->var_values[VAR_W] = inlink->w;
228  s->var_values[VAR_H] = inlink->h;
229  s->is_16bit = desc->comp[0].depth > 8;
230 
231  switch (inlink->format) {
232  case AV_PIX_FMT_YUV410P:
233  case AV_PIX_FMT_YUV411P:
234  case AV_PIX_FMT_YUV420P:
235  case AV_PIX_FMT_YUV422P:
236  case AV_PIX_FMT_YUV440P:
237  case AV_PIX_FMT_YUV444P:
238  case AV_PIX_FMT_YUVA420P:
239  case AV_PIX_FMT_YUVA422P:
240  case AV_PIX_FMT_YUVA444P:
267  min[Y] = 16 * (1 << (desc->comp[0].depth - 8));
268  min[U] = 16 * (1 << (desc->comp[1].depth - 8));
269  min[V] = 16 * (1 << (desc->comp[2].depth - 8));
270  min[A] = 0;
271  max[Y] = 235 * (1 << (desc->comp[0].depth - 8));
272  max[U] = 240 * (1 << (desc->comp[1].depth - 8));
273  max[V] = 240 * (1 << (desc->comp[2].depth - 8));
274  max[A] = (1 << desc->comp[0].depth) - 1;
275  break;
276  case AV_PIX_FMT_RGB48LE:
277  case AV_PIX_FMT_RGBA64LE:
278  min[0] = min[1] = min[2] = min[3] = 0;
279  max[0] = max[1] = max[2] = max[3] = 65535;
280  break;
281  default:
282  min[0] = min[1] = min[2] = min[3] = 0;
283  max[0] = max[1] = max[2] = max[3] = 255 * (1 << (desc->comp[0].depth - 8));
284  }
285 
286  s->is_yuv = s->is_rgb = 0;
287  s->is_planar = desc->flags & AV_PIX_FMT_FLAG_PLANAR;
288  if (ff_fmt_is_in(inlink->format, yuv_pix_fmts)) s->is_yuv = 1;
289  else if (ff_fmt_is_in(inlink->format, rgb_pix_fmts)) s->is_rgb = 1;
290 
291  if (s->is_rgb) {
292  ff_fill_rgba_map(rgba_map, inlink->format);
293  s->step = av_get_bits_per_pixel(desc) >> 3;
294  if (s->is_16bit) {
295  s->step = s->step >> 1;
296  }
297  }
298 
299  for (color = 0; color < desc->nb_components; color++) {
300  double res;
301  int comp = s->is_rgb ? rgba_map[color] : color;
302 
303  /* create the parsed expression */
304  av_expr_free(s->comp_expr[color]);
305  s->comp_expr[color] = NULL;
306  ret = av_expr_parse(&s->comp_expr[color], s->comp_expr_str[color],
308  if (ret < 0) {
310  "Error when parsing the expression '%s' for the component %d and color %d.\n",
311  s->comp_expr_str[comp], comp, color);
312  return AVERROR(EINVAL);
313  }
314 
315  /* compute the lut */
316  s->var_values[VAR_MAXVAL] = max[color];
317  s->var_values[VAR_MINVAL] = min[color];
318 
319  for (val = 0; val < FF_ARRAY_ELEMS(s->lut[comp]); val++) {
320  s->var_values[VAR_VAL] = val;
321  s->var_values[VAR_CLIPVAL] = av_clip(val, min[color], max[color]);
322  s->var_values[VAR_NEGVAL] =
323  av_clip(min[color] + max[color] - s->var_values[VAR_VAL],
324  min[color], max[color]);
325 
326  res = av_expr_eval(s->comp_expr[color], s->var_values, s);
327  if (isnan(res)) {
329  "Error when evaluating the expression '%s' for the value %d for the component %d.\n",
330  s->comp_expr_str[color], val, comp);
331  return AVERROR(EINVAL);
332  }
333  s->lut[comp][val] = av_clip((int)res, 0, max[A]);
334  av_log(ctx, AV_LOG_DEBUG, "val[%d][%d] = %d\n", comp, val, s->lut[comp][val]);
335  }
336  }
337 
338  return 0;
339 }
340 
341 struct thread_data {
344 
345  int w;
346  int h;
347 };
348 
349 #define LOAD_PACKED_COMMON\
350  LutContext *s = ctx->priv;\
351  const struct thread_data *td = arg;\
352 \
353  int i, j;\
354  const int w = td->w;\
355  const int h = td->h;\
356  AVFrame *in = td->in;\
357  AVFrame *out = td->out;\
358  const uint16_t (*tab)[256*256] = (const uint16_t (*)[256*256])s->lut;\
359  const int step = s->step;\
360 \
361  const int slice_start = ff_slice_pos(h, jobnr, nb_jobs); \
362  const int slice_end = ff_slice_pos(h, jobnr + 1, nb_jobs); \
363 
364 /* packed, 16-bit */
365 static int lut_packed_16bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
366 {
368 
369  uint16_t *inrow, *outrow, *inrow0, *outrow0;
370  const int in_linesize = in->linesize[0] / 2;
371  const int out_linesize = out->linesize[0] / 2;
372  inrow0 = (uint16_t *)in ->data[0];
373  outrow0 = (uint16_t *)out->data[0];
374 
375  for (i = slice_start; i < slice_end; i++) {
376  inrow = inrow0 + i * in_linesize;
377  outrow = outrow0 + i * out_linesize;
378  for (j = 0; j < w; j++) {
379 
380  switch (step) {
381 #if HAVE_BIGENDIAN
382  case 4: outrow[3] = av_bswap16(tab[3][av_bswap16(inrow[3])]); // Fall-through
383  case 3: outrow[2] = av_bswap16(tab[2][av_bswap16(inrow[2])]); // Fall-through
384  case 2: outrow[1] = av_bswap16(tab[1][av_bswap16(inrow[1])]); // Fall-through
385  default: outrow[0] = av_bswap16(tab[0][av_bswap16(inrow[0])]);
386 #else
387  case 4: outrow[3] = tab[3][inrow[3]]; // Fall-through
388  case 3: outrow[2] = tab[2][inrow[2]]; // Fall-through
389  case 2: outrow[1] = tab[1][inrow[1]]; // Fall-through
390  default: outrow[0] = tab[0][inrow[0]];
391 #endif
392  }
393  outrow += step;
394  inrow += step;
395  }
396  }
397 
398  return 0;
399 }
400 
401 /* packed, 8-bit */
402 static int lut_packed_8bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
403 {
405 
406  uint8_t *inrow, *outrow, *inrow0, *outrow0;
407  const int in_linesize = in->linesize[0];
408  const int out_linesize = out->linesize[0];
409  inrow0 = in ->data[0];
410  outrow0 = out->data[0];
411 
412  for (i = slice_start; i < slice_end; i++) {
413  inrow = inrow0 + i * in_linesize;
414  outrow = outrow0 + i * out_linesize;
415  for (j = 0; j < w; j++) {
416  switch (step) {
417  case 4: outrow[3] = tab[3][inrow[3]]; // Fall-through
418  case 3: outrow[2] = tab[2][inrow[2]]; // Fall-through
419  case 2: outrow[1] = tab[1][inrow[1]]; // Fall-through
420  default: outrow[0] = tab[0][inrow[0]];
421  }
422  outrow += step;
423  inrow += step;
424  }
425  }
426 
427  return 0;
428 }
429 
430 #define LOAD_PLANAR_COMMON\
431  LutContext *s = ctx->priv;\
432  const struct thread_data *td = arg;\
433  int i, j, plane;\
434  AVFrame *in = td->in;\
435  AVFrame *out = td->out;\
436 
437 #define PLANAR_COMMON\
438  int vsub = plane == 1 || plane == 2 ? s->vsub : 0;\
439  int hsub = plane == 1 || plane == 2 ? s->hsub : 0;\
440  int h = AV_CEIL_RSHIFT(td->h, vsub);\
441  int w = AV_CEIL_RSHIFT(td->w, hsub);\
442  const uint16_t *tab = s->lut[plane];\
443 \
444  const int slice_start = ff_slice_pos(h, jobnr, nb_jobs); \
445  const int slice_end = ff_slice_pos(h, jobnr + 1, nb_jobs); \
446 
447 /* planar >8 bit depth */
448 static int lut_planar_16bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
449 {
451 
452  uint16_t *inrow, *outrow;
453 
454  for (plane = 0; plane < 4 && in->data[plane] && in->linesize[plane]; plane++) {
456 
457  const int in_linesize = in->linesize[plane] / 2;
458  const int out_linesize = out->linesize[plane] / 2;
459 
460  inrow = (uint16_t *)in ->data[plane] + slice_start * in_linesize;
461  outrow = (uint16_t *)out->data[plane] + slice_start * out_linesize;
462 
463  for (i = slice_start; i < slice_end; i++) {
464  for (j = 0; j < w; j++) {
465 #if HAVE_BIGENDIAN
466  outrow[j] = av_bswap16(tab[av_bswap16(inrow[j])]);
467 #else
468  outrow[j] = tab[inrow[j]];
469 #endif
470  }
471  inrow += in_linesize;
472  outrow += out_linesize;
473  }
474  }
475 
476  return 0;
477 }
478 
479 /* planar 8bit depth */
480 static int lut_planar_8bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
481 {
483 
484  uint8_t *inrow, *outrow;
485 
486  for (plane = 0; plane < 4 && in->data[plane] && in->linesize[plane]; plane++) {
488 
489  const int in_linesize = in->linesize[plane];
490  const int out_linesize = out->linesize[plane];
491 
492  inrow = in ->data[plane] + slice_start * in_linesize;
493  outrow = out->data[plane] + slice_start * out_linesize;
494 
495  for (i = slice_start; i < slice_end; i++) {
496  for (j = 0; j < w; j++)
497  outrow[j] = tab[inrow[j]];
498  inrow += in_linesize;
499  outrow += out_linesize;
500  }
501  }
502 
503  return 0;
504 }
505 
506 #define PACKED_THREAD_DATA\
507  struct thread_data td = {\
508  .in = in,\
509  .out = out,\
510  .w = inlink->w,\
511  .h = in->height,\
512  };\
513 
514 #define PLANAR_THREAD_DATA\
515  struct thread_data td = {\
516  .in = in,\
517  .out = out,\
518  .w = inlink->w,\
519  .h = inlink->h,\
520  };\
521 
522 static int filter_frame(AVFilterLink *inlink, AVFrame *in)
523 {
524  AVFilterContext *ctx = inlink->dst;
525  LutContext *s = ctx->priv;
526  AVFilterLink *outlink = ctx->outputs[0];
527  AVFrame *out;
528  int direct = 0;
529 
530  if (av_frame_is_writable(in)) {
531  direct = 1;
532  out = in;
533  } else {
534  out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
535  if (!out) {
536  av_frame_free(&in);
537  return AVERROR(ENOMEM);
538  }
540  }
541 
542  if (s->is_rgb && s->is_16bit && !s->is_planar) {
543  /* packed, 16-bit */
545  ctx->internal->execute(ctx, lut_packed_16bits, &td, NULL,
546  FFMIN(in->height, ff_filter_get_nb_threads(ctx)));
547  } else if (s->is_rgb && !s->is_planar) {
548  /* packed 8 bits */
550  ctx->internal->execute(ctx, lut_packed_8bits, &td, NULL,
551  FFMIN(in->height, ff_filter_get_nb_threads(ctx)));
552  } else if (s->is_16bit) {
553  /* planar >8 bit depth */
555  ctx->internal->execute(ctx, lut_planar_16bits, &td, NULL,
556  FFMIN(in->height, ff_filter_get_nb_threads(ctx)));
557  } else {
558  /* planar 8bit depth */
560  ctx->internal->execute(ctx, lut_planar_8bits, &td, NULL,
561  FFMIN(in->height, ff_filter_get_nb_threads(ctx)));
562  }
563 
564  if (!direct)
565  av_frame_free(&in);
566 
567  return ff_filter_frame(outlink, out);
568 }
569 
570 static int process_command(AVFilterContext *ctx, const char *cmd, const char *args,
571  char *res, int res_len, int flags)
572 {
573  int ret = ff_filter_process_command(ctx, cmd, args, res, res_len, flags);
574 
575  if (ret < 0)
576  return ret;
577 
578  return config_props(ctx->inputs[0]);
579 }
580 
581 static const AVFilterPad inputs[] = {
582  { .name = "default",
583  .type = AVMEDIA_TYPE_VIDEO,
584  .filter_frame = filter_frame,
585  .config_props = config_props,
586  },
587  { NULL }
588 };
589 static const AVFilterPad outputs[] = {
590  { .name = "default",
591  .type = AVMEDIA_TYPE_VIDEO,
592  },
593  { NULL }
594 };
595 
596 #define DEFINE_LUT_FILTER(name_, description_) \
597  AVFilter ff_vf_##name_ = { \
598  .name = #name_, \
599  .description = NULL_IF_CONFIG_SMALL(description_), \
600  .priv_size = sizeof(LutContext), \
601  .priv_class = &name_ ## _class, \
602  .init = name_##_init, \
603  .uninit = uninit, \
604  .query_formats = query_formats, \
605  .inputs = inputs, \
606  .outputs = outputs, \
607  .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC | \
608  AVFILTER_FLAG_SLICE_THREADS, \
609  .process_command = process_command, \
610  }
611 
612 #if CONFIG_LUT_FILTER
613 
614 #define lut_options options
616 
617 static int lut_init(AVFilterContext *ctx)
618 {
619  return 0;
620 }
621 
622 DEFINE_LUT_FILTER(lut, "Compute and apply a lookup table to the RGB/YUV input video.");
623 #endif
624 
625 #if CONFIG_LUTYUV_FILTER
626 
627 #define lutyuv_options options
628 AVFILTER_DEFINE_CLASS(lutyuv);
629 
630 static av_cold int lutyuv_init(AVFilterContext *ctx)
631 {
632  LutContext *s = ctx->priv;
633 
634  s->is_yuv = 1;
635 
636  return 0;
637 }
638 
639 DEFINE_LUT_FILTER(lutyuv, "Compute and apply a lookup table to the YUV input video.");
640 #endif
641 
642 #if CONFIG_LUTRGB_FILTER
643 
644 #define lutrgb_options options
645 AVFILTER_DEFINE_CLASS(lutrgb);
646 
647 static av_cold int lutrgb_init(AVFilterContext *ctx)
648 {
649  LutContext *s = ctx->priv;
650 
651  s->is_rgb = 1;
652 
653  return 0;
654 }
655 
656 DEFINE_LUT_FILTER(lutrgb, "Compute and apply a lookup table to the RGB input video.");
657 #endif
658 
659 #if CONFIG_NEGATE_FILTER
660 
661 static const AVOption negate_options[] = {
662  { "negate_alpha", NULL, OFFSET(negate_alpha), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, FLAGS },
663  { NULL }
664 };
665 
666 AVFILTER_DEFINE_CLASS(negate);
667 
668 static av_cold int negate_init(AVFilterContext *ctx)
669 {
670  LutContext *s = ctx->priv;
671 
672  for (int i = 0; i < 4; i++) {
673  s->comp_expr_str[i] = av_strdup((i == 3 && !s->negate_alpha) ?
674  "val" : "negval");
675  if (!s->comp_expr_str[i])
676  return AVERROR(ENOMEM);
677  }
678 
679  return 0;
680 }
681 
682 DEFINE_LUT_FILTER(negate, "Negate input video.");
683 
684 #endif
static double val(void *priv, double ch)
Definition: aeval.c:76
Macro definitions for various function/variable attributes.
#define av_cold
Definition: attributes.h:88
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
uint8_t
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1096
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:882
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Definition: avfilter.c:802
Main libavfilter public API header.
byte swapping routines
#define flags(name, subs,...)
Definition: cbs_av1.c:572
#define s(width, name)
Definition: cbs_vp9.c:257
common internal and external API header
#define FFMIN(a, b)
Definition: common.h:105
#define av_clip
Definition: common.h:122
#define NULL
Definition: coverity.c:32
#define max(a, b)
Definition: cuda_runtime.h:33
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
Definition: drawutils.c:35
misc drawing utilities
static void comp(unsigned char *dst, ptrdiff_t dst_stride, unsigned char *src, ptrdiff_t src_stride, int add)
Definition: eamad.c:85
void av_expr_free(AVExpr *e)
Free a parsed expression previously created with av_expr_parse().
Definition: eval.c:339
double av_expr_eval(AVExpr *e, const double *const_values, void *opaque)
Evaluate a previously parsed expression.
Definition: eval.c:781
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:700
simple arithmetic expression evaluator
int ff_fmt_is_in(int fmt, const int *fmts)
Tell if an integer is contained in the provided -1-terminated list of integers.
Definition: formats.c:257
int ff_set_common_formats(AVFilterContext *ctx, AVFilterFormats *formats)
A helper for query_formats() which sets all links to the same list of formats.
Definition: formats.c:587
AVFilterFormats * ff_make_format_list(const int *fmts)
Create a list of supported formats.
Definition: formats.c:286
@ AV_OPT_TYPE_BOOL
Definition: opt.h:242
@ AV_OPT_TYPE_STRING
Definition: opt.h:229
#define AVERROR(e)
Definition: error.h:43
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
Definition: frame.c:594
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:203
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:658
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:215
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
char * av_strdup(const char *s)
Duplicate a string.
Definition: mem.c:253
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
for(j=16;j >0;--j)
int i
Definition: input.c:407
const char * arg
Definition: jacosubdec.c:66
static void direct(const float *in, const FFTComplex *ir, int len, float *out)
Definition: af_afir.c:60
#define AVFILTER_DEFINE_CLASS(fname)
Definition: internal.h:288
common internal API header
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:309
#define isnan(x)
Definition: libm.h:340
const char * desc
Definition: libsvtav1.c:79
uint8_t w
Definition: llviddspenc.c:39
static const uint32_t color[16+AV_CLASS_CATEGORY_NB]
Definition: log.c:92
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
Definition: mpeg12dec.c:2033
const char data[16]
Definition: mxf.c:142
AVOptions.
int av_get_bits_per_pixel(const AVPixFmtDescriptor *pixdesc)
Return the number of bits per pixel used by the pixel format described by pixdesc.
Definition: pixdesc.c:2525
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2573
#define AV_PIX_FMT_FLAG_PLANAR
At least one pixel component is not in the first data plane.
Definition: pixdesc.h:144
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
@ AV_PIX_FMT_YUV420P14LE
planar YUV 4:2:0,21bpp, (1 Cr & Cb sample per 2x2 Y samples), little-endian
Definition: pixfmt.h:245
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:99
@ AV_PIX_FMT_YUVA444P9LE
planar YUV 4:4:4 36bpp, (1 Cr & Cb sample per 1x1 Y & A samples), little-endian
Definition: pixfmt.h:183
@ AV_PIX_FMT_YUVA444P10LE
planar YUV 4:4:4 40bpp, (1 Cr & Cb sample per 1x1 Y & A samples, little-endian)
Definition: pixfmt.h:189
@ AV_PIX_FMT_YUV422P
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:70
@ AV_PIX_FMT_YUVA422P9LE
planar YUV 4:2:2 27bpp, (1 Cr & Cb sample per 2x1 Y & A samples), little-endian
Definition: pixfmt.h:181
@ AV_PIX_FMT_YUV420P9LE
planar YUV 4:2:0, 13.5bpp, (1 Cr & Cb sample per 2x2 Y samples), little-endian
Definition: pixfmt.h:157
@ AV_PIX_FMT_YUVA420P10LE
planar YUV 4:2:0 25bpp, (1 Cr & Cb sample per 2x2 Y & A samples, little-endian)
Definition: pixfmt.h:185
@ AV_PIX_FMT_YUVA420P
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
Definition: pixfmt.h:101
@ AV_PIX_FMT_YUVA422P10LE
planar YUV 4:2:2 30bpp, (1 Cr & Cb sample per 2x1 Y & A samples, little-endian)
Definition: pixfmt.h:187
@ AV_PIX_FMT_YUV420P10LE
planar YUV 4:2:0, 15bpp, (1 Cr & Cb sample per 2x2 Y samples), little-endian
Definition: pixfmt.h:159
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
Definition: pixfmt.h:72
@ AV_PIX_FMT_RGBA64LE
packed RGBA 16:16:16:16, 64bpp, 16R, 16G, 16B, 16A, the 2-byte value for each R/G/B/A component is st...
Definition: pixfmt.h:206
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:73
@ AV_PIX_FMT_YUV422P16LE
planar YUV 4:2:2, 32bpp, (1 Cr & Cb sample per 2x1 Y samples), little-endian
Definition: pixfmt.h:133
@ AV_PIX_FMT_YUV444P14LE
planar YUV 4:4:4,42bpp, (1 Cr & Cb sample per 1x1 Y samples), little-endian
Definition: pixfmt.h:253
@ AV_PIX_FMT_YUV422P12LE
planar YUV 4:2:2,24bpp, (1 Cr & Cb sample per 2x1 Y samples), little-endian
Definition: pixfmt.h:247
@ AV_PIX_FMT_YUV440P12LE
planar YUV 4:4:0,24bpp, (1 Cr & Cb sample per 1x2 Y samples), little-endian
Definition: pixfmt.h:277
@ AV_PIX_FMT_YUVA420P9LE
planar YUV 4:2:0 22.5bpp, (1 Cr & Cb sample per 2x2 Y & A samples), little-endian
Definition: pixfmt.h:179
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:71
@ AV_PIX_FMT_YUV444P9LE
planar YUV 4:4:4, 27bpp, (1 Cr & Cb sample per 1x1 Y samples), little-endian
Definition: pixfmt.h:163
@ AV_PIX_FMT_YUVA444P
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
Definition: pixfmt.h:177
@ AV_PIX_FMT_YUVA444P16LE
planar YUV 4:4:4 64bpp, (1 Cr & Cb sample per 1x1 Y & A samples, little-endian)
Definition: pixfmt.h:195
@ AV_PIX_FMT_YUVA420P16LE
planar YUV 4:2:0 40bpp, (1 Cr & Cb sample per 2x2 Y & A samples, little-endian)
Definition: pixfmt.h:191
@ AV_PIX_FMT_RGB48LE
packed RGB 16:16:16, 48bpp, 16R, 16G, 16B, the 2-byte value for each R/G/B component is stored as lit...
Definition: pixfmt.h:103
@ AV_PIX_FMT_YUV420P12LE
planar YUV 4:2:0,18bpp, (1 Cr & Cb sample per 2x2 Y samples), little-endian
Definition: pixfmt.h:243
@ AV_PIX_FMT_YUV422P9LE
planar YUV 4:2:2, 18bpp, (1 Cr & Cb sample per 2x1 Y samples), little-endian
Definition: pixfmt.h:167
@ AV_PIX_FMT_YUV422P10LE
planar YUV 4:2:2, 20bpp, (1 Cr & Cb sample per 2x1 Y samples), little-endian
Definition: pixfmt.h:161
@ AV_PIX_FMT_YUVA422P
planar YUV 4:2:2 24bpp, (1 Cr & Cb sample per 2x1 Y & A samples)
Definition: pixfmt.h:176
@ AV_PIX_FMT_YUV422P14LE
planar YUV 4:2:2,28bpp, (1 Cr & Cb sample per 2x1 Y samples), little-endian
Definition: pixfmt.h:249
@ AV_PIX_FMT_YUV440P10LE
planar YUV 4:4:0,20bpp, (1 Cr & Cb sample per 1x2 Y samples), little-endian
Definition: pixfmt.h:275
@ AV_PIX_FMT_YUV444P16LE
planar YUV 4:4:4, 48bpp, (1 Cr & Cb sample per 1x1 Y samples), little-endian
Definition: pixfmt.h:135
@ AV_PIX_FMT_YUVA422P16LE
planar YUV 4:2:2 48bpp, (1 Cr & Cb sample per 2x1 Y & A samples, little-endian)
Definition: pixfmt.h:193
@ AV_PIX_FMT_YUV444P10LE
planar YUV 4:4:4, 30bpp, (1 Cr & Cb sample per 1x1 Y samples), little-endian
Definition: pixfmt.h:165
@ AV_PIX_FMT_YUV420P16LE
planar YUV 4:2:0, 24bpp, (1 Cr & Cb sample per 2x2 Y samples), little-endian
Definition: pixfmt.h:131
@ AV_PIX_FMT_YUV444P12LE
planar YUV 4:4:4,36bpp, (1 Cr & Cb sample per 1x1 Y samples), little-endian
Definition: pixfmt.h:251
#define td
Definition: regdef.h:70
var_name
Definition: setts_bsf.c:50
#define av_bswap16
Definition: bswap.h:31
#define FF_ARRAY_ELEMS(a)
Describe the class of an AVClass context structure.
Definition: log.h:67
Definition: eval.c:159
An instance of a filter.
Definition: avfilter.h:341
A list of supported formats for one end of a filter link.
Definition: formats.h:65
A filter pad used for either input or output.
Definition: internal.h:54
const char * name
Pad name.
Definition: internal.h:60
AVFormatInternal * internal
An opaque field for libavformat internal usage.
Definition: avformat.h:1699
This structure describes decoded (raw) audio or video data.
Definition: frame.h:318
AVOption.
Definition: opt.h:248
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
uint16_t lut[4][256 *256]
lookup table for each component
Definition: vf_lut.c:64
double var_values[VAR_VARS_NB]
Definition: vf_lut.c:68
char * comp_expr_str[4]
Definition: vf_lut.c:65
int is_rgb
Definition: vf_lut.c:69
int is_planar
Definition: vf_lut.c:70
int vsub
Definition: vf_lut.c:67
int step
Definition: vf_lut.c:72
int hsub
Definition: vf_lut.c:67
int negate_alpha
Definition: vf_lut.c:73
AVExpr * comp_expr[4]
Definition: vf_lut.c:66
int is_16bit
Definition: vf_lut.c:71
int is_yuv
Definition: vf_lut.c:69
AVFrame * out
Definition: vf_lut.c:343
AVFrame * in
Definition: vf_lut.c:342
uint8_t level
Definition: svq3.c:206
#define av_freep(p)
#define av_log(a,...)
FILE * out
Definition: movenc.c:54
AVFormatContext * ctx
Definition: movenc.c:48
static const struct twinvq_data tab
static enum AVPixelFormat all_pix_fmts[]
Definition: vf_lut.c:145
static int lut_packed_8bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_lut.c:402
#define PACKED_THREAD_DATA
Definition: vf_lut.c:506
static double(*const funcs1[])(void *, double)
Definition: vf_lut.c:201
@ VAR_H
Definition: vf_lut.c:53
@ VAR_VAL
Definition: vf_lut.c:54
@ VAR_W
Definition: vf_lut.c:52
@ VAR_VARS_NB
Definition: vf_lut.c:59
@ VAR_CLIPVAL
Definition: vf_lut.c:58
@ VAR_NEGVAL
Definition: vf_lut.c:57
@ VAR_MINVAL
Definition: vf_lut.c:56
@ VAR_MAXVAL
Definition: vf_lut.c:55
#define B
Definition: vf_lut.c:81
#define LOAD_PLANAR_COMMON
Definition: vf_lut.c:430
static const AVOption options[]
Definition: vf_lut.c:87
#define PLANAR_THREAD_DATA
Definition: vf_lut.c:514
#define RGB_FORMATS
Definition: vf_lut.c:127
static enum AVPixelFormat rgb_pix_fmts[]
Definition: vf_lut.c:144
static const char *const funcs1_names[]
Definition: vf_lut.c:208
static int query_formats(AVFilterContext *ctx)
Definition: vf_lut.c:147
#define R
Definition: vf_lut.c:79
#define FLAGS
Definition: vf_lut.c:85
#define PLANAR_COMMON
Definition: vf_lut.c:437
static const AVFilterPad inputs[]
Definition: vf_lut.c:581
#define Y
Definition: vf_lut.c:76
static const AVFilterPad outputs[]
Definition: vf_lut.c:589
static int lut_planar_8bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_lut.c:480
static int lut_packed_16bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_lut.c:365
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: vf_lut.c:522
#define A
Definition: vf_lut.c:82
static int lut_planar_16bits(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_lut.c:448
static double compute_gammaval709(void *opaque, double gamma)
Compute ITU Rec.709 gamma correction of value val.
Definition: vf_lut.c:189
static int config_props(AVFilterLink *inlink)
Definition: vf_lut.c:215
static const char *const var_names[]
Definition: vf_lut.c:40
static int process_command(AVFilterContext *ctx, const char *cmd, const char *args, char *res, int res_len, int flags)
Definition: vf_lut.c:570
#define LOAD_PACKED_COMMON
Definition: vf_lut.c:349
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_lut.c:102
#define DEFINE_LUT_FILTER(name_, description_)
Definition: vf_lut.c:596
#define YUV_FORMATS
Definition: vf_lut.c:114
static double compute_gammaval(void *opaque, double gamma)
Compute gamma correction for value val, assuming the minval-maxval range, val is clipped to a value c...
Definition: vf_lut.c:176
#define OFFSET(x)
Definition: vf_lut.c:84
static double clip(void *opaque, double val)
Clip value val in the minval - maxval range.
Definition: vf_lut.c:163
#define GRAY_FORMATS
Definition: vf_lut.c:139
#define G
Definition: vf_lut.c:80
static enum AVPixelFormat yuv_pix_fmts[]
Definition: vf_lut.c:143
#define V
Definition: vf_lut.c:78
#define U
Definition: vf_lut.c:77
if(ret< 0)
Definition: vf_mcdeint.c:282
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
Definition: video.c:104
float min