--- /dev/null
+// Unit tests for MixEffect.
+
+#include "test_util.h"
+#include "gtest/gtest.h"
+#include "mix_effect.h"
+
+TEST(MixEffectTest, FiftyFiftyMix) {
+ float data_a[] = {
+ 0.0f, 0.25f,
+ 0.75f, 1.0f,
+ };
+ float data_b[] = {
+ 1.0f, 0.5f,
+ 0.75f, 0.6f,
+ };
+ float expected_data[] = {
+ 0.5f, 0.375f,
+ 0.75f, 0.8f,
+ };
+ float out_data[4];
+ EffectChainTester tester(data_a, 2, 2, COLORSPACE_sRGB, GAMMA_LINEAR);
+ Effect *input1 = tester.get_chain()->last_added_effect();
+ Effect *input2 = tester.add_input(data_b, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ Effect *mix_effect = tester.get_chain()->add_effect(new MixEffect(), input1, input2);
+ ASSERT_TRUE(mix_effect->set_float("strength_first", 0.5f));
+ ASSERT_TRUE(mix_effect->set_float("strength_second", 0.5f));
+ tester.run(out_data, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ expect_equal(expected_data, out_data, 2, 2);
+}
+
+TEST(MixEffectTest, OnlyA) {
+ float data_a[] = {
+ 0.0f, 0.25f,
+ 0.75f, 1.0f,
+ };
+ float data_b[] = {
+ 1.0f, 0.5f,
+ 0.75f, 0.6f,
+ };
+ float out_data[4];
+ EffectChainTester tester(data_a, 2, 2, COLORSPACE_sRGB, GAMMA_LINEAR);
+ Effect *input1 = tester.get_chain()->last_added_effect();
+ Effect *input2 = tester.add_input(data_b, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ Effect *mix_effect = tester.get_chain()->add_effect(new MixEffect(), input1, input2);
+ ASSERT_TRUE(mix_effect->set_float("strength_first", 1.0f));
+ ASSERT_TRUE(mix_effect->set_float("strength_second", 0.0f));
+ tester.run(out_data, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ expect_equal(data_a, out_data, 2, 2);
+}
+
+TEST(MixEffectTest, DoesNotSumToOne) {
+ float data_a[] = {
+ 1.0f, 0.5f,
+ 0.75f, 1.0f,
+ };
+ float data_b[] = {
+ 1.0f, 0.25f,
+ 0.15f, 0.6f,
+ };
+ float expected_data[] = {
+ 0.0f, 0.25f,
+ 0.6f, 0.4f,
+ };
+ float out_data[4];
+ EffectChainTester tester(data_a, 2, 2, COLORSPACE_sRGB, GAMMA_LINEAR);
+ Effect *input1 = tester.get_chain()->last_added_effect();
+ Effect *input2 = tester.add_input(data_b, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ Effect *mix_effect = tester.get_chain()->add_effect(new MixEffect(), input1, input2);
+ ASSERT_TRUE(mix_effect->set_float("strength_first", 1.0f));
+ ASSERT_TRUE(mix_effect->set_float("strength_second", -1.0f));
+ tester.run(out_data, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ expect_equal(expected_data, out_data, 2, 2);
+}
+
+TEST(MixEffectTest, MixesLinearlyDespitesRGBInputsAndOutputs) {
+ float data_a[] = {
+ 0.0f, 0.25f,
+ 0.75f, 1.0f,
+ };
+ float data_b[] = {
+ 0.0f, 0.0f,
+ 0.0f, 0.0f,
+ };
+ float expected_data[] = { // sRGB(0.5 * inv_sRGB(a)).
+ 0.00000f, 0.17349f,
+ 0.54807f, 0.73536f,
+ };
+ float out_data[4];
+ EffectChainTester tester(data_a, 2, 2, COLORSPACE_sRGB, GAMMA_sRGB);
+ Effect *input1 = tester.get_chain()->last_added_effect();
+ Effect *input2 = tester.add_input(data_b, COLORSPACE_sRGB, GAMMA_sRGB);
+
+ Effect *mix_effect = tester.get_chain()->add_effect(new MixEffect(), input1, input2);
+ ASSERT_TRUE(mix_effect->set_float("strength_first", 0.5f));
+ ASSERT_TRUE(mix_effect->set_float("strength_second", 0.5f));
+ tester.run(out_data, COLORSPACE_sRGB, GAMMA_sRGB);
+
+ expect_equal(expected_data, out_data, 2, 2);
+}
EffectChainTester::EffectChainTester(const float *data, unsigned width, unsigned height, ColorSpace color_space, GammaCurve gamma_curve)
: chain(width, height), width(width), height(height)
+{
+ add_input(data, color_space, gamma_curve);
+}
+
+Input *EffectChainTester::add_input(const float *data, ColorSpace color_space, GammaCurve gamma_curve)
{
ImageFormat format;
format.color_space = color_space;
FlatInput *input = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, width, height);
input->set_pixel_data(data);
chain.add_input(input);
+ return input;
}
void EffectChainTester::run(float *out_data, ColorSpace color_space, GammaCurve gamma_curve)