1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* 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/imgutils.c"
#include "libavutil/crc.h"
#include "libavutil/mem.h"
#undef printf
static int check_image_fill(enum AVPixelFormat pix_fmt, int w, int h) {
uint8_t *data[4];
size_t sizes[4];
ptrdiff_t linesizes1[4], offsets[3] = { 0 };
int i, total_size, linesizes[4];
if (av_image_fill_linesizes(linesizes, pix_fmt, w) < 0)
return -1;
for (i = 0; i < 4; i++)
linesizes1[i] = linesizes[i];
if (av_image_fill_plane_sizes(sizes, pix_fmt, h, linesizes1) < 0)
return -1;
total_size = av_image_fill_pointers(data, pix_fmt, h, (void *)1, linesizes);
if (total_size < 0)
return -1;
for (i = 0; i < 4 && data[i]; i++);
printf("planes: %d", i);
// Test the output of av_image_fill_linesizes()
printf(", linesizes:");
for (i = 0; i < 4; i++)
printf(" %3d", linesizes[i]);
// Test the output of av_image_fill_plane_sizes()
printf(", plane_sizes:");
for (i = 0; i < 4; i++)
printf(" %5"SIZE_SPECIFIER, sizes[i]);
// Test the output of av_image_fill_pointers()
for (i = 0; i < 3 && data[i + 1]; i++)
offsets[i] = data[i + 1] - data[i];
printf(", plane_offsets:");
for (i = 0; i < 3; i++)
printf(" %5"PTRDIFF_SPECIFIER, offsets[i]);
printf(", total_size: %d", total_size);
return 0;
}
static int check_image_fill_black(const AVPixFmtDescriptor *desc, enum AVPixelFormat pix_fmt, int w, int h)
{
uint8_t *data[4];
ptrdiff_t linesizes1[4];
int ret, total_size, linesizes[4];
ret = av_image_fill_linesizes(linesizes, pix_fmt, w);
if (ret < 0)
return ret;
total_size = av_image_alloc(data, linesizes, w, h, pix_fmt, 4);
if (total_size < 0) {
printf("alloc failure");
return total_size;
}
printf("total_size: %6d", total_size);
if (desc->flags & AV_PIX_FMT_FLAG_PAL)
total_size -= 256 * 4;
// Make it non-black by default...
memset(data[0], 0xA3, total_size);
for (int i = 0; i < 4; i++)
linesizes1[i] = linesizes[i];
for (enum AVColorRange range = 0; range < AVCOL_RANGE_NB; range++) {
ret = av_image_fill_black(data, linesizes1, pix_fmt, range, w, h);
printf(", black_%s_crc: ", av_color_range_name(range));
if (ret < 0) {
printf("----------");
} else {
const AVCRC *crc = av_crc_get_table(AV_CRC_32_IEEE_LE);
printf("0x%08"PRIx32, av_crc(crc, 0, data[0], total_size));
}
}
av_freep(&data[0]);
return 0;
}
int main(void)
{
int64_t x, y;
for (y = -1; y<UINT_MAX; y+= y/2 + 1) {
for (x = -1; x<UINT_MAX; x+= x/2 + 1) {
int ret = av_image_check_size(x, y, 0, NULL);
printf("%d", ret >= 0);
}
printf("\n");
}
printf("\n");
for (int i = 0; i < 2; i++) {
printf(i ? "\nimage_fill_black tests\n" : "image_fill tests\n");
for (const AVPixFmtDescriptor *desc = NULL; desc = av_pix_fmt_desc_next(desc);) {
int w = 64, h = 48;
enum AVPixelFormat pix_fmt = av_pix_fmt_desc_get_id(desc);
if (desc->flags & AV_PIX_FMT_FLAG_HWACCEL)
continue;
printf("%-16s", desc->name);
if (i == 0)
check_image_fill(pix_fmt, w, h);
else
check_image_fill_black(desc, pix_fmt, w, h);
printf("\n");
}
}
return 0;
}
|
