X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fglsl%2Fgenerate.h;h=a4ab75719e29b2416ad36e3be6918ace0beeddd1;hp=63f1b0770ba435b287157eef975514b7e41cab9d;hb=HEAD;hpb=de87bb70ae10de39a39b2415407a234ab28099cf

diff --git a/source/glsl/generate.h b/source/glsl/generate.h
index 63f1b077..aba2c353 100644
--- a/source/glsl/generate.h
+++ b/source/glsl/generate.h
@@ -11,159 +11,96 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
-/** Combines multiple declarations of the same identifier into one. */
-class DeclarationCombiner: private TraversingVisitor
+/** Assigns IDs to specialization constants with an automatic ID. */
+class ConstantIdAssigner: private TraversingVisitor
 {
 private:
-	std::map<std::string, std::vector<FunctionDeclaration *> > functions;
-	std::map<std::string, VariableDeclaration *> variables;
-	std::set<Node *> nodes_to_remove;
+	std::set<unsigned> used_ids;
+	std::map<std::string, unsigned> existing_constants;
+	std::vector<VariableDeclaration *> auto_constants;
 
 public:
-	void apply(Stage &);
+	void apply(Module &, const Features &);
 
 private:
-	virtual void visit(Block &);
 	virtual void visit(VariableDeclaration &);
-	virtual void visit(FunctionDeclaration &) { }
 };
 
-/** Manipulates specialization constants.  If values are specified, turns
-specialization constants into normal constants.  Without values assigns
-automatic constant_ids to specialization constants. */
-class ConstantSpecializer: private TraversingVisitor
-{
-private:
-	const std::map<std::string, int> *values;
-
-public:
-	ConstantSpecializer();
-
-	void apply(Stage &, const std::map<std::string, int> *);
-
-private:
-	virtual void visit(VariableDeclaration &);
-};
+/** Materializes implicitly declared interfaces.
 
-/** Forms links between nested blocks in the syntax tree. */
-class BlockHierarchyResolver: private TraversingVisitor
-{
-public:
-	void apply(Stage &s) { s.content.visit(*this); }
+Out variable declarations inside functions are moved to the global scope.
 
-private:
-	virtual void enter(Block &);
-};
+Passthrough statements are processed, generating out variables to match in
+variables and copying values.
 
-/** Resolves types of variables and base types of other types. */
-class TypeResolver: private TraversingVisitor
+Unresolved variables are looked up in the previous stage's out variables. */
+class InterfaceGenerator: private TraversingVisitor
 {
 private:
-	Stage *stage;
-	std::map<TypeDeclaration *, TypeDeclaration *> alias_map;
-	std::map<TypeDeclaration *, TypeDeclaration *> array_types;
-	NodeList<Statement>::iterator type_insert_point;
+	Stage *stage = 0;
+	std::string in_prefix;
+	std::string out_prefix;
+	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:
-	TypeResolver();
-
 	void apply(Stage &);
 
 private:
-	TypeDeclaration *resolve_type(const std::string &);
+	static std::string get_out_prefix(Stage::Type);
+	std::string change_prefix(const std::string &, const std::string &) const;
 	virtual void visit(Block &);
-	virtual void visit(BasicTypeDeclaration &);
-	virtual void visit(ImageTypeDeclaration &);
-	virtual void visit(StructDeclaration &);
-	virtual void visit(VariableDeclaration &);
-	virtual void visit(FunctionDeclaration &);
-};
-
-/** Resolves variable references.  Variable references which match the name
-of an interface block are turned into interface block references. */
-class VariableResolver: private TraversingVisitor
-{
-private:
-	Stage *stage;
-	std::map<std::string, VariableDeclaration *> *r_members;
-	RefPtr<InterfaceBlockReference> r_iface_ref;
-	std::string block_interface;
-	bool record_target;
-	bool r_self_referencing;
-	VariableDeclaration *r_assignment_target;
-
-public:
-	VariableResolver();
-
-	void apply(Stage &);
-
-private:
-	virtual void enter(Block &);
+	VariableDeclaration *generate_interface(VariableDeclaration &, const std::string &, const std::string &);
+	ExpressionStatement &insert_assignment(const std::string &, Expression *);
 	virtual void visit(VariableReference &);
-	virtual void visit(InterfaceBlockReference &);
-	virtual void visit(MemberAccess &);
-	virtual void visit(UnaryExpression &);
-	virtual void visit(BinaryExpression &);
-	virtual void visit(Assignment &);
-	virtual void visit(FunctionCall &);
 	virtual void visit(VariableDeclaration &);
-	virtual void visit(InterfaceBlock &);
+	virtual void visit(FunctionDeclaration &);
+	virtual void visit(Passthrough &);
 };
 
-/** Resolves function declarations and calls. */
-class FunctionResolver: private TraversingVisitor
+class LayoutDefaulter: private TraversingVisitor
 {
 private:
-	Stage *stage;
-	std::map<std::string, std::vector<FunctionDeclaration *> > declarations;
+	InterfaceLayout *in_iface = 0;
+	bool need_winding = true;
+	bool need_spacing = true;
 
 public:
 	void apply(Stage &);
 
 private:
-	virtual void visit(FunctionCall &);
-	virtual void visit(FunctionDeclaration &);
+	virtual void visit(InterfaceLayout &);
 };
 
-/** Materializes implicitly declared interfaces.
-
-Out variable declarations inside functions are moved to the global scope.
-
-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
+/**
+Assigns sizes to arrays which don't have a size.  Geometry shader inputs are
+sized by topology.  Other arrays are sized by their use with literal indices.
+*/
+class ArraySizer: private TraversingVisitor
 {
 private:
-	Stage *stage;
-	std::string in_prefix;
-	std::string out_prefix;
-	bool function_scope;
-	InterfaceBlock *iface_block;
-	bool copy_block;
-	Block *iface_target_block;
-	NodeList<Statement>::iterator iface_insert_point;
-	NodeList<Statement>::iterator assignment_insert_point;
-	std::set<Node *> nodes_to_remove;
+	std::map<VariableDeclaration *, int> max_indices;
+	unsigned input_size = 0;
+	VariableDeclaration *r_declaration;
 
 public:
-	InterfaceGenerator();
-
 	void apply(Stage &);
 
 private:
-	static std::string get_out_prefix(Stage::Type);
-	std::string change_prefix(const std::string &, const std::string &) const;
-	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(MemberAccess &);
+	virtual void visit(Swizzle &);
+	virtual void visit(UnaryExpression&);
+	virtual void visit(BinaryExpression &);
+	virtual void visit(TernaryExpression &);
+	virtual void visit(FunctionCall &);
+	virtual void visit(InterfaceLayout &);
 	virtual void visit(VariableDeclaration &);
-	virtual void visit(InterfaceBlock &);
-	virtual void visit(FunctionDeclaration &);
-	virtual void visit(Passthrough &);
 };
 
 } // namespace SL