void VisitBlockDecl(BlockDecl *D);
void VisitDeclContext(DeclContext *D);
- void VisitEnumDecl(EnumDecl *D);
void VisitFieldDecl(FieldDecl *D);
void VisitFunctionDecl(FunctionDecl *D);
void VisitNamedDecl(NamedDecl *D);
void VisitObjCContainerDecl(ObjCContainerDecl *CD);
void VisitObjCMethodDecl(ObjCMethodDecl *MD);
void VisitObjCPropertyDecl(ObjCPropertyDecl *D);
- void VisitRecordDecl(RecordDecl *D);
- void VisitTagDeclCommon(TagDecl *D);
+ void VisitTagDecl(TagDecl *D);
void VisitTypedefDecl(TypedefDecl *D);
};
} // end anonymous namespace
Visit(D);
}
-void USRGenerator::VisitEnumDecl(EnumDecl *D) {
- VisitDeclContext(D->getDeclContext());
- Out << "@E^";
- VisitTagDeclCommon(D);
-}
-
void USRGenerator::VisitFieldDecl(FieldDecl *D) {
const std::string &s = D->getNameAsString();
if (s.empty()) {
Out << "@N^" << D->getNameAsString();
}
-void USRGenerator::VisitRecordDecl(RecordDecl *D) {
- VisitDeclContext(D->getDeclContext());
- Out << "@S^";
- VisitTagDeclCommon(D);
-}
-
void USRGenerator::VisitObjCMethodDecl(ObjCMethodDecl *D) {
Visit(cast<Decl>(D->getDeclContext()));
Out << (D->isInstanceMethod() ? "(im)" : "(cm)");
Out << "(py)" << D->getName();
}
-void USRGenerator::VisitTagDeclCommon(TagDecl *D) {
+void USRGenerator::VisitTagDecl(TagDecl *D) {
+ VisitDeclContext(D->getDeclContext());
+ switch (D->getTagKind()) {
+ case TagDecl::TK_struct: Out << "@S^"; break;
+ case TagDecl::TK_class: Out << "@C^"; break;
+ case TagDecl::TK_union: Out << "@U^"; break;
+ case TagDecl::TK_enum: Out << "@E^"; break;
+ }
+
// FIXME: Better support for anonymous structures and enums.
const std::string &s = D->getNameAsString();
if (s.empty()) {