1 // Unit tests for VignetteEffect.
6 #include "effect_chain.h"
7 #include "gtest/gtest.h"
8 #include "image_format.h"
10 #include "vignette_effect.h"
12 TEST(VignetteEffectTest, HugeInnerRadiusDoesNothing) {
15 float data[size * size] = {
21 float out_data[size * size];
23 EffectChainTester tester(data, size, size, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
24 Effect *vignette_effect = tester.get_chain()->add_effect(new VignetteEffect());
25 ASSERT_TRUE(vignette_effect->set_float("inner_radius", 10.0f));
26 tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
28 expect_equal(data, out_data, size, size);
31 TEST(VignetteEffectTest, HardCircle) {
34 float data[size * size], out_data[size * size], expected_data[size * size];
35 for (int y = 0; y < size; ++y) {
36 for (int x = 0; x < size; ++x) {
37 data[y * size + x] = 1.0f;
40 for (int y = 0; y < size; ++y) {
41 const float yf = (y + 0.5f) / size;
42 for (int x = 0; x < size; ++x) {
43 const float xf = (x + 0.5f) / size;
44 if (hypot(xf - 0.5, yf - 0.5) < 0.3) {
45 expected_data[y * size + x] = 1.0f;
47 expected_data[y * size + x] = 0.0f;
52 EffectChainTester tester(data, size, size, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
53 Effect *vignette_effect = tester.get_chain()->add_effect(new VignetteEffect());
54 ASSERT_TRUE(vignette_effect->set_float("radius", 0.0f));
55 ASSERT_TRUE(vignette_effect->set_float("inner_radius", 0.3f));
56 tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
58 expect_equal(expected_data, out_data, size, size);
61 TEST(VignetteEffectTest, BurstFromUpperLeftCorner) {
62 const int width = 16, height = 24;
65 float data[width * height], out_data[width * height], expected_data[width * height];
66 for (int y = 0; y < height; ++y) {
67 for (int x = 0; x < width; ++x) {
68 data[y * width + x] = 1.0f;
71 for (int y = 0; y < height; ++y) {
72 const float yf = (y + 0.5f) / width; // Note: Division by width.
73 for (int x = 0; x < width; ++x) {
74 const float xf = (x + 0.5f) / width;
75 const float d = hypot(xf, yf) / radius;
77 expected_data[y * width + x] = 0.0f;
79 expected_data[y * width + x] = cos(d * 0.5 * M_PI) * cos(d * 0.5 * M_PI);
84 EffectChainTester tester(data, width, height, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
85 Effect *vignette_effect = tester.get_chain()->add_effect(new VignetteEffect());
86 float center[] = { 0.0f, 0.0f };
87 ASSERT_TRUE(vignette_effect->set_vec2("center", center));
88 ASSERT_TRUE(vignette_effect->set_float("radius", radius));
89 ASSERT_TRUE(vignette_effect->set_float("inner_radius", 0.0f));
90 tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
92 expect_equal(expected_data, out_data, width, height);