FFmpeg  4.3.9
vf_colorbalance.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2013 Paul B Mahol
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/opt.h"
22 #include "libavutil/pixdesc.h"
23 #include "avfilter.h"
24 #include "drawutils.h"
25 #include "formats.h"
26 #include "internal.h"
27 #include "video.h"
28 
29 #define R 0
30 #define G 1
31 #define B 2
32 #define A 3
33 
34 typedef struct ThreadData {
35  AVFrame *in, *out;
36 } ThreadData;
37 
38 typedef struct Range {
39  float shadows;
40  float midtones;
41  float highlights;
42 } Range;
43 
44 typedef struct ColorBalanceContext {
45  const AVClass *class;
50 
51  uint8_t rgba_map[4];
52  int depth;
53  int max;
54  int step;
55 
56  int (*color_balance)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs);
58 
59 #define OFFSET(x) offsetof(ColorBalanceContext, x)
60 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
61 static const AVOption colorbalance_options[] = {
62  { "rs", "set red shadows", OFFSET(cyan_red.shadows), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
63  { "gs", "set green shadows", OFFSET(magenta_green.shadows), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
64  { "bs", "set blue shadows", OFFSET(yellow_blue.shadows), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
65  { "rm", "set red midtones", OFFSET(cyan_red.midtones), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
66  { "gm", "set green midtones", OFFSET(magenta_green.midtones), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
67  { "bm", "set blue midtones", OFFSET(yellow_blue.midtones), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
68  { "rh", "set red highlights", OFFSET(cyan_red.highlights), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
69  { "gh", "set green highlights", OFFSET(magenta_green.highlights), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
70  { "bh", "set blue highlights", OFFSET(yellow_blue.highlights), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, FLAGS },
71  { "pl", "preserve lightness", OFFSET(preserve_lightness), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS },
72  { NULL }
73 };
74 
75 AVFILTER_DEFINE_CLASS(colorbalance);
76 
78 {
79  static const enum AVPixelFormat pix_fmts[] = {
94  };
95  AVFilterFormats *fmts_list = ff_make_format_list(pix_fmts);
96  if (!fmts_list)
97  return AVERROR(ENOMEM);
98  return ff_set_common_formats(ctx, fmts_list);
99 }
100 
101 static float get_component(float v, float l,
102  float s, float m, float h)
103 {
104  const float a = 4.f, b = 0.333f, scale = 0.7f;
105 
106  s *= av_clipf((b - l) * a + 0.5f, 0, 1) * scale;
107  m *= av_clipf((l - b) * a + 0.5f, 0, 1) * av_clipf((1.0 - l - b) * a + 0.5f, 0, 1) * scale;
108  h *= av_clipf((l + b - 1) * a + 0.5f, 0, 1) * scale;
109 
110  v += s;
111  v += m;
112  v += h;
113 
114  return av_clipf(v + 0.5f, 0, 1);
115 }
116 
117 static float hfun(float n, float h, float s, float l)
118 {
119  float a = s * FFMIN(l, 1. - l);
120  float k = fmodf(n + h / 30.f, 12.f);
121 
122  return av_clipf(l - a * FFMAX(FFMIN3(k - 3.f, 9.f - k, 1), -1.f), 0, 1);
123 }
124 
125 static void preservel(float *r, float *g, float *b, float l)
126 {
127  float max = FFMAX3(*r, *g, *b);
128  float min = FFMIN3(*r, *g, *b);
129  float h, s;
130 
131  l *= 0.5;
132 
133  if (*r == *g && *g == *b) {
134  h = 0.;
135  } else if (max == *r) {
136  h = 60. * (0. + (*g - *b) / (max - min));
137  } else if (max == *g) {
138  h = 60. * (2. + (*b - *r) / (max - min));
139  } else if (max == *b) {
140  h = 60. * (4. + (*r - *g) / (max - min));
141  } else {
142  h = 0.;
143  }
144  if (h < 0.)
145  h += 360.;
146 
147  if (max == 0. || min == 1.) {
148  s = 0.;
149  } else {
150  s = (max - min) / (1. - FFABS(2. * l - 1));
151  }
152 
153  *r = hfun(0, h, s, l);
154  *g = hfun(8, h, s, l);
155  *b = hfun(4, h, s, l);
156 }
157 
158 static int color_balance8_p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
159 {
160  ColorBalanceContext *s = ctx->priv;
161  ThreadData *td = arg;
162  AVFrame *in = td->in;
163  AVFrame *out = td->out;
164  const int slice_start = (out->height * jobnr) / nb_jobs;
165  const int slice_end = (out->height * (jobnr+1)) / nb_jobs;
166  const uint8_t *srcg = in->data[0] + slice_start * in->linesize[0];
167  const uint8_t *srcb = in->data[1] + slice_start * in->linesize[1];
168  const uint8_t *srcr = in->data[2] + slice_start * in->linesize[2];
169  const uint8_t *srca = in->data[3] + slice_start * in->linesize[3];
170  uint8_t *dstg = out->data[0] + slice_start * out->linesize[0];
171  uint8_t *dstb = out->data[1] + slice_start * out->linesize[1];
172  uint8_t *dstr = out->data[2] + slice_start * out->linesize[2];
173  uint8_t *dsta = out->data[3] + slice_start * out->linesize[3];
174  const float max = s->max;
175  int i, j;
176 
177  for (i = slice_start; i < slice_end; i++) {
178  for (j = 0; j < out->width; j++) {
179  float r = srcr[j] / max;
180  float g = srcg[j] / max;
181  float b = srcb[j] / max;
182  const float l = FFMAX3(r, g, b) + FFMIN3(r, g, b);
183 
187 
188  if (s->preserve_lightness)
189  preservel(&r, &g, &b, l);
190 
191  dstr[j] = av_clip_uint8(r * max);
192  dstg[j] = av_clip_uint8(g * max);
193  dstb[j] = av_clip_uint8(b * max);
194  if (in != out && out->linesize[3])
195  dsta[j] = srca[j];
196  }
197 
198  srcg += in->linesize[0];
199  srcb += in->linesize[1];
200  srcr += in->linesize[2];
201  srca += in->linesize[3];
202  dstg += out->linesize[0];
203  dstb += out->linesize[1];
204  dstr += out->linesize[2];
205  dsta += out->linesize[3];
206  }
207 
208  return 0;
209 }
210 
211 static int color_balance16_p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
212 {
213  ColorBalanceContext *s = ctx->priv;
214  ThreadData *td = arg;
215  AVFrame *in = td->in;
216  AVFrame *out = td->out;
217  const int slice_start = (out->height * jobnr) / nb_jobs;
218  const int slice_end = (out->height * (jobnr+1)) / nb_jobs;
219  const uint16_t *srcg = (const uint16_t *)in->data[0] + slice_start * in->linesize[0] / 2;
220  const uint16_t *srcb = (const uint16_t *)in->data[1] + slice_start * in->linesize[1] / 2;
221  const uint16_t *srcr = (const uint16_t *)in->data[2] + slice_start * in->linesize[2] / 2;
222  const uint16_t *srca = (const uint16_t *)in->data[3] + slice_start * in->linesize[3] / 2;
223  uint16_t *dstg = (uint16_t *)out->data[0] + slice_start * out->linesize[0] / 2;
224  uint16_t *dstb = (uint16_t *)out->data[1] + slice_start * out->linesize[1] / 2;
225  uint16_t *dstr = (uint16_t *)out->data[2] + slice_start * out->linesize[2] / 2;
226  uint16_t *dsta = (uint16_t *)out->data[3] + slice_start * out->linesize[3] / 2;
227  const int depth = s->depth;
228  const float max = s->max;
229  int i, j;
230 
231  for (i = slice_start; i < slice_end; i++) {
232  for (j = 0; j < out->width; j++) {
233  float r = srcr[j] / max;
234  float g = srcg[j] / max;
235  float b = srcb[j] / max;
236  const float l = (FFMAX3(r, g, b) + FFMIN3(r, g, b));
237 
241 
242  if (s->preserve_lightness)
243  preservel(&r, &g, &b, l);
244 
245  dstr[j] = av_clip_uintp2_c(r * max, depth);
246  dstg[j] = av_clip_uintp2_c(g * max, depth);
247  dstb[j] = av_clip_uintp2_c(b * max, depth);
248  if (in != out && out->linesize[3])
249  dsta[j] = srca[j];
250  }
251 
252  srcg += in->linesize[0] / 2;
253  srcb += in->linesize[1] / 2;
254  srcr += in->linesize[2] / 2;
255  srca += in->linesize[3] / 2;
256  dstg += out->linesize[0] / 2;
257  dstb += out->linesize[1] / 2;
258  dstr += out->linesize[2] / 2;
259  dsta += out->linesize[3] / 2;
260  }
261 
262  return 0;
263 }
264 
265 static int color_balance8(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
266 {
267  ColorBalanceContext *s = ctx->priv;
268  ThreadData *td = arg;
269  AVFrame *in = td->in;
270  AVFrame *out = td->out;
271  AVFilterLink *outlink = ctx->outputs[0];
272  const int slice_start = (out->height * jobnr) / nb_jobs;
273  const int slice_end = (out->height * (jobnr+1)) / nb_jobs;
274  const uint8_t *srcrow = in->data[0] + slice_start * in->linesize[0];
275  const uint8_t roffset = s->rgba_map[R];
276  const uint8_t goffset = s->rgba_map[G];
277  const uint8_t boffset = s->rgba_map[B];
278  const uint8_t aoffset = s->rgba_map[A];
279  const float max = s->max;
280  const int step = s->step;
281  uint8_t *dstrow;
282  int i, j;
283 
284  dstrow = out->data[0] + slice_start * out->linesize[0];
285  for (i = slice_start; i < slice_end; i++) {
286  const uint8_t *src = srcrow;
287  uint8_t *dst = dstrow;
288 
289  for (j = 0; j < outlink->w * step; j += step) {
290  float r = src[j + roffset] / max;
291  float g = src[j + goffset] / max;
292  float b = src[j + boffset] / max;
293  const float l = (FFMAX3(r, g, b) + FFMIN3(r, g, b));
294 
298 
299  if (s->preserve_lightness)
300  preservel(&r, &g, &b, l);
301 
302  dst[j + roffset] = av_clip_uint8(r * max);
303  dst[j + goffset] = av_clip_uint8(g * max);
304  dst[j + boffset] = av_clip_uint8(b * max);
305  if (in != out && step == 4)
306  dst[j + aoffset] = src[j + aoffset];
307  }
308 
309  srcrow += in->linesize[0];
310  dstrow += out->linesize[0];
311  }
312 
313  return 0;
314 }
315 
316 static int color_balance16(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
317 {
318  ColorBalanceContext *s = ctx->priv;
319  ThreadData *td = arg;
320  AVFrame *in = td->in;
321  AVFrame *out = td->out;
322  AVFilterLink *outlink = ctx->outputs[0];
323  const int slice_start = (out->height * jobnr) / nb_jobs;
324  const int slice_end = (out->height * (jobnr+1)) / nb_jobs;
325  const uint16_t *srcrow = (const uint16_t *)in->data[0] + slice_start * in->linesize[0] / 2;
326  const uint8_t roffset = s->rgba_map[R];
327  const uint8_t goffset = s->rgba_map[G];
328  const uint8_t boffset = s->rgba_map[B];
329  const uint8_t aoffset = s->rgba_map[A];
330  const int step = s->step / 2;
331  const int depth = s->depth;
332  const float max = s->max;
333  uint16_t *dstrow;
334  int i, j;
335 
336  dstrow = (uint16_t *)out->data[0] + slice_start * out->linesize[0] / 2;
337  for (i = slice_start; i < slice_end; i++) {
338  const uint16_t *src = srcrow;
339  uint16_t *dst = dstrow;
340 
341  for (j = 0; j < outlink->w * step; j += step) {
342  float r = src[j + roffset] / max;
343  float g = src[j + goffset] / max;
344  float b = src[j + boffset] / max;
345  const float l = (FFMAX3(r, g, b) + FFMIN3(r, g, b));
346 
350 
351  if (s->preserve_lightness)
352  preservel(&r, &g, &b, l);
353 
354  dst[j + roffset] = av_clip_uintp2_c(r * max, depth);
355  dst[j + goffset] = av_clip_uintp2_c(g * max, depth);
356  dst[j + boffset] = av_clip_uintp2_c(b * max, depth);
357  if (in != out && step == 4)
358  dst[j + aoffset] = src[j + aoffset];
359  }
360 
361  srcrow += in->linesize[0] / 2;
362  dstrow += out->linesize[0] / 2;
363  }
364 
365  return 0;
366 }
367 
368 static int config_output(AVFilterLink *outlink)
369 {
370  AVFilterContext *ctx = outlink->src;
371  ColorBalanceContext *s = ctx->priv;
373  const int depth = desc->comp[0].depth;
374  const int max = (1 << depth) - 1;
375  const int planar = av_pix_fmt_count_planes(outlink->format) > 1;
376 
377  s->depth = depth;
378  s->max = max;
379 
380  if (max == 255 && planar) {
382  } else if (planar) {
384  } else if (max == 255) {
386  } else {
388  }
389 
390  ff_fill_rgba_map(s->rgba_map, outlink->format);
391  s->step = av_get_padded_bits_per_pixel(desc) >> 3;
392 
393  return 0;
394 }
395 
397 {
398  AVFilterContext *ctx = inlink->dst;
399  ColorBalanceContext *s = ctx->priv;
400  AVFilterLink *outlink = ctx->outputs[0];
401  ThreadData td;
402  AVFrame *out;
403 
404  if (av_frame_is_writable(in)) {
405  out = in;
406  } else {
407  out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
408  if (!out) {
409  av_frame_free(&in);
410  return AVERROR(ENOMEM);
411  }
412  av_frame_copy_props(out, in);
413  }
414 
415  td.in = in;
416  td.out = out;
417  ctx->internal->execute(ctx, s->color_balance, &td, NULL, FFMIN(outlink->h, ff_filter_get_nb_threads(ctx)));
418 
419  if (in != out)
420  av_frame_free(&in);
421  return ff_filter_frame(outlink, out);
422 }
423 
425  {
426  .name = "default",
427  .type = AVMEDIA_TYPE_VIDEO,
428  .filter_frame = filter_frame,
429  },
430  { NULL }
431 };
432 
434  {
435  .name = "default",
436  .type = AVMEDIA_TYPE_VIDEO,
437  .config_props = config_output,
438  },
439  { NULL }
440 };
441 
443  .name = "colorbalance",
444  .description = NULL_IF_CONFIG_SMALL("Adjust the color balance."),
445  .priv_size = sizeof(ColorBalanceContext),
446  .priv_class = &colorbalance_class,
448  .inputs = colorbalance_inputs,
449  .outputs = colorbalance_outputs,
452 };
#define NULL
Definition: coverity.c:32
static const AVFilterPad colorbalance_inputs[]
AVFrame * out
Definition: af_adeclick.c:494
#define A
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2549
This structure describes decoded (raw) audio or video data.
Definition: frame.h:300
AVOption.
Definition: opt.h:246
#define AV_PIX_FMT_GBRAP10
Definition: pixfmt.h:417
static const AVFilterPad colorbalance_outputs[]
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2589
Main libavfilter public API header.
packed RGB 8:8:8, 24bpp, RGBRGB...
Definition: pixfmt.h:68
const char * g
Definition: vf_curves.c:115
#define AV_PIX_FMT_RGBA64
Definition: pixfmt.h:387
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:168
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:413
#define AV_PIX_FMT_BGRA64
Definition: pixfmt.h:392
static void preservel(float *r, float *g, float *b, float l)
packed BGR 8:8:8, 32bpp, XBGRXBGR... X=unused/undefined
Definition: pixfmt.h:239
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
static int color_balance16_p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
AVFilterFormats * ff_make_format_list(const int *fmts)
Create a list of supported formats.
Definition: formats.c:300
#define AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
Some filters support a generic "enable" expression option that can be used to enable or disable a fil...
Definition: avfilter.h:125
const char * name
Pad name.
Definition: internal.h:60
static int query_formats(AVFilterContext *ctx)
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1075
AVComponentDescriptor comp[4]
Parameters that describe how pixels are packed.
Definition: pixdesc.h:117
uint8_t
packed RGB 8:8:8, 32bpp, RGBXRGBX... X=unused/undefined
Definition: pixfmt.h:238
static int config_output(AVFilterLink *outlink)
AVOptions.
#define f(width, name)
Definition: cbs_vp9.c:255
float highlights
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:412
packed ABGR 8:8:8:8, 32bpp, ABGRABGR...
Definition: pixfmt.h:94
static int color_balance16(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
AVFrame * dst
Definition: vf_blend.c:56
#define FFMIN3(a, b, c)
Definition: common.h:97
#define AV_PIX_FMT_BGR48
Definition: pixfmt.h:388
float midtones
A filter pad used for either input or output.
Definition: internal.h:54
static int color_balance8(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:269
int width
Definition: frame.h:358
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:605
#define td
Definition: regdef.h:70
int step
Definition: vf_remap.c:82
#define AVERROR(e)
Definition: error.h:43
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:203
AVFILTER_DEFINE_CLASS(colorbalance)
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:188
const uint8_t * src
Definition: vf_bm3d.c:56
packed BGRA 8:8:8:8, 32bpp, BGRABGRA...
Definition: pixfmt.h:95
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:869
const char * r
Definition: vf_curves.c:114
void * priv
private data for use by the filter
Definition: avfilter.h:353
int av_get_padded_bits_per_pixel(const AVPixFmtDescriptor *pixdesc)
Return the number of bits per pixel for the pixel format described by pixdesc, including any padding ...
Definition: pixdesc.c:2514
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
Definition: avfilter.h:116
const char * arg
Definition: jacosubdec.c:66
#define AV_PIX_FMT_GBRAP12
Definition: pixfmt.h:418
#define AV_PIX_FMT_RGB48
Definition: pixfmt.h:383
static float hfun(float n, float h, float s, float l)
#define B
#define FFMAX(a, b)
Definition: common.h:94
packed ARGB 8:8:8:8, 32bpp, ARGBARGB...
Definition: pixfmt.h:92
#define AV_PIX_FMT_GBRAP16
Definition: pixfmt.h:419
packed RGBA 8:8:8:8, 32bpp, RGBARGBA...
Definition: pixfmt.h:93
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:416
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Definition: avfilter.c:784
#define FFMIN(a, b)
Definition: common.h:96
AVFrame * m
AVFormatContext * ctx
Definition: movenc.c:48
#define FFABS(a)
Absolute value, Note, INT_MIN / INT64_MIN result in undefined behavior as they are not representable ...
Definition: common.h:72
packed RGB 8:8:8, 24bpp, BGRBGR...
Definition: pixfmt.h:69
static const AVFilterPad inputs[]
Definition: af_acontrast.c:193
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:415
static const AVFilterPad outputs[]
Definition: af_acontrast.c:203
#define R
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
Definition: drawutils.c:35
#define G
misc drawing utilities
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
Definition: frame.c:595
int(* color_balance)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Used for passing data between threads.
Definition: dsddec.c:67
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:331
AVFilter ff_vf_colorbalance
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
Describe the class of an AVClass context structure.
Definition: log.h:67
static float get_component(float v, float l, float s, float m, float h)
Filter definition.
Definition: avfilter.h:144
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
AVFrame * b
static int color_balance8_p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
packed BGR 8:8:8, 32bpp, BGRXBGRX... X=unused/undefined
Definition: pixfmt.h:240
const char * name
Filter name.
Definition: avfilter.h:148
AVFilterLink ** outputs
array of pointers to output links
Definition: avfilter.h:350
AVFrame * min
Definition: vf_threshold.c:74
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:275
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:414
#define flags(name, subs,...)
Definition: cbs_av1.c:576
AVFilterInternal * internal
An opaque struct for libavfilter internal use.
Definition: avfilter.h:378
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:314
int
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:215
uint8_t pi<< 24) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_U8,(uint64_t)((*(const uint8_t *) pi - 0x80U))<< 56) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16,(*(const int16_t *) pi >>8)+0x80) CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_S16, *(const int16_t *) pi *(1<< 16)) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_S16,(uint64_t)(*(const int16_t *) pi)<< 48) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32,(*(const int32_t *) pi >>24)+0x80) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_S32,(uint64_t)(*(const int32_t *) pi)<< 32) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S64,(*(const int64_t *) pi >>56)+0x80) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S64, *(const int64_t *) pi *(1.0f/(UINT64_C(1)<< 63))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S64, *(const int64_t *) pi *(1.0/(UINT64_C(1)<< 63))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_FLT, llrintf(*(const float *) pi *(UINT64_C(1)<< 63))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_DBL, llrint(*(const double *) pi *(UINT64_C(1)<< 63))) #define FMT_PAIR_FUNC(out, in) static conv_func_type *const fmt_pair_to_conv_functions[AV_SAMPLE_FMT_NB *AV_SAMPLE_FMT_NB]={ FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_S64), };static void cpy1(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, len);} static void cpy2(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, 2 *len);} static void cpy4(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, 4 *len);} static void cpy8(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, 8 *len);} AudioConvert *swri_audio_convert_alloc(enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, const int *ch_map, int flags) { AudioConvert *ctx;conv_func_type *f=fmt_pair_to_conv_functions[av_get_packed_sample_fmt(out_fmt)+AV_SAMPLE_FMT_NB *av_get_packed_sample_fmt(in_fmt)];if(!f) return NULL;ctx=av_mallocz(sizeof(*ctx));if(!ctx) return NULL;if(channels==1){ in_fmt=av_get_planar_sample_fmt(in_fmt);out_fmt=av_get_planar_sample_fmt(out_fmt);} ctx->channels=channels;ctx->conv_f=f;ctx->ch_map=ch_map;if(in_fmt==AV_SAMPLE_FMT_U8||in_fmt==AV_SAMPLE_FMT_U8P) memset(ctx->silence, 0x80, sizeof(ctx->silence));if(out_fmt==in_fmt &&!ch_map) { switch(av_get_bytes_per_sample(in_fmt)){ case 1:ctx->simd_f=cpy1;break;case 2:ctx->simd_f=cpy2;break;case 4:ctx->simd_f=cpy4;break;case 8:ctx->simd_f=cpy8;break;} } if(HAVE_X86ASM &&HAVE_MMX) swri_audio_convert_init_x86(ctx, out_fmt, in_fmt, channels);if(ARCH_ARM) swri_audio_convert_init_arm(ctx, out_fmt, in_fmt, channels);if(ARCH_AARCH64) swri_audio_convert_init_aarch64(ctx, out_fmt, in_fmt, channels);return ctx;} void swri_audio_convert_free(AudioConvert **ctx) { av_freep(ctx);} int swri_audio_convert(AudioConvert *ctx, AudioData *out, AudioData *in, int len) { int ch;int off=0;const int os=(out->planar ? 1 :out->ch_count) *out->bps;unsigned misaligned=0;av_assert0(ctx->channels==out->ch_count);if(ctx->in_simd_align_mask) { int planes=in->planar ? in->ch_count :1;unsigned m=0;for(ch=0;ch< planes;ch++) m|=(intptr_t) in->ch[ch];misaligned|=m &ctx->in_simd_align_mask;} if(ctx->out_simd_align_mask) { int planes=out->planar ? out->ch_count :1;unsigned m=0;for(ch=0;ch< planes;ch++) m|=(intptr_t) out->ch[ch];misaligned|=m &ctx->out_simd_align_mask;} if(ctx->simd_f &&!ctx->ch_map &&!misaligned){ off=len &~15;av_assert1(off >=0);av_assert1(off<=len);av_assert2(ctx->channels==SWR_CH_MAX||!in->ch[ctx->channels]);if(off >0){ if(out->planar==in->planar){ int planes=out->planar ? out->ch_count :1;for(ch=0;ch< planes;ch++){ ctx->simd_f(out->ch+ch,(const uint8_t **) in->ch+ch, off *(out-> planar
Definition: audioconvert.c:56
avfilter_execute_func * execute
Definition: internal.h:144
#define OFFSET(x)
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
Definition: mpeg12dec.c:2040
const AVPixFmtDescriptor * desc
Definition: vf_tonemap.c:196
A list of supported formats for one end of a filter link.
Definition: formats.h:64
float shadows
An instance of a filter.
Definition: avfilter.h:338
static const AVOption colorbalance_options[]
int height
Definition: frame.h:358
static int process_command(AVFilterContext *ctx, const char *cmd, const char *args, char *res, int res_len, int flags)
Definition: af_afftdn.c:1374
const void ** s
AVFrame * in
Definition: af_afftdn.c:1083
AVFilterLink * inlink
Definition: vf_blend.c:57
internal API functions
int depth
Number of bits in the component.
Definition: pixdesc.h:58
#define FLAGS
packed RGB 8:8:8, 32bpp, XRGBXRGB... X=unused/undefined
Definition: pixfmt.h:237
AVFrame * a
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:659
#define FFMAX3(a, b, c)
Definition: common.h:95
AVFrame * max
Definition: vf_threshold.c:75
static av_always_inline av_const unsigned av_clip_uintp2_c(int a, int p)
Clip a signed integer to an unsigned power of two range.
Definition: common.h:229