git.sesse.net Git - movit/blob - luma_mix_effect_test.cpp

   1 // Unit tests for LumaMixEffect.
   2
   3 #include <GL/glew.h>
   4
   5 #include "effect_chain.h"
   6 #include "gtest/gtest.h"
   7 #include "image_format.h"
   8 #include "input.h"
   9 #include "luma_mix_effect.h"
  10 #include "test_util.h"
  11
  12 namespace movit {
  13
  14 TEST(LumaMixEffectTest, HardWipe) {
  15         float data_a[] = {
  16                 0.0f, 0.25f,
  17                 0.75f, 1.0f,
  18         };
  19         float data_b[] = {
  20                 1.0f, 0.5f,
  21                 0.65f, 0.6f,
  22         };
  23         float data_luma[] = {
  24                 0.0f, 0.25f,
  25                 0.5f, 0.75f,
  26         };
  27
  28         EffectChainTester tester(data_a, 2, 2, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
  29         Effect *input1 = tester.get_chain()->last_added_effect();
  30         Effect *input2 = tester.add_input(data_b, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
  31         Effect *input3 = tester.add_input(data_luma, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
  32
  33         Effect *luma_mix_effect = tester.get_chain()->add_effect(new LumaMixEffect(), input1, input2, input3);
  34         ASSERT_TRUE(luma_mix_effect->set_float("transition_width", 100000.0f));
  35
  36         float out_data[4];
  37         ASSERT_TRUE(luma_mix_effect->set_float("progress", 0.0f));
  38         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
  39         expect_equal(data_a, out_data, 2, 2);
  40
  41         // Lower right from B, the rest from A.
  42         float expected_data_049[] = {
  43                 0.0f, 0.25f,
  44                 0.75f, 0.6f,
  45         };
  46         ASSERT_TRUE(luma_mix_effect->set_float("progress", 0.49f));
  47         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
  48         expect_equal(expected_data_049, out_data, 2, 2);
  49
  50         // Lower two from B, the rest from A.
  51         float expected_data_051[] = {
  52                 0.0f, 0.25f,
  53                 0.65f, 0.6f,
  54         };
  55         ASSERT_TRUE(luma_mix_effect->set_float("progress", 0.51f));
  56         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
  57         expect_equal(expected_data_051, out_data, 2, 2);
  58
  59         ASSERT_TRUE(luma_mix_effect->set_float("progress", 1.0f));
  60         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
  61         expect_equal(data_b, out_data, 2, 2);
  62 }
  63
  64 TEST(LumaMixEffectTest, SoftWipeHalfWayThrough) {
  65         float data_a[] = {
  66                 0.0f, 0.25f,
  67                 0.75f, 1.0f,
  68         };
  69         float data_b[] = {
  70                 1.0f, 0.5f,
  71                 0.65f, 0.6f,
  72         };
  73         float data_luma[] = {
  74                 0.0f, 0.25f,
  75                 0.5f, 0.75f,
  76         };
  77         // At this point, the luma range and the mix range should exactly line up,
  78         // so we get a straight-up fade by luma.
  79         float expected_data[] = {
  80                 data_a[0] + (data_b[0] - data_a[0]) * data_luma[0],
  81                 data_a[1] + (data_b[1] - data_a[1]) * data_luma[1],
  82                 data_a[2] + (data_b[2] - data_a[2]) * data_luma[2],
  83                 data_a[3] + (data_b[3] - data_a[3]) * data_luma[3],
  84         };
  85         float out_data[4];
  86
  87         EffectChainTester tester(data_a, 2, 2, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
  88         Effect *input1 = tester.get_chain()->last_added_effect();
  89         Effect *input2 = tester.add_input(data_b, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
  90         Effect *input3 = tester.add_input(data_luma, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
  91
  92         Effect *luma_mix_effect = tester.get_chain()->add_effect(new LumaMixEffect(), input1, input2, input3);
  93         ASSERT_TRUE(luma_mix_effect->set_float("transition_width", 1.0f));
  94         ASSERT_TRUE(luma_mix_effect->set_float("progress", 0.5f));
  95         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
  96         expect_equal(expected_data, out_data, 2, 2);
  97 }
  98
  99 }  // namespace movit