1 // Unit tests for WhiteBalanceEffect.
5 #include "effect_chain.h"
6 #include "gtest/gtest.h"
7 #include "image_format.h"
9 #include "white_balance_effect.h"
11 TEST(WhiteBalanceEffectTest, GrayNeutralDoesNothing) {
13 0.0f, 0.0f, 0.0f, 1.0f,
14 0.5f, 0.5f, 0.5f, 0.3f,
15 1.0f, 0.0f, 0.0f, 1.0f,
16 0.0f, 1.0f, 0.0f, 0.7f,
17 0.0f, 0.0f, 1.0f, 1.0f,
23 float out_data[5 * 4];
24 EffectChainTester tester(data, 1, 5, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
25 Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
26 ASSERT_TRUE(white_balance_effect->set_vec3("neutral_color", neutral));
27 tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
29 expect_equal(data, out_data, 4, 5);
32 TEST(WhiteBalanceEffectTest, SettingReddishNeutralColorNeutralizesReddishColor) {
34 0.0f, 0.0f, 0.0f, 1.0f,
35 0.6f, 0.5f, 0.5f, 1.0f,
36 0.5f, 0.5f, 0.5f, 1.0f,
42 float out_data[3 * 4];
43 EffectChainTester tester(data, 1, 3, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
44 Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
45 ASSERT_TRUE(white_balance_effect->set_vec3("neutral_color", neutral));
46 tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
48 // Black should stay black.
49 EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 0]);
50 EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 1]);
51 EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 2]);
52 EXPECT_FLOAT_EQ(1.0f, out_data[4 * 0 + 3]);
54 // The neutral color should now have R=G=B.
55 EXPECT_NEAR(out_data[4 * 1 + 0], out_data[4 * 1 + 1], 0.001);
56 EXPECT_NEAR(out_data[4 * 1 + 0], out_data[4 * 1 + 2], 0.001);
57 EXPECT_FLOAT_EQ(1.0f, out_data[4 * 1 + 3]);
59 // It should also have kept its luminance.
60 float old_luminance = 0.2126 * data[4 * 1 + 0] +
61 0.7152 * data[4 * 1 + 1] +
62 0.0722 * data[4 * 1 + 2];
63 float new_luminance = 0.2126 * out_data[4 * 1 + 0] +
64 0.7152 * out_data[4 * 1 + 1] +
65 0.0722 * out_data[4 * 1 + 2];
66 EXPECT_NEAR(old_luminance, new_luminance, 0.001);
68 // Finally, the old gray should now have significantly less red than green and blue.
69 EXPECT_GT(out_data[4 * 2 + 1] - out_data[4 * 2 + 0], 0.05);
70 EXPECT_GT(out_data[4 * 2 + 2] - out_data[4 * 2 + 0], 0.05);
73 TEST(WhiteBalanceEffectTest, HigherColorTemperatureIncreasesBlue) {
75 0.0f, 0.0f, 0.0f, 1.0f,
76 0.5f, 0.5f, 0.5f, 1.0f,
79 float out_data[2 * 4];
80 EffectChainTester tester(data, 1, 2, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
81 Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
82 ASSERT_TRUE(white_balance_effect->set_float("output_color_temperature", 10000.0f));
83 tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
85 // Black should stay black.
86 EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 0]);
87 EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 1]);
88 EXPECT_FLOAT_EQ(0.0f, out_data[4 * 0 + 2]);
89 EXPECT_FLOAT_EQ(1.0f, out_data[4 * 0 + 3]);
91 // The neutral color should have kept its luminance.
92 float old_luminance = 0.2126 * data[4 * 1 + 0] +
93 0.7152 * data[4 * 1 + 1] +
94 0.0722 * data[4 * 1 + 2];
95 float new_luminance = 0.2126 * out_data[4 * 1 + 0] +
96 0.7152 * out_data[4 * 1 + 1] +
97 0.0722 * out_data[4 * 1 + 2];
98 EXPECT_NEAR(old_luminance, new_luminance, 0.001);
100 // It should also have significantly more blue then green,
101 // and significantly less red than green.
102 EXPECT_GT(out_data[4 * 1 + 2] - out_data[4 * 1 + 1], 0.05);
103 EXPECT_GT(out_data[4 * 1 + 1] - out_data[4 * 1 + 0], 0.05);