avfilter/af_afftdn: add alias for sample_noise end
[ffmpeg.git] / libavfilter / drawutils.h
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 #ifndef AVFILTER_DRAWUTILS_H
20 #define AVFILTER_DRAWUTILS_H
21
22 /**
23  * @file
24  * misc drawing utilities
25  */
26
27 #include <stdint.h>
28 #include "avfilter.h"
29 #include "libavutil/pixfmt.h"
30
31 int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt);
32
33 int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w,
34                             uint8_t dst_color[4],
35                             enum AVPixelFormat pix_fmt, uint8_t rgba_color[4],
36                             int *is_packed_rgba, uint8_t rgba_map[4]);
37
38 void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4],
39                        uint8_t *src[4], int pixelstep[4],
40                        int hsub, int vsub, int x, int y, int w, int h);
41
42 void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4],
43                        uint8_t *src[4], int src_linesize[4], int pixelstep[4],
44                        int hsub, int vsub, int x, int y, int y2, int w, int h);
45
46 #define MAX_PLANES 4
47
48 typedef struct FFDrawContext {
49     const struct AVPixFmtDescriptor *desc;
50     enum AVPixelFormat format;
51     unsigned nb_planes;
52     int pixelstep[MAX_PLANES]; /*< offset between pixels */
53     uint8_t comp_mask[MAX_PLANES]; /*< bitmask of used non-alpha components */
54     uint8_t hsub[MAX_PLANES];  /*< horizontal subsampling */
55     uint8_t vsub[MAX_PLANES];  /*< vertical subsampling */
56     uint8_t hsub_max;
57     uint8_t vsub_max;
58     int full_range;
59     unsigned flags;
60 } FFDrawContext;
61
62 typedef struct FFDrawColor {
63     uint8_t rgba[4];
64     union {
65         uint32_t u32[4];
66         uint16_t u16[8];
67         uint8_t  u8[16];
68     } comp[MAX_PLANES];
69 } FFDrawColor;
70
71 /**
72   * Process alpha pixel component.
73   */
74 #define FF_DRAW_PROCESS_ALPHA 1
75
76 /**
77  * Init a draw context.
78  *
79  * Only a limited number of pixel formats are supported, if format is not
80  * supported the function will return an error.
81  * flags is combination of FF_DRAW_* flags.
82  * @return  0 for success, < 0 for error
83  */
84 int ff_draw_init(FFDrawContext *draw, enum AVPixelFormat format, unsigned flags);
85
86 /**
87  * Prepare a color.
88  */
89 void ff_draw_color(FFDrawContext *draw, FFDrawColor *color, const uint8_t rgba[4]);
90
91 /**
92  * Copy a rectangle from an image to another.
93  *
94  * The coordinates must be as even as the subsampling requires.
95  */
96 void ff_copy_rectangle2(FFDrawContext *draw,
97                         uint8_t *dst[], int dst_linesize[],
98                         uint8_t *src[], int src_linesize[],
99                         int dst_x, int dst_y, int src_x, int src_y,
100                         int w, int h);
101
102 /**
103  * Fill a rectangle with an uniform color.
104  *
105  * The coordinates must be as even as the subsampling requires.
106  * The color needs to be inited with ff_draw_color.
107  */
108 void ff_fill_rectangle(FFDrawContext *draw, FFDrawColor *color,
109                        uint8_t *dst[], int dst_linesize[],
110                        int dst_x, int dst_y, int w, int h);
111
112 /**
113  * Blend a rectangle with an uniform color.
114  */
115 void ff_blend_rectangle(FFDrawContext *draw, FFDrawColor *color,
116                         uint8_t *dst[], int dst_linesize[],
117                         int dst_w, int dst_h,
118                         int x0, int y0, int w, int h);
119
120 /**
121  * Blend an alpha mask with an uniform color.
122  *
123  * @param draw           draw context
124  * @param color          color for the overlay;
125  * @param dst            destination image
126  * @param dst_linesize   line stride of the destination
127  * @param dst_w          width of the destination image
128  * @param dst_h          height of the destination image
129  * @param mask           mask
130  * @param mask_linesize  line stride of the mask
131  * @param mask_w         width of the mask
132  * @param mask_h         height of the mask
133  * @param l2depth        log2 of depth of the mask (0 for 1bpp, 3 for 8bpp)
134  * @param endianness     bit order of the mask (0: MSB to the left)
135  * @param x0             horizontal position of the overlay
136  * @param y0             vertical position of the overlay
137  */
138 void ff_blend_mask(FFDrawContext *draw, FFDrawColor *color,
139                    uint8_t *dst[], int dst_linesize[], int dst_w, int dst_h,
140                    const uint8_t *mask, int mask_linesize, int mask_w, int mask_h,
141                    int l2depth, unsigned endianness, int x0, int y0);
142
143 /**
144  * Round a dimension according to subsampling.
145  *
146  * @param draw       draw context
147  * @param sub_dir    0 for horizontal, 1 for vertical
148  * @param round_dir  0 nearest, -1 round down, +1 round up
149  * @param value      value to round
150  * @return  the rounded value
151  */
152 int ff_draw_round_to_sub(FFDrawContext *draw, int sub_dir, int round_dir,
153                          int value);
154
155 /**
156  * Return the list of pixel formats supported by the draw functions.
157  *
158  * The flags are the same as ff_draw_init, i.e., none currently.
159  */
160 AVFilterFormats *ff_draw_supported_pixel_formats(unsigned flags);
161
162 #endif /* AVFILTER_DRAWUTILS_H */