llvm/lib/Support/StringSaver.cpp

   1 //===-- StringSaver.cpp ---------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm/Support/StringSaver.h"
  10
  11 #include "llvm/ADT/SmallString.h"
  12
  13 using namespace llvm;
  14
  15 StringRef StringSaver::save(StringRef S) {
  16   char *P = Alloc.Allocate<char>(S.size() + 1);
  17   if (!S.empty())
  18     memcpy(P, S.data(), S.size());
  19   P[S.size()] = '\0';
  20   return StringRef(P, S.size());
  21 }
  22
  23 StringRef StringSaver::save(const Twine &S) {
  24   SmallString<128> Storage;
  25   return save(S.toStringRef(Storage));
  26 }
  27
  28 StringRef UniqueStringSaver::save(StringRef S) {
  29   auto R = Unique.insert(S);
  30   if (R.second)                 // cache miss, need to actually save the string
  31     *R.first = Strings.save(S); // safe replacement with equal value
  32   return *R.first;
  33 }
  34
  35 StringRef UniqueStringSaver::save(const Twine &S) {
  36   SmallString<128> Storage;
  37   return save(S.toStringRef(Storage));
  38 }