Clean up stuff from the vertex shader.

[movit] / effect.cpp

diff --git a/effect.cpp b/effect.cpp
index f89b1a99c8c2350d8d5a82ddde26d701c5d373e2..cab91bf51814ae435632381754486c8f72063d80 100644 (file)
--- a/effect.cpp
+++ b/effect.cpp
@@ -1,6 +1,37 @@
+#define GL_GLEXT_PROTOTYPES 1
+
+#include <stdio.h>
 #include <string.h>
 #include <assert.h>
 #include "effect.h"
+#include "util.h"
+
+#include <GL/gl.h>
+#include <GL/glext.h>
+
+void set_uniform_float(GLhandleARB glsl_program_num, const std::string &prefix, const std::string &key, float value)
+{
+       std::string name = prefix + "_" + key;
+       GLint l = glGetUniformLocation(glsl_program_num, name.c_str());
+       if (l == -1) {
+               return;
+       }
+       check_error();
+       glUniform1f(l, value);
+       check_error();
+}
+
+void set_uniform_vec3(GLhandleARB glsl_program_num, const std::string &prefix, const std::string &key, const float *values)
+{
+       std::string name = prefix + "_" + key;
+       GLint l = glGetUniformLocation(glsl_program_num, name.c_str());
+       if (l == -1) {
+               return;
+       }
+       check_error();
+       glUniform3fv(l, 1, values);
+       check_error();
+}
 
 bool Effect::set_int(const std::string &key, int value)
 {
@@ -47,3 +78,38 @@ void Effect::register_vec3(const std::string &key, float *values)
        params_vec3[key] = values;
 }
 
+// Output convenience uniforms for each parameter.
+// These will be filled in per-frame.
+std::string Effect::output_convenience_uniforms()
+{
+       std::string output = "";
+       for (std::map<std::string, float*>::const_iterator it = params_float.begin();
+            it != params_float.end();
+            ++it) {
+               char buf[256];
+               sprintf(buf, "uniform float PREFIX(%s);\n", it->first.c_str());
+               output.append(buf);
+       }
+       for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
+            it != params_vec3.end();
+            ++it) {
+               char buf[256];
+               sprintf(buf, "uniform vec3 PREFIX(%s);\n", it->first.c_str());
+               output.append(buf);
+       }
+       return output;
+}
+
+void Effect::set_uniforms(GLhandleARB glsl_program_num, const std::string& prefix)
+{
+       for (std::map<std::string, float*>::const_iterator it = params_float.begin();
+            it != params_float.end();
+            ++it) {
+               set_uniform_float(glsl_program_num, prefix, it->first, *it->second);
+       }
+       for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
+            it != params_vec3.end();
+            ++it) {
+               set_uniform_vec3(glsl_program_num, prefix, it->first, it->second);
+       }
+}