Add support for compute shaders.
[movit] / compute_shader_test.cpp
1 #include <string>
2
3 #include <epoxy/gl.h>
4 #include <assert.h>
5
6 #include "effect.h"
7 #include "flat_input.h"
8 #include "gtest/gtest.h"
9 #include "init.h"
10 #include "resource_pool.h"
11 #include "test_util.h"
12 #include "util.h"
13
14 using namespace std;
15
16 namespace movit {
17
18 // An effect that does nothing.
19 class IdentityComputeEffect : public Effect {
20 public:
21         IdentityComputeEffect() {}
22         virtual string effect_type_id() const { return "IdentityComputeEffect"; }
23         virtual bool is_compute_shader() const { return true; }
24         string output_fragment_shader() { return read_file("identity.compute"); }
25 };
26
27 TEST(ComputeShaderTest, Identity) {
28         float data[] = {
29                 0.0f, 0.25f, 0.3f,
30                 0.75f, 1.0f, 1.0f,
31         };
32         float out_data[6];
33         EffectChainTester tester(data, 3, 2, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
34         if (!movit_compute_shaders_supported) {
35                 fprintf(stderr, "Skipping test; no support for compile shaders.\n");
36                 return;
37         }
38         tester.get_chain()->add_effect(new IdentityComputeEffect());
39         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
40
41         expect_equal(data, out_data, 3, 2);
42 }
43
44 // Like IdentityComputeEffect, but due to the alpha handling, this will be
45 // the very last effect in the chain, which means we can't output it directly
46 // to the screen.
47 class IdentityAlphaComputeEffect : public IdentityComputeEffect {
48         AlphaHandling alpha_handling() const { return DONT_CARE_ALPHA_TYPE; }
49 };
50
51 TEST(ComputeShaderTest, LastEffectInChain) {
52         float data[] = {
53                 0.0f, 0.25f, 0.3f,
54                 0.75f, 1.0f, 1.0f,
55         };
56         float out_data[6];
57         EffectChainTester tester(data, 3, 2, FORMAT_GRAYSCALE, COLORSPACE_sRGB, GAMMA_LINEAR);
58         if (!movit_compute_shaders_supported) {
59                 fprintf(stderr, "Skipping test; no support for compile shaders.\n");
60                 return;
61         }
62         tester.get_chain()->add_effect(new IdentityAlphaComputeEffect());
63         tester.run(out_data, GL_RED, COLORSPACE_sRGB, GAMMA_LINEAR);
64
65         expect_equal(data, out_data, 3, 2);
66 }
67
68 }  // namespace movit