X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=white_balance_effect.h;h=2370847e966872746e0a75e967550bedd9979e45;hp=80416cba39340df11b0464a173e5ca07da5be3a5;hb=90ac46cdc5845432df13385f946c63b5496c685e;hpb=cfe0bc4fa1e2a56eeb12c33e596f79c1292292c8

diff --git a/white_balance_effect.h b/white_balance_effect.h
index 80416cb..2370847 100644
--- a/white_balance_effect.h
+++ b/white_balance_effect.h
@@ -1,14 +1,22 @@
-#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 <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; }
+	virtual bool one_to_one_sampling() const { return true; }
 	std::string output_fragment_shader();
 
 	void set_gl_state(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num);
@@ -20,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)