source/programcompiler.h

   1 #ifndef MSP_GL_PROGRAMCOMPILER_H_
   2 #define MSP_GL_PROGRAMCOMPILER_H_
   3
   4 #include <set>
   5 #include "programparser.h"
   6 #include "programsyntax.h"
   7
   8 namespace Msp {
   9 namespace GL {
  10
  11 class Program;
  12
  13 class ProgramCompiler
  14 {
  15 public:
  16         struct Formatter: ProgramSyntax::NodeVisitor
  17         {
  18                 std::string formatted;
  19                 unsigned indent;
  20                 bool parameter_list;
  21                 bool else_if;
  22
  23                 Formatter();
  24
  25                 virtual void visit(ProgramSyntax::Block &);
  26                 virtual void visit(ProgramSyntax::Literal &);
  27                 virtual void visit(ProgramSyntax::ParenthesizedExpression &);
  28                 virtual void visit(ProgramSyntax::VariableReference &);
  29                 virtual void visit(ProgramSyntax::MemberAccess &);
  30                 virtual void visit(ProgramSyntax::UnaryExpression &);
  31                 virtual void visit(ProgramSyntax::BinaryExpression &);
  32                 virtual void visit(ProgramSyntax::FunctionCall &);
  33                 virtual void visit(ProgramSyntax::ExpressionStatement &);
  34                 virtual void visit(ProgramSyntax::Layout &);
  35                 virtual void visit(ProgramSyntax::StructDeclaration &);
  36                 virtual void visit(ProgramSyntax::VariableDeclaration &);
  37                 virtual void visit(ProgramSyntax::InterfaceBlock &);
  38                 virtual void visit(ProgramSyntax::FunctionDeclaration &);
  39                 virtual void visit(ProgramSyntax::Conditional &);
  40                 virtual void visit(ProgramSyntax::Iteration &);
  41                 virtual void visit(ProgramSyntax::Return &);
  42         };
  43
  44 private:
  45         struct VariableResolver: ProgramSyntax::TraversingVisitor
  46         {
  47                 std::vector<ProgramSyntax::Block *> blocks;
  48                 ProgramSyntax::StructDeclaration *type;
  49                 bool anonymous;
  50
  51                 VariableResolver();
  52
  53                 virtual void visit(ProgramSyntax::Block &);
  54                 virtual void visit(ProgramSyntax::VariableReference &);
  55                 virtual void visit(ProgramSyntax::MemberAccess &);
  56                 virtual void visit(ProgramSyntax::BinaryExpression &);
  57                 virtual void visit(ProgramSyntax::StructDeclaration &);
  58                 virtual void visit(ProgramSyntax::VariableDeclaration &);
  59                 virtual void visit(ProgramSyntax::InterfaceBlock &);
  60         };
  61
  62         struct InterfaceGenerator: ProgramSyntax::TraversingVisitor
  63         {
  64                 ProgramSyntax::Context *context;
  65                 std::string in_prefix;
  66                 std::string out_prefix;
  67                 unsigned scope_level;
  68                 std::map<std::string, ProgramSyntax::NodePtr<ProgramSyntax::Node> > iface_declarations;
  69                 bool remove_node;
  70                 std::list<ProgramSyntax::NodePtr<ProgramSyntax::Node> > insert_nodes;
  71
  72                 InterfaceGenerator();
  73
  74                 static std::string get_out_prefix(ProgramSyntax::ContextType);
  75                 void visit(ProgramSyntax::Context &);
  76                 virtual void visit(ProgramSyntax::Block &);
  77                 std::string change_prefix(const std::string &, const std::string &) const;
  78                 bool generate_interface(ProgramSyntax::VariableDeclaration &, const std::string &, const std::string &);
  79                 void insert_assignment(const std::string &, ProgramSyntax::Expression *);
  80                 virtual void visit(ProgramSyntax::VariableReference &);
  81                 virtual void visit(ProgramSyntax::VariableDeclaration &);
  82                 virtual void visit(ProgramSyntax::Passthrough &);
  83         };
  84
  85         struct VariableRenamer: ProgramSyntax::TraversingVisitor
  86         {
  87                 virtual void visit(ProgramSyntax::VariableReference &);
  88                 virtual void visit(ProgramSyntax::VariableDeclaration &);
  89         };
  90
  91         struct UnusedVariableLocator: ProgramSyntax::TraversingVisitor
  92         {
  93                 std::set<ProgramSyntax::VariableDeclaration *> unused_variables;
  94
  95                 virtual void visit(ProgramSyntax::VariableReference &);
  96                 virtual void visit(ProgramSyntax::MemberAccess &);
  97                 virtual void visit(ProgramSyntax::VariableDeclaration &);
  98         };
  99
 100         struct NodeRemover: ProgramSyntax::TraversingVisitor
 101         {
 102                 std::set<ProgramSyntax::Node *> to_remove;
 103                 unsigned n_removed;
 104                 bool immutable_block;
 105                 bool remove_block;
 106
 107                 NodeRemover();
 108
 109                 virtual void visit(ProgramSyntax::Block &);
 110                 virtual void visit(ProgramSyntax::StructDeclaration &);
 111                 virtual void visit(ProgramSyntax::InterfaceBlock &);
 112         };
 113
 114         ProgramParser parser;
 115         ProgramSyntax::Module *module;
 116
 117 public:
 118         ProgramCompiler();
 119
 120         void compile(const std::string &);
 121         void compile(IO::Base &);
 122         void add_shaders(Program &);
 123
 124 private:
 125         void process();
 126         void process(ProgramSyntax::Context &);
 127         static void inject_block(ProgramSyntax::Block &, const ProgramSyntax::Block &);
 128         static void resolve_variables(ProgramSyntax::Context &);
 129         std::string format_context(ProgramSyntax::Context &);
 130 };
 131
 132 } // namespace GL
 133 } // namespace Msp
 134
 135 #endif