FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavfilter/vf_vectorscope.c
Date: 2024-11-20 23:03:26
Exec Total Coverage
Lines: 209 891 23.5%
Functions: 8 28 28.6%
Branches: 107 640 16.7%
Line Branch Exec Source
1 /*
2 * Copyright (c) 2015 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/avassert.h"
22 #include "libavutil/intreadwrite.h"
23 #include "libavutil/mem.h"
24 #include "libavutil/opt.h"
25 #include "libavutil/pixdesc.h"
26 #include "libavutil/xga_font_data.h"
27 #include "avfilter.h"
28 #include "filters.h"
29 #include "formats.h"
30 #include "video.h"
31
32 enum GraticuleType {
33 GRAT_NONE,
34 GRAT_GREEN,
35 GRAT_COLOR,
36 GRAT_INVERT,
37 NB_GRATICULES
38 };
39
40 enum VectorscopeMode {
41 TINT,
42 COLOR,
43 COLOR2,
44 COLOR3,
45 COLOR4,
46 COLOR5,
47 MODE_NB
48 };
49
50 typedef struct VectorscopeContext {
51 const AVClass *class;
52 int mode;
53 int intensity;
54 float fintensity;
55 uint16_t bg_color[4];
56 float ftint[2];
57 int planewidth[4];
58 int planeheight[4];
59 int hsub, vsub;
60 int x, y, pd;
61 int is_yuv;
62 int size;
63 int depth;
64 int mult;
65 int envelope;
66 int graticule;
67 float opacity;
68 float bgopacity;
69 float lthreshold;
70 float hthreshold;
71 int tint[2];
72 int tmin;
73 int tmax;
74 int flags;
75 int colorspace;
76 int cs;
77 uint8_t *peak_memory;
78 uint8_t **peak;
79
80 void (*vectorscope)(struct VectorscopeContext *s,
81 AVFrame *in, AVFrame *out, int pd);
82 void (*graticulef)(struct VectorscopeContext *s, AVFrame *out,
83 int X, int Y, int D, int P);
84 } VectorscopeContext;
85
86 #define OFFSET(x) offsetof(VectorscopeContext, x)
87 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
88 #define TFLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
89
90 static const AVOption vectorscope_options[] = {
91 { "mode", "set vectorscope mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64=0}, 0, MODE_NB-1, FLAGS, .unit = "mode"},
92 { "m", "set vectorscope mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64=0}, 0, MODE_NB-1, FLAGS, .unit = "mode"},
93 { "gray", 0, 0, AV_OPT_TYPE_CONST, {.i64=TINT}, 0, 0, FLAGS, .unit = "mode" },
94 { "tint", 0, 0, AV_OPT_TYPE_CONST, {.i64=TINT}, 0, 0, FLAGS, .unit = "mode" },
95 { "color", 0, 0, AV_OPT_TYPE_CONST, {.i64=COLOR}, 0, 0, FLAGS, .unit = "mode" },
96 { "color2", 0, 0, AV_OPT_TYPE_CONST, {.i64=COLOR2}, 0, 0, FLAGS, .unit = "mode" },
97 { "color3", 0, 0, AV_OPT_TYPE_CONST, {.i64=COLOR3}, 0, 0, FLAGS, .unit = "mode" },
98 { "color4", 0, 0, AV_OPT_TYPE_CONST, {.i64=COLOR4}, 0, 0, FLAGS, .unit = "mode" },
99 { "color5", 0, 0, AV_OPT_TYPE_CONST, {.i64=COLOR5}, 0, 0, FLAGS, .unit = "mode" },
100 { "x", "set color component on X axis", OFFSET(x), AV_OPT_TYPE_INT, {.i64=1}, 0, 2, FLAGS},
101 { "y", "set color component on Y axis", OFFSET(y), AV_OPT_TYPE_INT, {.i64=2}, 0, 2, FLAGS},
102 { "intensity", "set intensity", OFFSET(fintensity), AV_OPT_TYPE_FLOAT, {.dbl=0.004}, 0, 1, TFLAGS},
103 { "i", "set intensity", OFFSET(fintensity), AV_OPT_TYPE_FLOAT, {.dbl=0.004}, 0, 1, TFLAGS},
104 { "envelope", "set envelope", OFFSET(envelope), AV_OPT_TYPE_INT, {.i64=0}, 0, 3, TFLAGS, .unit = "envelope"},
105 { "e", "set envelope", OFFSET(envelope), AV_OPT_TYPE_INT, {.i64=0}, 0, 3, TFLAGS, .unit = "envelope"},
106 { "none", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, TFLAGS, .unit = "envelope" },
107 { "instant", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, TFLAGS, .unit = "envelope" },
108 { "peak", 0, 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, TFLAGS, .unit = "envelope" },
109 { "peak+instant", 0, 0, AV_OPT_TYPE_CONST, {.i64=3}, 0, 0, TFLAGS, .unit = "envelope" },
110 { "graticule", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=GRAT_NONE}, 0, NB_GRATICULES-1, FLAGS, .unit = "graticule"},
111 { "g", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=GRAT_NONE}, 0, NB_GRATICULES-1, FLAGS, .unit = "graticule"},
112 { "none", 0, 0, AV_OPT_TYPE_CONST, {.i64=GRAT_NONE}, 0, 0, FLAGS, .unit = "graticule" },
113 { "green", 0, 0, AV_OPT_TYPE_CONST, {.i64=GRAT_GREEN}, 0, 0, FLAGS, .unit = "graticule" },
114 { "color", 0, 0, AV_OPT_TYPE_CONST, {.i64=GRAT_COLOR}, 0, 0, FLAGS, .unit = "graticule" },
115 { "invert", 0, 0, AV_OPT_TYPE_CONST, {.i64=GRAT_INVERT},0, 0, FLAGS, .unit = "graticule" },
116 { "opacity", "set graticule opacity", OFFSET(opacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, TFLAGS},
117 { "o", "set graticule opacity", OFFSET(opacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, TFLAGS},
118 { "flags", "set graticule flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64=4}, 0, 7, TFLAGS, .unit = "flags"},
119 { "f", "set graticule flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64=4}, 0, 7, TFLAGS, .unit = "flags"},
120 { "white", "draw white point", 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, TFLAGS, .unit = "flags" },
121 { "black", "draw black point", 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, TFLAGS, .unit = "flags" },
122 { "name", "draw point name", 0, AV_OPT_TYPE_CONST, {.i64=4}, 0, 0, TFLAGS, .unit = "flags" },
123 { "bgopacity", "set background opacity", OFFSET(bgopacity), AV_OPT_TYPE_FLOAT, {.dbl=0.3}, 0, 1, TFLAGS},
124 { "b", "set background opacity", OFFSET(bgopacity), AV_OPT_TYPE_FLOAT, {.dbl=0.3}, 0, 1, TFLAGS},
125 { "lthreshold", "set low threshold", OFFSET(lthreshold), AV_OPT_TYPE_FLOAT, {.dbl=0}, 0, 1, FLAGS},
126 { "l", "set low threshold", OFFSET(lthreshold), AV_OPT_TYPE_FLOAT, {.dbl=0}, 0, 1, FLAGS},
127 { "hthreshold", "set high threshold", OFFSET(hthreshold), AV_OPT_TYPE_FLOAT, {.dbl=1}, 0, 1, FLAGS},
128 { "h", "set high threshold", OFFSET(hthreshold), AV_OPT_TYPE_FLOAT, {.dbl=1}, 0, 1, FLAGS},
129 { "colorspace", "set colorspace", OFFSET(colorspace), AV_OPT_TYPE_INT, {.i64=0}, 0, 2, FLAGS, .unit = "colorspace"},
130 { "c", "set colorspace", OFFSET(colorspace), AV_OPT_TYPE_INT, {.i64=0}, 0, 2, FLAGS, .unit = "colorspace"},
131 { "auto", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, .unit = "colorspace" },
132 { "601", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, .unit = "colorspace" },
133 { "709", 0, 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, FLAGS, .unit = "colorspace" },
134 { "tint0", "set 1st tint", OFFSET(ftint[0]), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, TFLAGS},
135 { "t0", "set 1st tint", OFFSET(ftint[0]), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, TFLAGS},
136 { "tint1", "set 2nd tint", OFFSET(ftint[1]), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, TFLAGS},
137 { "t1", "set 2nd tint", OFFSET(ftint[1]), AV_OPT_TYPE_FLOAT, {.dbl=0}, -1, 1, TFLAGS},
138 { NULL }
139 };
140
141 AVFILTER_DEFINE_CLASS(vectorscope);
142
143 static const enum AVPixelFormat out_yuv8_pix_fmts[] = {
144 AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUV444P,
145 AV_PIX_FMT_NONE
146 };
147
148 static const enum AVPixelFormat out_yuv9_pix_fmts[] = {
149 AV_PIX_FMT_YUVA444P9, AV_PIX_FMT_YUV444P9,
150 AV_PIX_FMT_NONE
151 };
152
153 static const enum AVPixelFormat out_yuv10_pix_fmts[] = {
154 AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_YUV444P10,
155 AV_PIX_FMT_NONE
156 };
157
158 static const enum AVPixelFormat out_yuv12_pix_fmts[] = {
159 AV_PIX_FMT_YUVA444P12, AV_PIX_FMT_YUV444P12,
160 AV_PIX_FMT_NONE
161 };
162
163 static const enum AVPixelFormat out_rgb8_pix_fmts[] = {
164 AV_PIX_FMT_GBRAP, AV_PIX_FMT_GBRP,
165 AV_PIX_FMT_NONE
166 };
167
168 static const enum AVPixelFormat out_rgb9_pix_fmts[] = {
169 AV_PIX_FMT_GBRP9,
170 AV_PIX_FMT_NONE
171 };
172
173 static const enum AVPixelFormat out_rgb10_pix_fmts[] = {
174 AV_PIX_FMT_GBRP10, AV_PIX_FMT_GBRAP10,
175 AV_PIX_FMT_NONE
176 };
177
178 static const enum AVPixelFormat out_rgb12_pix_fmts[] = {
179 AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRAP12,
180 AV_PIX_FMT_NONE
181 };
182
183 static const enum AVPixelFormat in1_pix_fmts[] = {
184 AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUVJ444P,
185 AV_PIX_FMT_YUV444P9, AV_PIX_FMT_YUV444P10,
186 AV_PIX_FMT_YUVA444P9, AV_PIX_FMT_YUVA444P10,
187 AV_PIX_FMT_YUVA444P12, AV_PIX_FMT_YUV444P12,
188 AV_PIX_FMT_GBRAP, AV_PIX_FMT_GBRP,
189 AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10, AV_PIX_FMT_GBRAP10,
190 AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRAP12,
191 AV_PIX_FMT_NONE
192 };
193
194 static const enum AVPixelFormat in2_pix_fmts[] = {
195 AV_PIX_FMT_YUVA420P, AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUVJ420P,
196 AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUVJ422P,
197 AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUVJ444P,
198 AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUVJ411P,
199 AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV410P,
200 AV_PIX_FMT_GBRAP, AV_PIX_FMT_GBRP,
201 AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10, AV_PIX_FMT_GBRAP10,
202 AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRAP12,
203 AV_PIX_FMT_YUV420P9, AV_PIX_FMT_YUV422P9, AV_PIX_FMT_YUV444P9,
204 AV_PIX_FMT_YUV420P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV444P10,
205 AV_PIX_FMT_YUVA420P9, AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA444P9,
206 AV_PIX_FMT_YUVA420P10, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA444P10,
207 AV_PIX_FMT_YUV420P12, AV_PIX_FMT_YUV422P12, AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV440P12,
208 AV_PIX_FMT_YUVA422P12, AV_PIX_FMT_YUVA444P12,
209 AV_PIX_FMT_NONE
210 };
211
212 12 static int query_formats(AVFilterContext *ctx)
213 {
214 12 VectorscopeContext *s = ctx->priv;
215 const enum AVPixelFormat *out_pix_fmts;
216 const AVPixFmtDescriptor *desc;
217 AVFilterFormats *avff;
218 int depth, rgb, i, ret;
219
220
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 6 times.
 12 if (!ctx->inputs[0]->incfg.formats ||
221
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 !ctx->inputs[0]->incfg.formats->nb_formats) {
222 6 return AVERROR(EAGAIN);
223 }
224
225
1/2
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
 6 if (!ctx->inputs[0]->outcfg.formats) {
226 const enum AVPixelFormat *in_pix_fmts;
227
228
4/8
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 5 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 6 if ((s->x == 1 && s->y == 2) || (s->x == 2 && s->y == 1))
229 5 in_pix_fmts = in2_pix_fmts;
230 else
231 1 in_pix_fmts = in1_pix_fmts;
232
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
 6 if ((ret = ff_formats_ref(ff_make_format_list(in_pix_fmts), &ctx->inputs[0]->outcfg.formats)) < 0)
233 ✗ return ret;
234 }
235
236 6 avff = ctx->inputs[0]->incfg.formats;
237 6 desc = av_pix_fmt_desc_get(avff->formats[0]);
238 6 rgb = desc->flags & AV_PIX_FMT_FLAG_RGB;
239 6 depth = desc->comp[0].depth;
240
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 for (i = 1; i < avff->nb_formats; i++) {
241 ✗ desc = av_pix_fmt_desc_get(avff->formats[i]);
242 ✗ if (rgb != (desc->flags & AV_PIX_FMT_FLAG_RGB) ||
243 ✗ depth != desc->comp[0].depth)
244 ✗ return AVERROR(EAGAIN);
245 }
246
247
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 6 if (rgb && depth == 8)
248 ✗ out_pix_fmts = out_rgb8_pix_fmts;
249
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 6 else if (rgb && depth == 9)
250 ✗ out_pix_fmts = out_rgb9_pix_fmts;
251
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 6 else if (rgb && depth == 10)
252 ✗ out_pix_fmts = out_rgb10_pix_fmts;
253
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 6 else if (rgb && depth == 12)
254 ✗ out_pix_fmts = out_rgb12_pix_fmts;
255
1/2
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
 6 else if (depth == 8)
256 6 out_pix_fmts = out_yuv8_pix_fmts;
257 ✗ else if (depth == 9)
258 ✗ out_pix_fmts = out_yuv9_pix_fmts;
259 ✗ else if (depth == 10)
260 ✗ out_pix_fmts = out_yuv10_pix_fmts;
261 ✗ else if (depth == 12)
262 ✗ out_pix_fmts = out_yuv12_pix_fmts;
263 else
264 ✗ return AVERROR(EAGAIN);
265
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
 6 if ((ret = ff_formats_ref(ff_make_format_list(out_pix_fmts), &ctx->outputs[0]->incfg.formats)) < 0)
266 ✗ return ret;
267
268 6 return 0;
269 }
270
271 6 static int config_output(AVFilterLink *outlink)
272 {
273 6 VectorscopeContext *s = outlink->src->priv;
274 int i;
275
276 6 outlink->h = outlink->w = s->size;
277 6 outlink->sample_aspect_ratio = (AVRational){1,1};
278
279 6 s->peak_memory = av_calloc(s->size, s->size);
280
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (!s->peak_memory)
281 ✗ return AVERROR(ENOMEM);
282
283 6 s->peak = av_calloc(s->size, sizeof(*s->peak));
284
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (!s->peak)
285 ✗ return AVERROR(ENOMEM);
286
287
2/2
✓ Branch 0 taken 1536 times.
✓ Branch 1 taken 6 times.
 1542 for (i = 0; i < s->size; i++)
288 1536 s->peak[i] = s->peak_memory + s->size * i;
289
290 6 return 0;
291 }
292
293 ✗ static void envelope_instant16(VectorscopeContext *s, AVFrame *out)
294 {
295 ✗ const int dlinesize = out->linesize[0] / 2;
296 ✗ uint16_t *dpd = s->mode == COLOR || !s->is_yuv ? (uint16_t *)out->data[s->pd] : (uint16_t *)out->data[0];
297 ✗ const int max = s->size - 1;
298 int i, j;
299
300 ✗ for (i = 0; i < out->height; i++) {
301 ✗ for (j = 0; j < out->width; j++) {
302 ✗ const int pos = i * dlinesize + j;
303 ✗ const int poa = (i - 1) * dlinesize + j;
304 ✗ const int pob = (i + 1) * dlinesize + j;
305
306 ✗ if (dpd[pos] && (((!j || !dpd[pos - 1]) || ((j == (out->width - 1)) || !dpd[pos + 1]))
307 ✗ || ((!i || !dpd[poa]) || ((i == (out->height - 1)) || !dpd[pob])))) {
308 ✗ dpd[pos] = max;
309 }
310 }
311 }
312 ✗ }
313
314 ✗ static void envelope_peak16(VectorscopeContext *s, AVFrame *out)
315 {
316 ✗ const int dlinesize = out->linesize[0] / 2;
317 ✗ uint16_t *dpd = s->mode == COLOR || !s->is_yuv ? (uint16_t *)out->data[s->pd] : (uint16_t *)out->data[0];
318 ✗ const int max = s->size - 1;
319 int i, j;
320
321 ✗ for (i = 0; i < out->height; i++) {
322 ✗ for (j = 0; j < out->width; j++) {
323 ✗ const int pos = i * dlinesize + j;
324
325 ✗ if (dpd[pos])
326 ✗ s->peak[i][j] = 1;
327 }
328 }
329
330 ✗ if (s->envelope == 3)
331 ✗ envelope_instant16(s, out);
332
333 ✗ for (i = 0; i < out->height; i++) {
334 ✗ for (j = 0; j < out->width; j++) {
335 ✗ const int pos = i * dlinesize + j;
336
337 ✗ if (s->peak[i][j] && (((!j || !s->peak[i][j-1]) || ((j == (out->width - 1)) || !s->peak[i][j + 1]))
338 ✗ || ((!i || !s->peak[i-1][j]) || ((i == (out->height - 1)) || !s->peak[i + 1][j])))) {
339 ✗ dpd[pos] = max;
340 }
341 }
342 }
343 ✗ }
344
345 ✗ static void envelope_instant(VectorscopeContext *s, AVFrame *out)
346 {
347 ✗ const int dlinesize = out->linesize[0];
348 ✗ uint8_t *dpd = s->mode == COLOR || !s->is_yuv ? out->data[s->pd] : out->data[0];
349 int i, j;
350
351 ✗ for (i = 0; i < out->height; i++) {
352 ✗ for (j = 0; j < out->width; j++) {
353 ✗ const int pos = i * dlinesize + j;
354 ✗ const int poa = (i - 1) * dlinesize + j;
355 ✗ const int pob = (i + 1) * dlinesize + j;
356
357 ✗ if (dpd[pos] && (((!j || !dpd[pos - 1]) || ((j == (out->width - 1)) || !dpd[pos + 1]))
358 ✗ || ((!i || !dpd[poa]) || ((i == (out->height - 1)) || !dpd[pob])))) {
359 ✗ dpd[pos] = 255;
360 }
361 }
362 }
363 ✗ }
364
365 ✗ static void envelope_peak(VectorscopeContext *s, AVFrame *out)
366 {
367 ✗ const int dlinesize = out->linesize[0];
368 ✗ uint8_t *dpd = s->mode == COLOR || !s->is_yuv ? out->data[s->pd] : out->data[0];
369 int i, j;
370
371 ✗ for (i = 0; i < out->height; i++) {
372 ✗ for (j = 0; j < out->width; j++) {
373 ✗ const int pos = i * dlinesize + j;
374
375 ✗ if (dpd[pos])
376 ✗ s->peak[i][j] = 1;
377 }
378 }
379
380 ✗ if (s->envelope == 3)
381 ✗ envelope_instant(s, out);
382
383 ✗ for (i = 0; i < out->height; i++) {
384 ✗ for (j = 0; j < out->width; j++) {
385 ✗ const int pos = i * dlinesize + j;
386
387 ✗ if (s->peak[i][j] && (((!j || !s->peak[i][j-1]) || ((j == (out->width - 1)) || !s->peak[i][j + 1]))
388 ✗ || ((!i || !s->peak[i-1][j]) || ((i == (out->height - 1)) || !s->peak[i + 1][j])))) {
389 ✗ dpd[pos] = 255;
390 }
391 }
392 }
393 ✗ }
394
395 ✗ static void envelope16(VectorscopeContext *s, AVFrame *out)
396 {
397 ✗ if (!s->envelope) {
398 ✗ return;
399 ✗ } else if (s->envelope == 1) {
400 ✗ envelope_instant16(s, out);
401 } else {
402 ✗ envelope_peak16(s, out);
403 }
404 }
405
406 18 static void envelope(VectorscopeContext *s, AVFrame *out)
407 {
408
1/2
✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
 18 if (!s->envelope) {
409 18 return;
410 ✗ } else if (s->envelope == 1) {
411 ✗ envelope_instant(s, out);
412 } else {
413 ✗ envelope_peak(s, out);
414 }
415 }
416
417 ✗ static void vectorscope16(VectorscopeContext *s, AVFrame *in, AVFrame *out, int pd)
418 {
419 ✗ const uint16_t * const *src = (const uint16_t * const *)in->data;
420 ✗ const int slinesizex = in->linesize[s->x] / 2;
421 ✗ const int slinesizey = in->linesize[s->y] / 2;
422 ✗ const int slinesized = in->linesize[pd] / 2;
423 ✗ const int dlinesize = out->linesize[0] / 2;
424 ✗ const int intensity = s->intensity;
425 ✗ const int px = s->x, py = s->y;
426 ✗ const int h = s->planeheight[py];
427 ✗ const int w = s->planewidth[px];
428 ✗ const uint16_t *spx = src[px];
429 ✗ const uint16_t *spy = src[py];
430 ✗ const uint16_t *spd = src[pd];
431 ✗ const int hsub = s->hsub;
432 ✗ const int vsub = s->vsub;
433 ✗ uint16_t **dst = (uint16_t **)out->data;
434 ✗ uint16_t *dpx = dst[px];
435 ✗ uint16_t *dpy = dst[py];
436 ✗ uint16_t *dpd = dst[pd];
437 ✗ uint16_t *dp1 = dst[1];
438 ✗ uint16_t *dp2 = dst[2];
439 ✗ const int max = s->size - 1;
440 ✗ const int mid = s->size / 2;
441 ✗ const int tmin = s->tmin;
442 ✗ const int tmax = s->tmax;
443 int i, j, k;
444
445 ✗ for (k = 0; k < 4 && dst[k]; k++) {
446 ✗ for (i = 0; i < out->height ; i++)
447 ✗ for (j = 0; j < out->width; j++)
448 ✗ AV_WN16(out->data[k] + i * out->linesize[k] + j * 2,
449 (s->mode == COLOR || s->mode == COLOR5) && k == s->pd ? 0 : s->bg_color[k]);
450 }
451
452 ✗ switch (s->mode) {
453 ✗ case COLOR:
454 case COLOR5:
455 case TINT:
456 ✗ for (i = 0; i < h; i++) {
457 ✗ const int iwx = i * slinesizex;
458 ✗ const int iwy = i * slinesizey;
459 ✗ const int iwd = i * slinesized;
460 ✗ for (j = 0; j < w; j++) {
461 ✗ const int x = FFMIN(spx[iwx + j], max);
462 ✗ const int y = FFMIN(spy[iwy + j], max);
463 ✗ const int z = spd[iwd + j];
464 ✗ const int pos = y * dlinesize + x;
465
466 ✗ if (z < tmin || z > tmax)
467 ✗ continue;
468
469 ✗ dpd[pos] = FFMIN(dpd[pos] + intensity, max);
470 }
471 }
472 ✗ break;
473 ✗ case COLOR2:
474 ✗ if (s->is_yuv) {
475 ✗ for (i = 0; i < h; i++) {
476 ✗ const int iw1 = i * slinesizex;
477 ✗ const int iw2 = i * slinesizey;
478 ✗ const int iwd = i * slinesized;
479 ✗ for (j = 0; j < w; j++) {
480 ✗ const int x = FFMIN(spx[iw1 + j], max);
481 ✗ const int y = FFMIN(spy[iw2 + j], max);
482 ✗ const int z = spd[iwd + j];
483 ✗ const int pos = y * dlinesize + x;
484
485 ✗ if (z < tmin || z > tmax)
486 ✗ continue;
487
488 ✗ if (!dpd[pos])
489 ✗ dpd[pos] = FFABS(mid - x) + FFABS(mid - y);
490 ✗ dpx[pos] = x;
491 ✗ dpy[pos] = y;
492 }
493 }
494 } else {
495 ✗ for (i = 0; i < h; i++) {
496 ✗ const int iw1 = i * slinesizex;
497 ✗ const int iw2 = i * slinesizey;
498 ✗ const int iwd = i * slinesized;
499 ✗ for (j = 0; j < w; j++) {
500 ✗ const int x = FFMIN(spx[iw1 + j], max);
501 ✗ const int y = FFMIN(spy[iw2 + j], max);
502 ✗ const int z = spd[iwd + j];
503 ✗ const int pos = y * dlinesize + x;
504
505 ✗ if (z < tmin || z > tmax)
506 ✗ continue;
507
508 ✗ if (!dpd[pos])
509 ✗ dpd[pos] = FFMIN(x + y, max);
510 ✗ dpx[pos] = x;
511 ✗ dpy[pos] = y;
512 }
513 }
514 }
515 ✗ break;
516 ✗ case COLOR3:
517 ✗ for (i = 0; i < h; i++) {
518 ✗ const int iw1 = i * slinesizex;
519 ✗ const int iw2 = i * slinesizey;
520 ✗ const int iwd = i * slinesized;
521 ✗ for (j = 0; j < w; j++) {
522 ✗ const int x = FFMIN(spx[iw1 + j], max);
523 ✗ const int y = FFMIN(spy[iw2 + j], max);
524 ✗ const int z = spd[iwd + j];
525 ✗ const int pos = y * dlinesize + x;
526
527 ✗ if (z < tmin || z > tmax)
528 ✗ continue;
529
530 ✗ dpd[pos] = FFMIN(max, dpd[pos] + intensity);
531 ✗ dpx[pos] = x;
532 ✗ dpy[pos] = y;
533 }
534 }
535 ✗ break;
536 ✗ case COLOR4:
537 ✗ for (i = 0; i < in->height; i++) {
538 ✗ const int iwx = (i >> vsub) * slinesizex;
539 ✗ const int iwy = (i >> vsub) * slinesizey;
540 ✗ const int iwd = i * slinesized;
541 ✗ for (j = 0; j < in->width; j++) {
542 ✗ const int x = FFMIN(spx[iwx + (j >> hsub)], max);
543 ✗ const int y = FFMIN(spy[iwy + (j >> hsub)], max);
544 ✗ const int z = spd[iwd + j];
545 ✗ const int pos = y * dlinesize + x;
546
547 ✗ if (z < tmin || z > tmax)
548 ✗ continue;
549
550 ✗ dpd[pos] = FFMAX(z, dpd[pos]);
551 ✗ dpx[pos] = x;
552 ✗ dpy[pos] = y;
553 }
554 }
555 ✗ break;
556 ✗ default:
557 ✗ av_assert0(0);
558 }
559
560 ✗ envelope16(s, out);
561
562 ✗ if (dst[3]) {
563 ✗ for (i = 0; i < out->height; i++) {
564 ✗ for (j = 0; j < out->width; j++) {
565 ✗ int pos = i * dlinesize + j;
566
567 ✗ if (dpd[pos])
568 ✗ dst[3][pos] = max;
569 }
570 }
571 }
572
573 ✗ if (s->mode == TINT && s->is_yuv &&
574 ✗ (s->tint[0] != mid || s->tint[1] != mid)) {
575 ✗ for (i = 0; i < out->height; i++) {
576 ✗ for (j = 0; j < out->width; j++) {
577 ✗ const int pos = i * dlinesize + j;
578 ✗ if (dpd[pos]) {
579 ✗ dp1[pos] = s->tint[0];
580 ✗ dp2[pos] = s->tint[1];
581 }
582 }
583 }
584 ✗ } else if (s->mode == TINT && !s->is_yuv) {
585 ✗ for (i = 0; i < out->height; i++) {
586 ✗ for (j = 0; j < out->width; j++) {
587 ✗ const int pos = i * dlinesize + j;
588 ✗ if (dpd[pos]) {
589 ✗ dpx[pos] = av_clip(dpd[pos] + dpd[pos] * s->ftint[0], 0, max);
590 ✗ dpy[pos] = av_clip(dpd[pos] + dpd[pos] * s->ftint[1], 0, max);
591 }
592 }
593 }
594 ✗ } else if (s->mode == COLOR) {
595 ✗ for (i = 0; i < out->height; i++) {
596 ✗ for (j = 0; j < out->width; j++) {
597 ✗ if (!dpd[i * dlinesize + j]) {
598 ✗ dpx[i * dlinesize + j] = j;
599 ✗ dpy[i * dlinesize + j] = i;
600 ✗ dpd[i * dlinesize + j] = mid;
601 }
602 }
603 }
604 ✗ } else if (s->mode == COLOR5) {
605 ✗ for (i = 0; i < out->height; i++) {
606 ✗ for (j = 0; j < out->width; j++) {
607 ✗ if (!dpd[i * dlinesize + j]) {
608 ✗ dpx[i * dlinesize + j] = j;
609 ✗ dpy[i * dlinesize + j] = i;
610 ✗ dpd[i * dlinesize + j] = mid * M_SQRT2 - hypot(i - mid, j - mid);
611 }
612 }
613 }
614 }
615 ✗ }
616
617 18 static void vectorscope8(VectorscopeContext *s, AVFrame *in, AVFrame *out, int pd)
618 {
619 18 const uint8_t * const *src = (const uint8_t * const *)in->data;
620 18 const int slinesizex = in->linesize[s->x];
621 18 const int slinesizey = in->linesize[s->y];
622 18 const int slinesized = in->linesize[pd];
623 18 const int dlinesize = out->linesize[0];
624 18 const int intensity = s->intensity;
625 18 const int px = s->x, py = s->y;
626 18 const int h = s->planeheight[py];
627 18 const int w = s->planewidth[px];
628 18 const uint8_t *spx = src[px];
629 18 const uint8_t *spy = src[py];
630 18 const uint8_t *spd = src[pd];
631 18 const int hsub = s->hsub;
632 18 const int vsub = s->vsub;
633 18 uint8_t **dst = out->data;
634 18 uint8_t *dpx = dst[px];
635 18 uint8_t *dpy = dst[py];
636 18 uint8_t *dpd = dst[pd];
637 18 uint8_t *dp1 = dst[1];
638 18 uint8_t *dp2 = dst[2];
639 18 const int tmin = s->tmin;
640 18 const int tmax = s->tmax;
641 int i, j, k;
642
643
3/4
✓ Branch 0 taken 72 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 54 times.
✓ Branch 3 taken 18 times.
 72 for (k = 0; k < 4 && dst[k]; k++)
644
2/2
✓ Branch 0 taken 13824 times.
✓ Branch 1 taken 54 times.
 13878 for (i = 0; i < out->height ; i++)
645 13824 memset(dst[k] + i * out->linesize[k],
646
5/6
✓ Branch 0 taken 11520 times.
✓ Branch 1 taken 2304 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 11520 times.
✓ Branch 4 taken 1536 times.
✓ Branch 5 taken 768 times.
 13824 (s->mode == COLOR || s->mode == COLOR5) && k == s->pd ? 0 : s->bg_color[k], out->width);
647
648
4/5
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 3 times.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 3 times.
✗ Branch 4 not taken.
 18 switch (s->mode) {
649 9 case COLOR5:
650 case COLOR:
651 case TINT:
652
2/2
✓ Branch 0 taken 1728 times.
✓ Branch 1 taken 9 times.
 1737 for (i = 0; i < h; i++) {
653 1728 const int iwx = i * slinesizex;
654 1728 const int iwy = i * slinesizey;
655 1728 const int iwd = i * slinesized;
656
2/2
✓ Branch 0 taken 456192 times.
✓ Branch 1 taken 1728 times.
 457920 for (j = 0; j < w; j++) {
657 456192 const int x = spx[iwx + j];
658 456192 const int y = spy[iwy + j];
659 456192 const int z = spd[iwd + j];
660 456192 const int pos = y * dlinesize + x;
661
662
2/4
✓ Branch 0 taken 456192 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 456192 times.
 456192 if (z < tmin || z > tmax)
663 ✗ continue;
664
665 456192 dpd[pos] = FFMIN(dpd[pos] + intensity, 255);
666 }
667 }
668 9 break;
669 3 case COLOR2:
670
1/2
✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
 3 if (s->is_yuv) {
671
2/2
✓ Branch 0 taken 432 times.
✓ Branch 1 taken 3 times.
 435 for (i = 0; i < h; i++) {
672 432 const int iw1 = i * slinesizex;
673 432 const int iw2 = i * slinesizey;
674 432 const int iwd = i * slinesized;
675
2/2
✓ Branch 0 taken 76032 times.
✓ Branch 1 taken 432 times.
 76464 for (j = 0; j < w; j++) {
676 76032 const int x = spx[iw1 + j];
677 76032 const int y = spy[iw2 + j];
678 76032 const int z = spd[iwd + j];
679 76032 const int pos = y * dlinesize + x;
680
681
2/4
✓ Branch 0 taken 76032 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 76032 times.
 76032 if (z < tmin || z > tmax)
682 ✗ continue;
683
684
2/2
✓ Branch 0 taken 44303 times.
✓ Branch 1 taken 31729 times.
 76032 if (!dpd[pos])
685
4/4
✓ Branch 0 taken 22627 times.
✓ Branch 1 taken 21676 times.
✓ Branch 2 taken 21498 times.
✓ Branch 3 taken 22805 times.
 44303 dpd[pos] = FFABS(128 - x) + FFABS(128 - y);
686 76032 dpx[pos] = x;
687 76032 dpy[pos] = y;
688 }
689 }
690 } else {
691 ✗ for (i = 0; i < h; i++) {
692 ✗ const int iw1 = i * slinesizex;
693 ✗ const int iw2 = i * slinesizey;
694 ✗ const int iwd = i * slinesized;
695 ✗ for (j = 0; j < w; j++) {
696 ✗ const int x = spx[iw1 + j];
697 ✗ const int y = spy[iw2 + j];
698 ✗ const int z = spd[iwd + j];
699 ✗ const int pos = y * dlinesize + x;
700
701 ✗ if (z < tmin || z > tmax)
702 ✗ continue;
703
704 ✗ if (!dpd[pos])
705 ✗ dpd[pos] = FFMIN(x + y, 255);
706 ✗ dpx[pos] = x;
707 ✗ dpy[pos] = y;
708 }
709 }
710 }
711 3 break;
712 3 case COLOR3:
713
2/2
✓ Branch 0 taken 432 times.
✓ Branch 1 taken 3 times.
 435 for (i = 0; i < h; i++) {
714 432 const int iw1 = i * slinesizex;
715 432 const int iw2 = i * slinesizey;
716 432 const int iwd = i * slinesized;
717
2/2
✓ Branch 0 taken 76032 times.
✓ Branch 1 taken 432 times.
 76464 for (j = 0; j < w; j++) {
718 76032 const int x = spx[iw1 + j];
719 76032 const int y = spy[iw2 + j];
720 76032 const int z = spd[iwd + j];
721 76032 const int pos = y * dlinesize + x;
722
723
2/4
✓ Branch 0 taken 76032 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 76032 times.
 76032 if (z < tmin || z > tmax)
724 ✗ continue;
725
726 76032 dpd[pos] = FFMIN(255, dpd[pos] + intensity);
727 76032 dpx[pos] = x;
728 76032 dpy[pos] = y;
729 }
730 }
731 3 break;
732 3 case COLOR4:
733
2/2
✓ Branch 0 taken 864 times.
✓ Branch 1 taken 3 times.
 867 for (i = 0; i < in->height; i++) {
734 864 const int iwx = (i >> vsub) * slinesizex;
735 864 const int iwy = (i >> vsub) * slinesizey;
736 864 const int iwd = i * slinesized;
737
2/2
✓ Branch 0 taken 304128 times.
✓ Branch 1 taken 864 times.
 304992 for (j = 0; j < in->width; j++) {
738 304128 const int x = spx[iwx + (j >> hsub)];
739 304128 const int y = spy[iwy + (j >> hsub)];
740 304128 const int z = spd[iwd + j];
741 304128 const int pos = y * dlinesize + x;
742
743
2/4
✓ Branch 0 taken 304128 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 304128 times.
 304128 if (z < tmin || z > tmax)
744 ✗ continue;
745
746 304128 dpd[pos] = FFMAX(z, dpd[pos]);
747 304128 dpx[pos] = x;
748 304128 dpy[pos] = y;
749 }
750 }
751 3 break;
752 ✗ default:
753 ✗ av_assert0(0);
754 }
755
756 18 envelope(s, out);
757
758
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 if (dst[3]) {
759 ✗ for (i = 0; i < out->height; i++) {
760 ✗ for (j = 0; j < out->width; j++) {
761 ✗ int pos = i * dlinesize + j;
762
763 ✗ if (dpd[pos])
764 ✗ dst[3][pos] = 255;
765 }
766 }
767 }
768
769
3/4
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 12 times.
✓ Branch 2 taken 6 times.
✗ Branch 3 not taken.
 18 if (s->mode == TINT && s->is_yuv &&
770
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 6 (s->tint[0] != 128 || s->tint[1] != 128)) {
771
2/2
✓ Branch 0 taken 1536 times.
✓ Branch 1 taken 6 times.
 1542 for (i = 0; i < out->height; i++) {
772
2/2
✓ Branch 0 taken 393216 times.
✓ Branch 1 taken 1536 times.
 394752 for (j = 0; j < out->width; j++) {
773 393216 const int pos = i * dlinesize + j;
774
2/2
✓ Branch 0 taken 123766 times.
✓ Branch 1 taken 269450 times.
 393216 if (dpd[pos]) {
775 123766 dp1[pos] = s->tint[0];
776 123766 dp2[pos] = s->tint[1];
777 }
778 }
779 }
780
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 12 } else if (s->mode == TINT && !s->is_yuv) {
781 ✗ for (i = 0; i < out->height; i++) {
782 ✗ for (j = 0; j < out->width; j++) {
783 ✗ const int pos = i * dlinesize + j;
784 ✗ if (dpd[pos]) {
785 ✗ dpx[pos] = av_clip_uint8(dpd[pos] + dpd[pos] * s->ftint[0]);
786 ✗ dpy[pos] = av_clip_uint8(dpd[pos] + dpd[pos] * s->ftint[1]);
787 }
788 }
789 }
790
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 9 times.
 12 } else if (s->mode == COLOR) {
791
2/2
✓ Branch 0 taken 768 times.
✓ Branch 1 taken 3 times.
 771 for (i = 0; i < out->height; i++) {
792
2/2
✓ Branch 0 taken 196608 times.
✓ Branch 1 taken 768 times.
 197376 for (j = 0; j < out->width; j++) {
793
2/2
✓ Branch 0 taken 152318 times.
✓ Branch 1 taken 44290 times.
 196608 if (!dpd[i * out->linesize[pd] + j]) {
794 152318 dpx[i * out->linesize[px] + j] = j;
795 152318 dpy[i * out->linesize[py] + j] = i;
796 152318 dpd[i * out->linesize[pd] + j] = 128;
797 }
798 }
799 }
800
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
 9 } else if (s->mode == COLOR5) {
801 ✗ for (i = 0; i < out->height; i++) {
802 ✗ for (j = 0; j < out->width; j++) {
803 ✗ if (!dpd[i * out->linesize[pd] + j]) {
804 ✗ dpx[i * out->linesize[px] + j] = j;
805 ✗ dpy[i * out->linesize[py] + j] = i;
806 ✗ dpd[i * out->linesize[pd] + j] = 128 * M_SQRT2 - hypot(i - 128, j - 128);
807 }
808 }
809 }
810 }
811 18 }
812
813 const static char *positions_name[] = {
814 "R", "B", "Cy", "Yl", "G", "Mg",
815 };
816
817 const static uint16_t positions[][14][3] = {
818 {
819 { 81, 90, 240 }, { 41, 240, 110 }, { 170, 166, 16 },
820 { 210, 16, 146 }, { 145, 54, 34 }, { 106, 202, 222 },
821 { 162, 44, 142 }, { 131, 156, 44 }, { 112, 72, 58 },
822 { 84, 184, 198 }, { 65, 100, 212 }, { 35, 212, 114 },
823 { 235, 128, 128 }, { 16, 128, 128 } },
824 { { 63, 102, 240 }, { 32, 240, 118 }, { 188, 154, 16 },
825 { 219, 16, 138 }, { 173, 42, 26 }, { 78, 214, 230 },
826 { 28, 212, 120 }, { 51, 109, 212 }, { 63, 193, 204 },
827 { 133, 63, 52 }, { 145, 147, 44 }, { 168, 44, 136 },
828 { 235, 128, 128 }, { 16, 128, 128 } },
829 { { 81*2, 90*2, 240*2 }, { 41*2, 240*2, 110*2 }, { 170*2, 166*2, 16*2 },
830 { 210*2, 16*2, 146*2 }, { 145*2, 54*2, 34*2 }, { 106*2, 202*2, 222*2 },
831 { 162*2, 44*2, 142*2 }, { 131*2, 156*2, 44*2 }, { 112*2, 72*2, 58*2 },
832 { 84*2, 184*2, 198*2 }, { 65*2, 100*2, 212*2 }, { 35*2, 212*2, 114*2 },
833 { 470, 256, 256 }, { 32, 256, 256 } },
834 { { 63*2, 102*2, 240*2 }, { 32*2, 240*2, 118*2 }, { 188*2, 154*2, 16*2 },
835 { 219*2, 16*2, 138*2 }, { 173*2, 42*2, 26*2 }, { 78*2, 214*2, 230*2 },
836 { 28*2, 212*2, 120*2 }, { 51*2, 109*2, 212*2 }, { 63*2, 193*2, 204*2 },
837 { 133*2, 63*2, 52*2 }, { 145*2, 147*2, 44*2 }, { 168*2, 44*2, 136*2 },
838 { 470, 256, 256 }, { 32, 256, 256 } },
839 { { 81*4, 90*4, 240*4 }, { 41*4, 240*4, 110*4 }, { 170*4, 166*4, 16*4 },
840 { 210*4, 16*4, 146*4 }, { 145*4, 54*4, 34*4 }, { 106*4, 202*4, 222*4 },
841 { 162*4, 44*4, 142*4 }, { 131*4, 156*4, 44*4 }, { 112*4, 72*4, 58*4 },
842 { 84*4, 184*4, 198*4 }, { 65*4, 100*4, 212*4 }, { 35*4, 212*4, 114*4 },
843 { 940, 512, 512 }, { 64, 512, 512 } },
844 { { 63*4, 102*4, 240*4 }, { 32*4, 240*4, 118*4 }, { 188*4, 154*4, 16*4 },
845 { 219*4, 16*4, 138*4 }, { 173*4, 42*4, 26*4 }, { 78*4, 214*4, 230*4 },
846 { 28*4, 212*4, 120*4 }, { 51*4, 109*4, 212*4 }, { 63*4, 193*4, 204*4 },
847 { 133*4, 63*4, 52*4 }, { 145*4, 147*4, 44*4 }, { 168*4, 44*4, 136*4 },
848 { 940, 512, 512 }, { 64, 512, 512 } },
849 { { 81*8, 90*4, 240*8 }, { 41*8, 240*8, 110*8 }, { 170*8, 166*8, 16*8 },
850 { 210*8, 16*4, 146*8 }, { 145*8, 54*8, 34*8 }, { 106*8, 202*8, 222*8 },
851 { 162*8, 44*4, 142*8 }, { 131*8, 156*8, 44*8 }, { 112*8, 72*8, 58*8 },
852 { 84*8, 184*4, 198*8 }, { 65*8, 100*8, 212*8 }, { 35*8, 212*8, 114*8 },
853 { 1880, 1024, 1024 }, { 128, 1024, 1024 } },
854 { { 63*8, 102*8, 240*8 }, { 32*8, 240*8, 118*8 }, { 188*8, 154*8, 16*8 },
855 { 219*8, 16*8, 138*8 }, { 173*8, 42*8, 26*8 }, { 78*8, 214*8, 230*8 },
856 { 28*8, 212*8, 120*8 }, { 51*8, 109*8, 212*8 }, { 63*8, 193*8, 204*8 },
857 { 133*8, 63*8, 52*8 }, { 145*8, 147*8, 44*8 }, { 168*8, 44*8, 136*8 },
858 { 1880, 1024, 1024 }, { 128, 1024, 1024 } },
859 { { 81*16, 90*16, 240*16 }, { 41*16, 240*16, 110*16 }, { 170*16, 166*16, 16*16 },
860 { 210*16, 16*16, 146*16 }, { 145*16, 54*16, 34*16 }, { 106*16, 202*16, 222*16 },
861 { 162*16, 44*16, 142*16 }, { 131*16, 156*16, 44*16 }, { 112*16, 72*16, 58*16 },
862 { 84*16, 184*16, 198*16 }, { 65*16, 100*16, 212*16 }, { 35*16, 212*16, 114*16 },
863 { 3760, 2048, 2048 }, { 256, 2048, 2048 } },
864 { { 63*16, 102*16, 240*16 }, { 32*16, 240*16, 118*16 }, { 188*16, 154*16, 16*16 },
865 { 219*16, 16*16, 138*16 }, { 173*16, 42*16, 26*16 }, { 78*16, 214*16, 230*16 },
866 { 28*16, 212*16, 120*16 }, { 51*16, 109*16, 212*16 }, { 63*16, 193*16, 204*16 },
867 { 133*16, 63*16, 52*16 }, { 145*16, 147*16, 44*16 }, { 168*16, 44*16, 136*16 },
868 { 3760, 2048, 2048 }, { 256, 2048, 2048 } },
869 };
870
871 ✗ static void draw_dots(uint8_t *dst, int L, int v, float o)
872 {
873 ✗ const float f = 1. - o;
874 ✗ const float V = o * v;
875 ✗ int l = L * 2;
876
877 ✗ dst[ l - 3] = dst[ l - 3] * f + V;
878 ✗ dst[ l + 3] = dst[ l + 3] * f + V;
879 ✗ dst[-l - 3] = dst[-l - 3] * f + V;
880 ✗ dst[-l + 3] = dst[-l + 3] * f + V;
881
882 ✗ l += L;
883
884 ✗ dst[ l - 3] = dst[ l - 3] * f + V;
885 ✗ dst[ l + 3] = dst[ l + 3] * f + V;
886 ✗ dst[ l - 2] = dst[ l - 2] * f + V;
887 ✗ dst[ l + 2] = dst[ l + 2] * f + V;
888 ✗ dst[-l - 3] = dst[-l - 3] * f + V;
889 ✗ dst[-l + 3] = dst[-l + 3] * f + V;
890 ✗ dst[-l - 2] = dst[-l - 2] * f + V;
891 ✗ dst[-l + 2] = dst[-l + 2] * f + V;
892 ✗ }
893
894 ✗ static void draw_idots(uint8_t *dst, int L, float o)
895 {
896 ✗ const float f = 1. - o;
897 ✗ int l = L * 2;
898
899 ✗ dst[ l - 3] = dst[ l - 3] * f + (255 - dst[ l - 3]) * o;
900 ✗ dst[ l + 3] = dst[ l + 3] * f + (255 - dst[ l + 3]) * o;
901 ✗ dst[-l - 3] = dst[-l - 3] * f + (255 - dst[-l - 3]) * o;
902 ✗ dst[-l + 3] = dst[-l + 3] * f + (255 - dst[-l + 3]) * o;
903
904 ✗ l += L;
905
906 ✗ dst[ l - 3] = dst[ l - 3] * f + (255 - dst[ l - 3]) * o;
907 ✗ dst[ l + 3] = dst[ l + 3] * f + (255 - dst[ l + 3]) * o;
908 ✗ dst[ l - 2] = dst[ l - 2] * f + (255 - dst[ l - 2]) * o;
909 ✗ dst[ l + 2] = dst[ l + 2] * f + (255 - dst[ l + 2]) * o;
910 ✗ dst[-l - 3] = dst[-l - 3] * f + (255 - dst[-l - 3]) * o;
911 ✗ dst[-l + 3] = dst[-l + 3] * f + (255 - dst[-l + 3]) * o;
912 ✗ dst[-l - 2] = dst[-l - 2] * f + (255 - dst[-l - 2]) * o;
913 ✗ dst[-l + 2] = dst[-l + 2] * f + (255 - dst[-l + 2]) * o;
914 ✗ }
915
916 ✗ static void draw_dots16(uint16_t *dst, int L, int v, float o)
917 {
918 ✗ const float f = 1. - o;
919 ✗ const float V = o * v;
920 ✗ int l = L * 2;
921
922 ✗ dst[ l - 3] = dst[ l - 3] * f + V;
923 ✗ dst[ l + 3] = dst[ l + 3] * f + V;
924 ✗ dst[-l - 3] = dst[-l - 3] * f + V;
925 ✗ dst[-l + 3] = dst[-l + 3] * f + V;
926
927 ✗ l += L;
928
929 ✗ dst[ l - 3] = dst[ l - 3] * f + V;
930 ✗ dst[ l + 3] = dst[ l + 3] * f + V;
931 ✗ dst[ l - 2] = dst[ l - 2] * f + V;
932 ✗ dst[ l + 2] = dst[ l + 2] * f + V;
933 ✗ dst[-l - 3] = dst[-l - 3] * f + V;
934 ✗ dst[-l + 3] = dst[-l + 3] * f + V;
935 ✗ dst[-l - 2] = dst[-l - 2] * f + V;
936 ✗ dst[-l + 2] = dst[-l + 2] * f + V;
937 ✗ }
938
939 ✗ static void draw_idots16(uint16_t *dst, int L, int v, float o)
940 {
941 ✗ const float f = 1. - o;
942 ✗ int l = L * 2;
943
944 ✗ dst[ l - 3] = dst[ l - 3] * f + (v - dst[ l - 3]) * o;
945 ✗ dst[ l + 3] = dst[ l + 3] * f + (v - dst[ l + 3]) * o;
946 ✗ dst[-l - 3] = dst[-l - 3] * f + (v - dst[-l - 3]) * o;
947 ✗ dst[-l + 3] = dst[-l + 3] * f + (v - dst[-l + 3]) * o;
948
949 ✗ l += L;
950
951 ✗ dst[ l - 3] = dst[ l - 3] * f + (v - dst[ l - 3]) * o;
952 ✗ dst[ l + 3] = dst[ l + 3] * f + (v - dst[ l + 3]) * o;
953 ✗ dst[ l - 2] = dst[ l - 2] * f + (v - dst[ l - 2]) * o;
954 ✗ dst[ l + 2] = dst[ l + 2] * f + (v - dst[ l + 2]) * o;
955 ✗ dst[-l - 3] = dst[-l - 3] * f + (v - dst[-l - 3]) * o;
956 ✗ dst[-l + 3] = dst[-l + 3] * f + (v - dst[-l + 3]) * o;
957 ✗ dst[-l - 2] = dst[-l - 2] * f + (v - dst[-l - 2]) * o;
958 ✗ dst[-l + 2] = dst[-l + 2] * f + (v - dst[-l + 2]) * o;
959 ✗ }
960
961 18 static void none_graticule(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
962 {
963 18 }
964
965 ✗ static void draw_ihtext(AVFrame *out, int x, int y, float o1, float o2, const char *txt, const uint8_t color[4])
966 {
967 const uint8_t *font;
968 int font_height;
969 int i, plane;
970
971 ✗ font = avpriv_cga_font, font_height = 8;
972
973 ✗ for (plane = 0; plane < 4 && out->data[plane]; plane++) {
974 ✗ for (i = 0; txt[i]; i++) {
975 int char_y, mask;
976
977 ✗ uint8_t *p = out->data[plane] + y * out->linesize[plane] + (x + i * 8);
978 ✗ for (char_y = font_height - 1; char_y >= 0; char_y--) {
979 ✗ for (mask = 0x80; mask; mask >>= 1) {
980 ✗ if (font[txt[i] * font_height + char_y] & mask)
981 ✗ p[0] = p[0] * o2 + (255 - p[0]) * o1;
982 ✗ p++;
983 }
984 ✗ p += out->linesize[plane] - 8;
985 }
986 }
987 }
988 ✗ }
989
990 ✗ static void draw_ihtext16(AVFrame *out, int x, int y, float o1, float o2, const char *txt, const uint16_t color[4])
991 {
992 const uint8_t *font;
993 int font_height;
994 int i, plane;
995
996 ✗ font = avpriv_cga_font, font_height = 8;
997
998 ✗ for (plane = 0; plane < 4 && out->data[plane]; plane++) {
999 ✗ for (i = 0; txt[i]; i++) {
1000 int char_y, mask;
1001 ✗ int v = color[plane];
1002
1003 ✗ uint16_t *p = (uint16_t *)(out->data[plane] + y * out->linesize[plane]) + (x + i * 8);
1004 ✗ for (char_y = font_height - 1; char_y >= 0; char_y--) {
1005 ✗ for (mask = 0x80; mask; mask >>= 1) {
1006 ✗ if (font[txt[i] * font_height + char_y] & mask)
1007 ✗ p[0] = p[0] * o2 + (v - p[0]) * o1;
1008 ✗ p++;
1009 }
1010 ✗ p += out->linesize[plane] / 2 - 8;
1011 }
1012 }
1013 }
1014 ✗ }
1015
1016 ✗ static void draw_htext(AVFrame *out, int x, int y, float o1, float o2, const char *txt, const uint8_t color[4])
1017 {
1018 const uint8_t *font;
1019 int font_height;
1020 int i, plane;
1021
1022 ✗ font = avpriv_cga_font, font_height = 8;
1023
1024 ✗ for (plane = 0; plane < 4 && out->data[plane]; plane++) {
1025 ✗ for (i = 0; txt[i]; i++) {
1026 int char_y, mask;
1027 ✗ int v = color[plane];
1028
1029 ✗ uint8_t *p = out->data[plane] + y * out->linesize[plane] + (x + i * 8);
1030 ✗ for (char_y = font_height - 1; char_y >= 0; char_y--) {
1031 ✗ for (mask = 0x80; mask; mask >>= 1) {
1032 ✗ if (font[txt[i] * font_height + char_y] & mask)
1033 ✗ p[0] = p[0] * o2 + v * o1;
1034 ✗ p++;
1035 }
1036 ✗ p += out->linesize[plane] - 8;
1037 }
1038 }
1039 }
1040 ✗ }
1041
1042 ✗ static void draw_htext16(AVFrame *out, int x, int y, float o1, float o2, const char *txt, const uint16_t color[4])
1043 {
1044 const uint8_t *font;
1045 int font_height;
1046 int i, plane;
1047
1048 ✗ font = avpriv_cga_font, font_height = 8;
1049
1050 ✗ for (plane = 0; plane < 4 && out->data[plane]; plane++) {
1051 ✗ for (i = 0; txt[i]; i++) {
1052 int char_y, mask;
1053 ✗ int v = color[plane];
1054
1055 ✗ uint16_t *p = (uint16_t *)(out->data[plane] + y * out->linesize[plane]) + (x + i * 8);
1056 ✗ for (char_y = font_height - 1; char_y >= 0; char_y--) {
1057 ✗ for (mask = 0x80; mask; mask >>= 1) {
1058 ✗ if (font[txt[i] * font_height + char_y] & mask)
1059 ✗ p[0] = p[0] * o2 + v * o1;
1060 ✗ p++;
1061 }
1062 ✗ p += out->linesize[plane] / 2 - 8;
1063 }
1064 }
1065 }
1066 ✗ }
1067
1068 ✗ static void color_graticule16(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
1069 {
1070 ✗ const int max = s->size - 1;
1071 ✗ const float o = s->opacity;
1072 int i;
1073
1074 ✗ for (i = 0; i < 12; i++) {
1075 ✗ int x = positions[P][i][X];
1076 ✗ int y = positions[P][i][Y];
1077 ✗ int d = positions[P][i][D];
1078
1079 ✗ draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, d, o);
1080 ✗ draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, x, o);
1081 ✗ draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, y, o);
1082 ✗ if (out->data[3])
1083 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1084 }
1085
1086 ✗ if (s->flags & 1) {
1087 ✗ int x = positions[P][12][X];
1088 ✗ int y = positions[P][12][Y];
1089 ✗ int d = positions[P][12][D];
1090
1091 ✗ draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, d, o);
1092 ✗ draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, x, o);
1093 ✗ draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, y, o);
1094 ✗ if (out->data[3])
1095 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1096 }
1097
1098 ✗ if (s->flags & 2) {
1099 ✗ int x = positions[P][13][X];
1100 ✗ int y = positions[P][13][Y];
1101 ✗ int d = positions[P][13][D];
1102
1103 ✗ draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, d, o);
1104 ✗ draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, x, o);
1105 ✗ draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, y, o);
1106 ✗ if (out->data[3])
1107 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1108 }
1109
1110 ✗ for (i = 0; i < 6 && s->flags & 4; i++) {
1111 ✗ uint16_t color[4] = { 0, 0, 0, 0 };
1112 ✗ int x = positions[P][i][X];
1113 ✗ int y = positions[P][i][Y];
1114 ✗ int d = positions[P][i][D];
1115
1116 ✗ color[D] = d;
1117 ✗ color[X] = x;
1118 ✗ color[Y] = y;
1119 ✗ color[3] = max;
1120
1121 ✗ if (x > max / 2)
1122 ✗ x += 8;
1123 else
1124 ✗ x -= 14;
1125 ✗ if (y > max / 2)
1126 ✗ y += 8;
1127 else
1128 ✗ y -= 14;
1129
1130 ✗ x = av_clip(x, 0, out->width - 9);
1131 ✗ y = av_clip(y, 0, out->height - 9);
1132 ✗ draw_htext16(out, x, y, o, 1. - o, positions_name[i], color);
1133 }
1134 ✗ }
1135
1136 ✗ static void color_graticule(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
1137 {
1138 ✗ const float o = s->opacity;
1139 int i;
1140
1141 ✗ for (i = 0; i < 12; i++) {
1142 ✗ int x = positions[P][i][X];
1143 ✗ int y = positions[P][i][Y];
1144 ✗ int d = positions[P][i][D];
1145
1146 ✗ draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], d, o);
1147 ✗ draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], x, o);
1148 ✗ draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], y, o);
1149 ✗ if (out->data[3])
1150 ✗ draw_dots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], 255, o);
1151 }
1152
1153 ✗ if (s->flags & 1) {
1154 ✗ int x = positions[P][12][X];
1155 ✗ int y = positions[P][12][Y];
1156 ✗ int d = positions[P][12][D];
1157
1158 ✗ draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], d, o);
1159 ✗ draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], x, o);
1160 ✗ draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], y, o);
1161 ✗ if (out->data[3])
1162 ✗ draw_dots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], 255, o);
1163 }
1164
1165 ✗ if (s->flags & 2) {
1166 ✗ int x = positions[P][13][X];
1167 ✗ int y = positions[P][13][Y];
1168 ✗ int d = positions[P][12][D];
1169
1170 ✗ draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], d, o);
1171 ✗ draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], x, o);
1172 ✗ draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], y, o);
1173 ✗ if (out->data[3])
1174 ✗ draw_dots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], 255, o);
1175 }
1176
1177 ✗ for (i = 0; i < 6 && s->flags & 4; i++) {
1178 ✗ uint8_t color[4] = { 0, 0, 0, 255 };
1179 ✗ int x = positions[P][i][X];
1180 ✗ int y = positions[P][i][Y];
1181 ✗ int d = positions[P][i][D];
1182
1183 ✗ color[D] = d;
1184 ✗ color[X] = x;
1185 ✗ color[Y] = y;
1186
1187 ✗ if (x > 128)
1188 ✗ x += 8;
1189 else
1190 ✗ x -= 14;
1191 ✗ if (y > 128)
1192 ✗ y += 8;
1193 else
1194 ✗ y -= 14;
1195
1196 ✗ x = av_clip(x, 0, out->width - 9);
1197 ✗ y = av_clip(y, 0, out->height - 9);
1198 ✗ draw_htext(out, x, y, o, 1. - o, positions_name[i], color);
1199 }
1200 ✗ }
1201
1202 ✗ static void green_graticule16(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
1203 {
1204 ✗ const int max = s->size - 1;
1205 ✗ const float o = s->opacity;
1206 ✗ const int m = s->mult;
1207 int i;
1208
1209 ✗ for (i = 0; i < 12; i++) {
1210 ✗ int x = positions[P][i][X];
1211 ✗ int y = positions[P][i][Y];
1212
1213 ✗ draw_dots16((uint16_t *)(out->data[0] + y * out->linesize[0] + x * 2), out->linesize[0] / 2, 128 * m, o);
1214 ✗ draw_dots16((uint16_t *)(out->data[1] + y * out->linesize[1] + x * 2), out->linesize[1] / 2, 0, o);
1215 ✗ draw_dots16((uint16_t *)(out->data[2] + y * out->linesize[2] + x * 2), out->linesize[2] / 2, 0, o);
1216 ✗ if (out->data[3])
1217 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1218 }
1219
1220 ✗ if (s->flags & 1) {
1221 ✗ int x = positions[P][12][X];
1222 ✗ int y = positions[P][12][Y];
1223
1224 ✗ draw_dots16((uint16_t *)(out->data[0] + y * out->linesize[0] + x * 2), out->linesize[0] / 2, 128 * m, o);
1225 ✗ draw_dots16((uint16_t *)(out->data[1] + y * out->linesize[1] + x * 2), out->linesize[1] / 2, 0, o);
1226 ✗ draw_dots16((uint16_t *)(out->data[2] + y * out->linesize[2] + x * 2), out->linesize[2] / 2, 0, o);
1227 ✗ if (out->data[3])
1228 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1229 }
1230
1231 ✗ if (s->flags & 2) {
1232 ✗ int x = positions[P][13][X];
1233 ✗ int y = positions[P][13][Y];
1234
1235 ✗ draw_dots16((uint16_t *)(out->data[0] + y * out->linesize[0] + x * 2), out->linesize[0] / 2, 128 * m, o);
1236 ✗ draw_dots16((uint16_t *)(out->data[1] + y * out->linesize[1] + x * 2), out->linesize[1] / 2, 0, o);
1237 ✗ draw_dots16((uint16_t *)(out->data[2] + y * out->linesize[2] + x * 2), out->linesize[2] / 2, 0, o);
1238 ✗ if (out->data[3])
1239 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1240 }
1241
1242 ✗ for (i = 0; i < 6 && s->flags & 4; i++) {
1243 ✗ const uint16_t color[4] = { 128 * m, 0, 0, max };
1244 ✗ int x = positions[P][i][X];
1245 ✗ int y = positions[P][i][Y];
1246
1247 ✗ if (x > max / 2)
1248 ✗ x += 8;
1249 else
1250 ✗ x -= 14;
1251 ✗ if (y > max / 2)
1252 ✗ y += 8;
1253 else
1254 ✗ y -= 14;
1255
1256 ✗ x = av_clip(x, 0, out->width - 9);
1257 ✗ y = av_clip(y, 0, out->height - 9);
1258 ✗ draw_htext16(out, x, y, o, 1. - o, positions_name[i], color);
1259 }
1260 ✗ }
1261
1262 ✗ static void green_graticule(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
1263 {
1264 ✗ const float o = s->opacity;
1265 int i;
1266
1267 ✗ for (i = 0; i < 12; i++) {
1268 ✗ int x = positions[P][i][X];
1269 ✗ int y = positions[P][i][Y];
1270
1271 ✗ draw_dots(out->data[0] + y * out->linesize[0] + x, out->linesize[0], 128, o);
1272 ✗ draw_dots(out->data[1] + y * out->linesize[1] + x, out->linesize[1], 0, o);
1273 ✗ draw_dots(out->data[2] + y * out->linesize[2] + x, out->linesize[2], 0, o);
1274 ✗ if (out->data[3])
1275 ✗ draw_dots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], 255, o);
1276 }
1277
1278 ✗ if (s->flags & 1) {
1279 ✗ int x = positions[P][12][X];
1280 ✗ int y = positions[P][12][Y];
1281
1282 ✗ draw_dots(out->data[0] + y * out->linesize[0] + x, out->linesize[0], 128, o);
1283 ✗ draw_dots(out->data[1] + y * out->linesize[1] + x, out->linesize[1], 0, o);
1284 ✗ draw_dots(out->data[2] + y * out->linesize[2] + x, out->linesize[2], 0, o);
1285 ✗ if (out->data[3])
1286 ✗ draw_dots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], 255, o);
1287 }
1288
1289 ✗ if (s->flags & 2) {
1290 ✗ int x = positions[P][13][X];
1291 ✗ int y = positions[P][13][Y];
1292
1293 ✗ draw_dots(out->data[0] + y * out->linesize[0] + x, out->linesize[0], 128, o);
1294 ✗ draw_dots(out->data[1] + y * out->linesize[1] + x, out->linesize[1], 0, o);
1295 ✗ draw_dots(out->data[2] + y * out->linesize[2] + x, out->linesize[2], 0, o);
1296 ✗ if (out->data[3])
1297 ✗ draw_dots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], 255, o);
1298 }
1299
1300 ✗ for (i = 0; i < 6 && s->flags & 4; i++) {
1301 ✗ const uint8_t color[4] = { 128, 0, 0, 255 };
1302 ✗ int x = positions[P][i][X];
1303 ✗ int y = positions[P][i][Y];
1304
1305 ✗ if (x > 128)
1306 ✗ x += 8;
1307 else
1308 ✗ x -= 14;
1309 ✗ if (y > 128)
1310 ✗ y += 8;
1311 else
1312 ✗ y -= 14;
1313
1314 ✗ x = av_clip(x, 0, out->width - 9);
1315 ✗ y = av_clip(y, 0, out->height - 9);
1316 ✗ draw_htext(out, x, y, o, 1. - o, positions_name[i], color);
1317 }
1318 ✗ }
1319
1320 ✗ static void invert_graticule16(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
1321 {
1322 ✗ const int max = s->size - 1;
1323 ✗ const float o = s->opacity;
1324 int i;
1325
1326 ✗ for (i = 0; i < 12; i++) {
1327 ✗ int x = positions[P][i][X];
1328 ✗ int y = positions[P][i][Y];
1329
1330 ✗ draw_idots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, max, o);
1331 ✗ draw_idots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, max, o);
1332 ✗ draw_idots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, max, o);
1333 ✗ if (out->data[3])
1334 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1335 }
1336
1337 ✗ if (s->flags & 1) {
1338 ✗ int x = positions[P][12][X];
1339 ✗ int y = positions[P][12][Y];
1340
1341 ✗ draw_idots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, max, o);
1342 ✗ draw_idots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, max, o);
1343 ✗ draw_idots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, max, o);
1344 ✗ if (out->data[3])
1345 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1346 }
1347
1348 ✗ if (s->flags & 2) {
1349 ✗ int x = positions[P][13][X];
1350 ✗ int y = positions[P][13][Y];
1351
1352 ✗ draw_idots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, max, o);
1353 ✗ draw_idots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, max, o);
1354 ✗ draw_idots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, max, o);
1355 ✗ if (out->data[3])
1356 ✗ draw_dots16((uint16_t *)(out->data[3] + y * out->linesize[3] + x * 2), out->linesize[3] / 2, max, o);
1357 }
1358
1359 ✗ for (i = 0; i < 6 && s->flags & 4; i++) {
1360 ✗ uint16_t color[4] = { max, max, max, max };
1361 ✗ int x = positions[P][i][X];
1362 ✗ int y = positions[P][i][Y];
1363
1364 ✗ if (x > max / 2)
1365 ✗ x += 8;
1366 else
1367 ✗ x -= 14;
1368 ✗ if (y > max / 2)
1369 ✗ y += 8;
1370 else
1371 ✗ y -= 14;
1372
1373 ✗ x = av_clip(x, 0, out->width - 9);
1374 ✗ y = av_clip(y, 0, out->height - 9);
1375 ✗ draw_ihtext16(out, x, y, o, 1. - o, positions_name[i], color);
1376 }
1377 ✗ }
1378
1379 ✗ static void invert_graticule(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
1380 {
1381 ✗ const float o = s->opacity;
1382 int i;
1383
1384 ✗ for (i = 0; i < 12; i++) {
1385 ✗ int x = positions[P][i][X];
1386 ✗ int y = positions[P][i][Y];
1387
1388 ✗ draw_idots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], o);
1389 ✗ draw_idots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], o);
1390 ✗ draw_idots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], o);
1391 ✗ if (out->data[3])
1392 ✗ draw_idots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], o);
1393 }
1394
1395 ✗ if (s->flags & 1) {
1396 ✗ int x = positions[P][12][X];
1397 ✗ int y = positions[P][12][Y];
1398
1399 ✗ draw_idots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], o);
1400 ✗ draw_idots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], o);
1401 ✗ draw_idots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], o);
1402 ✗ if (out->data[3])
1403 ✗ draw_idots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], o);
1404 }
1405
1406 ✗ if (s->flags & 2) {
1407 ✗ int x = positions[P][13][X];
1408 ✗ int y = positions[P][13][Y];
1409
1410 ✗ draw_idots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], o);
1411 ✗ draw_idots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], o);
1412 ✗ draw_idots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], o);
1413 ✗ if (out->data[3])
1414 ✗ draw_idots(out->data[3] + y * out->linesize[3] + x, out->linesize[3], o);
1415 }
1416
1417 ✗ for (i = 0; i < 6 && s->flags & 4; i++) {
1418 ✗ uint8_t color[4] = { 255, 255, 255, 255 };
1419 ✗ int x = positions[P][i][X];
1420 ✗ int y = positions[P][i][Y];
1421
1422 ✗ if (x > 128)
1423 ✗ x += 8;
1424 else
1425 ✗ x -= 14;
1426 ✗ if (y > 128)
1427 ✗ y += 8;
1428 else
1429 ✗ y -= 14;
1430
1431 ✗ x = av_clip(x, 0, out->width - 9);
1432 ✗ y = av_clip(y, 0, out->height - 9);
1433 ✗ draw_ihtext(out, x, y, o, 1. - o, positions_name[i], color);
1434 }
1435 ✗ }
1436
1437 18 static int filter_frame(AVFilterLink *inlink, AVFrame *in)
1438 {
1439 18 AVFilterContext *ctx = inlink->dst;
1440 18 VectorscopeContext *s = ctx->priv;
1441 18 AVFilterLink *outlink = ctx->outputs[0];
1442 AVFrame *out;
1443 int plane;
1444
1445 18 s->bg_color[3] = s->bgopacity * (s->size - 1);
1446
1447 18 s->tint[0] = .5f * (s->ftint[0] + 1.f) * (s->size - 1);
1448 18 s->tint[1] = .5f * (s->ftint[1] + 1.f) * (s->size - 1);
1449
1450 18 s->intensity = s->fintensity * (s->size - 1);
1451
1452
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 if (s->colorspace) {
1453 ✗ s->cs = (s->depth - 8) * 2 + s->colorspace - 1;
1454 } else {
1455
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 switch (in->colorspace) {
1456 ✗ case AVCOL_SPC_SMPTE170M:
1457 case AVCOL_SPC_BT470BG:
1458 ✗ s->cs = (s->depth - 8) * 2 + 0;
1459 ✗ break;
1460 18 case AVCOL_SPC_BT709:
1461 default:
1462 18 s->cs = (s->depth - 8) * 2 + 1;
1463 }
1464 }
1465
1466 18 out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
1467
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 if (!out) {
1468 ✗ av_frame_free(&in);
1469 ✗ return AVERROR(ENOMEM);
1470 }
1471 18 av_frame_copy_props(out, in);
1472
1473 18 s->vectorscope(s, in, out, s->pd);
1474 18 s->graticulef(s, out, s->x, s->y, s->pd, s->cs);
1475
1476
2/2
✓ Branch 0 taken 72 times.
✓ Branch 1 taken 18 times.
 90 for (plane = 0; plane < 4; plane++) {
1477
2/2
✓ Branch 0 taken 54 times.
✓ Branch 1 taken 18 times.
 72 if (out->data[plane]) {
1478 54 out->data[plane] += (s->size - 1) * out->linesize[plane];
1479 54 out->linesize[plane] = -out->linesize[plane];
1480 }
1481 }
1482
1483 18 av_frame_free(&in);
1484 18 return ff_filter_frame(outlink, out);
1485 }
1486
1487 6 static int config_input(AVFilterLink *inlink)
1488 {
1489 6 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
1490 6 AVFilterContext *ctx = inlink->dst;
1491 6 VectorscopeContext *s = ctx->priv;
1492
1493 6 s->is_yuv = !(desc->flags & AV_PIX_FMT_FLAG_RGB);
1494 6 s->size = 1 << desc->comp[0].depth;
1495 6 s->mult = s->size / 256;
1496 6 s->depth = desc->comp[0].depth;
1497 6 s->tmin = s->lthreshold * (s->size - 1);
1498 6 s->tmax = s->hthreshold * (s->size - 1);
1499
1500
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (s->tmin > s->tmax) {
1501 ✗ av_log(ctx, AV_LOG_ERROR, "low threshold should be less than high threshold\n");
1502 ✗ return AVERROR(EINVAL);
1503 }
1504
1505
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
 6 if (s->mode == TINT && s->is_yuv) {
1506 2 s->pd = 0;
1507 } else {
1508
2/8
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 4 if ((s->x == 1 && s->y == 2) || (s->x == 2 && s->y == 1))
1509 4 s->pd = 0;
1510 ✗ else if ((s->x == 0 && s->y == 2) || (s->x == 2 && s->y == 0))
1511 ✗ s->pd = 1;
1512 ✗ else if ((s->x == 0 && s->y == 1) || (s->x == 1 && s->y == 0))
1513 ✗ s->pd = 2;
1514 }
1515
1516
1/2
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
 6 if (s->size == 256)
1517 6 s->vectorscope = vectorscope8;
1518 else
1519 ✗ s->vectorscope = vectorscope16;
1520
1521 6 s->graticulef = none_graticule;
1522
1523
2/4
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
✗ Branch 3 not taken.
 6 if (s->is_yuv && s->size == 256) {
1524
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (s->graticule == GRAT_GREEN)
1525 ✗ s->graticulef = green_graticule;
1526
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 else if (s->graticule == GRAT_COLOR)
1527 ✗ s->graticulef = color_graticule;
1528
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 else if (s->graticule == GRAT_INVERT)
1529 ✗ s->graticulef = invert_graticule;
1530 ✗ } else if (s->is_yuv) {
1531 ✗ if (s->graticule == GRAT_GREEN)
1532 ✗ s->graticulef = green_graticule16;
1533 ✗ else if (s->graticule == GRAT_COLOR)
1534 ✗ s->graticulef = color_graticule16;
1535 ✗ else if (s->graticule == GRAT_INVERT)
1536 ✗ s->graticulef = invert_graticule16;
1537 }
1538
1539
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 switch (inlink->format) {
1540 ✗ case AV_PIX_FMT_GBRP12:
1541 case AV_PIX_FMT_GBRP10:
1542 case AV_PIX_FMT_GBRP9:
1543 case AV_PIX_FMT_GBRAP:
1544 case AV_PIX_FMT_GBRP:
1545 ✗ s->bg_color[0] = 0;
1546 ✗ s->bg_color[1] = 0;
1547 ✗ s->bg_color[2] = 0;
1548 ✗ break;
1549 6 default:
1550 6 s->bg_color[0] = 0;
1551 6 s->bg_color[1] = s->size / 2;
1552 6 s->bg_color[2] = s->size / 2;
1553 }
1554
1555 6 s->hsub = desc->log2_chroma_w;
1556 6 s->vsub = desc->log2_chroma_h;
1557 6 s->planeheight[1] = s->planeheight[2] = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
1558 6 s->planeheight[0] = s->planeheight[3] = inlink->h;
1559 6 s->planewidth[1] = s->planewidth[2] = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
1560 6 s->planewidth[0] = s->planewidth[3] = inlink->w;
1561
1562 6 return 0;
1563 }
1564
1565 12 static av_cold void uninit(AVFilterContext *ctx)
1566 {
1567 12 VectorscopeContext *s = ctx->priv;
1568
1569 12 av_freep(&s->peak);
1570 12 av_freep(&s->peak_memory);
1571 12 }
1572
1573 static const AVFilterPad inputs[] = {
1574 {
1575 .name = "default",
1576 .type = AVMEDIA_TYPE_VIDEO,
1577 .filter_frame = filter_frame,
1578 .config_props = config_input,
1579 },
1580 };
1581
1582 static const AVFilterPad outputs[] = {
1583 {
1584 .name = "default",
1585 .type = AVMEDIA_TYPE_VIDEO,
1586 .config_props = config_output,
1587 },
1588 };
1589
1590 const AVFilter ff_vf_vectorscope = {
1591 .name = "vectorscope",
1592 .description = NULL_IF_CONFIG_SMALL("Video vectorscope."),
1593 .priv_size = sizeof(VectorscopeContext),
1594 .priv_class = &vectorscope_class,
1595 .uninit = uninit,
1596 FILTER_INPUTS(inputs),
1597 FILTER_OUTPUTS(outputs),
1598 FILTER_QUERY_FUNC(query_formats),
1599 .process_command = ff_filter_process_command,
1600 };
1601