+/** Assigns offset layout qualifiers to struct members. */
+class StructOrganizer: private TraversingVisitor
+{
+private:
+ int offset = -1;
+
+public:
+ 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 = -1;
+ int desc_set = 0;
+ int bind_point = -1;
+ };
+
+ Features features;
+ bool alloc_new = true;
+ 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 &, bool = true);
+private:
+ void apply(Stage &);
+
+ void allocate_locations(const std::string &);
+ void bind_uniform(RefPtr<Layout> &, const std::string &, unsigned);
+
+ bool visit_uniform(const std::string &, RefPtr<Layout> &);
+ virtual void visit(VariableDeclaration &);
+ virtual void visit(InterfaceBlock &);
+ virtual void visit(FunctionDeclaration &) { }
+};
+