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/blend.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, ¶m, NULL); \
79 call_new(top2 + src_offset, w, bot2 + src_offset, w, \
80 dst2 + dst_offset, w, w, HEIGHT, ¶m, 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, ¶m, NULL); \
86 } while (0)
87
checkasm_check_blend(void)88 void 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(¶m, depth - 1); \
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