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

diff --git a/source/glsl/generate.h b/source/glsl/generate.h
index d828daa5..aba2c353 100644
--- a/source/glsl/generate.h
+++ b/source/glsl/generate.h
@@ -56,15 +56,53 @@ private:
 	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(VariableDeclaration &);
-	virtual void visit(InterfaceBlock &);
 	virtual void visit(FunctionDeclaration &);
 	virtual void visit(Passthrough &);
 };
 
+class LayoutDefaulter: private TraversingVisitor
+{
+private:
+	InterfaceLayout *in_iface = 0;
+	bool need_winding = true;
+	bool need_spacing = true;
+
+public:
+	void apply(Stage &);
+
+private:
+	virtual void visit(InterfaceLayout &);
+};
+
+/**
+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:
+	std::map<VariableDeclaration *, int> max_indices;
+	unsigned input_size = 0;
+	VariableDeclaration *r_declaration;
+
+public:
+	void apply(Stage &);
+
+private:
+	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 &);
+};
+
 } // namespace SL
 } // namespace GL
 } // namespace Msp