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

diff --git a/deinterlace_effect_test.cpp b/deinterlace_effect_test.cpp
index f880f0f..0c625c5 100644
--- a/deinterlace_effect_test.cpp
+++ b/deinterlace_effect_test.cpp
@@ -6,6 +6,7 @@
 #include <epoxy/gl.h>
 
 #include <algorithm>
+#include <memory>
 
 #include "effect_chain.h"
 #include "gtest/gtest.h"
@@ -197,19 +198,31 @@ TEST(DeinterlaceTest, FlickerBox) {
 }
 
 #ifdef HAVE_BENCHMARK
-void BM_DeinterlaceEffect_Gray(benchmark::State &state)
+namespace {
+
+struct TestFormat {
+	MovitPixelFormat input_format;
+	GLenum output_format;
+	size_t bytes_per_pixel;
+};
+TestFormat gray_format = { FORMAT_GRAYSCALE, GL_RED, 1 };
+TestFormat bgra_format = { FORMAT_BGRA_PREMULTIPLIED_ALPHA, GL_BGRA, 4 };
+
+}  // namespace
+
+void BM_DeinterlaceEffect(benchmark::State &state, TestFormat format, bool spatial_interlacing_check)
 {
 	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];
+	unique_ptr<float[]> field1(new float[width * field_height * format.bytes_per_pixel]);
+	unique_ptr<float[]> field2(new float[width * field_height * format.bytes_per_pixel]);
+	unique_ptr<float[]> field3(new float[width * field_height * format.bytes_per_pixel]);
+	unique_ptr<float[]> field4(new float[width * field_height * format.bytes_per_pixel]);
+	unique_ptr<float[]> field5(new float[width * field_height * format.bytes_per_pixel]);
+	unique_ptr<float[]> out_data(new float[width * height * format.bytes_per_pixel]);
 
-	for (unsigned i = 0; i < width * field_height; ++i) {
+	for (unsigned i = 0; i < width * field_height * format.bytes_per_pixel; ++i) {
 		field1[i] = rand();
 		field2[i] = rand();
 		field3[i] = rand();
@@ -218,25 +231,22 @@ void BM_DeinterlaceEffect_Gray(benchmark::State &state)
 	}
 
 	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 *input1 = tester.add_input(field1.get(), format.input_format, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input2 = tester.add_input(field2.get(), format.input_format, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input3 = tester.add_input(field3.get(), format.input_format, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input4 = tester.add_input(field4.get(), format.input_format, COLORSPACE_sRGB, GAMMA_LINEAR, width, field_height);
+	Effect *input5 = tester.add_input(field5.get(), format.input_format, 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));
+	ASSERT_TRUE(deinterlace_effect->set_int("enable_spatial_interlacing_check", spatial_interlacing_check));
 
-	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;
+	tester.benchmark(state, out_data.get(), format.output_format, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_PREMULTIPLIED);
 }
-BENCHMARK(BM_DeinterlaceEffect_Gray)->Args({720, 576})->Args({1280, 720})->Args({1920, 1080})->UseRealTime()->Unit(benchmark::kMicrosecond);
+BENCHMARK_CAPTURE(BM_DeinterlaceEffect, Gray, gray_format, true)->Args({720, 576})->Args({1280, 720})->Args({1920, 1080})->UseRealTime()->Unit(benchmark::kMicrosecond);
+BENCHMARK_CAPTURE(BM_DeinterlaceEffect, BGRA, bgra_format, true)->Args({720, 576})->Args({1280, 720})->Args({1920, 1080})->UseRealTime()->Unit(benchmark::kMicrosecond);
+BENCHMARK_CAPTURE(BM_DeinterlaceEffect, BGRANoSpatialCheck, bgra_format, false)->Args({720, 576})->Args({1280, 720})->Args({1920, 1080})->UseRealTime()->Unit(benchmark::kMicrosecond);
+
 #endif
 
 }  // namespace movit