Viewing file:  StringTableBuilder.h (2.57 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
//===- 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 "llvm/Support/Alignment.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,
DXContainer
};
private:
DenseMap<CachedHashStringRef, size_t> StringIndexMap;
size_t Size = 0;
Kind K;
Align Alignment;
bool Finalized = false;
void finalizeStringTable(bool Optimize);
void initSize();
public:
StringTableBuilder(Kind K, Align Alignment = Align(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;
bool isFinalized() const { return Finalized; }
};
} // end namespace llvm
#endif // LLVM_MC_STRINGTABLEBUILDER_H
|