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
103
104
105
|
#pragma once
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif
//===- StringTableBuilder.h - String table building utility -----*- 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
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_MC_STRINGTABLEBUILDER_H
#define LLVM_MC_STRINGTABLEBUILDER_H
#include "llvm/ADT/CachedHashString.h"
#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/StringRef.h"
#include <cstddef>
#include <cstdint>
namespace llvm {
class raw_ostream;
/// Utility for building string tables with deduplicated suffixes.
class StringTableBuilder {
public:
enum Kind {
ELF,
WinCOFF,
MachO,
MachO64,
MachOLinked,
MachO64Linked,
RAW,
DWARF,
XCOFF
};
private:
DenseMap<CachedHashStringRef, size_t> StringIndexMap;
size_t Size = 0;
Kind K;
unsigned Alignment;
bool Finalized = false;
void finalizeStringTable(bool Optimize);
void initSize();
public:
StringTableBuilder(Kind K, unsigned Alignment = 1);
~StringTableBuilder();
/// Add a string to the builder. Returns the position of S in the
/// table. The position will be changed if finalize is used.
/// Can only be used before the table is finalized.
size_t add(CachedHashStringRef S);
size_t add(StringRef S) { return add(CachedHashStringRef(S)); }
/// Analyze the strings and build the final table. No more strings can
/// be added after this point.
void finalize();
/// Finalize the string table without reording it. In this mode, offsets
/// returned by add will still be valid.
void finalizeInOrder();
/// Get the offest of a string in the string table. Can only be used
/// after the table is finalized.
size_t getOffset(CachedHashStringRef S) const;
size_t getOffset(StringRef S) const {
return getOffset(CachedHashStringRef(S));
}
/// Check if a string is contained in the string table. Since this class
/// doesn't store the string values, this function can be used to check if
/// storage needs to be done prior to adding the string.
bool contains(StringRef S) const {
return contains(CachedHashStringRef(S));
}
bool contains(CachedHashStringRef S) const {
return StringIndexMap.count(S);
}
size_t getSize() const { return Size; }
void clear();
void write(raw_ostream &OS) const;
void write(uint8_t *Buf) const;
private:
bool isFinalized() const { return Finalized; }
};
} // end namespace llvm
#endif // LLVM_MC_STRINGTABLEBUILDER_H
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
|
