avfilter/drawutils: support gbrap10 too
[ffmpeg.git] / libavfilter / drawutils.c
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
35 int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
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:
44     case AV_PIX_FMT_RGBA64BE:
45     case AV_PIX_FMT_RGBA64LE:
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:
51     case AV_PIX_FMT_BGRA64BE:
52     case AV_PIX_FMT_BGRA64LE:
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:
58     case AV_PIX_FMT_GBRP10LE:
59     case AV_PIX_FMT_GBRP10BE:
60     case AV_PIX_FMT_GBRP12LE:
61     case AV_PIX_FMT_GBRP12BE:
62     case AV_PIX_FMT_GBRP14LE:
63     case AV_PIX_FMT_GBRP14BE:
64     case AV_PIX_FMT_GBRP16LE:
65     case AV_PIX_FMT_GBRP16BE:
66     case AV_PIX_FMT_GBRAP:
67     case AV_PIX_FMT_GBRAP10LE:
68     case AV_PIX_FMT_GBRAP10BE:
69     case AV_PIX_FMT_GBRAP12LE:
70     case AV_PIX_FMT_GBRAP12BE:
71     case AV_PIX_FMT_GBRAP16LE:
72     case AV_PIX_FMT_GBRAP16BE:
73     case AV_PIX_FMT_GBRP:  rgba_map[GREEN] = 0; rgba_map[BLUE ] = 1; rgba_map[RED  ] = 2; rgba_map[ALPHA] = 3; break;
74     default:                    /* unsupported */
75         return AVERROR(EINVAL);
76     }
77     return 0;
78 }
79
80 int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w, uint8_t dst_color[4],
81                             enum AVPixelFormat pix_fmt, uint8_t rgba_color[4],
82                             int *is_packed_rgba, uint8_t rgba_map_ptr[4])
83 {
84     uint8_t rgba_map[4] = {0};
85     int i;
86     const AVPixFmtDescriptor *pix_desc = av_pix_fmt_desc_get(pix_fmt);
87     int hsub;
88
89     av_assert0(pix_desc);
90
91     hsub = pix_desc->log2_chroma_w;
92
93     *is_packed_rgba = ff_fill_rgba_map(rgba_map, pix_fmt) >= 0;
94
95     if (*is_packed_rgba) {
96         pixel_step[0] = (av_get_bits_per_pixel(pix_desc))>>3;
97         for (i = 0; i < 4; i++)
98             dst_color[rgba_map[i]] = rgba_color[i];
99
100         line[0] = av_malloc_array(w, pixel_step[0]);
101         if (!line[0])
102             return AVERROR(ENOMEM);
103         for (i = 0; i < w; i++)
104             memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]);
105         if (rgba_map_ptr)
106             memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4);
107     } else {
108         int plane;
109
110         dst_color[0] = RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]);
111         dst_color[1] = RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
112         dst_color[2] = RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
113         dst_color[3] = rgba_color[3];
114
115         for (plane = 0; plane < 4; plane++) {
116             int line_size;
117             int hsub1 = (plane == 1 || plane == 2) ? hsub : 0;
118
119             pixel_step[plane] = 1;
120             line_size = AV_CEIL_RSHIFT(w, hsub1) * pixel_step[plane];
121             line[plane] = av_malloc(line_size);
122             if (!line[plane]) {
123                 while(plane && line[plane-1])
124                     av_freep(&line[--plane]);
125                 return AVERROR(ENOMEM);
126             }
127             memset(line[plane], dst_color[plane], line_size);
128         }
129     }
130
131     return 0;
132 }
133
134 void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4],
135                        uint8_t *src[4], int pixelstep[4],
136                        int hsub, int vsub, int x, int y, int w, int h)
137 {
138     int i, plane;
139     uint8_t *p;
140
141     for (plane = 0; plane < 4 && dst[plane]; plane++) {
142         int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
143         int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
144         int width  = AV_CEIL_RSHIFT(w, hsub1);
145         int height = AV_CEIL_RSHIFT(h, vsub1);
146
147         p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
148         for (i = 0; i < height; i++) {
149             memcpy(p + (x >> hsub1) * pixelstep[plane],
150                    src[plane], width * pixelstep[plane]);
151             p += dst_linesize[plane];
152         }
153     }
154 }
155
156 void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4],
157                        uint8_t *src[4], int src_linesize[4], int pixelstep[4],
158                        int hsub, int vsub, int x, int y, int y2, int w, int h)
159 {
160     int i, plane;
161     uint8_t *p;
162
163     for (plane = 0; plane < 4 && dst[plane]; plane++) {
164         int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
165         int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
166         int width  = AV_CEIL_RSHIFT(w, hsub1);
167         int height = AV_CEIL_RSHIFT(h, vsub1);
168
169         p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
170         for (i = 0; i < height; i++) {
171             memcpy(p + (x >> hsub1) * pixelstep[plane],
172                    src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), width * pixelstep[plane]);
173             p += dst_linesize[plane];
174         }
175     }
176 }
177
178 int ff_draw_init(FFDrawContext *draw, enum AVPixelFormat format, unsigned flags)
179 {
180     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(format);
181     const AVComponentDescriptor *c;
182     unsigned i, nb_planes = 0;
183     int pixelstep[MAX_PLANES] = { 0 };
184
185     if (!desc || !desc->name)
186         return AVERROR(EINVAL);
187     if (desc->flags & ~(AV_PIX_FMT_FLAG_PLANAR | AV_PIX_FMT_FLAG_RGB | AV_PIX_FMT_FLAG_PSEUDOPAL | AV_PIX_FMT_FLAG_ALPHA))
188         return AVERROR(ENOSYS);
189     if (format == AV_PIX_FMT_P010LE || format == AV_PIX_FMT_P010BE)
190         return AVERROR(ENOSYS);
191     for (i = 0; i < desc->nb_components; i++) {
192         c = &desc->comp[i];
193         /* for now, only 8-16 bits formats */
194         if (c->depth < 8 || c->depth > 16)
195             return AVERROR(ENOSYS);
196         if (desc->flags & AV_PIX_FMT_FLAG_BE)
197             return AVERROR(ENOSYS);
198         if (c->plane >= MAX_PLANES)
199             return AVERROR(ENOSYS);
200         /* strange interleaving */
201         if (pixelstep[c->plane] != 0 &&
202             pixelstep[c->plane] != c->step)
203             return AVERROR(ENOSYS);
204         if (pixelstep[c->plane] == 6 &&
205             c->depth == 16)
206             return AVERROR(ENOSYS);
207         pixelstep[c->plane] = c->step;
208         if (pixelstep[c->plane] >= 8)
209             return AVERROR(ENOSYS);
210         nb_planes = FFMAX(nb_planes, c->plane + 1);
211     }
212     memset(draw, 0, sizeof(*draw));
213     draw->desc      = desc;
214     draw->format    = format;
215     draw->nb_planes = nb_planes;
216     draw->flags     = flags;
217     memcpy(draw->pixelstep, pixelstep, sizeof(draw->pixelstep));
218     draw->hsub[1] = draw->hsub[2] = draw->hsub_max = desc->log2_chroma_w;
219     draw->vsub[1] = draw->vsub[2] = draw->vsub_max = desc->log2_chroma_h;
220     for (i = 0; i < (desc->nb_components - !!(desc->flags & AV_PIX_FMT_FLAG_ALPHA && !(flags & FF_DRAW_PROCESS_ALPHA))); i++)
221         draw->comp_mask[desc->comp[i].plane] |=
222             1 << desc->comp[i].offset;
223     return 0;
224 }
225
226 void ff_draw_color(FFDrawContext *draw, FFDrawColor *color, const uint8_t rgba[4])
227 {
228     unsigned i;
229     uint8_t rgba_map[4];
230
231     if (rgba != color->rgba)
232         memcpy(color->rgba, rgba, sizeof(color->rgba));
233     if ((draw->desc->flags & AV_PIX_FMT_FLAG_RGB) &&
234         ff_fill_rgba_map(rgba_map, draw->format) >= 0) {
235         if (draw->nb_planes == 1) {
236             for (i = 0; i < 4; i++) {
237                 color->comp[0].u8[rgba_map[i]] = rgba[i];
238                 if (draw->desc->comp[rgba_map[i]].depth > 8) {
239                     color->comp[0].u16[rgba_map[i]] = color->comp[0].u8[rgba_map[i]] << 8;
240                 }
241             }
242         } else {
243             for (i = 0; i < 4; i++) {
244                 color->comp[rgba_map[i]].u8[0] = rgba[i];
245                 if (draw->desc->comp[rgba_map[i]].depth > 8)
246                     color->comp[rgba_map[i]].u16[0] = color->comp[rgba_map[i]].u8[0] << (draw->desc->comp[rgba_map[i]].depth - 8);
247             }
248         }
249     } else if (draw->nb_planes >= 2) {
250         /* assume YUV */
251         const AVPixFmtDescriptor *desc = draw->desc;
252         color->comp[desc->comp[0].plane].u8[desc->comp[0].offset] = RGB_TO_Y_CCIR(rgba[0], rgba[1], rgba[2]);
253         color->comp[desc->comp[1].plane].u8[desc->comp[1].offset] = RGB_TO_U_CCIR(rgba[0], rgba[1], rgba[2], 0);
254         color->comp[desc->comp[2].plane].u8[desc->comp[2].offset] = RGB_TO_V_CCIR(rgba[0], rgba[1], rgba[2], 0);
255         color->comp[3].u8[0] = rgba[3];
256 #define EXPAND(compn) \
257         if (desc->comp[compn].depth > 8) \
258             color->comp[desc->comp[compn].plane].u16[desc->comp[compn].offset] = \
259             color->comp[desc->comp[compn].plane].u8[desc->comp[compn].offset] << \
260                 (draw->desc->comp[compn].depth + draw->desc->comp[compn].shift - 8)
261         EXPAND(3);
262         EXPAND(2);
263         EXPAND(1);
264         EXPAND(0);
265     } else if (draw->format == AV_PIX_FMT_GRAY8 || draw->format == AV_PIX_FMT_GRAY8A) {
266         color->comp[0].u8[0] = RGB_TO_Y_CCIR(rgba[0], rgba[1], rgba[2]);
267         color->comp[1].u8[0] = rgba[3];
268     } else if (draw->format == AV_PIX_FMT_GRAY16LE || draw->format == AV_PIX_FMT_YA16LE) {
269         color->comp[0].u8[0] = RGB_TO_Y_CCIR(rgba[0], rgba[1], rgba[2]);
270         color->comp[0].u16[0] = color->comp[0].u8[0] << 8;
271         color->comp[1].u8[0] = rgba[3];
272         color->comp[1].u16[0] = color->comp[1].u8[0] << 8;
273     } else {
274         av_log(NULL, AV_LOG_WARNING,
275                "Color conversion not implemented for %s\n", draw->desc->name);
276         memset(color, 128, sizeof(*color));
277     }
278 }
279
280 static uint8_t *pointer_at(FFDrawContext *draw, uint8_t *data[], int linesize[],
281                            int plane, int x, int y)
282 {
283     return data[plane] +
284            (y >> draw->vsub[plane]) * linesize[plane] +
285            (x >> draw->hsub[plane]) * draw->pixelstep[plane];
286 }
287
288 void ff_copy_rectangle2(FFDrawContext *draw,
289                         uint8_t *dst[], int dst_linesize[],
290                         uint8_t *src[], int src_linesize[],
291                         int dst_x, int dst_y, int src_x, int src_y,
292                         int w, int h)
293 {
294     int plane, y, wp, hp;
295     uint8_t *p, *q;
296
297     for (plane = 0; plane < draw->nb_planes; plane++) {
298         p = pointer_at(draw, src, src_linesize, plane, src_x, src_y);
299         q = pointer_at(draw, dst, dst_linesize, plane, dst_x, dst_y);
300         wp = AV_CEIL_RSHIFT(w, draw->hsub[plane]) * draw->pixelstep[plane];
301         hp = AV_CEIL_RSHIFT(h, draw->vsub[plane]);
302         for (y = 0; y < hp; y++) {
303             memcpy(q, p, wp);
304             p += src_linesize[plane];
305             q += dst_linesize[plane];
306         }
307     }
308 }
309
310 void ff_fill_rectangle(FFDrawContext *draw, FFDrawColor *color,
311                        uint8_t *dst[], int dst_linesize[],
312                        int dst_x, int dst_y, int w, int h)
313 {
314     int plane, x, y, wp, hp;
315     uint8_t *p0, *p;
316     FFDrawColor color_tmp = *color;
317
318     for (plane = 0; plane < draw->nb_planes; plane++) {
319         p0 = pointer_at(draw, dst, dst_linesize, plane, dst_x, dst_y);
320         wp = AV_CEIL_RSHIFT(w, draw->hsub[plane]);
321         hp = AV_CEIL_RSHIFT(h, draw->vsub[plane]);
322         if (!hp)
323             return;
324         p = p0;
325
326         if (HAVE_BIGENDIAN && draw->desc->comp[0].depth > 8) {
327             for (x = 0; 2*x < draw->pixelstep[plane]; x++)
328                 color_tmp.comp[plane].u16[x] = av_bswap16(color_tmp.comp[plane].u16[x]);
329         }
330
331         /* copy first line from color */
332         for (x = 0; x < wp; x++) {
333             memcpy(p, color_tmp.comp[plane].u8, draw->pixelstep[plane]);
334             p += draw->pixelstep[plane];
335         }
336         wp *= draw->pixelstep[plane];
337         /* copy next lines from first line */
338         p = p0 + dst_linesize[plane];
339         for (y = 1; y < hp; y++) {
340             memcpy(p, p0, wp);
341             p += dst_linesize[plane];
342         }
343     }
344 }
345
346 /**
347  * Clip interval [x; x+w[ within [0; wmax[.
348  * The resulting w may be negative if the final interval is empty.
349  * dx, if not null, return the difference between in and out value of x.
350  */
351 static void clip_interval(int wmax, int *x, int *w, int *dx)
352 {
353     if (dx)
354         *dx = 0;
355     if (*x < 0) {
356         if (dx)
357             *dx = -*x;
358         *w += *x;
359         *x = 0;
360     }
361     if (*x + *w > wmax)
362         *w = wmax - *x;
363 }
364
365 /**
366  * Decompose w pixels starting at x
367  * into start + (w starting at x) + end
368  * with x and w aligned on multiples of 1<<sub.
369  */
370 static void subsampling_bounds(int sub, int *x, int *w, int *start, int *end)
371 {
372     int mask = (1 << sub) - 1;
373
374     *start = (-*x) & mask;
375     *x += *start;
376     *start = FFMIN(*start, *w);
377     *w -= *start;
378     *end = *w & mask;
379     *w >>= sub;
380 }
381
382 static int component_used(FFDrawContext *draw, int plane, int comp)
383 {
384     return (draw->comp_mask[plane] >> comp) & 1;
385 }
386
387 /* If alpha is in the [ 0 ; 0x1010101 ] range,
388    then alpha * value is in the [ 0 ; 0xFFFFFFFF ] range,
389    and >> 24 gives a correct rounding. */
390 static void blend_line(uint8_t *dst, unsigned src, unsigned alpha,
391                        int dx, int w, unsigned hsub, int left, int right)
392 {
393     unsigned asrc = alpha * src;
394     unsigned tau = 0x1010101 - alpha;
395     int x;
396
397     if (left) {
398         unsigned suba = (left * alpha) >> hsub;
399         *dst = (*dst * (0x1010101 - suba) + src * suba) >> 24;
400         dst += dx;
401     }
402     for (x = 0; x < w; x++) {
403         *dst = (*dst * tau + asrc) >> 24;
404         dst += dx;
405     }
406     if (right) {
407         unsigned suba = (right * alpha) >> hsub;
408         *dst = (*dst * (0x1010101 - suba) + src * suba) >> 24;
409     }
410 }
411
412 static void blend_line16(uint8_t *dst, unsigned src, unsigned alpha,
413                          int dx, int w, unsigned hsub, int left, int right)
414 {
415     unsigned asrc = alpha * src;
416     unsigned tau = 0x10001 - alpha;
417     int x;
418
419     if (left) {
420         unsigned suba = (left * alpha) >> hsub;
421         uint16_t value = AV_RL16(dst);
422         AV_WL16(dst, (value * (0x10001 - suba) + src * suba) >> 16);
423         dst += dx;
424     }
425     for (x = 0; x < w; x++) {
426         uint16_t value = AV_RL16(dst);
427         AV_WL16(dst, (value * tau + asrc) >> 16);
428         dst += dx;
429     }
430     if (right) {
431         unsigned suba = (right * alpha) >> hsub;
432         uint16_t value = AV_RL16(dst);
433         AV_WL16(dst, (value * (0x10001 - suba) + src * suba) >> 16);
434     }
435 }
436
437 void ff_blend_rectangle(FFDrawContext *draw, FFDrawColor *color,
438                         uint8_t *dst[], int dst_linesize[],
439                         int dst_w, int dst_h,
440                         int x0, int y0, int w, int h)
441 {
442     unsigned alpha, nb_planes, nb_comp, plane, comp;
443     int w_sub, h_sub, x_sub, y_sub, left, right, top, bottom, y;
444     uint8_t *p0, *p;
445
446     /* TODO optimize if alpha = 0xFF */
447     clip_interval(dst_w, &x0, &w, NULL);
448     clip_interval(dst_h, &y0, &h, NULL);
449     if (w <= 0 || h <= 0 || !color->rgba[3])
450         return;
451     if (draw->desc->comp[0].depth <= 8) {
452         /* 0x10203 * alpha + 2 is in the [ 2 ; 0x1010101 - 2 ] range */
453         alpha = 0x10203 * color->rgba[3] + 0x2;
454     } else {
455         /* 0x101 * alpha is in the [ 2 ; 0x1001] range */
456         alpha = 0x101 * color->rgba[3] + 0x2;
457     }
458     nb_planes = draw->nb_planes - !!(draw->desc->flags & AV_PIX_FMT_FLAG_ALPHA && !(draw->flags & FF_DRAW_PROCESS_ALPHA));
459     nb_planes += !nb_planes;
460     for (plane = 0; plane < nb_planes; plane++) {
461         nb_comp = draw->pixelstep[plane];
462         p0 = pointer_at(draw, dst, dst_linesize, plane, x0, y0);
463         w_sub = w;
464         h_sub = h;
465         x_sub = x0;
466         y_sub = y0;
467         subsampling_bounds(draw->hsub[plane], &x_sub, &w_sub, &left, &right);
468         subsampling_bounds(draw->vsub[plane], &y_sub, &h_sub, &top, &bottom);
469         for (comp = 0; comp < nb_comp; comp++) {
470             const int depth = draw->desc->comp[comp].depth;
471
472             if (!component_used(draw, plane, comp))
473                 continue;
474             p = p0 + comp;
475             if (top) {
476                 if (depth <= 8) {
477                     blend_line(p, color->comp[plane].u8[comp], alpha >> 1,
478                                draw->pixelstep[plane], w_sub,
479                                draw->hsub[plane], left, right);
480                 } else {
481                     blend_line16(p, color->comp[plane].u16[comp], alpha >> 1,
482                                  draw->pixelstep[plane], w_sub,
483                                  draw->hsub[plane], left, right);
484                 }
485                 p += dst_linesize[plane];
486             }
487             if (depth <= 8) {
488                 for (y = 0; y < h_sub; y++) {
489                     blend_line(p, color->comp[plane].u8[comp], alpha,
490                                draw->pixelstep[plane], w_sub,
491                                draw->hsub[plane], left, right);
492                     p += dst_linesize[plane];
493                 }
494             } else {
495                 for (y = 0; y < h_sub; y++) {
496                     blend_line16(p, color->comp[plane].u16[comp], alpha,
497                                  draw->pixelstep[plane], w_sub,
498                                  draw->hsub[plane], left, right);
499                     p += dst_linesize[plane];
500                 }
501             }
502             if (bottom) {
503                 if (depth <= 8) {
504                     blend_line(p, color->comp[plane].u8[comp], alpha >> 1,
505                                draw->pixelstep[plane], w_sub,
506                                draw->hsub[plane], left, right);
507                 } else {
508                     blend_line16(p, color->comp[plane].u16[comp], alpha >> 1,
509                                  draw->pixelstep[plane], w_sub,
510                                  draw->hsub[plane], left, right);
511                 }
512             }
513         }
514     }
515 }
516
517 static void blend_pixel16(uint8_t *dst, unsigned src, unsigned alpha,
518                           const uint8_t *mask, int mask_linesize, int l2depth,
519                           unsigned w, unsigned h, unsigned shift, unsigned xm0)
520 {
521     unsigned xm, x, y, t = 0;
522     unsigned xmshf = 3 - l2depth;
523     unsigned xmmod = 7 >> l2depth;
524     unsigned mbits = (1 << (1 << l2depth)) - 1;
525     unsigned mmult = 255 / mbits;
526     uint16_t value = AV_RL16(dst);
527
528     for (y = 0; y < h; y++) {
529         xm = xm0;
530         for (x = 0; x < w; x++) {
531             t += ((mask[xm >> xmshf] >> ((~xm & xmmod) << l2depth)) & mbits)
532                  * mmult;
533             xm++;
534         }
535         mask += mask_linesize;
536     }
537     alpha = (t >> shift) * alpha;
538     AV_WL16(dst, ((0x10001 - alpha) * value + alpha * src) >> 16);
539 }
540
541 static void blend_pixel(uint8_t *dst, unsigned src, unsigned alpha,
542                         const uint8_t *mask, int mask_linesize, int l2depth,
543                         unsigned w, unsigned h, unsigned shift, unsigned xm0)
544 {
545     unsigned xm, x, y, t = 0;
546     unsigned xmshf = 3 - l2depth;
547     unsigned xmmod = 7 >> l2depth;
548     unsigned mbits = (1 << (1 << l2depth)) - 1;
549     unsigned mmult = 255 / mbits;
550
551     for (y = 0; y < h; y++) {
552         xm = xm0;
553         for (x = 0; x < w; x++) {
554             t += ((mask[xm >> xmshf] >> ((~xm & xmmod) << l2depth)) & mbits)
555                  * mmult;
556             xm++;
557         }
558         mask += mask_linesize;
559     }
560     alpha = (t >> shift) * alpha;
561     *dst = ((0x1010101 - alpha) * *dst + alpha * src) >> 24;
562 }
563
564 static void blend_line_hv16(uint8_t *dst, int dst_delta,
565                             unsigned src, unsigned alpha,
566                             const uint8_t *mask, int mask_linesize, int l2depth, int w,
567                             unsigned hsub, unsigned vsub,
568                             int xm, int left, int right, int hband)
569 {
570     int x;
571
572     if (left) {
573         blend_pixel16(dst, src, alpha, mask, mask_linesize, l2depth,
574                       left, hband, hsub + vsub, xm);
575         dst += dst_delta;
576         xm += left;
577     }
578     for (x = 0; x < w; x++) {
579         blend_pixel16(dst, src, alpha, mask, mask_linesize, l2depth,
580                       1 << hsub, hband, hsub + vsub, xm);
581         dst += dst_delta;
582         xm += 1 << hsub;
583     }
584     if (right)
585         blend_pixel16(dst, src, alpha, mask, mask_linesize, l2depth,
586                       right, hband, hsub + vsub, xm);
587 }
588
589 static void blend_line_hv(uint8_t *dst, int dst_delta,
590                           unsigned src, unsigned alpha,
591                           const uint8_t *mask, int mask_linesize, int l2depth, int w,
592                           unsigned hsub, unsigned vsub,
593                           int xm, int left, int right, int hband)
594 {
595     int x;
596
597     if (left) {
598         blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
599                     left, hband, hsub + vsub, xm);
600         dst += dst_delta;
601         xm += left;
602     }
603     for (x = 0; x < w; x++) {
604         blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
605                     1 << hsub, hband, hsub + vsub, xm);
606         dst += dst_delta;
607         xm += 1 << hsub;
608     }
609     if (right)
610         blend_pixel(dst, src, alpha, mask, mask_linesize, l2depth,
611                     right, hband, hsub + vsub, xm);
612 }
613
614 void ff_blend_mask(FFDrawContext *draw, FFDrawColor *color,
615                    uint8_t *dst[], int dst_linesize[], int dst_w, int dst_h,
616                    const uint8_t *mask,  int mask_linesize, int mask_w, int mask_h,
617                    int l2depth, unsigned endianness, int x0, int y0)
618 {
619     unsigned alpha, nb_planes, nb_comp, plane, comp;
620     int xm0, ym0, w_sub, h_sub, x_sub, y_sub, left, right, top, bottom, y;
621     uint8_t *p0, *p;
622     const uint8_t *m;
623
624     clip_interval(dst_w, &x0, &mask_w, &xm0);
625     clip_interval(dst_h, &y0, &mask_h, &ym0);
626     mask += ym0 * mask_linesize;
627     if (mask_w <= 0 || mask_h <= 0 || !color->rgba[3])
628         return;
629     if (draw->desc->comp[0].depth <= 8) {
630         /* alpha is in the [ 0 ; 0x10203 ] range,
631            alpha * mask is in the [ 0 ; 0x1010101 - 4 ] range */
632         alpha = (0x10307 * color->rgba[3] + 0x3) >> 8;
633     } else {
634         alpha = (0x101 * color->rgba[3] + 0x2) >> 8;
635     }
636     nb_planes = draw->nb_planes - !!(draw->desc->flags & AV_PIX_FMT_FLAG_ALPHA && !(draw->flags & FF_DRAW_PROCESS_ALPHA));
637     nb_planes += !nb_planes;
638     for (plane = 0; plane < nb_planes; plane++) {
639         nb_comp = draw->pixelstep[plane];
640         p0 = pointer_at(draw, dst, dst_linesize, plane, x0, y0);
641         w_sub = mask_w;
642         h_sub = mask_h;
643         x_sub = x0;
644         y_sub = y0;
645         subsampling_bounds(draw->hsub[plane], &x_sub, &w_sub, &left, &right);
646         subsampling_bounds(draw->vsub[plane], &y_sub, &h_sub, &top, &bottom);
647         for (comp = 0; comp < nb_comp; comp++) {
648             const int depth = draw->desc->comp[comp].depth;
649
650             if (!component_used(draw, plane, comp))
651                 continue;
652             p = p0 + comp;
653             m = mask;
654             if (top) {
655                 if (depth <= 8) {
656                     blend_line_hv(p, draw->pixelstep[plane],
657                                   color->comp[plane].u8[comp], alpha,
658                                   m, mask_linesize, l2depth, w_sub,
659                                   draw->hsub[plane], draw->vsub[plane],
660                                   xm0, left, right, top);
661                 } else {
662                     blend_line_hv16(p, draw->pixelstep[plane],
663                                     color->comp[plane].u16[comp], alpha,
664                                     m, mask_linesize, l2depth, w_sub,
665                                     draw->hsub[plane], draw->vsub[plane],
666                                     xm0, left, right, top);
667                 }
668                 p += dst_linesize[plane];
669                 m += top * mask_linesize;
670             }
671             if (depth <= 8) {
672                 for (y = 0; y < h_sub; y++) {
673                     blend_line_hv(p, draw->pixelstep[plane],
674                                   color->comp[plane].u8[comp], alpha,
675                                   m, mask_linesize, l2depth, w_sub,
676                                   draw->hsub[plane], draw->vsub[plane],
677                                   xm0, left, right, 1 << draw->vsub[plane]);
678                     p += dst_linesize[plane];
679                     m += mask_linesize << draw->vsub[plane];
680                 }
681             } else {
682                 for (y = 0; y < h_sub; y++) {
683                     blend_line_hv16(p, draw->pixelstep[plane],
684                                     color->comp[plane].u16[comp], alpha,
685                                     m, mask_linesize, l2depth, w_sub,
686                                     draw->hsub[plane], draw->vsub[plane],
687                                     xm0, left, right, 1 << draw->vsub[plane]);
688                     p += dst_linesize[plane];
689                     m += mask_linesize << draw->vsub[plane];
690                 }
691             }
692             if (bottom) {
693                 if (depth <= 8) {
694                     blend_line_hv(p, draw->pixelstep[plane],
695                                   color->comp[plane].u8[comp], alpha,
696                                   m, mask_linesize, l2depth, w_sub,
697                                   draw->hsub[plane], draw->vsub[plane],
698                                   xm0, left, right, bottom);
699                 } else {
700                     blend_line_hv16(p, draw->pixelstep[plane],
701                                     color->comp[plane].u16[comp], alpha,
702                                     m, mask_linesize, l2depth, w_sub,
703                                     draw->hsub[plane], draw->vsub[plane],
704                                     xm0, left, right, bottom);
705                 }
706             }
707         }
708     }
709 }
710
711 int ff_draw_round_to_sub(FFDrawContext *draw, int sub_dir, int round_dir,
712                          int value)
713 {
714     unsigned shift = sub_dir ? draw->vsub_max : draw->hsub_max;
715
716     if (!shift)
717         return value;
718     if (round_dir >= 0)
719         value += round_dir ? (1 << shift) - 1 : 1 << (shift - 1);
720     return (value >> shift) << shift;
721 }
722
723 AVFilterFormats *ff_draw_supported_pixel_formats(unsigned flags)
724 {
725     enum AVPixelFormat i;
726     FFDrawContext draw;
727     AVFilterFormats *fmts = NULL;
728     int ret;
729
730     for (i = 0; av_pix_fmt_desc_get(i); i++)
731         if (ff_draw_init(&draw, i, flags) >= 0 &&
732             (ret = ff_add_format(&fmts, i)) < 0)
733             return NULL;
734     return fmts;
735 }