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 }
39