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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
//=- Serialization.h - Generic Object Serialization to Bitcode ---*- C++ -*-=//
//
// The LLVM Compiler Infrastructure
//
// This file was developed by Ted Kremenek and is distributed under the
// University of Illinois Open Source License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file defines the interface for generic object serialization to
// LLVM bitcode.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_BITCODE_SERIALIZE
#define LLVM_BITCODE_SERIALIZE
#include "llvm/Bitcode/BitstreamWriter.h"
#include "llvm/Bitcode/BitstreamReader.h"
#include "llvm/ADT/SmallVector.h"
#include <vector>
namespace llvm {
template <typename T> struct SerializeTrait;
class Serializer {
BitstreamWriter& Stream;
SmallVector<uint64_t,10> Record;
bool inBlock;
public:
Serializer(BitstreamWriter& stream, unsigned BlockID = 0);
~Serializer();
template <typename T>
inline void Emit(const T& X) { SerializeTrait<T>::Serialize(*this,X); }
void EmitInt(unsigned X, unsigned bits);
// FIXME: Substitute a better implementation which calculates the minimum
// number of bits needed to serialize the enum.
void EmitEnum(unsigned X, unsigned MinVal, unsigned MaxVal) { EmitInt(X,32); }
void EmitCString(const char* cstr);
void Flush() { if (inRecord()) EmitRecord(); }
private:
void EmitRecord();
inline bool inRecord() { return Record.size() > 0; }
};
class Deserializer {
BitstreamReader& Stream;
SmallVector<uint64_t,10> Record;
unsigned RecIdx;
public:
Deserializer(BitstreamReader& stream);
~Deserializer();
template <typename T>
inline T& Read(T& X) { SerializeTrait<T>::Deserialize(*this,X); return X; }
template <typename T>
inline T* Materialize() {
T* X = SerializeTrait<T>::Instantiate();
Read(*X);
return X;
}
uint64_t ReadInt(unsigned bits = 32);
bool ReadBool() { return ReadInt(1); }
// FIXME: Substitute a better implementation which calculates the minimum
// number of bits needed to serialize the enum.
template <typename EnumT>
EnumT ReadEnum(unsigned MinVal, unsigned MaxVal) {
return static_cast<EnumT>(ReadInt(32));
}
char* ReadCString(char* cstr = NULL, unsigned MaxLen=0, bool isNullTerm=true);
void ReadCString(std::vector<char>& buff, bool isNullTerm=false);
private:
void ReadRecord();
inline bool inRecord() {
if (Record.size() > 0) {
if (RecIdx >= Record.size()) {
RecIdx = 0;
Record.clear();
return false;
}
else return true;
}
else return false;
}
};
template <typename uintty, unsigned Bits>
struct SerializeIntTrait {
static inline void Serialize(Serializer& S, uintty X) {
S.EmitInt(X,Bits);
}
static inline void Deserialize(Deserializer& S, uintty& X) {
X = (uintty) S.ReadInt(Bits);
}
};
template <> struct SerializeTrait<bool>
: public SerializeIntTrait<bool,1> {};
template <> struct SerializeTrait<char>
: public SerializeIntTrait<char,8> {};
template <> struct SerializeTrait<short>
: public SerializeIntTrait<short,16> {};
template <> struct SerializeTrait<unsigned>
: public SerializeIntTrait<unsigned,32> {};
} // end namespace llvm
#endif
|
