Add a unit test for FlatInput.

[movit] / flat_input_test.cpp

diff --git a/flat_input_test.cpp b/flat_input_test.cpp
new file mode 100644 (file)
index 0000000..17ee9ce
--- /dev/null
+++ b/flat_input_test.cpp
@@ -0,0 +1,194 @@
+// Unit tests for FlatInput.
+
+#include "test_util.h"
+#include "gtest/gtest.h"
+#include "flat_input.h"
+
+TEST(FlatInput, SimpleGrayscale) {
+       const int size = 4;
+
+       float data[size] = {
+               0.0,
+               0.5,
+               0.7,
+               1.0,
+       };
+       float expected_data[4 * size] = {
+               0.0, 0.0, 0.0, 1.0,
+               0.5, 0.5, 0.5, 1.0,
+               0.7, 0.7, 0.7, 1.0,
+               1.0, 1.0, 1.0, 1.0,
+       };
+       float out_data[4 * size];
+
+       EffectChainTester tester(data, 1, size, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
+       tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+       expect_equal(expected_data, out_data, 4, size);
+}
+
+TEST(FlatInput, RGB) {
+       const int size = 5;
+
+       float data[3 * size] = {
+               0.0, 0.0, 0.0,
+               0.5, 0.0, 0.0,
+               0.0, 0.5, 0.0,
+               0.0, 0.0, 0.7,
+               0.0, 0.3, 0.7,
+       };
+       float expected_data[4 * size] = {
+               0.0, 0.0, 0.0, 1.0,
+               0.5, 0.0, 0.0, 1.0,
+               0.0, 0.5, 0.0, 1.0,
+               0.0, 0.0, 0.7, 1.0,
+               0.0, 0.3, 0.7, 1.0,
+       };
+       float out_data[4 * size];
+
+       EffectChainTester tester(data, 1, size, FORMAT_RGB, COLORSPACE_sRGB, GAMMA_LINEAR);
+       tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+       expect_equal(expected_data, out_data, 4, size);
+}
+
+TEST(FlatInput, RGBA) {
+       const int size = 5;
+
+       float data[4 * size] = {
+               0.0, 0.0, 0.0, 1.0,
+               0.5, 0.0, 0.0, 0.3,
+               0.0, 0.5, 0.0, 0.7,
+               0.0, 0.0, 0.7, 1.0,
+               0.0, 0.3, 0.7, 0.2,
+       };
+       float expected_data[4 * size] = {
+               0.0, 0.0, 0.0, 1.0,
+               0.5, 0.0, 0.0, 0.3,
+               0.0, 0.5, 0.0, 0.7,
+               0.0, 0.0, 0.7, 1.0,
+               0.0, 0.3, 0.7, 0.2,
+       };
+       float out_data[4 * size];
+
+       EffectChainTester tester(data, 1, size, FORMAT_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+       tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+       expect_equal(expected_data, out_data, 4, size);
+}
+
+TEST(FlatInput, BGR) {
+       const int size = 5;
+
+       float data[3 * size] = {
+               0.0, 0.0, 0.0,
+               0.5, 0.0, 0.0,
+               0.0, 0.5, 0.0,
+               0.0, 0.0, 0.7,
+               0.0, 0.3, 0.7,
+       };
+       float expected_data[4 * size] = {
+               0.0, 0.0, 0.0, 1.0,
+               0.0, 0.0, 0.5, 1.0,
+               0.0, 0.5, 0.0, 1.0,
+               0.7, 0.0, 0.0, 1.0,
+               0.7, 0.3, 0.0, 1.0,
+       };
+       float out_data[4 * size];
+
+       EffectChainTester tester(data, 1, size, FORMAT_BGR, COLORSPACE_sRGB, GAMMA_LINEAR);
+       tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+       expect_equal(expected_data, out_data, 4, size);
+}
+
+TEST(FlatInput, BGRA) {
+       const int size = 5;
+
+       float data[4 * size] = {
+               0.0, 0.0, 0.0, 1.0,
+               0.5, 0.0, 0.0, 0.3,
+               0.0, 0.5, 0.0, 0.7,
+               0.0, 0.0, 0.7, 1.0,
+               0.0, 0.3, 0.7, 0.2,
+       };
+       float expected_data[4 * size] = {
+               0.0, 0.0, 0.0, 1.0,
+               0.0, 0.0, 0.5, 0.3,
+               0.0, 0.5, 0.0, 0.7,
+               0.7, 0.0, 0.0, 1.0,
+               0.7, 0.3, 0.0, 0.2,
+       };
+       float out_data[4 * size];
+
+       EffectChainTester tester(data, 1, size, FORMAT_BGRA, COLORSPACE_sRGB, GAMMA_LINEAR);
+       tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+       expect_equal(expected_data, out_data, 4, size);
+}
+
+TEST(FlatInput, Pitch) {
+       const int pitch = 3;
+       const int width = 2;
+       const int height = 4;
+
+       float data[pitch * height] = {
+               0.0, 1.0, 999.0f,
+               0.5, 0.5, 999.0f,
+               0.7, 0.2, 999.0f,
+               1.0, 0.6, 999.0f,
+       };
+       float expected_data[4 * width * height] = {
+               0.0, 0.0, 0.0, 1.0,  1.0, 1.0, 1.0, 1.0,
+               0.5, 0.5, 0.5, 1.0,  0.5, 0.5, 0.5, 1.0,
+               0.7, 0.7, 0.7, 1.0,  0.2, 0.2, 0.2, 1.0,
+               1.0, 1.0, 1.0, 1.0,  0.6, 0.6, 0.6, 1.0,
+       };
+       float out_data[4 * width * height];
+
+       EffectChainTester tester(NULL, width, height);
+
+       ImageFormat format;
+       format.color_space = COLORSPACE_sRGB;
+       format.gamma_curve = GAMMA_LINEAR;
+
+       FlatInput *input = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, width, height);
+       input->set_pitch(pitch);
+       input->set_pixel_data(data);
+       tester.get_chain()->add_input(input);
+
+       tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+       expect_equal(expected_data, out_data, 4 * width, height);
+}
+
+TEST(FlatInput, UpdatedData) {
+       const int width = 2;
+       const int height = 4;
+
+       float data[width * height] = {
+               0.0, 1.0,
+               0.5, 0.5,
+               0.7, 0.2,
+               1.0, 0.6,
+       };
+       float out_data[width * height];
+
+       EffectChainTester tester(NULL, width, height);
+
+       ImageFormat format;
+       format.color_space = COLORSPACE_sRGB;
+       format.gamma_curve = GAMMA_LINEAR;
+
+       FlatInput *input = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, width, height);
+       input->set_pixel_data(data);
+       tester.get_chain()->add_input(input);
+
+       tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+       expect_equal(data, out_data, width, height);
+
+       data[6] = 0.3;
+       input->invalidate_pixel_data();
+
+       tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+       expect_equal(data, out_data, width, height);
+}