]> git.sesse.net Git - movit/blob - white_balance_effect_test.cpp
Fix the uniform name for the compute shader output. (It would always be 0 by accident...
[movit] / white_balance_effect_test.cpp
1 // Unit tests for WhiteBalanceEffect.
2
3 #include <epoxy/gl.h>
4
5 #include "effect_chain.h"
6 #include "gtest/gtest.h"
7 #include "image_format.h"
8 #include "test_util.h"
9 #include "white_balance_effect.h"
10
11 namespace movit {
12
13 TEST(WhiteBalanceEffectTest, GrayNeutralDoesNothing) {
14         float data[] = {
15                 0.0f, 0.0f, 0.0f, 1.0f,
16                 0.5f, 0.5f, 0.5f, 0.3f,
17                 1.0f, 0.0f, 0.0f, 1.0f,
18                 0.0f, 1.0f, 0.0f, 0.7f,
19                 0.0f, 0.0f, 1.0f, 1.0f,
20         };
21         float neutral[] = {
22                 0.5f, 0.5f, 0.5f
23         };
24
25         float out_data[5 * 4];
26         EffectChainTester tester(data, 1, 5, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
27         Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
28         ASSERT_TRUE(white_balance_effect->set_vec3("neutral_color", neutral));
29         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
30
31         expect_equal(data, out_data, 4, 5);
32 }
33
34 TEST(WhiteBalanceEffectTest, SettingReddishNeutralColorNeutralizesReddishColor) {
35         float data[] = {
36                 0.0f, 0.0f, 0.0f, 1.0f,
37                 0.6f, 0.5f, 0.5f, 1.0f,
38                 0.5f, 0.5f, 0.5f, 1.0f,
39         };
40         float neutral[] = {
41                 0.6f, 0.5f, 0.5f
42         };
43
44         float out_data[3 * 4];
45         EffectChainTester tester(data, 1, 3, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
46         Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
47         ASSERT_TRUE(white_balance_effect->set_vec3("neutral_color", neutral));
48         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
49
50         // Black should stay black.
51         EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 0]);
52         EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 1]);
53         EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 2]);
54         EXPECT_FLOAT_EQ(1.0f, out_data[4 * 0 + 3]);
55
56         // The neutral color should now have R=G=B.
57         EXPECT_NEAR(out_data[4 * 1 + 0], out_data[4 * 1 + 1], 0.001);
58         EXPECT_NEAR(out_data[4 * 1 + 0], out_data[4 * 1 + 2], 0.001);
59         EXPECT_FLOAT_EQ(1.0f, out_data[4 * 1 + 3]);
60
61         // It should also have kept its luminance.
62         float old_luminance = 0.2126 * data[4 * 1 + 0] +
63                 0.7152 * data[4 * 1 + 1] +
64                 0.0722 * data[4 * 1 + 2];       
65         float new_luminance = 0.2126 * out_data[4 * 1 + 0] +
66                 0.7152 * out_data[4 * 1 + 1] +
67                 0.0722 * out_data[4 * 1 + 2];
68         EXPECT_NEAR(old_luminance, new_luminance, 0.001);
69
70         // Finally, the old gray should now have significantly less red than green and blue.
71         EXPECT_GT(out_data[4 * 2 + 1] - out_data[4 * 2 + 0], 0.05);
72         EXPECT_GT(out_data[4 * 2 + 2] - out_data[4 * 2 + 0], 0.05);
73 }
74
75 TEST(WhiteBalanceEffectTest, HigherColorTemperatureIncreasesBlue) {
76         float data[] = {
77                 0.0f, 0.0f, 0.0f, 1.0f,
78                 0.5f, 0.5f, 0.5f, 1.0f,
79         };
80
81         float out_data[2 * 4];
82         EffectChainTester tester(data, 1, 2, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
83         Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
84         ASSERT_TRUE(white_balance_effect->set_float("output_color_temperature", 10000.0f));
85         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
86
87         // Black should stay black.
88         EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 0]);
89         EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 1]);
90         EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 2]);
91         EXPECT_FLOAT_EQ(1.0f, out_data[4 * 0 + 3]);
92
93         // The neutral color should have kept its luminance.
94         float old_luminance = 0.2126 * data[4 * 1 + 0] +
95                 0.7152 * data[4 * 1 + 1] +
96                 0.0722 * data[4 * 1 + 2];       
97         float new_luminance = 0.2126 * out_data[4 * 1 + 0] +
98                 0.7152 * out_data[4 * 1 + 1] +
99                 0.0722 * out_data[4 * 1 + 2];
100         EXPECT_NEAR(old_luminance, new_luminance, 0.001);
101
102         // It should also have significantly more blue then green,
103         // and significantly less red than green.
104         EXPECT_GT(out_data[4 * 1 + 2] - out_data[4 * 1 + 1], 0.05);
105         EXPECT_GT(out_data[4 * 1 + 1] - out_data[4 * 1 + 0], 0.05);
106 }
107
108 }  // namespace movit