• Main Page
  • Related Pages
  • Modules
  • Data Structures
  • Files
  • File List
  • Globals

libavfilter/drawutils.c

Go to the documentation of this file.
00001 /*
00002  * This file is part of FFmpeg.
00003  *
00004  * FFmpeg is free software; you can redistribute it and/or
00005  * modify it under the terms of the GNU Lesser General Public
00006  * License as published by the Free Software Foundation; either
00007  * version 2.1 of the License, or (at your option) any later version.
00008  *
00009  * FFmpeg is distributed in the hope that it will be useful,
00010  * but WITHOUT ANY WARRANTY; without even the implied warranty of
00011  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00012  * Lesser General Public License for more details.
00013  *
00014  * You should have received a copy of the GNU Lesser General Public
00015  * License along with FFmpeg; if not, write to the Free Software
00016  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
00017  */
00018 
00019 #include "libavutil/avutil.h"
00020 #include "libavutil/colorspace.h"
00021 #include "libavutil/pixdesc.h"
00022 #include "drawutils.h"
00023 
00024 enum { RED = 0, GREEN, BLUE, ALPHA };
00025 
00026 int ff_fill_rgba_map(uint8_t *rgba_map, enum PixelFormat pix_fmt)
00027 {
00028     switch (pix_fmt) {
00029     case PIX_FMT_ARGB:  rgba_map[ALPHA] = 0; rgba_map[RED  ] = 1; rgba_map[GREEN] = 2; rgba_map[BLUE ] = 3; break;
00030     case PIX_FMT_ABGR:  rgba_map[ALPHA] = 0; rgba_map[BLUE ] = 1; rgba_map[GREEN] = 2; rgba_map[RED  ] = 3; break;
00031     case PIX_FMT_RGBA:
00032     case PIX_FMT_RGB24: rgba_map[RED  ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break;
00033     case PIX_FMT_BGRA:
00034     case PIX_FMT_BGR24: rgba_map[BLUE ] = 0; rgba_map[GREEN] = 1; rgba_map[RED  ] = 2; rgba_map[ALPHA] = 3; break;
00035     default:                    /* unsupported */
00036         return AVERROR(EINVAL);
00037     }
00038     return 0;
00039 }
00040 
00041 int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w, uint8_t dst_color[4],
00042                             enum PixelFormat pix_fmt, uint8_t rgba_color[4],
00043                             int *is_packed_rgba, uint8_t rgba_map_ptr[4])
00044 {
00045     uint8_t rgba_map[4] = {0};
00046     int i;
00047     const AVPixFmtDescriptor *pix_desc = &av_pix_fmt_descriptors[pix_fmt];
00048     int hsub = pix_desc->log2_chroma_w;
00049 
00050     *is_packed_rgba = ff_fill_rgba_map(rgba_map, pix_fmt) >= 0;
00051 
00052     if (*is_packed_rgba) {
00053         pixel_step[0] = (av_get_bits_per_pixel(pix_desc))>>3;
00054         for (i = 0; i < 4; i++)
00055             dst_color[rgba_map[i]] = rgba_color[i];
00056 
00057         line[0] = av_malloc(w * pixel_step[0]);
00058         for (i = 0; i < w; i++)
00059             memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]);
00060         if (rgba_map_ptr)
00061             memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4);
00062     } else {
00063         int plane;
00064 
00065         dst_color[0] = RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]);
00066         dst_color[1] = RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
00067         dst_color[2] = RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
00068         dst_color[3] = rgba_color[3];
00069 
00070         for (plane = 0; plane < 4; plane++) {
00071             int line_size;
00072             int hsub1 = (plane == 1 || plane == 2) ? hsub : 0;
00073 
00074             pixel_step[plane] = 1;
00075             line_size = (w >> hsub1) * pixel_step[plane];
00076             line[plane] = av_malloc(line_size);
00077             memset(line[plane], dst_color[plane], line_size);
00078         }
00079     }
00080 
00081     return 0;
00082 }
00083 
00084 void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4],
00085                        uint8_t *src[4], int pixelstep[4],
00086                        int hsub, int vsub, int x, int y, int w, int h)
00087 {
00088     int i, plane;
00089     uint8_t *p;
00090 
00091     for (plane = 0; plane < 4 && dst[plane]; plane++) {
00092         int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
00093         int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
00094 
00095         p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
00096         for (i = 0; i < (h >> vsub1); i++) {
00097             memcpy(p + (x >> hsub1) * pixelstep[plane],
00098                    src[plane], (w >> hsub1) * pixelstep[plane]);
00099             p += dst_linesize[plane];
00100         }
00101     }
00102 }
00103 
00104 void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4],
00105                        uint8_t *src[4], int src_linesize[4], int pixelstep[4],
00106                        int hsub, int vsub, int x, int y, int y2, int w, int h)
00107 {
00108     int i, plane;
00109     uint8_t *p;
00110 
00111     for (plane = 0; plane < 4 && dst[plane]; plane++) {
00112         int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
00113         int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
00114 
00115         p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
00116         for (i = 0; i < (h >> vsub1); i++) {
00117             memcpy(p + (x >> hsub1) * pixelstep[plane],
00118                    src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), (w >> hsub1) * pixelstep[plane]);
00119             p += dst_linesize[plane];
00120         }
00121     }
00122 }
Generated on Fri Feb 1 2013 14:34:48 for FFmpeg by doxygen 1.7.1