X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=fft_pass_effect.h;h=a09304d22f264cd53b1890db90ac217031881224;hp=b3e025b1a1284bb0415b04320ba08e491b1fc3f9;hb=f5e3256da7d8e3a56c002da47bedf8ec1a2133f4;hpb=c4f0d4e876a8177db5738596f22349e030e0a1dc diff --git a/fft_pass_effect.h b/fft_pass_effect.h index b3e025b..a09304d 100644 --- a/fft_pass_effect.h +++ b/fft_pass_effect.h @@ -50,17 +50,20 @@ // scaling), and as fp16 has quite limited range at times, this can be relevant // on some GPUs for larger sizes. +#include +#include #include -#include #include #include "effect.h" +namespace movit { + class FFTPassEffect : public Effect { public: FFTPassEffect(); ~FFTPassEffect(); - virtual std::string effect_type_id() const { + std::string effect_type_id() const override { char buf[256]; if (inverse) { snprintf(buf, sizeof(buf), "IFFTPassEffect[%d]", (1 << pass_number)); @@ -69,9 +72,9 @@ public: } return buf; } - std::string output_fragment_shader(); + std::string output_fragment_shader() override; - void set_gl_state(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num); + void set_gl_state(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num) override; // We don't actually change the output size, but this flag makes sure // that no other effect is chained after us. This is important since @@ -80,31 +83,48 @@ public: // mode to GL_NEAREST, which other effects are not ready for; so, the // combination of these two flags guarantee that we're run entirely alone // in our own phase, which is exactly what we want. - virtual bool needs_texture_bounce() const { return true; } - virtual bool changes_output_size() const { return true; } + bool needs_texture_bounce() const override { return true; } + bool changes_output_size() const override { return true; } + bool sets_virtual_output_size() const override { return false; } - virtual void inform_input_size(unsigned input_num, unsigned width, unsigned height) + void inform_input_size(unsigned input_num, unsigned width, unsigned height) override { assert(input_num == 0); input_width = width; input_height = height; } - virtual void get_output_size(unsigned *width, unsigned *height, - unsigned *virtual_width, unsigned *virtual_height) const { + void get_output_size(unsigned *width, unsigned *height, + unsigned *virtual_width, unsigned *virtual_height) const override { *width = *virtual_width = input_width; *height = *virtual_height = input_height; } + + void inform_added(EffectChain *chain) override { this->chain = chain; } - enum Direction { HORIZONTAL = 0, VERTICAL = 1 }; + enum Direction { INVALID = -1, HORIZONTAL = 0, VERTICAL = 1 }; private: + void generate_support_texture(); + + EffectChain *chain; int input_width, input_height; GLuint tex; + float uniform_num_repeats; + GLint uniform_support_tex; + int fft_size; Direction direction; int pass_number; // From 1..n. int inverse; // 0 = forward (FFT), 1 = reverse (IFFT). + + int last_fft_size; + Direction last_direction; + int last_pass_number; + int last_inverse; + int last_input_size; }; +} // namespace movit + #endif // !defined(_MOVIT_FFT_PASS_EFFECT_H)