FFmpeg  4.4
drawutils.c
Go to the documentation of this file.
1 /*
2  * Copyright 2011 Stefano Sabatini <stefano.sabatini-lala poste it>
3  * Copyright 2012 Nicolas George <nicolas.george normalesup org>
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 <string.h>
23 
24 #include "libavutil/avassert.h"
25 #include "libavutil/avutil.h"
26 #include "libavutil/colorspace.h"
27 #include "libavutil/intreadwrite.h"
28 #include "libavutil/mem.h"
29 #include "libavutil/pixdesc.h"
30 #include "drawutils.h"
31 #include "formats.h"
32 
33 enum { RED = 0, GREEN, BLUE, ALPHA };
34 
36 {
37  switch (pix_fmt) {
38  case AV_PIX_FMT_0RGB:
39  case AV_PIX_FMT_ARGB: rgba_map[ALPHA] = 0; rgba_map[RED ] = 1; rgba_map[GREEN] = 2; rgba_map[BLUE ] = 3; break;
40  case AV_PIX_FMT_0BGR:
41  case AV_PIX_FMT_ABGR: rgba_map[ALPHA] = 0; rgba_map[BLUE ] = 1; rgba_map[GREEN] = 2; rgba_map[RED ] = 3; break;
42  case AV_PIX_FMT_RGB48LE:
43  case AV_PIX_FMT_RGB48BE:
46  case AV_PIX_FMT_RGB0:
47  case AV_PIX_FMT_RGBA:
48  case AV_PIX_FMT_RGB24: rgba_map[RED ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break;
49  case AV_PIX_FMT_BGR48LE:
50  case AV_PIX_FMT_BGR48BE:
53  case AV_PIX_FMT_BGRA:
54  case AV_PIX_FMT_BGR0:
55  case AV_PIX_FMT_BGR24: rgba_map[BLUE ] = 0; rgba_map[GREEN] = 1; rgba_map[RED ] = 2; rgba_map[ALPHA] = 3; break;
56  case AV_PIX_FMT_GBRP9LE:
57  case AV_PIX_FMT_GBRP9BE:
66  case AV_PIX_FMT_GBRAP:
77  case AV_PIX_FMT_GBRP: rgba_map[GREEN] = 0; rgba_map[BLUE ] = 1; rgba_map[RED ] = 2; rgba_map[ALPHA] = 3; break;
78  default: /* unsupported */
79  return AVERROR(EINVAL);
80  }
81  return 0;
82 }
83 
85 {
87  const AVComponentDescriptor *c;
88  unsigned i, nb_planes = 0;
89  int pixelstep[MAX_PLANES] = { 0 };
90  int full_range = 0;
91 
92  if (!desc || !desc->name)
93  return AVERROR(EINVAL);
95  return AVERROR(ENOSYS);
97  return AVERROR(ENOSYS);
100  full_range = 1;
101  for (i = 0; i < desc->nb_components; i++) {
102  c = &desc->comp[i];
103  /* for now, only 8-16 bits formats */
104  if (c->depth < 8 || c->depth > 16)
105  return AVERROR(ENOSYS);
106  if (desc->flags & AV_PIX_FMT_FLAG_BE)
107  return AVERROR(ENOSYS);
108  if (c->plane >= MAX_PLANES)
109  return AVERROR(ENOSYS);
110  /* strange interleaving */
111  if (pixelstep[c->plane] != 0 &&
112  pixelstep[c->plane] != c->step)
113  return AVERROR(ENOSYS);
114  if (pixelstep[c->plane] == 6 &&
115  c->depth == 16)
116  return AVERROR(ENOSYS);
117  pixelstep[c->plane] = c->step;
118  if (pixelstep[c->plane] >= 8)
119  return AVERROR(ENOSYS);
120  nb_planes = FFMAX(nb_planes, c->plane + 1);
121  }
122  memset(draw, 0, sizeof(*draw));
123  draw->desc = desc;
124  draw->format = format;
125  draw->nb_planes = nb_planes;
126  draw->flags = flags;
127  draw->full_range = full_range;
128  memcpy(draw->pixelstep, pixelstep, sizeof(draw->pixelstep));
129  draw->hsub[1] = draw->hsub[2] = draw->hsub_max = desc->log2_chroma_w;
130  draw->vsub[1] = draw->vsub[2] = draw->vsub_max = desc->log2_chroma_h;
131  for (i = 0; i < (desc->nb_components - !!(desc->flags & AV_PIX_FMT_FLAG_ALPHA && !(flags & FF_DRAW_PROCESS_ALPHA))); i++)
132  draw->comp_mask[desc->comp[i].plane] |=
133  1 << desc->comp[i].offset;
134  return 0;
135 }
136 
138 {
139  unsigned i;
140  uint8_t rgba_map[4];
141 
142  if (rgba != color->rgba)
143  memcpy(color->rgba, rgba, sizeof(color->rgba));
144  if ((draw->desc->flags & AV_PIX_FMT_FLAG_RGB) &&
145  ff_fill_rgba_map(rgba_map, draw->format) >= 0) {
146  if (draw->nb_planes == 1) {
147  for (i = 0; i < 4; i++) {
148  color->comp[0].u8[rgba_map[i]] = rgba[i];
149  if (draw->desc->comp[rgba_map[i]].depth > 8) {
150  color->comp[0].u16[rgba_map[i]] = color->comp[0].u8[rgba_map[i]] << 8;
151  }
152  }
153  } else {
154  for (i = 0; i < 4; i++) {
155  color->comp[rgba_map[i]].u8[0] = rgba[i];
156  if (draw->desc->comp[rgba_map[i]].depth > 8)
157  color->comp[rgba_map[i]].u16[0] = color->comp[rgba_map[i]].u8[0] << (draw->desc->comp[rgba_map[i]].depth - 8);
158  }
159  }
160  } else if (draw->nb_planes >= 2) {
161  /* assume YUV */
162  const AVPixFmtDescriptor *desc = draw->desc;
163  color->comp[desc->comp[0].plane].u8[desc->comp[0].offset] = draw->full_range ? RGB_TO_Y_JPEG(rgba[0], rgba[1], rgba[2]) : RGB_TO_Y_CCIR(rgba[0], rgba[1], rgba[2]);
164  color->comp[desc->comp[1].plane].u8[desc->comp[1].offset] = draw->full_range ? RGB_TO_U_JPEG(rgba[0], rgba[1], rgba[2]) : RGB_TO_U_CCIR(rgba[0], rgba[1], rgba[2], 0);
165  color->comp[desc->comp[2].plane].u8[desc->comp[2].offset] = draw->full_range ? RGB_TO_V_JPEG(rgba[0], rgba[1], rgba[2]) : RGB_TO_V_CCIR(rgba[0], rgba[1], rgba[2], 0);
166  color->comp[3].u8[0] = rgba[3];
167 #define EXPAND(compn) \
168  if (desc->comp[compn].depth > 8) \
169  color->comp[desc->comp[compn].plane].u16[desc->comp[compn].offset] = \
170  color->comp[desc->comp[compn].plane].u8[desc->comp[compn].offset] << \
171  (draw->desc->comp[compn].depth + draw->desc->comp[compn].shift - 8)
172  EXPAND(3);
173  EXPAND(2);
174  EXPAND(1);
175  EXPAND(0);
176  } else if (draw->format == AV_PIX_FMT_GRAY8 || draw->format == AV_PIX_FMT_GRAY8A ||
177  draw->format == AV_PIX_FMT_GRAY16LE || draw->format == AV_PIX_FMT_YA16LE ||
178  draw->format == AV_PIX_FMT_GRAY9LE ||
179  draw->format == AV_PIX_FMT_GRAY10LE ||
180  draw->format == AV_PIX_FMT_GRAY12LE ||
181  draw->format == AV_PIX_FMT_GRAY14LE) {
182  const AVPixFmtDescriptor *desc = draw->desc;
183  color->comp[0].u8[0] = RGB_TO_Y_CCIR(rgba[0], rgba[1], rgba[2]);
184  EXPAND(0);
185  color->comp[1].u8[0] = rgba[3];
186  EXPAND(1);
187  } else {
189  "Color conversion not implemented for %s\n", draw->desc->name);
190  memset(color, 128, sizeof(*color));
191  }
192 }
193 
194 static uint8_t *pointer_at(FFDrawContext *draw, uint8_t *data[], int linesize[],
195  int plane, int x, int y)
196 {
197  return data[plane] +
198  (y >> draw->vsub[plane]) * linesize[plane] +
199  (x >> draw->hsub[plane]) * draw->pixelstep[plane];
200 }
201 
203  uint8_t *dst[], int dst_linesize[],
204  uint8_t *src[], int src_linesize[],
205  int dst_x, int dst_y, int src_x, int src_y,
206  int w, int h)
207 {
208  int plane, y, wp, hp;
209  uint8_t *p, *q;
210 
211  for (plane = 0; plane < draw->nb_planes; plane++) {
212  p = pointer_at(draw, src, src_linesize, plane, src_x, src_y);
213  q = pointer_at(draw, dst, dst_linesize, plane, dst_x, dst_y);
214  wp = AV_CEIL_RSHIFT(w, draw->hsub[plane]) * draw->pixelstep[plane];
215  hp = AV_CEIL_RSHIFT(h, draw->vsub[plane]);
216  for (y = 0; y < hp; y++) {
217  memcpy(q, p, wp);
218  p += src_linesize[plane];
219  q += dst_linesize[plane];
220  }
221  }
222 }
223 
225  uint8_t *dst[], int dst_linesize[],
226  int dst_x, int dst_y, int w, int h)
227 {
228  int plane, x, y, wp, hp;
229  uint8_t *p0, *p;
230  FFDrawColor color_tmp = *color;
231 
232  for (plane = 0; plane < draw->nb_planes; plane++) {
233  p0 = pointer_at(draw, dst, dst_linesize, plane, dst_x, dst_y);
234  wp = AV_CEIL_RSHIFT(w, draw->hsub[plane]);
235  hp = AV_CEIL_RSHIFT(h, draw->vsub[plane]);
236  if (!hp)
237  return;
238  p = p0;
239 
240  if (HAVE_BIGENDIAN && draw->desc->comp[0].depth > 8) {
241  for (x = 0; 2*x < draw->pixelstep[plane]; x++)
242  color_tmp.comp[plane].u16[x] = av_bswap16(color_tmp.comp[plane].u16[x]);
243  }
244 
245  /* copy first line from color */
246  for (x = 0; x < wp; x++) {
247  memcpy(p, color_tmp.comp[plane].u8, draw->pixelstep[plane]);
248  p += draw->pixelstep[plane];
249  }
250  wp *= draw->pixelstep[plane];
251  /* copy next lines from first line */
252  p = p0 + dst_linesize[plane];
253  for (y = 1; y < hp; y++) {
254  memcpy(p, p0, wp);
255  p += dst_linesize[plane];
256  }
257  }
258 }
259 
260 /**
261  * Clip interval [x; x+w[ within [0; wmax[.
262  * The resulting w may be negative if the final interval is empty.
263  * dx, if not null, return the difference between in and out value of x.
264  */
265 static void clip_interval(int wmax, int *x, int *w, int *dx)
266 {
267  if (dx)
268  *dx = 0;
269  if (*x < 0) {
270  if (dx)
271  *dx = -*x;
272  *w += *x;
273  *x = 0;
274  }
275  if (*x + *w > wmax)
276  *w = wmax - *x;
277 }
278 
279 /**
280  * Decompose w pixels starting at x
281  * into start + (w starting at x) + end
282  * with x and w aligned on multiples of 1<<sub.
283  */
284 static void subsampling_bounds(int sub, int *x, int *w, int *start, int *end)
285 {
286  int mask = (1 << sub) - 1;
287 
288  *start = (-*x) & mask;
289  *x += *start;
290  *start = FFMIN(*start, *w);
291  *w -= *start;
292  *end = *w & mask;
293  *w >>= sub;
294 }
295 
296 static int component_used(FFDrawContext *draw, int plane, int comp)
297 {
298  return (draw->comp_mask[plane] >> comp) & 1;
299 }
300 
301 /* If alpha is in the [ 0 ; 0x1010101 ] range,
302  then alpha * value is in the [ 0 ; 0xFFFFFFFF ] range,
303  and >> 24 gives a correct rounding. */
304 static void blend_line(uint8_t *dst, unsigned src, unsigned alpha,
305  int dx, int w, unsigned hsub, int left, int right)
306 {
307  unsigned asrc = alpha * src;
308  unsigned tau = 0x1010101 - alpha;
309  int x;
310 
311  if (left) {
312  unsigned suba = (left * alpha) >> hsub;
313  *dst = (*dst * (0x1010101 - suba) + src * suba) >> 24;
314  dst += dx;
315  }
316  for (x = 0; x < w; x++) {
317  *dst = (*dst * tau + asrc) >> 24;
318  dst += dx;
319  }
320  if (right) {
321  unsigned suba = (right * alpha) >> hsub;
322  *dst = (*dst * (0x1010101 - suba) + src * suba) >> 24;
323  }
324 }
325 
326 static void blend_line16(uint8_t *dst, unsigned src, unsigned alpha,
327  int dx, int w, unsigned hsub, int left, int right)
328 {
329  unsigned asrc = alpha * src;
330  unsigned tau = 0x10001 - alpha;
331  int x;
332 
333  if (left) {
334  unsigned suba = (left * alpha) >> hsub;
335  uint16_t value = AV_RL16(dst);
336  AV_WL16(dst, (value * (0x10001 - suba) + src * suba) >> 16);
337  dst += dx;
338  }
339  for (x = 0; x < w; x++) {
340  uint16_t value = AV_RL16(dst);
341  AV_WL16(dst, (value * tau + asrc) >> 16);
342  dst += dx;
343  }
344  if (right) {
345  unsigned suba = (right * alpha) >> hsub;
346  uint16_t value = AV_RL16(dst);
347  AV_WL16(dst, (value * (0x10001 - suba) + src * suba) >> 16);
348  }
349 }
350 
352  uint8_t *dst[], int dst_linesize[],
353  int dst_w, int dst_h,
354  int x0, int y0, int w, int h)
355 {
356  unsigned alpha, nb_planes, nb_comp, plane, comp;
357  int w_sub, h_sub, x_sub, y_sub, left, right, top, bottom, y;
358  uint8_t *p0, *p;
359 
360  /* TODO optimize if alpha = 0xFF */
361  clip_interval(dst_w, &x0, &w, NULL);
362  clip_interval(dst_h, &y0, &h, NULL);
363  if (w <= 0 || h <= 0 || !color->rgba[3])
364  return;
365  if (draw->desc->comp[0].depth <= 8) {
366  /* 0x10203 * alpha + 2 is in the [ 2 ; 0x1010101 - 2 ] range */
367  alpha = 0x10203 * color->rgba[3] + 0x2;
368  } else {
369  /* 0x101 * alpha is in the [ 2 ; 0x1001] range */
370  alpha = 0x101 * color->rgba[3] + 0x2;
371  }
372  nb_planes = draw->nb_planes - !!(draw->desc->flags & AV_PIX_FMT_FLAG_ALPHA && !(draw->flags & FF_DRAW_PROCESS_ALPHA));
373  nb_planes += !nb_planes;
374  for (plane = 0; plane < nb_planes; plane++) {
375  nb_comp = draw->pixelstep[plane];
376  p0 = pointer_at(draw, dst, dst_linesize, plane, x0, y0);
377  w_sub = w;
378  h_sub = h;
379  x_sub = x0;
380  y_sub = y0;
381  subsampling_bounds(draw->hsub[plane], &x_sub, &w_sub, &left, &right);
382  subsampling_bounds(draw->vsub[plane], &y_sub, &h_sub, &top, &bottom);
383  for (comp = 0; comp < nb_comp; comp++) {
384  const int depth = draw->desc->comp[comp].depth;
385 
386  if (!component_used(draw, plane, comp))
387  continue;
388  p = p0 + comp;
389  if (top) {
390  if (depth <= 8) {
391  blend_line(p, color->comp[plane].u8[comp], alpha >> 1,
392  draw->pixelstep[plane], w_sub,
393  draw->hsub[plane], left, right);
394  } else {
395  blend_line16(p, color->comp[plane].u16[comp], alpha >> 1,
396  draw->pixelstep[plane], w_sub,
397  draw->hsub[plane], left, right);
398  }
399  p += dst_linesize[plane];
400  }
401  if (depth <= 8) {
402  for (y = 0; y < h_sub; y++) {
403  blend_line(p, color->comp[plane].u8[comp], alpha,
404  draw->pixelstep[plane], w_sub,
405  draw->hsub[plane], left, right);
406  p += dst_linesize[plane];
407  }
408  } else {
409  for (y = 0; y < h_sub; y++) {
410  blend_line16(p, color->comp[plane].u16[comp], alpha,
411  draw->pixelstep[plane], w_sub,
412  draw->hsub[plane], left, right);
413  p += dst_linesize[plane];
414  }
415  }
416  if (bottom) {
417  if (depth <= 8) {
418  blend_line(p, color->comp[plane].u8[comp], alpha >> 1,
419  draw->pixelstep[plane], w_sub,
420  draw->hsub[plane], left, right);
421  } else {
422  blend_line16(p, color->comp[plane].u16[comp], alpha >> 1,
423  draw->pixelstep[plane], w_sub,
424  draw->hsub[plane], left, right);
425  }
426  }
427  }
428  }
429 }
430 
431 static void blend_pixel16(uint8_t *dst, unsigned src, unsigned alpha,
432  const uint8_t *mask, int mask_linesize, int l2depth,
433  unsigned w, unsigned h, unsigned shift, unsigned xm0)
434 {
435  unsigned xm, x, y, t = 0;
436  unsigned xmshf = 3 - l2depth;
437  unsigned xmmod = 7 >> l2depth;
438  unsigned mbits = (1 << (1 << l2depth)) - 1;
439  unsigned mmult = 255 / mbits;
440  uint16_t value = AV_RL16(dst);
441 
442  for (y = 0; y < h; y++) {
443  xm = xm0;
444  for (x = 0; x < w; x++) {
445  t += ((mask[xm >> xmshf] >> ((~xm & xmmod) << l2depth)) & mbits)
446  * mmult;
447  xm++;
448  }
449  mask += mask_linesize;
450  }
451  alpha = (t >> shift) * alpha;
452  AV_WL16(dst, ((0x10001 - alpha) * value + alpha * src) >> 16);
453 }
454 
455 static void blend_pixel(uint8_t *dst, unsigned src, unsigned alpha,
456  const uint8_t *mask, int mask_linesize, int l2depth,
457  unsigned w, unsigned h, unsigned shift, unsigned xm0)
458 {
459  unsigned xm, x, y, t = 0;
460  unsigned xmshf = 3 - l2depth;
461  unsigned xmmod = 7 >> l2depth;
462  unsigned mbits = (1 << (1 << l2depth)) - 1;
463  unsigned mmult = 255 / mbits;
464 
465  for (y = 0; y < h; y++) {
466  xm = xm0;
467  for (x = 0; x < w; x++) {
468  t += ((mask[xm >> xmshf] >> ((~xm & xmmod) << l2depth)) & mbits)
469  * mmult;
470  xm++;
471  }
472  mask += mask_linesize;
473  }
474  alpha = (t >> shift) * alpha;
475  *dst = ((0x1010101 - alpha) * *dst + alpha * src) >> 24;
476 }
477 
478 static void blend_line_hv16(uint8_t *dst, int dst_delta,
479  unsigned src, unsigned alpha,
480  const uint8_t *mask, int mask_linesize, int l2depth, int w,
481  unsigned hsub, unsigned vsub,
482  int xm, int left, int right, int hband)
483 {
484  int x;
485 
486  if (left) {
487  blend_pixel16(dst, src, alpha, mask, mask_linesize, l2depth,
488  left, hband, hsub + vsub, xm);
489  dst += dst_delta;
490  xm += left;
491  }
492  for (x = 0; x < w; x++) {
493  blend_pixel16(dst, src, alpha, mask, mask_linesize, l2depth,
494  1 << hsub, hband, hsub + vsub, xm);
495  dst += dst_delta;
496  xm += 1 << hsub;
497  }
498  if (right)
499  blend_pixel16(dst, src, alpha, mask, mask_linesize, l2depth,
500  right, hband, hsub + vsub, xm);
501 }
502 
503 static void blend_line_hv(uint8_t *dst, int dst_delta,
504  unsigned src, unsigned alpha,
505  const uint8_t *mask, int mask_linesize, int l2depth, int w,
506  unsigned hsub, unsigned vsub,
507  int xm, int left, int right, int hband)
508 {
509  int x;
510 
511  if (left) {
512  blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
513  left, hband, hsub + vsub, xm);
514  dst += dst_delta;
515  xm += left;
516  }
517  for (x = 0; x < w; x++) {
518  blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
519  1 << hsub, hband, hsub + vsub, xm);
520  dst += dst_delta;
521  xm += 1 << hsub;
522  }
523  if (right)
524  blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
525  right, hband, hsub + vsub, xm);
526 }
527 
529  uint8_t *dst[], int dst_linesize[], int dst_w, int dst_h,
530  const uint8_t *mask, int mask_linesize, int mask_w, int mask_h,
531  int l2depth, unsigned endianness, int x0, int y0)
532 {
533  unsigned alpha, nb_planes, nb_comp, plane, comp;
534  int xm0, ym0, w_sub, h_sub, x_sub, y_sub, left, right, top, bottom, y;
535  uint8_t *p0, *p;
536  const uint8_t *m;
537 
538  clip_interval(dst_w, &x0, &mask_w, &xm0);
539  clip_interval(dst_h, &y0, &mask_h, &ym0);
540  mask += ym0 * mask_linesize;
541  if (mask_w <= 0 || mask_h <= 0 || !color->rgba[3])
542  return;
543  if (draw->desc->comp[0].depth <= 8) {
544  /* alpha is in the [ 0 ; 0x10203 ] range,
545  alpha * mask is in the [ 0 ; 0x1010101 - 4 ] range */
546  alpha = (0x10307 * color->rgba[3] + 0x3) >> 8;
547  } else {
548  alpha = (0x101 * color->rgba[3] + 0x2) >> 8;
549  }
550  nb_planes = draw->nb_planes - !!(draw->desc->flags & AV_PIX_FMT_FLAG_ALPHA && !(draw->flags & FF_DRAW_PROCESS_ALPHA));
551  nb_planes += !nb_planes;
552  for (plane = 0; plane < nb_planes; plane++) {
553  nb_comp = draw->pixelstep[plane];
554  p0 = pointer_at(draw, dst, dst_linesize, plane, x0, y0);
555  w_sub = mask_w;
556  h_sub = mask_h;
557  x_sub = x0;
558  y_sub = y0;
559  subsampling_bounds(draw->hsub[plane], &x_sub, &w_sub, &left, &right);
560  subsampling_bounds(draw->vsub[plane], &y_sub, &h_sub, &top, &bottom);
561  for (comp = 0; comp < nb_comp; comp++) {
562  const int depth = draw->desc->comp[comp].depth;
563 
564  if (!component_used(draw, plane, comp))
565  continue;
566  p = p0 + comp;
567  m = mask;
568  if (top) {
569  if (depth <= 8) {
570  blend_line_hv(p, draw->pixelstep[plane],
571  color->comp[plane].u8[comp], alpha,
572  m, mask_linesize, l2depth, w_sub,
573  draw->hsub[plane], draw->vsub[plane],
574  xm0, left, right, top);
575  } else {
576  blend_line_hv16(p, draw->pixelstep[plane],
577  color->comp[plane].u16[comp], alpha,
578  m, mask_linesize, l2depth, w_sub,
579  draw->hsub[plane], draw->vsub[plane],
580  xm0, left, right, top);
581  }
582  p += dst_linesize[plane];
583  m += top * mask_linesize;
584  }
585  if (depth <= 8) {
586  for (y = 0; y < h_sub; y++) {
587  blend_line_hv(p, draw->pixelstep[plane],
588  color->comp[plane].u8[comp], alpha,
589  m, mask_linesize, l2depth, w_sub,
590  draw->hsub[plane], draw->vsub[plane],
591  xm0, left, right, 1 << draw->vsub[plane]);
592  p += dst_linesize[plane];
593  m += mask_linesize << draw->vsub[plane];
594  }
595  } else {
596  for (y = 0; y < h_sub; y++) {
597  blend_line_hv16(p, draw->pixelstep[plane],
598  color->comp[plane].u16[comp], alpha,
599  m, mask_linesize, l2depth, w_sub,
600  draw->hsub[plane], draw->vsub[plane],
601  xm0, left, right, 1 << draw->vsub[plane]);
602  p += dst_linesize[plane];
603  m += mask_linesize << draw->vsub[plane];
604  }
605  }
606  if (bottom) {
607  if (depth <= 8) {
608  blend_line_hv(p, draw->pixelstep[plane],
609  color->comp[plane].u8[comp], alpha,
610  m, mask_linesize, l2depth, w_sub,
611  draw->hsub[plane], draw->vsub[plane],
612  xm0, left, right, bottom);
613  } else {
614  blend_line_hv16(p, draw->pixelstep[plane],
615  color->comp[plane].u16[comp], alpha,
616  m, mask_linesize, l2depth, w_sub,
617  draw->hsub[plane], draw->vsub[plane],
618  xm0, left, right, bottom);
619  }
620  }
621  }
622  }
623 }
624 
625 int ff_draw_round_to_sub(FFDrawContext *draw, int sub_dir, int round_dir,
626  int value)
627 {
628  unsigned shift = sub_dir ? draw->vsub_max : draw->hsub_max;
629 
630  if (!shift)
631  return value;
632  if (round_dir >= 0)
633  value += round_dir ? (1 << shift) - 1 : 1 << (shift - 1);
634  return (value >> shift) << shift;
635 }
636 
638 {
639  enum AVPixelFormat i;
640  FFDrawContext draw;
641  AVFilterFormats *fmts = NULL;
642  int ret;
643 
644  for (i = 0; av_pix_fmt_desc_get(i); i++)
645  if (ff_draw_init(&draw, i, flags) >= 0 &&
646  (ret = ff_add_format(&fmts, i)) < 0)
647  return NULL;
648  return fmts;
649 }
static const char *const format[]
Definition: af_aiir.c:456
uint8_t
simple assert() macros that are a bit more flexible than ISO C assert().
#define AV_RL16
Definition: intreadwrite.h:42
Convenience header that includes libavutil's core.
#define flags(name, subs,...)
Definition: cbs_av1.c:561
#define FFMIN(a, b)
Definition: common.h:105
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:58
#define FFMAX(a, b)
Definition: common.h:103
#define HAVE_BIGENDIAN
Definition: config.h:200
#define NULL
Definition: coverity.c:32
static enum AVPixelFormat pix_fmt
static float sub(float src0, float src1)
void ff_blend_rectangle(FFDrawContext *draw, FFDrawColor *color, uint8_t *dst[], int dst_linesize[], int dst_w, int dst_h, int x0, int y0, int w, int h)
Blend a rectangle with an uniform color.
Definition: drawutils.c:351
static void blend_pixel(uint8_t *dst, unsigned src, unsigned alpha, const uint8_t *mask, int mask_linesize, int l2depth, unsigned w, unsigned h, unsigned shift, unsigned xm0)
Definition: drawutils.c:455
void ff_copy_rectangle2(FFDrawContext *draw, uint8_t *dst[], int dst_linesize[], uint8_t *src[], int src_linesize[], int dst_x, int dst_y, int src_x, int src_y, int w, int h)
Copy a rectangle from an image to another.
Definition: drawutils.c:202
int ff_draw_round_to_sub(FFDrawContext *draw, int sub_dir, int round_dir, int value)
Round a dimension according to subsampling.
Definition: drawutils.c:625
static void clip_interval(int wmax, int *x, int *w, int *dx)
Clip interval [x; x+w[ within [0; wmax[.
Definition: drawutils.c:265
#define EXPAND(compn)
static void blend_line16(uint8_t *dst, unsigned src, unsigned alpha, int dx, int w, unsigned hsub, int left, int right)
Definition: drawutils.c:326
int ff_draw_init(FFDrawContext *draw, enum AVPixelFormat format, unsigned flags)
Init a draw context.
Definition: drawutils.c:84
void ff_draw_color(FFDrawContext *draw, FFDrawColor *color, const uint8_t rgba[4])
Prepare a color.
Definition: drawutils.c:137
@ BLUE
Definition: drawutils.c:33
@ ALPHA
Definition: drawutils.c:33
@ GREEN
Definition: drawutils.c:33
@ RED
Definition: drawutils.c:33
static uint8_t * pointer_at(FFDrawContext *draw, uint8_t *data[], int linesize[], int plane, int x, int y)
Definition: drawutils.c:194
static void blend_pixel16(uint8_t *dst, unsigned src, unsigned alpha, const uint8_t *mask, int mask_linesize, int l2depth, unsigned w, unsigned h, unsigned shift, unsigned xm0)
Definition: drawutils.c:431
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
Definition: drawutils.c:35
static void blend_line_hv(uint8_t *dst, int dst_delta, unsigned src, unsigned alpha, const uint8_t *mask, int mask_linesize, int l2depth, int w, unsigned hsub, unsigned vsub, int xm, int left, int right, int hband)
Definition: drawutils.c:503
static void subsampling_bounds(int sub, int *x, int *w, int *start, int *end)
Decompose w pixels starting at x into start + (w starting at x) + end with x and w aligned on multipl...
Definition: drawutils.c:284
static void blend_line_hv16(uint8_t *dst, int dst_delta, unsigned src, unsigned alpha, const uint8_t *mask, int mask_linesize, int l2depth, int w, unsigned hsub, unsigned vsub, int xm, int left, int right, int hband)
Definition: drawutils.c:478
void ff_fill_rectangle(FFDrawContext *draw, FFDrawColor *color, uint8_t *dst[], int dst_linesize[], int dst_x, int dst_y, int w, int h)
Fill a rectangle with an uniform color.
Definition: drawutils.c:224
static void blend_line(uint8_t *dst, unsigned src, unsigned alpha, int dx, int w, unsigned hsub, int left, int right)
Definition: drawutils.c:304
void ff_blend_mask(FFDrawContext *draw, FFDrawColor *color, uint8_t *dst[], int dst_linesize[], int dst_w, int dst_h, const uint8_t *mask, int mask_linesize, int mask_w, int mask_h, int l2depth, unsigned endianness, int x0, int y0)
Blend an alpha mask with an uniform color.
Definition: drawutils.c:528
AVFilterFormats * ff_draw_supported_pixel_formats(unsigned flags)
Return the list of pixel formats supported by the draw functions.
Definition: drawutils.c:637
static int component_used(FFDrawContext *draw, int plane, int comp)
Definition: drawutils.c:296
misc drawing utilities
#define FF_DRAW_PROCESS_ALPHA
Process alpha pixel component.
Definition: drawutils.h:61
static void comp(unsigned char *dst, ptrdiff_t dst_stride, unsigned char *src, ptrdiff_t src_stride, int add)
Definition: eamad.c:85
double value
Definition: eval.c:98
#define MAX_PLANES
Definition: ffv1.h:49
int ff_add_format(AVFilterFormats **avff, int64_t fmt)
Add fmt to the list of media formats contained in *avff.
Definition: formats.c:332
#define AVERROR(e)
Definition: error.h:43
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:200
bool full_range
static const int16_t alpha[]
Definition: ilbcdata.h:55
int i
Definition: input.c:407
#define AV_WL16(p, v)
Definition: intreadwrite.h:412
Various defines for YUV<->RGB conversion.
#define RGB_TO_V_JPEG(r1, g1, b1)
Definition: colorspace.h:118
#define RGB_TO_U_CCIR(r1, g1, b1, shift)
Definition: colorspace.h:102
#define RGB_TO_Y_CCIR(r, g, b)
Definition: colorspace.h:98
#define RGB_TO_Y_JPEG(r, g, b)
Definition: colorspace.h:110
#define RGB_TO_V_CCIR(r1, g1, b1, shift)
Definition: colorspace.h:106
#define RGB_TO_U_JPEG(r1, g1, b1)
Definition: colorspace.h:114
#define FF_PSEUDOPAL
Definition: internal.h:299
const char * desc
Definition: libsvtav1.c:79
uint8_t w
Definition: llviddspenc.c:39
static const uint32_t color[16+AV_CLASS_CATEGORY_NB]
Definition: log.c:92
static const uint16_t mask[17]
Definition: lzw.c:38
Memory handling functions.
const char data[16]
Definition: mxf.c:142
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2573
#define AV_PIX_FMT_FLAG_ALPHA
The pixel format has an alpha channel.
Definition: pixdesc.h:179
#define AV_PIX_FMT_FLAG_RGB
The pixel format contains RGB-like data (as opposed to YUV/grayscale).
Definition: pixdesc.h:148
#define AV_PIX_FMT_FLAG_PLANAR
At least one pixel component is not in the first data plane.
Definition: pixdesc.h:144
#define AV_PIX_FMT_FLAG_BE
Pixel format is big-endian.
Definition: pixdesc.h:128
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
@ AV_PIX_FMT_P010LE
like NV12, with 10bpp per component, data in the high bits, zeros in the low bits,...
Definition: pixfmt.h:284
@ AV_PIX_FMT_RGB24
packed RGB 8:8:8, 24bpp, RGBRGB...
Definition: pixfmt.h:68
@ AV_PIX_FMT_GRAY10LE
Y , 10bpp, little-endian.
Definition: pixfmt.h:298
@ AV_PIX_FMT_GBRP10BE
planar GBR 4:4:4 30bpp, big-endian
Definition: pixfmt.h:172
@ AV_PIX_FMT_GBRPF32BE
IEEE-754 single precision planar GBR 4:4:4, 96bpp, big-endian.
Definition: pixfmt.h:318
@ AV_PIX_FMT_BGR0
packed BGR 8:8:8, 32bpp, BGRXBGRX... X=unused/undefined
Definition: pixfmt.h:240
@ AV_PIX_FMT_GBRP9LE
planar GBR 4:4:4 27bpp, little-endian
Definition: pixfmt.h:171
@ AV_PIX_FMT_P016LE
like NV12, with 16bpp per component, little-endian
Definition: pixfmt.h:300
@ AV_PIX_FMT_P010BE
like NV12, with 10bpp per component, data in the high bits, zeros in the low bits,...
Definition: pixfmt.h:285
@ AV_PIX_FMT_ARGB
packed ARGB 8:8:8:8, 32bpp, ARGBARGB...
Definition: pixfmt.h:92
@ AV_PIX_FMT_GBRP12BE
planar GBR 4:4:4 36bpp, big-endian
Definition: pixfmt.h:254
@ AV_PIX_FMT_GRAY8A
alias for AV_PIX_FMT_YA8
Definition: pixfmt.h:146
@ AV_PIX_FMT_GRAY12LE
Y , 12bpp, little-endian.
Definition: pixfmt.h:296
@ AV_PIX_FMT_GBRAP12BE
planar GBR 4:4:4:4 48bpp, big-endian
Definition: pixfmt.h:287
@ AV_PIX_FMT_BGRA
packed BGRA 8:8:8:8, 32bpp, BGRABGRA...
Definition: pixfmt.h:95
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
Definition: pixfmt.h:74
@ AV_PIX_FMT_BGR48BE
packed RGB 16:16:16, 48bpp, 16B, 16G, 16R, the 2-byte value for each R/G/B component is stored as big...
Definition: pixfmt.h:148
@ AV_PIX_FMT_YA16LE
16 bits gray, 16 bits alpha (little-endian)
Definition: pixfmt.h:213
@ AV_PIX_FMT_GRAY14LE
Y , 14bpp, little-endian.
Definition: pixfmt.h:338
@ AV_PIX_FMT_RGB48BE
packed RGB 16:16:16, 48bpp, 16R, 16G, 16B, the 2-byte value for each R/G/B component is stored as big...
Definition: pixfmt.h:102
@ AV_PIX_FMT_ABGR
packed ABGR 8:8:8:8, 32bpp, ABGRABGR...
Definition: pixfmt.h:94
@ AV_PIX_FMT_P016BE
like NV12, with 16bpp per component, big-endian
Definition: pixfmt.h:301
@ AV_PIX_FMT_YUVJ440P
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range
Definition: pixfmt.h:100
@ AV_PIX_FMT_RGBA64BE
packed RGBA 16:16:16:16, 64bpp, 16R, 16G, 16B, 16A, the 2-byte value for each R/G/B/A component is st...
Definition: pixfmt.h:205
@ AV_PIX_FMT_RGBA64LE
packed RGBA 16:16:16:16, 64bpp, 16R, 16G, 16B, 16A, the 2-byte value for each R/G/B/A component is st...
Definition: pixfmt.h:206
@ AV_PIX_FMT_0BGR
packed BGR 8:8:8, 32bpp, XBGRXBGR... X=unused/undefined
Definition: pixfmt.h:239
@ AV_PIX_FMT_GBRAP16BE
planar GBRA 4:4:4:4 64bpp, big-endian
Definition: pixfmt.h:216
@ AV_PIX_FMT_GBRPF32LE
IEEE-754 single precision planar GBR 4:4:4, 96bpp, little-endian.
Definition: pixfmt.h:319
@ AV_PIX_FMT_RGBA
packed RGBA 8:8:8:8, 32bpp, RGBARGBA...
Definition: pixfmt.h:93
@ AV_PIX_FMT_YUVJ411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
Definition: pixfmt.h:258
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:215
@ AV_PIX_FMT_GBRP12LE
planar GBR 4:4:4 36bpp, little-endian
Definition: pixfmt.h:255
@ AV_PIX_FMT_BGRA64BE
packed RGBA 16:16:16:16, 64bpp, 16B, 16G, 16R, 16A, the 2-byte value for each R/G/B/A component is st...
Definition: pixfmt.h:207
@ AV_PIX_FMT_GBRP16BE
planar GBR 4:4:4 48bpp, big-endian
Definition: pixfmt.h:174
@ AV_PIX_FMT_GBRAP12LE
planar GBR 4:4:4:4 48bpp, little-endian
Definition: pixfmt.h:288
@ AV_PIX_FMT_GBRP9BE
planar GBR 4:4:4 27bpp, big-endian
Definition: pixfmt.h:170
@ AV_PIX_FMT_RGB48LE
packed RGB 16:16:16, 48bpp, 16R, 16G, 16B, the 2-byte value for each R/G/B component is stored as lit...
Definition: pixfmt.h:103
@ AV_PIX_FMT_GBRAPF32BE
IEEE-754 single precision planar GBRA 4:4:4:4, 128bpp, big-endian.
Definition: pixfmt.h:320
@ AV_PIX_FMT_YUVJ422P
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:79
@ AV_PIX_FMT_GBRAPF32LE
IEEE-754 single precision planar GBRA 4:4:4:4, 128bpp, little-endian.
Definition: pixfmt.h:321
@ AV_PIX_FMT_BGR48LE
packed RGB 16:16:16, 48bpp, 16B, 16G, 16R, the 2-byte value for each R/G/B component is stored as lit...
Definition: pixfmt.h:149
@ AV_PIX_FMT_GBRP14LE
planar GBR 4:4:4 42bpp, little-endian
Definition: pixfmt.h:257
@ AV_PIX_FMT_RGB0
packed RGB 8:8:8, 32bpp, RGBXRGBX... X=unused/undefined
Definition: pixfmt.h:238
@ AV_PIX_FMT_GRAY16LE
Y , 16bpp, little-endian.
Definition: pixfmt.h:98
@ AV_PIX_FMT_GBRP10LE
planar GBR 4:4:4 30bpp, little-endian
Definition: pixfmt.h:173
@ AV_PIX_FMT_GBRAP10BE
planar GBR 4:4:4:4 40bpp, big-endian
Definition: pixfmt.h:290
@ AV_PIX_FMT_BGRA64LE
packed RGBA 16:16:16:16, 64bpp, 16B, 16G, 16R, 16A, the 2-byte value for each R/G/B/A component is st...
Definition: pixfmt.h:208
@ AV_PIX_FMT_GBRAP10LE
planar GBR 4:4:4:4 40bpp, little-endian
Definition: pixfmt.h:291
@ AV_PIX_FMT_BGR24
packed RGB 8:8:8, 24bpp, BGRBGR...
Definition: pixfmt.h:69
@ AV_PIX_FMT_GRAY9LE
Y , 9bpp, little-endian.
Definition: pixfmt.h:316
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:168
@ AV_PIX_FMT_GBRAP16LE
planar GBRA 4:4:4:4 64bpp, little-endian
Definition: pixfmt.h:217
@ AV_PIX_FMT_0RGB
packed RGB 8:8:8, 32bpp, XRGBXRGB... X=unused/undefined
Definition: pixfmt.h:237
@ AV_PIX_FMT_GBRP14BE
planar GBR 4:4:4 42bpp, big-endian
Definition: pixfmt.h:256
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:80
@ AV_PIX_FMT_GBRP16LE
planar GBR 4:4:4 48bpp, little-endian
Definition: pixfmt.h:175
@ AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:78
#define av_bswap16
Definition: bswap.h:31
static int shift(int a, int b)
Definition: sonic.c:82
int depth
Number of bits in the component.
Definition: pixdesc.h:58
A list of supported formats for one end of a filter link.
Definition: formats.h:65
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
const char * name
Definition: pixdesc.h:82
AVComponentDescriptor comp[4]
Parameters that describe how pixels are packed.
Definition: pixdesc.h:117
uint64_t flags
Combination of AV_PIX_FMT_FLAG_...
Definition: pixdesc.h:106
union FFDrawColor::@204 comp[MAX_PLANES]
uint16_t u16[8]
Definition: drawutils.h:53
uint8_t u8[16]
Definition: drawutils.h:54
unsigned nb_planes
Definition: drawutils.h:38
unsigned flags
Definition: drawutils.h:46
const struct AVPixFmtDescriptor * desc
Definition: drawutils.h:36
uint8_t vsub[MAX_PLANES]
Definition: drawutils.h:42
enum AVPixelFormat format
Definition: drawutils.h:37
int full_range
Definition: drawutils.h:45
uint8_t hsub[MAX_PLANES]
Definition: drawutils.h:41
uint8_t vsub_max
Definition: drawutils.h:44
uint8_t comp_mask[MAX_PLANES]
Definition: drawutils.h:40
int pixelstep[MAX_PLANES]
Definition: drawutils.h:39
uint8_t hsub_max
Definition: drawutils.h:43
#define av_log(a,...)
#define src
Definition: vp8dsp.c:255
static void hsub(htype *dst, const htype *src, int bins)
Definition: vf_median.c:75
static double c[64]