39 lines · 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-exception6//7//===----------------------------------------------------------------------===//8 9#include "llvm/Support/StringSaver.h"10 11#include "llvm/ADT/SmallString.h"12 13using namespace llvm;14 15StringRef 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 23StringRef StringSaver::save(const Twine &S) {24 SmallString<128> Storage;25 return save(S.toStringRef(Storage));26}27 28StringRef UniqueStringSaver::save(StringRef S) {29 auto R = Unique.insert(S);30 if (R.second) // cache miss, need to actually save the string31 *R.first = Strings.save(S); // safe replacement with equal value32 return *R.first;33}34 35StringRef UniqueStringSaver::save(const Twine &S) {36 SmallString<128> Storage;37 return save(S.toStringRef(Storage));38}39