1/*
2 * Copyright (c) 2016 Tiancheng "Timothy" Gu
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along
17 * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 */
20
21#include <string.h>
22#include "checkasm.h"
23#include "libavfilter/vf_blend_init.h"
24#include "libavutil/common.h"
25#include "libavutil/internal.h"
26#include "libavutil/intreadwrite.h"
27
28#define WIDTH 256
29#define HEIGHT 256
30#define BUF_UNITS 3
31#define SIZE_PER_UNIT (WIDTH * HEIGHT)
32#define BUF_SIZE (BUF_UNITS * SIZE_PER_UNIT)
33
34#define randomize_buffers()                   \
35    do {                                      \
36        int i, j;                             \
37        for (i = 0; i < HEIGHT; i++) {        \
38            for (j = 0; j < WIDTH; j++) {     \
39                top1[i * WIDTH + j] =         \
40                top2[i * WIDTH + j] = i;      \
41                bot1[i * WIDTH + j] =         \
42                bot2[i * WIDTH + j] = j;      \
43            }                                 \
44        }                                     \
45        for (i = 0; i < SIZE_PER_UNIT; i += 4) { \
46            uint32_t r = rnd();               \
47            AV_WN32A(dst1 + i, r);            \
48            AV_WN32A(dst2 + i, r);            \
49        }                                     \
50        for (; i < BUF_SIZE; i += 4) {        \
51            uint32_t r = rnd();               \
52            AV_WN32A(top1 + i, r);            \
53            AV_WN32A(top2 + i, r);            \
54            r = rnd();                        \
55            AV_WN32A(bot1 + i, r);            \
56            AV_WN32A(bot2 + i, r);            \
57            r = rnd();                        \
58            AV_WN32A(dst1 + i, r);            \
59            AV_WN32A(dst2 + i, r);            \
60        }                                     \
61    } while (0)
62
63#define check_blend_func(depth)                                                            \
64    do {                                                                                   \
65        int i, w;                                                                          \
66        declare_func(void, const uint8_t *top, ptrdiff_t top_linesize,                     \
67                     const uint8_t *bottom, ptrdiff_t bottom_linesize,                     \
68                     uint8_t *dst, ptrdiff_t dst_linesize,                                 \
69                     ptrdiff_t width, ptrdiff_t height,                                    \
70                     struct FilterParams *param, double *values);                          \
71        w = WIDTH / depth;                                                                 \
72                                                                                           \
73        for (i = 0; i < BUF_UNITS - 1; i++) {                                              \
74            int src_offset = i * SIZE_PER_UNIT + (BUF_UNITS - 1 - i) * depth; /* Test various alignments */  \
75            int dst_offset = i * SIZE_PER_UNIT; /* dst must be aligned */                  \
76            randomize_buffers();                                                           \
77            call_ref(top1 + src_offset, w, bot1 + src_offset, w,                           \
78                     dst1 + dst_offset, w, w, HEIGHT, &param, NULL);                       \
79            call_new(top2 + src_offset, w, bot2 + src_offset, w,                           \
80                     dst2 + dst_offset, w, w, HEIGHT, &param, NULL);                       \
81            if (memcmp(top1, top2, BUF_SIZE) || memcmp(bot1, bot2, BUF_SIZE) || memcmp(dst1, dst2, BUF_SIZE)) \
82                fail();                                                                    \
83        }                                                                                  \
84        bench_new(top2, w / 4, bot2, w / 4, dst2, w / 4,                                   \
85                  w / 4, HEIGHT / 4, &param, NULL);                                        \
86    } while (0)
87
88void checkasm_check_blend(void)
89{
90    uint8_t *top1 = av_malloc(BUF_SIZE);
91    uint8_t *top2 = av_malloc(BUF_SIZE);
92    uint8_t *bot1 = av_malloc(BUF_SIZE);
93    uint8_t *bot2 = av_malloc(BUF_SIZE);
94    uint8_t *dst1 = av_malloc(BUF_SIZE);
95    uint8_t *dst2 = av_malloc(BUF_SIZE);
96    FilterParams param = {
97        .opacity = 1.0,
98    };
99
100#define check_and_report(name, val, depth)        \
101    param.mode = val;                             \
102    ff_blend_init(&param, depth * 8);             \
103    if (check_func(param.blend, #name))           \
104        check_blend_func(depth);
105
106    check_and_report(addition, BLEND_ADDITION, 1)
107    check_and_report(grainmerge, BLEND_GRAINMERGE, 1)
108    check_and_report(and, BLEND_AND, 1)
109    check_and_report(average, BLEND_AVERAGE, 1)
110    check_and_report(darken, BLEND_DARKEN, 1)
111    check_and_report(grainextract, BLEND_GRAINEXTRACT, 1)
112    check_and_report(hardmix, BLEND_HARDMIX, 1)
113    check_and_report(lighten, BLEND_LIGHTEN, 1)
114    check_and_report(multiply, BLEND_MULTIPLY, 1)
115    check_and_report(or, BLEND_OR, 1)
116    check_and_report(phoenix, BLEND_PHOENIX, 1)
117    check_and_report(screen, BLEND_SCREEN, 1)
118    check_and_report(subtract, BLEND_SUBTRACT, 1)
119    check_and_report(xor, BLEND_XOR, 1)
120    check_and_report(difference, BLEND_DIFFERENCE, 1)
121    check_and_report(extremity, BLEND_EXTREMITY, 1)
122    check_and_report(negation, BLEND_NEGATION, 1)
123
124    report("8bit");
125
126    check_and_report(addition_16, BLEND_ADDITION, 2)
127    check_and_report(grainmerge_16, BLEND_GRAINMERGE, 2)
128    check_and_report(and_16, BLEND_AND, 2)
129    check_and_report(average_16, BLEND_AVERAGE, 2)
130    check_and_report(darken_16, BLEND_DARKEN, 2)
131    check_and_report(grainextract_16, BLEND_GRAINEXTRACT, 2)
132    check_and_report(difference_16, BLEND_DIFFERENCE, 2)
133    check_and_report(extremity_16, BLEND_EXTREMITY, 2)
134    check_and_report(negation_16, BLEND_NEGATION, 2)
135    check_and_report(lighten_16, BLEND_LIGHTEN, 2)
136    check_and_report(or_16, BLEND_OR, 2)
137    check_and_report(phoenix_16, BLEND_PHOENIX, 2)
138    check_and_report(subtract_16, BLEND_SUBTRACT, 2)
139    check_and_report(xor_16, BLEND_SUBTRACT, 2)
140
141    report("16bit");
142
143    av_freep(&top1);
144    av_freep(&top2);
145    av_freep(&bot1);
146    av_freep(&bot2);
147    av_freep(&dst1);
148    av_freep(&dst2);
149}
150