+TypeResolver::TypeResolver():
+ stage(0)
+{ }
+
+void TypeResolver::apply(Stage &s)
+{
+ stage = &s;
+ s.types.clear();
+ s.content.visit(*this);
+}
+
+void TypeResolver::visit(BasicTypeDeclaration &type)
+{
+ map<string, TypeDeclaration *>::iterator i = stage->types.find(type.base);
+ type.base_type = (i!=stage->types.end() ? i->second : 0);
+
+ if(type.kind==BasicTypeDeclaration::VECTOR && type.base_type)
+ if(BasicTypeDeclaration *basic_base = dynamic_cast<BasicTypeDeclaration *>(type.base_type))
+ if(basic_base->kind==BasicTypeDeclaration::VECTOR)
+ {
+ type.kind = BasicTypeDeclaration::MATRIX;
+ type.size |= basic_base->size<<16;
+ }
+
+ stage->types.insert(make_pair(type.name, &type));
+}
+
+void TypeResolver::visit(ImageTypeDeclaration &type)
+{
+ map<string, TypeDeclaration *>::iterator i = stage->types.find(type.base);
+ type.base_type = (i!=stage->types.end() ? i->second : 0);
+
+ stage->types.insert(make_pair(type.name, &type));
+}
+
+void TypeResolver::visit(StructDeclaration &strct)
+{
+ stage->types.insert(make_pair(strct.name, &strct));
+ TraversingVisitor::visit(strct);
+}
+
+void TypeResolver::visit(VariableDeclaration &var)
+{
+ map<string, TypeDeclaration *>::iterator i = stage->types.find(var.type);
+ if(i!=stage->types.end())
+ var.type_declaration = i->second;
+}
+
+