]> git.sesse.net Git - movit/blobdiff - test_util.cpp
Do not unconditionally profile (that was a miscommit). Instead, make a PROFILE=1...
[movit] / test_util.cpp
index a52eb797782fbad2f0f4aae297d2248322903a75..4e37cec3fdd82900831f5b55fa34d905b80bb4c1 100644 (file)
@@ -9,6 +9,32 @@
 
 EffectChainTester::EffectChainTester(const float *data, unsigned width, unsigned height, ColorSpace color_space, GammaCurve gamma_curve)
        : chain(width, height), width(width), height(height)
+{
+       add_input(data, color_space, gamma_curve);
+
+       glGenTextures(1, &texnum);
+       check_error();
+       glBindTexture(GL_TEXTURE_2D, texnum);
+       check_error();
+       glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA16F_ARB, width, height, 0, GL_RGBA, GL_UNSIGNED_BYTE, NULL);
+       check_error();
+
+       glGenFramebuffers(1, &fbo);
+       check_error();
+       glBindFramebuffer(GL_FRAMEBUFFER, fbo);
+       check_error();
+       glFramebufferTexture2D(
+               GL_FRAMEBUFFER,
+               GL_COLOR_ATTACHMENT0,
+               GL_TEXTURE_2D,
+               texnum,
+               0);
+       check_error();
+       glBindFramebuffer(GL_FRAMEBUFFER, 0);
+       check_error();
+}
+
+Input *EffectChainTester::add_input(const float *data, ColorSpace color_space, GammaCurve gamma_curve)
 {
        ImageFormat format;
        format.color_space = color_space;
@@ -17,6 +43,7 @@ EffectChainTester::EffectChainTester(const float *data, unsigned width, unsigned
        FlatInput *input = new FlatInput(format, FORMAT_GRAYSCALE, GL_FLOAT, width, height);
        input->set_pixel_data(data);
        chain.add_input(input);
+       return input;
 }
 
 void EffectChainTester::run(float *out_data, ColorSpace color_space, GammaCurve gamma_curve)
@@ -27,9 +54,9 @@ void EffectChainTester::run(float *out_data, ColorSpace color_space, GammaCurve
        chain.add_output(format);
        chain.finalize();
 
-       glViewport(0, 0, width, height);
-       chain.render_to_screen();
+       chain.render_to_fbo(fbo, width, height);
 
+       glBindFramebuffer(GL_FRAMEBUFFER, fbo);
        glReadPixels(0, 0, width, height, GL_RED, GL_FLOAT, out_data);
 
        // Flip upside-down to compensate for different origin.