X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Ffinalize.h;h=36b349125f51eb236cff061d9c08c0ad49c2a542;hb=7b03ca83ca6c48b9f8c3f77c13760e7f87e9632d;hp=e0904b2e40eab2272accfd1cf4c8e6d4972ac4fd;hpb=6ff7aea73057e4a650d8b0ac659d9bba05ba02e2;p=libs%2Fgl.git

diff --git a/source/glsl/finalize.h b/source/glsl/finalize.h
index e0904b2e..36b34912 100644
--- a/source/glsl/finalize.h
+++ b/source/glsl/finalize.h
@@ -8,6 +8,26 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
+class LocationAllocator: private TraversingVisitor
+{
+private:
+	std::map<std::string, std::set<unsigned> > used_locations;
+	std::map<std::string, unsigned> uniform_locations;
+	std::vector<VariableDeclaration *> unplaced_variables;
+
+public:
+	void apply(Module &);
+private:
+	void apply(Stage &);
+
+	void allocate_locations(const std::string &);
+	void add_location(RefPtr<Layout> &, unsigned);
+
+	virtual void visit(VariableDeclaration &);
+	virtual void visit(InterfaceBlock &) { }
+	virtual void visit(FunctionDeclaration &) { }
+};
+
 /** Generates default precision declarations or removes precision declarations
 according to the requirements of the target API. */
 class PrecisionConverter: private TraversingVisitor
@@ -58,10 +78,13 @@ private:
 	bool supports_unified_sampling_functions() const;
 	virtual void visit(FunctionCall &);
 	bool supports_interface_layouts() const;
+	bool supports_stage_interface_layouts() const;
 	bool supports_centroid_sampling() const;
 	bool supports_sample_sampling() const;
+	bool supports_uniform_location() const;
 	virtual void visit(VariableDeclaration &);
 	bool supports_interface_blocks(const std::string &) const;
+	bool supports_interface_block_location() const;
 	virtual void visit(InterfaceBlock &);
 };