X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=deinterlace_effect_test.cpp;h=f880f0fc2629cb131e2f68cab97185fccdd13882;hp=31bd363024bb22ef3db70b51a1a29608b3b29abc;hb=f34b1c36acd27944f00885edfc55363432bfec8e;hpb=c06f1c4cc39bbebe13fe8e42a9278a55b5d0a216

diff --git a/deinterlace_effect_test.cpp b/deinterlace_effect_test.cpp
index 31bd363..f880f0f 100644
--- a/deinterlace_effect_test.cpp
+++ b/deinterlace_effect_test.cpp
@@ -1,5 +1,8 @@
 // Unit tests for DeinterlaceEffect.
 
+#ifdef HAVE_BENCHMARK
+#include <benchmark/benchmark.h>
+#endif
 #include <epoxy/gl.h>
 
 #include <algorithm>
@@ -30,7 +33,7 @@ TEST(DeinterlaceTest, ConstantColor) {
 		0.3f, 0.3f,
 	};
 	float out_data[12];
-	EffectChainTester tester(NULL, 2, 6);
+	EffectChainTester tester(nullptr, 2, 6);
 	Effect *input1 = tester.add_input(data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, 2, 3);
 	Effect *input2 = tester.add_input(data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, 2, 3);
 	Effect *input3 = tester.add_input(data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, 2, 3);
@@ -74,9 +77,9 @@ TEST(DeinterlaceTest, VerticalInterpolation) {
 	// Set previous and next fields to something so big that all the temporal checks
 	// are effectively turned off.
 	fill(neg_blowout_data, neg_blowout_data + width * height, -100.0f);
-	fill(neg_blowout_data, pos_blowout_data + width * height,  100.0f);
+	fill(pos_blowout_data, pos_blowout_data + width * height,  100.0f);
 
-	EffectChainTester tester(NULL, width, height * 2);
+	EffectChainTester tester(nullptr, width, height * 2);
 	Effect *input1 = tester.add_input(neg_blowout_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 	Effect *input2 = tester.add_input(neg_blowout_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 	Effect *input3 = tester.add_input(data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
@@ -128,7 +131,7 @@ TEST(DeinterlaceTest, DiagonalInterpolation) {
 	fill(neg_blowout_data, neg_blowout_data + width * height, -100.0f);
 	fill(pos_blowout_data, pos_blowout_data + width * height,  100.0f);
 
-	EffectChainTester tester(NULL, width, height * 2);
+	EffectChainTester tester(nullptr, width, height * 2);
 	Effect *input1 = tester.add_input(neg_blowout_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 	Effect *input2 = tester.add_input(neg_blowout_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 	Effect *input3 = tester.add_input(data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
@@ -169,7 +172,7 @@ TEST(DeinterlaceTest, FlickerBox) {
 	float out_data[width * height * 2];
 
 	{
-		EffectChainTester tester(NULL, width, height * 2);
+		EffectChainTester tester(nullptr, width, height * 2);
 		Effect *white_input = tester.add_input(white_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 		Effect *black_input = tester.add_input(black_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 		Effect *deinterlace_effect = tester.get_chain()->add_effect(new DeinterlaceEffect(), white_input, black_input, white_input, black_input, white_input);
@@ -181,7 +184,7 @@ TEST(DeinterlaceTest, FlickerBox) {
 	}
 
 	{
-		EffectChainTester tester(NULL, width, height * 2);
+		EffectChainTester tester(nullptr, width, height * 2);
 		Effect *white_input = tester.add_input(white_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 		Effect *black_input = tester.add_input(black_data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, height);
 		Effect *deinterlace_effect = tester.get_chain()->add_effect(new DeinterlaceEffect(), white_input, black_input, white_input, black_input, white_input);
@@ -193,4 +196,47 @@ TEST(DeinterlaceTest, FlickerBox) {
 	}
 }
 
+#ifdef HAVE_BENCHMARK
+void BM_DeinterlaceEffect_Gray(benchmark::State &state)
+{
+	unsigned width = state.range(0), height = state.range(1);
+	unsigned field_height = height / 2;
+
+	float *field1 = new float[width * field_height];
+	float *field2 = new float[width * field_height];
+	float *field3 = new float[width * field_height];
+	float *field4 = new float[width * field_height];
+	float *field5 = new float[width * field_height];
+	float *out_data = new float[width * height];
+
+	for (unsigned i = 0; i < width * field_height; ++i) {
+		field1[i] = rand();
+		field2[i] = rand();
+		field3[i] = rand();
+		field4[i] = rand();
+		field5[i] = rand();
+	}
+
+	EffectChainTester tester(nullptr, width, height);
+	Effect *input1 = tester.add_input(field1, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input2 = tester.add_input(field2, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input3 = tester.add_input(field3, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input4 = tester.add_input(field4, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input5 = tester.add_input(field5, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *deinterlace_effect = tester.get_chain()->add_effect(new DeinterlaceEffect(), input1, input2, input3, input4, input5);
+
+	ASSERT_TRUE(deinterlace_effect->set_int("current_field_position", 0));
+
+	tester.benchmark(state, out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_PREMULTIPLIED);
+
+	delete[] field1;
+	delete[] field2;
+	delete[] field3;
+	delete[] field4;
+	delete[] field5;
+	delete[] out_data;
+}
+BENCHMARK(BM_DeinterlaceEffect_Gray)->Args({720, 576})->Args({1280, 720})->Args({1920, 1080})->UseRealTime()->Unit(benchmark::kMicrosecond);
+#endif
+
 }  // namespace movit