]>
git.proxmox.com Git - rustc.git/blob - src/llvm/include/llvm/TableGen/StringToOffsetTable.h
1 //===- StringToOffsetTable.h - Emit a big concatenated string ---*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #ifndef LLVM_TABLEGEN_STRINGTOOFFSETTABLE_H
11 #define LLVM_TABLEGEN_STRINGTOOFFSETTABLE_H
13 #include "llvm/ADT/SmallString.h"
14 #include "llvm/ADT/StringExtras.h"
15 #include "llvm/ADT/StringMap.h"
16 #include "llvm/Support/raw_ostream.h"
21 /// StringToOffsetTable - This class uniques a bunch of nul-terminated strings
22 /// and keeps track of their offset in a massive contiguous string allocation.
23 /// It can then output this string blob and use indexes into the string to
24 /// reference each piece.
25 class StringToOffsetTable
{
26 StringMap
<unsigned> StringOffset
;
27 std::string AggregateString
;
30 unsigned GetOrAddStringOffset(StringRef Str
, bool appendZero
= true) {
32 StringOffset
.insert(std::make_pair(Str
, AggregateString
.size()));
33 if (IterBool
.second
) {
34 // Add the string to the aggregate if this is the first time found.
35 AggregateString
.append(Str
.begin(), Str
.end());
37 AggregateString
+= '\0';
40 return IterBool
.first
->second
;
43 void EmitString(raw_ostream
&O
) {
46 raw_svector_ostream(Str
).write_escaped(AggregateString
);
47 AggregateString
= Str
.str();
50 unsigned CharsPrinted
= 0;
51 for (unsigned i
= 0, e
= AggregateString
.size(); i
!= e
; ++i
) {
52 if (CharsPrinted
> 70) {
56 O
<< AggregateString
[i
];
59 // Print escape sequences all together.
60 if (AggregateString
[i
] != '\\')
63 assert(i
+1 < AggregateString
.size() && "Incomplete escape sequence!");
64 if (isdigit(AggregateString
[i
+1])) {
65 assert(isdigit(AggregateString
[i
+2]) &&
66 isdigit(AggregateString
[i
+3]) &&
67 "Expected 3 digit octal escape!");
68 O
<< AggregateString
[++i
];
69 O
<< AggregateString
[++i
];
70 O
<< AggregateString
[++i
];
73 O
<< AggregateString
[++i
];
81 } // end namespace llvm