X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=mix_effect_test.cpp;fp=mix_effect_test.cpp;h=cae4b28795ac078e7c7ce525e65bf4e35836f067;hb=f52ae2984b179e8d91e3d78ecc72888644fce406;hp=0000000000000000000000000000000000000000;hpb=58094bb7c80300a2f67a7558d755abf5705c57df;p=movit

diff --git a/mix_effect_test.cpp b/mix_effect_test.cpp
new file mode 100644
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);
+}