Steve, pointer out that getName() is available for selectors. Made code much smaller, thanks.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@43042 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/AST/StmtDumper.cpp b/AST/StmtDumper.cpp
index c2a93a6..604d473 100644
--- a/AST/StmtDumper.cpp
+++ b/AST/StmtDumper.cpp
@@ -412,15 +412,7 @@
fprintf(F, " ");
Selector &selector = Node->getSelector();
- if (selector.isUnarySelector())
- fprintf(F, "%s", selector.getIdentifierInfoForSlot(0)->getName());
- else {
- for (unsigned i = 0, e = Node->getNumArgs(); i != e; ++i)
- if (selector.getIdentifierInfoForSlot(i))
- fprintf(F, "%s:", selector.getIdentifierInfoForSlot(i)->getName());
- else
- fprintf(F, ":");
- }
+ fprintf(F, "%s", selector.getName().c_str());
}
//===----------------------------------------------------------------------===//
diff --git a/AST/StmtPrinter.cpp b/AST/StmtPrinter.cpp
index 2db46f4..a19ad37 100644
--- a/AST/StmtPrinter.cpp
+++ b/AST/StmtPrinter.cpp
@@ -623,15 +623,7 @@
void StmtPrinter::VisitObjCSelectorExpr(ObjCSelectorExpr *Node) {
OS << "@selector(";
Selector &selector = Node->getSelector();
- if (selector.isUnarySelector())
- OS << " " << selector.getIdentifierInfoForSlot(0)->getName();
- else {
- for (unsigned i = 0, e = Node->getNumArgs(); i != e; ++i)
- if (selector.getIdentifierInfoForSlot(i))
- OS << selector.getIdentifierInfoForSlot(i)->getName() << ":";
- else
- OS << ":";
- }
+ OS << " " << selector.getName();
OS << ")";
}