Split glsl/generate.cpp in two

[libs/gl.git] / source / glsl / resolve.h

diff --git a/source/glsl/resolve.h b/source/glsl/resolve.h
new file mode 100644 (file)
index 0000000..ca9c4f3
--- /dev/null
+++ b/source/glsl/resolve.h
@@ -0,0 +1,170 @@
+#ifndef MSP_GL_SL_RESOLVE_H_
+#define MSP_GL_SL_RESOLVE_H_
+
+#include <map>
+#include <set>
+#include <string>
+#include <vector>
+#include "visitor.h"
+
+namespace Msp {
+namespace GL {
+namespace SL {
+
+/** Forms links between nested blocks in the syntax tree. */
+class BlockHierarchyResolver: private TraversingVisitor
+{
+private:
+       bool r_any_resolved;
+
+public:
+       BlockHierarchyResolver(): r_any_resolved(false) { }
+
+       bool apply(Stage &s) { r_any_resolved = false; s.content.visit(*this); return r_any_resolved; }
+
+private:
+       virtual void enter(Block &);
+};
+
+/** Resolves types of variables and base types of other types. */
+class TypeResolver: private TraversingVisitor
+{
+private:
+       Stage *stage;
+       std::map<TypeDeclaration *, TypeDeclaration *> alias_map;
+       std::map<TypeDeclaration *, TypeDeclaration *> array_types;
+       NodeList<Statement>::iterator type_insert_point;
+       InterfaceBlock *iface_block;
+       bool r_any_resolved;
+
+public:
+       TypeResolver();
+
+       bool apply(Stage &);
+
+private:
+       TypeDeclaration *get_or_create_array_type(TypeDeclaration &);
+       void resolve_type(TypeDeclaration *&, const std::string &, bool);
+       virtual void visit(Block &);
+       virtual void visit(BasicTypeDeclaration &);
+       virtual void visit(ImageTypeDeclaration &);
+       virtual void visit(StructDeclaration &);
+       virtual void visit(VariableDeclaration &);
+       virtual void visit(InterfaceBlock &);
+       virtual void visit(FunctionDeclaration &);
+};
+
+/** Resolves variable references.  Variable references which match the name
+of an interface block are turned into interface block references. */
+class VariableResolver: private TraversingVisitor
+{
+private:
+       Stage *stage;
+       RefPtr<Expression> r_replacement_expr;
+       bool r_any_resolved;
+       bool record_target;
+       bool r_self_referencing;
+       Assignment::Target r_assignment_target;
+       std::vector<VariableDeclaration *> redeclared_builtins;
+       std::set<Node *> nodes_to_remove;
+
+public:
+       VariableResolver();
+
+       bool apply(Stage &);
+
+private:
+       virtual void enter(Block &);
+       virtual void visit(RefPtr<Expression> &);
+       void check_assignment_target(Statement *);
+       virtual void visit(VariableReference &);
+       virtual void visit(InterfaceBlockReference &);
+       void add_to_chain(Assignment::Target::ChainType, unsigned);
+       virtual void visit(MemberAccess &);
+       virtual void visit(Swizzle &);
+       virtual void visit(BinaryExpression &);
+       virtual void visit(Assignment &);
+       void merge_layouts(Layout &, const Layout &);
+       virtual void visit(VariableDeclaration &);
+       virtual void visit(InterfaceBlock &);
+};
+
+/** Resolves types and lvalueness of expressions. */
+class ExpressionResolver: private TraversingVisitor
+{
+private:
+       enum Compatibility
+       {
+               NOT_COMPATIBLE,
+               LEFT_CONVERTIBLE,
+               RIGHT_CONVERTIBLE,
+               SAME_TYPE
+       };
+
+       struct ArgumentInfo
+       {
+               BasicTypeDeclaration *type;
+               unsigned component_count;
+       };
+
+       Stage *stage;
+       std::vector<BasicTypeDeclaration *> basic_types;
+       NodeList<Statement>::iterator insert_point;
+       bool r_any_resolved;
+
+public:
+       ExpressionResolver();
+
+       bool apply(Stage &);
+
+private:
+       static bool is_scalar(BasicTypeDeclaration &);
+       static bool is_vector_or_matrix(BasicTypeDeclaration &);
+       static BasicTypeDeclaration *get_element_type(BasicTypeDeclaration &);
+       static bool can_convert(BasicTypeDeclaration &, BasicTypeDeclaration &);
+       static Compatibility get_compatibility(BasicTypeDeclaration &, BasicTypeDeclaration &);
+       BasicTypeDeclaration *find_type(BasicTypeDeclaration::Kind, unsigned);
+       BasicTypeDeclaration *find_type(BasicTypeDeclaration &, BasicTypeDeclaration::Kind, unsigned);
+       void convert_to(RefPtr<Expression> &, BasicTypeDeclaration &);
+       bool convert_to_element(RefPtr<Expression> &, BasicTypeDeclaration &);
+       bool truncate_vector(RefPtr<Expression> &, unsigned);
+       void resolve(Expression &, TypeDeclaration *, bool);
+
+       virtual void visit(Block &);
+       virtual void visit(Literal &);
+       virtual void visit(VariableReference &);
+       virtual void visit(InterfaceBlockReference &);
+       virtual void visit(MemberAccess &);
+       virtual void visit(Swizzle &);
+       virtual void visit(UnaryExpression &);
+       void visit(BinaryExpression &, bool);
+       virtual void visit(BinaryExpression &);
+       virtual void visit(Assignment &);
+       virtual void visit(TernaryExpression &);
+       void visit_constructor(FunctionCall &);
+       virtual void visit(FunctionCall &);
+       virtual void visit(BasicTypeDeclaration &);
+       virtual void visit(VariableDeclaration &);
+};
+
+/** Resolves function declarations and calls. */
+class FunctionResolver: private TraversingVisitor
+{
+private:
+       Stage *stage;
+       std::map<std::string, std::vector<FunctionDeclaration *> > declarations;
+       bool r_any_resolved;
+
+public:
+       bool apply(Stage &);
+
+private:
+       virtual void visit(FunctionCall &);
+       virtual void visit(FunctionDeclaration &);
+};
+
+} // namespace SL
+} // namespace GL
+} // namespace Msp
+
+#endif