X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Ffinalize.h;h=1743074849831066a4ff038b093c67b75496395a;hb=3fe1aab63922eec99d8bf6fd4fd60bec10df173c;hp=212007131e80a3d0e37173364005d68f3b8d1166;hpb=c8de66203565f07f10898a74dbd55072432790fc;p=libs%2Fgl.git

diff --git a/source/glsl/finalize.h b/source/glsl/finalize.h
index 21200713..17430748 100644
--- a/source/glsl/finalize.h
+++ b/source/glsl/finalize.h
@@ -1,5 +1,5 @@
-#ifndef MSP_GL_SL_COMPATIBILITY_H_
-#define MSP_GL_SL_COMPATIBILITY_H_
+#ifndef MSP_GL_SL_FINALIZE_H_
+#define MSP_GL_SL_FINALIZE_H_
 
 #include <string>
 #include "visitor.h"
@@ -8,75 +8,143 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
-/** Generates default precision declarations if they are missing, to satisfy
-GLSL ES requirements. */
-class DefaultPrecisionGenerator: private TraversingVisitor
+/** Assigns offset layout qualifiers to struct members. */
+class StructOrganizer: private TraversingVisitor
 {
 private:
-	Stage *stage;
-	std::set<std::string> have_default;
-	NodeList<Statement>::iterator insert_point;
+	int offset = -1;
 
 public:
-	DefaultPrecisionGenerator();
+	void apply(Stage &s) { s.content.visit(*this); }
 
-	void apply(Stage &);
+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:
-	virtual void visit(Block &);
-	virtual void visit(Precision &);
+	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);
+
+	bool visit_uniform(const std::string &, RefPtr<Layout> &);
 	virtual void visit(VariableDeclaration &);
+	virtual void visit(InterfaceBlock &);
+	virtual void visit(FunctionDeclaration &) { }
 };
 
-/** Removes precision qualifiers from variable declarations, as well as
-default precision declarations. */
-class PrecisionRemover: private TraversingVisitor
+/** Generates default precision declarations or removes precision declarations
+according to the requirements of the target API. */
+class PrecisionConverter: private TraversingVisitor
 {
 private:
+	Stage *stage = 0;
+	std::set<std::string> have_default;
+	NodeList<Statement>::iterator insert_point;
 	std::set<Node *> nodes_to_remove;
 
 public:
 	void apply(Stage &);
 
 private:
+	virtual void visit(Block &);
 	virtual void visit(Precision &);
 	virtual void visit(VariableDeclaration &);
 };
 
+/** Base class for feature converters. */
+class FeatureConverter: protected TraversingVisitor
+{
+protected:
+	Stage *stage = 0;
+	Features features;
+
+	FeatureConverter() = default;
+
+public:
+	void apply(Stage &, const Features &);
+protected:
+	virtual void apply() = 0;
+
+	void unsupported(const std::string &);
+
+	bool check_version(const Version &) const;
+	bool check_extension(bool Features::*) const;
+};
+
 /** Converts structures of the syntax tree to match a particular set of
 features. */
-class LegacyConverter: private TraversingVisitor
+class StructuralFeatureConverter: public FeatureConverter
 {
 private:
-	Stage *stage;
-	Features features;
-	std::string r_type;
-	VariableDeclaration *frag_out;
+	VariableDeclaration *frag_out = 0;
 	NodeList<Statement>::iterator uniform_insert_point;
 	std::set<Node *> nodes_to_remove;
+	RefPtr<Expression> r_replaced_reference;
+	bool r_flattened_interface = false;
 
 public:
-	LegacyConverter();
-
-	virtual void apply(Stage &, const Features &);
-
+	void apply(Stage &s, const Features &f) { FeatureConverter::apply(s, f); }
 private:
-	void unsupported(const std::string &);
+	virtual void apply();
 
 	virtual void visit(Block &);
-	bool check_version(const Version &) const;
-	bool check_extension(bool Features::*) const;
+	virtual void visit(RefPtr<Expression> &);
 	bool supports_stage(Stage::Type) const;
 	bool supports_unified_interface_syntax() const;
 	virtual void visit(VariableReference &);
+	virtual void visit(InterfaceBlockReference &);
+	virtual void visit(MemberAccess &);
 	virtual void visit(Assignment &);
 	bool supports_unified_sampling_functions() const;
 	virtual void visit(FunctionCall &);
+	virtual void visit(VariableDeclaration &);
+	bool supports_interface_blocks(const std::string &) const;
+	virtual void visit(InterfaceBlock &);
+};
+
+/** Converts qualifiers on variables and blocksto match a particular set of
+features. */
+class QualifierConverter: private FeatureConverter
+{
+public:
+	void apply(Stage &s, const Features &f) { FeatureConverter::apply(s, f); }
+private:
+	virtual void apply();
+
 	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;
+	bool supports_binding() const;
 	virtual void visit(VariableDeclaration &);
-	bool supports_interface_blocks(const std::string &) const;
+	bool supports_interface_block_location() const;
 	virtual void visit(InterfaceBlock &);
 };