mirror of https://git.videolan.org/git/ffmpeg.git
129 lines
4.1 KiB
C
129 lines
4.1 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
|
|
*/
|
|
|
|
#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;
|
|
}
|