X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=effect.cpp;h=1bbb89f08417821b9ffbdd2759763cd4517d3a9a;hp=927acd52e7934774ba0b43eea4b61202f7b9bf4e;hb=fe785e6f62ab58ef05972850d9275005794505b7;hpb=8f070dcdd579e19e092a6619115f4b084fbb2c33 diff --git a/effect.cpp b/effect.cpp index 927acd5..1bbb89f 100644 --- a/effect.cpp +++ b/effect.cpp @@ -1,7 +1,86 @@ +#define GL_GLEXT_PROTOTYPES 1 + #include #include #include #include "effect.h" +#include "effect_chain.h" +#include "util.h" + +#include +#include + +GLint get_uniform_location(GLuint glsl_program_num, const std::string &prefix, const std::string &key) +{ + std::string name = prefix + "_" + key; + return glGetUniformLocation(glsl_program_num, name.c_str()); +} + +void set_uniform_int(GLuint glsl_program_num, const std::string &prefix, const std::string &key, int value) +{ + GLint location = get_uniform_location(glsl_program_num, prefix, key); + if (location == -1) { + return; + } + check_error(); + glUniform1i(location, value); + check_error(); +} + +void set_uniform_float(GLuint glsl_program_num, const std::string &prefix, const std::string &key, float value) +{ + GLint location = get_uniform_location(glsl_program_num, prefix, key); + if (location == -1) { + return; + } + check_error(); + glUniform1f(location, value); + check_error(); +} + +void set_uniform_float_array(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values, size_t num_values) +{ + GLint location = get_uniform_location(glsl_program_num, prefix, key); + if (location == -1) { + return; + } + check_error(); + glUniform1fv(location, num_values, values); + check_error(); +} + +void set_uniform_vec2(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values) +{ + GLint location = get_uniform_location(glsl_program_num, prefix, key); + if (location == -1) { + return; + } + check_error(); + glUniform2fv(location, 1, values); + check_error(); +} + +void set_uniform_vec3(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values) +{ + GLint location = get_uniform_location(glsl_program_num, prefix, key); + if (location == -1) { + return; + } + check_error(); + glUniform3fv(location, 1, values); + check_error(); +} + +void set_uniform_vec4_array(GLuint glsl_program_num, const std::string &prefix, const std::string &key, const float *values, size_t num_values) +{ + GLint location = get_uniform_location(glsl_program_num, prefix, key); + if (location == -1) { + return; + } + check_error(); + glUniform4fv(location, num_values, values); + check_error(); +} bool Effect::set_int(const std::string &key, int value) { @@ -21,6 +100,15 @@ bool Effect::set_float(const std::string &key, float value) return true; } +bool Effect::set_vec2(const std::string &key, const float *values) +{ + if (params_vec2.count(key) == 0) { + return false; + } + memcpy(params_vec2[key], values, sizeof(float) * 2); + return true; +} + bool Effect::set_vec3(const std::string &key, const float *values) { if (params_vec3.count(key) == 0) { @@ -42,15 +130,54 @@ void Effect::register_float(const std::string &key, float *value) params_float[key] = value; } +void Effect::register_vec2(const std::string &key, float *values) +{ + assert(params_vec2.count(key) == 0); + params_vec2[key] = values; +} + void Effect::register_vec3(const std::string &key, float *values) { assert(params_vec3.count(key) == 0); params_vec3[key] = values; } +void Effect::register_1d_texture(const std::string &key, float *values, size_t size) +{ + assert(params_tex_1d.count(key) == 0); + + Texture1D tex; + tex.values = values; + tex.size = size; + tex.needs_update = false; + glGenTextures(1, &tex.texture_num); + + glBindTexture(GL_TEXTURE_1D, tex.texture_num); + check_error(); + glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_MIN_FILTER, GL_LINEAR); + check_error(); + glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE); + check_error(); + glTexImage1D(GL_TEXTURE_1D, 0, GL_LUMINANCE16F_ARB, size, 0, GL_LUMINANCE, GL_FLOAT, values); + check_error(); + + params_tex_1d[key] = tex; +} + +void Effect::invalidate_1d_texture(const std::string &key) +{ + assert(params_tex_1d.count(key) != 0); + params_tex_1d[key].needs_update = true; +} + +void Effect::add_self_to_effect_chain(EffectChain *chain, const std::vector &inputs) +{ + chain->add_effect_raw(this, inputs); +} + // Output convenience uniforms for each parameter. // These will be filled in per-frame. -std::string Effect::output_convenience_uniforms() +std::string Effect::output_convenience_uniforms() const { std::string output = ""; for (std::map::const_iterator it = params_float.begin(); @@ -60,6 +187,13 @@ std::string Effect::output_convenience_uniforms() sprintf(buf, "uniform float PREFIX(%s);\n", it->first.c_str()); output.append(buf); } + for (std::map::const_iterator it = params_vec2.begin(); + it != params_vec2.end(); + ++it) { + char buf[256]; + sprintf(buf, "uniform vec2 PREFIX(%s);\n", it->first.c_str()); + output.append(buf); + } for (std::map::const_iterator it = params_vec3.begin(); it != params_vec3.end(); ++it) { @@ -67,5 +201,51 @@ std::string Effect::output_convenience_uniforms() sprintf(buf, "uniform vec3 PREFIX(%s);\n", it->first.c_str()); output.append(buf); } + for (std::map::const_iterator it = params_tex_1d.begin(); + it != params_tex_1d.end(); + ++it) { + char buf[256]; + sprintf(buf, "uniform sampler1D PREFIX(%s);\n", it->first.c_str()); + output.append(buf); + } return output; } + +void Effect::set_gl_state(GLuint glsl_program_num, const std::string& prefix, unsigned *sampler_num) +{ + for (std::map::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::const_iterator it = params_vec2.begin(); + it != params_vec2.end(); + ++it) { + set_uniform_vec2(glsl_program_num, prefix, it->first, it->second); + } + for (std::map::const_iterator it = params_vec3.begin(); + it != params_vec3.end(); + ++it) { + set_uniform_vec3(glsl_program_num, prefix, it->first, it->second); + } + + for (std::map::iterator it = params_tex_1d.begin(); + it != params_tex_1d.end(); + ++it) { + glActiveTexture(GL_TEXTURE0 + *sampler_num); + check_error(); + glBindTexture(GL_TEXTURE_1D, it->second.texture_num); + check_error(); + + if (it->second.needs_update) { + glTexImage1D(GL_TEXTURE_1D, 0, GL_LUMINANCE16F_ARB, it->second.size, 0, GL_LUMINANCE, GL_FLOAT, it->second.values); + check_error(); + it->second.needs_update = false; + } + + set_uniform_int(glsl_program_num, prefix, it->first, *sampler_num); + ++*sampler_num; + } +} + +void Effect::clear_gl_state() {}