+/** 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
+{
+private:
+ struct Uniform
+ {
+ int location;
+ int desc_set;
+ int bind_point;
+
+ Uniform(): location(-1), desc_set(-1), bind_point(-1) { }
+ };
+
+ std::map<std::string, std::set<unsigned> > used_locations;
+ std::map<std::string, Uniform> uniforms;
+ std::map<unsigned, std::set<unsigned> > used_bindings;
+ std::vector<VariableDeclaration *> unplaced_variables;
+ std::vector<VariableDeclaration *> unbound_textures;
+ std::vector<InterfaceBlock *> unbound_blocks;
+
+public:
+ void apply(Module &, const Features &);
+private:
+ void apply(Stage &);
+
+ void allocate_locations(const std::string &);
+ void bind_uniform(RefPtr<Layout> &, const std::string &, unsigned);
+ void add_layout_value(RefPtr<Layout> &, const std::string &, unsigned);
+
+ virtual void visit(VariableDeclaration &);
+ virtual void visit(InterfaceBlock &);
+ virtual void visit(FunctionDeclaration &) { }
+};
+