Libav 0.7.1
|
00001 /* 00002 * This file is part of Libav. 00003 * 00004 * Libav 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 * Libav 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 Libav; 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_line_with_color(uint8_t *line[4], int pixel_step[4], int w, uint8_t dst_color[4], 00027 enum PixelFormat pix_fmt, uint8_t rgba_color[4], 00028 int *is_packed_rgba, uint8_t rgba_map_ptr[4]) 00029 { 00030 uint8_t rgba_map[4] = {0}; 00031 int i; 00032 const AVPixFmtDescriptor *pix_desc = &av_pix_fmt_descriptors[pix_fmt]; 00033 int hsub = pix_desc->log2_chroma_w; 00034 00035 *is_packed_rgba = 1; 00036 switch (pix_fmt) { 00037 case PIX_FMT_ARGB: rgba_map[ALPHA] = 0; rgba_map[RED ] = 1; rgba_map[GREEN] = 2; rgba_map[BLUE ] = 3; break; 00038 case PIX_FMT_ABGR: rgba_map[ALPHA] = 0; rgba_map[BLUE ] = 1; rgba_map[GREEN] = 2; rgba_map[RED ] = 3; break; 00039 case PIX_FMT_RGBA: 00040 case PIX_FMT_RGB24: rgba_map[RED ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break; 00041 case PIX_FMT_BGRA: 00042 case PIX_FMT_BGR24: rgba_map[BLUE ] = 0; rgba_map[GREEN] = 1; rgba_map[RED ] = 2; rgba_map[ALPHA] = 3; break; 00043 default: 00044 *is_packed_rgba = 0; 00045 } 00046 00047 if (*is_packed_rgba) { 00048 pixel_step[0] = (av_get_bits_per_pixel(pix_desc))>>3; 00049 for (i = 0; i < 4; i++) 00050 dst_color[rgba_map[i]] = rgba_color[i]; 00051 00052 line[0] = av_malloc(w * pixel_step[0]); 00053 for (i = 0; i < w; i++) 00054 memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]); 00055 if (rgba_map_ptr) 00056 memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4); 00057 } else { 00058 int plane; 00059 00060 dst_color[0] = RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]); 00061 dst_color[1] = RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0); 00062 dst_color[2] = RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0); 00063 dst_color[3] = rgba_color[3]; 00064 00065 for (plane = 0; plane < 4; plane++) { 00066 int line_size; 00067 int hsub1 = (plane == 1 || plane == 2) ? hsub : 0; 00068 00069 pixel_step[plane] = 1; 00070 line_size = (w >> hsub1) * pixel_step[plane]; 00071 line[plane] = av_malloc(line_size); 00072 memset(line[plane], dst_color[plane], line_size); 00073 } 00074 } 00075 00076 return 0; 00077 } 00078 00079 void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4], 00080 uint8_t *src[4], int pixelstep[4], 00081 int hsub, int vsub, int x, int y, int w, int h) 00082 { 00083 int i, plane; 00084 uint8_t *p; 00085 00086 for (plane = 0; plane < 4 && dst[plane]; plane++) { 00087 int hsub1 = plane == 1 || plane == 2 ? hsub : 0; 00088 int vsub1 = plane == 1 || plane == 2 ? vsub : 0; 00089 00090 p = dst[plane] + (y >> vsub1) * dst_linesize[plane]; 00091 for (i = 0; i < (h >> vsub1); i++) { 00092 memcpy(p + (x >> hsub1) * pixelstep[plane], 00093 src[plane], (w >> hsub1) * pixelstep[plane]); 00094 p += dst_linesize[plane]; 00095 } 00096 } 00097 } 00098 00099 void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4], 00100 uint8_t *src[4], int src_linesize[4], int pixelstep[4], 00101 int hsub, int vsub, int x, int y, int y2, int w, int h) 00102 { 00103 int i, plane; 00104 uint8_t *p; 00105 00106 for (plane = 0; plane < 4 && dst[plane]; plane++) { 00107 int hsub1 = plane == 1 || plane == 2 ? hsub : 0; 00108 int vsub1 = plane == 1 || plane == 2 ? vsub : 0; 00109 00110 p = dst[plane] + (y >> vsub1) * dst_linesize[plane]; 00111 for (i = 0; i < (h >> vsub1); i++) { 00112 memcpy(p + (x >> hsub1) * pixelstep[plane], 00113 src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), (w >> hsub1) * pixelstep[plane]); 00114 p += dst_linesize[plane]; 00115 } 00116 } 00117 }