diff --git a/include/llvm/Bitcode/Serialize.h b/include/llvm/Bitcode/Serialize.h index b0b39de3b79..5e6eaa05c22 100644 --- a/include/llvm/Bitcode/Serialize.h +++ b/include/llvm/Bitcode/Serialize.h @@ -33,39 +33,41 @@ class Serializer { public: explicit Serializer(BitstreamWriter& stream); ~Serializer(); + + //==------------------------------------------------==// + // Template-based dispatch to emit arbitrary types. + //==------------------------------------------------==// - template + template inline void Emit(const T& X) { SerializeTrait::Emit(*this,X); } - template - struct Emitter { - Serializer &S; - - Emitter(Serializer& s) : S(s) {} - void operator()(const T& x) const { S.Emit(x); } - }; - - template - Emitter MakeEmitter() { return Emitter(*this); } + //==------------------------------------------------==// + // Methods to emit primitive types. + //==------------------------------------------------==// void EmitInt(uint64_t X); void EmitSInt(int64_t X); - void EmitBool(bool X) { EmitInt(X); } + inline void EmitBool(bool X) { EmitInt(X); } void EmitCStr(const char* beg, const char* end); void EmitCStr(const char* cstr); void EmitPtr(const void* ptr) { EmitInt(getPtrId(ptr)); } template - void EmitRef(const T& ref) { EmitPtr(&ref); } + inline void EmitRef(const T& ref) { EmitPtr(&ref); } template - void EmitOwnedPtr(T* ptr) { + inline void EmitOwnedPtr(T* ptr) { EmitPtr(ptr); if (ptr) SerializeTrait::Emit(*this,*ptr); } + + //==------------------------------------------------==// + // Batch emission of pointers. + //==------------------------------------------------==// + template void BatchEmitOwnedPtrs(T1* p1, T2* p2) { EmitPtr(p1); @@ -135,6 +137,61 @@ public: if (p2) SerializeTrait::Emit(*this,*p2); if (p3) SerializeTrait::Emit(*this,*p3); } + + //==------------------------------------------------==// + // Emitter Functors + //==------------------------------------------------==// + + template + struct Emitter0 { + Serializer& S; + Emitter0(Serializer& s) : S(s) {} + void operator()(const T& x) const { + SerializeTrait::Emit(S,x); + } + }; + + template + struct Emitter1 { + Serializer& S; + Arg1 A1; + + Emitter1(Serializer& s, Arg1 a1) : S(s), A1(a1) {} + void operator()(const T& x) const { + SerializeTrait::Emit(S,x,A1); + } + }; + + template + struct Emitter2 { + Serializer& S; + Arg1 A1; + Arg2 A2; + + Emitter2(Serializer& s, Arg1 a1, Arg2 a2) : S(s), A1(a1), A2(a2) {} + void operator()(const T& x) const { + SerializeTrait::Emit(S,x,A1,A2); + } + }; + + template + Emitter0 MakeEmitter() { + return Emitter0(*this); + } + + template + Emitter1 MakeEmitter(Arg1 a1) { + return Emitter1(*this,a1); + } + + template + Emitter2 MakeEmitter(Arg1 a1, Arg2 a2) { + return Emitter2(*this,a1,a2); + } + + //==------------------------------------------------==// + // Misc. query and block/record manipulation methods. + //==------------------------------------------------==// bool isRegistered(const void* p) const;