// Unit tests for WhiteBalanceEffect.
-#include "test_util.h"
+#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,
};
float out_data[5 * 4];
- EffectChainTester tester(data, 1, 5, FORMAT_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+ 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);
};
float out_data[3 * 4];
- EffectChainTester tester(data, 1, 3, FORMAT_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+ 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);
};
float out_data[2 * 4];
- EffectChainTester tester(data, 1, 2, FORMAT_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+ 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);
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