mirror of https://github.com/FFmpeg/FFmpeg.git
Add drawutils.h and drawutils.c, and use them in the pad filter. The new functions are going to be shared by other filters. Signed-off-by: Michael Niedermayer <michaelni@gmx.at>oldabi
parent
bbcaaf752f
commit
bcfd9e821b
4 changed files with 182 additions and 105 deletions
@ -0,0 +1,117 @@ |
||||
/*
|
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "libavutil/avutil.h" |
||||
#include "libavutil/colorspace.h" |
||||
#include "libavutil/pixdesc.h" |
||||
#include "drawutils.h" |
||||
|
||||
enum { RED = 0, GREEN, BLUE, ALPHA }; |
||||
|
||||
int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w, uint8_t dst_color[4], |
||||
enum PixelFormat pix_fmt, uint8_t rgba_color[4], |
||||
int *is_packed_rgba, uint8_t rgba_map_ptr[4]) |
||||
{ |
||||
uint8_t rgba_map[4] = {0}; |
||||
int i; |
||||
const AVPixFmtDescriptor *pix_desc = &av_pix_fmt_descriptors[pix_fmt]; |
||||
int hsub = pix_desc->log2_chroma_w; |
||||
|
||||
*is_packed_rgba = 1; |
||||
switch (pix_fmt) { |
||||
case PIX_FMT_ARGB: rgba_map[ALPHA] = 0; rgba_map[RED ] = 1; rgba_map[GREEN] = 2; rgba_map[BLUE ] = 3; break; |
||||
case PIX_FMT_ABGR: rgba_map[ALPHA] = 0; rgba_map[BLUE ] = 1; rgba_map[GREEN] = 2; rgba_map[RED ] = 3; break; |
||||
case PIX_FMT_RGBA: |
||||
case PIX_FMT_RGB24: rgba_map[RED ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break; |
||||
case PIX_FMT_BGRA: |
||||
case PIX_FMT_BGR24: rgba_map[BLUE ] = 0; rgba_map[GREEN] = 1; rgba_map[RED ] = 2; rgba_map[ALPHA] = 3; break; |
||||
default: |
||||
*is_packed_rgba = 0; |
||||
} |
||||
|
||||
if (*is_packed_rgba) { |
||||
pixel_step[0] = (av_get_bits_per_pixel(pix_desc))>>3; |
||||
for (i = 0; i < 4; i++) |
||||
dst_color[rgba_map[i]] = rgba_color[i]; |
||||
|
||||
line[0] = av_malloc(w * pixel_step[0]); |
||||
for (i = 0; i < w; i++) |
||||
memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]); |
||||
if (rgba_map_ptr) |
||||
memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4); |
||||
} else { |
||||
int plane; |
||||
|
||||
dst_color[0] = RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]); |
||||
dst_color[1] = RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0); |
||||
dst_color[2] = RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0); |
||||
dst_color[3] = rgba_color[3]; |
||||
|
||||
for (plane = 0; plane < 4; plane++) { |
||||
int line_size; |
||||
int hsub1 = (plane == 1 || plane == 2) ? hsub : 0; |
||||
|
||||
pixel_step[plane] = 1; |
||||
line_size = (w >> hsub1) * pixel_step[plane]; |
||||
line[plane] = av_malloc(line_size); |
||||
memset(line[plane], dst_color[plane], line_size); |
||||
} |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4], |
||||
uint8_t *src[4], int pixelstep[4], |
||||
int hsub, int vsub, int x, int y, int w, int h) |
||||
{ |
||||
int i, plane; |
||||
uint8_t *p; |
||||
|
||||
for (plane = 0; plane < 4 && dst[plane]; plane++) { |
||||
int hsub1 = plane == 1 || plane == 2 ? hsub : 0; |
||||
int vsub1 = plane == 1 || plane == 2 ? vsub : 0; |
||||
|
||||
p = dst[plane] + (y >> vsub1) * dst_linesize[plane]; |
||||
for (i = 0; i < (h >> vsub1); i++) { |
||||
memcpy(p + (x >> hsub1) * pixelstep[plane], |
||||
src[plane], (w >> hsub1) * pixelstep[plane]); |
||||
p += dst_linesize[plane]; |
||||
} |
||||
} |
||||
} |
||||
|
||||
void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4], |
||||
uint8_t *src[4], int src_linesize[4], int pixelstep[4], |
||||
int hsub, int vsub, int x, int y, int y2, int w, int h) |
||||
{ |
||||
int i, plane; |
||||
uint8_t *p; |
||||
|
||||
for (plane = 0; plane < 4 && dst[plane]; plane++) { |
||||
int hsub1 = plane == 1 || plane == 2 ? hsub : 0; |
||||
int vsub1 = plane == 1 || plane == 2 ? vsub : 0; |
||||
|
||||
p = dst[plane] + (y >> vsub1) * dst_linesize[plane]; |
||||
for (i = 0; i < (h >> vsub1); i++) { |
||||
memcpy(p + (x >> hsub1) * pixelstep[plane], |
||||
src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), (w >> hsub1) * pixelstep[plane]); |
||||
p += dst_linesize[plane]; |
||||
} |
||||
} |
||||
} |
@ -0,0 +1,43 @@ |
||||
/*
|
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#ifndef AVFILTER_DRAWUTILS_H |
||||
#define AVFILTER_DRAWUTILS_H |
||||
|
||||
/**
|
||||
* @file |
||||
* misc drawing utilities |
||||
*/ |
||||
|
||||
#include <stdint.h> |
||||
#include "libavutil/pixfmt.h" |
||||
|
||||
int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w, |
||||
uint8_t dst_color[4], |
||||
enum PixelFormat pix_fmt, uint8_t rgba_color[4], |
||||
int *is_packed_rgba, uint8_t rgba_map[4]); |
||||
|
||||
void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4], |
||||
uint8_t *src[4], int pixelstep[4], |
||||
int hsub, int vsub, int x, int y, int w, int h); |
||||
|
||||
void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4], |
||||
uint8_t *src[4], int src_linesize[4], int pixelstep[4], |
||||
int hsub, int vsub, int x, int y, int y2, int w, int h); |
||||
|
||||
#endif /* AVFILTER_DRAWUTILS_H */ |
Loading…
Reference in new issue