Change so that all modifications to the graph (meta-effects, colorspace information...
[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_float_array(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values, size_t num_values)
39 {
40         GLint location = get_uniform_location(glsl_program_num, prefix, key);
41         if (location == -1) {
42                 return;
43         }
44         check_error();
45         glUniform1fv(location, num_values, values);
46         check_error();
47 }
48
49 void set_uniform_vec2(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         glUniform2fv(location, 1, values);
57         check_error();
58 }
59
60 void set_uniform_vec3(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values)
61 {
62         GLint location = get_uniform_location(glsl_program_num, prefix, key);
63         if (location == -1) {
64                 return;
65         }
66         check_error();
67         glUniform3fv(location, 1, values);
68         check_error();
69 }
70
71 void set_uniform_vec4_array(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values, size_t num_values)
72 {
73         GLint location = get_uniform_location(glsl_program_num, prefix, key);
74         if (location == -1) {
75                 return;
76         }
77         check_error();
78         glUniform4fv(location, num_values, values);
79         check_error();
80 }
81
82 bool Effect::set_int(const std::string &key, int value)
83 {
84         if (params_int.count(key) == 0) {
85                 return false;
86         }
87         *params_int[key] = value;
88         return true;
89 }
90
91 bool Effect::set_float(const std::string &key, float value)
92 {
93         if (params_float.count(key) == 0) {
94                 return false;
95         }
96         *params_float[key] = value;
97         return true;
98 }
99
100 bool Effect::set_vec2(const std::string &key, const float *values)
101 {
102         if (params_vec2.count(key) == 0) {
103                 return false;
104         }
105         memcpy(params_vec2[key], values, sizeof(float) * 2);
106         return true;
107 }
108
109 bool Effect::set_vec3(const std::string &key, const float *values)
110 {
111         if (params_vec3.count(key) == 0) {
112                 return false;
113         }
114         memcpy(params_vec3[key], values, sizeof(float) * 3);
115         return true;
116 }
117
118 void Effect::register_int(const std::string &key, int *value)
119 {
120         assert(params_int.count(key) == 0);
121         params_int[key] = value;
122 }
123
124 void Effect::register_float(const std::string &key, float *value)
125 {
126         assert(params_float.count(key) == 0);
127         params_float[key] = value;
128 }
129
130 void Effect::register_vec2(const std::string &key, float *values)
131 {
132         assert(params_vec2.count(key) == 0);
133         params_vec2[key] = values;
134 }
135
136 void Effect::register_vec3(const std::string &key, float *values)
137 {
138         assert(params_vec3.count(key) == 0);
139         params_vec3[key] = values;
140 }
141
142 void Effect::register_1d_texture(const std::string &key, float *values, size_t size)
143 {
144         assert(params_tex_1d.count(key) == 0);
145
146         Texture1D tex;
147         tex.values = values;
148         tex.size = size;
149         tex.needs_update = false;
150         glGenTextures(1, &tex.texture_num);
151
152         glBindTexture(GL_TEXTURE_1D, tex.texture_num);
153         check_error();
154         glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
155         check_error();
156         glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
157         check_error();
158         glTexImage1D(GL_TEXTURE_1D, 0, GL_LUMINANCE16F_ARB, size, 0, GL_LUMINANCE, GL_FLOAT, values);
159         check_error();
160
161         params_tex_1d[key] = tex;
162 }
163
164 void Effect::invalidate_1d_texture(const std::string &key)
165 {
166         assert(params_tex_1d.count(key) != 0);
167         params_tex_1d[key].needs_update = true;
168 }
169
170 // Output convenience uniforms for each parameter.
171 // These will be filled in per-frame.
172 std::string Effect::output_convenience_uniforms() const
173 {
174         std::string output = "";
175         for (std::map<std::string, float*>::const_iterator it = params_float.begin();
176              it != params_float.end();
177              ++it) {
178                 char buf[256];
179                 sprintf(buf, "uniform float PREFIX(%s);\n", it->first.c_str());
180                 output.append(buf);
181         }
182         for (std::map<std::string, float*>::const_iterator it = params_vec2.begin();
183              it != params_vec2.end();
184              ++it) {
185                 char buf[256];
186                 sprintf(buf, "uniform vec2 PREFIX(%s);\n", it->first.c_str());
187                 output.append(buf);
188         }
189         for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
190              it != params_vec3.end();
191              ++it) {
192                 char buf[256];
193                 sprintf(buf, "uniform vec3 PREFIX(%s);\n", it->first.c_str());
194                 output.append(buf);
195         }
196         for (std::map<std::string, Texture1D>::const_iterator it = params_tex_1d.begin();
197              it != params_tex_1d.end();
198              ++it) {
199                 char buf[256];
200                 sprintf(buf, "uniform sampler1D PREFIX(%s);\n", it->first.c_str());
201                 output.append(buf);
202         }
203         return output;
204 }
205
206 void Effect::set_gl_state(GLuint glsl_program_num, const std::string& prefix, unsigned *sampler_num)
207 {
208         for (std::map<std::string, float*>::const_iterator it = params_float.begin();
209              it != params_float.end();
210              ++it) {
211                 set_uniform_float(glsl_program_num, prefix, it->first, *it->second);
212         }
213         for (std::map<std::string, float*>::const_iterator it = params_vec2.begin();
214              it != params_vec2.end();
215              ++it) {
216                 set_uniform_vec2(glsl_program_num, prefix, it->first, it->second);
217         }
218         for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
219              it != params_vec3.end();
220              ++it) {
221                 set_uniform_vec3(glsl_program_num, prefix, it->first, it->second);
222         }
223
224         for (std::map<std::string, Texture1D>::iterator it = params_tex_1d.begin();
225              it != params_tex_1d.end();
226              ++it) {
227                 glActiveTexture(GL_TEXTURE0 + *sampler_num);
228                 check_error();
229                 glBindTexture(GL_TEXTURE_1D, it->second.texture_num);
230                 check_error();
231
232                 if (it->second.needs_update) {
233                         glTexImage1D(GL_TEXTURE_1D, 0, GL_LUMINANCE16F_ARB, it->second.size, 0, GL_LUMINANCE, GL_FLOAT, it->second.values);
234                         check_error();
235                         it->second.needs_update = false;
236                 }
237
238                 set_uniform_int(glsl_program_num, prefix, it->first, *sampler_num);
239                 ++*sampler_num;
240         }
241 }
242
243 void Effect::clear_gl_state() {}