-static av_always_inline void aflat(WaveformContext *s,
- AVFrame *in, AVFrame *out,
- int component, int intensity,
- int offset_y, int offset_x,
- int column, int mirror)
-{
- const int plane = s->desc->comp[component].plane;
- const int c0_linesize = in->linesize[ plane + 0 ];
- const int c1_linesize = in->linesize[(plane + 1) % s->ncomp];
- const int c2_linesize = in->linesize[(plane + 2) % s->ncomp];
- const int c0_shift_w = s->shift_w[ component + 0 ];
- const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp];
- const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp];
- const int c0_shift_h = s->shift_h[ component + 0 ];
- const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp];
- const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp];
- const int d0_linesize = out->linesize[ plane + 0 ];
- const int d1_linesize = out->linesize[(plane + 1) % s->ncomp];
- const int d2_linesize = out->linesize[(plane + 2) % s->ncomp];
- const int max = 255 - intensity;
- const int src_h = in->height;
- const int src_w = in->width;
- int x, y;
-
- if (column) {
- const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1);
- const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1);
- const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1);
-
- for (x = 0; x < src_w; x++) {
- const uint8_t *c0_data = in->data[plane + 0];
- const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
- const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
- uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
- uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
- uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
- uint8_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1);
- uint8_t * const d0 = (mirror ? d0_bottom_line : d0_data);
- uint8_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1);
- uint8_t * const d1 = (mirror ? d1_bottom_line : d1_data);
- uint8_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1);
- uint8_t * const d2 = (mirror ? d2_bottom_line : d2_data);
-
- for (y = 0; y < src_h; y++) {
- const int c0 = c0_data[x >> c0_shift_w] + 128;
- const int c1 = c1_data[x >> c1_shift_w] - 128;
- const int c2 = c2_data[x >> c2_shift_w] - 128;
- uint8_t *target;
-
- target = d0 + x + d0_signed_linesize * c0;
- update(target, max, intensity);
-
- target = d1 + x + d1_signed_linesize * (c0 + c1);
- update(target, max, intensity);
-
- target = d2 + x + d2_signed_linesize * (c0 + c2);
- update(target, max, intensity);
-
- if (!c0_shift_h || (y & c0_shift_h))
- c0_data += c0_linesize;
- if (!c1_shift_h || (y & c1_shift_h))
- c1_data += c1_linesize;
- if (!c1_shift_h || (y & c1_shift_h))
- c2_data += c1_linesize;
- d0_data += d0_linesize;
- d1_data += d1_linesize;
- d2_data += d2_linesize;
- }
- }
- } else {
- const uint8_t *c0_data = in->data[plane];
- const uint8_t *c1_data = in->data[(plane + 1) % s->ncomp];
- const uint8_t *c2_data = in->data[(plane + 2) % s->ncomp];
- uint8_t *d0_data = out->data[plane] + offset_y * d0_linesize + offset_x;
- uint8_t *d1_data = out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x;
- uint8_t *d2_data = out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x;
-
- if (mirror) {
- d0_data += s->size - 1;
- d1_data += s->size - 1;
- d2_data += s->size - 1;
- }
-
- for (y = 0; y < src_h; y++) {
- for (x = 0; x < src_w; x++) {
- const int c0 = c0_data[x >> c0_shift_w] + 128;
- const int c1 = c1_data[x >> c1_shift_w] - 128;
- const int c2 = c2_data[x >> c2_shift_w] - 128;
- uint8_t *target;
-
- if (mirror) {
- target = d0_data - c0;
- update(target, max, intensity);
- target = d1_data - (c0 + c1);
- update(target, max, intensity);
- target = d2_data - (c0 + c2);
- update(target, max, intensity);
- } else {
- target = d0_data + c0;
- update(target, max, intensity);
- target = d1_data + (c0 + c1);
- update(target, max, intensity);
- target = d2_data + (c0 + c2);
- update(target, max, intensity);
- }
- }
-
- if (!c0_shift_h || (y & c0_shift_h))
- c0_data += c0_linesize;
- if (!c1_shift_h || (y & c1_shift_h))
- c1_data += c1_linesize;
- if (!c2_shift_h || (y & c2_shift_h))
- c2_data += c2_linesize;
- d0_data += d0_linesize;
- d1_data += d1_linesize;
- d2_data += d2_linesize;
- }
- }
+FLAT_FUNC(column_mirror, 1, 1)
+FLAT_FUNC(column, 1, 0)
+FLAT_FUNC(row_mirror, 0, 1)
+FLAT_FUNC(row, 0, 0)
+
+#define AFLAT16(name, update_cr, column, mirror) \
+static int name(AVFilterContext *ctx, \
+ void *arg, int jobnr, \
+ int nb_jobs) \
+{ \
+ WaveformContext *s = ctx->priv; \
+ ThreadData *td = arg; \
+ AVFrame *in = td->in; \
+ AVFrame *out = td->out; \
+ int component = td->component; \
+ int offset_y = td->offset_y; \
+ int offset_x = td->offset_x; \
+ const int intensity = s->intensity; \
+ const int plane = s->desc->comp[component].plane; \
+ const int c0_linesize = in->linesize[ plane + 0 ] / 2; \
+ const int c1_linesize = in->linesize[(plane + 1) % s->ncomp] / 2; \
+ const int c2_linesize = in->linesize[(plane + 2) % s->ncomp] / 2; \
+ const int c0_shift_w = s->shift_w[ component + 0 ]; \
+ const int c1_shift_w = s->shift_w[(component + 1) % s->ncomp]; \
+ const int c2_shift_w = s->shift_w[(component + 2) % s->ncomp]; \
+ const int c0_shift_h = s->shift_h[ component + 0 ]; \
+ const int c1_shift_h = s->shift_h[(component + 1) % s->ncomp]; \
+ const int c2_shift_h = s->shift_h[(component + 2) % s->ncomp]; \
+ const int d0_linesize = out->linesize[ plane + 0 ] / 2; \
+ const int d1_linesize = out->linesize[(plane + 1) % s->ncomp] / 2; \
+ const int d2_linesize = out->linesize[(plane + 2) % s->ncomp] / 2; \
+ const int limit = s->max - 1; \
+ const int max = limit - intensity; \
+ const int mid = s->max / 2; \
+ const int src_h = in->height; \
+ const int src_w = in->width; \
+ const int sliceh_start = !column ? (src_h * jobnr) / nb_jobs : 0; \
+ const int sliceh_end = !column ? (src_h * (jobnr+1)) / nb_jobs : src_h; \
+ const int slicew_start = column ? (src_w * jobnr) / nb_jobs : 0; \
+ const int slicew_end = column ? (src_w * (jobnr+1)) / nb_jobs : src_w; \
+ int x, y; \
+ \
+ if (column) { \
+ const int d0_signed_linesize = d0_linesize * (mirror == 1 ? -1 : 1); \
+ const int d1_signed_linesize = d1_linesize * (mirror == 1 ? -1 : 1); \
+ const int d2_signed_linesize = d2_linesize * (mirror == 1 ? -1 : 1); \
+ \
+ for (x = slicew_start; x < slicew_end; x++) { \
+ const uint16_t *c0_data = (uint16_t *)in->data[plane + 0]; \
+ const uint16_t *c1_data = (uint16_t *)in->data[(plane + 1) % s->ncomp]; \
+ const uint16_t *c2_data = (uint16_t *)in->data[(plane + 2) % s->ncomp]; \
+ uint16_t *d0_data = (uint16_t *)out->data[plane] + offset_y * d0_linesize + offset_x; \
+ uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + offset_y * d1_linesize + offset_x; \
+ uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + offset_y * d2_linesize + offset_x; \
+ uint16_t * const d0_bottom_line = d0_data + d0_linesize * (s->size - 1); \
+ uint16_t * const d0 = (mirror ? d0_bottom_line : d0_data); \
+ uint16_t * const d1_bottom_line = d1_data + d1_linesize * (s->size - 1); \
+ uint16_t * const d1 = (mirror ? d1_bottom_line : d1_data); \
+ uint16_t * const d2_bottom_line = d2_data + d2_linesize * (s->size - 1); \
+ uint16_t * const d2 = (mirror ? d2_bottom_line : d2_data); \
+ \
+ for (y = 0; y < src_h; y++) { \
+ const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit) + mid; \
+ const int c1 = FFMIN(c1_data[x >> c1_shift_w], limit) - mid; \
+ const int c2 = FFMIN(c2_data[x >> c2_shift_w], limit) - mid; \
+ uint16_t *target; \
+ \
+ target = d0 + x + d0_signed_linesize * c0; \
+ update16(target, max, intensity, limit); \
+ \
+ target = d1 + x + d1_signed_linesize * (c0 + c1); \
+ update16(target, max, intensity, limit); \
+ \
+ target = d2 + x + d2_signed_linesize * (c0 + c2); \
+ update_cr(target, max, intensity, limit); \
+ \
+ if (!c0_shift_h || (y & c0_shift_h)) \
+ c0_data += c0_linesize; \
+ if (!c1_shift_h || (y & c1_shift_h)) \
+ c1_data += c1_linesize; \
+ if (!c2_shift_h || (y & c2_shift_h)) \
+ c2_data += c2_linesize; \
+ d0_data += d0_linesize; \
+ d1_data += d1_linesize; \
+ d2_data += d2_linesize; \
+ } \
+ } \
+ } else { \
+ const uint16_t *c0_data = (uint16_t *)in->data[plane] + (sliceh_start >> c0_shift_h) * c0_linesize; \
+ const uint16_t *c1_data = (uint16_t *)in->data[(plane + 1) % s->ncomp] + (sliceh_start >> c1_shift_h) * c1_linesize; \
+ const uint16_t *c2_data = (uint16_t *)in->data[(plane + 2) % s->ncomp] + (sliceh_start >> c2_shift_h) * c2_linesize; \
+ uint16_t *d0_data = (uint16_t *)out->data[plane] + (offset_y + sliceh_start) * d0_linesize + offset_x; \
+ uint16_t *d1_data = (uint16_t *)out->data[(plane + 1) % s->ncomp] + (offset_y + sliceh_start) * d1_linesize + offset_x; \
+ uint16_t *d2_data = (uint16_t *)out->data[(plane + 2) % s->ncomp] + (offset_y + sliceh_start) * d2_linesize + offset_x; \
+ \
+ if (mirror) { \
+ d0_data += s->size - 1; \
+ d1_data += s->size - 1; \
+ d2_data += s->size - 1; \
+ } \
+ \
+ for (y = sliceh_start; y < sliceh_end; y++) { \
+ for (x = 0; x < src_w; x++) { \
+ const int c0 = FFMIN(c0_data[x >> c0_shift_w], limit) + mid; \
+ const int c1 = FFMIN(c1_data[x >> c1_shift_w], limit) - mid; \
+ const int c2 = FFMIN(c2_data[x >> c2_shift_w], limit) - mid; \
+ uint16_t *target; \
+ \
+ if (mirror) { \
+ target = d0_data - c0; \
+ update16(target, max, intensity, limit); \
+ target = d1_data - (c0 + c1); \
+ update16(target, max, intensity, limit); \
+ target = d2_data - (c0 + c2); \
+ update_cr(target, max, intensity, limit); \
+ } else { \
+ target = d0_data + c0; \
+ update16(target, max, intensity, limit); \
+ target = d1_data + (c0 + c1); \
+ update16(target, max, intensity, limit); \
+ target = d2_data + (c0 + c2); \
+ update_cr(target, max, intensity, limit); \
+ } \
+ } \
+ \
+ if (!c0_shift_h || (y & c0_shift_h)) \
+ c0_data += c0_linesize; \
+ if (!c1_shift_h || (y & c1_shift_h)) \
+ c1_data += c1_linesize; \
+ if (!c2_shift_h || (y & c2_shift_h)) \
+ c2_data += c2_linesize; \
+ d0_data += d0_linesize; \
+ d1_data += d1_linesize; \
+ d2_data += d2_linesize; \
+ } \
+ } \
+ return 0; \
+}