Rewrite type comparisons as a visitor

[libs/gl.git] / source / glsl / reflect.h

diff --git a/source/glsl/reflect.h b/source/glsl/reflect.h
index 72a50c3eb6368b303ba5bd6e2fad8f7095e0189b..9520feb6a924c797ee50b7e8a062dbd7b1f74795 100644 (file)
--- a/source/glsl/reflect.h
+++ b/source/glsl/reflect.h
@@ -7,6 +7,32 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
+/** Compares two types for equality.  Struct types are compared recursively. */
+class TypeComparer: private NodeVisitor
+{
+private:
+       Node *first;
+       Node *second;
+       unsigned first_tag;
+       bool r_result;
+
+       static unsigned next_tag;
+
+public:
+       TypeComparer();
+
+       bool apply(TypeDeclaration &t1, TypeDeclaration &t2) { compare(t1, t2); return r_result; }
+
+private:
+       void compare(Node &, Node &);
+       template<typename T>
+       T *multi_visit(T &);
+       virtual void visit(BasicTypeDeclaration &);
+       virtual void visit(ImageTypeDeclaration &);
+       virtual void visit(StructDeclaration &);
+       virtual void visit(VariableDeclaration &);
+};
+
 /** Determines the number of interface locations required by a variable. */
 class LocationCounter: private NodeVisitor
 {
@@ -25,6 +51,31 @@ private:
        virtual void visit(VariableDeclaration &);
 };
 
+/** Determines the size and alignment of a variable, in bytes. */
+class MemoryRequirementsCalculator: private NodeVisitor
+{
+public:
+       struct Result
+       {
+               unsigned size;
+               unsigned alignment;
+
+               Result(unsigned s, unsigned a): size(s), alignment(a) { }
+       };
+private:
+       unsigned r_size;
+       unsigned r_alignment;
+       int r_offset;
+
+public:
+       Result apply(VariableDeclaration &v) { v.visit(*this); return Result(r_size, r_alignment); }
+
+private:
+       virtual void visit(BasicTypeDeclaration &);
+       virtual void visit(StructDeclaration &);
+       virtual void visit(VariableDeclaration &);
+};
+
 /** Collects dependencies of a function.  This includes global variables,
 interface blocks, other functions and types. */
 class DependencyCollector: private TraversingVisitor