Steve, pointer out that getName() is available for selectors. Made code much smaller...

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 c2a93a6f0f1068cfbee0cbac381a68532fb7dd1e..604d473479040db5155ad206fcf98f1948ba3856 100644 (file)
--- a/AST/StmtDumper.cpp
+++ b/AST/StmtDumper.cpp
@@ -412,15 +412,7 @@ void StmtDumper::VisitObjCSelectorExpr(ObjCSelectorExpr *Node) {
   
   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 2db46f4e0cb2a695a92afc69211e3f98e593ca17..a19ad37a0c3101cfbb059f090334e1f8eeb8d662 100644 (file)
--- a/AST/StmtPrinter.cpp
+++ b/AST/StmtPrinter.cpp
@@ -623,15 +623,7 @@ void StmtPrinter::VisitObjCEncodeExpr(ObjCEncodeExpr *Node) {
 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 << ")";
 }