expect_equal(expected_data, out_data, 3, 2);
}
-TEST(EffectChainTest, TopLeftOrigin) {
+class WithAndWithoutComputeShaderTest : public testing::TestWithParam<string> {
+};
+INSTANTIATE_TEST_CASE_P(WithAndWithoutComputeShaderTest,
+ WithAndWithoutComputeShaderTest,
+ testing::Values("fragment", "compute"));
+
+// An effect that does nothing, but as a compute shader.
+class IdentityComputeEffect : public Effect {
+public:
+ IdentityComputeEffect() {}
+ virtual string effect_type_id() const { return "IdentityComputeEffect"; }
+ virtual bool is_compute_shader() const { return true; }
+ string output_fragment_shader() { return read_file("identity.comp"); }
+};
+
+TEST_P(WithAndWithoutComputeShaderTest, TopLeftOrigin) {
float data[] = {
0.0f, 0.25f, 0.3f,
0.75f, 1.0f, 1.0f,
float out_data[6];
EffectChainTester tester(data, 3, 2, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
tester.get_chain()->set_output_origin(OUTPUT_ORIGIN_TOP_LEFT);
+ if (GetParam() == "compute") {
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
+ tester.get_chain()->add_effect(new IdentityComputeEffect());
+ }
tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
expect_equal(expected_data, out_data, 3, 2);
template<class T>
class RewritingEffect : public Effect {
public:
- RewritingEffect() : effect(new T()), replaced_node(nullptr) {}
+ template<class... Args>
+ RewritingEffect(Args &&... args) : effect(new T(std::forward<Args>(args)...)), replaced_node(nullptr) {}
string effect_type_id() const override { return "RewritingEffect[" + effect->effect_type_id() + "]"; }
string output_fragment_shader() override { EXPECT_TRUE(false); return read_file("identity.frag"); }
void rewrite_graph(EffectChain *graph, Node *self) override {
tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_sRGB);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- ASSERT_EQ(1, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ ASSERT_EQ(1u, node->outgoing_links.size());
EXPECT_EQ("GammaExpansionEffect", node->incoming_links[0]->effect->effect_type_id());
EXPECT_EQ("GammaCompressionEffect", node->outgoing_links[0]->effect->effect_type_id());
tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_sRGB);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- ASSERT_EQ(1, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ ASSERT_EQ(1u, node->outgoing_links.size());
EXPECT_EQ("FlatInput", node->incoming_links[0]->effect->effect_type_id());
EXPECT_EQ("GammaCompressionEffect", node->outgoing_links[0]->effect->effect_type_id());
tester.run(out_data, GL_RED, COLORSPACE_REC_601_525, GAMMA_LINEAR);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- ASSERT_EQ(1, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ ASSERT_EQ(1u, node->outgoing_links.size());
EXPECT_EQ("ColorspaceConversionEffect", node->incoming_links[0]->effect->effect_type_id());
EXPECT_EQ("ColorspaceConversionEffect", node->outgoing_links[0]->effect->effect_type_id());
tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_sRGB);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- EXPECT_EQ(0, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ EXPECT_EQ(0u, node->outgoing_links.size());
EXPECT_EQ("FlatInput", node->incoming_links[0]->effect->effect_type_id());
expect_equal(expected_data, out_data, 3, 2);
tester.run(out_data, GL_RED, COLORSPACE_REC_601_525, GAMMA_LINEAR);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- EXPECT_EQ(0, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ EXPECT_EQ(0u, node->outgoing_links.size());
EXPECT_EQ("FlatInput", node->incoming_links[0]->effect->effect_type_id());
expect_equal(expected_data, out_data, 3, 2);
tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- EXPECT_EQ(0, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ EXPECT_EQ(0u, node->outgoing_links.size());
EXPECT_EQ("FlatInput", node->incoming_links[0]->effect->effect_type_id());
expect_equal(expected_data, out_data, 4, size);
tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_PREMULTIPLIED);
Node *node = input->blue_node;
- EXPECT_EQ(0, node->incoming_links.size());
- EXPECT_EQ(0, node->outgoing_links.size());
+ EXPECT_EQ(0u, node->incoming_links.size());
+ EXPECT_EQ(0u, node->outgoing_links.size());
expect_equal(data, out_data, 4, size);
}
tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED);
Node *node = effect->replaced_node;
- EXPECT_EQ(1, node->incoming_links.size());
- EXPECT_EQ(0, node->outgoing_links.size());
+ EXPECT_EQ(1u, node->incoming_links.size());
+ EXPECT_EQ(0u, node->outgoing_links.size());
expect_equal(data, out_data, 4, size);
}
tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED);
Node *node = effect->replaced_node;
- EXPECT_EQ(1, node->incoming_links.size());
- EXPECT_EQ(1, node->outgoing_links.size());
+ EXPECT_EQ(1u, node->incoming_links.size());
+ EXPECT_EQ(1u, node->outgoing_links.size());
EXPECT_EQ("AlphaDivisionEffect", node->outgoing_links[0]->effect->effect_type_id());
expect_equal(data, out_data, 4, size);
class MipmapNeedingEffect : public Effect {
public:
MipmapNeedingEffect() {}
- bool needs_mipmaps() const override { return true; }
+ MipmapRequirements needs_mipmaps() const override { return NEEDS_MIPMAPS; }
// To be allowed to mess with the sampler state.
bool needs_texture_bounce() const override { return true; }
expect_equal(expected_data, out_data, 4, 16);
}
+// An effect to verify that you can turn off mipmaps; it downscales by two,
+// which gives blur with mipmaps and aliasing (picks out every other pixel)
+// without.
+class Downscale2xEffect : public Effect {
+public:
+ explicit Downscale2xEffect(MipmapRequirements mipmap_requirements)
+ : mipmap_requirements(mipmap_requirements)
+ {
+ register_vec2("offset", offset);
+ }
+ MipmapRequirements needs_mipmaps() const override { return mipmap_requirements; }
+
+ string effect_type_id() const override { return "Downscale2xEffect"; }
+ string output_fragment_shader() override { return read_file("downscale2x.frag"); }
+
+private:
+ const MipmapRequirements mipmap_requirements;
+ float offset[2] { 0.0f, 0.0f };
+};
+
+TEST(EffectChainTest, MipmapChainGetsSplit) {
+ float data[] = {
+ 0.0f, 0.0f, 0.0f, 0.0f,
+ 1.0f, 0.0f, 1.0f, 0.0f,
+ 0.0f, 0.0f, 0.0f, 0.0f,
+ 1.0f, 0.0f, 1.0f, 0.0f,
+ };
+
+ // The intermediate result after the first step looks like this,
+ // assuming there are no mipmaps (the zeros are due to border behavior):
+ //
+ // 0 0 0 0
+ // 0 0 0 0
+ // 1 1 0 0
+ // 1 1 0 0
+ //
+ // so another 2x downscale towards the bottom left will give
+ //
+ // 0 0
+ // 1 0
+ //
+ // with yet more zeros coming in on the top and the right from the border.
+ float expected_data[] = {
+ 0.0f, 0.0f, 0.0f, 0.0f,
+ 0.0f, 0.0f, 0.0f, 0.0f,
+ 0.0f, 0.0f, 0.0f, 0.0f,
+ 1.0f, 0.0f, 0.0f, 0.0f,
+ };
+ float out_data[4 * 4];
+
+ float offset[] = { -0.5f / 4.0f, -0.5f / 4.0f };
+ RewritingEffect<Downscale2xEffect> *pick_out_bottom_left = new RewritingEffect<Downscale2xEffect>(Effect::CANNOT_ACCEPT_MIPMAPS);
+ ASSERT_TRUE(pick_out_bottom_left->effect->set_vec2("offset", offset));
+
+ RewritingEffect<Downscale2xEffect> *downscale2x = new RewritingEffect<Downscale2xEffect>(Effect::NEEDS_MIPMAPS);
+
+ EffectChainTester tester(data, 4, 4, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
+ tester.get_chain()->add_effect(pick_out_bottom_left);
+ tester.get_chain()->add_effect(downscale2x);
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ EXPECT_NE(pick_out_bottom_left->replaced_node->containing_phase,
+ downscale2x->replaced_node->containing_phase);
+
+ expect_equal(expected_data, out_data, 4, 4);
+}
+
// An effect that adds its two inputs together. Used below.
class AddEffect : public Effect {
public:
expect_equal(expected_data, out_data, 2, 2);
}
+// Constructs the graph
+//
+// FlatInput |
+// / \ |
+// Downscale2xEffect (mipmaps) Downscale2xEffect (no mipmaps) |
+// | | |
+// Downscale2xEffect (mipmaps) Downscale2xEffect (no mipmaps) |
+// \ / |
+// AddEffect |
+//
+// and verifies that it gives the correct output. Due to the conflicting
+// mipmap demands, EffectChain needs to make two phases; exactly where it's
+// split is less important, though (this is a fairly obscure situation that
+// is unlikely to happen in practice).
+TEST(EffectChainTest, DiamondGraphWithConflictingMipmaps) {
+ float data[] = {
+ 0.0f, 0.0f, 0.0f, 0.0f,
+ 1.0f, 0.0f, 1.0f, 0.0f,
+ 0.0f, 0.0f, 0.0f, 0.0f,
+ 1.0f, 0.0f, 1.0f, 0.0f,
+ };
+
+ // Same situation as MipmapChainGetsSplit. The output of the two
+ // downscales with no mipmaps looks like this:
+ //
+ // 0 0 0 0
+ // 0 0 0 0
+ // 0 0 0 0
+ // 1 0 0 0
+ //
+ // and the one with mipmaps is 0.25 everywhere. Due to postmultiplied
+ // alpha, we get the average even though we are using AddEffect.
+ float expected_data[] = {
+ 0.125f, 0.125f, 0.125f, 0.125f,
+ 0.125f, 0.125f, 0.125f, 0.125f,
+ 0.125f, 0.125f, 0.125f, 0.125f,
+ 0.625f, 0.125f, 0.125f, 0.125f,
+ };
+ float out_data[4 * 4];
+
+ float offset[] = { -0.5f / 4.0f, -0.5f / 4.0f };
+ Downscale2xEffect *nomipmap1 = new Downscale2xEffect(Effect::CANNOT_ACCEPT_MIPMAPS);
+ Downscale2xEffect *nomipmap2 = new Downscale2xEffect(Effect::CANNOT_ACCEPT_MIPMAPS);
+ ASSERT_TRUE(nomipmap1->set_vec2("offset", offset));
+ ASSERT_TRUE(nomipmap2->set_vec2("offset", offset));
+
+ Downscale2xEffect *mipmap1 = new Downscale2xEffect(Effect::NEEDS_MIPMAPS);
+ Downscale2xEffect *mipmap2 = new Downscale2xEffect(Effect::NEEDS_MIPMAPS);
+
+ EffectChainTester tester(nullptr, 4, 4);
+
+ ImageFormat format;
+ format.color_space = COLORSPACE_sRGB;
+ format.gamma_curve = GAMMA_LINEAR;
+
+ FlatInput *input = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 4);
+ input->set_pixel_data(data);
+
+ tester.get_chain()->add_input(input);
+
+ tester.get_chain()->add_effect(nomipmap1, input);
+ tester.get_chain()->add_effect(nomipmap2, nomipmap1);
+
+ tester.get_chain()->add_effect(mipmap1, input);
+ tester.get_chain()->add_effect(mipmap2, mipmap1);
+
+ tester.get_chain()->add_effect(new AddEffect(), nomipmap2, mipmap2);
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ expect_equal(expected_data, out_data, 4, 4);
+}
+
TEST(EffectChainTest, EffectUsedTwiceOnlyGetsOneGammaConversion) {
float data[] = {
0.735f, 0.0f,
expect_equal(expected_data, out_data, 2, 2);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- ASSERT_EQ(1, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ ASSERT_EQ(1u, node->outgoing_links.size());
EXPECT_EQ("FlatInput", node->incoming_links[0]->effect->effect_type_id());
EXPECT_EQ("GammaExpansionEffect", node->outgoing_links[0]->effect->effect_type_id());
}
expect_equal(expected_data, out_data, 2, 2);
Node *node = effect->replaced_node;
- ASSERT_EQ(1, node->incoming_links.size());
- ASSERT_EQ(1, node->outgoing_links.size());
+ ASSERT_EQ(1u, node->incoming_links.size());
+ ASSERT_EQ(1u, node->outgoing_links.size());
EXPECT_EQ("FlatInput", node->incoming_links[0]->effect->effect_type_id());
EXPECT_EQ("ColorspaceConversionEffect", node->outgoing_links[0]->effect->effect_type_id());
}
bool sets_virtual_output_size() const override { return false; }
};
-class WithAndWithoutComputeShaderTest : public testing::TestWithParam<string> {
-};
-INSTANTIATE_TEST_CASE_P(WithAndWithoutComputeShaderTest,
- WithAndWithoutComputeShaderTest,
- testing::Values("fragment", "compute"));
-
-// An effect that does nothing, but as a compute shader.
-class IdentityComputeEffect : public Effect {
-public:
- IdentityComputeEffect() {}
- virtual string effect_type_id() const { return "IdentityComputeEffect"; }
- virtual bool is_compute_shader() const { return true; }
- string output_fragment_shader() { return read_file("identity.comp"); }
-};
-
// An effect that promises one-to-one sampling (unlike IdentityEffect).
class OneToOneEffect : public Effect {
public:
RewritingEffect<OneToOneEffect> *effect2 = new RewritingEffect<OneToOneEffect>();
if (GetParam() == "compute") {
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
tester.get_chain()->add_effect(new IdentityComputeEffect());
} else {
tester.get_chain()->add_effect(new NonVirtualResizeEffect(size, size));
expect_equal(data, out_data, size, size);
// The first OneToOneEffect should be in the same phase as its input.
- ASSERT_EQ(1, effect1->replaced_node->incoming_links.size());
+ ASSERT_EQ(1u, effect1->replaced_node->incoming_links.size());
EXPECT_EQ(effect1->replaced_node->incoming_links[0]->containing_phase,
effect1->replaced_node->containing_phase);
// the IdentityEffect (since the latter is not one-to-one),
// ie., the chain should be broken somewhere between them, but exactly
// where doesn't matter.
- ASSERT_EQ(1, effect1->replaced_node->incoming_links.size());
+ ASSERT_EQ(1u, effect1->replaced_node->incoming_links.size());
EXPECT_NE(effect1->replaced_node->incoming_links[0]->containing_phase,
effect3->replaced_node->containing_phase);
float out_data[6], temp[6 * 4];
EffectChain chain(width, height);
+ MovitDebugLevel old_movit_debug_level = movit_debug_level;
movit_debug_level = MOVIT_DEBUG_ON;
ImageFormat format;
expect_equal(expected_data, out_data, width, height);
// Reset the debug status again.
- movit_debug_level = MOVIT_DEBUG_OFF;
+ movit_debug_level = old_movit_debug_level;
}
// A dummy effect whose only purpose is to test sprintf decimal behavior.
EffectChainTester tester(data, size, 1, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, GL_RGBA32F);
tester.get_chain()->set_intermediate_format(GL_RGB10_A2, SQUARE_ROOT_FRAMEBUFFER_TRANSFORMATION);
if (GetParam() == "compute") {
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
tester.get_chain()->add_effect(new IdentityComputeEffect());
} else {
tester.get_chain()->add_effect(new IdentityEffect());
expect_equal(data, out_data, 3, 2);
- ASSERT_NE(0, effect->last_glsl_program_num);
+ ASSERT_NE(0u, effect->last_glsl_program_num);
// Now pretend some other effect is using this program number;
// ResourcePool will then need to clone it.
expect_equal(data, out_data, 3, 2);
}
+// A compute shader to mirror the inputs, in 2x2 blocks.
+class MirrorComputeEffect : public Effect {
+public:
+ MirrorComputeEffect() {}
+ string effect_type_id() const override { return "MirrorComputeEffect"; }
+ bool is_compute_shader() const override { return true; }
+ string output_fragment_shader() override { return read_file("mirror.comp"); }
+ void get_compute_dimensions(unsigned output_width, unsigned output_height,
+ unsigned *x, unsigned *y, unsigned *z) const override {
+ *x = output_width / 2;
+ *y = output_height / 2;
+ *z = 1;
+ }
+};
+
+TEST(ComputeShaderTest, ComputeThenOneToOne) {
+ float data[] = {
+ 0.0f, 0.25f, 0.3f, 0.8f,
+ 0.75f, 1.0f, 1.0f, 0.2f,
+ };
+ float expected_data[] = {
+ 0.8f, 0.3f, 0.25f, 0.0f,
+ 0.2f, 1.0f, 1.0f, 0.75f,
+ };
+ float out_data[8];
+ EffectChainTester tester(data, 4, 2, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
+ tester.get_chain()->add_effect(new MirrorComputeEffect());
+ tester.get_chain()->add_effect(new OneToOneEffect());
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ expect_equal(expected_data, out_data, 4, 2);
+}
+
+// A compute shader that also resizes its input, taking the upper-left pixel
+// of every 2x2 group. (The shader is hard-coded to 4x2 input for simplicity.)
+class Downscale2xComputeEffect : public Effect {
+public:
+ Downscale2xComputeEffect() {}
+ string effect_type_id() const override { return "Downscale2xComputeEffect"; }
+ bool is_compute_shader() const override { return true; }
+ string output_fragment_shader() override { return read_file("downscale2x.comp"); }
+ bool changes_output_size() const override { return true; }
+ void inform_input_size(unsigned input_num, unsigned width, unsigned height) override
+ {
+ this->width = width;
+ this->height = height;
+ }
+ void get_output_size(unsigned *width, unsigned *height,
+ unsigned *virtual_width, unsigned *virtual_height) const override {
+ *width = *virtual_width = this->width / 2;
+ *height = *virtual_height = this->height / 2;
+ }
+
+private:
+ unsigned width, height;
+};
+
+// Even if the compute shader is not the last effect, it's the one that should decide
+// the output size of the phase.
+TEST(ComputeShaderTest, ResizingComputeThenOneToOne) {
+ float data[] = {
+ 0.0f, 0.25f, 0.3f, 0.8f,
+ 0.75f, 1.0f, 1.0f, 0.2f,
+ };
+ float expected_data[] = {
+ 0.0f, 0.3f,
+ };
+ float out_data[2];
+ EffectChainTester tester(nullptr, 2, 1);
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
+ tester.add_input(data, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR, 4, 2);
+
+ RewritingEffect<Downscale2xComputeEffect> *downscale_effect = new RewritingEffect<Downscale2xComputeEffect>();
+ tester.get_chain()->add_effect(downscale_effect);
+ tester.get_chain()->add_effect(new OneToOneEffect());
+ tester.get_chain()->add_effect(new BouncingIdentityEffect());
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ expect_equal(expected_data, out_data, 2, 1);
+
+ Phase *phase = downscale_effect->replaced_node->containing_phase;
+ EXPECT_EQ(2u, phase->output_width);
+ EXPECT_EQ(1u, phase->output_height);
+}
+
+class StrongOneToOneAddEffect : public AddEffect {
+public:
+ StrongOneToOneAddEffect() {}
+ string effect_type_id() const override { return "StrongOneToOneAddEffect"; }
+ bool strong_one_to_one_sampling() const override { return true; }
+};
+
+TEST(ComputeShaderTest, NoTwoComputeInSamePhase) {
+ float data[] = {
+ 0.0f, 0.25f, 0.3f, 0.8f,
+ 0.75f, 1.0f, 1.0f, 0.2f,
+ };
+ float expected_data[] = {
+ 0.0f, 0.3f,
+ };
+ float out_data[2];
+
+ ImageFormat format;
+ format.color_space = COLORSPACE_sRGB;
+ format.gamma_curve = GAMMA_LINEAR;
+
+ EffectChainTester tester(nullptr, 2, 1);
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
+
+ FlatInput *input1 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input1->set_pixel_data(data);
+ tester.get_chain()->add_input(input1);
+ Effect *downscale1 = tester.get_chain()->add_effect(new Downscale2xComputeEffect());
+
+ FlatInput *input2 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input2->set_pixel_data(data);
+ tester.get_chain()->add_input(input2);
+ Effect *downscale2 = tester.get_chain()->add_effect(new Downscale2xComputeEffect());
+
+ tester.get_chain()->add_effect(new StrongOneToOneAddEffect(), downscale1, downscale2);
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+ expect_equal(expected_data, out_data, 2, 1);
+}
+
+// Like the previous test, but the adder effect is not directly connected
+// to the compute shaders (so the status has to be propagated through those effects).
+TEST(ComputeShaderTest, NoTwoComputeInSamePhaseIndirect) {
+ float data[] = {
+ 0.0f, 0.25f, 0.3f, 0.8f,
+ 0.75f, 1.0f, 1.0f, 0.2f,
+ };
+ float expected_data[] = {
+ 0.0f, 0.3f,
+ };
+ float out_data[2];
+
+ ImageFormat format;
+ format.color_space = COLORSPACE_sRGB;
+ format.gamma_curve = GAMMA_LINEAR;
+
+ EffectChainTester tester(nullptr, 2, 1);
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
+
+ FlatInput *input1 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input1->set_pixel_data(data);
+ tester.get_chain()->add_input(input1);
+ tester.get_chain()->add_effect(new Downscale2xComputeEffect());
+ Effect *identity1 = tester.get_chain()->add_effect(new OneToOneEffect());
+
+ FlatInput *input2 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input2->set_pixel_data(data);
+ tester.get_chain()->add_input(input2);
+ tester.get_chain()->add_effect(new Downscale2xComputeEffect());
+ Effect *identity2 = tester.get_chain()->add_effect(new OneToOneEffect());
+
+ tester.get_chain()->add_effect(new StrongOneToOneAddEffect(), identity1, identity2);
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+ expect_equal(expected_data, out_data, 2, 1);
+}
+
+// Like the previous test, but the adder is not strong one-to-one
+// (so there are two different compute shader inputs, but none of them
+// are in the same phase).
+TEST(ComputeShaderTest, BounceTextureFromTwoComputeShaders) {
+ float data[] = {
+ 0.0f, 0.25f, 0.3f, 0.8f,
+ 0.75f, 1.0f, 1.0f, 0.2f,
+ };
+ float expected_data[] = {
+ 0.0f, 0.3f,
+ };
+ float out_data[2];
+
+ ImageFormat format;
+ format.color_space = COLORSPACE_sRGB;
+ format.gamma_curve = GAMMA_LINEAR;
+
+ EffectChainTester tester(nullptr, 2, 1);
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
+
+ FlatInput *input1 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input1->set_pixel_data(data);
+ tester.get_chain()->add_input(input1);
+ tester.get_chain()->add_effect(new Downscale2xComputeEffect());
+ Effect *identity1 = tester.get_chain()->add_effect(new OneToOneEffect());
+
+ FlatInput *input2 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input2->set_pixel_data(data);
+ tester.get_chain()->add_input(input2);
+ tester.get_chain()->add_effect(new Downscale2xComputeEffect());
+ Effect *identity2 = tester.get_chain()->add_effect(new OneToOneEffect());
+
+ tester.get_chain()->add_effect(new AddEffect(), identity1, identity2);
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+ expect_equal(expected_data, out_data, 2, 1);
+}
+
+// Requires mipmaps, but is otherwise like IdentityEffect.
+class MipmapNeedingIdentityEffect : public IdentityEffect {
+public:
+ MipmapNeedingIdentityEffect() {}
+ MipmapRequirements needs_mipmaps() const override { return NEEDS_MIPMAPS; }
+ string effect_type_id() const override { return "MipmapNeedingIdentityEffect"; }
+ bool strong_one_to_one_sampling() const override { return true; }
+};
+
+TEST(ComputeShaderTest, StrongOneToOneButStillNotChained) {
+ float data[] = {
+ 0.0f, 0.25f, 0.3f, 0.8f,
+ 0.75f, 1.0f, 1.0f, 0.2f,
+ };
+ float out_data[8];
+
+ ImageFormat format;
+ format.color_space = COLORSPACE_sRGB;
+ format.gamma_curve = GAMMA_LINEAR;
+
+ EffectChainTester tester(nullptr, 4, 2);
+ if (!movit_compute_shaders_supported) {
+ fprintf(stderr, "Skipping test; no support for compile shaders.\n");
+ return;
+ }
+
+ FlatInput *input1 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input1->set_pixel_data(data);
+ tester.get_chain()->add_input(input1);
+ Effect *compute_effect = tester.get_chain()->add_effect(new IdentityComputeEffect());
+
+ FlatInput *input2 = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 4, 2);
+ input2->set_pixel_data(data);
+ tester.get_chain()->add_input(input2);
+
+ // Not chained with the compute shader because MipmapNeedingIdentityEffect comes in
+ // the same phase, and compute shaders cannot supply mipmaps.
+ tester.get_chain()->add_effect(new StrongOneToOneAddEffect(), compute_effect, input2);
+ tester.get_chain()->add_effect(new MipmapNeedingIdentityEffect());
+
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+ expect_equal(data, out_data, 4, 2);
+}
+
+TEST(EffectChainTest, BounceResetsMipmapNeeds) {
+ float data[] = {
+ 0.0f, 0.25f,
+ 0.75f, 1.0f,
+ };
+ float out_data[1];
+
+ ImageFormat format;
+ format.color_space = COLORSPACE_sRGB;
+ format.gamma_curve = GAMMA_LINEAR;
+
+ NonMipmapCapableInput *input = new NonMipmapCapableInput(format, FORMAT_GRAYSCALE, GL_FLOAT, 2, 2);
+ input->set_pixel_data(data);
+
+ RewritingEffect<IdentityEffect> *identity = new RewritingEffect<IdentityEffect>();
+
+ RewritingEffect<ResizeEffect> *downscale = new RewritingEffect<ResizeEffect>(); // Needs mipmaps.
+ ASSERT_TRUE(downscale->effect->set_int("width", 1));
+ ASSERT_TRUE(downscale->effect->set_int("height", 1));
+
+ EffectChainTester tester(nullptr, 1, 1);
+ tester.get_chain()->add_input(input);
+ tester.get_chain()->add_effect(identity);
+ tester.get_chain()->add_effect(downscale);
+ tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
+
+ Node *input_node = identity->replaced_node->incoming_links[0];
+
+ // The ResizeEffect needs mipmaps. Normally, that would mean that it should
+ // propagate this tatus down through the IdentityEffect. However, since we
+ // bounce (due to the resize), the dependency breaks there, and we don't
+ // need to bounce again between the input and the IdentityEffect.
+ EXPECT_EQ(input_node->containing_phase,
+ identity->replaced_node->containing_phase);
+ EXPECT_NE(identity->replaced_node->containing_phase,
+ downscale->replaced_node->containing_phase);
+}
+
} // namespace movit