mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-07 11:13:41 +02:00
6c3a82f043
- Introduce ff_draw_init2, which takes explicit colorspace and range args - Use lavu/csp and lavfi/colorspace for conversion, rather than the lavu/colorspace.h macros - Use the passed-in colorspace when performing RGB->YUV conversions The upshot of this is: - Support for YUV spaces other than BT601 - Better rounding for all conversions - Particular rounding improvements in >8-bit formats, which previously used simple left-shifts - Support for limited-range RGB - Support for full-range YUV in non-J pixfmts Due to the rounding improvements, this results in a large number of minor changes to FATE tests. Signed-off-by: rcombs <rcombs@rcombs.me>
167 lines
5.6 KiB
C
167 lines
5.6 KiB
C
/*
|
|
* 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 "avfilter.h"
|
|
#include "libavutil/pixfmt.h"
|
|
|
|
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt);
|
|
|
|
#define MAX_PLANES 4
|
|
|
|
typedef struct FFDrawContext {
|
|
const struct AVPixFmtDescriptor *desc;
|
|
enum AVPixelFormat format;
|
|
unsigned nb_planes;
|
|
int pixelstep[MAX_PLANES]; /*< offset between pixels */
|
|
uint8_t hsub[MAX_PLANES]; /*< horizontal subsampling */
|
|
uint8_t vsub[MAX_PLANES]; /*< vertical subsampling */
|
|
uint8_t hsub_max;
|
|
uint8_t vsub_max;
|
|
enum AVColorRange range;
|
|
unsigned flags;
|
|
enum AVColorSpace csp;
|
|
double rgb2yuv[3][3];
|
|
} FFDrawContext;
|
|
|
|
typedef struct FFDrawColor {
|
|
uint8_t rgba[4];
|
|
union {
|
|
uint32_t u32[4];
|
|
uint16_t u16[8];
|
|
uint8_t u8[16];
|
|
} comp[MAX_PLANES];
|
|
} FFDrawColor;
|
|
|
|
/**
|
|
* Process alpha pixel component.
|
|
*/
|
|
#define FF_DRAW_PROCESS_ALPHA 1
|
|
|
|
/**
|
|
* Init a draw context.
|
|
*
|
|
* Only a limited number of pixel formats are supported, if format is not
|
|
* supported the function will return an error.
|
|
* @param format pixel format of the frames that will be drawn onto
|
|
* @param csp color space of the frames that will be drawn onto,
|
|
* defaulting to BT601 or RGB depending on the specified format
|
|
* when AVCOL_SPC_UNSPECIFIED is passed.
|
|
* @param range sample value range of the frames that will be drawn onto,
|
|
* defaulting to TV-range unless using a legacy J format
|
|
* when AVCOL_RANGE_UNSPECIFIED is passed.
|
|
* @param flags combination of FF_DRAW_* flags.
|
|
* @return 0 for success, < 0 for error
|
|
*/
|
|
int ff_draw_init2(FFDrawContext *draw, enum AVPixelFormat format, enum AVColorSpace csp,
|
|
enum AVColorRange range, unsigned flags);
|
|
|
|
/*
|
|
* Legacy wrapper for ff_draw_init2.
|
|
*/
|
|
int ff_draw_init(FFDrawContext *draw, enum AVPixelFormat format, unsigned flags);
|
|
|
|
|
|
|
|
/**
|
|
* Prepare a color. The rgba value passed is always 8-bit full-range in the RGB space
|
|
* corresponding to the space set at initialization.
|
|
*/
|
|
void ff_draw_color(FFDrawContext *draw, FFDrawColor *color, const uint8_t rgba[4]);
|
|
|
|
/**
|
|
* Copy a rectangle from an image to another.
|
|
*
|
|
* The coordinates must be as even as the subsampling requires.
|
|
*/
|
|
void ff_copy_rectangle2(FFDrawContext *draw,
|
|
uint8_t *dst[], int dst_linesize[],
|
|
uint8_t *src[], int src_linesize[],
|
|
int dst_x, int dst_y, int src_x, int src_y,
|
|
int w, int h);
|
|
|
|
/**
|
|
* Fill a rectangle with an uniform color.
|
|
*
|
|
* The coordinates must be as even as the subsampling requires.
|
|
* The color needs to be inited with ff_draw_color.
|
|
*/
|
|
void ff_fill_rectangle(FFDrawContext *draw, FFDrawColor *color,
|
|
uint8_t *dst[], int dst_linesize[],
|
|
int dst_x, int dst_y, int w, int h);
|
|
|
|
/**
|
|
* Blend a rectangle with an uniform color.
|
|
*/
|
|
void ff_blend_rectangle(FFDrawContext *draw, FFDrawColor *color,
|
|
uint8_t *dst[], int dst_linesize[],
|
|
int dst_w, int dst_h,
|
|
int x0, int y0, int w, int h);
|
|
|
|
/**
|
|
* Blend an alpha mask with an uniform color.
|
|
*
|
|
* @param draw draw context
|
|
* @param color color for the overlay;
|
|
* @param dst destination image
|
|
* @param dst_linesize line stride of the destination
|
|
* @param dst_w width of the destination image
|
|
* @param dst_h height of the destination image
|
|
* @param mask mask
|
|
* @param mask_linesize line stride of the mask
|
|
* @param mask_w width of the mask
|
|
* @param mask_h height of the mask
|
|
* @param l2depth log2 of depth of the mask (0 for 1bpp, 3 for 8bpp)
|
|
* @param endianness bit order of the mask (0: MSB to the left)
|
|
* @param x0 horizontal position of the overlay
|
|
* @param y0 vertical position of the overlay
|
|
*/
|
|
void ff_blend_mask(FFDrawContext *draw, FFDrawColor *color,
|
|
uint8_t *dst[], int dst_linesize[], int dst_w, int dst_h,
|
|
const uint8_t *mask, int mask_linesize, int mask_w, int mask_h,
|
|
int l2depth, unsigned endianness, int x0, int y0);
|
|
|
|
/**
|
|
* Round a dimension according to subsampling.
|
|
*
|
|
* @param draw draw context
|
|
* @param sub_dir 0 for horizontal, 1 for vertical
|
|
* @param round_dir 0 nearest, -1 round down, +1 round up
|
|
* @param value value to round
|
|
* @return the rounded value
|
|
*/
|
|
int ff_draw_round_to_sub(FFDrawContext *draw, int sub_dir, int round_dir,
|
|
int value);
|
|
|
|
/**
|
|
* Return the list of pixel formats supported by the draw functions.
|
|
*
|
|
* The flags are the same as ff_draw_init, i.e., none currently.
|
|
*/
|
|
AVFilterFormats *ff_draw_supported_pixel_formats(unsigned flags);
|
|
|
|
#endif /* AVFILTER_DRAWUTILS_H */
|