]> git.sesse.net Git - movit/blob - effect.cpp
When correcting for white balance, move the D65 normalization into compute_lms_scalin...
[movit] / effect.cpp
1 #include <stdio.h>
2 #include <string.h>
3 #include <assert.h>
4 #include "effect.h"
5 #include "effect_chain.h"
6 #include "util.h"
7
8 #include "opengl.h"
9
10 GLint get_uniform_location(GLuint glsl_program_num, const std::string &prefix, const std::string &key)
11 {
12         std::string name = prefix + "_" + key;
13         return glGetUniformLocation(glsl_program_num, name.c_str());
14 }
15
16 void set_uniform_int(GLuint glsl_program_num, const std::string &prefix, const std::string &key, int value)
17 {
18         GLint location = get_uniform_location(glsl_program_num, prefix, key);
19         if (location == -1) {
20                 return;
21         }
22         check_error();
23         glUniform1i(location, value);
24         check_error();
25 }
26
27 void set_uniform_float(GLuint glsl_program_num, const std::string &prefix, const std::string &key, float value)
28 {
29         GLint location = get_uniform_location(glsl_program_num, prefix, key);
30         if (location == -1) {
31                 return;
32         }
33         check_error();
34         glUniform1f(location, value);
35         check_error();
36 }
37
38 void set_uniform_vec2(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values)
39 {
40         GLint location = get_uniform_location(glsl_program_num, prefix, key);
41         if (location == -1) {
42                 return;
43         }
44         check_error();
45         glUniform2fv(location, 1, values);
46         check_error();
47 }
48
49 void set_uniform_vec3(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values)
50 {
51         GLint location = get_uniform_location(glsl_program_num, prefix, key);
52         if (location == -1) {
53                 return;
54         }
55         check_error();
56         glUniform3fv(location, 1, values);
57         check_error();
58 }
59
60 void set_uniform_vec4_array(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values, size_t num_values)
61 {
62         GLint location = get_uniform_location(glsl_program_num, prefix, key);
63         if (location == -1) {
64                 return;
65         }
66         check_error();
67         glUniform4fv(location, num_values, values);
68         check_error();
69 }
70
71 void set_uniform_mat3(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const Eigen::Matrix3d& matrix)
72 {
73         GLint location = get_uniform_location(glsl_program_num, prefix, key);
74         if (location == -1) {
75                 return;
76         }
77         check_error();
78
79         // Convert to float (GLSL has no double matrices).
80         float matrixf[9];
81         for (unsigned y = 0; y < 3; ++y) {
82                 for (unsigned x = 0; x < 3; ++x) {
83                         matrixf[y + x * 3] = matrix(y, x);
84                 }
85         }
86
87         glUniformMatrix3fv(location, 1, GL_FALSE, matrixf);
88         check_error();
89 }
90
91 bool Effect::set_int(const std::string &key, int value)
92 {
93         if (params_int.count(key) == 0) {
94                 return false;
95         }
96         *params_int[key] = value;
97         return true;
98 }
99
100 bool Effect::set_float(const std::string &key, float value)
101 {
102         if (params_float.count(key) == 0) {
103                 return false;
104         }
105         *params_float[key] = value;
106         return true;
107 }
108
109 bool Effect::set_vec2(const std::string &key, const float *values)
110 {
111         if (params_vec2.count(key) == 0) {
112                 return false;
113         }
114         memcpy(params_vec2[key], values, sizeof(float) * 2);
115         return true;
116 }
117
118 bool Effect::set_vec3(const std::string &key, const float *values)
119 {
120         if (params_vec3.count(key) == 0) {
121                 return false;
122         }
123         memcpy(params_vec3[key], values, sizeof(float) * 3);
124         return true;
125 }
126
127 void Effect::register_int(const std::string &key, int *value)
128 {
129         assert(params_int.count(key) == 0);
130         params_int[key] = value;
131 }
132
133 void Effect::register_float(const std::string &key, float *value)
134 {
135         assert(params_float.count(key) == 0);
136         params_float[key] = value;
137 }
138
139 void Effect::register_vec2(const std::string &key, float *values)
140 {
141         assert(params_vec2.count(key) == 0);
142         params_vec2[key] = values;
143 }
144
145 void Effect::register_vec3(const std::string &key, float *values)
146 {
147         assert(params_vec3.count(key) == 0);
148         params_vec3[key] = values;
149 }
150
151 void Effect::register_1d_texture(const std::string &key, float *values, size_t size)
152 {
153         assert(params_tex_1d.count(key) == 0);
154
155         Texture1D tex;
156         tex.values = values;
157         tex.size = size;
158         tex.needs_update = false;
159         glGenTextures(1, &tex.texture_num);
160
161         glBindTexture(GL_TEXTURE_1D, tex.texture_num);
162         check_error();
163         glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
164         check_error();
165         glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
166         check_error();
167         glTexImage1D(GL_TEXTURE_1D, 0, GL_LUMINANCE16F_ARB, size, 0, GL_LUMINANCE, GL_FLOAT, values);
168         check_error();
169
170         params_tex_1d[key] = tex;
171 }
172
173 void Effect::invalidate_1d_texture(const std::string &key)
174 {
175         assert(params_tex_1d.count(key) != 0);
176         params_tex_1d[key].needs_update = true;
177 }
178
179 // Output convenience uniforms for each parameter.
180 // These will be filled in per-frame.
181 std::string Effect::output_convenience_uniforms() const
182 {
183         std::string output = "";
184         for (std::map<std::string, float*>::const_iterator it = params_float.begin();
185              it != params_float.end();
186              ++it) {
187                 char buf[256];
188                 sprintf(buf, "uniform float PREFIX(%s);\n", it->first.c_str());
189                 output.append(buf);
190         }
191         for (std::map<std::string, float*>::const_iterator it = params_vec2.begin();
192              it != params_vec2.end();
193              ++it) {
194                 char buf[256];
195                 sprintf(buf, "uniform vec2 PREFIX(%s);\n", it->first.c_str());
196                 output.append(buf);
197         }
198         for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
199              it != params_vec3.end();
200              ++it) {
201                 char buf[256];
202                 sprintf(buf, "uniform vec3 PREFIX(%s);\n", it->first.c_str());
203                 output.append(buf);
204         }
205         for (std::map<std::string, Texture1D>::const_iterator it = params_tex_1d.begin();
206              it != params_tex_1d.end();
207              ++it) {
208                 char buf[256];
209                 sprintf(buf, "uniform sampler1D PREFIX(%s);\n", it->first.c_str());
210                 output.append(buf);
211         }
212         return output;
213 }
214
215 void Effect::set_gl_state(GLuint glsl_program_num, const std::string& prefix, unsigned *sampler_num)
216 {
217         for (std::map<std::string, float*>::const_iterator it = params_float.begin();
218              it != params_float.end();
219              ++it) {
220                 set_uniform_float(glsl_program_num, prefix, it->first, *it->second);
221         }
222         for (std::map<std::string, float*>::const_iterator it = params_vec2.begin();
223              it != params_vec2.end();
224              ++it) {
225                 set_uniform_vec2(glsl_program_num, prefix, it->first, it->second);
226         }
227         for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
228              it != params_vec3.end();
229              ++it) {
230                 set_uniform_vec3(glsl_program_num, prefix, it->first, it->second);
231         }
232
233         for (std::map<std::string, Texture1D>::iterator it = params_tex_1d.begin();
234              it != params_tex_1d.end();
235              ++it) {
236                 glActiveTexture(GL_TEXTURE0 + *sampler_num);
237                 check_error();
238                 glBindTexture(GL_TEXTURE_1D, it->second.texture_num);
239                 check_error();
240
241                 if (it->second.needs_update) {
242                         glTexImage1D(GL_TEXTURE_1D, 0, GL_LUMINANCE16F_ARB, it->second.size, 0, GL_LUMINANCE, GL_FLOAT, it->second.values);
243                         check_error();
244                         it->second.needs_update = false;
245                 }
246
247                 set_uniform_int(glsl_program_num, prefix, it->first, *sampler_num);
248                 ++*sampler_num;
249         }
250 }
251
252 void Effect::clear_gl_state() {}