summaryrefslogtreecommitdiff
path: root/libavfilter/vf_waveform.c (plain)
blob: 70995eebeee3bedef1669928fb425828c41a5e31
1/*
2 * Copyright (c) 2012-2016 Paul B Mahol
3 * Copyright (c) 2013 Marton Balint
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#include "libavutil/avassert.h"
23#include "libavutil/opt.h"
24#include "libavutil/parseutils.h"
25#include "libavutil/pixdesc.h"
26#include "libavutil/xga_font_data.h"
27#include "avfilter.h"
28#include "formats.h"
29#include "internal.h"
30#include "video.h"
31
32enum FilterType {
33 LOWPASS,
34 FLAT,
35 AFLAT,
36 CHROMA,
37 COLOR,
38 ACOLOR,
39 NB_FILTERS
40};
41
42enum DisplayType {
43 OVERLAY,
44 STACK,
45 PARADE,
46 NB_DISPLAYS
47};
48
49enum ScaleType {
50 DIGITAL,
51 MILLIVOLTS,
52 IRE,
53 NB_SCALES
54};
55
56typedef struct GraticuleLine {
57 const char *name;
58 uint16_t pos;
59} GraticuleLine;
60
61typedef struct GraticuleLines {
62 struct GraticuleLine line[4];
63} GraticuleLines;
64
65typedef struct WaveformContext {
66 const AVClass *class;
67 int mode;
68 int acomp;
69 int dcomp;
70 int ncomp;
71 int pcomp;
72 uint8_t bg_color[4];
73 float fintensity;
74 int intensity;
75 int mirror;
76 int display;
77 int envelope;
78 int graticule;
79 float opacity;
80 float bgopacity;
81 int estart[4];
82 int eend[4];
83 int *emax[4][4];
84 int *emin[4][4];
85 int *peak;
86 int filter;
87 int flags;
88 int bits;
89 int max;
90 int size;
91 int scale;
92 int shift_w[4], shift_h[4];
93 GraticuleLines *glines;
94 int nb_glines;
95 void (*waveform)(struct WaveformContext *s,
96 AVFrame *in, AVFrame *out,
97 int component, int intensity,
98 int offset_y, int offset_x,
99 int column, int mirror);
100 void (*graticulef)(struct WaveformContext *s, AVFrame *out);
101 const AVPixFmtDescriptor *desc;
102 const AVPixFmtDescriptor *odesc;
103} WaveformContext;
104
105#define OFFSET(x) offsetof(WaveformContext, x)
106#define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
107
108static const AVOption waveform_options[] = {
109 { "mode", "set mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64=1}, 0, 1, FLAGS, "mode" },
110 { "m", "set mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64=1}, 0, 1, FLAGS, "mode" },
111 { "row", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "mode" },
112 { "column", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "mode" },
113 { "intensity", "set intensity", OFFSET(fintensity), AV_OPT_TYPE_FLOAT, {.dbl=0.04}, 0, 1, FLAGS },
114 { "i", "set intensity", OFFSET(fintensity), AV_OPT_TYPE_FLOAT, {.dbl=0.04}, 0, 1, FLAGS },
115 { "mirror", "set mirroring", OFFSET(mirror), AV_OPT_TYPE_BOOL, {.i64=1}, 0, 1, FLAGS },
116 { "r", "set mirroring", OFFSET(mirror), AV_OPT_TYPE_BOOL, {.i64=1}, 0, 1, FLAGS },
117 { "display", "set display mode", OFFSET(display), AV_OPT_TYPE_INT, {.i64=STACK}, 0, NB_DISPLAYS-1, FLAGS, "display" },
118 { "d", "set display mode", OFFSET(display), AV_OPT_TYPE_INT, {.i64=STACK}, 0, NB_DISPLAYS-1, FLAGS, "display" },
119 { "overlay", NULL, 0, AV_OPT_TYPE_CONST, {.i64=OVERLAY}, 0, 0, FLAGS, "display" },
120 { "stack", NULL, 0, AV_OPT_TYPE_CONST, {.i64=STACK}, 0, 0, FLAGS, "display" },
121 { "parade", NULL, 0, AV_OPT_TYPE_CONST, {.i64=PARADE}, 0, 0, FLAGS, "display" },
122 { "components", "set components to display", OFFSET(pcomp), AV_OPT_TYPE_INT, {.i64=1}, 1, 15, FLAGS },
123 { "c", "set components to display", OFFSET(pcomp), AV_OPT_TYPE_INT, {.i64=1}, 1, 15, FLAGS },
124 { "envelope", "set envelope to display", OFFSET(envelope), AV_OPT_TYPE_INT, {.i64=0}, 0, 3, FLAGS, "envelope" },
125 { "e", "set envelope to display", OFFSET(envelope), AV_OPT_TYPE_INT, {.i64=0}, 0, 3, FLAGS, "envelope" },
126 { "none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "envelope" },
127 { "instant", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "envelope" },
128 { "peak", NULL, 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, FLAGS, "envelope" },
129 { "peak+instant", NULL, 0, AV_OPT_TYPE_CONST, {.i64=3}, 0, 0, FLAGS, "envelope" },
130 { "filter", "set filter", OFFSET(filter), AV_OPT_TYPE_INT, {.i64=0}, 0, NB_FILTERS-1, FLAGS, "filter" },
131 { "f", "set filter", OFFSET(filter), AV_OPT_TYPE_INT, {.i64=0}, 0, NB_FILTERS-1, FLAGS, "filter" },
132 { "lowpass", NULL, 0, AV_OPT_TYPE_CONST, {.i64=LOWPASS}, 0, 0, FLAGS, "filter" },
133 { "flat" , NULL, 0, AV_OPT_TYPE_CONST, {.i64=FLAT}, 0, 0, FLAGS, "filter" },
134 { "aflat" , NULL, 0, AV_OPT_TYPE_CONST, {.i64=AFLAT}, 0, 0, FLAGS, "filter" },
135 { "chroma", NULL, 0, AV_OPT_TYPE_CONST, {.i64=CHROMA}, 0, 0, FLAGS, "filter" },
136 { "color", NULL, 0, AV_OPT_TYPE_CONST, {.i64=COLOR}, 0, 0, FLAGS, "filter" },
137 { "acolor", NULL, 0, AV_OPT_TYPE_CONST, {.i64=ACOLOR}, 0, 0, FLAGS, "filter" },
138 { "graticule", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "graticule" },
139 { "g", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "graticule" },
140 { "none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "graticule" },
141 { "green", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "graticule" },
142 { "opacity", "set graticule opacity", OFFSET(opacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, FLAGS },
143 { "o", "set graticule opacity", OFFSET(opacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, FLAGS },
144 { "flags", "set graticule flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64=1}, 0, 3, FLAGS, "flags" },
145 { "fl", "set graticule flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64=1}, 0, 3, FLAGS, "flags" },
146 { "numbers", "draw numbers", 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "flags" },
147 { "dots", "draw dots instead of lines", 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, FLAGS, "flags" },
148 { "scale", "set scale", OFFSET(scale), AV_OPT_TYPE_INT, {.i64=0}, 0, NB_SCALES-1, FLAGS, "scale" },
149 { "s", "set scale", OFFSET(scale), AV_OPT_TYPE_INT, {.i64=0}, 0, NB_SCALES-1, FLAGS, "scale" },
150 { "digital", NULL, 0, AV_OPT_TYPE_CONST, {.i64=DIGITAL}, 0, 0, FLAGS, "scale" },
151 { "millivolts", NULL, 0, AV_OPT_TYPE_CONST, {.i64=MILLIVOLTS}, 0, 0, FLAGS, "scale" },
152 { "ire", NULL, 0, AV_OPT_TYPE_CONST, {.i64=IRE}, 0, 0, FLAGS, "scale" },
153 { "bgopacity", "set background opacity", OFFSET(bgopacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, FLAGS },
154 { "b", "set background opacity", OFFSET(bgopacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, FLAGS },
155 { NULL }
156};
157
158AVFILTER_DEFINE_CLASS(waveform);
159
160static const enum AVPixelFormat in_lowpass_pix_fmts[] = {
161 AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRAP,
162 AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10, AV_PIX_FMT_GBRP12,
163 AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV420P,
164 AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV440P,
165 AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV410P,
166 AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ411P, AV_PIX_FMT_YUVJ420P,
167 AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P,
168 AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUVA420P,
169 AV_PIX_FMT_GRAY8, AV_PIX_FMT_GRAY10, AV_PIX_FMT_GRAY12,
170 AV_PIX_FMT_YUV444P9, AV_PIX_FMT_YUV422P9, AV_PIX_FMT_YUV420P9,
171 AV_PIX_FMT_YUVA444P9, AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA420P9,
172 AV_PIX_FMT_YUV444P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV420P10,
173 AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA420P10,
174 AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV422P12, AV_PIX_FMT_YUV420P12, AV_PIX_FMT_YUV440P12,
175 AV_PIX_FMT_NONE
176};
177
178static const enum AVPixelFormat in_color_pix_fmts[] = {
179 AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRAP,
180 AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10, AV_PIX_FMT_GBRP12,
181 AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV420P,
182 AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV440P,
183 AV_PIX_FMT_YUV411P,
184 AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ411P, AV_PIX_FMT_YUVJ420P,
185 AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P,
186 AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUVA420P,
187 AV_PIX_FMT_YUV444P9, AV_PIX_FMT_YUV422P9, AV_PIX_FMT_YUV420P9,
188 AV_PIX_FMT_YUVA444P9, AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA420P9,
189 AV_PIX_FMT_YUV444P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV420P10,
190 AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA420P10,
191 AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV422P12, AV_PIX_FMT_YUV420P12, AV_PIX_FMT_YUV440P12,
192 AV_PIX_FMT_NONE
193};
194
195static const enum AVPixelFormat in_flat_pix_fmts[] = {
196 AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV420P,
197 AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV440P,
198 AV_PIX_FMT_YUV411P,
199 AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ411P, AV_PIX_FMT_YUVJ420P,
200 AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P,
201 AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUVA420P,
202 AV_PIX_FMT_YUV444P9, AV_PIX_FMT_YUV422P9, AV_PIX_FMT_YUV420P9,
203 AV_PIX_FMT_YUVA444P9, AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA420P9,
204 AV_PIX_FMT_YUV444P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV420P10,
205 AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA420P10,
206 AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV422P12, AV_PIX_FMT_YUV420P12, AV_PIX_FMT_YUV440P12,
207 AV_PIX_FMT_NONE
208};
209
210static const enum AVPixelFormat out_rgb8_lowpass_pix_fmts[] = {
211 AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRAP,
212 AV_PIX_FMT_NONE
213};
214
215static const enum AVPixelFormat out_rgb9_lowpass_pix_fmts[] = {
216 AV_PIX_FMT_GBRP9,
217 AV_PIX_FMT_NONE
218};
219
220static const enum AVPixelFormat out_rgb10_lowpass_pix_fmts[] = {
221 AV_PIX_FMT_GBRP10,
222 AV_PIX_FMT_NONE
223};
224
225static const enum AVPixelFormat out_rgb12_lowpass_pix_fmts[] = {
226 AV_PIX_FMT_GBRP12,
227 AV_PIX_FMT_NONE
228};
229
230static const enum AVPixelFormat out_yuv8_lowpass_pix_fmts[] = {
231 AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVA444P,
232 AV_PIX_FMT_NONE
233};
234
235static const enum AVPixelFormat out_yuv9_lowpass_pix_fmts[] = {
236 AV_PIX_FMT_YUV444P9, AV_PIX_FMT_YUVA444P9,
237 AV_PIX_FMT_NONE
238};
239
240static const enum AVPixelFormat out_yuv10_lowpass_pix_fmts[] = {
241 AV_PIX_FMT_YUV444P10, AV_PIX_FMT_YUVA444P10,
242 AV_PIX_FMT_NONE
243};
244
245static const enum AVPixelFormat out_yuv12_lowpass_pix_fmts[] = {
246 AV_PIX_FMT_YUV444P12,
247 AV_PIX_FMT_NONE
248};
249
250static const enum AVPixelFormat out_gray8_lowpass_pix_fmts[] = {
251 AV_PIX_FMT_GRAY8,
252 AV_PIX_FMT_NONE
253};
254
255static const enum AVPixelFormat out_gray10_lowpass_pix_fmts[] = {
256 AV_PIX_FMT_GRAY10,
257 AV_PIX_FMT_NONE
258};
259
260static const enum AVPixelFormat out_gray12_lowpass_pix_fmts[] = {
261 AV_PIX_FMT_GRAY12,
262 AV_PIX_FMT_NONE
263};
264
265static const enum AVPixelFormat flat_pix_fmts[] = {
266 AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUVJ444P,
267 AV_PIX_FMT_YUV444P9, AV_PIX_FMT_YUV444P10,
268 AV_PIX_FMT_YUV444P12,
269 AV_PIX_FMT_NONE
270};
271
272static int query_formats(AVFilterContext *ctx)
273{
274 WaveformContext *s = ctx->priv;
275 const enum AVPixelFormat *out_pix_fmts;
276 const enum AVPixelFormat *in_pix_fmts;
277 const AVPixFmtDescriptor *desc;
278 AVFilterFormats *avff;
279 int depth, rgb, i, ret, ncomp;
280
281 if (!ctx->inputs[0]->in_formats ||
282 !ctx->inputs[0]->in_formats->nb_formats) {
283 return AVERROR(EAGAIN);
284 }
285
286 switch (s->filter) {
287 case LOWPASS: in_pix_fmts = in_lowpass_pix_fmts; break;
288 case CHROMA:
289 case AFLAT:
290 case FLAT: in_pix_fmts = in_flat_pix_fmts; break;
291 case ACOLOR:
292 case COLOR: in_pix_fmts = in_color_pix_fmts; break;
293 }
294
295 if (!ctx->inputs[0]->out_formats) {
296 if ((ret = ff_formats_ref(ff_make_format_list(in_pix_fmts), &ctx->inputs[0]->out_formats)) < 0)
297 return ret;
298 }
299
300 avff = ctx->inputs[0]->in_formats;
301 desc = av_pix_fmt_desc_get(avff->formats[0]);
302 ncomp = desc->nb_components;
303 rgb = desc->flags & AV_PIX_FMT_FLAG_RGB;
304 depth = desc->comp[0].depth;
305 for (i = 1; i < avff->nb_formats; i++) {
306 desc = av_pix_fmt_desc_get(avff->formats[i]);
307 if (rgb != (desc->flags & AV_PIX_FMT_FLAG_RGB) ||
308 depth != desc->comp[0].depth)
309 return AVERROR(EAGAIN);
310 }
311
312 if (s->filter == LOWPASS && ncomp == 1 && depth == 8)
313 out_pix_fmts = out_gray8_lowpass_pix_fmts;
314 else if (s->filter == LOWPASS && ncomp == 1 && depth == 10)
315 out_pix_fmts = out_gray10_lowpass_pix_fmts;
316 else if (s->filter == LOWPASS && ncomp == 1 && depth == 12)
317 out_pix_fmts = out_gray12_lowpass_pix_fmts;
318 else if (rgb && depth == 8 && ncomp > 2)
319 out_pix_fmts = out_rgb8_lowpass_pix_fmts;
320 else if (rgb && depth == 9 && ncomp > 2)
321 out_pix_fmts = out_rgb9_lowpass_pix_fmts;
322 else if (rgb && depth == 10 && ncomp > 2)
323 out_pix_fmts = out_rgb10_lowpass_pix_fmts;
324 else if (rgb && depth == 12 && ncomp > 2)
325 out_pix_fmts = out_rgb12_lowpass_pix_fmts;
326 else if (depth == 8 && ncomp > 2)
327 out_pix_fmts = out_yuv8_lowpass_pix_fmts;
328 else if (depth == 9 && ncomp > 2)
329 out_pix_fmts = out_yuv9_lowpass_pix_fmts;
330 else if (depth == 10 && ncomp > 2)
331 out_pix_fmts = out_yuv10_lowpass_pix_fmts;
332 else if (depth == 12 && ncomp > 2)
333 out_pix_fmts = out_yuv12_lowpass_pix_fmts;
334 else
335 return AVERROR(EAGAIN);
336 if ((ret = ff_formats_ref(ff_make_format_list(out_pix_fmts), &ctx->outputs[0]->in_formats)) < 0)
337 return ret;
338
339 return 0;
340}
341
342static void envelope_instant16(WaveformContext *s, AVFrame *out, int plane, int component, int offset)
343{
344 const int dst_linesize = out->linesize[component] / 2;
345 const int bg = s->bg_color[component] * (s->max / 256);
346 const int limit = s->max - 1;
347 const int dst_h = s->display == PARADE ? out->height / s->acomp : out->height;
348 const int dst_w = s->display == PARADE ? out->width / s->acomp : out->width;
349 const int start = s->estart[plane];
350 const int end = s->eend[plane];
351 uint16_t *dst;
352 int x, y;
353
354 if (s->mode) {
355 for (x = offset; x < offset + dst_w; x++) {
356 for (y = start; y < end; y++) {
357 dst = (uint16_t *)out->data[component] + y * dst_linesize + x;
358 if (dst[0] != bg) {
359 dst[0] = limit;
360 break;
361 }
362 }
363 for (y = end - 1; y >= start; y--) {
364 dst = (uint16_t *)out->data[component] + y * dst_linesize + x;
365 if (dst[0] != bg) {
366 dst[0] = limit;
367 break;
368 }
369 }
370 }
371 } else {
372 for (y = offset; y < offset + dst_h; y++) {
373 dst = (uint16_t *)out->data[component] + y * dst_linesize;
374 for (x = start; x < end; x++) {
375 if (dst[x] != bg) {
376 dst[x] = limit;
377 break;
378 }
379 }
380 for (x = end - 1; x >= start; x--) {
381 if (dst[x] != bg) {
382 dst[x] = limit;
383 break;
384 }
385 }
386 }
387 }
388}
389
390static void envelope_instant(WaveformContext *s, AVFrame *out, int plane, int component, int offset)
391{
392 const int dst_linesize = out->linesize[component];
393 const uint8_t bg = s->bg_color[component];
394 const int dst_h = s->display == PARADE ? out->height / s->acomp : out->height;
395 const int dst_w = s->display == PARADE ? out->width / s->acomp : out->width;
396 const int start = s->estart[plane];
397 const int end = s->eend[plane];
398 uint8_t *dst;
399 int x, y;
400
401 if (s->mode) {
402 for (x = offset; x < offset + dst_w; x++) {
403 for (y = start; y < end; y++) {
404 dst = out->data[component] + y * dst_linesize + x;
405 if (dst[0] != bg) {
406 dst[0] = 255;
407 break;
408 }
409 }
410 for (y = end - 1; y >= start; y--) {
411 dst = out->data[component] + y * dst_linesize + x;
412 if (dst[0] != bg) {
413 dst[0] = 255;
414 break;
415 }
416 }
417 }
418 } else {
419 for (y = offset; y < offset + dst_h; y++) {
420 dst = out->data[component] + y * dst_linesize;
421 for (x = start; x < end; x++) {
422 if (dst[x] != bg) {
423 dst[x] = 255;
424 break;
425 }
426 }
427 for (x = end - 1; x >= start; x--) {
428 if (dst[x] != bg) {
429 dst[x] = 255;
430 break;
431 }
432 }
433 }
434 }
435}
436
437static void envelope_peak16(WaveformContext *s, AVFrame *out, int plane, int component, int offset)
438{
439 const int dst_linesize = out->linesize[component] / 2;
440 const int bg = s->bg_color[component] * (s->max / 256);
441 const int limit = s->max - 1;
442 const int dst_h = s->display == PARADE ? out->height / s->acomp : out->height;
443 const int dst_w = s->display == PARADE ? out->width / s->acomp : out->width;
444 const int start = s->estart[plane];
445 const int end = s->eend[plane];
446 int *emax = s->emax[plane][component];
447 int *emin = s->emin[plane][component];
448 uint16_t *dst;
449 int x, y;
450
451 if (s->mode) {
452 for (x = offset; x < offset + dst_w; x++) {
453 for (y = start; y < end && y < emin[x - offset]; y++) {
454 dst = (uint16_t *)out->data[component] + y * dst_linesize + x;
455 if (dst[0] != bg) {
456 emin[x - offset] = y;
457 break;
458 }
459 }
460 for (y = end - 1; y >= start && y >= emax[x - offset]; y--) {
461 dst = (uint16_t *)out->data[component] + y * dst_linesize + x;
462 if (dst[0] != bg) {
463 emax[x - offset] = y;
464 break;
465 }
466 }
467 }
468
469 if (s->envelope == 3)
470 envelope_instant16(s, out, plane, component, offset);
471
472 for (x = offset; x < offset + dst_w; x++) {
473 dst = (uint16_t *)out->data[component] + emin[x - offset] * dst_linesize + x;
474 dst[0] = limit;
475 dst = (uint16_t *)out->data[component] + emax[x - offset] * dst_linesize + x;
476 dst[0] = limit;
477 }
478 } else {
479 for (y = offset; y < offset + dst_h; y++) {
480 dst = (uint16_t *)out->data[component] + y * dst_linesize;
481 for (x = start; x < end && x < emin[y - offset]; x++) {
482 if (dst[x] != bg) {
483 emin[y - offset] = x;
484 break;
485 }
486 }
487 for (x = end - 1; x >= start && x >= emax[y - offset]; x--) {
488 if (dst[x] != bg) {
489 emax[y - offset] = x;
490 break;
491 }
492 }
493 }
494
495 if (s->envelope == 3)
496 envelope_instant16(s, out, plane, component, offset);
497
498 for (y = offset; y < offset + dst_h; y++) {
499 dst = (uint16_t *)out->data[component] + y * dst_linesize + emin[y - offset];
500 dst[0] = limit;
501 dst = (uint16_t *)out->data[component] + y * dst_linesize + emax[y - offset];
502 dst[0] = limit;
503 }
504 }
505}
506
507static void envelope_peak(WaveformContext *s, AVFrame *out, int plane, int component, int offset)
508{
509 const int dst_linesize = out->linesize[component];
510 const int bg = s->bg_color[component];
511 const int dst_h = s->display == PARADE ? out->height / s->acomp : out->height;
512 const int dst_w = s->display == PARADE ? out->width / s->acomp : out->width;
513 const int start = s->estart[plane];
514 const int end = s->eend[plane];
515 int *emax = s->emax[plane][component];
516 int *emin = s->emin[plane][component];
517 uint8_t *dst;
518 int x, y;
519
520 if (s->mode) {
521 for (x = offset; x < offset + dst_w; x++) {
522 for (y = start; y < end && y < emin[x - offset]; y++) {
523 dst = out->data[component] + y * dst_linesize + x;
524 if (dst[0] != bg) {
525 emin[x - offset] = y;
526 break;
527 }
528 }
529 for (y = end - 1; y >= start && y >= emax[x - offset]; y--) {
530 dst = out->data[component] + y * dst_linesize + x;
531 if (dst[0] != bg) {
532 emax[x - offset] = y;
533 break;
534 }
535 }
536 }
537
538 if (s->envelope == 3)
539 envelope_instant(s, out, plane, component, offset);
540
541 for (x = offset; x < offset + dst_w; x++) {
542 dst = out->data[component] + emin[x - offset] * dst_linesize + x;
543 dst[0] = 255;
544 dst = out->data[component] + emax[x - offset] * dst_linesize + x;
545 dst[0] = 255;
546 }
547 } else {
548 for (y = offset; y < offset + dst_h; y++) {
549 dst = out->data[component] + y * dst_linesize;
550 for (x = start; x < end && x < emin[y - offset]; x++) {
551 if (dst[x] != bg) {
552 emin[y - offset] = x;
553 break;
554 }
555 }
556 for (x = end - 1; x >= start && x >= emax[y - offset]; x--) {
557 if (dst[x] != bg) {
558 emax[y - offset] = x;
559 break;
560 }
561 }
562 }
563
564 if (s->envelope == 3)
565 envelope_instant(s, out, plane, component, offset);
566
567 for (y = offset; y < offset + dst_h; y++) {
568 dst = out->data[component] + y * dst_linesize + emin[y - offset];
569 dst[0] = 255;
570 dst = out->data[component] + y * dst_linesize + emax[y - offset];
571 dst[0] = 255;
572 }
573 }
574}
575
576static void envelope16(WaveformContext *s, AVFrame *out, int plane, int component, int offset)
577{
578 if (s->envelope == 0) {
579 return;
580 } else if (s->envelope == 1) {
581 envelope_instant16(s, out, plane, component, offset);
582 } else {
583 envelope_peak16(s, out, plane, component, offset);
584 }
585}
586
587static void envelope(WaveformContext *s, AVFrame *out, int plane, int component, int offset)
588{
589 if (s->envelope == 0) {
590 return;
591 } else if (s->envelope == 1) {
592 envelope_instant(s, out, plane, component, offset);
593 } else {
594 envelope_peak(s, out, plane, component, offset);
595 }
596}
597
598static void update16(uint16_t *target, int max, int intensity, int limit)
599{
600 if (*target <= max)
601 *target += intensity;
602 else
603 *target = limit;
604}
605
606static void update(uint8_t *target, int max, int intensity)
607{
608 if (*target <= max)
609 *target += intensity;
610 else
611 *target = 255;
612}
613
614static av_always_inline void lowpass16(WaveformContext *s,
615 AVFrame *in, AVFrame *out,
616 int component, int intensity,
617 int offset_y, int offset_x,
618 int column, int mirror)
619{
620 const int plane = s->desc->comp[component].plane;
621 const int shift_w = s->shift_w[component];
622 const int shift_h = s->shift_h[component];
623 const int src_linesize = in->linesize[plane] / 2;
624 const int dst_linesize = out->linesize[plane] / 2;
625 const int dst_signed_linesize = dst_linesize * (mirror == 1 ? -1 : 1);
626 const int limit = s->max - 1;
627 const int max = limit - intensity;
628 const int src_h = AV_CEIL_RSHIFT(in->height, shift_h);
629 const int src_w = AV_CEIL_RSHIFT(in->width, shift_w);
630 const uint16_t *src_data = (const uint16_t *)in->data[plane];
631 uint16_t *dst_data = (uint16_t *)out->data[plane] + offset_y * dst_linesize + offset_x;
632 uint16_t * const dst_bottom_line = dst_data + dst_linesize * (s->size - 1);
633 uint16_t * const dst_line = (mirror ? dst_bottom_line : dst_data);
634 const int step = column ? 1 << shift_w : 1 << shift_h;
635 const uint16_t *p;
636 int y;
637
638 if (!column && mirror)
639 dst_data += s->size;
640
641 for (y = 0; y < src_h; y++) {
642 const uint16_t *src_data_end = src_data + src_w;
643 uint16_t *dst = dst_line;
644
645 for (p = src_data; p < src_data_end; p++) {
646 uint16_t *target;
647 int i = 0, v = FFMIN(*p, limit);
648
649 if (column) {
650 do {
651 target = dst++ + dst_signed_linesize * v;
652 update16(target, max, intensity, limit);
653 } while (++i < step);
654 } else {
655 uint16_t *row = dst_data;
656 do {
657 if (mirror)
658 target = row - v - 1;
659 else
660 target = row + v;
661 update16(target, max, intensity, limit);
662 row += dst_linesize;
663 } while (++i < step);
664 }
665 }
666 src_data += src_linesize;
667 dst_data += dst_linesize * step;
668 }
669
670 envelope16(s, out, plane, plane, column ? offset_x : offset_y);
671}
672
673#define LOWPASS16_FUNC(name, column, mirror) \
674static void lowpass16_##name(WaveformContext *s, \
675 AVFrame *in, AVFrame *out, \
676 int component, int intensity, \
677 int offset_y, int offset_x, \
678 int unused1, int unused2) \
679{ \
680 lowpass16(s, in, out, component, intensity, \
681 offset_y, offset_x, column, mirror); \
682}
683
684LOWPASS16_FUNC(column_mirror, 1, 1)
685LOWPASS16_FUNC(column, 1, 0)
686LOWPASS16_FUNC(row_mirror, 0, 1)
687LOWPASS16_FUNC(row, 0, 0)
688
689static av_always_inline void lowpass(WaveformContext *s,
690 AVFrame *in, AVFrame *out,
691 int component, int intensity,
692 int offset_y, int offset_x,
693 int column, int mirror)
694{
695 const int plane = s->desc->comp[component].plane;
696 const int shift_w = s->shift_w[component];
697 const int shift_h = s->shift_h[component];
698 const int src_linesize = in->linesize[plane];
699 const int dst_linesize = out->linesize[plane];
700 const int dst_signed_linesize = dst_linesize * (mirror == 1 ? -1 : 1);
701 const int max = 255 - intensity;
702 const int src_h = AV_CEIL_RSHIFT(in->height, shift_h);
703 const int src_w = AV_CEIL_RSHIFT(in->width, shift_w);
704 const uint8_t *src_data = in->data[plane];
705 uint8_t *dst_data = out->data[plane] + offset_y * dst_linesize + offset_x;
706 uint8_t * const dst_bottom_line = dst_data + dst_linesize * (s->size - 1);
707 uint8_t * const dst_line = (mirror ? dst_bottom_line : dst_data);
708 const int step = column ? 1 << shift_w : 1 << shift_h;
709 const uint8_t *p;
710 int y;
711
712 if (!column && mirror)
713 dst_data += s->size;
714
715 for (y = 0; y < src_h; y++) {
716 const uint8_t *src_data_end = src_data + src_w;
717 uint8_t *dst = dst_line;
718
719 for (p = src_data; p < src_data_end; p++) {
720 uint8_t *target;
721 if (column) {
722 target = dst + dst_signed_linesize * *p;
723 dst += step;
724 update(target, max, intensity);
725 } else {
726 uint8_t *row = dst_data;
727 if (mirror)
728 target = row - *p - 1;
729 else
730 target = row + *p;
731 update(target, max, intensity);
732 row += dst_linesize;
733 }
734 }
735 src_data += src_linesize;
736 dst_data += dst_linesize * step;
737 }
738
739 if (column && step > 1) {
740 const int dst_w = s->display == PARADE ? out->width / s->acomp : out->width;
741 const int dst_h = 256;
742 uint8_t *dst;
743 int x, z;
744
745 dst = out->data[plane] + offset_y * dst_linesize + offset_x;
746 for (y = 0; y < dst_h; y++) {
747 for (x = 0; x < dst_w; x+=step) {
748 for (z = 1; z < step; z++) {
749 dst[x + z] = dst[x];
750 }
751 }
752 dst += dst_linesize;
753 }
754 } else if (step > 1) {
755 const int dst_h = s->display == PARADE ? out->height / s->acomp : out->height;
756 const int dst_w = 256;
757 uint8_t *dst;
758 int z;
759
760 dst = out->data[plane] + offset_y * dst_linesize + offset_x;
761 for (y = 0; y < dst_h; y+=step) {
762 for (z = 1; z < step; z++)
763 memcpy(dst + dst_linesize * z, dst, dst_w);
764 dst += dst_linesize * step;
765 }
766 }
767
768 envelope(s, out, plane, plane, column ? offset_x : offset_y);
769}
770
771#define LOWPASS_FUNC(name, column, mirror) \
772static void lowpass_##name(WaveformContext *s, \
773 AVFrame *in, AVFrame *out, \
774 int component, int intensity, \
775 int offset_y, int offset_x, \
776 int unused1, int unused2) \
777{ \
778 lowpass(s, in, out, component, intensity, \
779 offset_y, offset_x, column, mirror); \
780}
781
782LOWPASS_FUNC(column_mirror, 1, 1)
783LOWPASS_FUNC(column, 1, 0)
784LOWPASS_FUNC(row_mirror, 0, 1)
785LOWPASS_FUNC(row, 0, 0)
786
787static av_always_inline void flat16(WaveformContext *s,
788 AVFrame *in, AVFrame *out,
789 int component, int intensity,
790 int offset_y, int offset_x,
791 int column, int mirror)
792{
793 const int plane = s->desc->comp[component].plane;
794 const int c0_linesize = in->linesize[ plane + 0 ] / 2;
795 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp] / 2;
796 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp] / 2;
797 const int c0_shift_w = s->shift_w[ component + 0 ];
798 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
799 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
800 const int c0_shift_h = s->shift_h[ component + 0 ];
801 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
802 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
803 const int d0_linesize = out->linesize[ plane + 0 ] / 2;
804 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp] / 2;
805 const int limit = s->max - 1;
806 const int max = limit - intensity;
807 const int mid = s->max / 2;
808 const int src_h = in->height;
809 const int src_w = in->width;
810 int x, y;
811
812 if (column) {
813 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
814 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
815
816 for (x = 0; x < src_w; x++) {
817 const uint16_t *c0_data = (uint16_t *)in->data[plane + 0];
818 const uint16_t *c1_data = (uint16_t *)in->data[(plane + 1) % s->ncomp];
819 const uint16_t *c2_data = (uint16_t *)in->data[(plane + 2) % s->ncomp];
820 uint16_t *d0_data = (uint16_t *)(out->data[plane]) + offset_y * d0_linesize + offset_x;
821 uint16_t *d1_data = (uint16_t *)(out->data[(plane + 1) % s->ncomp]) + offset_y * d1_linesize + offset_x;
822 uint16_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
823 uint16_t * const d0 = (mirror ? d0_bottom_line : d0_data);
824 uint16_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
825 uint16_t * const d1 = (mirror ? d1_bottom_line : d1_data);
826
827 for (y = 0; y < src_h; y++) {
828 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit) + s->max;
829 const int c1 = FFMIN(FFABS(c1_data[x >> c1_shift_w] - mid) + FFABS(c2_data[x >> c2_shift_w] - mid), limit);
830 uint16_t *target;
831
832 target = d0 + x + d0_signed_linesize * c0;
833 update16(target, max, intensity, limit);
834 target = d1 + x + d1_signed_linesize * (c0 - c1);
835 update16(target, max, intensity, limit);
836 target = d1 + x + d1_signed_linesize * (c0 + c1);
837 update16(target, max, intensity, limit);
838
839 if (!c0_shift_h || (y & c0_shift_h))
840 c0_data += c0_linesize;
841 if (!c1_shift_h || (y & c1_shift_h))
842 c1_data += c1_linesize;
843 if (!c2_shift_h || (y & c2_shift_h))
844 c2_data += c2_linesize;
845 d0_data += d0_linesize;
846 d1_data += d1_linesize;
847 }
848 }
849 } else {
850 const uint16_t *c0_data = (uint16_t *)in->data[plane];
851 const uint16_t *c1_data = (uint16_t *)in->data[(plane + 1) % s->ncomp];
852 const uint16_t *c2_data = (uint16_t *)in->data[(plane + 2) % s->ncomp];
853 uint16_t *d0_data = (uint16_t *)(out->data[plane]) + offset_y * d0_linesize + offset_x;
854 uint16_t *d1_data = (uint16_t *)(out->data[(plane + 1) % s->ncomp]) + offset_y * d1_linesize + offset_x;
855
856 if (mirror) {
857 d0_data += s->size - 1;
858 d1_data += s->size - 1;
859 }
860
861 for (y = 0; y < src_h; y++) {
862 for (x = 0; x < src_w; x++) {
863 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit) + s->max;
864 const int c1 = FFMIN(FFABS(c1_data[x >> c1_shift_w] - mid) + FFABS(c2_data[x >> c2_shift_w] - mid), limit);
865 uint16_t *target;
866
867 if (mirror) {
868 target = d0_data - c0;
869 update16(target, max, intensity, limit);
870 target = d1_data - (c0 - c1);
871 update16(target, max, intensity, limit);
872 target = d1_data - (c0 + c1);
873 update16(target, max, intensity, limit);
874 } else {
875 target = d0_data + c0;
876 update16(target, max, intensity, limit);
877 target = d1_data + (c0 - c1);
878 update16(target, max, intensity, limit);
879 target = d1_data + (c0 + c1);
880 update16(target, max, intensity, limit);
881 }
882 }
883
884 if (!c0_shift_h || (y & c0_shift_h))
885 c0_data += c0_linesize;
886 if (!c1_shift_h || (y & c1_shift_h))
887 c1_data += c1_linesize;
888 if (!c2_shift_h || (y & c2_shift_h))
889 c2_data += c2_linesize;
890 d0_data += d0_linesize;
891 d1_data += d1_linesize;
892 }
893 }
894
895 envelope16(s, out, plane, plane, column ? offset_x : offset_y);
896 envelope16(s, out, plane, (plane + 1) % s->ncomp, column ? offset_x : offset_y);
897}
898
899static av_always_inline void flat(WaveformContext *s,
900 AVFrame *in, AVFrame *out,
901 int component, int intensity,
902 int offset_y, int offset_x,
903 int column, int mirror)
904{
905 const int plane = s->desc->comp[component].plane;
906 const int c0_linesize = in->linesize[ plane + 0 ];
907 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp];
908 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp];
909 const int c0_shift_w = s->shift_w[ component + 0 ];
910 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
911 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
912 const int c0_shift_h = s->shift_h[ component + 0 ];
913 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
914 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
915 const int d0_linesize = out->linesize[ plane + 0 ];
916 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp];
917 const int max = 255 - intensity;
918 const int src_h = in->height;
919 const int src_w = in->width;
920 int x, y;
921
922 if (column) {
923 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
924 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
925
926 for (x = 0; x < src_w; x++) {
927 const uint8_t *c0_data = in->data[plane + 0];
928 const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
929 const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
930 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
931 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
932 uint8_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
933 uint8_t * const d0 = (mirror ? d0_bottom_line : d0_data);
934 uint8_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
935 uint8_t * const d1 = (mirror ? d1_bottom_line : d1_data);
936
937 for (y = 0; y < src_h; y++) {
938 const int c0 = c0_data[x >> c0_shift_w] + 256;
939 const int c1 = FFABS(c1_data[x >> c1_shift_w] - 128) + FFABS(c2_data[x >> c2_shift_w] - 128);
940 uint8_t *target;
941
942 target = d0 + x + d0_signed_linesize * c0;
943 update(target, max, intensity);
944 target = d1 + x + d1_signed_linesize * (c0 - c1);
945 update(target, max, intensity);
946 target = d1 + x + d1_signed_linesize * (c0 + c1);
947 update(target, max, intensity);
948
949 if (!c0_shift_h || (y & c0_shift_h))
950 c0_data += c0_linesize;
951 if (!c1_shift_h || (y & c1_shift_h))
952 c1_data += c1_linesize;
953 if (!c2_shift_h || (y & c2_shift_h))
954 c2_data += c2_linesize;
955 d0_data += d0_linesize;
956 d1_data += d1_linesize;
957 }
958 }
959 } else {
960 const uint8_t *c0_data = in->data[plane];
961 const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
962 const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
963 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
964 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
965
966 if (mirror) {
967 d0_data += s->size - 1;
968 d1_data += s->size - 1;
969 }
970
971 for (y = 0; y < src_h; y++) {
972 for (x = 0; x < src_w; x++) {
973 int c0 = c0_data[x >> c0_shift_w] + 256;
974 const int c1 = FFABS(c1_data[x >> c1_shift_w] - 128) + FFABS(c2_data[x >> c2_shift_w] - 128);
975 uint8_t *target;
976
977 if (mirror) {
978 target = d0_data - c0;
979 update(target, max, intensity);
980 target = d1_data - (c0 - c1);
981 update(target, max, intensity);
982 target = d1_data - (c0 + c1);
983 update(target, max, intensity);
984 } else {
985 target = d0_data + c0;
986 update(target, max, intensity);
987 target = d1_data + (c0 - c1);
988 update(target, max, intensity);
989 target = d1_data + (c0 + c1);
990 update(target, max, intensity);
991 }
992 }
993
994 if (!c0_shift_h || (y & c0_shift_h))
995 c0_data += c0_linesize;
996 if (!c1_shift_h || (y & c1_shift_h))
997 c1_data += c1_linesize;
998 if (!c2_shift_h || (y & c2_shift_h))
999 c2_data += c2_linesize;
1000 d0_data += d0_linesize;
1001 d1_data += d1_linesize;
1002 }
1003 }
1004
1005 envelope(s, out, plane, plane, column ? offset_x : offset_y);
1006 envelope(s, out, plane, (plane + 1) % s->ncomp, column ? offset_x : offset_y);
1007}
1008
1009static av_always_inline void aflat16(WaveformContext *s,
1010 AVFrame *in, AVFrame *out,
1011 int component, int intensity,
1012 int offset_y, int offset_x,
1013 int column, int mirror)
1014{
1015 const int plane = s->desc->comp[component].plane;
1016 const int c0_linesize = in->linesize[ plane + 0 ] / 2;
1017 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp] / 2;
1018 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp] / 2;
1019 const int c0_shift_w = s->shift_w[ component + 0 ];
1020 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
1021 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
1022 const int c0_shift_h = s->shift_h[ component + 0 ];
1023 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
1024 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
1025 const int d0_linesize = out->linesize[ plane + 0 ] / 2;
1026 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp] / 2;
1027 const int d2_linesize = out->linesize[(plane + 2) % s->ncomp] / 2;
1028 const int limit = s->max - 1;
1029 const int max = limit - intensity;
1030 const int mid = s->max / 2;
1031 const int src_h = in->height;
1032 const int src_w = in->width;
1033 int x, y;
1034
1035 if (column) {
1036 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
1037 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
1038 const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1);
1039
1040 for (x = 0; x < src_w; x++) {
1041 const uint16_t *c0_data = (uint16_t *)in->data[plane + 0];
1042 const uint16_t *c1_data = (uint16_t *)in->data[(plane + 1) % s->ncomp];
1043 const uint16_t *c2_data = (uint16_t *)in->data[(plane + 2) % s->ncomp];
1044 uint16_t *d0_data = (uint16_t *)out->data[plane] + offset_y * d0_linesize + offset_x;
1045 uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1046 uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1047 uint16_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
1048 uint16_t * const d0 = (mirror ? d0_bottom_line : d0_data);
1049 uint16_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
1050 uint16_t * const d1 = (mirror ? d1_bottom_line : d1_data);
1051 uint16_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1);
1052 uint16_t * const d2 = (mirror ? d2_bottom_line : d2_data);
1053
1054 for (y = 0; y < src_h; y++) {
1055 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit) + mid;
1056 const int c1 = FFMIN(c1_data[x >> c1_shift_w], limit) - mid;
1057 const int c2 = FFMIN(c2_data[x >> c2_shift_w], limit) - mid;
1058 uint16_t *target;
1059
1060 target = d0 + x + d0_signed_linesize * c0;
1061 update16(target, max, intensity, limit);
1062
1063 target = d1 + x + d1_signed_linesize * (c0 + c1);
1064 update16(target, max, intensity, limit);
1065
1066 target = d2 + x + d2_signed_linesize * (c0 + c2);
1067 update16(target, max, intensity, limit);
1068
1069 if (!c0_shift_h || (y & c0_shift_h))
1070 c0_data += c0_linesize;
1071 if (!c1_shift_h || (y & c1_shift_h))
1072 c1_data += c1_linesize;
1073 if (!c2_shift_h || (y & c2_shift_h))
1074 c2_data += c2_linesize;
1075 d0_data += d0_linesize;
1076 d1_data += d1_linesize;
1077 d2_data += d2_linesize;
1078 }
1079 }
1080 } else {
1081 const uint16_t *c0_data = (uint16_t *)in->data[plane];
1082 const uint16_t *c1_data = (uint16_t *)in->data[(plane + 1) % s->ncomp];
1083 const uint16_t *c2_data = (uint16_t *)in->data[(plane + 2) % s->ncomp];
1084 uint16_t *d0_data = (uint16_t *)out->data[plane] + offset_y * d0_linesize + offset_x;
1085 uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1086 uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1087
1088 if (mirror) {
1089 d0_data += s->size - 1;
1090 d1_data += s->size - 1;
1091 d2_data += s->size - 1;
1092 }
1093
1094 for (y = 0; y < src_h; y++) {
1095 for (x = 0; x < src_w; x++) {
1096 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit) + mid;
1097 const int c1 = FFMIN(c1_data[x >> c1_shift_w], limit) - mid;
1098 const int c2 = FFMIN(c2_data[x >> c2_shift_w], limit) - mid;
1099 uint16_t *target;
1100
1101 if (mirror) {
1102 target = d0_data - c0;
1103 update16(target, max, intensity, limit);
1104 target = d1_data - (c0 + c1);
1105 update16(target, max, intensity, limit);
1106 target = d2_data - (c0 + c2);
1107 update16(target, max, intensity, limit);
1108 } else {
1109 target = d0_data + c0;
1110 update16(target, max, intensity, limit);
1111 target = d1_data + (c0 + c1);
1112 update16(target, max, intensity, limit);
1113 target = d2_data + (c0 + c2);
1114 update16(target, max, intensity, limit);
1115 }
1116 }
1117
1118 if (!c0_shift_h || (y & c0_shift_h))
1119 c0_data += c0_linesize;
1120 if (!c1_shift_h || (y & c1_shift_h))
1121 c1_data += c1_linesize;
1122 if (!c2_shift_h || (y & c2_shift_h))
1123 c2_data += c2_linesize;
1124 d0_data += d0_linesize;
1125 d1_data += d1_linesize;
1126 d2_data += d2_linesize;
1127 }
1128 }
1129
1130 envelope16(s, out, plane, (plane + 0) % s->ncomp, column ? offset_x : offset_y);
1131 envelope16(s, out, plane, (plane + 1) % s->ncomp, column ? offset_x : offset_y);
1132 envelope16(s, out, plane, (plane + 2) % s->ncomp, column ? offset_x : offset_y);
1133}
1134
1135static av_always_inline void aflat(WaveformContext *s,
1136 AVFrame *in, AVFrame *out,
1137 int component, int intensity,
1138 int offset_y, int offset_x,
1139 int column, int mirror)
1140{
1141 const int plane = s->desc->comp[component].plane;
1142 const int c0_linesize = in->linesize[ plane + 0 ];
1143 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp];
1144 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp];
1145 const int c0_shift_w = s->shift_w[ component + 0 ];
1146 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
1147 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
1148 const int c0_shift_h = s->shift_h[ component + 0 ];
1149 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
1150 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
1151 const int d0_linesize = out->linesize[ plane + 0 ];
1152 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp];
1153 const int d2_linesize = out->linesize[(plane + 2) % s->ncomp];
1154 const int max = 255 - intensity;
1155 const int src_h = in->height;
1156 const int src_w = in->width;
1157 int x, y;
1158
1159 if (column) {
1160 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
1161 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
1162 const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1);
1163
1164 for (x = 0; x < src_w; x++) {
1165 const uint8_t *c0_data = in->data[plane + 0];
1166 const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
1167 const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
1168 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
1169 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1170 uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1171 uint8_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
1172 uint8_t * const d0 = (mirror ? d0_bottom_line : d0_data);
1173 uint8_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
1174 uint8_t * const d1 = (mirror ? d1_bottom_line : d1_data);
1175 uint8_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1);
1176 uint8_t * const d2 = (mirror ? d2_bottom_line : d2_data);
1177
1178 for (y = 0; y < src_h; y++) {
1179 const int c0 = c0_data[x >> c0_shift_w] + 128;
1180 const int c1 = c1_data[x >> c1_shift_w] - 128;
1181 const int c2 = c2_data[x >> c2_shift_w] - 128;
1182 uint8_t *target;
1183
1184 target = d0 + x + d0_signed_linesize * c0;
1185 update(target, max, intensity);
1186
1187 target = d1 + x + d1_signed_linesize * (c0 + c1);
1188 update(target, max, intensity);
1189
1190 target = d2 + x + d2_signed_linesize * (c0 + c2);
1191 update(target, max, intensity);
1192
1193 if (!c0_shift_h || (y & c0_shift_h))
1194 c0_data += c0_linesize;
1195 if (!c1_shift_h || (y & c1_shift_h))
1196 c1_data += c1_linesize;
1197 if (!c1_shift_h || (y & c1_shift_h))
1198 c2_data += c1_linesize;
1199 d0_data += d0_linesize;
1200 d1_data += d1_linesize;
1201 d2_data += d2_linesize;
1202 }
1203 }
1204 } else {
1205 const uint8_t *c0_data = in->data[plane];
1206 const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
1207 const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
1208 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
1209 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1210 uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1211
1212 if (mirror) {
1213 d0_data += s->size - 1;
1214 d1_data += s->size - 1;
1215 d2_data += s->size - 1;
1216 }
1217
1218 for (y = 0; y < src_h; y++) {
1219 for (x = 0; x < src_w; x++) {
1220 const int c0 = c0_data[x >> c0_shift_w] + 128;
1221 const int c1 = c1_data[x >> c1_shift_w] - 128;
1222 const int c2 = c2_data[x >> c2_shift_w] - 128;
1223 uint8_t *target;
1224
1225 if (mirror) {
1226 target = d0_data - c0;
1227 update(target, max, intensity);
1228 target = d1_data - (c0 + c1);
1229 update(target, max, intensity);
1230 target = d2_data - (c0 + c2);
1231 update(target, max, intensity);
1232 } else {
1233 target = d0_data + c0;
1234 update(target, max, intensity);
1235 target = d1_data + (c0 + c1);
1236 update(target, max, intensity);
1237 target = d2_data + (c0 + c2);
1238 update(target, max, intensity);
1239 }
1240 }
1241
1242 if (!c0_shift_h || (y & c0_shift_h))
1243 c0_data += c0_linesize;
1244 if (!c1_shift_h || (y & c1_shift_h))
1245 c1_data += c1_linesize;
1246 if (!c2_shift_h || (y & c2_shift_h))
1247 c2_data += c2_linesize;
1248 d0_data += d0_linesize;
1249 d1_data += d1_linesize;
1250 d2_data += d2_linesize;
1251 }
1252 }
1253
1254 envelope(s, out, plane, (plane + 0) % s->ncomp, column ? offset_x : offset_y);
1255 envelope(s, out, plane, (plane + 1) % s->ncomp, column ? offset_x : offset_y);
1256 envelope(s, out, plane, (plane + 2) % s->ncomp, column ? offset_x : offset_y);
1257}
1258
1259static av_always_inline void chroma16(WaveformContext *s,
1260 AVFrame *in, AVFrame *out,
1261 int component, int intensity,
1262 int offset_y, int offset_x,
1263 int column, int mirror)
1264{
1265 const int plane = s->desc->comp[component].plane;
1266 const int c0_linesize = in->linesize[(plane + 1) % s->ncomp] / 2;
1267 const int c1_linesize = in->linesize[(plane + 2) % s->ncomp] / 2;
1268 const int dst_linesize = out->linesize[plane] / 2;
1269 const int limit = s->max - 1;
1270 const int max = limit - intensity;
1271 const int mid = s->max / 2;
1272 const int c0_shift_w = s->shift_w[(component + 1) % s->ncomp];
1273 const int c1_shift_w = s->shift_w[(component + 2) % s->ncomp];
1274 const int c0_shift_h = s->shift_h[(component + 1) % s->ncomp];
1275 const int c1_shift_h = s->shift_h[(component + 2) % s->ncomp];
1276 const int src_h = in->height;
1277 const int src_w = in->width;
1278 int x, y;
1279
1280 if (column) {
1281 const int dst_signed_linesize = dst_linesize * (mirror == 1 ? -1 : 1);
1282
1283 for (x = 0; x < src_w; x++) {
1284 const uint16_t *c0_data = (uint16_t *)in->data[(plane + 1) % s->ncomp];
1285 const uint16_t *c1_data = (uint16_t *)in->data[(plane + 2) % s->ncomp];
1286 uint16_t *dst_data = (uint16_t *)out->data[plane] + offset_y * dst_linesize + offset_x;
1287 uint16_t * const dst_bottom_line = dst_data + dst_linesize * (s->size - 1);
1288 uint16_t * const dst_line = (mirror ? dst_bottom_line : dst_data);
1289 uint16_t *dst = dst_line;
1290
1291 for (y = 0; y < src_h; y++) {
1292 const int sum = FFMIN(FFABS(c0_data[x >> c0_shift_w] - mid) + FFABS(c1_data[x >> c1_shift_w] - mid - 1), limit);
1293 uint16_t *target;
1294
1295 target = dst + x + dst_signed_linesize * sum;
1296 update16(target, max, intensity, limit);
1297
1298 if (!c0_shift_h || (y & c0_shift_h))
1299 c0_data += c0_linesize;
1300 if (!c1_shift_h || (y & c1_shift_h))
1301 c1_data += c1_linesize;
1302 dst_data += dst_linesize;
1303 }
1304 }
1305 } else {
1306 const uint16_t *c0_data = (uint16_t *)in->data[(plane + 1) % s->ncomp];
1307 const uint16_t *c1_data = (uint16_t *)in->data[(plane + 2) % s->ncomp];
1308 uint16_t *dst_data = (uint16_t *)out->data[plane] + offset_y * dst_linesize + offset_x;
1309
1310 if (mirror)
1311 dst_data += s->size - 1;
1312 for (y = 0; y < src_h; y++) {
1313 for (x = 0; x < src_w; x++) {
1314 const int sum = FFMIN(FFABS(c0_data[x >> c0_shift_w] - mid) + FFABS(c1_data[x >> c1_shift_w] - mid - 1), limit);
1315 uint16_t *target;
1316
1317 if (mirror) {
1318 target = dst_data - sum;
1319 update16(target, max, intensity, limit);
1320 } else {
1321 target = dst_data + sum;
1322 update16(target, max, intensity, limit);
1323 }
1324 }
1325
1326 if (!c0_shift_h || (y & c0_shift_h))
1327 c0_data += c0_linesize;
1328 if (!c1_shift_h || (y & c1_shift_h))
1329 c1_data += c1_linesize;
1330 dst_data += dst_linesize;
1331 }
1332 }
1333
1334 envelope16(s, out, plane, plane, column ? offset_x : offset_y);
1335}
1336
1337static av_always_inline void chroma(WaveformContext *s,
1338 AVFrame *in, AVFrame *out,
1339 int component, int intensity,
1340 int offset_y, int offset_x,
1341 int column, int mirror)
1342{
1343 const int plane = s->desc->comp[component].plane;
1344 const int c0_linesize = in->linesize[(plane + 1) % s->ncomp];
1345 const int c1_linesize = in->linesize[(plane + 2) % s->ncomp];
1346 const int dst_linesize = out->linesize[plane];
1347 const int max = 255 - intensity;
1348 const int c0_shift_w = s->shift_w[(component + 1) % s->ncomp];
1349 const int c1_shift_w = s->shift_w[(component + 2) % s->ncomp];
1350 const int c0_shift_h = s->shift_h[(component + 1) % s->ncomp];
1351 const int c1_shift_h = s->shift_h[(component + 2) % s->ncomp];
1352 const int src_h = in->height;
1353 const int src_w = in->width;
1354 int x, y;
1355
1356 if (column) {
1357 const int dst_signed_linesize = dst_linesize * (mirror == 1 ? -1 : 1);
1358
1359 for (x = 0; x < src_w; x++) {
1360 const uint8_t *c0_data = in->data[(plane + 1) % s->ncomp];
1361 const uint8_t *c1_data = in->data[(plane + 2) % s->ncomp];
1362 uint8_t *dst_data = out->data[plane] + offset_y * dst_linesize + offset_x;
1363 uint8_t * const dst_bottom_line = dst_data + dst_linesize * (s->size - 1);
1364 uint8_t * const dst_line = (mirror ? dst_bottom_line : dst_data);
1365 uint8_t *dst = dst_line;
1366
1367 for (y = 0; y < src_h; y++) {
1368 const int sum = FFABS(c0_data[x >> c0_shift_w] - 128) + FFABS(c1_data[x >> c1_shift_w] - 127);
1369 uint8_t *target;
1370
1371 target = dst + x + dst_signed_linesize * sum;
1372 update(target, max, intensity);
1373
1374 if (!c0_shift_h || (y & c0_shift_h))
1375 c0_data += c0_linesize;
1376 if (!c1_shift_h || (y & c1_shift_h))
1377 c1_data += c1_linesize;
1378 dst_data += dst_linesize;
1379 }
1380 }
1381 } else {
1382 const uint8_t *c0_data = in->data[(plane + 1) % s->ncomp];
1383 const uint8_t *c1_data = in->data[(plane + 2) % s->ncomp];
1384 uint8_t *dst_data = out->data[plane] + offset_y * dst_linesize + offset_x;
1385
1386 if (mirror)
1387 dst_data += s->size - 1;
1388 for (y = 0; y < src_h; y++) {
1389 for (x = 0; x < src_w; x++) {
1390 const int sum = FFABS(c0_data[x >> c0_shift_w] - 128) + FFABS(c1_data[x >> c1_shift_w] - 127);
1391 uint8_t *target;
1392
1393 if (mirror) {
1394 target = dst_data - sum;
1395 update(target, max, intensity);
1396 } else {
1397 target = dst_data + sum;
1398 update(target, max, intensity);
1399 }
1400 }
1401
1402 if (!c0_shift_h || (y & c0_shift_h))
1403 c0_data += c0_linesize;
1404 if (!c1_shift_h || (y & c1_shift_h))
1405 c1_data += c1_linesize;
1406 dst_data += dst_linesize;
1407 }
1408 }
1409
1410 envelope(s, out, plane, plane, column ? offset_x : offset_y);
1411}
1412
1413static av_always_inline void color16(WaveformContext *s,
1414 AVFrame *in, AVFrame *out,
1415 int component, int intensity,
1416 int offset_y, int offset_x,
1417 int column, int mirror)
1418{
1419 const int plane = s->desc->comp[component].plane;
1420 const int limit = s->max - 1;
1421 const uint16_t *c0_data = (const uint16_t *)in->data[plane + 0];
1422 const uint16_t *c1_data = (const uint16_t *)in->data[(plane + 1) % s->ncomp];
1423 const uint16_t *c2_data = (const uint16_t *)in->data[(plane + 2) % s->ncomp];
1424 const int c0_linesize = in->linesize[ plane + 0 ] / 2;
1425 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp] / 2;
1426 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp] / 2;
1427 const int d0_linesize = out->linesize[ plane + 0 ] / 2;
1428 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp] / 2;
1429 const int d2_linesize = out->linesize[(plane + 2) % s->ncomp] / 2;
1430 const int c0_shift_w = s->shift_w[ component + 0 ];
1431 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
1432 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
1433 const int c0_shift_h = s->shift_h[ component + 0 ];
1434 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
1435 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
1436 const int src_h = in->height;
1437 const int src_w = in->width;
1438 int x, y;
1439
1440 if (column) {
1441 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
1442 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
1443 const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1);
1444 uint16_t *d0_data = (uint16_t *)out->data[plane] + offset_y * d0_linesize + offset_x;
1445 uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1446 uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1447 uint16_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
1448 uint16_t * const d0 = (mirror ? d0_bottom_line : d0_data);
1449 uint16_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
1450 uint16_t * const d1 = (mirror ? d1_bottom_line : d1_data);
1451 uint16_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1);
1452 uint16_t * const d2 = (mirror ? d2_bottom_line : d2_data);
1453
1454 for (y = 0; y < src_h; y++) {
1455 for (x = 0; x < src_w; x++) {
1456 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit);
1457 const int c1 = c1_data[x >> c1_shift_w];
1458 const int c2 = c2_data[x >> c2_shift_w];
1459
1460 *(d0 + d0_signed_linesize * c0 + x) = c0;
1461 *(d1 + d1_signed_linesize * c0 + x) = c1;
1462 *(d2 + d2_signed_linesize * c0 + x) = c2;
1463 }
1464
1465 if (!c0_shift_h || (y & c0_shift_h))
1466 c0_data += c0_linesize;
1467 if (!c1_shift_h || (y & c1_shift_h))
1468 c1_data += c1_linesize;
1469 if (!c2_shift_h || (y & c2_shift_h))
1470 c2_data += c2_linesize;
1471 d0_data += d0_linesize;
1472 d1_data += d1_linesize;
1473 d2_data += d2_linesize;
1474 }
1475 } else {
1476 uint16_t *d0_data = (uint16_t *)out->data[plane] + offset_y * d0_linesize + offset_x;
1477 uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1478 uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1479
1480 if (mirror) {
1481 d0_data += s->size - 1;
1482 d1_data += s->size - 1;
1483 d2_data += s->size - 1;
1484 }
1485
1486 for (y = 0; y < src_h; y++) {
1487 for (x = 0; x < src_w; x++) {
1488 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit);
1489 const int c1 = c1_data[x >> c1_shift_w];
1490 const int c2 = c2_data[x >> c2_shift_w];
1491
1492 if (mirror) {
1493 *(d0_data - c0) = c0;
1494 *(d1_data - c0) = c1;
1495 *(d2_data - c0) = c2;
1496 } else {
1497 *(d0_data + c0) = c0;
1498 *(d1_data + c0) = c1;
1499 *(d2_data + c0) = c2;
1500 }
1501 }
1502
1503 if (!c0_shift_h || (y & c0_shift_h))
1504 c0_data += c0_linesize;
1505 if (!c1_shift_h || (y & c1_shift_h))
1506 c1_data += c1_linesize;
1507 if (!c2_shift_h || (y & c2_shift_h))
1508 c2_data += c2_linesize;
1509 d0_data += d0_linesize;
1510 d1_data += d1_linesize;
1511 d2_data += d2_linesize;
1512 }
1513 }
1514
1515 envelope16(s, out, plane, plane, column ? offset_x : offset_y);
1516}
1517
1518static av_always_inline void color(WaveformContext *s,
1519 AVFrame *in, AVFrame *out,
1520 int component, int intensity,
1521 int offset_y, int offset_x,
1522 int column, int mirror)
1523{
1524 const int plane = s->desc->comp[component].plane;
1525 const uint8_t *c0_data = in->data[plane + 0];
1526 const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
1527 const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
1528 const int c0_linesize = in->linesize[ plane + 0 ];
1529 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp];
1530 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp];
1531 const int d0_linesize = out->linesize[ plane + 0 ];
1532 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp];
1533 const int d2_linesize = out->linesize[(plane + 2) % s->ncomp];
1534 const int c0_shift_w = s->shift_w[ component + 0 ];
1535 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
1536 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
1537 const int c0_shift_h = s->shift_h[ component + 0 ];
1538 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
1539 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
1540 const int src_h = in->height;
1541 const int src_w = in->width;
1542 int x, y;
1543
1544 if (s->mode) {
1545 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
1546 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
1547 const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1);
1548 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
1549 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1550 uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1551 uint8_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
1552 uint8_t * const d0 = (mirror ? d0_bottom_line : d0_data);
1553 uint8_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
1554 uint8_t * const d1 = (mirror ? d1_bottom_line : d1_data);
1555 uint8_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1);
1556 uint8_t * const d2 = (mirror ? d2_bottom_line : d2_data);
1557
1558 for (y = 0; y < src_h; y++) {
1559 for (x = 0; x < src_w; x++) {
1560 const int c0 = c0_data[x >> c0_shift_w];
1561 const int c1 = c1_data[x >> c1_shift_w];
1562 const int c2 = c2_data[x >> c2_shift_w];
1563
1564 *(d0 + d0_signed_linesize * c0 + x) = c0;
1565 *(d1 + d1_signed_linesize * c0 + x) = c1;
1566 *(d2 + d2_signed_linesize * c0 + x) = c2;
1567 }
1568
1569 if (!c0_shift_h || (y & c0_shift_h))
1570 c0_data += c0_linesize;
1571 if (!c1_shift_h || (y & c1_shift_h))
1572 c1_data += c1_linesize;
1573 if (!c2_shift_h || (y & c2_shift_h))
1574 c2_data += c2_linesize;
1575 d0_data += d0_linesize;
1576 d1_data += d1_linesize;
1577 d2_data += d2_linesize;
1578 }
1579 } else {
1580 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
1581 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1582 uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1583
1584 if (mirror) {
1585 d0_data += s->size - 1;
1586 d1_data += s->size - 1;
1587 d2_data += s->size - 1;
1588 }
1589
1590 for (y = 0; y < src_h; y++) {
1591 for (x = 0; x < src_w; x++) {
1592 const int c0 = c0_data[x >> c0_shift_w];
1593 const int c1 = c1_data[x >> c1_shift_w];
1594 const int c2 = c2_data[x >> c2_shift_w];
1595
1596 if (mirror) {
1597 *(d0_data - c0) = c0;
1598 *(d1_data - c0) = c1;
1599 *(d2_data - c0) = c2;
1600 } else {
1601 *(d0_data + c0) = c0;
1602 *(d1_data + c0) = c1;
1603 *(d2_data + c0) = c2;
1604 }
1605 }
1606
1607 if (!c0_shift_h || (y & c0_shift_h))
1608 c0_data += c0_linesize;
1609 if (!c1_shift_h || (y & c1_shift_h))
1610 c1_data += c1_linesize;
1611 if (!c2_shift_h || (y & c2_shift_h))
1612 c2_data += c2_linesize;
1613 d0_data += d0_linesize;
1614 d1_data += d1_linesize;
1615 d2_data += d2_linesize;
1616 }
1617 }
1618
1619 envelope(s, out, plane, plane, column ? offset_x : offset_y);
1620}
1621
1622static av_always_inline void acolor16(WaveformContext *s,
1623 AVFrame *in, AVFrame *out,
1624 int component, int intensity,
1625 int offset_y, int offset_x,
1626 int column, int mirror)
1627{
1628 const int plane = s->desc->comp[component].plane;
1629 const int limit = s->max - 1;
1630 const int max = limit - intensity;
1631 const uint16_t *c0_data = (const uint16_t *)in->data[plane + 0];
1632 const uint16_t *c1_data = (const uint16_t *)in->data[(plane + 1) % s->ncomp];
1633 const uint16_t *c2_data = (const uint16_t *)in->data[(plane + 2) % s->ncomp];
1634 const int c0_linesize = in->linesize[ plane + 0 ] / 2;
1635 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp] / 2;
1636 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp] / 2;
1637 const int d0_linesize = out->linesize[ plane + 0 ] / 2;
1638 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp] / 2;
1639 const int d2_linesize = out->linesize[(plane + 2) % s->ncomp] / 2;
1640 const int c0_shift_w = s->shift_w[ component + 0 ];
1641 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
1642 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
1643 const int c0_shift_h = s->shift_h[ component + 0 ];
1644 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
1645 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
1646 const int src_h = in->height;
1647 const int src_w = in->width;
1648 int x, y;
1649
1650 if (s->mode) {
1651 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
1652 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
1653 const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1);
1654 uint16_t *d0_data = (uint16_t *)out->data[plane] + offset_y * d0_linesize + offset_x;
1655 uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1656 uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1657 uint16_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
1658 uint16_t * const d0 = (mirror ? d0_bottom_line : d0_data);
1659 uint16_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
1660 uint16_t * const d1 = (mirror ? d1_bottom_line : d1_data);
1661 uint16_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1);
1662 uint16_t * const d2 = (mirror ? d2_bottom_line : d2_data);
1663
1664 for (y = 0; y < src_h; y++) {
1665 for (x = 0; x < src_w; x++) {
1666 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit);
1667 const int c1 = c1_data[x >> c1_shift_w];
1668 const int c2 = c2_data[x >> c2_shift_w];
1669
1670 update16(d0 + d0_signed_linesize * c0 + x, max, intensity, limit);
1671 *(d1 + d1_signed_linesize * c0 + x) = c1;
1672 *(d2 + d2_signed_linesize * c0 + x) = c2;
1673 }
1674
1675 if (!c0_shift_h || (y & c0_shift_h))
1676 c0_data += c0_linesize;
1677 if (!c1_shift_h || (y & c1_shift_h))
1678 c1_data += c1_linesize;
1679 if (!c2_shift_h || (y & c2_shift_h))
1680 c2_data += c2_linesize;
1681 d0_data += d0_linesize;
1682 d1_data += d1_linesize;
1683 d2_data += d2_linesize;
1684 }
1685 } else {
1686 uint16_t *d0_data = (uint16_t *)out->data[plane] + offset_y * d0_linesize + offset_x;
1687 uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1688 uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1689
1690 if (mirror) {
1691 d0_data += s->size - 1;
1692 d1_data += s->size - 1;
1693 d2_data += s->size - 1;
1694 }
1695
1696 for (y = 0; y < src_h; y++) {
1697 for (x = 0; x < src_w; x++) {
1698 const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit);
1699 const int c1 = c1_data[x >> c1_shift_w];
1700 const int c2 = c2_data[x >> c2_shift_w];
1701
1702 if (mirror) {
1703 update16(d0_data - c0, max, intensity, limit);
1704 *(d1_data - c0) = c1;
1705 *(d2_data - c0) = c2;
1706 } else {
1707 update16(d0_data + c0, max, intensity, limit);
1708 *(d1_data + c0) = c1;
1709 *(d2_data + c0) = c2;
1710 }
1711 }
1712
1713 if (!c0_shift_h || (y & c0_shift_h))
1714 c0_data += c0_linesize;
1715 if (!c1_shift_h || (y & c1_shift_h))
1716 c1_data += c1_linesize;
1717 if (!c2_shift_h || (y & c2_shift_h))
1718 c2_data += c2_linesize;
1719 d0_data += d0_linesize;
1720 d1_data += d1_linesize;
1721 d2_data += d2_linesize;
1722 }
1723 }
1724
1725 envelope16(s, out, plane, plane, column ? offset_x : offset_y);
1726}
1727
1728static av_always_inline void acolor(WaveformContext *s,
1729 AVFrame *in, AVFrame *out,
1730 int component, int intensity,
1731 int offset_y, int offset_x,
1732 int column, int mirror)
1733{
1734 const int plane = s->desc->comp[component].plane;
1735 const uint8_t *c0_data = in->data[plane + 0];
1736 const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
1737 const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
1738 const int c0_linesize = in->linesize[ plane + 0 ];
1739 const int c1_linesize = in->linesize[(plane + 1) % s->ncomp];
1740 const int c2_linesize = in->linesize[(plane + 2) % s->ncomp];
1741 const int d0_linesize = out->linesize[ plane + 0 ];
1742 const int d1_linesize = out->linesize[(plane + 1) % s->ncomp];
1743 const int d2_linesize = out->linesize[(plane + 2) % s->ncomp];
1744 const int c0_shift_w = s->shift_w[ component + 0 ];
1745 const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
1746 const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
1747 const int c0_shift_h = s->shift_h[ component + 0 ];
1748 const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
1749 const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
1750 const int max = 255 - intensity;
1751 const int src_h = in->height;
1752 const int src_w = in->width;
1753 int x, y;
1754
1755 if (s->mode) {
1756 const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
1757 const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
1758 const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1);
1759 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
1760 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1761 uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1762 uint8_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
1763 uint8_t * const d0 = (mirror ? d0_bottom_line : d0_data);
1764 uint8_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
1765 uint8_t * const d1 = (mirror ? d1_bottom_line : d1_data);
1766 uint8_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1);
1767 uint8_t * const d2 = (mirror ? d2_bottom_line : d2_data);
1768
1769 for (y = 0; y < src_h; y++) {
1770 for (x = 0; x < src_w; x++) {
1771 const int c0 = c0_data[x >> c0_shift_w];
1772 const int c1 = c1_data[x >> c1_shift_w];
1773 const int c2 = c2_data[x >> c2_shift_w];
1774
1775 update(d0 + d0_signed_linesize * c0 + x, max, intensity);
1776 *(d1 + d1_signed_linesize * c0 + x) = c1;
1777 *(d2 + d2_signed_linesize * c0 + x) = c2;
1778 }
1779
1780 if (!c0_shift_h || (y & c0_shift_h))
1781 c0_data += c0_linesize;
1782 if (!c1_shift_h || (y & c1_shift_h))
1783 c1_data += c1_linesize;
1784 if (!c2_shift_h || (y & c2_shift_h))
1785 c2_data += c2_linesize;
1786 d0_data += d0_linesize;
1787 d1_data += d1_linesize;
1788 d2_data += d2_linesize;
1789 }
1790 } else {
1791 uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
1792 uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
1793 uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
1794
1795 if (mirror) {
1796 d0_data += s->size - 1;
1797 d1_data += s->size - 1;
1798 d2_data += s->size - 1;
1799 }
1800
1801 for (y = 0; y < src_h; y++) {
1802 for (x = 0; x < src_w; x++) {
1803 const int c0 = c0_data[x >> c0_shift_w];
1804 const int c1 = c1_data[x >> c1_shift_w];
1805 const int c2 = c2_data[x >> c2_shift_w];
1806
1807 if (mirror) {
1808 update(d0_data - c0, max, intensity);
1809 *(d1_data - c0) = c1;
1810 *(d2_data - c0) = c2;
1811 } else {
1812 update(d0_data + c0, max, intensity);
1813 *(d1_data + c0) = c1;
1814 *(d2_data + c0) = c2;
1815 }
1816 }
1817
1818 if (!c0_shift_h || (y & c0_shift_h))
1819 c0_data += c0_linesize;
1820 if (!c1_shift_h || (y & c1_shift_h))
1821 c1_data += c1_linesize;
1822 if (!c2_shift_h || (y & c2_shift_h))
1823 c2_data += c2_linesize;
1824 d0_data += d0_linesize;
1825 d1_data += d1_linesize;
1826 d2_data += d2_linesize;
1827 }
1828 }
1829
1830 envelope(s, out, plane, plane, column ? offset_x : offset_y);
1831}
1832
1833static const uint8_t black_yuva_color[4] = { 0, 127, 127, 255 };
1834static const uint8_t green_yuva_color[4] = { 255, 0, 0, 255 };
1835static const uint8_t black_gbrp_color[4] = { 0, 0, 0, 255 };
1836
1837static const GraticuleLines aflat_digital8[] = {
1838 { { { "16", 16+128 }, { "16", 16+128 }, { "16", 16+128 }, { "0", 0+128 } } },
1839 { { { "128", 128+128 }, { "128", 128+128 }, { "128", 128+128 }, { "128", 128+128 } } },
1840 { { { "235", 235+128 }, { "240", 240+128 }, { "240", 240+128 }, { "255", 255+128 } } },
1841};
1842
1843static const GraticuleLines aflat_digital9[] = {
1844 { { { "32", 32+256 }, { "32", 32+256 }, { "32", 32+256 }, { "0", 0+256 } } },
1845 { { { "256", 256+256 }, { "256", 256+256 }, { "256", 256+256 }, { "256", 256+256 } } },
1846 { { { "470", 470+256 }, { "480", 480+256 }, { "480", 480+256 }, { "511", 511+256 } } },
1847};
1848
1849static const GraticuleLines aflat_digital10[] = {
1850 { { { "64", 64+512 }, { "64", 64+512 }, { "64", 64+512 }, { "0", 0+512 } } },
1851 { { { "512", 512+512 }, { "512", 512+512 }, { "512", 512+512 }, { "512", 512+512 } } },
1852 { { { "940", 940+512 }, { "960", 960+512 }, { "960", 960+512 }, { "1023", 1023+512 } } },
1853};
1854
1855static const GraticuleLines aflat_digital12[] = {
1856 { { { "256", 256+2048 }, { "256", 256+2048 }, { "256", 256+2048 }, { "0", 0+2048 } } },
1857 { { { "2048", 2048+2048 }, { "2048", 2048+2048 }, { "2048", 2048+2048 }, { "2048", 2048+2048 } } },
1858 { { { "3760", 3760+2048 }, { "3840", 3840+2048 }, { "3840", 3840+2048 }, { "4095", 4095+2048 } } },
1859};
1860
1861static const GraticuleLines aflat_millivolts8[] = {
1862 { { { "0", 16+128 }, { "0", 16+128 }, { "0", 16+128 }, { "0", 0+128 } } },
1863 { { { "175", 71+128 }, { "175", 72+128 }, { "175", 72+128 }, { "175", 64+128 } } },
1864 { { { "350", 126+128 }, { "350", 128+128 }, { "350", 128+128 }, { "350", 128+128 } } },
1865 { { { "525", 180+128 }, { "525", 184+128 }, { "525", 184+128 }, { "525", 192+128 } } },
1866 { { { "700", 235+128 }, { "700", 240+128 }, { "700", 240+128 }, { "700", 255+128 } } },
1867};
1868
1869static const GraticuleLines aflat_millivolts9[] = {
1870 { { { "0", 32+256 }, { "0", 32+256 }, { "0", 32+256 }, { "0", 0+256 } } },
1871 { { { "175", 142+256 }, { "175", 144+256 }, { "175", 144+256 }, { "175", 128+256 } } },
1872 { { { "350", 251+256 }, { "350", 256+256 }, { "350", 256+256 }, { "350", 256+256 } } },
1873 { { { "525", 361+256 }, { "525", 368+256 }, { "525", 368+256 }, { "525", 384+256 } } },
1874 { { { "700", 470+256 }, { "700", 480+256 }, { "700", 480+256 }, { "700", 511+256 } } },
1875};
1876
1877static const GraticuleLines aflat_millivolts10[] = {
1878 { { { "0", 64+512 }, { "0", 64+512 }, { "0", 64+512 }, { "0", 0+512 } } },
1879 { { { "175", 283+512 }, { "175", 288+512 }, { "175", 288+512 }, { "175", 256+512 } } },
1880 { { { "350", 502+512 }, { "350", 512+512 }, { "350", 512+512 }, { "350", 512+512 } } },
1881 { { { "525", 721+512 }, { "525", 736+512 }, { "525", 736+512 }, { "525", 768+512 } } },
1882 { { { "700", 940+512 }, { "700", 960+512 }, { "700", 960+512 }, { "700", 1023+512 } } },
1883};
1884
1885static const GraticuleLines aflat_millivolts12[] = {
1886 { { { "0", 256+2048 }, { "0", 256+2048 }, { "0", 256+2048 }, { "0", 0+2048 } } },
1887 { { { "175", 1132+2048 }, { "175", 1152+2048 }, { "175", 1152+2048 }, { "175", 1024+2048 } } },
1888 { { { "350", 2008+2048 }, { "350", 2048+2048 }, { "350", 2048+2048 }, { "350", 2048+2048 } } },
1889 { { { "525", 2884+2048 }, { "525", 2944+2048 }, { "525", 2944+2048 }, { "525", 3072+2048 } } },
1890 { { { "700", 3760+2048 }, { "700", 3840+2048 }, { "700", 3840+2048 }, { "700", 4095+2048 } } },
1891};
1892
1893static const GraticuleLines aflat_ire8[] = {
1894 { { { "-25", -39+128 }, { "-25", -40+128 }, { "-25", -40+128 }, { "-25", -64+128 } } },
1895 { { { "0", 16+128 }, { "0", 16+128 }, { "0", 16+128 }, { "0", 0+128 } } },
1896 { { { "25", 71+128 }, { "25", 72+128 }, { "25", 72+128 }, { "25", 64+128 } } },
1897 { { { "50", 126+128 }, { "50", 128+128 }, { "50", 128+128 }, { "50", 128+128 } } },
1898 { { { "75", 180+128 }, { "75", 184+128 }, { "75", 184+128 }, { "75", 192+128 } } },
1899 { { { "100", 235+128 }, { "100", 240+128 }, { "100", 240+128 }, { "100", 256+128 } } },
1900 { { { "125", 290+128 }, { "125", 296+128 }, { "125", 296+128 }, { "125", 320+128 } } },
1901};
1902
1903static const GraticuleLines aflat_ire9[] = {
1904 { { { "-25", -78+256 }, { "-25", -80+256 }, { "-25", -80+256 }, { "-25",-128+256 } } },
1905 { { { "0", 32+256 }, { "0", 32+256 }, { "0", 32+256 }, { "0", 0+256 } } },
1906 { { { "25", 142+256 }, { "25", 144+256 }, { "25", 144+256 }, { "25", 128+256 } } },
1907 { { { "50", 251+256 }, { "50", 256+256 }, { "50", 256+256 }, { "50", 256+256 } } },
1908 { { { "75", 361+256 }, { "75", 368+256 }, { "75", 368+256 }, { "75", 384+256 } } },
1909 { { { "100", 470+256 }, { "100", 480+256 }, { "100", 480+256 }, { "100", 512+256 } } },
1910 { { { "125", 580+256 }, { "125", 592+256 }, { "125", 592+256 }, { "125", 640+256 } } },
1911};
1912
1913static const GraticuleLines aflat_ire10[] = {
1914 { { { "-25",-156+512 }, { "-25",-160+512 }, { "-25",-160+512 }, { "-25", -256+512 } } },
1915 { { { "0", 64+512 }, { "0", 64+512 }, { "0", 64+512 }, { "0", 0+512 } } },
1916 { { { "25", 283+512 }, { "25", 288+512 }, { "25", 288+512 }, { "25", 256+512 } } },
1917 { { { "50", 502+512 }, { "50", 512+512 }, { "50", 512+512 }, { "50", 512+512 } } },
1918 { { { "75", 721+512 }, { "75", 736+512 }, { "75", 736+512 }, { "75", 768+512 } } },
1919 { { { "100", 940+512 }, { "100", 960+512 }, { "100", 960+512 }, { "100", 1024+512 } } },
1920 { { { "125",1160+512 }, { "125",1184+512 }, { "125",1184+512 }, { "125", 1280+512 } } },
1921};
1922
1923static const GraticuleLines aflat_ire12[] = {
1924 { { { "-25", -624+2048 }, { "-25", -640+2048 }, { "-25", -640+2048 }, { "-25",-1024+2048 } } },
1925 { { { "0", 256+2048 }, { "0", 256+2048 }, { "0", 256+2048 }, { "0", 0+2048 } } },
1926 { { { "25", 1132+2048 }, { "25", 1152+2048 }, { "25", 1152+2048 }, { "25", 1024+2048 } } },
1927 { { { "50", 2008+2048 }, { "50", 2048+2048 }, { "50", 2048+2048 }, { "50", 2048+2048 } } },
1928 { { { "75", 2884+2048 }, { "75", 2944+2048 }, { "75", 2944+2048 }, { "75", 3072+2048 } } },
1929 { { { "100", 3760+2048 }, { "100", 3840+2048 }, { "100", 3840+2048 }, { "100", 4096+2048 } } },
1930 { { { "125", 4640+2048 }, { "125", 4736+2048 }, { "125", 4736+2048 }, { "125", 5120+2048 } } },
1931};
1932
1933static const GraticuleLines flat_digital8[] = {
1934 { { { "16", 16+256 }, { "16", 16+256 }, { "16", 16+256 }, { "0", 0+256 } } },
1935 { { { "128", 128+256 }, { "128", 128+256 }, { "128", 128+256 }, { "128", 128+256 } } },
1936 { { { "235", 235+256 }, { "240", 240+256 }, { "240", 240+256 }, { "255", 255+256 } } },
1937};
1938
1939static const GraticuleLines flat_digital9[] = {
1940 { { { "32", 32+512 }, { "32", 32+512 }, { "32", 32+512 }, { "0", 0+512 } } },
1941 { { { "256", 256+512 }, { "256", 256+512 }, { "256", 256+512 }, { "256", 256+512 } } },
1942 { { { "470", 470+512 }, { "480", 480+512 }, { "480", 480+512 }, { "511", 511+512 } } },
1943};
1944
1945static const GraticuleLines flat_digital10[] = {
1946 { { { "64", 64+1024 }, { "64", 64+1024 }, { "64", 64+1024 }, { "0", 0+1024 } } },
1947 { { { "512", 512+1024 }, { "512", 512+1024 }, { "512", 512+1024 }, { "512", 512+1024 } } },
1948 { { { "940", 940+1024 }, { "960", 960+1024 }, { "960", 960+1024 }, { "1023", 1023+1024 } } },
1949};
1950
1951static const GraticuleLines flat_digital12[] = {
1952 { { { "256", 256+4096 }, { "256", 256+4096 }, { "256", 256+4096 }, { "0", 0+4096 } } },
1953 { { { "2048", 2048+4096 }, { "2048", 2048+4096 }, { "2048", 2048+4096 }, { "2048", 2048+4096 } } },
1954 { { { "3760", 3760+4096 }, { "3840", 3840+4096 }, { "3840", 3840+4096 }, { "4095", 4095+4096 } } },
1955};
1956
1957static const GraticuleLines flat_millivolts8[] = {
1958 { { { "0", 16+256 }, { "0", 16+256 }, { "0", 16+256 }, { "0", 0+256 } } },
1959 { { { "175", 71+256 }, { "175", 72+256 }, { "175", 72+256 }, { "175", 64+256 } } },
1960 { { { "350", 126+256 }, { "350", 128+256 }, { "350", 128+256 }, { "350", 128+256 } } },
1961 { { { "525", 180+256 }, { "525", 184+256 }, { "525", 184+256 }, { "525", 192+256 } } },
1962 { { { "700", 235+256 }, { "700", 240+256 }, { "700", 240+256 }, { "700", 255+256 } } },
1963};
1964
1965static const GraticuleLines flat_millivolts9[] = {
1966 { { { "0", 32+512 }, { "0", 32+512 }, { "0", 32+512 }, { "0", 0+512 } } },
1967 { { { "175", 142+512 }, { "175", 144+512 }, { "175", 144+512 }, { "175", 128+512 } } },
1968 { { { "350", 251+512 }, { "350", 256+512 }, { "350", 256+512 }, { "350", 256+512 } } },
1969 { { { "525", 361+512 }, { "525", 368+512 }, { "525", 368+512 }, { "525", 384+512 } } },
1970 { { { "700", 470+512 }, { "700", 480+512 }, { "700", 480+512 }, { "700", 511+512 } } },
1971};
1972
1973static const GraticuleLines flat_millivolts10[] = {
1974 { { { "0", 64+1024 }, { "0", 64+1024 }, { "0", 64+1024 }, { "0", 0+1024 } } },
1975 { { { "175", 283+1024 }, { "175", 288+1024 }, { "175", 288+1024 }, { "175", 256+1024 } } },
1976 { { { "350", 502+1024 }, { "350", 512+1024 }, { "350", 512+1024 }, { "350", 512+1024 } } },
1977 { { { "525", 721+1024 }, { "525", 736+1024 }, { "525", 736+1024 }, { "525", 768+1024 } } },
1978 { { { "700", 940+1024 }, { "700", 960+1024 }, { "700", 960+1024 }, { "700", 1023+1024 } } },
1979};
1980
1981static const GraticuleLines flat_millivolts12[] = {
1982 { { { "0", 256+4096 }, { "0", 256+4096 }, { "0", 256+4096 }, { "0", 0+4096 } } },
1983 { { { "175", 1132+4096 }, { "175", 1152+4096 }, { "175", 1152+4096 }, { "175", 1024+4096 } } },
1984 { { { "350", 2008+4096 }, { "350", 2048+4096 }, { "350", 2048+4096 }, { "350", 2048+4096 } } },
1985 { { { "525", 2884+4096 }, { "525", 2944+4096 }, { "525", 2944+4096 }, { "525", 3072+4096 } } },
1986 { { { "700", 3760+4096 }, { "700", 3840+4096 }, { "700", 3840+4096 }, { "700", 4095+4096 } } },
1987};
1988
1989static const GraticuleLines flat_ire8[] = {
1990 { { { "-25", -39+256 }, { "-25", -40+256 }, { "-25", -40+256 }, { "-25", -64+256 } } },
1991 { { { "0", 16+256 }, { "0", 16+256 }, { "0", 16+256 }, { "0", 0+256 } } },
1992 { { { "25", 71+256 }, { "25", 72+256 }, { "25", 72+256 }, { "25", 64+256 } } },
1993 { { { "50", 126+256 }, { "50", 128+256 }, { "50", 128+256 }, { "50", 128+256 } } },
1994 { { { "75", 180+256 }, { "75", 184+256 }, { "75", 184+256 }, { "75", 192+256 } } },
1995 { { { "100", 235+256 }, { "100", 240+256 }, { "100", 240+256 }, { "100", 256+256 } } },
1996 { { { "125", 290+256 }, { "125", 296+256 }, { "125", 296+256 }, { "125", 320+256 } } },
1997};
1998
1999static const GraticuleLines flat_ire9[] = {
2000 { { { "-25", -78+512 }, { "-25", -80+512 }, { "-25", -80+512 }, { "-25",-128+512 } } },
2001 { { { "0", 32+512 }, { "0", 32+512 }, { "0", 32+512 }, { "0", 0+512 } } },
2002 { { { "25", 142+512 }, { "25", 144+512 }, { "25", 144+512 }, { "25", 128+512 } } },
2003 { { { "50", 251+512 }, { "50", 256+512 }, { "50", 256+512 }, { "50", 256+512 } } },
2004 { { { "75", 361+512 }, { "75", 368+512 }, { "75", 368+512 }, { "75", 384+512 } } },
2005 { { { "100", 470+512 }, { "100", 480+512 }, { "100", 480+512 }, { "100", 512+512 } } },
2006 { { { "125", 580+512 }, { "125", 592+512 }, { "125", 592+512 }, { "125", 640+512 } } },
2007};
2008
2009static const GraticuleLines flat_ire10[] = {
2010 { { { "-25",-156+1024 }, { "-25",-160+1024 }, { "-25",-160+1024 }, { "-25", -256+1024 } } },
2011 { { { "0", 64+1024 }, { "0", 64+1024 }, { "0", 64+1024 }, { "0", 0+1024 } } },
2012 { { { "25", 283+1024 }, { "25", 288+1024 }, { "25", 288+1024 }, { "25", 256+1024 } } },
2013 { { { "50", 502+1024 }, { "50", 512+1024 }, { "50", 512+1024 }, { "50", 512+1024 } } },
2014 { { { "75", 721+1024 }, { "75", 736+1024 }, { "75", 736+1024 }, { "75", 768+1024 } } },
2015 { { { "100", 940+1024 }, { "100", 960+1024 }, { "100", 960+1024 }, { "100", 1024+1024 } } },
2016 { { { "125",1160+1024 }, { "125",1184+1024 }, { "125",1184+1024 }, { "125", 1280+1024 } } },
2017};
2018
2019static const GraticuleLines flat_ire12[] = {
2020 { { { "-25", -624+4096 }, { "-25", -640+4096 }, { "-25", -640+4096 }, { "-25",-1024+4096 } } },
2021 { { { "0", 256+4096 }, { "0", 256+4096 }, { "0", 256+4096 }, { "0", 0+4096 } } },
2022 { { { "25", 1132+4096 }, { "25", 1152+4096 }, { "25", 1152+4096 }, { "25", 1024+4096 } } },
2023 { { { "50", 2008+4096 }, { "50", 2048+4096 }, { "50", 2048+4096 }, { "50", 2048+4096 } } },
2024 { { { "75", 2884+4096 }, { "75", 2944+4096 }, { "75", 2944+4096 }, { "75", 3072+4096 } } },
2025 { { { "100", 3760+4096 }, { "100", 3840+4096 }, { "100", 3840+4096 }, { "100", 4096+4096 } } },
2026 { { { "125", 4640+4096 }, { "125", 4736+4096 }, { "125", 4736+4096 }, { "125", 5120+4096 } } },
2027};
2028
2029static const GraticuleLines digital8[] = {
2030 { { { "16", 16 }, { "16", 16 }, { "16", 16 }, { "0", 0 } } },
2031 { { { "128", 128 }, { "128", 128 }, { "128", 128 }, { "128", 128 } } },
2032 { { { "235", 235 }, { "240", 240 }, { "240", 240 }, { "255", 255 } } },
2033};
2034
2035static const GraticuleLines digital9[] = {
2036 { { { "32", 32 }, { "32", 32 }, { "32", 32 }, { "0", 0 } } },
2037 { { { "256", 256 }, { "256", 256 }, { "256", 256 }, { "256", 256 } } },
2038 { { { "470", 470 }, { "480", 480 }, { "480", 480 }, { "511", 511 } } },
2039};
2040
2041static const GraticuleLines digital10[] = {
2042 { { { "64", 64 }, { "64", 64 }, { "64", 64 }, { "0", 0 } } },
2043 { { { "512", 512 }, { "512", 512 }, { "512", 512 }, { "512", 512 } } },
2044 { { { "940", 940 }, { "960", 960 }, { "960", 960 }, { "1023", 1023 } } },
2045};
2046
2047static const GraticuleLines digital12[] = {
2048 { { { "256", 256 }, { "256", 256 }, { "256", 256 }, { "0", 0 } } },
2049 { { { "2048", 2048 }, { "2048", 2048 }, { "2048", 2048 }, { "2048", 2048 } } },
2050 { { { "3760", 3760 }, { "3840", 3840 }, { "3840", 3840 }, { "4095", 4095 } } },
2051};
2052
2053static const GraticuleLines millivolts8[] = {
2054 { { { "0", 16 }, { "0", 16 }, { "0", 16 }, { "0", 0 } } },
2055 { { { "175", 71 }, { "175", 72 }, { "175", 72 }, { "175", 64 } } },
2056 { { { "350", 126 }, { "350", 128 }, { "350", 128 }, { "350", 128 } } },
2057 { { { "525", 180 }, { "525", 184 }, { "525", 184 }, { "525", 192 } } },
2058 { { { "700", 235 }, { "700", 240 }, { "700", 240 }, { "700", 255 } } },
2059};
2060
2061static const GraticuleLines millivolts9[] = {
2062 { { { "0", 32 }, { "0", 32 }, { "0", 32 }, { "0", 0 } } },
2063 { { { "175", 142 }, { "175", 144 }, { "175", 144 }, { "175", 128 } } },
2064 { { { "350", 251 }, { "350", 256 }, { "350", 256 }, { "350", 256 } } },
2065 { { { "525", 361 }, { "525", 368 }, { "525", 368 }, { "525", 384 } } },
2066 { { { "700", 470 }, { "700", 480 }, { "700", 480 }, { "700", 511 } } },
2067};
2068
2069static const GraticuleLines millivolts10[] = {
2070 { { { "0", 64 }, { "0", 64 }, { "0", 64 }, { "0", 0 } } },
2071 { { { "175", 283 }, { "175", 288 }, { "175", 288 }, { "175", 256 } } },
2072 { { { "350", 502 }, { "350", 512 }, { "350", 512 }, { "350", 512 } } },
2073 { { { "525", 721 }, { "525", 736 }, { "525", 736 }, { "525", 768 } } },
2074 { { { "700", 940 }, { "700", 960 }, { "700", 960 }, { "700", 1023 } } },
2075};
2076
2077static const GraticuleLines millivolts12[] = {
2078 { { { "0", 256 }, { "0", 256 }, { "0", 256 }, { "0", 0 } } },
2079 { { { "175", 1132 }, { "175", 1152 }, { "175", 1152 }, { "175", 1024 } } },
2080 { { { "350", 2008 }, { "350", 2048 }, { "350", 2048 }, { "350", 2048 } } },
2081 { { { "525", 2884 }, { "525", 2944 }, { "525", 2944 }, { "525", 3072 } } },
2082 { { { "700", 3760 }, { "700", 3840 }, { "700", 3840 }, { "700", 4095 } } },
2083};
2084
2085static const GraticuleLines ire8[] = {
2086 { { { "0", 16 }, { "0", 16 }, { "0", 16 }, { "0", 0 } } },
2087 { { { "25", 71 }, { "25", 72 }, { "25", 72 }, { "25", 64 } } },
2088 { { { "50", 126 }, { "50", 128 }, { "50", 128 }, { "50", 128 } } },
2089 { { { "75", 180 }, { "75", 184 }, { "75", 184 }, { "75", 192 } } },
2090 { { { "100", 235 }, { "100", 240 }, { "100", 240 }, { "100", 255 } } },
2091};
2092
2093static const GraticuleLines ire9[] = {
2094 { { { "0", 32 }, { "0", 32 }, { "0", 32 }, { "0", 0 } } },
2095 { { { "25", 142 }, { "25", 144 }, { "25", 144 }, { "25", 128 } } },
2096 { { { "50", 251 }, { "50", 256 }, { "50", 256 }, { "50", 256 } } },
2097 { { { "75", 361 }, { "75", 368 }, { "75", 368 }, { "75", 384 } } },
2098 { { { "100", 470 }, { "100", 480 }, { "100", 480 }, { "100", 511 } } },
2099};
2100
2101static const GraticuleLines ire10[] = {
2102 { { { "0", 64 }, { "0", 64 }, { "0", 64 }, { "0", 0 } } },
2103 { { { "25", 283 }, { "25", 288 }, { "25", 288 }, { "25", 256 } } },
2104 { { { "50", 502 }, { "50", 512 }, { "50", 512 }, { "50", 512 } } },
2105 { { { "75", 721 }, { "75", 736 }, { "75", 736 }, { "75", 768 } } },
2106 { { { "100", 940 }, { "100", 960 }, { "100", 960 }, { "100", 1023 } } },
2107};
2108
2109static const GraticuleLines ire12[] = {
2110 { { { "0", 256 }, { "0", 256 }, { "0", 256 }, { "0", 0 } } },
2111 { { { "25", 1132 }, { "25", 1152 }, { "25", 1152 }, { "25", 1024 } } },
2112 { { { "50", 2008 }, { "50", 2048 }, { "50", 2048 }, { "50", 2048 } } },
2113 { { { "75", 2884 }, { "75", 2944 }, { "75", 2944 }, { "75", 3072 } } },
2114 { { { "100", 3760 }, { "100", 3840 }, { "100", 3840 }, { "100", 4095 } } },
2115};
2116
2117static const GraticuleLines chroma_digital8[] = {
2118 { { { "50", 50 }, { "50", 50 }, { "50", 50 }, { "50", 50 } } },
2119 { { { "100", 100 }, { "100", 100 }, { "100", 100 }, { "100", 100 } } },
2120 { { { "150", 150 }, { "150", 150 }, { "150", 150 }, { "150", 150 } } },
2121 { { { "200", 200 }, { "200", 200 }, { "200", 200 }, { "200", 200 } } },
2122 { { { "255", 255 }, { "255", 255 }, { "255", 255 }, { "255", 255 } } },
2123};
2124
2125static const GraticuleLines chroma_digital9[] = {
2126 { { { "100", 100 }, { "100", 100 }, { "100", 100 }, { "100", 100 } } },
2127 { { { "200", 200 }, { "200", 200 }, { "200", 200 }, { "200", 200 } } },
2128 { { { "300", 300 }, { "300", 300 }, { "300", 300 }, { "300", 300 } } },
2129 { { { "400", 400 }, { "400", 400 }, { "400", 400 }, { "400", 400 } } },
2130 { { { "500", 500 }, { "500", 500 }, { "500", 500 }, { "500", 500 } } },
2131};
2132
2133static const GraticuleLines chroma_digital10[] = {
2134 { { { "200", 200 }, { "200", 200 }, { "200", 200 }, { "200", 200 } } },
2135 { { { "400", 400 }, { "400", 400 }, { "400", 400 }, { "400", 400 } } },
2136 { { { "600", 600 }, { "600", 600 }, { "600", 600 }, { "600", 600 } } },
2137 { { { "800", 800 }, { "800", 800 }, { "800", 800 }, { "800", 800 } } },
2138 { { {"1000",1000 }, {"1000",1000 }, {"1000",1000 }, {"1000",1000 } } },
2139};
2140
2141static const GraticuleLines chroma_digital12[] = {
2142 { { { "800", 800 }, { "800", 800 }, { "800", 800 }, { "800", 800 } } },
2143 { { { "1600", 1600 }, { "1600", 1600 }, { "1600", 1600 }, { "1600", 1600 } } },
2144 { { { "2400", 2400 }, { "2400", 2400 }, { "2400", 2400 }, { "2400", 2400 } } },
2145 { { { "3200", 3200 }, { "3200", 3200 }, { "3200", 3200 }, { "3200", 3200 } } },
2146 { { { "4000", 4000 }, { "4000", 4000 }, { "4000", 4000 }, { "4000", 4000 } } },
2147};
2148
2149static void blend_vline(uint8_t *dst, int height, int linesize, float o1, float o2, int v, int step)
2150{
2151 int y;
2152
2153 for (y = 0; y < height; y += step) {
2154 dst[0] = v * o1 + dst[0] * o2;
2155
2156 dst += linesize * step;
2157 }
2158}
2159
2160static void blend_vline16(uint16_t *dst, int height, int linesize, float o1, float o2, int v, int step)
2161{
2162 int y;
2163
2164 for (y = 0; y < height; y += step) {
2165 dst[0] = v * o1 + dst[0] * o2;
2166
2167 dst += (linesize / 2) * step;
2168 }
2169}
2170
2171static void blend_hline(uint8_t *dst, int width, float o1, float o2, int v, int step)
2172{
2173 int x;
2174
2175 for (x = 0; x < width; x += step) {
2176 dst[x] = v * o1 + dst[x] * o2;
2177 }
2178}
2179
2180static void blend_hline16(uint16_t *dst, int width, float o1, float o2, int v, int step)
2181{
2182 int x;
2183
2184 for (x = 0; x < width; x += step) {
2185 dst[x] = v * o1 + dst[x] * o2;
2186 }
2187}
2188
2189static void draw_htext(AVFrame *out, int x, int y, float o1, float o2, const char *txt, const uint8_t color[4])
2190{
2191 const uint8_t *font;
2192 int font_height;
2193 int i, plane;
2194
2195 font = avpriv_cga_font, font_height = 8;
2196
2197 for (plane = 0; plane < 4 && out->data[plane]; plane++) {
2198 for (i = 0; txt[i]; i++) {
2199 int char_y, mask;
2200 int v = color[plane];
2201
2202 uint8_t *p = out->data[plane] + y * out->linesize[plane] + (x + i * 8);
2203 for (char_y = 0; char_y < font_height; char_y++) {
2204 for (mask = 0x80; mask; mask >>= 1) {
2205 if (font[txt[i] * font_height + char_y] & mask)
2206 p[0] = p[0] * o2 + v * o1;
2207 p++;
2208 }
2209 p += out->linesize[plane] - 8;
2210 }
2211 }
2212 }
2213}
2214
2215static void draw_htext16(AVFrame *out, int x, int y, int mult, float o1, float o2, const char *txt, const uint8_t color[4])
2216{
2217 const uint8_t *font;
2218 int font_height;
2219 int i, plane;
2220
2221 font = avpriv_cga_font, font_height = 8;
2222
2223 for (plane = 0; plane < 4 && out->data[plane]; plane++) {
2224 for (i = 0; txt[i]; i++) {
2225 int char_y, mask;
2226 int v = color[plane] * mult;
2227
2228 uint16_t *p = (uint16_t *)(out->data[plane] + y * out->linesize[plane]) + (x + i * 8);
2229 for (char_y = 0; char_y < font_height; char_y++) {
2230 for (mask = 0x80; mask; mask >>= 1) {
2231 if (font[txt[i] * font_height + char_y] & mask)
2232 p[0] = p[0] * o2 + v * o1;
2233 p++;
2234 }
2235 p += out->linesize[plane] / 2 - 8;
2236 }
2237 }
2238 }
2239}
2240
2241static void draw_vtext(AVFrame *out, int x, int y, float o1, float o2, const char *txt, const uint8_t color[4])
2242{
2243 const uint8_t *font;
2244 int font_height;
2245 int i, plane;
2246
2247 font = avpriv_cga_font, font_height = 8;
2248
2249 for (plane = 0; plane < 4 && out->data[plane]; plane++) {
2250 for (i = 0; txt[i]; i++) {
2251 int char_y, mask;
2252 int v = color[plane];
2253
2254 for (char_y = font_height - 1; char_y >= 0; char_y--) {
2255 uint8_t *p = out->data[plane] + (y + i * 10) * out->linesize[plane] + x;
2256 for (mask = 0x80; mask; mask >>= 1) {
2257 if (font[txt[i] * font_height + font_height - 1 - char_y] & mask)
2258 p[char_y] = p[char_y] * o2 + v * o1;
2259 p += out->linesize[plane];
2260 }
2261 }
2262 }
2263 }
2264}
2265
2266static void draw_vtext16(AVFrame *out, int x, int y, int mult, float o1, float o2, const char *txt, const uint8_t color[4])
2267{
2268 const uint8_t *font;
2269 int font_height;
2270 int i, plane;
2271
2272 font = avpriv_cga_font, font_height = 8;
2273
2274 for (plane = 0; plane < 4 && out->data[plane]; plane++) {
2275 for (i = 0; txt[i]; i++) {
2276 int char_y, mask;
2277 int v = color[plane] * mult;
2278
2279 for (char_y = 0; char_y < font_height; char_y++) {
2280 uint16_t *p = (uint16_t *)(out->data[plane] + (y + i * 10) * out->linesize[plane]) + x;
2281 for (mask = 0x80; mask; mask >>= 1) {
2282 if (font[txt[i] * font_height + font_height - 1 - char_y] & mask)
2283 p[char_y] = p[char_y] * o2 + v * o1;
2284 p += out->linesize[plane] / 2;
2285 }
2286 }
2287 }
2288 }
2289}
2290
2291static void graticule_none(WaveformContext *s, AVFrame *out)
2292{
2293}
2294
2295static void graticule_green_row(WaveformContext *s, AVFrame *out)
2296{
2297 const int step = (s->flags & 2) + 1;
2298 const float o1 = s->opacity;
2299 const float o2 = 1. - o1;
2300 const int height = s->display == PARADE ? out->height / s->acomp : out->height;
2301 int k = 0, c, p, l, offset_x = 0, offset_y = 0;
2302
2303 for (c = 0; c < s->ncomp; c++) {
2304 if (!((1 << c) & s->pcomp) || (!s->display && k > 0))
2305 continue;
2306
2307 k++;
2308 for (p = 0; p < s->ncomp; p++) {
2309 const int v = green_yuva_color[p];
2310 for (l = 0; l < s->nb_glines; l++) {
2311 const uint16_t pos = s->glines[l].line[c].pos;
2312 int x = offset_x + (s->mirror ? s->size - 1 - pos : pos);
2313 uint8_t *dst = out->data[p] + offset_y * out->linesize[p] + x;
2314
2315 blend_vline(dst, height, out->linesize[p], o1, o2, v, step);
2316 }
2317 }
2318
2319 for (l = 0; l < s->nb_glines && (s->flags & 1); l++) {
2320 const char *name = s->glines[l].line[c].name;
2321 const uint16_t pos = s->glines[l].line[c].pos;
2322 int x = offset_x + (s->mirror ? s->size - 1 - pos : pos) - 10;
2323
2324 if (x < 0)
2325 x = 4;
2326
2327 draw_vtext(out, x, offset_y + 2, o1, o2, name, green_yuva_color);
2328 }
2329
2330 offset_x += s->size * (s->display == STACK);
2331 offset_y += height * (s->display == PARADE);
2332 }
2333}
2334
2335static void graticule16_green_row(WaveformContext *s, AVFrame *out)
2336{
2337 const int step = (s->flags & 2) + 1;
2338 const float o1 = s->opacity;
2339 const float o2 = 1. - o1;
2340 const int mult = s->size / 256;
2341 const int height = s->display == PARADE ? out->height / s->acomp : out->height;
2342 int k = 0, c, p, l, offset_x = 0, offset_y = 0;
2343
2344 for (c = 0; c < s->ncomp; c++) {
2345 if (!((1 << c) & s->pcomp) || (!s->display && k > 0))
2346 continue;
2347
2348 k++;
2349 for (p = 0; p < s->ncomp; p++) {
2350 const int v = green_yuva_color[p] * mult;
2351 for (l = 0; l < s->nb_glines ; l++) {
2352 const uint16_t pos = s->glines[l].line[c].pos;
2353 int x = offset_x + (s->mirror ? s->size - 1 - pos : pos);
2354 uint16_t *dst = (uint16_t *)(out->data[p] + offset_y * out->linesize[p]) + x;
2355
2356 blend_vline16(dst, height, out->linesize[p], o1, o2, v, step);
2357 }
2358 }
2359
2360 for (l = 0; l < s->nb_glines && (s->flags & 1); l++) {
2361 const char *name = s->glines[l].line[c].name;
2362 const uint16_t pos = s->glines[l].line[c].pos;
2363 int x = offset_x + (s->mirror ? s->size - 1 - pos : pos) - 10;
2364
2365 if (x < 0)
2366 x = 4;
2367
2368 draw_vtext16(out, x, offset_y + 2, mult, o1, o2, name, green_yuva_color);
2369 }
2370
2371 offset_x += s->size * (s->display == STACK);
2372 offset_y += height * (s->display == PARADE);
2373 }
2374}
2375
2376static void graticule_green_column(WaveformContext *s, AVFrame *out)
2377{
2378 const int step = (s->flags & 2) + 1;
2379 const float o1 = s->opacity;
2380 const float o2 = 1. - o1;
2381 const int width = s->display == PARADE ? out->width / s->acomp : out->width;
2382 int k = 0, c, p, l, offset_y = 0, offset_x = 0;
2383
2384 for (c = 0; c < s->ncomp; c++) {
2385 if ((!((1 << c) & s->pcomp) || (!s->display && k > 0)))
2386 continue;
2387
2388 k++;
2389 for (p = 0; p < s->ncomp; p++) {
2390 const int v = green_yuva_color[p];
2391 for (l = 0; l < s->nb_glines ; l++) {
2392 const uint16_t pos = s->glines[l].line[c].pos;
2393 int y = offset_y + (s->mirror ? s->size - 1 - pos : pos);
2394 uint8_t *dst = out->data[p] + y * out->linesize[p] + offset_x;
2395
2396 blend_hline(dst, width, o1, o2, v, step);
2397 }
2398 }
2399
2400 for (l = 0; l < s->nb_glines && (s->flags & 1); l++) {
2401 const char *name = s->glines[l].line[c].name;
2402 const uint16_t pos = s->glines[l].line[c].pos;
2403 int y = offset_y + (s->mirror ? s->size - 1 - pos : pos) - 10;
2404
2405 if (y < 0)
2406 y = 4;
2407
2408 draw_htext(out, 2 + offset_x, y, o1, o2, name, green_yuva_color);
2409 }
2410
2411 offset_y += s->size * (s->display == STACK);
2412 offset_x += width * (s->display == PARADE);
2413 }
2414}
2415
2416static void graticule16_green_column(WaveformContext *s, AVFrame *out)
2417{
2418 const int step = (s->flags & 2) + 1;
2419 const float o1 = s->opacity;
2420 const float o2 = 1. - o1;
2421 const int mult = s->size / 256;
2422 const int width = s->display == PARADE ? out->width / s->acomp : out->width;
2423 int k = 0, c, p, l, offset_x = 0, offset_y = 0;
2424
2425 for (c = 0; c < s->ncomp; c++) {
2426 if ((!((1 << c) & s->pcomp) || (!s->display && k > 0)))
2427 continue;
2428
2429 k++;
2430 for (p = 0; p < s->ncomp; p++) {
2431 const int v = green_yuva_color[p] * mult;
2432 for (l = 0; l < s->nb_glines ; l++) {
2433 const uint16_t pos = s->glines[l].line[c].pos;
2434 int y = offset_y + (s->mirror ? s->size - 1 - pos : pos);
2435 uint16_t *dst = (uint16_t *)(out->data[p] + y * out->linesize[p]) + offset_x;
2436
2437 blend_hline16(dst, width, o1, o2, v, step);
2438 }
2439 }
2440
2441 for (l = 0; l < s->nb_glines && (s->flags & 1); l++) {
2442 const char *name = s->glines[l].line[c].name;
2443 const uint16_t pos = s->glines[l].line[c].pos;
2444 int y = offset_y + (s->mirror ? s->size - 1 - pos: pos) - 10;
2445
2446 if (y < 0)
2447 y = 4;
2448
2449 draw_htext16(out, 2 + offset_x, y, mult, o1, o2, name, green_yuva_color);
2450 }
2451
2452 offset_y += s->size * (s->display == STACK);
2453 offset_x += width * (s->display == PARADE);
2454 }
2455}
2456
2457static int config_input(AVFilterLink *inlink)
2458{
2459 AVFilterContext *ctx = inlink->dst;
2460 WaveformContext *s = ctx->priv;
2461
2462 s->desc = av_pix_fmt_desc_get(inlink->format);
2463 s->ncomp = s->desc->nb_components;
2464 s->bits = s->desc->comp[0].depth;
2465 s->max = 1 << s->bits;
2466 s->intensity = s->fintensity * (s->max - 1);
2467
2468 s->shift_w[0] = s->shift_w[3] = 0;
2469 s->shift_h[0] = s->shift_h[3] = 0;
2470 s->shift_w[1] = s->shift_w[2] = s->desc->log2_chroma_w;
2471 s->shift_h[1] = s->shift_h[2] = s->desc->log2_chroma_h;
2472
2473 s->graticulef = graticule_none;
2474
2475 switch (s->filter) {
2476 case AFLAT: s->size = 256 * 2; break;
2477 case FLAT: s->size = 256 * 3; break;
2478 default: s->size = 256; break;
2479 }
2480
2481 switch (s->filter | ((s->bits > 8) << 4) |
2482 (s->mode << 8) | (s->mirror << 12)) {
2483 case 0x1100: s->waveform = lowpass_column_mirror; break;
2484 case 0x1000: s->waveform = lowpass_row_mirror; break;
2485 case 0x0100: s->waveform = lowpass_column; break;
2486 case 0x0000: s->waveform = lowpass_row; break;
2487 case 0x1110: s->waveform = lowpass16_column_mirror; break;
2488 case 0x1010: s->waveform = lowpass16_row_mirror; break;
2489 case 0x0110: s->waveform = lowpass16_column; break;
2490 case 0x0010: s->waveform = lowpass16_row; break;
2491 case 0x1101:
2492 case 0x1001:
2493 case 0x0101:
2494 case 0x0001: s->waveform = flat; break;
2495 case 0x1111:
2496 case 0x1011:
2497 case 0x0111:
2498 case 0x0011: s->waveform = flat16; break;
2499 case 0x1102:
2500 case 0x1002:
2501 case 0x0102:
2502 case 0x0002: s->waveform = aflat; break;
2503 case 0x1112:
2504 case 0x1012:
2505 case 0x0112:
2506 case 0x0012: s->waveform = aflat16; break;
2507 case 0x1103:
2508 case 0x1003:
2509 case 0x0103:
2510 case 0x0003: s->waveform = chroma; break;
2511 case 0x1113:
2512 case 0x1013:
2513 case 0x0113:
2514 case 0x0013: s->waveform = chroma16; break;
2515 case 0x1104:
2516 case 0x1004:
2517 case 0x0104:
2518 case 0x0004: s->waveform = color; break;
2519 case 0x1114:
2520 case 0x1014:
2521 case 0x0114:
2522 case 0x0014: s->waveform = color16; break;
2523 case 0x1105:
2524 case 0x1005:
2525 case 0x0105:
2526 case 0x0005: s->waveform = acolor; break;
2527 case 0x1115:
2528 case 0x1015:
2529 case 0x0115:
2530 case 0x0015: s->waveform = acolor16; break;
2531 }
2532
2533 switch (s->filter) {
2534 case LOWPASS:
2535 case COLOR:
2536 case ACOLOR:
2537 case CHROMA:
2538 case AFLAT:
2539 case FLAT:
2540 if (s->graticule && s->mode == 1)
2541 s->graticulef = s->bits > 8 ? graticule16_green_column : graticule_green_column;
2542 else if (s->graticule && s->mode == 0)
2543 s->graticulef = s->bits > 8 ? graticule16_green_row : graticule_green_row;
2544 break;
2545 }
2546
2547 switch (s->filter) {
2548 case COLOR:
2549 case ACOLOR:
2550 case LOWPASS:
2551 switch (s->scale) {
2552 case DIGITAL:
2553 switch (s->bits) {
2554 case 8: s->glines = (GraticuleLines *)digital8; s->nb_glines = FF_ARRAY_ELEMS(digital8); break;
2555 case 9: s->glines = (GraticuleLines *)digital9; s->nb_glines = FF_ARRAY_ELEMS(digital9); break;
2556 case 10: s->glines = (GraticuleLines *)digital10; s->nb_glines = FF_ARRAY_ELEMS(digital10); break;
2557 case 12: s->glines = (GraticuleLines *)digital12; s->nb_glines = FF_ARRAY_ELEMS(digital12); break;
2558 }
2559 break;
2560 case MILLIVOLTS:
2561 switch (s->bits) {
2562 case 8: s->glines = (GraticuleLines *)millivolts8; s->nb_glines = FF_ARRAY_ELEMS(millivolts8); break;
2563 case 9: s->glines = (GraticuleLines *)millivolts9; s->nb_glines = FF_ARRAY_ELEMS(millivolts9); break;
2564 case 10: s->glines = (GraticuleLines *)millivolts10; s->nb_glines = FF_ARRAY_ELEMS(millivolts10); break;
2565 case 12: s->glines = (GraticuleLines *)millivolts12; s->nb_glines = FF_ARRAY_ELEMS(millivolts12); break;
2566 }
2567 break;
2568 case IRE:
2569 switch (s->bits) {
2570 case 8: s->glines = (GraticuleLines *)ire8; s->nb_glines = FF_ARRAY_ELEMS(ire8); break;
2571 case 9: s->glines = (GraticuleLines *)ire9; s->nb_glines = FF_ARRAY_ELEMS(ire9); break;
2572 case 10: s->glines = (GraticuleLines *)ire10; s->nb_glines = FF_ARRAY_ELEMS(ire10); break;
2573 case 12: s->glines = (GraticuleLines *)ire12; s->nb_glines = FF_ARRAY_ELEMS(ire12); break;
2574 }
2575 break;
2576 }
2577 break;
2578 case CHROMA:
2579 switch (s->scale) {
2580 case DIGITAL:
2581 switch (s->bits) {
2582 case 8: s->glines = (GraticuleLines *)chroma_digital8; s->nb_glines = FF_ARRAY_ELEMS(chroma_digital8); break;
2583 case 9: s->glines = (GraticuleLines *)chroma_digital9; s->nb_glines = FF_ARRAY_ELEMS(chroma_digital9); break;
2584 case 10: s->glines = (GraticuleLines *)chroma_digital10; s->nb_glines = FF_ARRAY_ELEMS(chroma_digital10); break;
2585 case 12: s->glines = (GraticuleLines *)chroma_digital12; s->nb_glines = FF_ARRAY_ELEMS(chroma_digital12); break;
2586 }
2587 break;
2588 case MILLIVOLTS:
2589 switch (s->bits) {
2590 case 8: s->glines = (GraticuleLines *)millivolts8; s->nb_glines = FF_ARRAY_ELEMS(millivolts8); break;
2591 case 9: s->glines = (GraticuleLines *)millivolts9; s->nb_glines = FF_ARRAY_ELEMS(millivolts9); break;
2592 case 10: s->glines = (GraticuleLines *)millivolts10; s->nb_glines = FF_ARRAY_ELEMS(millivolts10); break;
2593 case 12: s->glines = (GraticuleLines *)millivolts12; s->nb_glines = FF_ARRAY_ELEMS(millivolts12); break;
2594 }
2595 break;
2596 case IRE:
2597 switch (s->bits) {
2598 case 8: s->glines = (GraticuleLines *)ire8; s->nb_glines = FF_ARRAY_ELEMS(ire8); break;
2599 case 9: s->glines = (GraticuleLines *)ire9; s->nb_glines = FF_ARRAY_ELEMS(ire9); break;
2600 case 10: s->glines = (GraticuleLines *)ire10; s->nb_glines = FF_ARRAY_ELEMS(ire10); break;
2601 case 12: s->glines = (GraticuleLines *)ire12; s->nb_glines = FF_ARRAY_ELEMS(ire12); break;
2602 }
2603 break;
2604 }
2605 break;
2606 case AFLAT:
2607 switch (s->scale) {
2608 case DIGITAL:
2609 switch (s->bits) {
2610 case 8: s->glines = (GraticuleLines *)aflat_digital8; s->nb_glines = FF_ARRAY_ELEMS(aflat_digital8); break;
2611 case 9: s->glines = (GraticuleLines *)aflat_digital9; s->nb_glines = FF_ARRAY_ELEMS(aflat_digital9); break;
2612 case 10: s->glines = (GraticuleLines *)aflat_digital10; s->nb_glines = FF_ARRAY_ELEMS(aflat_digital10); break;
2613 case 12: s->glines = (GraticuleLines *)aflat_digital12; s->nb_glines = FF_ARRAY_ELEMS(aflat_digital12); break;
2614 }
2615 break;
2616 case MILLIVOLTS:
2617 switch (s->bits) {
2618 case 8: s->glines = (GraticuleLines *)aflat_millivolts8; s->nb_glines = FF_ARRAY_ELEMS(aflat_millivolts8); break;
2619 case 9: s->glines = (GraticuleLines *)aflat_millivolts9; s->nb_glines = FF_ARRAY_ELEMS(aflat_millivolts9); break;
2620 case 10: s->glines = (GraticuleLines *)aflat_millivolts10; s->nb_glines = FF_ARRAY_ELEMS(aflat_millivolts10); break;
2621 case 12: s->glines = (GraticuleLines *)aflat_millivolts12; s->nb_glines = FF_ARRAY_ELEMS(aflat_millivolts12); break;
2622 }
2623 break;
2624 case IRE:
2625 switch (s->bits) {
2626 case 8: s->glines = (GraticuleLines *)aflat_ire8; s->nb_glines = FF_ARRAY_ELEMS(aflat_ire8); break;
2627 case 9: s->glines = (GraticuleLines *)aflat_ire9; s->nb_glines = FF_ARRAY_ELEMS(aflat_ire9); break;
2628 case 10: s->glines = (GraticuleLines *)aflat_ire10; s->nb_glines = FF_ARRAY_ELEMS(aflat_ire10); break;
2629 case 12: s->glines = (GraticuleLines *)aflat_ire12; s->nb_glines = FF_ARRAY_ELEMS(aflat_ire12); break;
2630 }
2631 break;
2632 }
2633 break;
2634 case FLAT:
2635 switch (s->scale) {
2636 case DIGITAL:
2637 switch (s->bits) {
2638 case 8: s->glines = (GraticuleLines *)flat_digital8; s->nb_glines = FF_ARRAY_ELEMS(flat_digital8); break;
2639 case 9: s->glines = (GraticuleLines *)flat_digital9; s->nb_glines = FF_ARRAY_ELEMS(flat_digital9); break;
2640 case 10: s->glines = (GraticuleLines *)flat_digital10; s->nb_glines = FF_ARRAY_ELEMS(flat_digital10); break;
2641 case 12: s->glines = (GraticuleLines *)flat_digital12; s->nb_glines = FF_ARRAY_ELEMS(flat_digital12); break;
2642 }
2643 break;
2644 case MILLIVOLTS:
2645 switch (s->bits) {
2646 case 8: s->glines = (GraticuleLines *)flat_millivolts8; s->nb_glines = FF_ARRAY_ELEMS(flat_millivolts8); break;
2647 case 9: s->glines = (GraticuleLines *)flat_millivolts9; s->nb_glines = FF_ARRAY_ELEMS(flat_millivolts9); break;
2648 case 10: s->glines = (GraticuleLines *)flat_millivolts10; s->nb_glines = FF_ARRAY_ELEMS(flat_millivolts10); break;
2649 case 12: s->glines = (GraticuleLines *)flat_millivolts12; s->nb_glines = FF_ARRAY_ELEMS(flat_millivolts12); break;
2650 }
2651 break;
2652 case IRE:
2653 switch (s->bits) {
2654 case 8: s->glines = (GraticuleLines *)flat_ire8; s->nb_glines = FF_ARRAY_ELEMS(flat_ire8); break;
2655 case 9: s->glines = (GraticuleLines *)flat_ire9; s->nb_glines = FF_ARRAY_ELEMS(flat_ire9); break;
2656 case 10: s->glines = (GraticuleLines *)flat_ire10; s->nb_glines = FF_ARRAY_ELEMS(flat_ire10); break;
2657 case 12: s->glines = (GraticuleLines *)flat_ire12; s->nb_glines = FF_ARRAY_ELEMS(flat_ire12); break;
2658 }
2659 break;
2660 }
2661 break;
2662 }
2663
2664 s->size = s->size << (s->bits - 8);
2665
2666 switch (inlink->format) {
2667 case AV_PIX_FMT_GBRAP:
2668 case AV_PIX_FMT_GBRP:
2669 case AV_PIX_FMT_GBRP9:
2670 case AV_PIX_FMT_GBRP10:
2671 case AV_PIX_FMT_GBRP12:
2672 memcpy(s->bg_color, black_gbrp_color, sizeof(s->bg_color));
2673 s->graticulef = graticule_none;
2674 break;
2675 default:
2676 memcpy(s->bg_color, black_yuva_color, sizeof(s->bg_color));
2677 }
2678
2679 s->bg_color[3] *= s->bgopacity;
2680
2681 return 0;
2682}
2683
2684static int config_output(AVFilterLink *outlink)
2685{
2686 AVFilterContext *ctx = outlink->src;
2687 AVFilterLink *inlink = ctx->inputs[0];
2688 WaveformContext *s = ctx->priv;
2689 int comp = 0, i, j = 0, k, p, size;
2690
2691 for (i = 0; i < s->ncomp; i++) {
2692 if ((1 << i) & s->pcomp)
2693 comp++;
2694 }
2695 s->acomp = comp;
2696 s->odesc = av_pix_fmt_desc_get(outlink->format);
2697 s->dcomp = s->odesc->nb_components;
2698
2699 av_freep(&s->peak);
2700
2701 if (s->mode) {
2702 outlink->h = s->size * FFMAX(comp * (s->display == STACK), 1);
2703 outlink->w = inlink->w * FFMAX(comp * (s->display == PARADE), 1);
2704 size = inlink->w;
2705 } else {
2706 outlink->w = s->size * FFMAX(comp * (s->display == STACK), 1);
2707 outlink->h = inlink->h * FFMAX(comp * (s->display == PARADE), 1);
2708 size = inlink->h;
2709 }
2710
2711 s->peak = av_malloc_array(size, 32 * sizeof(*s->peak));
2712 if (!s->peak)
2713 return AVERROR(ENOMEM);
2714
2715 for (p = 0; p < s->ncomp; p++) {
2716 const int plane = s->desc->comp[p].plane;
2717 int offset;
2718
2719 if (!((1 << p) & s->pcomp))
2720 continue;
2721
2722 for (k = 0; k < 4; k++) {
2723 s->emax[plane][k] = s->peak + size * (plane * 4 + k + 0);
2724 s->emin[plane][k] = s->peak + size * (plane * 4 + k + 16);
2725 }
2726
2727 offset = j++ * s->size * (s->display == STACK);
2728 s->estart[plane] = offset;
2729 s->eend[plane] = (offset + s->size - 1);
2730 for (i = 0; i < size; i++) {
2731 for (k = 0; k < 4; k++) {
2732 s->emax[plane][k][i] = s->estart[plane];
2733 s->emin[plane][k][i] = s->eend[plane];
2734 }
2735 }
2736 }
2737
2738 outlink->sample_aspect_ratio = (AVRational){1,1};
2739
2740 return 0;
2741}
2742
2743static int filter_frame(AVFilterLink *inlink, AVFrame *in)
2744{
2745 AVFilterContext *ctx = inlink->dst;
2746 WaveformContext *s = ctx->priv;
2747 AVFilterLink *outlink = ctx->outputs[0];
2748 AVFrame *out;
2749 int i, j, k;
2750
2751 out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
2752 if (!out) {
2753 av_frame_free(&in);
2754 return AVERROR(ENOMEM);
2755 }
2756 out->pts = in->pts;
2757 av_frame_set_color_range(out, AVCOL_RANGE_JPEG);
2758
2759 for (k = 0; k < s->dcomp; k++) {
2760 if (s->bits <= 8) {
2761 for (i = 0; i < outlink->h ; i++)
2762 memset(out->data[s->odesc->comp[k].plane] +
2763 i * out->linesize[s->odesc->comp[k].plane],
2764 s->bg_color[k], outlink->w);
2765 } else {
2766 const int mult = s->max / 256;
2767 uint16_t *dst = (uint16_t *)out->data[s->odesc->comp[k].plane];
2768
2769 for (i = 0; i < outlink->h ; i++) {
2770 for (j = 0; j < outlink->w; j++)
2771 dst[j] = s->bg_color[k] * mult;
2772 dst += out->linesize[s->odesc->comp[k].plane] / 2;
2773 }
2774 }
2775 }
2776
2777 for (k = 0, i = 0; k < s->ncomp; k++) {
2778 if ((1 << k) & s->pcomp) {
2779 int offset_y;
2780 int offset_x;
2781
2782 if (s->display == PARADE) {
2783 offset_x = s->mode ? i++ * inlink->w : 0;
2784 offset_y = s->mode ? 0 : i++ * inlink->h;
2785 } else {
2786 offset_y = s->mode ? i++ * s->size * !!s->display : 0;
2787 offset_x = s->mode ? 0 : i++ * s->size * !!s->display;
2788 }
2789 s->waveform(s, in, out, k, s->intensity, offset_y, offset_x, s->mode, s->mirror);
2790 }
2791 }
2792 s->graticulef(s, out);
2793
2794 av_frame_free(&in);
2795 return ff_filter_frame(outlink, out);
2796}
2797
2798static av_cold void uninit(AVFilterContext *ctx)
2799{
2800 WaveformContext *s = ctx->priv;
2801
2802 av_freep(&s->peak);
2803}
2804
2805static const AVFilterPad inputs[] = {
2806 {
2807 .name = "default",
2808 .type = AVMEDIA_TYPE_VIDEO,
2809 .filter_frame = filter_frame,
2810 .config_props = config_input,
2811 },
2812 { NULL }
2813};
2814
2815static const AVFilterPad outputs[] = {
2816 {
2817 .name = "default",
2818 .type = AVMEDIA_TYPE_VIDEO,
2819 .config_props = config_output,
2820 },
2821 { NULL }
2822};
2823
2824AVFilter ff_vf_waveform = {
2825 .name = "waveform",
2826 .description = NULL_IF_CONFIG_SMALL("Video waveform monitor."),
2827 .priv_size = sizeof(WaveformContext),
2828 .priv_class = &waveform_class,
2829 .query_formats = query_formats,
2830 .uninit = uninit,
2831 .inputs = inputs,
2832 .outputs = outputs,
2833};
2834