aboutsummaryrefslogtreecommitdiffstats
path: root/libavutil/tests/imgutils.c
blob: 582a3581579d81d491b2a778d17676e9ee6945bf (plain) (blame)
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
/*
 * 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"

#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;
}