X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fglsl%2Fgenerate.h;h=d828daa5172716cb6c173e129cd5c8d6126fb452;hp=a09c552f1fd9f05c5329edabae112ece237017b0;hb=76cc18518fc8b0b4fa11fda153e7d9b3899ed112;hpb=7a16308e72aef363727b21348779673edf8e5c07

diff --git a/source/glsl/generate.h b/source/glsl/generate.h
index a09c552f..d828daa5 100644
--- a/source/glsl/generate.h
+++ b/source/glsl/generate.h
@@ -11,110 +11,58 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
-class DeclarationCombiner: public BlockModifier
+/** Assigns IDs to specialization constants with an automatic ID. */
+class ConstantIdAssigner: private TraversingVisitor
 {
 private:
-	bool toplevel;
-	std::map<std::string, std::vector<FunctionDeclaration *> > functions;
-	std::map<std::string, VariableDeclaration *> variables;
+	std::set<unsigned> used_ids;
+	std::map<std::string, unsigned> existing_constants;
+	std::vector<VariableDeclaration *> auto_constants;
 
 public:
-	DeclarationCombiner();
+	void apply(Module &, const Features &);
 
-	using StageVisitor::visit;
-	virtual void visit(Block &);
-	virtual void visit(FunctionDeclaration &);
-	virtual void visit(VariableDeclaration &);
-};
-
-class VariableResolver: public StageVisitor
-{
 private:
-	std::vector<Block *> blocks;
-	StructDeclaration *type;
-	bool anonymous;
-	std::string block_interface;
-	bool record_target;
-	VariableDeclaration *assignment_target;
-	bool self_referencing;
-
-public:
-	VariableResolver();
-
-	virtual void apply(Stage &);
-	using StageVisitor::visit;
-	virtual void visit(Block &);
-	virtual void visit(VariableReference &);
-	virtual void visit(MemberAccess &);
-	virtual void visit(BinaryExpression &);
-	virtual void visit(Assignment &);
-	virtual void visit(StructDeclaration &);
 	virtual void visit(VariableDeclaration &);
-	virtual void visit(InterfaceBlock &);
 };
 
-class FunctionResolver: public StageVisitor
-{
-private:
-	std::map<std::string, std::vector<FunctionDeclaration *> > functions;
+/** Materializes implicitly declared interfaces.
 
-public:
-	using StageVisitor::visit;
-	virtual void visit(FunctionCall &);
-	virtual void visit(FunctionDeclaration &);
-};
+Out variable declarations inside functions are moved to the global scope.
 
-class InterfaceGenerator: public BlockModifier
+Passthrough statements are processed, generating out variables to match in
+variables and copying values.
+
+Unresolved variables are looked up in the previous stage's out variables. */
+class InterfaceGenerator: private TraversingVisitor
 {
 private:
+	Stage *stage = 0;
 	std::string in_prefix;
 	std::string out_prefix;
-	unsigned scope_level;
-	std::map<std::string, RefPtr<VariableDeclaration> > iface_declarations;
+	bool function_scope = false;
+	bool copy_block = false;
+	std::vector<VariableDeclaration *> declared_inputs;
+	Block *iface_target_block = 0;
+	NodeList<Statement>::iterator iface_insert_point;
+	NodeList<Statement>::iterator assignment_insert_point;
+	std::set<Node *> nodes_to_remove;
 
 public:
-	InterfaceGenerator();
+	void apply(Stage &);
 
+private:
 	static std::string get_out_prefix(Stage::Type);
-	virtual void apply(Stage &);
-	using StageVisitor::visit;
-	virtual void visit(Block &);
 	std::string change_prefix(const std::string &, const std::string &) const;
-	bool generate_interface(VariableDeclaration &, const std::string &, const std::string &);
+	virtual void visit(Block &);
+	VariableDeclaration *generate_interface(VariableDeclaration &, const std::string &, const std::string &);
+	InterfaceBlock *generate_interface(InterfaceBlock &);
 	ExpressionStatement &insert_assignment(const std::string &, Expression *);
 	virtual void visit(VariableReference &);
 	virtual void visit(VariableDeclaration &);
-	virtual void visit(Passthrough &);
-};
-
-class DeclarationReorderer: public StageVisitor
-{
-private:
-	enum DeclarationKind
-	{
-		NO_DECLARATION,
-		LAYOUT,
-		STRUCT,
-		VARIABLE,
-		FUNCTION
-	};
-
-	unsigned scope_level;
-	DeclarationKind kind;
-	std::set<Node *> ordered_funcs;
-	std::set<Node *> needed_funcs;
-
-public:
-	DeclarationReorderer();
-
-	using StageVisitor::visit;
-	virtual void visit(Block &);
-	virtual void visit(FunctionCall &);
-	virtual void visit(InterfaceLayout &) { kind = LAYOUT; }
-	virtual void visit(StructDeclaration &) { kind = STRUCT; }
-	virtual void visit(VariableDeclaration &);
-	virtual void visit(InterfaceBlock &) { kind = VARIABLE; }
+	virtual void visit(InterfaceBlock &);
 	virtual void visit(FunctionDeclaration &);
+	virtual void visit(Passthrough &);
 };
 
 } // namespace SL