]> git.sesse.net Git - movit/blob - complex_modulate_effect_test.cpp
Add some asserts to EffectChain::execute_phase, so that we do not inadvertedly insert...
[movit] / complex_modulate_effect_test.cpp
1 // Unit tests for ComplexModulateEffect.
2
3 #include <epoxy/gl.h>
4
5 #include "effect_chain.h"
6 #include "gtest/gtest.h"
7 #include "complex_modulate_effect.h"
8 #include "image_format.h"
9 #include "input.h"
10 #include "test_util.h"
11
12 namespace movit {
13
14 TEST(ComplexModulateEffectTest, Identity) {
15         const int size = 3;
16         float data_a[size * 4] = {
17                 0.0f, 0.1f, 0.2f, 0.1f,
18                 0.4f, 0.3f, 0.8f, 2.0f,
19                 0.5f, 0.2f, 0.1f, 0.0f,
20         };
21         float data_b[size * 2] = {
22                 1.0f, 0.0f,
23                 1.0f, 0.0f,
24                 1.0f, 0.0f,
25         };
26         float out_data[size * 4];
27
28         EffectChainTester tester(data_a, 1, size, FORMAT_RGBA_PREMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
29         Effect *input1 = tester.get_chain()->last_added_effect();
30         Effect *input2 = tester.add_input(data_b, FORMAT_RG, COLORSPACE_sRGB, GAMMA_LINEAR);
31
32         tester.get_chain()->add_effect(new ComplexModulateEffect(), input1, input2);
33         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_PREMULTIPLIED);
34
35         expect_equal(data_a, out_data, 4, size);
36 }
37
38 TEST(ComplexModulateEffectTest, ComplexMultiplication) {
39         const int size = 2;
40         float data_a[size * 4] = {
41                 0.0f, 0.1f, 0.2f, 0.1f,
42                 0.4f, 0.3f, 0.8f, 2.0f,
43         };
44         float data_b[size * 2] = {
45                 0.0f,  1.0f,
46                 0.5f, -0.8f,
47         };
48         float expected_data[size * 4] = {
49                 -0.1f,   0.0f,  -0.1f, 0.2f,
50                  0.44f, -0.17f,  2.0f, 0.36f,
51         };
52         float out_data[size * 4];
53
54         EffectChainTester tester(data_a, 1, size, FORMAT_RGBA_PREMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
55         Effect *input1 = tester.get_chain()->last_added_effect();
56         Effect *input2 = tester.add_input(data_b, FORMAT_RG, COLORSPACE_sRGB, GAMMA_LINEAR);
57
58         tester.get_chain()->add_effect(new ComplexModulateEffect(), input1, input2);
59         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_PREMULTIPLIED);
60
61         expect_equal(expected_data, out_data, 4, size);
62 }
63
64 TEST(ComplexModulateEffectTest, Repeat) {
65         const int size = 2, repeats = 3;
66         float data_a[size * repeats * 4] = {
67                 0.0f, 0.1f, 0.2f, 0.3f,
68                 1.0f, 1.1f, 1.2f, 1.3f,
69                 2.0f, 2.1f, 2.2f, 2.3f,
70                 3.0f, 3.1f, 3.2f, 3.3f,
71                 4.0f, 4.1f, 4.2f, 4.3f,
72                 5.0f, 5.1f, 5.2f, 5.3f,
73         };
74         float data_b[size * 2] = {
75                 1.0f,  0.0f,
76                 0.0f, -1.0f,
77         };
78         float expected_data[size * repeats * 4] = {
79                 0.0f,  0.1f, 0.2f,  0.3f,
80                 1.1f, -1.0f, 1.3f, -1.2f,
81                 2.0f,  2.1f, 2.2f,  2.3f,
82                 3.1f, -3.0f, 3.3f, -3.2f,
83                 4.0f,  4.1f, 4.2f,  4.3f,
84                 5.1f, -5.0f, 5.3f, -5.2f,
85         };
86         float out_data[size * repeats * 4];
87
88         EffectChainTester tester(data_a, 1, repeats * size, FORMAT_RGBA_PREMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
89         Effect *input1 = tester.get_chain()->last_added_effect();
90         Effect *input2 = tester.add_input(data_b, FORMAT_RG, COLORSPACE_sRGB, GAMMA_LINEAR, 1, size);
91
92         Effect *effect = tester.get_chain()->add_effect(new ComplexModulateEffect(), input1, input2);
93         ASSERT_TRUE(effect->set_int("num_repeats_y", repeats));
94         tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_PREMULTIPLIED);
95
96         expect_equal(expected_data, out_data, 4, size * repeats);
97 }
98
99 }  // namespace movit