X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=effect_chain.h;h=0abd51805793a2d40b4b9fd0f5394a056bad7dc0;hp=561d823ffa4cf208c7dc239ffc86bb071843394a;hb=45636792070587fb38cc84927f4e3d9c2ce8bd33;hpb=34121ba75f6274c25c2cee53169819ae96e77d66 diff --git a/effect_chain.h b/effect_chain.h index 561d823..0abd518 100644 --- a/effect_chain.h +++ b/effect_chain.h @@ -5,28 +5,50 @@ #include #include "effect.h" -#include "effect_id.h" +#include "image_format.h" +#include "input.h" -enum PixelFormat { FORMAT_RGB, FORMAT_RGBA, FORMAT_BGR, FORMAT_BGRA }; +class EffectChain; +class Phase; -enum ColorSpace { - COLORSPACE_sRGB = 0, - COLORSPACE_REC_709 = 0, // Same as sRGB. - COLORSPACE_REC_601_525 = 1, - COLORSPACE_REC_601_625 = 2, -}; +// A node in the graph; basically an effect and some associated information. +class Node { +public: + Effect *effect; + + // Edges in the graph (forward and backward). + std::vector outgoing_links; + std::vector incoming_links; + +private: + // Identifier used to create unique variables in GLSL. + std::string effect_id; -enum GammaCurve { - GAMMA_LINEAR = 0, - GAMMA_sRGB = 1, - GAMMA_REC_601 = 2, - GAMMA_REC_709 = 2, // Same as Rec. 601. + // If output goes to RTT (otherwise, none of these are set). + // The Phsae pointer is a but ugly; we should probably fix so + // that Phase takes other phases as inputs, instead of Node. + GLuint output_texture; + unsigned output_texture_width, output_texture_height; + Phase *phase; + + // Used during the building of the effect chain. + ColorSpace output_color_space; + GammaCurve output_gamma_curve; + + friend class EffectChain; }; -struct ImageFormat { - PixelFormat pixel_format; - ColorSpace color_space; - GammaCurve gamma_curve; +// A rendering phase; a single GLSL program rendering a single quad. +struct Phase { + GLint glsl_program_num; + bool input_needs_mipmaps; + + // Inputs are only inputs from other phases (ie., those that come from RTT); + // input textures are not counted here. + std::vector inputs; + + std::vector effects; // In order. + unsigned output_width, output_height; }; class EffectChain { @@ -36,24 +58,27 @@ public: // User API: // input, effects, output, finalize need to come in that specific order. - void add_input(const ImageFormat &format); + // EffectChain takes ownership of the given input. + // input is returned back for convenience. + Input *add_input(Input *input); - // The returned pointer is owned by EffectChain. - Effect *add_effect(EffectId effect) { + // EffectChain takes ownership of the given effect. + // effect is returned back for convenience. + Effect *add_effect(Effect *effect) { return add_effect(effect, last_added_effect()); } - Effect *add_effect(EffectId effect, Effect *input) { + Effect *add_effect(Effect *effect, Effect *input) { std::vector inputs; inputs.push_back(input); return add_effect(effect, inputs); } - Effect *add_effect(EffectId effect, Effect *input1, Effect *input2) { + Effect *add_effect(Effect *effect, Effect *input1, Effect *input2) { std::vector inputs; inputs.push_back(input1); inputs.push_back(input2); return add_effect(effect, inputs); } - Effect *add_effect(EffectId effect, const std::vector &inputs); + Effect *add_effect(Effect *effect, const std::vector &inputs); // Similar to add_effect, but: // @@ -69,57 +94,52 @@ public: void finalize(); //void render(unsigned char *src, unsigned char *dst); - void render_to_screen(unsigned char *src); + void render_to_screen(); Effect *last_added_effect() { - if (effects.empty()) { + if (nodes.empty()) { return NULL; } else { - return effects.back(); + return nodes.back()->effect; } } private: - struct Phase { - GLint glsl_program_num; - bool input_needs_mipmaps; - std::vector inputs; - std::vector effects; // In order. - }; + // Determine the preferred output size of a given phase. + // Requires that all input phases (if any) already have output sizes set. + void find_output_size(Phase *phase); - Effect *normalize_to_linear_gamma(Effect *input); - Effect *normalize_to_srgb(Effect *input); - - void draw_vertex(float x, float y, const std::vector &inputs); + void find_all_nonlinear_inputs(Node *effect, + std::vector *nonlinear_inputs, + std::vector *intermediates); + Node *normalize_to_linear_gamma(Node *input); + Node *normalize_to_srgb(Node *input); // Create a GLSL program computing the given effects in order. - Phase compile_glsl_program(const std::vector &inputs, const std::vector &effects); + Phase *compile_glsl_program(const std::vector &inputs, + const std::vector &effects); // Create all GLSL programs needed to compute the given effect, and all outputs // that depends on it (whenever possible). - void construct_glsl_programs(Effect *start, std::set *completed_effects); + void construct_glsl_programs(Node *output); + + // Output the current graph to the given file in a Graphviz-compatible format; + // only useful for debugging. + void output_dot(const char *filename); unsigned width, height; - ImageFormat input_format, output_format; - std::vector effects, unexpanded_effects; - std::map effect_ids; - std::map effect_output_textures; - std::map > outgoing_links; - std::map > incoming_links; + ImageFormat output_format; - GLuint source_image_num; - bool use_srgb_texture_format; + std::vector nodes; + std::map node_map; + + std::vector inputs; // Also contained in nodes. GLuint fbo; - std::vector phases; + std::vector phases; GLenum format, bytes_per_pixel; bool finalized; - - // Used during the building of the effect chain. - std::map output_color_space; - std::map output_gamma_curve; }; - #endif // !defined(_EFFECT_CHAIN_H)