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 UnusedVariableLocator: ProgramSyntax::TraversingVisitor
  63         {
  64                 std::set<ProgramSyntax::VariableDeclaration *> unused_variables;
  65
  66                 virtual void visit(ProgramSyntax::VariableReference &);
  67                 virtual void visit(ProgramSyntax::MemberAccess &);
  68                 virtual void visit(ProgramSyntax::VariableDeclaration &);
  69         };
  70
  71         struct NodeRemover: ProgramSyntax::TraversingVisitor
  72         {
  73                 std::set<ProgramSyntax::Node *> to_remove;
  74                 unsigned n_removed;
  75                 bool immutable_block;
  76                 bool remove_block;
  77
  78                 NodeRemover();
  79
  80                 virtual void visit(ProgramSyntax::Block &);
  81                 virtual void visit(ProgramSyntax::StructDeclaration &);
  82                 virtual void visit(ProgramSyntax::InterfaceBlock &);
  83         };
  84
  85         ProgramParser parser;
  86         ProgramSyntax::Module *module;
  87
  88 public:
  89         ProgramCompiler();
  90
  91         void compile(const std::string &);
  92         void compile(IO::Base &);
  93         void add_shaders(Program &);
  94
  95 private:
  96         void process();
  97         void process(ProgramSyntax::Context &);
  98         static void inject_block(ProgramSyntax::Block &, const ProgramSyntax::Block &);
  99         std::string format_context(ProgramSyntax::Context &);
 100 };
 101
 102 } // namespace GL
 103 } // namespace Msp
 104
 105 #endif