X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=white_balance_effect.h;h=01dcc7e24d18a902eec255fb7f056dfbb43d303d;hp=c4e46a47548354c6df953fd2caf20d7c9112a80f;hb=eff011224abc5dc81f801f3ea44572287a55bcac;hpb=37f56fcbe571b2322243f6de59494bf9e0cbb37a

diff --git a/white_balance_effect.h b/white_balance_effect.h
index c4e46a4..01dcc7e 100644
--- a/white_balance_effect.h
+++ b/white_balance_effect.h
@@ -1,21 +1,25 @@
-#ifndef _WHITE_BALANCE_EFFECT_H
-#define _WHITE_BALANCE_EFFECT_H 1
+#ifndef _MOVIT_WHITE_BALANCE_EFFECT_H
+#define _MOVIT_WHITE_BALANCE_EFFECT_H 1
 
 // Color correction in LMS color space.
 
-#include <GL/glew.h>
+#include <epoxy/gl.h>
 #include <string>
+#include <Eigen/Core>
 
 #include "effect.h"
 
+namespace movit {
+
 class WhiteBalanceEffect : public Effect {
 public:
 	WhiteBalanceEffect();
-	virtual std::string effect_type_id() const { return "WhiteBalanceEffect"; }
-	virtual AlphaHandling alpha_handling() const { return DONT_CARE_ALPHA_TYPE; }
-	std::string output_fragment_shader();
+	std::string effect_type_id() const override { return "WhiteBalanceEffect"; }
+	AlphaHandling alpha_handling() const override { return DONT_CARE_ALPHA_TYPE; }
+	bool strong_one_to_one_sampling() const override { return true; }
+	std::string output_fragment_shader() override;
 
-	void set_gl_state(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num);
+	void set_gl_state(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num) override;
 
 private:
 	// The neutral color, in linear sRGB.
@@ -24,6 +28,10 @@ private:
 	// Output color temperature (in Kelvins).
 	// Choosing 6500 will lead to no color cast (ie., the neutral color becomes perfectly gray).
 	float output_color_temperature;
+
+	Eigen::Matrix3d uniform_correction_matrix;
 };
 
-#endif // !defined(_WHITE_BALANCE_EFFECT_H)
+}  // namespace movit
+
+#endif // !defined(_MOVIT_WHITE_BALANCE_EFFECT_H)