Add a function for EffectChainTester to get out the uint8 result.

[movit] / test_util.h

diff --git a/test_util.h b/test_util.h
index c66dfa47a0c2b427261034df0f8ff7c5a716515a..412acadd912f3423d8df35c4bbeff552636cd441 100644 (file)
--- a/test_util.h
+++ b/test_util.h
@@ -5,17 +5,25 @@
 
 class EffectChainTester {
 public:
-       EffectChainTester(const float *data, unsigned width, unsigned height, MovitPixelFormat pixel_format, Colorspace color_space, GammaCurve gamma_curve);
+       EffectChainTester(const float *data, unsigned width, unsigned height,
+                         MovitPixelFormat pixel_format = FORMAT_GRAYSCALE,
+                         Colorspace color_space = COLORSPACE_sRGB,
+                         GammaCurve gamma_curve = GAMMA_LINEAR);
        ~EffectChainTester();
        
        EffectChain *get_chain() { return &chain; }
        Input *add_input(const float *data, MovitPixelFormat pixel_format, Colorspace color_space, GammaCurve gamma_curve);
+       Input *add_input(const unsigned char *data, MovitPixelFormat pixel_format, Colorspace color_space, GammaCurve gamma_curve);
        void run(float *out_data, GLenum format, Colorspace color_space, GammaCurve gamma_curve);
+       void run(unsigned char *out_data, GLenum format, Colorspace color_space, GammaCurve gamma_curve);
 
 private:
+       void finalize_chain(Colorspace color_space, GammaCurve gamma_curve);
+
        EffectChain chain;
        GLuint fbo, texnum;
        unsigned width, height;
+       bool finalized;
 };
 
 void expect_equal(const float *ref, const float *result, unsigned width, unsigned height, float largest_difference_limit = 1.5 / 255.0, float rms_limit = 0.2 / 255.0);