X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=effect_chain.h;h=cf96293b7ebc79a0c4619201399eb254626f7be7;hp=9d389d53f3d25dee54bfcc130653bc8fc5d6416c;hb=cfe0bc4fa1e2a56eeb12c33e596f79c1292292c8;hpb=416abbec4dbde5ce6b05ada2b2977b18a6718190 diff --git a/effect_chain.h b/effect_chain.h index 9d389d5..cf96293 100644 --- a/effect_chain.h +++ b/effect_chain.h @@ -8,6 +8,50 @@ #include "image_format.h" #include "input.h" +class EffectChain; +class Phase; + +// A node in the graph; basically an effect and some associated information. +class Node { +public: + Effect *effect; + bool disabled; + + // 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; + + // 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; +}; + +// 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 { public: EffectChain(unsigned width, unsigned height); @@ -37,16 +81,6 @@ public: } Effect *add_effect(Effect *effect, const std::vector &inputs); - // Similar to add_effect, but: - // - // * Does not insert any normalizing effects. - // * Does not ask the effect to insert itself, so it won't work - // with meta-effects. - // - // We should really separate out these two “sides” of Effect in the - // type system soon. - void add_effect_raw(Effect *effect, const std::vector &inputs); - void add_output(const ImageFormat &format); void finalize(); @@ -61,56 +95,25 @@ public: } } -private: - struct Phase; - - // A node in the graph; basically an effect and some associated information. - struct Node { - Effect *effect; - - // Identifier used to create unique variables in GLSL. - std::string effect_id; - - // Edges in the graph (forward and backward). - std::vector outgoing_links; - std::vector incoming_links; - - // 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; - }; - - // 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; - }; + // API for manipulating the graph directly. Intended to be used from + // effects and by EffectChain itself. + // + // Note that for nodes with multiple inputs, the order of calls to + // connect_nodes() will matter. + Node *add_node(Effect *effect); + void connect_nodes(Node *sender, Node *receiver); + void replace_receiver(Node *old_receiver, Node *new_receiver); + void replace_sender(Node *new_sender, Node *receiver); + void insert_node_between(Node *sender, Node *middle, Node *receiver); +private: // 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); - 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); - - void draw_vertex(float x, float y, const std::vector &inputs); + // Find all inputs eventually feeding into this effect that have + // output gamma different from GAMMA_LINEAR. + void find_all_nonlinear_inputs(Node *effect, std::vector *nonlinear_inputs); // Create a GLSL program computing the given effects in order. Phase *compile_glsl_program(const std::vector &inputs, @@ -120,6 +123,29 @@ private: // that depends on it (whenever possible). 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); + + // Some of the graph algorithms assume that the nodes array is sorted + // topologically (inputs are always before outputs), but some operations + // (like graph rewriting) can change that. This function restores that order. + void sort_nodes_topologically(); + void topological_sort_visit_node(Node *node, std::set *visited_nodes, std::vector *sorted_list); + + // Used during finalize(). + void propagate_gamma_and_color_space(); + Node *find_output_node(); + + bool node_needs_colorspace_fix(Node *node); + void fix_internal_color_spaces(); + void fix_output_color_space(); + + bool node_needs_gamma_fix(Node *node); + void fix_internal_gamma_by_asking_inputs(unsigned step); + void fix_internal_gamma_by_inserting_nodes(unsigned step); + void fix_output_gamma(); + unsigned width, height; ImageFormat output_format; @@ -135,5 +161,4 @@ private: bool finalized; }; - #endif // !defined(_EFFECT_CHAIN_H)