Statistics
| Branch: | Revision:

ffmpeg / libavfilter / drawutils.c @ master

History | View | Annotate | Download (4.38 KB)

1 bcfd9e82 Stefano Sabatini
/*
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
#include "libavutil/avutil.h"
20
#include "libavutil/colorspace.h"
21
#include "libavutil/pixdesc.h"
22
#include "drawutils.h"
23
24
enum { RED = 0, GREEN, BLUE, ALPHA };
25
26
int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w, uint8_t dst_color[4],
27
                            enum PixelFormat pix_fmt, uint8_t rgba_color[4],
28
                            int *is_packed_rgba, uint8_t rgba_map_ptr[4])
29
{
30
    uint8_t rgba_map[4] = {0};
31
    int i;
32
    const AVPixFmtDescriptor *pix_desc = &av_pix_fmt_descriptors[pix_fmt];
33
    int hsub = pix_desc->log2_chroma_w;
34
35
    *is_packed_rgba = 1;
36
    switch (pix_fmt) {
37
    case PIX_FMT_ARGB:  rgba_map[ALPHA] = 0; rgba_map[RED  ] = 1; rgba_map[GREEN] = 2; rgba_map[BLUE ] = 3; break;
38
    case PIX_FMT_ABGR:  rgba_map[ALPHA] = 0; rgba_map[BLUE ] = 1; rgba_map[GREEN] = 2; rgba_map[RED  ] = 3; break;
39
    case PIX_FMT_RGBA:
40
    case PIX_FMT_RGB24: rgba_map[RED  ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break;
41
    case PIX_FMT_BGRA:
42
    case PIX_FMT_BGR24: rgba_map[BLUE ] = 0; rgba_map[GREEN] = 1; rgba_map[RED  ] = 2; rgba_map[ALPHA] = 3; break;
43
    default:
44
        *is_packed_rgba = 0;
45
    }
46
47
    if (*is_packed_rgba) {
48
        pixel_step[0] = (av_get_bits_per_pixel(pix_desc))>>3;
49
        for (i = 0; i < 4; i++)
50
            dst_color[rgba_map[i]] = rgba_color[i];
51
52
        line[0] = av_malloc(w * pixel_step[0]);
53
        for (i = 0; i < w; i++)
54
            memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]);
55
        if (rgba_map_ptr)
56
            memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4);
57
    } else {
58
        int plane;
59
60
        dst_color[0] = RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]);
61
        dst_color[1] = RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
62
        dst_color[2] = RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);
63
        dst_color[3] = rgba_color[3];
64
65
        for (plane = 0; plane < 4; plane++) {
66
            int line_size;
67
            int hsub1 = (plane == 1 || plane == 2) ? hsub : 0;
68
69
            pixel_step[plane] = 1;
70
            line_size = (w >> hsub1) * pixel_step[plane];
71
            line[plane] = av_malloc(line_size);
72
            memset(line[plane], dst_color[plane], line_size);
73
        }
74
    }
75
76
    return 0;
77
}
78
79
void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4],
80
                       uint8_t *src[4], int pixelstep[4],
81
                       int hsub, int vsub, int x, int y, int w, int h)
82
{
83
    int i, plane;
84
    uint8_t *p;
85
86
    for (plane = 0; plane < 4 && dst[plane]; plane++) {
87
        int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
88
        int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
89
90
        p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
91
        for (i = 0; i < (h >> vsub1); i++) {
92
            memcpy(p + (x >> hsub1) * pixelstep[plane],
93
                   src[plane], (w >> hsub1) * pixelstep[plane]);
94
            p += dst_linesize[plane];
95
        }
96
    }
97
}
98
99
void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4],
100
                       uint8_t *src[4], int src_linesize[4], int pixelstep[4],
101
                       int hsub, int vsub, int x, int y, int y2, int w, int h)
102
{
103
    int i, plane;
104
    uint8_t *p;
105
106
    for (plane = 0; plane < 4 && dst[plane]; plane++) {
107
        int hsub1 = plane == 1 || plane == 2 ? hsub : 0;
108
        int vsub1 = plane == 1 || plane == 2 ? vsub : 0;
109
110
        p = dst[plane] + (y >> vsub1) * dst_linesize[plane];
111
        for (i = 0; i < (h >> vsub1); i++) {
112
            memcpy(p + (x >> hsub1) * pixelstep[plane],
113
                   src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), (w >> hsub1) * pixelstep[plane]);
114
            p += dst_linesize[plane];
115
        }
116
    }
117
}