source/programsyntax.h

   1 #ifndef MSP_GL_PROGRAMSYNTAX_H_
   2 #define MSP_GL_PROGRAMSYNTAX_H_
   3
   4 #include <list>
   5 #include <map>
   6 #include <string>
   7 #include <vector>
   8 #include <msp/core/refptr.h>
   9 #include "extension.h"
  10
  11 namespace Msp {
  12 namespace GL {
  13 namespace ProgramSyntax {
  14
  15 struct NodeVisitor;
  16
  17 struct Node
  18 {
  19 private:
  20         Node &operator=(const Node &);
  21 public:
  22         virtual ~Node() { }
  23
  24         virtual Node *clone() const = 0;
  25         virtual void visit(NodeVisitor &) = 0;
  26 };
  27
  28 template<typename T>
  29 class NodePtr: public RefPtr<T>
  30 {
  31 public:
  32         NodePtr() { }
  33         NodePtr(T *p): RefPtr<T>(p) { }
  34         NodePtr(const NodePtr &p): RefPtr<T>(p ? p->clone() : 0) { }
  35
  36         template<typename U>
  37         NodePtr(const RefPtr<U> &p): RefPtr<T>(p) { }
  38
  39         template<typename U>
  40         NodePtr(const NodePtr<U> &p): RefPtr<T>(p ? p->clone() : 0) { }
  41 };
  42
  43 template<typename C>
  44 class NodeContainer: public C
  45 {
  46 public:
  47         NodeContainer() { }
  48         NodeContainer(const NodeContainer &);
  49 };
  50
  51 template<typename T>
  52 class NodeList: public NodeContainer<std::list<RefPtr<T> > >
  53 { };
  54
  55 template<typename T>
  56 class NodeArray: public NodeContainer<std::vector<RefPtr<T> > >
  57 { };
  58
  59 struct StructDeclaration;
  60 struct VariableDeclaration;
  61 struct FunctionDeclaration;
  62
  63 struct Block: Node
  64 {
  65         NodeList<Node> body;
  66         bool use_braces;
  67         std::map<std::string, StructDeclaration *> types;
  68         std::map<std::string, VariableDeclaration *> variables;
  69
  70         Block();
  71
  72         virtual Block *clone() const { return new Block(*this); }
  73         virtual void visit(NodeVisitor &);
  74 };
  75
  76 struct Expression: Node
  77 {
  78         virtual Expression *clone() const = 0;
  79 };
  80
  81 struct Literal: Expression
  82 {
  83         std::string token;
  84
  85         virtual Literal *clone() const { return new Literal(*this); }
  86         virtual void visit(NodeVisitor &);
  87 };
  88
  89 struct ParenthesizedExpression: Expression
  90 {
  91         NodePtr<Expression> expression;
  92
  93         virtual ParenthesizedExpression *clone() const { return new ParenthesizedExpression(*this); }
  94         virtual void visit(NodeVisitor &);
  95 };
  96
  97 struct VariableReference: Expression
  98 {
  99         std::string name;
 100         VariableDeclaration *declaration;
 101
 102         VariableReference();
 103
 104         virtual VariableReference *clone() const { return new VariableReference(*this); }
 105         virtual void visit(NodeVisitor &);
 106 };
 107
 108 struct MemberAccess: Expression
 109 {
 110         NodePtr<Expression> left;
 111         std::string member;
 112         VariableDeclaration *declaration;
 113
 114         virtual MemberAccess *clone() const { return new MemberAccess(*this); }
 115         virtual void visit(NodeVisitor &);
 116 };
 117
 118 struct UnaryExpression: Expression
 119 {
 120         std::string oper;
 121         NodePtr<Expression> expression;
 122         bool prefix;
 123
 124         UnaryExpression();
 125
 126         virtual UnaryExpression *clone() const { return new UnaryExpression(*this); }
 127         virtual void visit(NodeVisitor &);
 128 };
 129
 130 struct BinaryExpression: Expression
 131 {
 132         NodePtr<Expression> left;
 133         std::string oper;
 134         NodePtr<Expression> right;
 135         std::string after;
 136
 137         virtual BinaryExpression *clone() const { return new BinaryExpression(*this); }
 138         virtual void visit(NodeVisitor &);
 139 };
 140
 141 struct Assignment: BinaryExpression
 142 {
 143         bool self_referencing;
 144         VariableDeclaration *target_declaration;
 145
 146         Assignment();
 147
 148         virtual Assignment *clone() const { return new Assignment(*this); }
 149         virtual void visit(NodeVisitor &);
 150 };
 151
 152 struct FunctionCall: Expression
 153 {
 154         std::string name;
 155         FunctionDeclaration *declaration;
 156         bool constructor;
 157         NodeArray<Expression> arguments;
 158
 159         FunctionCall();
 160
 161         virtual FunctionCall *clone() const { return new FunctionCall(*this); }
 162         virtual void visit(NodeVisitor &);
 163 };
 164
 165 struct ExpressionStatement: Node
 166 {
 167         NodePtr<Expression> expression;
 168
 169         virtual ExpressionStatement *clone() const { return new ExpressionStatement(*this); }
 170         virtual void visit(NodeVisitor &);
 171 };
 172
 173 struct Import: Node
 174 {
 175         std::string module;
 176
 177         virtual Import *clone() const { return new Import(*this); }
 178         virtual void visit(NodeVisitor &);
 179 };
 180
 181 struct Precision: Node
 182 {
 183         std::string precision;
 184         std::string type;
 185
 186         virtual Precision *clone() const { return new Precision(*this); }
 187         virtual void visit(NodeVisitor &);
 188 };
 189
 190 struct Layout: Node
 191 {
 192         struct Qualifier
 193         {
 194                 std::string identifier;
 195                 std::string value;
 196         };
 197
 198         std::vector<Qualifier> qualifiers;
 199
 200         virtual Layout *clone() const { return new Layout(*this); }
 201         virtual void visit(NodeVisitor &);
 202 };
 203
 204 struct InterfaceLayout: Node
 205 {
 206         std::string interface;
 207         Layout layout;
 208
 209         virtual InterfaceLayout *clone() const { return new InterfaceLayout(*this); }
 210         virtual void visit(NodeVisitor &);
 211 };
 212
 213 struct StructDeclaration: Node
 214 {
 215         std::string name;
 216         Block members;
 217
 218         StructDeclaration();
 219
 220         virtual StructDeclaration *clone() const { return new StructDeclaration(*this); }
 221         virtual void visit(NodeVisitor &);
 222 };
 223
 224 struct VariableDeclaration: Node
 225 {
 226         bool constant;
 227         std::string sampling;
 228         std::string interface;
 229         std::string precision;
 230         std::string type;
 231         StructDeclaration *type_declaration;
 232         std::string name;
 233         bool array;
 234         NodePtr<Expression> array_size;
 235         NodePtr<Expression> init_expression;
 236         VariableDeclaration *linked_declaration;
 237         NodePtr<Layout> layout;
 238
 239         VariableDeclaration();
 240
 241         virtual VariableDeclaration *clone() const { return new VariableDeclaration(*this); }
 242         virtual void visit(NodeVisitor &);
 243 };
 244
 245 struct InterfaceBlock: Node
 246 {
 247         std::string interface;
 248         std::string name;
 249         Block members;
 250         std::string instance_name;
 251         bool array;
 252
 253         InterfaceBlock();
 254
 255         virtual InterfaceBlock *clone() const { return new InterfaceBlock(*this); }
 256         virtual void visit(NodeVisitor &);
 257 };
 258
 259 struct FunctionDeclaration: Node
 260 {
 261         std::string return_type;
 262         std::string name;
 263         NodeArray<VariableDeclaration> parameters;
 264         FunctionDeclaration *definition;
 265         Block body;
 266
 267         FunctionDeclaration();
 268         FunctionDeclaration(const FunctionDeclaration &);
 269
 270         virtual FunctionDeclaration *clone() const { return new FunctionDeclaration(*this); }
 271         virtual void visit(NodeVisitor &);
 272 };
 273
 274 struct Conditional: Node
 275 {
 276         NodePtr<Expression> condition;
 277         Block body;
 278         Block else_body;
 279
 280         virtual Conditional *clone() const { return new Conditional(*this); }
 281         virtual void visit(NodeVisitor &);
 282 };
 283
 284 struct Iteration: Node
 285 {
 286         NodePtr<Node> init_statement;
 287         NodePtr<Expression> condition;
 288         NodePtr<Expression> loop_expression;
 289         Block body;
 290
 291         virtual Iteration *clone() const { return new Iteration(*this); }
 292         virtual void visit(NodeVisitor &);
 293 };
 294
 295 struct Passthrough: Node
 296 {
 297         NodePtr<Expression> subscript;
 298
 299         virtual Passthrough *clone() const { return new Passthrough(*this); }
 300         virtual void visit(NodeVisitor &);
 301 };
 302
 303 struct Return: Node
 304 {
 305         NodePtr<Expression> expression;
 306
 307         virtual Return *clone() const { return new Return(*this); }
 308         virtual void visit(NodeVisitor &);
 309 };
 310
 311 struct Jump: Node
 312 {
 313         std::string keyword;
 314
 315         virtual Jump *clone() const { return new Jump(*this); }
 316         virtual void visit(NodeVisitor &);
 317 };
 318
 319 struct NodeVisitor
 320 {
 321         virtual ~NodeVisitor() { }
 322
 323         virtual void visit(Block &) { }
 324         virtual void visit(Literal &) { }
 325         virtual void visit(ParenthesizedExpression &) { }
 326         virtual void visit(VariableReference &) { }
 327         virtual void visit(MemberAccess &) { }
 328         virtual void visit(UnaryExpression &) { }
 329         virtual void visit(BinaryExpression &) { }
 330         virtual void visit(Assignment &);
 331         virtual void visit(FunctionCall &) { }
 332         virtual void visit(ExpressionStatement &) { }
 333         virtual void visit(Import &) { }
 334         virtual void visit(Precision &) { }
 335         virtual void visit(Layout &) { }
 336         virtual void visit(InterfaceLayout &) { }
 337         virtual void visit(StructDeclaration &) { }
 338         virtual void visit(VariableDeclaration &) { }
 339         virtual void visit(InterfaceBlock &) { }
 340         virtual void visit(FunctionDeclaration &) { }
 341         virtual void visit(Conditional &) { }
 342         virtual void visit(Iteration &) { }
 343         virtual void visit(Passthrough &) { }
 344         virtual void visit(Return &) { }
 345         virtual void visit(Jump &) { }
 346 };
 347
 348 struct TraversingVisitor: NodeVisitor
 349 {
 350         using NodeVisitor::visit;
 351         virtual void visit(Block &);
 352         virtual void visit(ParenthesizedExpression &);
 353         virtual void visit(MemberAccess &);
 354         virtual void visit(UnaryExpression &);
 355         virtual void visit(BinaryExpression &);
 356         virtual void visit(FunctionCall &);
 357         virtual void visit(ExpressionStatement &);
 358         virtual void visit(InterfaceLayout &);
 359         virtual void visit(StructDeclaration &);
 360         virtual void visit(VariableDeclaration &);
 361         virtual void visit(InterfaceBlock &);
 362         virtual void visit(FunctionDeclaration &);
 363         virtual void visit(Conditional &);
 364         virtual void visit(Iteration &);
 365         virtual void visit(Passthrough &);
 366         virtual void visit(Return &);
 367 };
 368
 369 enum StageType
 370 {
 371         SHARED,
 372         VERTEX,
 373         GEOMETRY,
 374         FRAGMENT
 375 };
 376
 377 struct Stage
 378 {
 379         StageType type;
 380         Stage *previous;
 381         ProgramSyntax::Block content;
 382         std::map<std::string, VariableDeclaration *> in_variables;
 383         std::map<std::string, VariableDeclaration *> out_variables;
 384         std::map<std::string, unsigned> locations;
 385         Version required_version;
 386
 387         Stage(StageType);
 388 };
 389
 390 struct Module
 391 {
 392         Stage shared;
 393         std::list<Stage> stages;
 394
 395         Module();
 396 };
 397
 398 } // namespace ProgramSyntax
 399 } // namespace GL
 400 } // namespace Msp
 401
 402 #endif