git.sesse.net Git - ffmpeg/blob - libavfilter/drawutils.c

   1 /*
   2  * This file is part of FFmpeg.
   3  *
   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.
   8  *
   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.
  13  *
  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
  17  */
  18
  19 #include "libavutil/avutil.h"
  20 #include "libavutil/colorspace.h"
  21 #include "libavutil/pixdesc.h"
  22 #include "drawutils.h"
  23
  24 enum { RED = 0, GREEN, BLUE, ALPHA };
  25
  26 int ff_fill_rgba_map(uint8_t *rgba_map, enum PixelFormat pix_fmt)
  27 {
  28     switch (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;
  31     case PIX_FMT_RGBA:
  32     case PIX_FMT_RGB24: rgba_map[RED  ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break;
  33     case PIX_FMT_BGRA:
  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);
  37     }
  38     return 0;
  39 }
  40
  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])
  44 {
  45     uint8_t rgba_map[4] = {0};
  46     int i;
  47     const AVPixFmtDescriptor *pix_desc = &av_pix_fmt_descriptors[pix_fmt];
  48     int hsub = pix_desc->log2_chroma_w;
  49
  50     *is_packed_rgba = ff_fill_rgba_map(rgba_map, pix_fmt) >= 0;
  51
  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];
  56
  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]);
  60         if (rgba_map_ptr)
  61             memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4);
  62     } else {
  63         int plane;
  64
  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];
  69
  70         for (plane = 0; plane < 4; plane++) {
  71             int line_size;
  72             int hsub1 = (plane == 1 || plane == 2) ? hsub : 0;
  73
  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);
  78         }
  79     }
  80
  81     return 0;
  82 }
  83
  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)
  87 {
  88     int i, plane;
  89     uint8_t *p;
  90
  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;
  94
  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];
 100         }
 101     }
 102 }
 103
 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)
 107 {
 108     int i, plane;
 109     uint8_t *p;
 110
 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;
 114
 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];
 120         }
 121     }
 122 }