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