[llvm] r256781 - [PGO]: Use efficient 'join' API for uncompressed string

[Xinliang David Li via llvm-commits]

Author: davidxl
Date: Mon Jan  4 15:31:09 2016
New Revision: 256781

URL: http://llvm.org/viewvc/llvm-project?rev=256781&view=rev
Log:
[PGO]: Use efficient 'join' API for uncompressed string

Modified:
    llvm/trunk/lib/ProfileData/InstrProf.cpp

Modified: llvm/trunk/lib/ProfileData/InstrProf.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/ProfileData/InstrProf.cpp?rev=256781&r1=256780&r2=256781&view=diff
==============================================================================
--- llvm/trunk/lib/ProfileData/InstrProf.cpp (original)
+++ llvm/trunk/lib/ProfileData/InstrProf.cpp Mon Jan  4 15:31:09 2016
@@ -13,6 +13,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/ProfileData/InstrProf.h"
+#include "llvm/ADT/StringExtras.h"
 #include "llvm/IR/Constants.h"
 #include "llvm/IR/Function.h"
 #include "llvm/IR/GlobalVariable.h"
@@ -167,18 +168,9 @@ GlobalVariable *createPGOFuncNameVar(Fun
 int collectPGOFuncNameStrings(const std::vector<std::string> &NameStrs,
                               bool doCompression, std::string &Result) {
   uint8_t Header[16], *P = Header;
-  std::string UncompressedNameStrings;
-  size_t UncompressedStringLen = 0;
+  std::string UncompressedNameStrings =
+      join(NameStrs.begin(), NameStrs.end(), StringRef(" "));
 
-  for (auto NameStr : NameStrs)
-    UncompressedStringLen += (NameStr.length() + 1);
-
-  UncompressedNameStrings.reserve(UncompressedStringLen + 1);
-
-  for (auto NameStr : NameStrs) {
-    UncompressedNameStrings += NameStr;
-    UncompressedNameStrings.append(" ");
-  }
   unsigned EncLen = encodeULEB128(UncompressedNameStrings.length(), P);
   P += EncLen;
   if (!doCompression) {
@@ -252,8 +244,8 @@ int readPGOFuncNameStrings(StringRef Nam
     do {
       size_t NameStop = NameStrings.find(' ', NameStart);
       if (NameStop == StringRef::npos)
-        return 1;
-      if (NameStop == NameStrings.size() - 1)
+        NameStop = NameStrings.size();
+      if (NameStop >= NameStrings.size() - 1)
         isLast = true;
       StringRef Name = NameStrings.substr(NameStart, NameStop - NameStart);
       Symtab.addFuncName(Name);