namespace GL {
namespace SL {
-class LocationCounter: private NodeVisitor
+bool is_scalar(const BasicTypeDeclaration &);
+bool is_vector_or_matrix(const BasicTypeDeclaration &);
+BasicTypeDeclaration *get_element_type(BasicTypeDeclaration &);
+bool can_convert(const BasicTypeDeclaration &, const BasicTypeDeclaration &);
+
+/** Compares two types for equality. Struct types are compared recursively. */
+class TypeComparer: private NodeVisitor
{
private:
- unsigned r_count;
+ Node *first = 0;
+ Node *second = 0;
+ unsigned first_tag = 0;
+ bool r_result = false;
+
+ static unsigned next_tag;
public:
- LocationCounter();
+ 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(Literal &);
+ virtual void visit(VariableReference &);
+ virtual void visit(UnaryExpression &);
+ virtual void visit(BinaryExpression &);
+ virtual void visit(TernaryExpression &);
+ 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:
+ unsigned r_count = 0;
+
+public:
unsigned apply(VariableDeclaration &v) { v.visit(*this); return r_count; }
private:
virtual void visit(VariableDeclaration &);
};
+/** Determines the size and alignment of a variable or a type, in bytes. */
+class MemoryRequirementsCalculator: private NodeVisitor
+{
+public:
+ struct Result
+ {
+ unsigned size;
+ unsigned alignment;
+ unsigned stride;
+
+ Result(unsigned s, unsigned a): size(s), alignment(a), stride(s+a-1-(s+a-1)%a) { }
+ };
+private:
+ unsigned r_size = 0;
+ unsigned r_alignment = 1;
+ int r_offset = -1;
+
+public:
+ Result apply(VariableDeclaration &v) { v.visit(*this); return Result(r_size, r_alignment); }
+ Result apply(TypeDeclaration &t) { t.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;
+ std::set<FunctionDeclaration *> visited_functions;
+
+public:
+ std::set<Node *> apply(FunctionDeclaration &);
+
+private:
+ virtual void visit(VariableReference &);
+ virtual void visit(InterfaceBlockReference &);
+ virtual void visit(FunctionCall &);
+ virtual void visit(VariableDeclaration &);
+ virtual void visit(FunctionDeclaration &);
+};
+
+class AssignmentCollector: private TraversingVisitor
+{
+private:
+ std::set<VariableDeclaration *> assigned_variables;
+
+public:
+ std::set<VariableDeclaration *> apply(Node &);
+
+private:
+ virtual void visit(Assignment &);
+};
+
} // namespace SL
} // namespace GL
} // namespace Msp