1 /*
2 * Copyright 2011 The LibYuv Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include <stdlib.h>
12 #include <time.h>
13
14 #include "../unit_test/unit_test.h"
15 #include "libyuv/cpu_id.h"
16 #include "libyuv/scale_uv.h"
17
18 namespace libyuv {
19
20 #define STRINGIZE(line) #line
21 #define FILELINESTR(file, line) file ":" STRINGIZE(line)
22
23 #if !defined(DISABLE_SLOW_TESTS) || defined(__x86_64__) || defined(__i386__)
24 // SLOW TESTS are those that are unoptimized C code.
25 // FULL TESTS are optimized but test many variations of the same code.
26 #define ENABLE_FULL_TESTS
27 #endif
28
29 // Test scaling with C vs Opt and return maximum pixel difference. 0 = exact.
UVTestFilter(int src_width,int src_height,int dst_width,int dst_height,FilterMode f,int benchmark_iterations,int disable_cpu_flags,int benchmark_cpu_info)30 static int UVTestFilter(int src_width,
31 int src_height,
32 int dst_width,
33 int dst_height,
34 FilterMode f,
35 int benchmark_iterations,
36 int disable_cpu_flags,
37 int benchmark_cpu_info) {
38 if (!SizeValid(src_width, src_height, dst_width, dst_height)) {
39 return 0;
40 }
41
42 int i;
43 int64_t src_uv_plane_size = Abs(src_width) * Abs(src_height) * 2LL;
44 int src_stride_uv = Abs(src_width) * 2;
45 int64_t dst_uv_plane_size = dst_width * dst_height * 2LL;
46 int dst_stride_uv = dst_width * 2;
47
48 align_buffer_page_end(src_uv, src_uv_plane_size);
49 align_buffer_page_end(dst_uv_c, dst_uv_plane_size);
50 align_buffer_page_end(dst_uv_opt, dst_uv_plane_size);
51
52 if (!src_uv || !dst_uv_c || !dst_uv_opt) {
53 printf("Skipped. Alloc failed " FILELINESTR(__FILE__, __LINE__) "\n");
54 return 0;
55 }
56 MemRandomize(src_uv, src_uv_plane_size);
57 memset(dst_uv_c, 2, dst_uv_plane_size);
58 memset(dst_uv_opt, 123, dst_uv_plane_size);
59
60 MaskCpuFlags(disable_cpu_flags); // Disable all CPU optimization.
61 double c_time = get_time();
62 UVScale(src_uv, src_stride_uv, src_width, src_height, dst_uv_c, dst_stride_uv,
63 dst_width, dst_height, f);
64 c_time = (get_time() - c_time);
65
66 MaskCpuFlags(benchmark_cpu_info); // Enable all CPU optimization.
67 double opt_time = get_time();
68 for (i = 0; i < benchmark_iterations; ++i) {
69 UVScale(src_uv, src_stride_uv, src_width, src_height, dst_uv_opt,
70 dst_stride_uv, dst_width, dst_height, f);
71 }
72 opt_time = (get_time() - opt_time) / benchmark_iterations;
73
74 // Report performance of C vs OPT
75 printf("filter %d - %8d us C - %8d us OPT\n", f,
76 static_cast<int>(c_time * 1e6), static_cast<int>(opt_time * 1e6));
77
78 int max_diff = 0;
79 for (i = 0; i < dst_uv_plane_size; ++i) {
80 int abs_diff = Abs(dst_uv_c[i] - dst_uv_opt[i]);
81 if (abs_diff > max_diff) {
82 max_diff = abs_diff;
83 }
84 }
85
86 free_aligned_buffer_page_end(dst_uv_c);
87 free_aligned_buffer_page_end(dst_uv_opt);
88 free_aligned_buffer_page_end(src_uv);
89 return max_diff;
90 }
91
92 // The following adjustments in dimensions ensure the scale factor will be
93 // exactly achieved.
94 #define DX(x, nom, denom) static_cast<int>((Abs(x) / nom) * nom)
95 #define SX(x, nom, denom) static_cast<int>((x / nom) * denom)
96
97 #define TEST_FACTOR1(name, filter, nom, denom) \
98 TEST_F(LibYUVScaleTest, UVScaleDownBy##name##_##filter) { \
99 int diff = UVTestFilter( \
100 SX(benchmark_width_, nom, denom), SX(benchmark_height_, nom, denom), \
101 DX(benchmark_width_, nom, denom), DX(benchmark_height_, nom, denom), \
102 kFilter##filter, benchmark_iterations_, disable_cpu_flags_, \
103 benchmark_cpu_info_); \
104 EXPECT_EQ(0, diff); \
105 }
106
107 #if defined(ENABLE_FULL_TESTS)
108 // Test a scale factor with all 4 filters. Expect exact for SIMD vs C.
109 #define TEST_FACTOR(name, nom, denom) \
110 TEST_FACTOR1(name, None, nom, denom) \
111 TEST_FACTOR1(name, Linear, nom, denom) \
112 TEST_FACTOR1(name, Bilinear, nom, denom) \
113 TEST_FACTOR1(name, Box, nom, denom)
114 #else
115 // Test a scale factor with Bilinear.
116 #define TEST_FACTOR(name, nom, denom) TEST_FACTOR1(name, Bilinear, nom, denom)
117 #endif
118
119 TEST_FACTOR(2, 1, 2)
120 TEST_FACTOR(4, 1, 4)
121 // TEST_FACTOR(8, 1, 8) Disable for benchmark performance.
122 TEST_FACTOR(3by4, 3, 4)
123 TEST_FACTOR(3by8, 3, 8)
124 TEST_FACTOR(3, 1, 3)
125 #undef TEST_FACTOR1
126 #undef TEST_FACTOR
127 #undef SX
128 #undef DX
129
130 #define TEST_SCALETO1(name, width, height, filter, max_diff) \
131 TEST_F(LibYUVScaleTest, name##To##width##x##height##_##filter) { \
132 int diff = UVTestFilter(benchmark_width_, benchmark_height_, width, \
133 height, kFilter##filter, benchmark_iterations_, \
134 disable_cpu_flags_, benchmark_cpu_info_); \
135 EXPECT_LE(diff, max_diff); \
136 } \
137 TEST_F(LibYUVScaleTest, name##From##width##x##height##_##filter) { \
138 int diff = UVTestFilter(width, height, Abs(benchmark_width_), \
139 Abs(benchmark_height_), kFilter##filter, \
140 benchmark_iterations_, disable_cpu_flags_, \
141 benchmark_cpu_info_); \
142 EXPECT_LE(diff, max_diff); \
143 }
144
145 #if defined(ENABLE_FULL_TESTS)
146 /// Test scale to a specified size with all 4 filters.
147 #define TEST_SCALETO(name, width, height) \
148 TEST_SCALETO1(name, width, height, None, 0) \
149 TEST_SCALETO1(name, width, height, Linear, 3) \
150 TEST_SCALETO1(name, width, height, Bilinear, 3)
151 #else
152 #define TEST_SCALETO(name, width, height) \
153 TEST_SCALETO1(name, width, height, Bilinear, 3)
154 #endif
155
156 TEST_SCALETO(UVScale, 1, 1)
157 TEST_SCALETO(UVScale, 569, 480)
158 TEST_SCALETO(UVScale, 640, 360)
159 #ifndef DISABLE_SLOW_TESTS
160 TEST_SCALETO(UVScale, 256, 144) /* 128x72 * 2 */
161 TEST_SCALETO(UVScale, 320, 240)
162 TEST_SCALETO(UVScale, 1280, 720)
163 TEST_SCALETO(UVScale, 1920, 1080)
164 #endif // DISABLE_SLOW_TESTS
165 #undef TEST_SCALETO1
166 #undef TEST_SCALETO
167
168 #define TEST_SCALESWAPXY1(name, filter, max_diff) \
169 TEST_F(LibYUVScaleTest, name##SwapXY_##filter) { \
170 int diff = \
171 UVTestFilter(benchmark_width_, benchmark_height_, benchmark_height_, \
172 benchmark_width_, kFilter##filter, benchmark_iterations_, \
173 disable_cpu_flags_, benchmark_cpu_info_); \
174 EXPECT_LE(diff, max_diff); \
175 }
176
177 #if defined(ENABLE_FULL_TESTS)
178 // Test scale with swapped width and height with all 3 filters.
179 TEST_SCALESWAPXY1(UVScale, None, 0)
180 TEST_SCALESWAPXY1(UVScale, Linear, 0)
181 TEST_SCALESWAPXY1(UVScale, Bilinear, 0)
182 #else
183 TEST_SCALESWAPXY1(UVScale, Bilinear, 0)
184 #endif
185 #undef TEST_SCALESWAPXY1
186
TEST_F(LibYUVScaleTest,UVTest3x)187 TEST_F(LibYUVScaleTest, UVTest3x) {
188 const int kSrcStride = 480 * 2;
189 const int kDstStride = 160 * 2;
190 const int kSize = kSrcStride * 3;
191 align_buffer_page_end(orig_pixels, kSize);
192 for (int i = 0; i < 480 * 3; ++i) {
193 orig_pixels[i * 2 + 0] = i;
194 orig_pixels[i * 2 + 1] = 255 - i;
195 }
196 align_buffer_page_end(dest_pixels, kDstStride);
197
198 int iterations160 = (benchmark_width_ * benchmark_height_ + (160 - 1)) / 160 *
199 benchmark_iterations_;
200 for (int i = 0; i < iterations160; ++i) {
201 UVScale(orig_pixels, kSrcStride, 480, 3, dest_pixels, kDstStride, 160, 1,
202 kFilterBilinear);
203 }
204
205 EXPECT_EQ(225, dest_pixels[0]);
206 EXPECT_EQ(255 - 225, dest_pixels[1]);
207
208 UVScale(orig_pixels, kSrcStride, 480, 3, dest_pixels, kDstStride, 160, 1,
209 kFilterNone);
210
211 EXPECT_EQ(225, dest_pixels[0]);
212 EXPECT_EQ(255 - 225, dest_pixels[1]);
213
214 free_aligned_buffer_page_end(dest_pixels);
215 free_aligned_buffer_page_end(orig_pixels);
216 }
217
TEST_F(LibYUVScaleTest,UVTest4x)218 TEST_F(LibYUVScaleTest, UVTest4x) {
219 const int kSrcStride = 640 * 2;
220 const int kDstStride = 160 * 2;
221 const int kSize = kSrcStride * 4;
222 align_buffer_page_end(orig_pixels, kSize);
223 for (int i = 0; i < 640 * 4; ++i) {
224 orig_pixels[i * 2 + 0] = i;
225 orig_pixels[i * 2 + 1] = 255 - i;
226 }
227 align_buffer_page_end(dest_pixels, kDstStride);
228
229 int iterations160 = (benchmark_width_ * benchmark_height_ + (160 - 1)) / 160 *
230 benchmark_iterations_;
231 for (int i = 0; i < iterations160; ++i) {
232 UVScale(orig_pixels, kSrcStride, 640, 4, dest_pixels, kDstStride, 160, 1,
233 kFilterBilinear);
234 }
235
236 EXPECT_EQ(66, dest_pixels[0]);
237 EXPECT_EQ(190, dest_pixels[1]);
238
239 UVScale(orig_pixels, kSrcStride, 64, 4, dest_pixels, kDstStride, 16, 1,
240 kFilterNone);
241
242 EXPECT_EQ(2, dest_pixels[0]); // expect the 3rd pixel of the 3rd row
243 EXPECT_EQ(255 - 2, dest_pixels[1]);
244
245 free_aligned_buffer_page_end(dest_pixels);
246 free_aligned_buffer_page_end(orig_pixels);
247 }
248
249 } // namespace libyuv
250