]> git.sesse.net Git - movit/blob - ycbcr_422interleaved_input_test.cpp
Fix multiply-defined symbols with multiple YCbCrInputs in the same chain.
[movit] / ycbcr_422interleaved_input_test.cpp
1 // Unit tests for YCbCr422InterleavedInput.
2
3 #include <epoxy/gl.h>
4 #include <stddef.h>
5
6 #include "effect_chain.h"
7 #include "gtest/gtest.h"
8 #include "test_util.h"
9 #include "util.h"
10 #include "resize_effect.h"
11 #include "ycbcr_422interleaved_input.h"
12
13 namespace movit {
14
15 // Adapted from the Simple444 test from YCbCrInputTest.
16 TEST(YCbCr422InterleavedInputTest, Simple422) {
17         const int width = 2;
18         const int height = 5;
19
20         // Pure-color test inputs, calculated with the formulas in Rec. 601
21         // section 2.5.4.
22         unsigned char uyvy[width * height * 2] = {
23                 /*U=*/128, /*Y=*/ 16, /*V=*/128, /*Y=*/ 16,
24                 /*U=*/128, /*Y=*/235, /*V=*/128, /*Y=*/235,
25                 /*U=*/ 90, /*Y=*/ 81, /*V=*/240, /*Y=*/ 81,
26                 /*U=*/ 54, /*Y=*/145, /*V=*/ 34, /*Y=*/145,
27                 /*U=*/240, /*Y=*/ 41, /*V=*/110, /*Y=*/ 41,
28         };
29
30         float expected_data[4 * width * height] = {
31                 0.0, 0.0, 0.0, 1.0,   0.0, 0.0, 0.0, 1.0,
32                 1.0, 1.0, 1.0, 1.0,   1.0, 1.0, 1.0, 1.0,
33                 1.0, 0.0, 0.0, 1.0,   1.0, 0.0, 0.0, 1.0,
34                 0.0, 1.0, 0.0, 1.0,   0.0, 1.0, 0.0, 1.0,
35                 0.0, 0.0, 1.0, 1.0,   0.0, 0.0, 1.0, 1.0,
36         };
37         float out_data[4 * width * height];
38
39         EffectChainTester tester(NULL, width, height);
40
41         ImageFormat format;
42         format.color_space = COLORSPACE_sRGB;
43         format.gamma_curve = GAMMA_sRGB;
44
45         YCbCrFormat ycbcr_format;
46         ycbcr_format.luma_coefficients = YCBCR_REC_601;
47         ycbcr_format.full_range = false;
48         ycbcr_format.num_levels = 256;
49         ycbcr_format.chroma_subsampling_x = 2;
50         ycbcr_format.chroma_subsampling_y = 1;
51         ycbcr_format.cb_x_position = 0.0f;  // Doesn't really matter here, since Y is constant.
52         ycbcr_format.cb_y_position = 0.5f;
53         ycbcr_format.cr_x_position = 0.0f;
54         ycbcr_format.cr_y_position = 0.5f;
55
56         YCbCr422InterleavedInput *input = new YCbCr422InterleavedInput(format, ycbcr_format, width, height);
57         input->set_pixel_data(uyvy);
58         tester.get_chain()->add_input(input);
59
60         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_sRGB);
61
62         // Y'CbCr isn't 100% accurate (the input values are rounded),
63         // so we need some leeway.
64         expect_equal(expected_data, out_data, 4 * width, height, 0.025, 0.002);
65 }
66
67 TEST(YCbCr422InterleavedInputTest, LumaLinearInterpolation) {
68         const int width = 4;
69         const int height = 1;
70         const int out_width = width * 3;
71
72         // Black, white, black and then gray.
73         unsigned char uyvy[width * height * 2] = {
74                 /*U=*/128, /*Y=*/ 16,
75                 /*V=*/128, /*Y=*/235,
76                 /*U=*/128, /*Y=*/ 16,
77                 /*V=*/128, /*Y=*/128,
78         };
79
80         float expected_data[out_width * height] = {
81                 0.0, /**/0.0, 0.333, 0.667, /**/1.0, 0.667, 0.333, /**/0.0, 0.167, 0.333, /**/0.5, 0.5
82         };
83         float out_data[out_width * height];
84
85         EffectChainTester tester(NULL, out_width, height);
86
87         ImageFormat format;
88         format.color_space = COLORSPACE_sRGB;
89         format.gamma_curve = GAMMA_sRGB;
90
91         YCbCrFormat ycbcr_format;
92         ycbcr_format.luma_coefficients = YCBCR_REC_601;
93         ycbcr_format.full_range = false;
94         ycbcr_format.num_levels = 256;
95         ycbcr_format.chroma_subsampling_x = 2;
96         ycbcr_format.chroma_subsampling_y = 1;
97         ycbcr_format.cb_x_position = 0.0f;  // Doesn't really matter here, since U/V are constant.
98         ycbcr_format.cb_y_position = 0.5f;
99         ycbcr_format.cr_x_position = 0.0f;
100         ycbcr_format.cr_y_position = 0.5f;
101
102         YCbCr422InterleavedInput *input = new YCbCr422InterleavedInput(format, ycbcr_format, width, height);
103         input->set_pixel_data(uyvy);
104         tester.get_chain()->add_input(input);
105
106         ResizeEffect *upscale = new ResizeEffect();
107         ASSERT_TRUE(upscale->set_int("width", out_width));
108         ASSERT_TRUE(upscale->set_int("height", height));
109         tester.get_chain()->add_effect(upscale);
110
111         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_sRGB);
112
113         // Y'CbCr isn't 100% accurate (the input values are rounded),
114         // so we need some leeway.
115         expect_equal(expected_data, out_data, out_width, height, 0.025, 0.002);
116 }
117
118 // Adapted from the YCbCrInput test of the same name.
119 TEST(YCbCr422InterleavedInputTest, DifferentCbAndCrPositioning) {
120         const int width = 4;
121         const int height = 4;
122
123         unsigned char uyvy[width * height * 2] = {
124                 /*U=*/ 64, /*Y=*/126, /*V=*/ 48, /*Y=*/126,  /*U=*/128, /*Y=*/126, /*V=*/128, /*Y=*/126,
125                 /*U=*/128, /*Y=*/126, /*V=*/128, /*Y=*/126,  /*U=*/192, /*Y=*/126, /*V=*/208, /*Y=*/126,
126                 /*U=*/128, /*Y=*/126, /*V=*/128, /*Y=*/126,  /*U=*/128, /*Y=*/126, /*V=*/128, /*Y=*/126,
127                 /*U=*/128, /*Y=*/126, /*V=*/128, /*Y=*/126,  /*U=*/128, /*Y=*/126, /*V=*/128, /*Y=*/126,
128         };
129
130         // Chroma samples in this case are always co-sited with a luma sample;
131         // their associated color values and position are marked off in comments.
132         float expected_data_blue[width * height] = {
133                    0.000 /* 0.0 */, 0.250,           0.500 /* 0.5 */, 0.500, 
134                    0.500 /* 0.5 */, 0.750,           1.000 /* 1.0 */, 1.000, 
135                    0.500 /* 0.5 */, 0.500,           0.500 /* 0.5 */, 0.500, 
136                    0.500 /* 0.5 */, 0.500,           0.500 /* 0.5 */, 0.500, 
137         };
138         float expected_data_red[width * height] = {
139                    0.000,           0.000 /* 0.0 */, 0.250,           0.500 /* 0.5 */, 
140                    0.500,           0.500 /* 0.5 */, 0.750,           1.000 /* 1.0 */, 
141                    0.500,           0.500 /* 0.5 */, 0.500,           0.500 /* 0.5 */, 
142                    0.500,           0.500 /* 0.5 */, 0.500,           0.500 /* 0.5 */, 
143         };
144         float out_data[width * height];
145
146         EffectChainTester tester(NULL, width, height);
147
148         ImageFormat format;
149         format.color_space = COLORSPACE_sRGB;
150         format.gamma_curve = GAMMA_sRGB;
151
152         YCbCrFormat ycbcr_format;
153         ycbcr_format.luma_coefficients = YCBCR_REC_601;
154         ycbcr_format.full_range = false;
155         ycbcr_format.num_levels = 256;
156         ycbcr_format.chroma_subsampling_x = 2;
157         ycbcr_format.chroma_subsampling_y = 1;
158         ycbcr_format.cb_x_position = 0.0f;
159         ycbcr_format.cb_y_position = 0.5f;
160         ycbcr_format.cr_x_position = 1.0f;
161         ycbcr_format.cr_y_position = 0.5f;
162
163         YCbCr422InterleavedInput *input = new YCbCr422InterleavedInput(format, ycbcr_format, width, height);
164         input->set_pixel_data(uyvy);
165         tester.get_chain()->add_input(input);
166
167         // Y'CbCr isn't 100% accurate (the input values are rounded),
168         // so we need some leeway.
169         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_sRGB);
170         expect_equal(expected_data_red, out_data, width, height, 0.02, 0.002);
171
172         tester.run(out_data, GL_BLUE, COLORSPACE_sRGB, GAMMA_sRGB);
173         expect_equal(expected_data_blue, out_data, width, height, 0.01, 0.001);
174 }
175
176 TEST(YCbCr422InterleavedInputTest, PBO) {
177         const int width = 2;
178         const int height = 5;
179
180         // Pure-color test inputs, calculated with the formulas in Rec. 601
181         // section 2.5.4.
182         unsigned char uyvy[width * height * 2] = {
183                 /*U=*/128, /*Y=*/ 16, /*V=*/128, /*Y=*/ 16,
184                 /*U=*/128, /*Y=*/235, /*V=*/128, /*Y=*/235,
185                 /*U=*/ 90, /*Y=*/ 81, /*V=*/240, /*Y=*/ 81,
186                 /*U=*/ 54, /*Y=*/145, /*V=*/ 34, /*Y=*/145,
187                 /*U=*/240, /*Y=*/ 41, /*V=*/110, /*Y=*/ 41,
188         };
189
190         float expected_data[4 * width * height] = {
191                 0.0, 0.0, 0.0, 1.0,   0.0, 0.0, 0.0, 1.0,
192                 1.0, 1.0, 1.0, 1.0,   1.0, 1.0, 1.0, 1.0,
193                 1.0, 0.0, 0.0, 1.0,   1.0, 0.0, 0.0, 1.0,
194                 0.0, 1.0, 0.0, 1.0,   0.0, 1.0, 0.0, 1.0,
195                 0.0, 0.0, 1.0, 1.0,   0.0, 0.0, 1.0, 1.0,
196         };
197         float out_data[4 * width * height];
198
199         GLuint pbo;
200         glGenBuffers(1, &pbo);
201         glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, pbo);
202         glBufferData(GL_PIXEL_UNPACK_BUFFER_ARB, width * height * 2, uyvy, GL_STREAM_DRAW);
203         glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, 0);
204
205         EffectChainTester tester(NULL, width, height);
206
207         ImageFormat format;
208         format.color_space = COLORSPACE_sRGB;
209         format.gamma_curve = GAMMA_sRGB;
210
211         YCbCrFormat ycbcr_format;
212         ycbcr_format.luma_coefficients = YCBCR_REC_601;
213         ycbcr_format.full_range = false;
214         ycbcr_format.num_levels = 256;
215         ycbcr_format.chroma_subsampling_x = 2;
216         ycbcr_format.chroma_subsampling_y = 1;
217         ycbcr_format.cb_x_position = 0.0f;  // Doesn't really matter here, since Y is constant.
218         ycbcr_format.cb_y_position = 0.5f;
219         ycbcr_format.cr_x_position = 0.0f;
220         ycbcr_format.cr_y_position = 0.5f;
221
222         YCbCr422InterleavedInput *input = new YCbCr422InterleavedInput(format, ycbcr_format, width, height);
223         input->set_pixel_data((unsigned char *)BUFFER_OFFSET(0), pbo);
224         tester.get_chain()->add_input(input);
225
226         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_sRGB);
227
228         // Y'CbCr isn't 100% accurate (the input values are rounded),
229         // so we need some leeway.
230         expect_equal(expected_data, out_data, 4 * width, height, 0.025, 0.002);
231
232         glDeleteBuffers(1, &pbo);
233 }
234
235 }  // namespace movit