X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Freflect.h;h=9520feb6a924c797ee50b7e8a062dbd7b1f74795;hb=c701c8787cb19fbb6dc5b0bfae1a94e2b07dd549;hp=c03a5ccbd8b57044db6b28355863bac292c47a75;hpb=8f5f54a9e165dae424e5b0bb8e488c3d01849bf6;p=libs%2Fgl.git

diff --git a/source/glsl/reflect.h b/source/glsl/reflect.h
index c03a5ccb..9520feb6 100644
--- a/source/glsl/reflect.h
+++ b/source/glsl/reflect.h
@@ -7,6 +7,33 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
+/** Compares two types for equality.  Struct types are compared recursively. */
+class TypeComparer: private NodeVisitor
+{
+private:
+	Node *first;
+	Node *second;
+	unsigned first_tag;
+	bool r_result;
+
+	static unsigned next_tag;
+
+public:
+	TypeComparer();
+
+	bool apply(TypeDeclaration &t1, TypeDeclaration &t2) { compare(t1, t2); return r_result; }
+
+private:
+	void compare(Node &, Node &);
+	template<typename T>
+	T *multi_visit(T &);
+	virtual void visit(BasicTypeDeclaration &);
+	virtual void visit(ImageTypeDeclaration &);
+	virtual void visit(StructDeclaration &);
+	virtual void visit(VariableDeclaration &);
+};
+
+/** Determines the number of interface locations required by a variable. */
 class LocationCounter: private NodeVisitor
 {
 private:
@@ -24,6 +51,49 @@ private:
 	virtual void visit(VariableDeclaration &);
 };
 
+/** Determines the size and alignment of a variable, in bytes. */
+class MemoryRequirementsCalculator: private NodeVisitor
+{
+public:
+	struct Result
+	{
+		unsigned size;
+		unsigned alignment;
+
+		Result(unsigned s, unsigned a): size(s), alignment(a) { }
+	};
+private:
+	unsigned r_size;
+	unsigned r_alignment;
+	int r_offset;
+
+public:
+	Result apply(VariableDeclaration &v) { v.visit(*this); return Result(r_size, r_alignment); }
+
+private:
+	virtual void visit(BasicTypeDeclaration &);
+	virtual void visit(StructDeclaration &);
+	virtual void visit(VariableDeclaration &);
+};
+
+/** Collects dependencies of a function.  This includes global variables,
+interface blocks, other functions and types. */
+class DependencyCollector: private TraversingVisitor
+{
+private:
+	std::set<Node *> dependencies;
+	std::set<Node *> locals;
+
+public:
+	std::set<Node *> apply(FunctionDeclaration &);
+
+private:
+	virtual void visit(VariableReference &);
+	virtual void visit(InterfaceBlockReference &);
+	virtual void visit(FunctionCall &);
+	virtual void visit(VariableDeclaration &);
+};
+
 } // namespace SL
 } // namespace GL
 } // namespace Msp