source/glsl/visitor.h

   1 #ifndef MSP_GL_SL_VISITOR_H_
   2 #define MSP_GL_SL_VISITOR_H_
   3
   4 #include <set>
   5 #include <vector>
   6 #include "syntax.h"
   7
   8 namespace Msp {
   9 namespace GL {
  10 namespace SL {
  11
  12 class NodeVisitor
  13 {
  14 protected:
  15         NodeVisitor() { }
  16 public:
  17         virtual ~NodeVisitor() { }
  18
  19         virtual void visit(Block &) { }
  20         virtual void visit(Literal &) { }
  21         virtual void visit(ParenthesizedExpression &) { }
  22         virtual void visit(VariableReference &) { }
  23         virtual void visit(MemberAccess &) { }
  24         virtual void visit(UnaryExpression &) { }
  25         virtual void visit(BinaryExpression &) { }
  26         virtual void visit(Assignment &);
  27         virtual void visit(FunctionCall &) { }
  28         virtual void visit(ExpressionStatement &) { }
  29         virtual void visit(Import &) { }
  30         virtual void visit(Precision &) { }
  31         virtual void visit(Layout &) { }
  32         virtual void visit(InterfaceLayout &) { }
  33         virtual void visit(StructDeclaration &) { }
  34         virtual void visit(VariableDeclaration &) { }
  35         virtual void visit(InterfaceBlock &) { }
  36         virtual void visit(FunctionDeclaration &) { }
  37         virtual void visit(Conditional &) { }
  38         virtual void visit(Iteration &) { }
  39         virtual void visit(Passthrough &) { }
  40         virtual void visit(Return &) { }
  41         virtual void visit(Jump &) { }
  42 };
  43
  44 class TraversingVisitor: public NodeVisitor
  45 {
  46 protected:
  47         TraversingVisitor() { }
  48
  49 public:
  50         using NodeVisitor::visit;
  51         virtual void visit(Block &);
  52         virtual void visit(ParenthesizedExpression &);
  53         virtual void visit(MemberAccess &);
  54         virtual void visit(UnaryExpression &);
  55         virtual void visit(BinaryExpression &);
  56         virtual void visit(FunctionCall &);
  57         virtual void visit(ExpressionStatement &);
  58         virtual void visit(InterfaceLayout &);
  59         virtual void visit(StructDeclaration &);
  60         virtual void visit(VariableDeclaration &);
  61         virtual void visit(InterfaceBlock &);
  62         virtual void visit(FunctionDeclaration &);
  63         virtual void visit(Conditional &);
  64         virtual void visit(Iteration &);
  65         virtual void visit(Passthrough &);
  66         virtual void visit(Return &);
  67 };
  68
  69 class StageVisitor: public TraversingVisitor
  70 {
  71 public:
  72         typedef void ResultType;
  73
  74 protected:
  75         Stage *stage;
  76
  77         StageVisitor();
  78
  79 public:
  80         virtual void apply(Stage &);
  81         void get_result() const { }
  82 };
  83
  84 class BlockModifier: public StageVisitor
  85 {
  86 protected:
  87         bool remove_node;
  88         std::vector<RefPtr<Statement> > insert_nodes;
  89
  90         BlockModifier();
  91
  92         void flatten_block(Block &);
  93         void apply_and_increment(Block &, NodeList<Statement>::iterator &);
  94
  95 public:
  96         using StageVisitor::visit;
  97         virtual void visit(Block &);
  98 };
  99
 100 template<typename T>
 101 class NodeGatherer: public StageVisitor
 102 {
 103 public:
 104         typedef std::vector<T *> ResultType;
 105
 106 private:
 107         std::vector<T *> nodes;
 108
 109 public:
 110         const ResultType &get_result() const { return nodes; }
 111         using StageVisitor::visit;
 112         virtual void visit(T &n) { nodes.push_back(&n); }
 113 };
 114
 115 class NodeRemover: public StageVisitor
 116 {
 117 private:
 118         std::set<Node *> to_remove;
 119
 120 public:
 121         NodeRemover() { }
 122         NodeRemover(const std::set<Node *> &);
 123
 124         using StageVisitor::visit;
 125         virtual void visit(Block &);
 126         virtual void visit(VariableDeclaration &);
 127         virtual void visit(Iteration &);
 128 };
 129
 130 } // namespace SL
 131 } // namespace GL
 132 } // namespace Msp
 133
 134 #endif