2 * This file is part of FFmpeg.
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 #include "libavutil/avutil.h"
20 #include "libavutil/colorspace.h"
21 #include "libavutil/pixdesc.h"
22 #include "drawutils.h"
24 enum { RED = 0, GREEN, BLUE, ALPHA };
26 int ff_fill_rgba_map(uint8_t *rgba_map, enum PixelFormat pix_fmt)
29 case PIX_FMT_ARGB: rgba_map[ALPHA] = 0; rgba_map[RED ] = 1; rgba_map[GREEN] = 2; rgba_map[BLUE ] = 3; break;
30 case PIX_FMT_ABGR: rgba_map[ALPHA] = 0; rgba_map[BLUE ] = 1; rgba_map[GREEN] = 2; rgba_map[RED ] = 3; break;
32 case PIX_FMT_RGB24: rgba_map[RED ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break;
34 case PIX_FMT_BGR24: rgba_map[BLUE ] = 0; rgba_map[GREEN] = 1; rgba_map[RED ] = 2; rgba_map[ALPHA] = 3; break;
35 default: /* unsupported */
36 return AVERROR(EINVAL);
41 int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w, uint8_t dst_color[4],
42 enum PixelFormat pix_fmt, uint8_t rgba_color[4],
43 int *is_packed_rgba, uint8_t rgba_map_ptr[4])
45 uint8_t rgba_map[4] = {0};
47 const AVPixFmtDescriptor *pix_desc = &av_pix_fmt_descriptors[pix_fmt];
48 int hsub = pix_desc->log2_chroma_w;
50 *is_packed_rgba = ff_fill_rgba_map(rgba_map, pix_fmt) >= 0;
52 if (*is_packed_rgba) {
53 pixel_step[0] = (av_get_bits_per_pixel(pix_desc))>>3;
54 for (i = 0; i < 4; i++)
55 dst_color[rgba_map[i]] = rgba_color[i];
57 line[0] = av_malloc(w * pixel_step[0]);
58 for (i = 0; i < w; i++)
59 memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]);
61 memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4);
65 dst_color[0] = RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]);
66 dst_color[1] = RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
67 dst_color[2] = RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
68 dst_color[3] = rgba_color[3];
70 for (plane = 0; plane < 4; plane++) {
72 int hsub1 = (plane == 1 || plane == 2) ? hsub : 0;
74 pixel_step[plane] = 1;
75 line_size = (w >> hsub1) * pixel_step[plane];
76 line[plane] = av_malloc(line_size);
77 memset(line[plane], dst_color[plane], line_size);
84 void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4],
85 uint8_t *src[4], int pixelstep[4],
86 int hsub, int vsub, int x, int y, int w, int h)
91 for (plane = 0; plane < 4 && dst[plane]; plane++) {
92 int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
93 int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
95 p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
96 for (i = 0; i < (h >> vsub1); i++) {
97 memcpy(p + (x >> hsub1) * pixelstep[plane],
98 src[plane], (w >> hsub1) * pixelstep[plane]);
99 p += dst_linesize[plane];
104 void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4],
105 uint8_t *src[4], int src_linesize[4], int pixelstep[4],
106 int hsub, int vsub, int x, int y, int y2, int w, int h)
111 for (plane = 0; plane < 4 && dst[plane]; plane++) {
112 int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
113 int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
115 p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
116 for (i = 0; i < (h >> vsub1); i++) {
117 memcpy(p + (x >> hsub1) * pixelstep[plane],
118 src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), (w >> hsub1) * pixelstep[plane]);
119 p += dst_linesize[plane];