git.sesse.net Git - movit/blob - 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 void Effect::add_self_to_effect_chain(EffectChain *chain, const std::vector<Effect *> &inputs)
 171 {
 172         chain->add_effect_raw(this, inputs);
 173 }
 174
 175 // Output convenience uniforms for each parameter.
 176 // These will be filled in per-frame.
 177 std::string Effect::output_convenience_uniforms() const
 178 {
 179         std::string output = "";
 180         for (std::map<std::string, float*>::const_iterator it = params_float.begin();
 181              it != params_float.end();
 182              ++it) {
 183                 char buf[256];
 184                 sprintf(buf, "uniform float PREFIX(%s);\n", it->first.c_str());
 185                 output.append(buf);
 186         }
 187         for (std::map<std::string, float*>::const_iterator it = params_vec2.begin();
 188              it != params_vec2.end();
 189              ++it) {
 190                 char buf[256];
 191                 sprintf(buf, "uniform vec2 PREFIX(%s);\n", it->first.c_str());
 192                 output.append(buf);
 193         }
 194         for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
 195              it != params_vec3.end();
 196              ++it) {
 197                 char buf[256];
 198                 sprintf(buf, "uniform vec3 PREFIX(%s);\n", it->first.c_str());
 199                 output.append(buf);
 200         }
 201         for (std::map<std::string, Texture1D>::const_iterator it = params_tex_1d.begin();
 202              it != params_tex_1d.end();
 203              ++it) {
 204                 char buf[256];
 205                 sprintf(buf, "uniform sampler1D PREFIX(%s);\n", it->first.c_str());
 206                 output.append(buf);
 207         }
 208         return output;
 209 }
 210
 211 void Effect::set_gl_state(GLuint glsl_program_num, const std::string& prefix, unsigned *sampler_num)
 212 {
 213         for (std::map<std::string, float*>::const_iterator it = params_float.begin();
 214              it != params_float.end();
 215              ++it) {
 216                 set_uniform_float(glsl_program_num, prefix, it->first, *it->second);
 217         }
 218         for (std::map<std::string, float*>::const_iterator it = params_vec2.begin();
 219              it != params_vec2.end();
 220              ++it) {
 221                 set_uniform_vec2(glsl_program_num, prefix, it->first, it->second);
 222         }
 223         for (std::map<std::string, float*>::const_iterator it = params_vec3.begin();
 224              it != params_vec3.end();
 225              ++it) {
 226                 set_uniform_vec3(glsl_program_num, prefix, it->first, it->second);
 227         }
 228
 229         for (std::map<std::string, Texture1D>::iterator it = params_tex_1d.begin();
 230              it != params_tex_1d.end();
 231              ++it) {
 232                 glActiveTexture(GL_TEXTURE0 + *sampler_num);
 233                 check_error();
 234                 glBindTexture(GL_TEXTURE_1D, it->second.texture_num);
 235                 check_error();
 236
 237                 if (it->second.needs_update) {
 238                         glTexImage1D(GL_TEXTURE_1D, 0, GL_LUMINANCE16F_ARB, it->second.size, 0, GL_LUMINANCE, GL_FLOAT, it->second.values);
 239                         check_error();
 240                         it->second.needs_update = false;
 241                 }
 242
 243                 set_uniform_int(glsl_program_num, prefix, it->first, *sampler_num);
 244                 ++*sampler_num;
 245         }
 246 }
 247
 248 void Effect::clear_gl_state() {}