X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=test_util.h;h=412acadd912f3423d8df35c4bbeff552636cd441;hp=0c2e7b9d51b141e258963a30e6a0bc0d2c7e945b;hb=3140175e3160e75d659e596e420afffb596e3c2c;hpb=f52ae2984b179e8d91e3d78ecc72888644fce406

diff --git a/test_util.h b/test_util.h
index 0c2e7b9..412acad 100644
--- 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)