Add visitors to calculate offsets of struct members

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

diff --git a/source/glsl/finalize.h b/source/glsl/finalize.h
index 0638026dba207f2e7d48a209d2a2f3d6ea49c11c..fd29737a713b3ed331e9ba4b0bddb2e13b9f72a6 100644 (file)
--- a/source/glsl/finalize.h
+++ b/source/glsl/finalize.h
@@ -8,6 +8,22 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
+/** Assigns offset layout qualifiers to struct members. */
+class StructOrganizer: private TraversingVisitor
+{
+private:
+       int offset;
+
+public:
+       StructOrganizer();
+
+       void apply(Stage &s) { s.content.visit(*this); }
+
+private:
+       virtual void visit(StructDeclaration &);
+       virtual void visit(VariableDeclaration &);
+};
+
 /** Assigns location and binding layout qualifiers to interface variables and
 blocks. */
 class LocationAllocator: private TraversingVisitor