FFmpeg  4.4.8
vf_colorcontrast.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2021 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 <float.h>
22 
23 #include "libavutil/opt.h"
24 #include "libavutil/imgutils.h"
25 #include "avfilter.h"
26 #include "filters.h"
27 #include "drawutils.h"
28 #include "formats.h"
29 #include "internal.h"
30 #include "video.h"
31 
32 #define R 0
33 #define G 1
34 #define B 2
35 
36 typedef struct ColorContrastContext {
37  const AVClass *class;
38 
39  float rc, gm, by;
40  float rcw, gmw, byw;
41  float preserve;
42 
43  int step;
44  int depth;
46 
48  int jobnr, int nb_jobs);
50 
51 static inline float lerpf(float v0, float v1, float f)
52 {
53  return v0 + (v1 - v0) * f;
54 }
55 
56 #define PROCESS(max) \
57  br = (b + r) * 0.5f; \
58  gb = (g + b) * 0.5f; \
59  rg = (r + g) * 0.5f; \
60  \
61  gd = g - br; \
62  bd = b - rg; \
63  rd = r - gb; \
64  \
65  g0 = g + gd * gm; \
66  b0 = b - gd * gm; \
67  r0 = r - gd * gm; \
68  \
69  g1 = g - bd * by; \
70  b1 = b + bd * by; \
71  r1 = r - bd * by; \
72  \
73  g2 = g - rd * rc; \
74  b2 = b - rd * rc; \
75  r2 = r + rd * rc; \
76  \
77  ng = av_clipf((g0 * gmw + g1 * byw + g2 * rcw) * scale, 0.f, max); \
78  nb = av_clipf((b0 * gmw + b1 * byw + b2 * rcw) * scale, 0.f, max); \
79  nr = av_clipf((r0 * gmw + r1 * byw + r2 * rcw) * scale, 0.f, max); \
80  \
81  li = FFMAX3(r, g, b) + FFMIN3(r, g, b); \
82  lo = FFMAX3(nr, ng, nb) + FFMIN3(nr, ng, nb) + FLT_EPSILON; \
83  lf = li / lo; \
84  \
85  r = nr * lf; \
86  g = ng * lf; \
87  b = nb * lf; \
88  \
89  nr = lerpf(nr, r, preserve); \
90  ng = lerpf(ng, g, preserve); \
91  nb = lerpf(nb, b, preserve);
92 
93 static int colorcontrast_slice8(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
94 {
95  ColorContrastContext *s = ctx->priv;
96  AVFrame *frame = arg;
97  const int width = frame->width;
98  const int height = frame->height;
99  const int slice_start = ff_slice_pos(height, jobnr, nb_jobs);
100  const int slice_end = ff_slice_pos(height, jobnr + 1, nb_jobs);
101  const int glinesize = frame->linesize[0];
102  const int blinesize = frame->linesize[1];
103  const int rlinesize = frame->linesize[2];
104  uint8_t *gptr = frame->data[0] + slice_start * glinesize;
105  uint8_t *bptr = frame->data[1] + slice_start * blinesize;
106  uint8_t *rptr = frame->data[2] + slice_start * rlinesize;
107  const float preserve = s->preserve;
108  const float gm = s->gm * 0.5f;
109  const float by = s->by * 0.5f;
110  const float rc = s->rc * 0.5f;
111  const float gmw = s->gmw;
112  const float byw = s->byw;
113  const float rcw = s->rcw;
114  const float sum = gmw + byw + rcw;
115  const float scale = 1.f / sum;
116 
117  for (int y = slice_start; y < slice_end && sum > FLT_EPSILON; y++) {
118  for (int x = 0; x < width; x++) {
119  float g = gptr[x];
120  float b = bptr[x];
121  float r = rptr[x];
122  float g0, g1, g2;
123  float b0, b1, b2;
124  float r0, r1, r2;
125  float gd, bd, rd;
126  float gb, br, rg;
127  float nr, ng, nb;
128  float li, lo, lf;
129 
130  PROCESS(255.f);
131 
132  gptr[x] = av_clip_uint8(ng);
133  bptr[x] = av_clip_uint8(nb);
134  rptr[x] = av_clip_uint8(nr);
135  }
136 
137  gptr += glinesize;
138  bptr += blinesize;
139  rptr += rlinesize;
140  }
141 
142  return 0;
143 }
144 
145 static int colorcontrast_slice16(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
146 {
147  ColorContrastContext *s = ctx->priv;
148  AVFrame *frame = arg;
149  const int depth = s->depth;
150  const float max = (1 << depth) - 1;
151  const int width = frame->width;
152  const int height = frame->height;
153  const int slice_start = ff_slice_pos(height, jobnr, nb_jobs);
154  const int slice_end = ff_slice_pos(height, jobnr + 1, nb_jobs);
155  const int glinesize = frame->linesize[0] / 2;
156  const int blinesize = frame->linesize[1] / 2;
157  const int rlinesize = frame->linesize[2] / 2;
158  uint16_t *gptr = (uint16_t *)frame->data[0] + slice_start * glinesize;
159  uint16_t *bptr = (uint16_t *)frame->data[1] + slice_start * blinesize;
160  uint16_t *rptr = (uint16_t *)frame->data[2] + slice_start * rlinesize;
161  const float preserve = s->preserve;
162  const float gm = s->gm * 0.5f;
163  const float by = s->by * 0.5f;
164  const float rc = s->rc * 0.5f;
165  const float gmw = s->gmw;
166  const float byw = s->byw;
167  const float rcw = s->rcw;
168  const float sum = gmw + byw + rcw;
169  const float scale = 1.f / sum;
170 
171  for (int y = slice_start; y < slice_end && sum > FLT_EPSILON; y++) {
172  for (int x = 0; x < width; x++) {
173  float g = gptr[x];
174  float b = bptr[x];
175  float r = rptr[x];
176  float g0, g1, g2;
177  float b0, b1, b2;
178  float r0, r1, r2;
179  float gd, bd, rd;
180  float gb, br, rg;
181  float nr, ng, nb;
182  float li, lo, lf;
183 
184  PROCESS(max);
185 
186  gptr[x] = av_clip_uintp2_c(ng, depth);
187  bptr[x] = av_clip_uintp2_c(nb, depth);
188  rptr[x] = av_clip_uintp2_c(nr, depth);
189  }
190 
191  gptr += glinesize;
192  bptr += blinesize;
193  rptr += rlinesize;
194  }
195 
196  return 0;
197 }
198 
199 static int colorcontrast_slice8p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
200 {
201  ColorContrastContext *s = ctx->priv;
202  AVFrame *frame = arg;
203  const int step = s->step;
204  const int width = frame->width;
205  const int height = frame->height;
206  const int slice_start = ff_slice_pos(height, jobnr, nb_jobs);
207  const int slice_end = ff_slice_pos(height, jobnr + 1, nb_jobs);
208  const int linesize = frame->linesize[0];
209  const uint8_t roffset = s->rgba_map[R];
210  const uint8_t goffset = s->rgba_map[G];
211  const uint8_t boffset = s->rgba_map[B];
212  uint8_t *ptr = frame->data[0] + slice_start * linesize;
213  const float preserve = s->preserve;
214  const float gm = s->gm * 0.5f;
215  const float by = s->by * 0.5f;
216  const float rc = s->rc * 0.5f;
217  const float gmw = s->gmw;
218  const float byw = s->byw;
219  const float rcw = s->rcw;
220  const float sum = gmw + byw + rcw;
221  const float scale = 1.f / sum;
222 
223  for (int y = slice_start; y < slice_end && sum > FLT_EPSILON; y++) {
224  for (int x = 0; x < width; x++) {
225  float g = ptr[x * step + goffset];
226  float b = ptr[x * step + boffset];
227  float r = ptr[x * step + roffset];
228  float g0, g1, g2;
229  float b0, b1, b2;
230  float r0, r1, r2;
231  float gd, bd, rd;
232  float gb, br, rg;
233  float nr, ng, nb;
234  float li, lo, lf;
235 
236  PROCESS(255.f);
237 
238  ptr[x * step + goffset] = av_clip_uint8(ng);
239  ptr[x * step + boffset] = av_clip_uint8(nb);
240  ptr[x * step + roffset] = av_clip_uint8(nr);
241  }
242 
243  ptr += linesize;
244  }
245 
246  return 0;
247 }
248 
249 static int colorcontrast_slice16p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
250 {
251  ColorContrastContext *s = ctx->priv;
252  AVFrame *frame = arg;
253  const int step = s->step;
254  const int depth = s->depth;
255  const float max = (1 << depth) - 1;
256  const int width = frame->width;
257  const int height = frame->height;
258  const int slice_start = ff_slice_pos(height, jobnr, nb_jobs);
259  const int slice_end = ff_slice_pos(height, jobnr + 1, nb_jobs);
260  const int linesize = frame->linesize[0] / 2;
261  const uint8_t roffset = s->rgba_map[R];
262  const uint8_t goffset = s->rgba_map[G];
263  const uint8_t boffset = s->rgba_map[B];
264  uint16_t *ptr = (uint16_t *)frame->data[0] + slice_start * linesize;
265  const float preserve = s->preserve;
266  const float gm = s->gm * 0.5f;
267  const float by = s->by * 0.5f;
268  const float rc = s->rc * 0.5f;
269  const float gmw = s->gmw;
270  const float byw = s->byw;
271  const float rcw = s->rcw;
272  const float sum = gmw + byw + rcw;
273  const float scale = 1.f / sum;
274 
275  for (int y = slice_start; y < slice_end && sum > FLT_EPSILON; y++) {
276  for (int x = 0; x < width; x++) {
277  float g = ptr[x * step + goffset];
278  float b = ptr[x * step + boffset];
279  float r = ptr[x * step + roffset];
280  float g0, g1, g2;
281  float b0, b1, b2;
282  float r0, r1, r2;
283  float gd, bd, rd;
284  float gb, br, rg;
285  float nr, ng, nb;
286  float li, lo, lf;
287 
288  PROCESS(max);
289 
290  ptr[x * step + goffset] = av_clip_uintp2_c(ng, depth);
291  ptr[x * step + boffset] = av_clip_uintp2_c(nb, depth);
292  ptr[x * step + roffset] = av_clip_uintp2_c(nr, depth);
293  }
294 
295  ptr += linesize;
296  }
297 
298  return 0;
299 }
300 
302 {
303  AVFilterContext *ctx = link->dst;
304  ColorContrastContext *s = ctx->priv;
305  int res;
306 
307  if (res = ctx->internal->execute(ctx, s->do_slice, frame, NULL,
309  return res;
310 
311  return ff_filter_frame(ctx->outputs[0], frame);
312 }
313 
315 {
316  static const enum AVPixelFormat pixel_fmts[] = {
329  };
330 
332 
333  formats = ff_make_format_list(pixel_fmts);
334  if (!formats)
335  return AVERROR(ENOMEM);
336 
338 }
339 
340 static av_cold int config_input(AVFilterLink *inlink)
341 {
342  AVFilterContext *ctx = inlink->dst;
343  ColorContrastContext *s = ctx->priv;
345  int planar = desc->flags & AV_PIX_FMT_FLAG_PLANAR;
346 
347  s->step = desc->nb_components;
348  if (inlink->format == AV_PIX_FMT_RGB0 ||
349  inlink->format == AV_PIX_FMT_0RGB ||
350  inlink->format == AV_PIX_FMT_BGR0 ||
351  inlink->format == AV_PIX_FMT_0BGR)
352  s->step = 4;
353 
354  s->depth = desc->comp[0].depth;
355  s->do_slice = s->depth <= 8 ? colorcontrast_slice8 : colorcontrast_slice16;
356  if (!planar)
357  s->do_slice = s->depth <= 8 ? colorcontrast_slice8p : colorcontrast_slice16p;
358 
359  ff_fill_rgba_map(s->rgba_map, inlink->format);
360 
361  return 0;
362 }
363 
365  {
366  .name = "default",
367  .type = AVMEDIA_TYPE_VIDEO,
368  .needs_writable = 1,
369  .filter_frame = filter_frame,
370  .config_props = config_input,
371  },
372  { NULL }
373 };
374 
376  {
377  .name = "default",
378  .type = AVMEDIA_TYPE_VIDEO,
379  },
380  { NULL }
381 };
382 
383 #define OFFSET(x) offsetof(ColorContrastContext, x)
384 #define VF AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
385 
386 static const AVOption colorcontrast_options[] = {
387  { "rc", "set the red-cyan contrast", OFFSET(rc), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, VF },
388  { "gm", "set the green-magenta contrast", OFFSET(gm), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, VF },
389  { "by", "set the blue-yellow contrast", OFFSET(by), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, VF },
390  { "rcw", "set the red-cyan weight", OFFSET(rcw), AV_OPT_TYPE_FLOAT, {.dbl=0}, 0, 1, VF },
391  { "gmw", "set the green-magenta weight", OFFSET(gmw), AV_OPT_TYPE_FLOAT, {.dbl=0}, 0, 1, VF },
392  { "byw", "set the blue-yellow weight", OFFSET(byw), AV_OPT_TYPE_FLOAT, {.dbl=0}, 0, 1, VF },
393  { "pl", "set the amount of preserving lightness", OFFSET(preserve), AV_OPT_TYPE_FLOAT, {.dbl=0}, 0, 1, VF },
394  { NULL }
395 };
396 
397 AVFILTER_DEFINE_CLASS(colorcontrast);
398 
400  .name = "colorcontrast",
401  .description = NULL_IF_CONFIG_SMALL("Adjust color contrast between RGB components."),
402  .priv_size = sizeof(ColorContrastContext),
403  .priv_class = &colorcontrast_class,
409 };
static const AVFilterPad inputs[]
Definition: af_acontrast.c:193
static const AVFilterPad outputs[]
Definition: af_acontrast.c:203
static int process_command(AVFilterContext *ctx, const char *cmd, const char *args, char *res, int res_len, int flags)
Definition: af_acrusher.c:336
#define av_cold
Definition: attributes.h:88
uint8_t
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
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.
#define flags(name, subs,...)
Definition: cbs_av1.c:572
#define s(width, name)
Definition: cbs_vp9.c:257
#define f(width, name)
Definition: cbs_vp9.c:255
#define FFMIN(a, b)
Definition: common.h:105
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:302
#define av_clip_uint8
Definition: common.h:128
#define NULL
Definition: coverity.c:32
#define max(a, b)
Definition: cuda_runtime.h:33
static AVFrame * frame
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
Definition: drawutils.c:35
misc drawing utilities
int
static int ff_slice_pos(int total, int jobnr, int nb_jobs)
Compute the boundary index for a slice when work of size total is split into nb_jobs slices.
Definition: filters.h:271
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_FLOAT
Definition: opt.h:228
#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:126
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
Definition: avfilter.h:117
#define AVERROR(e)
Definition: error.h:43
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
for(j=16;j >0;--j)
misc image utilities
const char * arg
Definition: jacosubdec.c:66
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
const char * desc
Definition: libsvtav1.c:79
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
Definition: mpeg12dec.c:2033
AVOptions.
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
#define AV_PIX_FMT_GBRAP12
Definition: pixfmt.h:420
#define AV_PIX_FMT_GBRAP16
Definition: pixfmt.h:421
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:414
#define AV_PIX_FMT_BGR48
Definition: pixfmt.h:390
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:415
#define AV_PIX_FMT_RGBA64
Definition: pixfmt.h:389
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:416
#define AV_PIX_FMT_RGB48
Definition: pixfmt.h:385
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
@ AV_PIX_FMT_RGB24
packed RGB 8:8:8, 24bpp, RGBRGB...
Definition: pixfmt.h:68
@ AV_PIX_FMT_BGR0
packed BGR 8:8:8, 32bpp, BGRXBGRX... X=unused/undefined
Definition: pixfmt.h:240
@ AV_PIX_FMT_ARGB
packed ARGB 8:8:8:8, 32bpp, ARGBARGB...
Definition: pixfmt.h:92
@ AV_PIX_FMT_BGRA
packed BGRA 8:8:8:8, 32bpp, BGRABGRA...
Definition: pixfmt.h:95
@ AV_PIX_FMT_ABGR
packed ABGR 8:8:8:8, 32bpp, ABGRABGR...
Definition: pixfmt.h:94
@ AV_PIX_FMT_0BGR
packed BGR 8:8:8, 32bpp, XBGRXBGR... X=unused/undefined
Definition: pixfmt.h:239
@ AV_PIX_FMT_RGBA
packed RGBA 8:8:8:8, 32bpp, RGBARGBA...
Definition: pixfmt.h:93
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:215
@ AV_PIX_FMT_RGB0
packed RGB 8:8:8, 32bpp, RGBXRGBX... X=unused/undefined
Definition: pixfmt.h:238
@ AV_PIX_FMT_BGR24
packed RGB 8:8:8, 24bpp, BGRBGR...
Definition: pixfmt.h:69
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:168
@ AV_PIX_FMT_0RGB
packed RGB 8:8:8, 32bpp, XRGBXRGB... X=unused/undefined
Definition: pixfmt.h:237
#define AV_PIX_FMT_BGRA64
Definition: pixfmt.h:394
#define AV_PIX_FMT_GBRAP10
Definition: pixfmt.h:419
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:418
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:417
#define v0
Definition: regdef.h:26
formats
Definition: signature.h:48
Describe the class of an AVClass context structure.
Definition: log.h:67
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
Filter definition.
Definition: avfilter.h:145
const char * name
Filter name.
Definition: avfilter.h:149
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
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:332
int width
Definition: frame.h:376
int height
Definition: frame.h:376
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:349
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
int(* do_slice)(AVFilterContext *s, void *arg, int jobnr, int nb_jobs)
AVFormatContext * ctx
Definition: movenc.c:48
#define height
#define width
#define B
static int colorcontrast_slice16p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static const AVFilterPad colorcontrast_inputs[]
static const AVOption colorcontrast_options[]
#define R
static av_cold int query_formats(AVFilterContext *ctx)
static int colorcontrast_slice8(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
#define VF
static av_cold int config_input(AVFilterLink *inlink)
static int filter_frame(AVFilterLink *link, AVFrame *frame)
static float lerpf(float v0, float v1, float f)
AVFILTER_DEFINE_CLASS(colorcontrast)
#define PROCESS(max)
static const AVFilterPad colorcontrast_outputs[]
#define OFFSET(x)
AVFilter ff_vf_colorcontrast
static int colorcontrast_slice16(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
#define G
static int colorcontrast_slice8p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
const char * b
Definition: vf_curves.c:119
const char * g
Definition: vf_curves.c:118
const char * r
Definition: vf_curves.c:117
static double b1(void *priv, double x, double y)
Definition: vf_xfade.c:1665
static double b2(void *priv, double x, double y)
Definition: vf_xfade.c:1666
static double b0(void *priv, double x, double y)
Definition: vf_xfade.c:1664