blob: df1d8a53dd308df45b250638fcfc4c30f3608eb9 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#pragma once
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif
//===-- LVStringPool.h ------------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file defines the LVStringPool class, which is used to implement a
// basic string pool table.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_DEBUGINFO_LOGICALVIEW_CORE_LVSTRINGPOOL_H
#define LLVM_DEBUGINFO_LOGICALVIEW_CORE_LVSTRINGPOOL_H
#include "llvm/ADT/StringMap.h"
#include "llvm/Support/Allocator.h"
#include "llvm/Support/Debug.h"
#include "llvm/Support/Format.h"
#include "llvm/Support/raw_ostream.h"
#include <iomanip>
#include <vector>
namespace llvm {
namespace logicalview {
class LVStringPool {
static constexpr size_t BadIndex = std::numeric_limits<size_t>::max();
using TableType = StringMap<size_t, BumpPtrAllocator>;
using ValueType = TableType::value_type;
BumpPtrAllocator Allocator;
TableType StringTable;
std::vector<ValueType *> Entries;
public:
LVStringPool() { getIndex(""); }
LVStringPool(LVStringPool const &other) = delete;
LVStringPool(LVStringPool &&other) = delete;
~LVStringPool() = default;
bool isValidIndex(size_t Index) const { return Index != BadIndex; }
// Return number of strings in the pool. The empty string is allocated
// at the slot zero. We substract 1 to indicate the number of non empty
// strings.
size_t getSize() const { return Entries.size() - 1; }
// Return the index for the specified key, otherwise 'BadIndex'.
size_t findIndex(StringRef Key) const {
TableType::const_iterator Iter = StringTable.find(Key);
if (Iter != StringTable.end())
return Iter->second;
return BadIndex;
}
// Return an index for the specified key.
size_t getIndex(StringRef Key) {
size_t Index = findIndex(Key);
if (isValidIndex(Index))
return Index;
size_t Value = Entries.size();
ValueType *Entry = ValueType::create(Key, Allocator, std::move(Value));
StringTable.insert(Entry);
Entries.push_back(Entry);
return Value;
}
// Given the index, return its corresponding string.
StringRef getString(size_t Index) const {
return (Index >= Entries.size()) ? StringRef() : Entries[Index]->getKey();
}
void print(raw_ostream &OS) const {
if (!Entries.empty()) {
OS << "\nString Pool:\n";
for (const ValueType *Entry : Entries)
OS << "Index: " << Entry->getValue() << ", "
<< "Key: '" << Entry->getKey() << "'\n";
}
}
#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
void dump() const { print(dbgs()); }
#endif
};
} // namespace logicalview
} // end namespace llvm
#endif // LLVM_DEBUGINFO_LOGICALVIEW_CORE_LVSTRINGPOOL_H
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
|
