1480093f4SDimitry Andric //===-- NonRelocatableStringpool.cpp --------------------------------------===// 2480093f4SDimitry Andric // 3480093f4SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4480093f4SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5480093f4SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6480093f4SDimitry Andric // 7480093f4SDimitry Andric //===----------------------------------------------------------------------===// 8480093f4SDimitry Andric 9480093f4SDimitry Andric #include "llvm/CodeGen/NonRelocatableStringpool.h" 1004eeddc0SDimitry Andric #include "llvm/ADT/STLExtras.h" 11480093f4SDimitry Andric 12480093f4SDimitry Andric namespace llvm { 13480093f4SDimitry Andric 14480093f4SDimitry Andric DwarfStringPoolEntryRef NonRelocatableStringpool::getEntry(StringRef S) { 15480093f4SDimitry Andric if (S.empty() && !Strings.empty()) 16480093f4SDimitry Andric return EmptyString; 17480093f4SDimitry Andric 18480093f4SDimitry Andric if (Translator) 19480093f4SDimitry Andric S = Translator(S); 20480093f4SDimitry Andric auto I = Strings.insert({S, DwarfStringPoolEntry()}); 21480093f4SDimitry Andric auto &Entry = I.first->second; 22480093f4SDimitry Andric if (I.second || !Entry.isIndexed()) { 23480093f4SDimitry Andric Entry.Index = NumEntries++; 24480093f4SDimitry Andric Entry.Offset = CurrentEndOffset; 25480093f4SDimitry Andric Entry.Symbol = nullptr; 26480093f4SDimitry Andric CurrentEndOffset += S.size() + 1; 27480093f4SDimitry Andric } 28*81ad6265SDimitry Andric return DwarfStringPoolEntryRef(*I.first); 29480093f4SDimitry Andric } 30480093f4SDimitry Andric 31480093f4SDimitry Andric StringRef NonRelocatableStringpool::internString(StringRef S) { 32480093f4SDimitry Andric DwarfStringPoolEntry Entry{nullptr, 0, DwarfStringPoolEntry::NotIndexed}; 33480093f4SDimitry Andric 34480093f4SDimitry Andric if (Translator) 35480093f4SDimitry Andric S = Translator(S); 36480093f4SDimitry Andric 37480093f4SDimitry Andric auto InsertResult = Strings.insert({S, Entry}); 38480093f4SDimitry Andric return InsertResult.first->getKey(); 39480093f4SDimitry Andric } 40480093f4SDimitry Andric 41480093f4SDimitry Andric std::vector<DwarfStringPoolEntryRef> 42480093f4SDimitry Andric NonRelocatableStringpool::getEntriesForEmission() const { 43480093f4SDimitry Andric std::vector<DwarfStringPoolEntryRef> Result; 44480093f4SDimitry Andric Result.reserve(Strings.size()); 45480093f4SDimitry Andric for (const auto &E : Strings) 46480093f4SDimitry Andric if (E.getValue().isIndexed()) 47*81ad6265SDimitry Andric Result.emplace_back(E); 48480093f4SDimitry Andric llvm::sort(Result, [](const DwarfStringPoolEntryRef A, 49480093f4SDimitry Andric const DwarfStringPoolEntryRef B) { 50480093f4SDimitry Andric return A.getIndex() < B.getIndex(); 51480093f4SDimitry Andric }); 52480093f4SDimitry Andric return Result; 53480093f4SDimitry Andric } 54480093f4SDimitry Andric 55480093f4SDimitry Andric } // namespace llvm 56