Add a unit test for MixEffect.

[movit] / mix_effect_test.cpp

diff --git a/mix_effect_test.cpp b/mix_effect_test.cpp
new file mode 100644 (file)
index 0000000..cae4b28
--- /dev/null
+++ b/mix_effect_test.cpp
@@ -0,0 +1,105 @@
+// 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);
+}