]> git.sesse.net Git - movit/blob - test_util.h
Small refactoring in EffectChainTester.
[movit] / test_util.h
1 #ifndef _MOVIT_TEST_UTIL_H
2 #define _MOVIT_TEST_UTIL_H 1
3
4 #include <epoxy/gl.h>
5 #include "effect_chain.h"
6 #include "image_format.h"
7
8 namespace movit {
9
10 class Input;
11
12 class EffectChainTester {
13 public:
14         EffectChainTester(const float *data, unsigned width, unsigned height,
15                           MovitPixelFormat pixel_format = FORMAT_GRAYSCALE,
16                           Colorspace color_space = COLORSPACE_sRGB,
17                           GammaCurve gamma_curve = GAMMA_LINEAR,
18                           GLenum framebuffer_format = GL_RGBA16F_ARB);
19         ~EffectChainTester();
20         
21         EffectChain *get_chain() { return &chain; }
22         Input *add_input(const float *data, MovitPixelFormat pixel_format, Colorspace color_space, GammaCurve gamma_curve, int input_width = -1, int input_height = -1);
23         Input *add_input(const unsigned char *data, MovitPixelFormat pixel_format, Colorspace color_space, GammaCurve gamma_curve, int input_width = -1, int input_height = -1);
24         void run(float *out_data, GLenum format, Colorspace color_space, GammaCurve gamma_curve, OutputAlphaFormat alpha_format = OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED);
25         void run(unsigned char *out_data, GLenum format, Colorspace color_space, GammaCurve gamma_curve, OutputAlphaFormat alpha_format = OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED);
26         void add_output(const ImageFormat &format, OutputAlphaFormat alpha_format);
27         void add_ycbcr_output(const ImageFormat &format, OutputAlphaFormat alpha_format, const YCbCrFormat &ycbcr_format);
28
29 private:
30         void finalize_chain(Colorspace color_space, GammaCurve gamma_curve, OutputAlphaFormat alpha_format);
31
32         template<class T>
33         void internal_run(T *out_data, GLenum internal_format, GLenum format, Colorspace color_space, GammaCurve gamma_curve, OutputAlphaFormat alpha_format = OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED);
34
35         EffectChain chain;
36         GLuint fbo, texnum;
37         unsigned width, height;
38         bool output_added;
39         bool finalized;
40 };
41
42 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);
43 void expect_equal(const unsigned char *ref, const unsigned char *result, unsigned width, unsigned height, unsigned largest_difference_limit = 1, float rms_limit = 0.2);
44 void test_accuracy(const float *expected, const float *result, unsigned num_values, double absolute_error_limit, double relative_error_limit, double local_relative_error_limit, double rms_limit);
45
46 }  // namespace movit
47
48 #endif  // !defined(_MOVIT_TEST_UTIL_H)