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 0c2e7b9d51b141e258963a30e6a0bc0d2c7e945b..412acadd912f3423d8df35c4bbeff552636cd441 100644 (file)
--- a/test_util.h
+++ b/test_util.h
@@ -5,16 +5,27 @@
 
 class EffectChainTester {
 public:
-       EffectChainTester(const float *data, unsigned width, unsigned height, 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, ColorSpace color_space, GammaCurve gamma_curve);
-       void run(float *out_data, ColorSpace color_space, GammaCurve gamma_curve);
+       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);
+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);
 
 #endif  // !defined(_TEST_UTIL_H)