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(InterfaceBlockReference &) { }
  24         virtual void visit(MemberAccess &) { }
  25         virtual void visit(UnaryExpression &) { }
  26         virtual void visit(BinaryExpression &) { }
  27         virtual void visit(Assignment &);
  28         virtual void visit(FunctionCall &) { }
  29         virtual void visit(ExpressionStatement &) { }
  30         virtual void visit(Import &) { }
  31         virtual void visit(Precision &) { }
  32         virtual void visit(Layout &) { }
  33         virtual void visit(InterfaceLayout &) { }
  34         virtual void visit(StructDeclaration &) { }
  35         virtual void visit(VariableDeclaration &) { }
  36         virtual void visit(InterfaceBlock &) { }
  37         virtual void visit(FunctionDeclaration &) { }
  38         virtual void visit(Conditional &) { }
  39         virtual void visit(Iteration &) { }
  40         virtual void visit(Passthrough &) { }
  41         virtual void visit(Return &) { }
  42         virtual void visit(Jump &) { }
  43 };
  44
  45 class TraversingVisitor: public NodeVisitor
  46 {
  47 protected:
  48         Block *current_block;
  49
  50         TraversingVisitor(): current_block(0) { }
  51
  52 public:
  53         virtual void enter(Block &) { }
  54         virtual void visit(Block &);
  55         virtual void visit(ParenthesizedExpression &);
  56         virtual void visit(MemberAccess &);
  57         virtual void visit(UnaryExpression &);
  58         virtual void visit(BinaryExpression &);
  59         virtual void visit(FunctionCall &);
  60         virtual void visit(ExpressionStatement &);
  61         virtual void visit(InterfaceLayout &);
  62         virtual void visit(StructDeclaration &);
  63         virtual void visit(VariableDeclaration &);
  64         virtual void visit(InterfaceBlock &);
  65         virtual void visit(FunctionDeclaration &);
  66         virtual void visit(Conditional &);
  67         virtual void visit(Iteration &);
  68         virtual void visit(Passthrough &);
  69         virtual void visit(Return &);
  70 };
  71
  72 template<typename T>
  73 class NodeGatherer: private TraversingVisitor
  74 {
  75 private:
  76         std::vector<T *> nodes;
  77
  78 public:
  79         const std::vector<T *> &apply(Stage &s) { s.content.visit(*this); return nodes; }
  80
  81 private:
  82         virtual void visit(T &n) { nodes.push_back(&n); }
  83 };
  84
  85 class NodeRemover: private TraversingVisitor
  86 {
  87 private:
  88         Stage *stage;
  89         const std::set<Node *> *to_remove;
  90         bool recursive_remove;
  91
  92 public:
  93         NodeRemover();
  94
  95         void apply(Stage &, const std::set<Node *> &);
  96
  97 private:
  98         template<typename T>
  99         void remove_from_map(std::map<std::string, T *> &, const std::string &, T &);
 100
 101         virtual void visit(Block &);
 102         virtual void visit(StructDeclaration &);
 103         virtual void visit(VariableDeclaration &);
 104         virtual void visit(InterfaceBlock &);
 105         virtual void visit(Iteration &);
 106 };
 107
 108 } // namespace SL
 109 } // namespace GL
 110 } // namespace Msp
 111
 112 #endif