1*480093f4SDimitry Andric //===-- NonRelocatableStringpool.cpp --------------------------------------===//
2*480093f4SDimitry Andric //
3*480093f4SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*480093f4SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*480093f4SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*480093f4SDimitry Andric //
7*480093f4SDimitry Andric //===----------------------------------------------------------------------===//
8*480093f4SDimitry Andric 
9*480093f4SDimitry Andric #include "llvm/CodeGen/NonRelocatableStringpool.h"
10*480093f4SDimitry Andric 
11*480093f4SDimitry Andric namespace llvm {
12*480093f4SDimitry Andric 
getEntry(StringRef S)13*480093f4SDimitry Andric DwarfStringPoolEntryRef NonRelocatableStringpool::getEntry(StringRef S) {
14*480093f4SDimitry Andric   if (S.empty() && !Strings.empty())
15*480093f4SDimitry Andric     return EmptyString;
16*480093f4SDimitry Andric 
17*480093f4SDimitry Andric   if (Translator)
18*480093f4SDimitry Andric     S = Translator(S);
19*480093f4SDimitry Andric   auto I = Strings.insert({S, DwarfStringPoolEntry()});
20*480093f4SDimitry Andric   auto &Entry = I.first->second;
21*480093f4SDimitry Andric   if (I.second || !Entry.isIndexed()) {
22*480093f4SDimitry Andric     Entry.Index = NumEntries++;
23*480093f4SDimitry Andric     Entry.Offset = CurrentEndOffset;
24*480093f4SDimitry Andric     Entry.Symbol = nullptr;
25*480093f4SDimitry Andric     CurrentEndOffset += S.size() + 1;
26*480093f4SDimitry Andric   }
27*480093f4SDimitry Andric   return DwarfStringPoolEntryRef(*I.first, true);
28*480093f4SDimitry Andric }
29*480093f4SDimitry Andric 
internString(StringRef S)30*480093f4SDimitry Andric StringRef NonRelocatableStringpool::internString(StringRef S) {
31*480093f4SDimitry Andric   DwarfStringPoolEntry Entry{nullptr, 0, DwarfStringPoolEntry::NotIndexed};
32*480093f4SDimitry Andric 
33*480093f4SDimitry Andric   if (Translator)
34*480093f4SDimitry Andric     S = Translator(S);
35*480093f4SDimitry Andric 
36*480093f4SDimitry Andric   auto InsertResult = Strings.insert({S, Entry});
37*480093f4SDimitry Andric   return InsertResult.first->getKey();
38*480093f4SDimitry Andric }
39*480093f4SDimitry Andric 
40*480093f4SDimitry Andric std::vector<DwarfStringPoolEntryRef>
getEntriesForEmission() const41*480093f4SDimitry Andric NonRelocatableStringpool::getEntriesForEmission() const {
42*480093f4SDimitry Andric   std::vector<DwarfStringPoolEntryRef> Result;
43*480093f4SDimitry Andric   Result.reserve(Strings.size());
44*480093f4SDimitry Andric   for (const auto &E : Strings)
45*480093f4SDimitry Andric     if (E.getValue().isIndexed())
46*480093f4SDimitry Andric       Result.emplace_back(E, true);
47*480093f4SDimitry Andric   llvm::sort(Result, [](const DwarfStringPoolEntryRef A,
48*480093f4SDimitry Andric                         const DwarfStringPoolEntryRef B) {
49*480093f4SDimitry Andric     return A.getIndex() < B.getIndex();
50*480093f4SDimitry Andric   });
51*480093f4SDimitry Andric   return Result;
52*480093f4SDimitry Andric }
53*480093f4SDimitry Andric 
54*480093f4SDimitry Andric } // namespace llvm
55