#include <Eigen/Core>
#include <Eigen/LU>
-#include <GL/glew.h>
+#include <epoxy/gl.h>
#include <assert.h>
+#include "colorspace_conversion_effect.h"
#include "d65.h"
#include "effect_util.h"
+#include "image_format.h"
#include "util.h"
#include "white_balance_effect.h"
using namespace Eigen;
+using namespace std;
+
+namespace movit {
namespace {
return Vector3d(x, y, 1.0 - x - y);
}
-// Assuming sRGB primaries, from Wikipedia.
-const double rgb_to_xyz_matrix[9] = {
- 0.4124, 0.2126, 0.0193,
- 0.3576, 0.7152, 0.1192,
- 0.1805, 0.0722, 0.9505,
-};
-
/*
* There are several different perceptual color spaces with different intended
* uses; for instance, CIECAM02 uses one space (CAT02) for purposes of computing
{
register_vec3("neutral_color", (float *)&neutral_color);
register_float("output_color_temperature", &output_color_temperature);
+ register_uniform_mat3("correction_matrix", &uniform_correction_matrix);
}
-std::string WhiteBalanceEffect::output_fragment_shader()
+string WhiteBalanceEffect::output_fragment_shader()
{
return read_file("white_balance_effect.frag");
}
-void WhiteBalanceEffect::set_gl_state(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num)
+void WhiteBalanceEffect::set_gl_state(GLuint glsl_program_num, const string &prefix, unsigned *sampler_num)
{
+ Matrix3d rgb_to_xyz_matrix = ColorspaceConversionEffect::get_xyz_matrix(COLORSPACE_sRGB);
Vector3d rgb(neutral_color.r, neutral_color.g, neutral_color.b);
- Vector3d xyz = Map<const Matrix3d>(rgb_to_xyz_matrix) * rgb;
+ Vector3d xyz = rgb_to_xyz_matrix * rgb;
Vector3d lms_scale = compute_lms_scaling_factors(xyz);
/*
* Note that since we postmultiply our vectors, the order of the matrices
* has to be the opposite of the execution order.
*/
- Matrix3d corr_matrix =
- Map<const Matrix3d>(rgb_to_xyz_matrix).inverse() *
+ uniform_correction_matrix =
+ rgb_to_xyz_matrix.inverse() *
Map<const Matrix3d>(xyz_to_lms_matrix).inverse() *
lms_scale.asDiagonal() *
Map<const Matrix3d>(xyz_to_lms_matrix) *
- Map<const Matrix3d>(rgb_to_xyz_matrix);
- set_uniform_mat3(glsl_program_num, prefix, "correction_matrix", corr_matrix);
+ rgb_to_xyz_matrix;
}
+
+} // namespace movit