Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * Copyright (c) 2003 Michael Niedermayer | ||
3 | * Copyright (c) 2012 Jeremy Tran | ||
4 | * | ||
5 | * This file is part of FFmpeg. | ||
6 | * | ||
7 | * FFmpeg is free software; you can redistribute it and/or | ||
8 | * modify it under the terms of the GNU Lesser General Public | ||
9 | * License as published by the Free Software Foundation; either | ||
10 | * version 2.1 of the License, or (at your option) any later version. | ||
11 | * | ||
12 | * FFmpeg is distributed in the hope that it will be useful, | ||
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
15 | * Lesser General Public License for more details. | ||
16 | * | ||
17 | * You should have received a copy of the GNU Lesser General Public | ||
18 | * License along with FFmpeg; if not, write to the Free Software | ||
19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
20 | */ | ||
21 | |||
22 | /** | ||
23 | * @file | ||
24 | * Apply a hue/saturation filter to the input video | ||
25 | * Ported from MPlayer libmpcodecs/vf_hue.c. | ||
26 | */ | ||
27 | |||
28 | #include <float.h> | ||
29 | #include "libavutil/eval.h" | ||
30 | #include "libavutil/imgutils.h" | ||
31 | #include "libavutil/mem.h" | ||
32 | #include "libavutil/opt.h" | ||
33 | #include "libavutil/pixdesc.h" | ||
34 | |||
35 | #include "avfilter.h" | ||
36 | #include "filters.h" | ||
37 | #include "video.h" | ||
38 | |||
39 | #define SAT_MIN_VAL -10 | ||
40 | #define SAT_MAX_VAL 10 | ||
41 | |||
42 | static const char *const var_names[] = { | ||
43 | "n", // frame count | ||
44 | "pts", // presentation timestamp expressed in AV_TIME_BASE units | ||
45 | "r", // frame rate | ||
46 | "t", // timestamp expressed in seconds | ||
47 | "tb", // timebase | ||
48 | NULL | ||
49 | }; | ||
50 | |||
51 | enum var_name { | ||
52 | VAR_N, | ||
53 | VAR_PTS, | ||
54 | VAR_R, | ||
55 | VAR_T, | ||
56 | VAR_TB, | ||
57 | VAR_NB | ||
58 | }; | ||
59 | |||
60 | typedef struct HueContext { | ||
61 | const AVClass *class; | ||
62 | float hue_deg; /* hue expressed in degrees */ | ||
63 | float hue; /* hue expressed in radians */ | ||
64 | char *hue_deg_expr; | ||
65 | char *hue_expr; | ||
66 | AVExpr *hue_deg_pexpr; | ||
67 | AVExpr *hue_pexpr; | ||
68 | float saturation; | ||
69 | char *saturation_expr; | ||
70 | AVExpr *saturation_pexpr; | ||
71 | float brightness; | ||
72 | char *brightness_expr; | ||
73 | AVExpr *brightness_pexpr; | ||
74 | int hsub; | ||
75 | int vsub; | ||
76 | int is_first; | ||
77 | int32_t hue_sin; | ||
78 | int32_t hue_cos; | ||
79 | double var_values[VAR_NB]; | ||
80 | uint8_t lut_l[256]; | ||
81 | uint8_t lut_u[256][256]; | ||
82 | uint8_t lut_v[256][256]; | ||
83 | uint16_t lut_l16[65536]; | ||
84 | uint16_t lut_u10[1024][1024]; | ||
85 | uint16_t lut_v10[1024][1024]; | ||
86 | } HueContext; | ||
87 | |||
88 | #define OFFSET(x) offsetof(HueContext, x) | ||
89 | #define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM | ||
90 | static const AVOption hue_options[] = { | ||
91 | { "h", "set the hue angle degrees expression", OFFSET(hue_deg_expr), AV_OPT_TYPE_STRING, | ||
92 | { .str = NULL }, .flags = FLAGS }, | ||
93 | { "s", "set the saturation expression", OFFSET(saturation_expr), AV_OPT_TYPE_STRING, | ||
94 | { .str = "1" }, .flags = FLAGS }, | ||
95 | { "H", "set the hue angle radians expression", OFFSET(hue_expr), AV_OPT_TYPE_STRING, | ||
96 | { .str = NULL }, .flags = FLAGS }, | ||
97 | { "b", "set the brightness expression", OFFSET(brightness_expr), AV_OPT_TYPE_STRING, | ||
98 | { .str = "0" }, .flags = FLAGS }, | ||
99 | { NULL } | ||
100 | }; | ||
101 | |||
102 | AVFILTER_DEFINE_CLASS(hue); | ||
103 | |||
104 | 88 | static inline void compute_sin_and_cos(HueContext *hue) | |
105 | { | ||
106 | /* | ||
107 | * Scale the value to the norm of the resulting (U,V) vector, that is | ||
108 | * the saturation. | ||
109 | * This will be useful in the apply_lut function. | ||
110 | */ | ||
111 | 88 | hue->hue_sin = lrint(sin(hue->hue) * (1 << 16) * hue->saturation); | |
112 | 88 | hue->hue_cos = lrint(cos(hue->hue) * (1 << 16) * hue->saturation); | |
113 | 88 | } | |
114 | |||
115 | 22 | static inline void create_luma_lut(HueContext *h) | |
116 | { | ||
117 | 22 | const float b = h->brightness; | |
118 | int i; | ||
119 | |||
120 |
2/2✓ Branch 0 taken 5632 times.
✓ Branch 1 taken 22 times.
|
5654 | for (i = 0; i < 256; i++) { |
121 | 5632 | h->lut_l[i] = av_clip_uint8(i + b * 25.5); | |
122 | } | ||
123 |
2/2✓ Branch 0 taken 1441792 times.
✓ Branch 1 taken 22 times.
|
1441814 | for (i = 0; i < 65536; i++) { |
124 | 1441792 | h->lut_l16[i] = av_clip_uintp2(i + b * 102.4, 10); | |
125 | } | ||
126 | 22 | } | |
127 | |||
128 | 61 | static inline void create_chrominance_lut(HueContext *h, const int32_t c, | |
129 | const int32_t s) | ||
130 | { | ||
131 | int32_t i, j, u, v, new_u, new_v; | ||
132 | |||
133 | /* | ||
134 | * If we consider U and V as the components of a 2D vector then its angle | ||
135 | * is the hue and the norm is the saturation | ||
136 | */ | ||
137 |
2/2✓ Branch 0 taken 15616 times.
✓ Branch 1 taken 61 times.
|
15677 | for (i = 0; i < 256; i++) { |
138 |
2/2✓ Branch 0 taken 3997696 times.
✓ Branch 1 taken 15616 times.
|
4013312 | for (j = 0; j < 256; j++) { |
139 | /* Normalize the components from range [16;240] to [-112;112] */ | ||
140 | 3997696 | u = i - 128; | |
141 | 3997696 | v = j - 128; | |
142 | /* | ||
143 | * Apply the rotation of the vector : (c * u) - (s * v) | ||
144 | * (s * u) + (c * v) | ||
145 | * De-normalize the components (without forgetting to scale 128 | ||
146 | * by << 16) | ||
147 | * Finally scale back the result by >> 16 | ||
148 | */ | ||
149 | 3997696 | new_u = ((c * u) - (s * v) + (1 << 15) + (128 << 16)) >> 16; | |
150 | 3997696 | new_v = ((s * u) + (c * v) + (1 << 15) + (128 << 16)) >> 16; | |
151 | |||
152 | /* Prevent a potential overflow */ | ||
153 | 3997696 | h->lut_u[i][j] = av_clip_uint8(new_u); | |
154 | 3997696 | h->lut_v[i][j] = av_clip_uint8(new_v); | |
155 | } | ||
156 | } | ||
157 |
2/2✓ Branch 0 taken 62464 times.
✓ Branch 1 taken 61 times.
|
62525 | for (i = 0; i < 1024; i++) { |
158 |
2/2✓ Branch 0 taken 63963136 times.
✓ Branch 1 taken 62464 times.
|
64025600 | for (j = 0; j < 1024; j++) { |
159 | 63963136 | u = i - 512; | |
160 | 63963136 | v = j - 512; | |
161 | /* | ||
162 | * Apply the rotation of the vector : (c * u) - (s * v) | ||
163 | * (s * u) + (c * v) | ||
164 | * De-normalize the components (without forgetting to scale 512 | ||
165 | * by << 16) | ||
166 | * Finally scale back the result by >> 16 | ||
167 | */ | ||
168 | 63963136 | new_u = ((c * u) - (s * v) + (1 << 15) + (512 << 16)) >> 16; | |
169 | 63963136 | new_v = ((s * u) + (c * v) + (1 << 15) + (512 << 16)) >> 16; | |
170 | |||
171 | /* Prevent a potential overflow */ | ||
172 | 63963136 | h->lut_u10[i][j] = av_clip_uintp2(new_u, 10); | |
173 | 63963136 | h->lut_v10[i][j] = av_clip_uintp2(new_v, 10); | |
174 | } | ||
175 | } | ||
176 | 61 | } | |
177 | |||
178 | 20 | static int set_expr(AVExpr **pexpr_ptr, char **expr_ptr, | |
179 | const char *expr, const char *option, void *log_ctx) | ||
180 | { | ||
181 | int ret; | ||
182 | AVExpr *new_pexpr; | ||
183 | char *new_expr; | ||
184 | |||
185 | 20 | new_expr = av_strdup(expr); | |
186 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
|
20 | if (!new_expr) |
187 | ✗ | return AVERROR(ENOMEM); | |
188 | 20 | ret = av_expr_parse(&new_pexpr, expr, var_names, | |
189 | NULL, NULL, NULL, NULL, 0, log_ctx); | ||
190 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
|
20 | if (ret < 0) { |
191 | ✗ | av_log(log_ctx, AV_LOG_ERROR, | |
192 | "Error when evaluating the expression '%s' for %s\n", | ||
193 | expr, option); | ||
194 | ✗ | av_free(new_expr); | |
195 | ✗ | return ret; | |
196 | } | ||
197 | |||
198 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
|
20 | if (*pexpr_ptr) |
199 | ✗ | av_expr_free(*pexpr_ptr); | |
200 | 20 | *pexpr_ptr = new_pexpr; | |
201 | 20 | av_freep(expr_ptr); | |
202 | 20 | *expr_ptr = new_expr; | |
203 | |||
204 | 20 | return 0; | |
205 | } | ||
206 | |||
207 | 8 | static av_cold int init(AVFilterContext *ctx) | |
208 | { | ||
209 | 8 | HueContext *hue = ctx->priv; | |
210 | int ret; | ||
211 | |||
212 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
8 | if (hue->hue_expr && hue->hue_deg_expr) { |
213 | ✗ | av_log(ctx, AV_LOG_ERROR, | |
214 | "H and h options are incompatible and cannot be specified " | ||
215 | "at the same time\n"); | ||
216 | ✗ | return AVERROR(EINVAL); | |
217 | } | ||
218 | |||
219 | #define SET_EXPR(expr, option) \ | ||
220 | if (hue->expr##_expr) do { \ | ||
221 | ret = set_expr(&hue->expr##_pexpr, &hue->expr##_expr, \ | ||
222 | hue->expr##_expr, option, ctx); \ | ||
223 | if (ret < 0) \ | ||
224 | return ret; \ | ||
225 | } while (0) | ||
226 |
2/4✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 8 times.
|
8 | SET_EXPR(brightness, "b"); |
227 |
2/4✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 8 times.
|
8 | SET_EXPR(saturation, "s"); |
228 |
3/4✓ Branch 0 taken 4 times.
✓ Branch 1 taken 4 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 4 times.
|
8 | SET_EXPR(hue_deg, "h"); |
229 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
|
8 | SET_EXPR(hue, "H"); |
230 | #undef SET_EXPR | ||
231 | |||
232 | 8 | av_log(ctx, AV_LOG_VERBOSE, | |
233 | "H_expr:%s h_deg_expr:%s s_expr:%s b_expr:%s\n", | ||
234 | hue->hue_expr, hue->hue_deg_expr, hue->saturation_expr, hue->brightness_expr); | ||
235 | 8 | compute_sin_and_cos(hue); | |
236 | 8 | hue->is_first = 1; | |
237 | |||
238 | 8 | return 0; | |
239 | } | ||
240 | |||
241 | 8 | static av_cold void uninit(AVFilterContext *ctx) | |
242 | { | ||
243 | 8 | HueContext *hue = ctx->priv; | |
244 | |||
245 | 8 | av_expr_free(hue->brightness_pexpr); | |
246 | 8 | av_expr_free(hue->hue_deg_pexpr); | |
247 | 8 | av_expr_free(hue->hue_pexpr); | |
248 | 8 | av_expr_free(hue->saturation_pexpr); | |
249 | 8 | } | |
250 | |||
251 | static const enum AVPixelFormat pix_fmts[] = { | ||
252 | AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV422P, | ||
253 | AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV411P, | ||
254 | AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV440P, | ||
255 | AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUVA422P, | ||
256 | AV_PIX_FMT_YUVA420P, | ||
257 | AV_PIX_FMT_YUV444P10, AV_PIX_FMT_YUV422P10, | ||
258 | AV_PIX_FMT_YUV420P10, | ||
259 | AV_PIX_FMT_YUV440P10, | ||
260 | AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_YUVA422P10, | ||
261 | AV_PIX_FMT_YUVA420P10, | ||
262 | AV_PIX_FMT_NONE | ||
263 | }; | ||
264 | |||
265 | 4 | static int config_props(AVFilterLink *inlink) | |
266 | { | ||
267 | 4 | HueContext *hue = inlink->dst->priv; | |
268 | 4 | FilterLink *l = ff_filter_link(inlink); | |
269 | 4 | const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format); | |
270 | |||
271 | 4 | hue->hsub = desc->log2_chroma_w; | |
272 | 4 | hue->vsub = desc->log2_chroma_h; | |
273 | |||
274 | 4 | hue->var_values[VAR_N] = 0; | |
275 | 4 | hue->var_values[VAR_TB] = av_q2d(inlink->time_base); | |
276 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | hue->var_values[VAR_R] = l->frame_rate.num == 0 || l->frame_rate.den == 0 ? |
277 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
8 | NAN : av_q2d(l->frame_rate); |
278 | |||
279 | 4 | return 0; | |
280 | } | ||
281 | |||
282 | 19 | static void apply_luma_lut(HueContext *s, | |
283 | uint8_t *ldst, const int dst_linesize, | ||
284 | uint8_t *lsrc, const int src_linesize, | ||
285 | int w, int h) | ||
286 | { | ||
287 | int i; | ||
288 | |||
289 |
2/2✓ Branch 0 taken 5472 times.
✓ Branch 1 taken 19 times.
|
5491 | while (h--) { |
290 |
2/2✓ Branch 0 taken 1926144 times.
✓ Branch 1 taken 5472 times.
|
1931616 | for (i = 0; i < w; i++) |
291 | 1926144 | ldst[i] = s->lut_l[lsrc[i]]; | |
292 | |||
293 | 5472 | lsrc += src_linesize; | |
294 | 5472 | ldst += dst_linesize; | |
295 | } | ||
296 | 19 | } | |
297 | |||
298 | ✗ | static void apply_luma_lut10(HueContext *s, | |
299 | uint16_t *ldst, const int dst_linesize, | ||
300 | uint16_t *lsrc, const int src_linesize, | ||
301 | int w, int h) | ||
302 | { | ||
303 | int i; | ||
304 | |||
305 | ✗ | while (h--) { | |
306 | ✗ | for (i = 0; i < w; i++) | |
307 | ✗ | ldst[i] = s->lut_l16[lsrc[i]]; | |
308 | |||
309 | ✗ | lsrc += src_linesize; | |
310 | ✗ | ldst += dst_linesize; | |
311 | } | ||
312 | ✗ | } | |
313 | |||
314 | 60 | static void apply_lut(HueContext *s, | |
315 | uint8_t *udst, uint8_t *vdst, const int dst_linesize, | ||
316 | uint8_t *usrc, uint8_t *vsrc, const int src_linesize, | ||
317 | int w, int h) | ||
318 | { | ||
319 | int i; | ||
320 | |||
321 |
2/2✓ Branch 0 taken 8640 times.
✓ Branch 1 taken 60 times.
|
8700 | while (h--) { |
322 |
2/2✓ Branch 0 taken 1520640 times.
✓ Branch 1 taken 8640 times.
|
1529280 | for (i = 0; i < w; i++) { |
323 | 1520640 | const int u = usrc[i]; | |
324 | 1520640 | const int v = vsrc[i]; | |
325 | |||
326 | 1520640 | udst[i] = s->lut_u[u][v]; | |
327 | 1520640 | vdst[i] = s->lut_v[u][v]; | |
328 | } | ||
329 | |||
330 | 8640 | usrc += src_linesize; | |
331 | 8640 | vsrc += src_linesize; | |
332 | 8640 | udst += dst_linesize; | |
333 | 8640 | vdst += dst_linesize; | |
334 | } | ||
335 | 60 | } | |
336 | |||
337 | 20 | static void apply_lut10(HueContext *s, | |
338 | uint16_t *udst, uint16_t *vdst, const int dst_linesize, | ||
339 | uint16_t *usrc, uint16_t *vsrc, const int src_linesize, | ||
340 | int w, int h) | ||
341 | { | ||
342 | int i; | ||
343 | |||
344 |
2/2✓ Branch 0 taken 5760 times.
✓ Branch 1 taken 20 times.
|
5780 | while (h--) { |
345 |
2/2✓ Branch 0 taken 1013760 times.
✓ Branch 1 taken 5760 times.
|
1019520 | for (i = 0; i < w; i++) { |
346 | 1013760 | const int u = av_clip_uintp2(usrc[i], 10); | |
347 | 1013760 | const int v = av_clip_uintp2(vsrc[i], 10); | |
348 | |||
349 | 1013760 | udst[i] = s->lut_u10[u][v]; | |
350 | 1013760 | vdst[i] = s->lut_v10[u][v]; | |
351 | } | ||
352 | |||
353 | 5760 | usrc += src_linesize; | |
354 | 5760 | vsrc += src_linesize; | |
355 | 5760 | udst += dst_linesize; | |
356 | 5760 | vdst += dst_linesize; | |
357 | } | ||
358 | 20 | } | |
359 | |||
360 | 80 | static int filter_frame(AVFilterLink *inlink, AVFrame *inpic) | |
361 | { | ||
362 | 80 | FilterLink *inl = ff_filter_link(inlink); | |
363 | 80 | HueContext *hue = inlink->dst->priv; | |
364 | 80 | AVFilterLink *outlink = inlink->dst->outputs[0]; | |
365 | AVFrame *outpic; | ||
366 | 80 | const int32_t old_hue_sin = hue->hue_sin, old_hue_cos = hue->hue_cos; | |
367 | 80 | const float old_brightness = hue->brightness; | |
368 | 80 | int direct = 0; | |
369 | 80 | const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format); | |
370 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 60 times.
|
80 | const int bps = desc->comp[0].depth > 8 ? 2 : 1; |
371 | |||
372 |
1/2✓ Branch 1 taken 80 times.
✗ Branch 2 not taken.
|
80 | if (av_frame_is_writable(inpic)) { |
373 | 80 | direct = 1; | |
374 | 80 | outpic = inpic; | |
375 | } else { | ||
376 | ✗ | outpic = ff_get_video_buffer(outlink, outlink->w, outlink->h); | |
377 | ✗ | if (!outpic) { | |
378 | ✗ | av_frame_free(&inpic); | |
379 | ✗ | return AVERROR(ENOMEM); | |
380 | } | ||
381 | ✗ | av_frame_copy_props(outpic, inpic); | |
382 | } | ||
383 | |||
384 | 80 | hue->var_values[VAR_N] = inl->frame_count_out; | |
385 |
1/2✓ Branch 0 taken 80 times.
✗ Branch 1 not taken.
|
80 | hue->var_values[VAR_T] = TS2T(inpic->pts, inlink->time_base); |
386 |
1/2✓ Branch 0 taken 80 times.
✗ Branch 1 not taken.
|
80 | hue->var_values[VAR_PTS] = TS2D(inpic->pts); |
387 | |||
388 |
1/2✓ Branch 0 taken 80 times.
✗ Branch 1 not taken.
|
80 | if (hue->saturation_expr) { |
389 | 80 | hue->saturation = av_expr_eval(hue->saturation_pexpr, hue->var_values, NULL); | |
390 | |||
391 |
2/4✓ Branch 0 taken 80 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 80 times.
|
80 | if (hue->saturation < SAT_MIN_VAL || hue->saturation > SAT_MAX_VAL) { |
392 | ✗ | hue->saturation = av_clip(hue->saturation, SAT_MIN_VAL, SAT_MAX_VAL); | |
393 | ✗ | av_log(inlink->dst, AV_LOG_WARNING, | |
394 | "Saturation value not in range [%d,%d]: clipping value to %0.1f\n", | ||
395 | ✗ | SAT_MIN_VAL, SAT_MAX_VAL, hue->saturation); | |
396 | } | ||
397 | } | ||
398 | |||
399 |
1/2✓ Branch 0 taken 80 times.
✗ Branch 1 not taken.
|
80 | if (hue->brightness_expr) { |
400 | 80 | hue->brightness = av_expr_eval(hue->brightness_pexpr, hue->var_values, NULL); | |
401 | |||
402 |
2/4✓ Branch 0 taken 80 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 80 times.
|
80 | if (hue->brightness < -10 || hue->brightness > 10) { |
403 | ✗ | hue->brightness = av_clipf(hue->brightness, -10, 10); | |
404 | ✗ | av_log(inlink->dst, AV_LOG_WARNING, | |
405 | "Brightness value not in range [%d,%d]: clipping value to %0.1f\n", | ||
406 | ✗ | -10, 10, hue->brightness); | |
407 | } | ||
408 | } | ||
409 | |||
410 |
2/2✓ Branch 0 taken 40 times.
✓ Branch 1 taken 40 times.
|
80 | if (hue->hue_deg_expr) { |
411 | 40 | hue->hue_deg = av_expr_eval(hue->hue_deg_pexpr, hue->var_values, NULL); | |
412 | 40 | hue->hue = hue->hue_deg * M_PI / 180; | |
413 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 40 times.
|
40 | } else if (hue->hue_expr) { |
414 | ✗ | hue->hue = av_expr_eval(hue->hue_pexpr, hue->var_values, NULL); | |
415 | ✗ | hue->hue_deg = hue->hue * 180 / M_PI; | |
416 | } | ||
417 | |||
418 | 80 | av_log(inlink->dst, AV_LOG_DEBUG, | |
419 | "H:%0.1f*PI h:%0.1f s:%0.1f b:%0.f t:%0.1f n:%d\n", | ||
420 | 80 | hue->hue/M_PI, hue->hue_deg, hue->saturation, hue->brightness, | |
421 | 80 | hue->var_values[VAR_T], (int)hue->var_values[VAR_N]); | |
422 | |||
423 | 80 | compute_sin_and_cos(hue); | |
424 |
6/6✓ Branch 0 taken 76 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 38 times.
✓ Branch 3 taken 38 times.
✓ Branch 4 taken 19 times.
✓ Branch 5 taken 19 times.
|
80 | if (hue->is_first || (old_hue_sin != hue->hue_sin || old_hue_cos != hue->hue_cos)) |
425 | 61 | create_chrominance_lut(hue, hue->hue_cos, hue->hue_sin); | |
426 | |||
427 |
6/6✓ Branch 0 taken 76 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 19 times.
✓ Branch 3 taken 57 times.
✓ Branch 4 taken 18 times.
✓ Branch 5 taken 1 times.
|
80 | if (hue->is_first || (old_brightness != hue->brightness && hue->brightness)) |
428 | 22 | create_luma_lut(hue); | |
429 | |||
430 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 80 times.
|
80 | if (!direct) { |
431 | ✗ | if (!hue->brightness) | |
432 | ✗ | av_image_copy_plane(outpic->data[0], outpic->linesize[0], | |
433 | ✗ | inpic->data[0], inpic->linesize[0], | |
434 | ✗ | inlink->w * bps, inlink->h); | |
435 | ✗ | if (inpic->data[3]) | |
436 | ✗ | av_image_copy_plane(outpic->data[3], outpic->linesize[3], | |
437 | ✗ | inpic->data[3], inpic->linesize[3], | |
438 | ✗ | inlink->w * bps, inlink->h); | |
439 | } | ||
440 | |||
441 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 60 times.
|
80 | if (bps > 1) { |
442 | 20 | apply_lut10(hue, (uint16_t*)outpic->data[1], (uint16_t*)outpic->data[2], outpic->linesize[1]/2, | |
443 | 20 | (uint16_t*) inpic->data[1], (uint16_t*) inpic->data[2], inpic->linesize[1]/2, | |
444 | 20 | AV_CEIL_RSHIFT(inlink->w, hue->hsub), | |
445 | 20 | AV_CEIL_RSHIFT(inlink->h, hue->vsub)); | |
446 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
|
20 | if (hue->brightness) |
447 | ✗ | apply_luma_lut10(hue, (uint16_t*)outpic->data[0], outpic->linesize[0]/2, | |
448 | ✗ | (uint16_t*) inpic->data[0], inpic->linesize[0]/2, inlink->w, inlink->h); | |
449 | } else { | ||
450 | 60 | apply_lut(hue, outpic->data[1], outpic->data[2], outpic->linesize[1], | |
451 | 60 | inpic->data[1], inpic->data[2], inpic->linesize[1], | |
452 | 60 | AV_CEIL_RSHIFT(inlink->w, hue->hsub), | |
453 | 60 | AV_CEIL_RSHIFT(inlink->h, hue->vsub)); | |
454 |
2/2✓ Branch 0 taken 19 times.
✓ Branch 1 taken 41 times.
|
60 | if (hue->brightness) |
455 | 19 | apply_luma_lut(hue, outpic->data[0], outpic->linesize[0], | |
456 | 19 | inpic->data[0], inpic->linesize[0], inlink->w, inlink->h); | |
457 | } | ||
458 | |||
459 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 80 times.
|
80 | if (!direct) |
460 | ✗ | av_frame_free(&inpic); | |
461 | |||
462 | 80 | hue->is_first = 0; | |
463 | 80 | return ff_filter_frame(outlink, outpic); | |
464 | } | ||
465 | |||
466 | ✗ | static int process_command(AVFilterContext *ctx, const char *cmd, const char *args, | |
467 | char *res, int res_len, int flags) | ||
468 | { | ||
469 | ✗ | HueContext *hue = ctx->priv; | |
470 | int ret; | ||
471 | |||
472 | #define SET_EXPR(expr, option) \ | ||
473 | do { \ | ||
474 | ret = set_expr(&hue->expr##_pexpr, &hue->expr##_expr, \ | ||
475 | args, option, ctx); \ | ||
476 | if (ret < 0) \ | ||
477 | return ret; \ | ||
478 | } while (0) | ||
479 | |||
480 | ✗ | if (!strcmp(cmd, "h")) { | |
481 | ✗ | SET_EXPR(hue_deg, "h"); | |
482 | ✗ | av_freep(&hue->hue_expr); | |
483 | ✗ | } else if (!strcmp(cmd, "H")) { | |
484 | ✗ | SET_EXPR(hue, "H"); | |
485 | ✗ | av_freep(&hue->hue_deg_expr); | |
486 | ✗ | } else if (!strcmp(cmd, "s")) { | |
487 | ✗ | SET_EXPR(saturation, "s"); | |
488 | ✗ | } else if (!strcmp(cmd, "b")) { | |
489 | ✗ | SET_EXPR(brightness, "b"); | |
490 | } else | ||
491 | ✗ | return AVERROR(ENOSYS); | |
492 | |||
493 | ✗ | return 0; | |
494 | } | ||
495 | |||
496 | static const AVFilterPad hue_inputs[] = { | ||
497 | { | ||
498 | .name = "default", | ||
499 | .type = AVMEDIA_TYPE_VIDEO, | ||
500 | .filter_frame = filter_frame, | ||
501 | .config_props = config_props, | ||
502 | }, | ||
503 | }; | ||
504 | |||
505 | const AVFilter ff_vf_hue = { | ||
506 | .name = "hue", | ||
507 | .description = NULL_IF_CONFIG_SMALL("Adjust the hue and saturation of the input video."), | ||
508 | .priv_size = sizeof(HueContext), | ||
509 | .init = init, | ||
510 | .uninit = uninit, | ||
511 | .process_command = process_command, | ||
512 | FILTER_INPUTS(hue_inputs), | ||
513 | FILTER_OUTPUTS(ff_video_default_filterpad), | ||
514 | FILTER_PIXFMTS_ARRAY(pix_fmts), | ||
515 | .priv_class = &hue_class, | ||
516 | .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, | ||
517 | }; | ||
518 |