blob: d0e4802524bf912e8d611bcca16d716dad09e6e1 [file] [log] [blame]
Sebastian Redld6522cf2010-08-18 23:56:31 +00001//===--- ASTWriter.cpp - AST File Writer ----------------------------------===//
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Sebastian Redl55c0ad52010-08-18 23:56:21 +000010// This file defines the ASTWriter class, which writes AST files.
Douglas Gregoref84c4b2009-04-09 22:27:44 +000011//
12//===----------------------------------------------------------------------===//
13
Sebastian Redl1914c6f2010-08-18 23:56:37 +000014#include "clang/Serialization/ASTWriter.h"
Argyrios Kyrtzidis4bd97102010-08-20 16:03:52 +000015#include "ASTCommon.h"
Douglas Gregorc3a6ade2010-08-12 20:07:10 +000016#include "clang/Sema/Sema.h"
17#include "clang/Sema/IdentifierResolver.h"
Douglas Gregoref84c4b2009-04-09 22:27:44 +000018#include "clang/AST/ASTContext.h"
19#include "clang/AST/Decl.h"
20#include "clang/AST/DeclContextInternals.h"
John McCall19c1bfd2010-08-25 05:32:35 +000021#include "clang/AST/DeclTemplate.h"
Douglas Gregorfeb84b02009-04-14 21:18:50 +000022#include "clang/AST/Expr.h"
John McCallbfd822c2010-08-24 07:32:53 +000023#include "clang/AST/ExprCXX.h"
Douglas Gregoref84c4b2009-04-09 22:27:44 +000024#include "clang/AST/Type.h"
John McCall8f115c62009-10-16 21:56:05 +000025#include "clang/AST/TypeLocVisitor.h"
Sebastian Redlf5b13462010-08-18 23:57:17 +000026#include "clang/Serialization/ASTReader.h"
Chris Lattnerbaa52f42009-04-10 18:00:12 +000027#include "clang/Lex/MacroInfo.h"
Douglas Gregoraae92242010-03-19 21:51:54 +000028#include "clang/Lex/PreprocessingRecord.h"
Chris Lattnerbaa52f42009-04-10 18:00:12 +000029#include "clang/Lex/Preprocessor.h"
Steve Naroff3fa455a2009-04-24 20:03:17 +000030#include "clang/Lex/HeaderSearch.h"
Douglas Gregora7f71a92009-04-10 03:52:48 +000031#include "clang/Basic/FileManager.h"
Douglas Gregore84a9da2009-04-20 20:36:09 +000032#include "clang/Basic/OnDiskHashTable.h"
Douglas Gregora7f71a92009-04-10 03:52:48 +000033#include "clang/Basic/SourceManager.h"
Douglas Gregor4c7626e2009-04-13 16:31:14 +000034#include "clang/Basic/SourceManagerInternals.h"
Douglas Gregorbfbde532009-04-10 21:16:55 +000035#include "clang/Basic/TargetInfo.h"
Douglas Gregor7b71e632009-04-27 22:23:34 +000036#include "clang/Basic/Version.h"
Douglas Gregore0a3a512009-04-14 21:55:33 +000037#include "llvm/ADT/APFloat.h"
38#include "llvm/ADT/APInt.h"
Daniel Dunbarf8502d52009-10-17 23:52:28 +000039#include "llvm/ADT/StringExtras.h"
Douglas Gregoref84c4b2009-04-09 22:27:44 +000040#include "llvm/Bitcode/BitstreamWriter.h"
Douglas Gregora7f71a92009-04-10 03:52:48 +000041#include "llvm/Support/MemoryBuffer.h"
Douglas Gregor45fe0362009-05-12 01:31:05 +000042#include "llvm/System/Path.h"
Chris Lattner225dd6c2009-04-11 18:40:46 +000043#include <cstdio>
Douglas Gregoref84c4b2009-04-09 22:27:44 +000044using namespace clang;
Sebastian Redl539c5062010-08-18 23:57:32 +000045using namespace clang::serialization;
Douglas Gregoref84c4b2009-04-09 22:27:44 +000046
Sebastian Redl3df5a082010-07-30 17:03:48 +000047template <typename T, typename Allocator>
48T *data(std::vector<T, Allocator> &v) {
49 return v.empty() ? 0 : &v.front();
50}
51template <typename T, typename Allocator>
52const T *data(const std::vector<T, Allocator> &v) {
53 return v.empty() ? 0 : &v.front();
54}
55
Douglas Gregoref84c4b2009-04-09 22:27:44 +000056//===----------------------------------------------------------------------===//
57// Type serialization
58//===----------------------------------------------------------------------===//
Chris Lattner7099dbc2009-04-27 06:16:06 +000059
Douglas Gregoref84c4b2009-04-09 22:27:44 +000060namespace {
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000061 class ASTTypeWriter {
Sebastian Redl55c0ad52010-08-18 23:56:21 +000062 ASTWriter &Writer;
63 ASTWriter::RecordData &Record;
Douglas Gregoref84c4b2009-04-09 22:27:44 +000064
65 public:
66 /// \brief Type code that corresponds to the record generated.
Sebastian Redl539c5062010-08-18 23:57:32 +000067 TypeCode Code;
Douglas Gregoref84c4b2009-04-09 22:27:44 +000068
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000069 ASTTypeWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
Sebastian Redl539c5062010-08-18 23:57:32 +000070 : Writer(Writer), Record(Record), Code(TYPE_EXT_QUAL) { }
Douglas Gregoref84c4b2009-04-09 22:27:44 +000071
72 void VisitArrayType(const ArrayType *T);
73 void VisitFunctionType(const FunctionType *T);
74 void VisitTagType(const TagType *T);
75
76#define TYPE(Class, Base) void Visit##Class##Type(const Class##Type *T);
77#define ABSTRACT_TYPE(Class, Base)
Douglas Gregoref84c4b2009-04-09 22:27:44 +000078#include "clang/AST/TypeNodes.def"
79 };
80}
81
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000082void ASTTypeWriter::VisitBuiltinType(const BuiltinType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +000083 assert(false && "Built-in types are never serialized");
84}
85
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000086void ASTTypeWriter::VisitComplexType(const ComplexType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +000087 Writer.AddTypeRef(T->getElementType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +000088 Code = TYPE_COMPLEX;
Douglas Gregoref84c4b2009-04-09 22:27:44 +000089}
90
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000091void ASTTypeWriter::VisitPointerType(const PointerType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +000092 Writer.AddTypeRef(T->getPointeeType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +000093 Code = TYPE_POINTER;
Douglas Gregoref84c4b2009-04-09 22:27:44 +000094}
95
Sebastian Redl42a0f6a2010-08-18 23:56:27 +000096void ASTTypeWriter::VisitBlockPointerType(const BlockPointerType *T) {
Mike Stump11289f42009-09-09 15:08:12 +000097 Writer.AddTypeRef(T->getPointeeType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +000098 Code = TYPE_BLOCK_POINTER;
Douglas Gregoref84c4b2009-04-09 22:27:44 +000099}
100
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000101void ASTTypeWriter::VisitLValueReferenceType(const LValueReferenceType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000102 Writer.AddTypeRef(T->getPointeeType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000103 Code = TYPE_LVALUE_REFERENCE;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000104}
105
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000106void ASTTypeWriter::VisitRValueReferenceType(const RValueReferenceType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000107 Writer.AddTypeRef(T->getPointeeType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000108 Code = TYPE_RVALUE_REFERENCE;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000109}
110
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000111void ASTTypeWriter::VisitMemberPointerType(const MemberPointerType *T) {
Mike Stump11289f42009-09-09 15:08:12 +0000112 Writer.AddTypeRef(T->getPointeeType(), Record);
113 Writer.AddTypeRef(QualType(T->getClass(), 0), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000114 Code = TYPE_MEMBER_POINTER;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000115}
116
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000117void ASTTypeWriter::VisitArrayType(const ArrayType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000118 Writer.AddTypeRef(T->getElementType(), Record);
119 Record.push_back(T->getSizeModifier()); // FIXME: stable values
John McCall8ccfcb52009-09-24 19:53:00 +0000120 Record.push_back(T->getIndexTypeCVRQualifiers()); // FIXME: stable values
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000121}
122
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000123void ASTTypeWriter::VisitConstantArrayType(const ConstantArrayType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000124 VisitArrayType(T);
125 Writer.AddAPInt(T->getSize(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000126 Code = TYPE_CONSTANT_ARRAY;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000127}
128
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000129void ASTTypeWriter::VisitIncompleteArrayType(const IncompleteArrayType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000130 VisitArrayType(T);
Sebastian Redl539c5062010-08-18 23:57:32 +0000131 Code = TYPE_INCOMPLETE_ARRAY;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000132}
133
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000134void ASTTypeWriter::VisitVariableArrayType(const VariableArrayType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000135 VisitArrayType(T);
Douglas Gregor04318252009-07-06 15:59:29 +0000136 Writer.AddSourceLocation(T->getLBracketLoc(), Record);
137 Writer.AddSourceLocation(T->getRBracketLoc(), Record);
Douglas Gregor8f45df52009-04-16 22:23:12 +0000138 Writer.AddStmt(T->getSizeExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +0000139 Code = TYPE_VARIABLE_ARRAY;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000140}
141
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000142void ASTTypeWriter::VisitVectorType(const VectorType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000143 Writer.AddTypeRef(T->getElementType(), Record);
144 Record.push_back(T->getNumElements());
Chris Lattner37141f42010-06-23 06:00:24 +0000145 Record.push_back(T->getAltiVecSpecific());
Sebastian Redl539c5062010-08-18 23:57:32 +0000146 Code = TYPE_VECTOR;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000147}
148
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000149void ASTTypeWriter::VisitExtVectorType(const ExtVectorType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000150 VisitVectorType(T);
Sebastian Redl539c5062010-08-18 23:57:32 +0000151 Code = TYPE_EXT_VECTOR;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000152}
153
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000154void ASTTypeWriter::VisitFunctionType(const FunctionType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000155 Writer.AddTypeRef(T->getResultType(), Record);
Rafael Espindolac50c27c2010-03-30 20:24:48 +0000156 FunctionType::ExtInfo C = T->getExtInfo();
157 Record.push_back(C.getNoReturn());
Rafael Espindola49b85ab2010-03-30 22:15:11 +0000158 Record.push_back(C.getRegParm());
Douglas Gregor8c940862010-01-18 17:14:39 +0000159 // FIXME: need to stabilize encoding of calling convention...
Rafael Espindolac50c27c2010-03-30 20:24:48 +0000160 Record.push_back(C.getCC());
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000161}
162
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000163void ASTTypeWriter::VisitFunctionNoProtoType(const FunctionNoProtoType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000164 VisitFunctionType(T);
Sebastian Redl539c5062010-08-18 23:57:32 +0000165 Code = TYPE_FUNCTION_NO_PROTO;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000166}
167
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000168void ASTTypeWriter::VisitFunctionProtoType(const FunctionProtoType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000169 VisitFunctionType(T);
170 Record.push_back(T->getNumArgs());
171 for (unsigned I = 0, N = T->getNumArgs(); I != N; ++I)
172 Writer.AddTypeRef(T->getArgType(I), Record);
173 Record.push_back(T->isVariadic());
174 Record.push_back(T->getTypeQuals());
Sebastian Redl5068f77ac2009-05-27 22:11:52 +0000175 Record.push_back(T->hasExceptionSpec());
176 Record.push_back(T->hasAnyExceptionSpec());
177 Record.push_back(T->getNumExceptions());
178 for (unsigned I = 0, N = T->getNumExceptions(); I != N; ++I)
179 Writer.AddTypeRef(T->getExceptionType(I), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000180 Code = TYPE_FUNCTION_PROTO;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000181}
182
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000183void ASTTypeWriter::VisitUnresolvedUsingType(const UnresolvedUsingType *T) {
John McCallb96ec562009-12-04 22:46:56 +0000184 Writer.AddDeclRef(T->getDecl(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000185 Code = TYPE_UNRESOLVED_USING;
John McCallb96ec562009-12-04 22:46:56 +0000186}
John McCallb96ec562009-12-04 22:46:56 +0000187
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000188void ASTTypeWriter::VisitTypedefType(const TypedefType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000189 Writer.AddDeclRef(T->getDecl(), Record);
Argyrios Kyrtzidis45a83f92010-07-02 11:55:11 +0000190 assert(!T->isCanonicalUnqualified() && "Invalid typedef ?");
191 Writer.AddTypeRef(T->getCanonicalTypeInternal(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000192 Code = TYPE_TYPEDEF;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000193}
194
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000195void ASTTypeWriter::VisitTypeOfExprType(const TypeOfExprType *T) {
Douglas Gregor8f45df52009-04-16 22:23:12 +0000196 Writer.AddStmt(T->getUnderlyingExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +0000197 Code = TYPE_TYPEOF_EXPR;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000198}
199
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000200void ASTTypeWriter::VisitTypeOfType(const TypeOfType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000201 Writer.AddTypeRef(T->getUnderlyingType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000202 Code = TYPE_TYPEOF;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000203}
204
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000205void ASTTypeWriter::VisitDecltypeType(const DecltypeType *T) {
Anders Carlsson81df7b82009-06-24 19:06:50 +0000206 Writer.AddStmt(T->getUnderlyingExpr());
Sebastian Redl539c5062010-08-18 23:57:32 +0000207 Code = TYPE_DECLTYPE;
Anders Carlsson81df7b82009-06-24 19:06:50 +0000208}
209
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000210void ASTTypeWriter::VisitTagType(const TagType *T) {
Argyrios Kyrtzidisa4ed1812010-07-08 13:09:53 +0000211 Record.push_back(T->isDependentType());
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000212 Writer.AddDeclRef(T->getDecl(), Record);
Mike Stump11289f42009-09-09 15:08:12 +0000213 assert(!T->isBeingDefined() &&
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000214 "Cannot serialize in the middle of a type definition");
215}
216
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000217void ASTTypeWriter::VisitRecordType(const RecordType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000218 VisitTagType(T);
Sebastian Redl539c5062010-08-18 23:57:32 +0000219 Code = TYPE_RECORD;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000220}
221
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000222void ASTTypeWriter::VisitEnumType(const EnumType *T) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000223 VisitTagType(T);
Sebastian Redl539c5062010-08-18 23:57:32 +0000224 Code = TYPE_ENUM;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000225}
226
Mike Stump11289f42009-09-09 15:08:12 +0000227void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000228ASTTypeWriter::VisitSubstTemplateTypeParmType(
John McCallcebee162009-10-18 09:09:24 +0000229 const SubstTemplateTypeParmType *T) {
230 Writer.AddTypeRef(QualType(T->getReplacedParameter(), 0), Record);
231 Writer.AddTypeRef(T->getReplacementType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000232 Code = TYPE_SUBST_TEMPLATE_TYPE_PARM;
John McCallcebee162009-10-18 09:09:24 +0000233}
234
235void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000236ASTTypeWriter::VisitTemplateSpecializationType(
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000237 const TemplateSpecializationType *T) {
Argyrios Kyrtzidisa4ed1812010-07-08 13:09:53 +0000238 Record.push_back(T->isDependentType());
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000239 Writer.AddTemplateName(T->getTemplateName(), Record);
240 Record.push_back(T->getNumArgs());
241 for (TemplateSpecializationType::iterator ArgI = T->begin(), ArgE = T->end();
242 ArgI != ArgE; ++ArgI)
243 Writer.AddTemplateArgument(*ArgI, Record);
Argyrios Kyrtzidis45a83f92010-07-02 11:55:11 +0000244 Writer.AddTypeRef(T->isCanonicalUnqualified() ? QualType()
245 : T->getCanonicalTypeInternal(),
246 Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000247 Code = TYPE_TEMPLATE_SPECIALIZATION;
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000248}
249
250void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000251ASTTypeWriter::VisitDependentSizedArrayType(const DependentSizedArrayType *T) {
Argyrios Kyrtzidis4a57bd02010-06-30 08:49:25 +0000252 VisitArrayType(T);
253 Writer.AddStmt(T->getSizeExpr());
254 Writer.AddSourceRange(T->getBracketsRange(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000255 Code = TYPE_DEPENDENT_SIZED_ARRAY;
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000256}
257
258void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000259ASTTypeWriter::VisitDependentSizedExtVectorType(
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000260 const DependentSizedExtVectorType *T) {
261 // FIXME: Serialize this type (C++ only)
262 assert(false && "Cannot serialize dependent sized extended vector types");
263}
264
265void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000266ASTTypeWriter::VisitTemplateTypeParmType(const TemplateTypeParmType *T) {
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000267 Record.push_back(T->getDepth());
268 Record.push_back(T->getIndex());
269 Record.push_back(T->isParameterPack());
270 Writer.AddIdentifierRef(T->getName(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000271 Code = TYPE_TEMPLATE_TYPE_PARM;
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000272}
273
274void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000275ASTTypeWriter::VisitDependentNameType(const DependentNameType *T) {
Argyrios Kyrtzidisbfcacee2010-06-24 08:57:31 +0000276 Record.push_back(T->getKeyword());
277 Writer.AddNestedNameSpecifier(T->getQualifier(), Record);
278 Writer.AddIdentifierRef(T->getIdentifier(), Record);
Argyrios Kyrtzidise9290952010-07-02 11:55:24 +0000279 Writer.AddTypeRef(T->isCanonicalUnqualified() ? QualType()
280 : T->getCanonicalTypeInternal(),
281 Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000282 Code = TYPE_DEPENDENT_NAME;
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000283}
284
285void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000286ASTTypeWriter::VisitDependentTemplateSpecializationType(
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +0000287 const DependentTemplateSpecializationType *T) {
Argyrios Kyrtzidisf0f7a792010-06-25 16:24:58 +0000288 Record.push_back(T->getKeyword());
289 Writer.AddNestedNameSpecifier(T->getQualifier(), Record);
290 Writer.AddIdentifierRef(T->getIdentifier(), Record);
291 Record.push_back(T->getNumArgs());
292 for (DependentTemplateSpecializationType::iterator
293 I = T->begin(), E = T->end(); I != E; ++I)
294 Writer.AddTemplateArgument(*I, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000295 Code = TYPE_DEPENDENT_TEMPLATE_SPECIALIZATION;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000296}
297
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000298void ASTTypeWriter::VisitElaboratedType(const ElaboratedType *T) {
Abramo Bagnara6150c882010-05-11 21:36:43 +0000299 Record.push_back(T->getKeyword());
Argyrios Kyrtzidisf0f7a792010-06-25 16:24:58 +0000300 Writer.AddNestedNameSpecifier(T->getQualifier(), Record);
301 Writer.AddTypeRef(T->getNamedType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000302 Code = TYPE_ELABORATED;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000303}
304
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000305void ASTTypeWriter::VisitInjectedClassNameType(const InjectedClassNameType *T) {
John McCalle78aac42010-03-10 03:28:59 +0000306 Writer.AddDeclRef(T->getDecl(), Record);
John McCall2408e322010-04-27 00:57:59 +0000307 Writer.AddTypeRef(T->getInjectedSpecializationType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000308 Code = TYPE_INJECTED_CLASS_NAME;
John McCalle78aac42010-03-10 03:28:59 +0000309}
310
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000311void ASTTypeWriter::VisitObjCInterfaceType(const ObjCInterfaceType *T) {
Douglas Gregor1c283312010-08-11 12:19:30 +0000312 Writer.AddDeclRef(T->getDecl(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000313 Code = TYPE_OBJC_INTERFACE;
John McCall8b07ec22010-05-15 11:32:37 +0000314}
315
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000316void ASTTypeWriter::VisitObjCObjectType(const ObjCObjectType *T) {
John McCall8b07ec22010-05-15 11:32:37 +0000317 Writer.AddTypeRef(T->getBaseType(), Record);
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000318 Record.push_back(T->getNumProtocols());
John McCall8b07ec22010-05-15 11:32:37 +0000319 for (ObjCObjectType::qual_iterator I = T->qual_begin(),
Steve Naroff4fc95aa2009-05-27 16:21:00 +0000320 E = T->qual_end(); I != E; ++I)
321 Writer.AddDeclRef(*I, Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000322 Code = TYPE_OBJC_OBJECT;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000323}
324
Steve Narofffb4330f2009-06-17 22:40:22 +0000325void
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000326ASTTypeWriter::VisitObjCObjectPointerType(const ObjCObjectPointerType *T) {
Mike Stump11289f42009-09-09 15:08:12 +0000327 Writer.AddTypeRef(T->getPointeeType(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +0000328 Code = TYPE_OBJC_OBJECT_POINTER;
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000329}
330
John McCall8f115c62009-10-16 21:56:05 +0000331namespace {
332
333class TypeLocWriter : public TypeLocVisitor<TypeLocWriter> {
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000334 ASTWriter &Writer;
335 ASTWriter::RecordData &Record;
John McCall8f115c62009-10-16 21:56:05 +0000336
337public:
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000338 TypeLocWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
John McCall8f115c62009-10-16 21:56:05 +0000339 : Writer(Writer), Record(Record) { }
340
John McCall17001972009-10-18 01:05:36 +0000341#define ABSTRACT_TYPELOC(CLASS, PARENT)
John McCall8f115c62009-10-16 21:56:05 +0000342#define TYPELOC(CLASS, PARENT) \
John McCall17001972009-10-18 01:05:36 +0000343 void Visit##CLASS##TypeLoc(CLASS##TypeLoc TyLoc);
John McCall8f115c62009-10-16 21:56:05 +0000344#include "clang/AST/TypeLocNodes.def"
345
John McCall17001972009-10-18 01:05:36 +0000346 void VisitArrayTypeLoc(ArrayTypeLoc TyLoc);
347 void VisitFunctionTypeLoc(FunctionTypeLoc TyLoc);
John McCall8f115c62009-10-16 21:56:05 +0000348};
349
350}
351
John McCall17001972009-10-18 01:05:36 +0000352void TypeLocWriter::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
353 // nothing to do
John McCall8f115c62009-10-16 21:56:05 +0000354}
John McCall17001972009-10-18 01:05:36 +0000355void TypeLocWriter::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
Douglas Gregorc9b7a592010-01-18 18:04:31 +0000356 Writer.AddSourceLocation(TL.getBuiltinLoc(), Record);
357 if (TL.needsExtraLocalData()) {
358 Record.push_back(TL.getWrittenTypeSpec());
359 Record.push_back(TL.getWrittenSignSpec());
360 Record.push_back(TL.getWrittenWidthSpec());
361 Record.push_back(TL.hasModeAttr());
362 }
John McCall8f115c62009-10-16 21:56:05 +0000363}
John McCall17001972009-10-18 01:05:36 +0000364void TypeLocWriter::VisitComplexTypeLoc(ComplexTypeLoc TL) {
365 Writer.AddSourceLocation(TL.getNameLoc(), Record);
John McCall8f115c62009-10-16 21:56:05 +0000366}
John McCall17001972009-10-18 01:05:36 +0000367void TypeLocWriter::VisitPointerTypeLoc(PointerTypeLoc TL) {
368 Writer.AddSourceLocation(TL.getStarLoc(), Record);
John McCall8f115c62009-10-16 21:56:05 +0000369}
John McCall17001972009-10-18 01:05:36 +0000370void TypeLocWriter::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
371 Writer.AddSourceLocation(TL.getCaretLoc(), Record);
John McCall8f115c62009-10-16 21:56:05 +0000372}
John McCall17001972009-10-18 01:05:36 +0000373void TypeLocWriter::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
374 Writer.AddSourceLocation(TL.getAmpLoc(), Record);
John McCall8f115c62009-10-16 21:56:05 +0000375}
John McCall17001972009-10-18 01:05:36 +0000376void TypeLocWriter::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
377 Writer.AddSourceLocation(TL.getAmpAmpLoc(), Record);
John McCall8f115c62009-10-16 21:56:05 +0000378}
John McCall17001972009-10-18 01:05:36 +0000379void TypeLocWriter::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
380 Writer.AddSourceLocation(TL.getStarLoc(), Record);
John McCall8f115c62009-10-16 21:56:05 +0000381}
John McCall17001972009-10-18 01:05:36 +0000382void TypeLocWriter::VisitArrayTypeLoc(ArrayTypeLoc TL) {
383 Writer.AddSourceLocation(TL.getLBracketLoc(), Record);
384 Writer.AddSourceLocation(TL.getRBracketLoc(), Record);
385 Record.push_back(TL.getSizeExpr() ? 1 : 0);
386 if (TL.getSizeExpr())
387 Writer.AddStmt(TL.getSizeExpr());
John McCall8f115c62009-10-16 21:56:05 +0000388}
John McCall17001972009-10-18 01:05:36 +0000389void TypeLocWriter::VisitConstantArrayTypeLoc(ConstantArrayTypeLoc TL) {
390 VisitArrayTypeLoc(TL);
391}
392void TypeLocWriter::VisitIncompleteArrayTypeLoc(IncompleteArrayTypeLoc TL) {
393 VisitArrayTypeLoc(TL);
394}
395void TypeLocWriter::VisitVariableArrayTypeLoc(VariableArrayTypeLoc TL) {
396 VisitArrayTypeLoc(TL);
397}
398void TypeLocWriter::VisitDependentSizedArrayTypeLoc(
399 DependentSizedArrayTypeLoc TL) {
400 VisitArrayTypeLoc(TL);
401}
402void TypeLocWriter::VisitDependentSizedExtVectorTypeLoc(
403 DependentSizedExtVectorTypeLoc TL) {
404 Writer.AddSourceLocation(TL.getNameLoc(), Record);
405}
406void TypeLocWriter::VisitVectorTypeLoc(VectorTypeLoc TL) {
407 Writer.AddSourceLocation(TL.getNameLoc(), Record);
408}
409void TypeLocWriter::VisitExtVectorTypeLoc(ExtVectorTypeLoc TL) {
410 Writer.AddSourceLocation(TL.getNameLoc(), Record);
411}
412void TypeLocWriter::VisitFunctionTypeLoc(FunctionTypeLoc TL) {
413 Writer.AddSourceLocation(TL.getLParenLoc(), Record);
414 Writer.AddSourceLocation(TL.getRParenLoc(), Record);
Douglas Gregor7fb25412010-10-01 18:44:50 +0000415 Record.push_back(TL.getTrailingReturn());
John McCall17001972009-10-18 01:05:36 +0000416 for (unsigned i = 0, e = TL.getNumArgs(); i != e; ++i)
417 Writer.AddDeclRef(TL.getArg(i), Record);
418}
419void TypeLocWriter::VisitFunctionProtoTypeLoc(FunctionProtoTypeLoc TL) {
420 VisitFunctionTypeLoc(TL);
421}
422void TypeLocWriter::VisitFunctionNoProtoTypeLoc(FunctionNoProtoTypeLoc TL) {
423 VisitFunctionTypeLoc(TL);
424}
John McCallb96ec562009-12-04 22:46:56 +0000425void TypeLocWriter::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
426 Writer.AddSourceLocation(TL.getNameLoc(), Record);
427}
John McCall17001972009-10-18 01:05:36 +0000428void TypeLocWriter::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
429 Writer.AddSourceLocation(TL.getNameLoc(), Record);
430}
431void TypeLocWriter::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
John McCalle8595032010-01-13 20:03:27 +0000432 Writer.AddSourceLocation(TL.getTypeofLoc(), Record);
433 Writer.AddSourceLocation(TL.getLParenLoc(), Record);
434 Writer.AddSourceLocation(TL.getRParenLoc(), Record);
John McCall17001972009-10-18 01:05:36 +0000435}
436void TypeLocWriter::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
John McCalle8595032010-01-13 20:03:27 +0000437 Writer.AddSourceLocation(TL.getTypeofLoc(), Record);
438 Writer.AddSourceLocation(TL.getLParenLoc(), Record);
439 Writer.AddSourceLocation(TL.getRParenLoc(), Record);
440 Writer.AddTypeSourceInfo(TL.getUnderlyingTInfo(), Record);
John McCall17001972009-10-18 01:05:36 +0000441}
442void TypeLocWriter::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
443 Writer.AddSourceLocation(TL.getNameLoc(), Record);
444}
445void TypeLocWriter::VisitRecordTypeLoc(RecordTypeLoc TL) {
446 Writer.AddSourceLocation(TL.getNameLoc(), Record);
447}
448void TypeLocWriter::VisitEnumTypeLoc(EnumTypeLoc TL) {
449 Writer.AddSourceLocation(TL.getNameLoc(), Record);
450}
John McCall17001972009-10-18 01:05:36 +0000451void TypeLocWriter::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
452 Writer.AddSourceLocation(TL.getNameLoc(), Record);
453}
John McCallcebee162009-10-18 09:09:24 +0000454void TypeLocWriter::VisitSubstTemplateTypeParmTypeLoc(
455 SubstTemplateTypeParmTypeLoc TL) {
456 Writer.AddSourceLocation(TL.getNameLoc(), Record);
457}
John McCall17001972009-10-18 01:05:36 +0000458void TypeLocWriter::VisitTemplateSpecializationTypeLoc(
459 TemplateSpecializationTypeLoc TL) {
John McCall0ad16662009-10-29 08:12:44 +0000460 Writer.AddSourceLocation(TL.getTemplateNameLoc(), Record);
461 Writer.AddSourceLocation(TL.getLAngleLoc(), Record);
462 Writer.AddSourceLocation(TL.getRAngleLoc(), Record);
463 for (unsigned i = 0, e = TL.getNumArgs(); i != e; ++i)
Argyrios Kyrtzidisae85e242010-06-22 09:54:59 +0000464 Writer.AddTemplateArgumentLocInfo(TL.getArgLoc(i).getArgument().getKind(),
465 TL.getArgLoc(i).getLocInfo(), Record);
John McCall17001972009-10-18 01:05:36 +0000466}
Abramo Bagnara6150c882010-05-11 21:36:43 +0000467void TypeLocWriter::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
Abramo Bagnarad7548482010-05-19 21:37:53 +0000468 Writer.AddSourceLocation(TL.getKeywordLoc(), Record);
469 Writer.AddSourceRange(TL.getQualifierRange(), Record);
John McCall17001972009-10-18 01:05:36 +0000470}
John McCalle78aac42010-03-10 03:28:59 +0000471void TypeLocWriter::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
472 Writer.AddSourceLocation(TL.getNameLoc(), Record);
473}
Douglas Gregorc1d2d8a2010-03-31 17:34:00 +0000474void TypeLocWriter::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
Abramo Bagnarad7548482010-05-19 21:37:53 +0000475 Writer.AddSourceLocation(TL.getKeywordLoc(), Record);
476 Writer.AddSourceRange(TL.getQualifierRange(), Record);
John McCall17001972009-10-18 01:05:36 +0000477 Writer.AddSourceLocation(TL.getNameLoc(), Record);
478}
John McCallc392f372010-06-11 00:33:02 +0000479void TypeLocWriter::VisitDependentTemplateSpecializationTypeLoc(
480 DependentTemplateSpecializationTypeLoc TL) {
481 Writer.AddSourceLocation(TL.getKeywordLoc(), Record);
482 Writer.AddSourceRange(TL.getQualifierRange(), Record);
483 Writer.AddSourceLocation(TL.getNameLoc(), Record);
484 Writer.AddSourceLocation(TL.getLAngleLoc(), Record);
485 Writer.AddSourceLocation(TL.getRAngleLoc(), Record);
486 for (unsigned I = 0, E = TL.getNumArgs(); I != E; ++I)
Argyrios Kyrtzidisae85e242010-06-22 09:54:59 +0000487 Writer.AddTemplateArgumentLocInfo(TL.getArgLoc(I).getArgument().getKind(),
488 TL.getArgLoc(I).getLocInfo(), Record);
John McCallc392f372010-06-11 00:33:02 +0000489}
John McCall17001972009-10-18 01:05:36 +0000490void TypeLocWriter::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
491 Writer.AddSourceLocation(TL.getNameLoc(), Record);
John McCall8b07ec22010-05-15 11:32:37 +0000492}
493void TypeLocWriter::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
494 Record.push_back(TL.hasBaseTypeAsWritten());
John McCall17001972009-10-18 01:05:36 +0000495 Writer.AddSourceLocation(TL.getLAngleLoc(), Record);
496 Writer.AddSourceLocation(TL.getRAngleLoc(), Record);
497 for (unsigned i = 0, e = TL.getNumProtocols(); i != e; ++i)
498 Writer.AddSourceLocation(TL.getProtocolLoc(i), Record);
John McCall8f115c62009-10-16 21:56:05 +0000499}
John McCallfc93cf92009-10-22 22:37:11 +0000500void TypeLocWriter::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
501 Writer.AddSourceLocation(TL.getStarLoc(), Record);
John McCallfc93cf92009-10-22 22:37:11 +0000502}
John McCall8f115c62009-10-16 21:56:05 +0000503
Chris Lattner19cea4e2009-04-22 05:57:30 +0000504//===----------------------------------------------------------------------===//
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000505// ASTWriter Implementation
Douglas Gregoref84c4b2009-04-09 22:27:44 +0000506//===----------------------------------------------------------------------===//
507
Chris Lattner28fa4e62009-04-26 22:26:21 +0000508static void EmitBlockID(unsigned ID, const char *Name,
509 llvm::BitstreamWriter &Stream,
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000510 ASTWriter::RecordData &Record) {
Chris Lattner28fa4e62009-04-26 22:26:21 +0000511 Record.clear();
512 Record.push_back(ID);
513 Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_SETBID, Record);
514
515 // Emit the block name if present.
516 if (Name == 0 || Name[0] == 0) return;
517 Record.clear();
518 while (*Name)
519 Record.push_back(*Name++);
520 Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_BLOCKNAME, Record);
521}
522
523static void EmitRecordID(unsigned ID, const char *Name,
524 llvm::BitstreamWriter &Stream,
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000525 ASTWriter::RecordData &Record) {
Chris Lattner28fa4e62009-04-26 22:26:21 +0000526 Record.clear();
527 Record.push_back(ID);
528 while (*Name)
529 Record.push_back(*Name++);
530 Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_SETRECORDNAME, Record);
Chris Lattnerccac3a62009-04-27 00:49:53 +0000531}
532
533static void AddStmtsExprs(llvm::BitstreamWriter &Stream,
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000534 ASTWriter::RecordData &Record) {
Sebastian Redl539c5062010-08-18 23:57:32 +0000535#define RECORD(X) EmitRecordID(X, #X, Stream, Record)
Chris Lattnerccac3a62009-04-27 00:49:53 +0000536 RECORD(STMT_STOP);
537 RECORD(STMT_NULL_PTR);
538 RECORD(STMT_NULL);
539 RECORD(STMT_COMPOUND);
540 RECORD(STMT_CASE);
541 RECORD(STMT_DEFAULT);
542 RECORD(STMT_LABEL);
543 RECORD(STMT_IF);
544 RECORD(STMT_SWITCH);
545 RECORD(STMT_WHILE);
546 RECORD(STMT_DO);
547 RECORD(STMT_FOR);
548 RECORD(STMT_GOTO);
549 RECORD(STMT_INDIRECT_GOTO);
550 RECORD(STMT_CONTINUE);
551 RECORD(STMT_BREAK);
552 RECORD(STMT_RETURN);
553 RECORD(STMT_DECL);
554 RECORD(STMT_ASM);
555 RECORD(EXPR_PREDEFINED);
556 RECORD(EXPR_DECL_REF);
557 RECORD(EXPR_INTEGER_LITERAL);
558 RECORD(EXPR_FLOATING_LITERAL);
559 RECORD(EXPR_IMAGINARY_LITERAL);
560 RECORD(EXPR_STRING_LITERAL);
561 RECORD(EXPR_CHARACTER_LITERAL);
562 RECORD(EXPR_PAREN);
563 RECORD(EXPR_UNARY_OPERATOR);
564 RECORD(EXPR_SIZEOF_ALIGN_OF);
565 RECORD(EXPR_ARRAY_SUBSCRIPT);
566 RECORD(EXPR_CALL);
567 RECORD(EXPR_MEMBER);
568 RECORD(EXPR_BINARY_OPERATOR);
569 RECORD(EXPR_COMPOUND_ASSIGN_OPERATOR);
570 RECORD(EXPR_CONDITIONAL_OPERATOR);
571 RECORD(EXPR_IMPLICIT_CAST);
572 RECORD(EXPR_CSTYLE_CAST);
573 RECORD(EXPR_COMPOUND_LITERAL);
574 RECORD(EXPR_EXT_VECTOR_ELEMENT);
575 RECORD(EXPR_INIT_LIST);
576 RECORD(EXPR_DESIGNATED_INIT);
577 RECORD(EXPR_IMPLICIT_VALUE_INIT);
578 RECORD(EXPR_VA_ARG);
579 RECORD(EXPR_ADDR_LABEL);
580 RECORD(EXPR_STMT);
581 RECORD(EXPR_TYPES_COMPATIBLE);
582 RECORD(EXPR_CHOOSE);
583 RECORD(EXPR_GNU_NULL);
584 RECORD(EXPR_SHUFFLE_VECTOR);
585 RECORD(EXPR_BLOCK);
586 RECORD(EXPR_BLOCK_DECL_REF);
587 RECORD(EXPR_OBJC_STRING_LITERAL);
588 RECORD(EXPR_OBJC_ENCODE);
589 RECORD(EXPR_OBJC_SELECTOR_EXPR);
590 RECORD(EXPR_OBJC_PROTOCOL_EXPR);
591 RECORD(EXPR_OBJC_IVAR_REF_EXPR);
592 RECORD(EXPR_OBJC_PROPERTY_REF_EXPR);
593 RECORD(EXPR_OBJC_KVC_REF_EXPR);
594 RECORD(EXPR_OBJC_MESSAGE_EXPR);
Chris Lattnerccac3a62009-04-27 00:49:53 +0000595 RECORD(STMT_OBJC_FOR_COLLECTION);
596 RECORD(STMT_OBJC_CATCH);
597 RECORD(STMT_OBJC_FINALLY);
598 RECORD(STMT_OBJC_AT_TRY);
599 RECORD(STMT_OBJC_AT_SYNCHRONIZED);
600 RECORD(STMT_OBJC_AT_THROW);
Sam Weinige83b3ac2010-02-07 06:32:43 +0000601 RECORD(EXPR_CXX_OPERATOR_CALL);
602 RECORD(EXPR_CXX_CONSTRUCT);
603 RECORD(EXPR_CXX_STATIC_CAST);
604 RECORD(EXPR_CXX_DYNAMIC_CAST);
605 RECORD(EXPR_CXX_REINTERPRET_CAST);
606 RECORD(EXPR_CXX_CONST_CAST);
607 RECORD(EXPR_CXX_FUNCTIONAL_CAST);
608 RECORD(EXPR_CXX_BOOL_LITERAL);
609 RECORD(EXPR_CXX_NULL_PTR_LITERAL);
Chris Lattnerccac3a62009-04-27 00:49:53 +0000610#undef RECORD
Chris Lattner28fa4e62009-04-26 22:26:21 +0000611}
Mike Stump11289f42009-09-09 15:08:12 +0000612
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000613void ASTWriter::WriteBlockInfoBlock() {
Chris Lattner28fa4e62009-04-26 22:26:21 +0000614 RecordData Record;
615 Stream.EnterSubblock(llvm::bitc::BLOCKINFO_BLOCK_ID, 3);
Mike Stump11289f42009-09-09 15:08:12 +0000616
Sebastian Redl539c5062010-08-18 23:57:32 +0000617#define BLOCK(X) EmitBlockID(X ## _ID, #X, Stream, Record)
618#define RECORD(X) EmitRecordID(X, #X, Stream, Record)
Mike Stump11289f42009-09-09 15:08:12 +0000619
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000620 // AST Top-Level Block.
Sebastian Redlf1642042010-08-18 23:57:22 +0000621 BLOCK(AST_BLOCK);
Zhongxing Xub027cdf2009-06-03 09:23:28 +0000622 RECORD(ORIGINAL_FILE_NAME);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000623 RECORD(TYPE_OFFSET);
624 RECORD(DECL_OFFSET);
625 RECORD(LANGUAGE_OPTIONS);
Douglas Gregor7b71e632009-04-27 22:23:34 +0000626 RECORD(METADATA);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000627 RECORD(IDENTIFIER_OFFSET);
628 RECORD(IDENTIFIER_TABLE);
629 RECORD(EXTERNAL_DEFINITIONS);
630 RECORD(SPECIAL_TYPES);
631 RECORD(STATISTICS);
632 RECORD(TENTATIVE_DEFINITIONS);
Argyrios Kyrtzidis35672e72010-08-13 18:42:17 +0000633 RECORD(UNUSED_FILESCOPED_DECLS);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000634 RECORD(LOCALLY_SCOPED_EXTERNAL_DECLS);
635 RECORD(SELECTOR_OFFSETS);
636 RECORD(METHOD_POOL);
637 RECORD(PP_COUNTER_VALUE);
Douglas Gregor258ae542009-04-27 06:38:32 +0000638 RECORD(SOURCE_LOCATION_OFFSETS);
639 RECORD(SOURCE_LOCATION_PRELOADS);
Douglas Gregorc5046832009-04-27 18:38:38 +0000640 RECORD(STAT_CACHE);
Douglas Gregor61cac2b2009-04-27 20:06:05 +0000641 RECORD(EXT_VECTOR_DECLS);
Ted Kremenek17437132010-01-22 20:59:36 +0000642 RECORD(VERSION_CONTROL_BRANCH_REVISION);
Douglas Gregoraae92242010-03-19 21:51:54 +0000643 RECORD(MACRO_DEFINITION_OFFSETS);
Sebastian Redl595c5132010-07-08 22:01:51 +0000644 RECORD(CHAINED_METADATA);
Fariborz Jahanianc51609a2010-07-23 19:11:11 +0000645 RECORD(REFERENCED_SELECTOR_POOL);
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +0000646
Chris Lattner28fa4e62009-04-26 22:26:21 +0000647 // SourceManager Block.
Chris Lattner64031982009-04-27 00:40:25 +0000648 BLOCK(SOURCE_MANAGER_BLOCK);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000649 RECORD(SM_SLOC_FILE_ENTRY);
650 RECORD(SM_SLOC_BUFFER_ENTRY);
651 RECORD(SM_SLOC_BUFFER_BLOB);
652 RECORD(SM_SLOC_INSTANTIATION_ENTRY);
653 RECORD(SM_LINE_TABLE);
Mike Stump11289f42009-09-09 15:08:12 +0000654
Chris Lattner28fa4e62009-04-26 22:26:21 +0000655 // Preprocessor Block.
Chris Lattner64031982009-04-27 00:40:25 +0000656 BLOCK(PREPROCESSOR_BLOCK);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000657 RECORD(PP_MACRO_OBJECT_LIKE);
658 RECORD(PP_MACRO_FUNCTION_LIKE);
659 RECORD(PP_TOKEN);
Douglas Gregoraae92242010-03-19 21:51:54 +0000660 RECORD(PP_MACRO_INSTANTIATION);
661 RECORD(PP_MACRO_DEFINITION);
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +0000662
Douglas Gregor12bfa382009-10-17 00:13:19 +0000663 // Decls and Types block.
664 BLOCK(DECLTYPES_BLOCK);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000665 RECORD(TYPE_EXT_QUAL);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000666 RECORD(TYPE_COMPLEX);
667 RECORD(TYPE_POINTER);
668 RECORD(TYPE_BLOCK_POINTER);
669 RECORD(TYPE_LVALUE_REFERENCE);
670 RECORD(TYPE_RVALUE_REFERENCE);
671 RECORD(TYPE_MEMBER_POINTER);
672 RECORD(TYPE_CONSTANT_ARRAY);
673 RECORD(TYPE_INCOMPLETE_ARRAY);
674 RECORD(TYPE_VARIABLE_ARRAY);
675 RECORD(TYPE_VECTOR);
676 RECORD(TYPE_EXT_VECTOR);
677 RECORD(TYPE_FUNCTION_PROTO);
678 RECORD(TYPE_FUNCTION_NO_PROTO);
679 RECORD(TYPE_TYPEDEF);
680 RECORD(TYPE_TYPEOF_EXPR);
681 RECORD(TYPE_TYPEOF);
682 RECORD(TYPE_RECORD);
683 RECORD(TYPE_ENUM);
684 RECORD(TYPE_OBJC_INTERFACE);
John McCall94f619a2010-05-16 02:12:35 +0000685 RECORD(TYPE_OBJC_OBJECT);
Steve Narofffb4330f2009-06-17 22:40:22 +0000686 RECORD(TYPE_OBJC_OBJECT_POINTER);
Chris Lattnerdb397b62009-04-26 22:32:16 +0000687 RECORD(DECL_TRANSLATION_UNIT);
688 RECORD(DECL_TYPEDEF);
689 RECORD(DECL_ENUM);
690 RECORD(DECL_RECORD);
691 RECORD(DECL_ENUM_CONSTANT);
692 RECORD(DECL_FUNCTION);
693 RECORD(DECL_OBJC_METHOD);
694 RECORD(DECL_OBJC_INTERFACE);
695 RECORD(DECL_OBJC_PROTOCOL);
696 RECORD(DECL_OBJC_IVAR);
697 RECORD(DECL_OBJC_AT_DEFS_FIELD);
698 RECORD(DECL_OBJC_CLASS);
699 RECORD(DECL_OBJC_FORWARD_PROTOCOL);
700 RECORD(DECL_OBJC_CATEGORY);
701 RECORD(DECL_OBJC_CATEGORY_IMPL);
702 RECORD(DECL_OBJC_IMPLEMENTATION);
703 RECORD(DECL_OBJC_COMPATIBLE_ALIAS);
704 RECORD(DECL_OBJC_PROPERTY);
705 RECORD(DECL_OBJC_PROPERTY_IMPL);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000706 RECORD(DECL_FIELD);
707 RECORD(DECL_VAR);
Chris Lattnerdb397b62009-04-26 22:32:16 +0000708 RECORD(DECL_IMPLICIT_PARAM);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000709 RECORD(DECL_PARM_VAR);
Chris Lattnerdb397b62009-04-26 22:32:16 +0000710 RECORD(DECL_FILE_SCOPE_ASM);
711 RECORD(DECL_BLOCK);
712 RECORD(DECL_CONTEXT_LEXICAL);
713 RECORD(DECL_CONTEXT_VISIBLE);
Douglas Gregor12bfa382009-10-17 00:13:19 +0000714 // Statements and Exprs can occur in the Decls and Types block.
Chris Lattnerccac3a62009-04-27 00:49:53 +0000715 AddStmtsExprs(Stream, Record);
Chris Lattner28fa4e62009-04-26 22:26:21 +0000716#undef RECORD
717#undef BLOCK
718 Stream.ExitBlock();
719}
720
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000721/// \brief Adjusts the given filename to only write out the portion of the
722/// filename that is not part of the system root directory.
Mike Stump11289f42009-09-09 15:08:12 +0000723///
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000724/// \param Filename the file name to adjust.
725///
726/// \param isysroot When non-NULL, the PCH file is a relocatable PCH file and
727/// the returned filename will be adjusted by this system root.
728///
729/// \returns either the original filename (if it needs no adjustment) or the
730/// adjusted filename (which points into the @p Filename parameter).
Mike Stump11289f42009-09-09 15:08:12 +0000731static const char *
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000732adjustFilenameForRelocatablePCH(const char *Filename, const char *isysroot) {
733 assert(Filename && "No file name to adjust?");
Mike Stump11289f42009-09-09 15:08:12 +0000734
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000735 if (!isysroot)
736 return Filename;
Mike Stump11289f42009-09-09 15:08:12 +0000737
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000738 // Verify that the filename and the system root have the same prefix.
739 unsigned Pos = 0;
740 for (; Filename[Pos] && isysroot[Pos]; ++Pos)
741 if (Filename[Pos] != isysroot[Pos])
742 return Filename; // Prefixes don't match.
Mike Stump11289f42009-09-09 15:08:12 +0000743
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000744 // We hit the end of the filename before we hit the end of the system root.
745 if (!Filename[Pos])
746 return Filename;
Mike Stump11289f42009-09-09 15:08:12 +0000747
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000748 // If the file name has a '/' at the current position, skip over the '/'.
749 // We distinguish sysroot-based includes from absolute includes by the
750 // absence of '/' at the beginning of sysroot-based includes.
751 if (Filename[Pos] == '/')
752 ++Pos;
Mike Stump11289f42009-09-09 15:08:12 +0000753
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000754 return Filename + Pos;
755}
Chris Lattner28fa4e62009-04-26 22:26:21 +0000756
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000757/// \brief Write the AST metadata (e.g., i686-apple-darwin9).
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000758void ASTWriter::WriteMetadata(ASTContext &Context, const char *isysroot) {
Douglas Gregorbfbde532009-04-10 21:16:55 +0000759 using namespace llvm;
Douglas Gregor45fe0362009-05-12 01:31:05 +0000760
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000761 // Metadata
762 const TargetInfo &Target = Context.Target;
763 BitCodeAbbrev *MetaAbbrev = new BitCodeAbbrev();
Sebastian Redl4d3af3e2010-07-09 21:00:24 +0000764 MetaAbbrev->Add(BitCodeAbbrevOp(
Sebastian Redl539c5062010-08-18 23:57:32 +0000765 Chain ? CHAINED_METADATA : METADATA));
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000766 MetaAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // AST major
767 MetaAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // AST minor
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000768 MetaAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // Clang major
769 MetaAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // Clang minor
770 MetaAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // Relocatable
Sebastian Redl4d3af3e2010-07-09 21:00:24 +0000771 // Target triple or chained PCH name
772 MetaAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000773 unsigned MetaAbbrevCode = Stream.EmitAbbrev(MetaAbbrev);
Mike Stump11289f42009-09-09 15:08:12 +0000774
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000775 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +0000776 Record.push_back(Chain ? CHAINED_METADATA : METADATA);
777 Record.push_back(VERSION_MAJOR);
778 Record.push_back(VERSION_MINOR);
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000779 Record.push_back(CLANG_VERSION_MAJOR);
780 Record.push_back(CLANG_VERSION_MINOR);
781 Record.push_back(isysroot != 0);
Sebastian Redl4d3af3e2010-07-09 21:00:24 +0000782 // FIXME: This writes the absolute path for chained headers.
783 const std::string &BlobStr = Chain ? Chain->getFileName() : Target.getTriple().getTriple();
784 Stream.EmitRecordWithBlob(MetaAbbrevCode, Record, BlobStr);
Mike Stump11289f42009-09-09 15:08:12 +0000785
Douglas Gregor45fe0362009-05-12 01:31:05 +0000786 // Original file name
787 SourceManager &SM = Context.getSourceManager();
788 if (const FileEntry *MainFile = SM.getFileEntryForID(SM.getMainFileID())) {
789 BitCodeAbbrev *FileAbbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +0000790 FileAbbrev->Add(BitCodeAbbrevOp(ORIGINAL_FILE_NAME));
Douglas Gregor45fe0362009-05-12 01:31:05 +0000791 FileAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // File name
792 unsigned FileAbbrevCode = Stream.EmitAbbrev(FileAbbrev);
793
794 llvm::sys::Path MainFilePath(MainFile->getName());
Mike Stump11289f42009-09-09 15:08:12 +0000795
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +0000796 MainFilePath.makeAbsolute();
Douglas Gregor45fe0362009-05-12 01:31:05 +0000797
Kovarththanan Rajaratnamd16d38c2010-03-14 07:15:57 +0000798 const char *MainFileNameStr = MainFilePath.c_str();
Mike Stump11289f42009-09-09 15:08:12 +0000799 MainFileNameStr = adjustFilenameForRelocatablePCH(MainFileNameStr,
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000800 isysroot);
Douglas Gregor45fe0362009-05-12 01:31:05 +0000801 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +0000802 Record.push_back(ORIGINAL_FILE_NAME);
Daniel Dunbar8100d012009-08-24 09:31:37 +0000803 Stream.EmitRecordWithBlob(FileAbbrevCode, Record, MainFileNameStr);
Douglas Gregor45fe0362009-05-12 01:31:05 +0000804 }
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +0000805
Ted Kremenek18e066f2010-01-22 22:12:47 +0000806 // Repository branch/version information.
807 BitCodeAbbrev *RepoAbbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +0000808 RepoAbbrev->Add(BitCodeAbbrevOp(VERSION_CONTROL_BRANCH_REVISION));
Ted Kremenek18e066f2010-01-22 22:12:47 +0000809 RepoAbbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // SVN branch/tag
810 unsigned RepoAbbrevCode = Stream.EmitAbbrev(RepoAbbrev);
Douglas Gregord54f3a12009-10-05 21:07:28 +0000811 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +0000812 Record.push_back(VERSION_CONTROL_BRANCH_REVISION);
Ted Kremenek18e066f2010-01-22 22:12:47 +0000813 Stream.EmitRecordWithBlob(RepoAbbrevCode, Record,
814 getClangFullRepositoryVersion());
Douglas Gregorbfbde532009-04-10 21:16:55 +0000815}
816
817/// \brief Write the LangOptions structure.
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000818void ASTWriter::WriteLanguageOptions(const LangOptions &LangOpts) {
Douglas Gregor55abb232009-04-10 20:39:37 +0000819 RecordData Record;
820 Record.push_back(LangOpts.Trigraphs);
821 Record.push_back(LangOpts.BCPLComment); // BCPL-style '//' comments.
822 Record.push_back(LangOpts.DollarIdents); // '$' allowed in identifiers.
823 Record.push_back(LangOpts.AsmPreprocessor); // Preprocessor in asm mode.
824 Record.push_back(LangOpts.GNUMode); // True in gnu99 mode false in c99 mode (etc)
Chandler Carruthe03aa552010-04-17 20:17:31 +0000825 Record.push_back(LangOpts.GNUKeywords); // Allow GNU-extension keywords
Douglas Gregor55abb232009-04-10 20:39:37 +0000826 Record.push_back(LangOpts.ImplicitInt); // C89 implicit 'int'.
827 Record.push_back(LangOpts.Digraphs); // C94, C99 and C++
828 Record.push_back(LangOpts.HexFloats); // C99 Hexadecimal float constants.
829 Record.push_back(LangOpts.C99); // C99 Support
830 Record.push_back(LangOpts.Microsoft); // Microsoft extensions.
831 Record.push_back(LangOpts.CPlusPlus); // C++ Support
832 Record.push_back(LangOpts.CPlusPlus0x); // C++0x Support
Douglas Gregor55abb232009-04-10 20:39:37 +0000833 Record.push_back(LangOpts.CXXOperatorNames); // Treat C++ operator names as keywords.
Mike Stump11289f42009-09-09 15:08:12 +0000834
Douglas Gregor55abb232009-04-10 20:39:37 +0000835 Record.push_back(LangOpts.ObjC1); // Objective-C 1 support enabled.
836 Record.push_back(LangOpts.ObjC2); // Objective-C 2 support enabled.
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +0000837 Record.push_back(LangOpts.ObjCNonFragileABI); // Objective-C
Fariborz Jahanian45878032010-02-09 19:31:38 +0000838 // modern abi enabled.
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +0000839 Record.push_back(LangOpts.ObjCNonFragileABI2); // Objective-C enhanced
Fariborz Jahanian45878032010-02-09 19:31:38 +0000840 // modern abi enabled.
Fariborz Jahanian62c56022010-04-22 21:01:59 +0000841 Record.push_back(LangOpts.NoConstantCFStrings); // non cfstring generation enabled..
Mike Stump11289f42009-09-09 15:08:12 +0000842
Douglas Gregor55abb232009-04-10 20:39:37 +0000843 Record.push_back(LangOpts.PascalStrings); // Allow Pascal strings
Douglas Gregor55abb232009-04-10 20:39:37 +0000844 Record.push_back(LangOpts.WritableStrings); // Allow writable strings
845 Record.push_back(LangOpts.LaxVectorConversions);
Nate Begemanf2911662009-06-25 23:01:11 +0000846 Record.push_back(LangOpts.AltiVec);
Douglas Gregor55abb232009-04-10 20:39:37 +0000847 Record.push_back(LangOpts.Exceptions); // Support exception handling.
Daniel Dunbar925152c2010-02-10 18:48:44 +0000848 Record.push_back(LangOpts.SjLjExceptions);
Douglas Gregor55abb232009-04-10 20:39:37 +0000849
850 Record.push_back(LangOpts.NeXTRuntime); // Use NeXT runtime.
851 Record.push_back(LangOpts.Freestanding); // Freestanding implementation
852 Record.push_back(LangOpts.NoBuiltin); // Do not use builtin functions (-fno-builtin)
853
Chris Lattner258172e2009-04-27 07:35:58 +0000854 // Whether static initializers are protected by locks.
855 Record.push_back(LangOpts.ThreadsafeStatics);
Douglas Gregorb3286fe2009-09-03 14:36:33 +0000856 Record.push_back(LangOpts.POSIXThreads);
Douglas Gregor55abb232009-04-10 20:39:37 +0000857 Record.push_back(LangOpts.Blocks); // block extension to C
858 Record.push_back(LangOpts.EmitAllDecls); // Emit all declarations, even if
859 // they are unused.
860 Record.push_back(LangOpts.MathErrno); // Math functions must respect errno
861 // (modulo the platform support).
862
Chris Lattner51924e512010-06-26 21:25:03 +0000863 Record.push_back(LangOpts.getSignedOverflowBehavior());
864 Record.push_back(LangOpts.HeinousExtensions);
Douglas Gregor55abb232009-04-10 20:39:37 +0000865
866 Record.push_back(LangOpts.Optimize); // Whether __OPTIMIZE__ should be defined.
Mike Stump11289f42009-09-09 15:08:12 +0000867 Record.push_back(LangOpts.OptimizeSize); // Whether __OPTIMIZE_SIZE__ should be
Douglas Gregor55abb232009-04-10 20:39:37 +0000868 // defined.
869 Record.push_back(LangOpts.Static); // Should __STATIC__ be defined (as
870 // opposed to __DYNAMIC__).
871 Record.push_back(LangOpts.PICLevel); // The value for __PIC__, if non-zero.
872
873 Record.push_back(LangOpts.GNUInline); // Should GNU inline semantics be
874 // used (instead of C99 semantics).
875 Record.push_back(LangOpts.NoInline); // Should __NO_INLINE__ be defined.
Anders Carlsson5879fbd2009-05-13 19:49:53 +0000876 Record.push_back(LangOpts.AccessControl); // Whether C++ access control should
877 // be enabled.
Eli Friedman9ffd4a92009-06-05 07:05:05 +0000878 Record.push_back(LangOpts.CharIsSigned); // Whether char is a signed or
879 // unsigned type
John Thompsoned4e2952009-11-05 20:14:16 +0000880 Record.push_back(LangOpts.ShortWChar); // force wchar_t to be unsigned short
Douglas Gregor55abb232009-04-10 20:39:37 +0000881 Record.push_back(LangOpts.getGCMode());
882 Record.push_back(LangOpts.getVisibilityMode());
Daniel Dunbar143021e2009-09-21 04:16:19 +0000883 Record.push_back(LangOpts.getStackProtectorMode());
Douglas Gregor55abb232009-04-10 20:39:37 +0000884 Record.push_back(LangOpts.InstantiationDepth);
Nate Begemanf2911662009-06-25 23:01:11 +0000885 Record.push_back(LangOpts.OpenCL);
Mike Stumpd9546382009-12-12 01:27:46 +0000886 Record.push_back(LangOpts.CatchUndefined);
Anders Carlsson9cedbef2009-08-22 22:30:33 +0000887 Record.push_back(LangOpts.ElideConstructors);
Douglas Gregor8ed0c0b2010-07-09 17:35:33 +0000888 Record.push_back(LangOpts.SpellChecking);
Sebastian Redl539c5062010-08-18 23:57:32 +0000889 Stream.EmitRecord(LANGUAGE_OPTIONS, Record);
Douglas Gregor55abb232009-04-10 20:39:37 +0000890}
891
Douglas Gregora7f71a92009-04-10 03:52:48 +0000892//===----------------------------------------------------------------------===//
Douglas Gregorc5046832009-04-27 18:38:38 +0000893// stat cache Serialization
894//===----------------------------------------------------------------------===//
895
896namespace {
897// Trait used for the on-disk hash table of stat cache results.
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000898class ASTStatCacheTrait {
Douglas Gregorc5046832009-04-27 18:38:38 +0000899public:
900 typedef const char * key_type;
901 typedef key_type key_type_ref;
Mike Stump11289f42009-09-09 15:08:12 +0000902
Douglas Gregorc5046832009-04-27 18:38:38 +0000903 typedef std::pair<int, struct stat> data_type;
904 typedef const data_type& data_type_ref;
905
906 static unsigned ComputeHash(const char *path) {
Daniel Dunbarf8502d52009-10-17 23:52:28 +0000907 return llvm::HashString(path);
Douglas Gregorc5046832009-04-27 18:38:38 +0000908 }
Mike Stump11289f42009-09-09 15:08:12 +0000909
910 std::pair<unsigned,unsigned>
Douglas Gregorc5046832009-04-27 18:38:38 +0000911 EmitKeyDataLength(llvm::raw_ostream& Out, const char *path,
912 data_type_ref Data) {
913 unsigned StrLen = strlen(path);
914 clang::io::Emit16(Out, StrLen);
915 unsigned DataLen = 1; // result value
916 if (Data.first == 0)
917 DataLen += 4 + 4 + 2 + 8 + 8;
918 clang::io::Emit8(Out, DataLen);
919 return std::make_pair(StrLen + 1, DataLen);
920 }
Mike Stump11289f42009-09-09 15:08:12 +0000921
Douglas Gregorc5046832009-04-27 18:38:38 +0000922 void EmitKey(llvm::raw_ostream& Out, const char *path, unsigned KeyLen) {
923 Out.write(path, KeyLen);
924 }
Mike Stump11289f42009-09-09 15:08:12 +0000925
Douglas Gregorc5046832009-04-27 18:38:38 +0000926 void EmitData(llvm::raw_ostream& Out, key_type_ref,
927 data_type_ref Data, unsigned DataLen) {
928 using namespace clang::io;
929 uint64_t Start = Out.tell(); (void)Start;
Mike Stump11289f42009-09-09 15:08:12 +0000930
Douglas Gregorc5046832009-04-27 18:38:38 +0000931 // Result of stat()
932 Emit8(Out, Data.first? 1 : 0);
Mike Stump11289f42009-09-09 15:08:12 +0000933
Douglas Gregorc5046832009-04-27 18:38:38 +0000934 if (Data.first == 0) {
935 Emit32(Out, (uint32_t) Data.second.st_ino);
936 Emit32(Out, (uint32_t) Data.second.st_dev);
937 Emit16(Out, (uint16_t) Data.second.st_mode);
938 Emit64(Out, (uint64_t) Data.second.st_mtime);
939 Emit64(Out, (uint64_t) Data.second.st_size);
940 }
941
942 assert(Out.tell() - Start == DataLen && "Wrong data length");
943 }
944};
945} // end anonymous namespace
946
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000947/// \brief Write the stat() system call cache to the AST file.
Sebastian Redl55c0ad52010-08-18 23:56:21 +0000948void ASTWriter::WriteStatCache(MemorizeStatCalls &StatCalls) {
Douglas Gregorc5046832009-04-27 18:38:38 +0000949 // Build the on-disk hash table containing information about every
950 // stat() call.
Sebastian Redl42a0f6a2010-08-18 23:56:27 +0000951 OnDiskChainedHashTableGenerator<ASTStatCacheTrait> Generator;
Douglas Gregorc5046832009-04-27 18:38:38 +0000952 unsigned NumStatEntries = 0;
Mike Stump11289f42009-09-09 15:08:12 +0000953 for (MemorizeStatCalls::iterator Stat = StatCalls.begin(),
Douglas Gregorc5046832009-04-27 18:38:38 +0000954 StatEnd = StatCalls.end();
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000955 Stat != StatEnd; ++Stat, ++NumStatEntries) {
956 const char *Filename = Stat->first();
Douglas Gregor0086a5a2009-07-07 00:12:59 +0000957 Generator.insert(Filename, Stat->second);
958 }
Mike Stump11289f42009-09-09 15:08:12 +0000959
Douglas Gregorc5046832009-04-27 18:38:38 +0000960 // Create the on-disk hash table in a buffer.
Mike Stump11289f42009-09-09 15:08:12 +0000961 llvm::SmallString<4096> StatCacheData;
Douglas Gregorc5046832009-04-27 18:38:38 +0000962 uint32_t BucketOffset;
963 {
964 llvm::raw_svector_ostream Out(StatCacheData);
965 // Make sure that no bucket is at offset 0
966 clang::io::Emit32(Out, 0);
967 BucketOffset = Generator.Emit(Out);
968 }
969
970 // Create a blob abbreviation
971 using namespace llvm;
972 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +0000973 Abbrev->Add(BitCodeAbbrevOp(STAT_CACHE));
Douglas Gregorc5046832009-04-27 18:38:38 +0000974 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32));
975 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32));
976 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
977 unsigned StatCacheAbbrev = Stream.EmitAbbrev(Abbrev);
978
979 // Write the stat cache
980 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +0000981 Record.push_back(STAT_CACHE);
Douglas Gregorc5046832009-04-27 18:38:38 +0000982 Record.push_back(BucketOffset);
983 Record.push_back(NumStatEntries);
Daniel Dunbar8100d012009-08-24 09:31:37 +0000984 Stream.EmitRecordWithBlob(StatCacheAbbrev, Record, StatCacheData.str());
Douglas Gregorc5046832009-04-27 18:38:38 +0000985}
986
987//===----------------------------------------------------------------------===//
Douglas Gregora7f71a92009-04-10 03:52:48 +0000988// Source Manager Serialization
989//===----------------------------------------------------------------------===//
990
991/// \brief Create an abbreviation for the SLocEntry that refers to a
992/// file.
Douglas Gregor8f45df52009-04-16 22:23:12 +0000993static unsigned CreateSLocFileAbbrev(llvm::BitstreamWriter &Stream) {
Douglas Gregora7f71a92009-04-10 03:52:48 +0000994 using namespace llvm;
995 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +0000996 Abbrev->Add(BitCodeAbbrevOp(SM_SLOC_FILE_ENTRY));
Douglas Gregora7f71a92009-04-10 03:52:48 +0000997 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // Offset
998 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // Include location
999 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 2)); // Characteristic
1000 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // Line directives
Douglas Gregorb41ca8f2010-03-21 22:49:54 +00001001 // FileEntry fields.
1002 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 12)); // Size
1003 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 32)); // Modification time
Douglas Gregor5712ebc2010-03-16 16:35:32 +00001004 // HeaderFileInfo fields.
1005 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // isImport
1006 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 2)); // DirInfo
1007 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // NumIncludes
1008 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // ControllingMacro
Douglas Gregora7f71a92009-04-10 03:52:48 +00001009 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // File name
Douglas Gregor8f45df52009-04-16 22:23:12 +00001010 return Stream.EmitAbbrev(Abbrev);
Douglas Gregora7f71a92009-04-10 03:52:48 +00001011}
1012
1013/// \brief Create an abbreviation for the SLocEntry that refers to a
1014/// buffer.
Douglas Gregor8f45df52009-04-16 22:23:12 +00001015static unsigned CreateSLocBufferAbbrev(llvm::BitstreamWriter &Stream) {
Douglas Gregora7f71a92009-04-10 03:52:48 +00001016 using namespace llvm;
1017 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001018 Abbrev->Add(BitCodeAbbrevOp(SM_SLOC_BUFFER_ENTRY));
Douglas Gregora7f71a92009-04-10 03:52:48 +00001019 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // Offset
1020 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // Include location
1021 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 2)); // Characteristic
1022 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // Line directives
1023 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // Buffer name blob
Douglas Gregor8f45df52009-04-16 22:23:12 +00001024 return Stream.EmitAbbrev(Abbrev);
Douglas Gregora7f71a92009-04-10 03:52:48 +00001025}
1026
1027/// \brief Create an abbreviation for the SLocEntry that refers to a
1028/// buffer's blob.
Douglas Gregor8f45df52009-04-16 22:23:12 +00001029static unsigned CreateSLocBufferBlobAbbrev(llvm::BitstreamWriter &Stream) {
Douglas Gregora7f71a92009-04-10 03:52:48 +00001030 using namespace llvm;
1031 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001032 Abbrev->Add(BitCodeAbbrevOp(SM_SLOC_BUFFER_BLOB));
Douglas Gregora7f71a92009-04-10 03:52:48 +00001033 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // Blob
Douglas Gregor8f45df52009-04-16 22:23:12 +00001034 return Stream.EmitAbbrev(Abbrev);
Douglas Gregora7f71a92009-04-10 03:52:48 +00001035}
1036
1037/// \brief Create an abbreviation for the SLocEntry that refers to an
1038/// buffer.
Douglas Gregor8f45df52009-04-16 22:23:12 +00001039static unsigned CreateSLocInstantiationAbbrev(llvm::BitstreamWriter &Stream) {
Douglas Gregora7f71a92009-04-10 03:52:48 +00001040 using namespace llvm;
1041 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001042 Abbrev->Add(BitCodeAbbrevOp(SM_SLOC_INSTANTIATION_ENTRY));
Douglas Gregora7f71a92009-04-10 03:52:48 +00001043 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // Offset
1044 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // Spelling location
1045 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // Start location
1046 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // End location
Douglas Gregor83243272009-04-15 18:05:10 +00001047 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Token length
Douglas Gregor8f45df52009-04-16 22:23:12 +00001048 return Stream.EmitAbbrev(Abbrev);
Douglas Gregora7f71a92009-04-10 03:52:48 +00001049}
1050
1051/// \brief Writes the block containing the serialized form of the
1052/// source manager.
1053///
1054/// TODO: We should probably use an on-disk hash table (stored in a
1055/// blob), indexed based on the file name, so that we only create
1056/// entries for files that we actually need. In the common case (no
1057/// errors), we probably won't have to create file entries for any of
1058/// the files in the AST.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001059void ASTWriter::WriteSourceManagerBlock(SourceManager &SourceMgr,
Douglas Gregor0086a5a2009-07-07 00:12:59 +00001060 const Preprocessor &PP,
1061 const char *isysroot) {
Douglas Gregor258ae542009-04-27 06:38:32 +00001062 RecordData Record;
1063
Chris Lattner0910e3b2009-04-10 17:16:57 +00001064 // Enter the source manager block.
Sebastian Redl539c5062010-08-18 23:57:32 +00001065 Stream.EnterSubblock(SOURCE_MANAGER_BLOCK_ID, 3);
Douglas Gregora7f71a92009-04-10 03:52:48 +00001066
1067 // Abbreviations for the various kinds of source-location entries.
Chris Lattnerc4976c732009-04-27 19:03:22 +00001068 unsigned SLocFileAbbrv = CreateSLocFileAbbrev(Stream);
1069 unsigned SLocBufferAbbrv = CreateSLocBufferAbbrev(Stream);
1070 unsigned SLocBufferBlobAbbrv = CreateSLocBufferBlobAbbrev(Stream);
1071 unsigned SLocInstantiationAbbrv = CreateSLocInstantiationAbbrev(Stream);
Douglas Gregora7f71a92009-04-10 03:52:48 +00001072
Douglas Gregor4c7626e2009-04-13 16:31:14 +00001073 // Write the line table.
1074 if (SourceMgr.hasLineTable()) {
1075 LineTableInfo &LineTable = SourceMgr.getLineTable();
1076
1077 // Emit the file names
1078 Record.push_back(LineTable.getNumFilenames());
1079 for (unsigned I = 0, N = LineTable.getNumFilenames(); I != N; ++I) {
1080 // Emit the file name
1081 const char *Filename = LineTable.getFilename(I);
Douglas Gregor0086a5a2009-07-07 00:12:59 +00001082 Filename = adjustFilenameForRelocatablePCH(Filename, isysroot);
Douglas Gregor4c7626e2009-04-13 16:31:14 +00001083 unsigned FilenameLen = Filename? strlen(Filename) : 0;
1084 Record.push_back(FilenameLen);
1085 if (FilenameLen)
1086 Record.insert(Record.end(), Filename, Filename + FilenameLen);
1087 }
Mike Stump11289f42009-09-09 15:08:12 +00001088
Douglas Gregor4c7626e2009-04-13 16:31:14 +00001089 // Emit the line entries
1090 for (LineTableInfo::iterator L = LineTable.begin(), LEnd = LineTable.end();
1091 L != LEnd; ++L) {
1092 // Emit the file ID
1093 Record.push_back(L->first);
Mike Stump11289f42009-09-09 15:08:12 +00001094
Douglas Gregor4c7626e2009-04-13 16:31:14 +00001095 // Emit the line entries
1096 Record.push_back(L->second.size());
Mike Stump11289f42009-09-09 15:08:12 +00001097 for (std::vector<LineEntry>::iterator LE = L->second.begin(),
Douglas Gregor4c7626e2009-04-13 16:31:14 +00001098 LEEnd = L->second.end();
1099 LE != LEEnd; ++LE) {
1100 Record.push_back(LE->FileOffset);
1101 Record.push_back(LE->LineNo);
1102 Record.push_back(LE->FilenameID);
1103 Record.push_back((unsigned)LE->FileKind);
1104 Record.push_back(LE->IncludeOffset);
1105 }
Douglas Gregor4c7626e2009-04-13 16:31:14 +00001106 }
Sebastian Redl539c5062010-08-18 23:57:32 +00001107 Stream.EmitRecord(SM_LINE_TABLE, Record);
Douglas Gregor4c7626e2009-04-13 16:31:14 +00001108 }
1109
Douglas Gregor258ae542009-04-27 06:38:32 +00001110 // Write out the source location entry table. We skip the first
1111 // entry, which is always the same dummy entry.
Chris Lattner12d61d32009-04-27 19:01:47 +00001112 std::vector<uint32_t> SLocEntryOffsets;
Douglas Gregor258ae542009-04-27 06:38:32 +00001113 RecordData PreloadSLocs;
Sebastian Redl5c415f32010-07-22 17:01:13 +00001114 unsigned BaseSLocID = Chain ? Chain->getTotalNumSLocs() : 0;
1115 SLocEntryOffsets.reserve(SourceMgr.sloc_entry_size() - 1 - BaseSLocID);
1116 for (unsigned I = BaseSLocID + 1, N = SourceMgr.sloc_entry_size();
1117 I != N; ++I) {
Douglas Gregor8655e882009-10-16 22:46:09 +00001118 // Get this source location entry.
1119 const SrcMgr::SLocEntry *SLoc = &SourceMgr.getSLocEntry(I);
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +00001120
Douglas Gregor258ae542009-04-27 06:38:32 +00001121 // Record the offset of this source-location entry.
1122 SLocEntryOffsets.push_back(Stream.GetCurrentBitNo());
1123
1124 // Figure out which record code to use.
1125 unsigned Code;
1126 if (SLoc->isFile()) {
1127 if (SLoc->getFile().getContentCache()->Entry)
Sebastian Redl539c5062010-08-18 23:57:32 +00001128 Code = SM_SLOC_FILE_ENTRY;
Douglas Gregor258ae542009-04-27 06:38:32 +00001129 else
Sebastian Redl539c5062010-08-18 23:57:32 +00001130 Code = SM_SLOC_BUFFER_ENTRY;
Douglas Gregor258ae542009-04-27 06:38:32 +00001131 } else
Sebastian Redl539c5062010-08-18 23:57:32 +00001132 Code = SM_SLOC_INSTANTIATION_ENTRY;
Douglas Gregor258ae542009-04-27 06:38:32 +00001133 Record.clear();
1134 Record.push_back(Code);
1135
1136 Record.push_back(SLoc->getOffset());
1137 if (SLoc->isFile()) {
1138 const SrcMgr::FileInfo &File = SLoc->getFile();
1139 Record.push_back(File.getIncludeLoc().getRawEncoding());
1140 Record.push_back(File.getFileCharacteristic()); // FIXME: stable encoding
1141 Record.push_back(File.hasLineDirectives());
1142
1143 const SrcMgr::ContentCache *Content = File.getContentCache();
1144 if (Content->Entry) {
1145 // The source location entry is a file. The blob associated
1146 // with this entry is the file name.
Mike Stump11289f42009-09-09 15:08:12 +00001147
Douglas Gregorb41ca8f2010-03-21 22:49:54 +00001148 // Emit size/modification time for this file.
1149 Record.push_back(Content->Entry->getSize());
1150 Record.push_back(Content->Entry->getModificationTime());
1151
Douglas Gregor5712ebc2010-03-16 16:35:32 +00001152 // Emit header-search information associated with this file.
1153 HeaderFileInfo HFI;
1154 HeaderSearch &HS = PP.getHeaderSearchInfo();
1155 if (Content->Entry->getUID() < HS.header_file_size())
1156 HFI = HS.header_file_begin()[Content->Entry->getUID()];
1157 Record.push_back(HFI.isImport);
1158 Record.push_back(HFI.DirInfo);
1159 Record.push_back(HFI.NumIncludes);
1160 AddIdentifierRef(HFI.ControllingMacro, Record);
1161
Douglas Gregor0086a5a2009-07-07 00:12:59 +00001162 // Turn the file name into an absolute path, if it isn't already.
1163 const char *Filename = Content->Entry->getName();
1164 llvm::sys::Path FilePath(Filename, strlen(Filename));
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +00001165 FilePath.makeAbsolute();
Kovarththanan Rajaratnamd16d38c2010-03-14 07:15:57 +00001166 Filename = FilePath.c_str();
Mike Stump11289f42009-09-09 15:08:12 +00001167
Douglas Gregor0086a5a2009-07-07 00:12:59 +00001168 Filename = adjustFilenameForRelocatablePCH(Filename, isysroot);
Daniel Dunbar8100d012009-08-24 09:31:37 +00001169 Stream.EmitRecordWithBlob(SLocFileAbbrv, Record, Filename);
Douglas Gregor258ae542009-04-27 06:38:32 +00001170
1171 // FIXME: For now, preload all file source locations, so that
1172 // we get the appropriate File entries in the reader. This is
1173 // a temporary measure.
Sebastian Redl5c415f32010-07-22 17:01:13 +00001174 PreloadSLocs.push_back(BaseSLocID + SLocEntryOffsets.size());
Douglas Gregor258ae542009-04-27 06:38:32 +00001175 } else {
1176 // The source location entry is a buffer. The blob associated
1177 // with this entry contains the contents of the buffer.
1178
1179 // We add one to the size so that we capture the trailing NULL
1180 // that is required by llvm::MemoryBuffer::getMemBuffer (on
1181 // the reader side).
Douglas Gregor874cc622010-03-16 00:35:39 +00001182 const llvm::MemoryBuffer *Buffer
Chris Lattnerfb24a3a2010-04-20 20:35:58 +00001183 = Content->getBuffer(PP.getDiagnostics(), PP.getSourceManager());
Douglas Gregor258ae542009-04-27 06:38:32 +00001184 const char *Name = Buffer->getBufferIdentifier();
Daniel Dunbar8100d012009-08-24 09:31:37 +00001185 Stream.EmitRecordWithBlob(SLocBufferAbbrv, Record,
1186 llvm::StringRef(Name, strlen(Name) + 1));
Douglas Gregor258ae542009-04-27 06:38:32 +00001187 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +00001188 Record.push_back(SM_SLOC_BUFFER_BLOB);
Douglas Gregor258ae542009-04-27 06:38:32 +00001189 Stream.EmitRecordWithBlob(SLocBufferBlobAbbrv, Record,
Daniel Dunbar8100d012009-08-24 09:31:37 +00001190 llvm::StringRef(Buffer->getBufferStart(),
1191 Buffer->getBufferSize() + 1));
Douglas Gregor258ae542009-04-27 06:38:32 +00001192
1193 if (strcmp(Name, "<built-in>") == 0)
Sebastian Redl5c415f32010-07-22 17:01:13 +00001194 PreloadSLocs.push_back(BaseSLocID + SLocEntryOffsets.size());
Douglas Gregor258ae542009-04-27 06:38:32 +00001195 }
1196 } else {
1197 // The source location entry is an instantiation.
1198 const SrcMgr::InstantiationInfo &Inst = SLoc->getInstantiation();
1199 Record.push_back(Inst.getSpellingLoc().getRawEncoding());
1200 Record.push_back(Inst.getInstantiationLocStart().getRawEncoding());
1201 Record.push_back(Inst.getInstantiationLocEnd().getRawEncoding());
1202
1203 // Compute the token length for this macro expansion.
1204 unsigned NextOffset = SourceMgr.getNextOffset();
Douglas Gregor8655e882009-10-16 22:46:09 +00001205 if (I + 1 != N)
1206 NextOffset = SourceMgr.getSLocEntry(I + 1).getOffset();
Douglas Gregor258ae542009-04-27 06:38:32 +00001207 Record.push_back(NextOffset - SLoc->getOffset() - 1);
1208 Stream.EmitRecordWithAbbrev(SLocInstantiationAbbrv, Record);
1209 }
1210 }
1211
Douglas Gregor8f45df52009-04-16 22:23:12 +00001212 Stream.ExitBlock();
Douglas Gregor258ae542009-04-27 06:38:32 +00001213
1214 if (SLocEntryOffsets.empty())
1215 return;
1216
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001217 // Write the source-location offsets table into the AST block. This
Douglas Gregor258ae542009-04-27 06:38:32 +00001218 // table is used for lazily loading source-location information.
1219 using namespace llvm;
1220 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001221 Abbrev->Add(BitCodeAbbrevOp(SOURCE_LOCATION_OFFSETS));
Douglas Gregor258ae542009-04-27 06:38:32 +00001222 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 16)); // # of slocs
1223 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 16)); // next offset
1224 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // offsets
1225 unsigned SLocOffsetsAbbrev = Stream.EmitAbbrev(Abbrev);
Mike Stump11289f42009-09-09 15:08:12 +00001226
Douglas Gregor258ae542009-04-27 06:38:32 +00001227 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +00001228 Record.push_back(SOURCE_LOCATION_OFFSETS);
Douglas Gregor258ae542009-04-27 06:38:32 +00001229 Record.push_back(SLocEntryOffsets.size());
Sebastian Redlc1d035f2010-09-22 20:19:08 +00001230 unsigned BaseOffset = Chain ? Chain->getNextSLocOffset() : 0;
1231 Record.push_back(SourceMgr.getNextOffset() - BaseOffset);
Douglas Gregor258ae542009-04-27 06:38:32 +00001232 Stream.EmitRecordWithBlob(SLocOffsetsAbbrev, Record,
Sebastian Redl3df5a082010-07-30 17:03:48 +00001233 (const char *)data(SLocEntryOffsets),
Chris Lattner12d61d32009-04-27 19:01:47 +00001234 SLocEntryOffsets.size()*sizeof(SLocEntryOffsets[0]));
Douglas Gregor258ae542009-04-27 06:38:32 +00001235
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001236 // Write the source location entry preloads array, telling the AST
Douglas Gregor258ae542009-04-27 06:38:32 +00001237 // reader which source locations entries it should load eagerly.
Sebastian Redl539c5062010-08-18 23:57:32 +00001238 Stream.EmitRecord(SOURCE_LOCATION_PRELOADS, PreloadSLocs);
Douglas Gregora7f71a92009-04-10 03:52:48 +00001239}
1240
Douglas Gregorc5046832009-04-27 18:38:38 +00001241//===----------------------------------------------------------------------===//
1242// Preprocessor Serialization
1243//===----------------------------------------------------------------------===//
1244
Chris Lattnereeffaef2009-04-10 17:15:23 +00001245/// \brief Writes the block containing the serialized form of the
1246/// preprocessor.
1247///
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001248void ASTWriter::WritePreprocessor(const Preprocessor &PP) {
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001249 RecordData Record;
Chris Lattner0910e3b2009-04-10 17:16:57 +00001250
Chris Lattner0af3ba12009-04-13 01:29:17 +00001251 // If the preprocessor __COUNTER__ value has been bumped, remember it.
1252 if (PP.getCounterValue() != 0) {
1253 Record.push_back(PP.getCounterValue());
Sebastian Redl539c5062010-08-18 23:57:32 +00001254 Stream.EmitRecord(PP_COUNTER_VALUE, Record);
Chris Lattner0af3ba12009-04-13 01:29:17 +00001255 Record.clear();
Douglas Gregoreda6a892009-04-26 00:07:37 +00001256 }
1257
1258 // Enter the preprocessor block.
Douglas Gregor796d76a2010-10-20 22:00:55 +00001259 Stream.EnterSubblock(PREPROCESSOR_BLOCK_ID, 3);
Mike Stump11289f42009-09-09 15:08:12 +00001260
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001261 // If the AST file contains __DATE__ or __TIME__ emit a warning about this.
Douglas Gregoreda6a892009-04-26 00:07:37 +00001262 // FIXME: use diagnostics subsystem for localization etc.
1263 if (PP.SawDateOrTime())
1264 fprintf(stderr, "warning: precompiled header used __DATE__ or __TIME__.\n");
Mike Stump11289f42009-09-09 15:08:12 +00001265
Douglas Gregor796d76a2010-10-20 22:00:55 +00001266
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001267 // Loop over all the macro definitions that are live at the end of the file,
1268 // emitting each to the PP section.
Douglas Gregoraae92242010-03-19 21:51:54 +00001269 PreprocessingRecord *PPRec = PP.getPreprocessingRecord();
Douglas Gregor796d76a2010-10-20 22:00:55 +00001270 unsigned InclusionAbbrev = 0;
1271 if (PPRec) {
1272 using namespace llvm;
1273 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
1274 Abbrev->Add(BitCodeAbbrevOp(PP_INCLUSION_DIRECTIVE));
1275 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // index
1276 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // start location
1277 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // end location
1278 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // filename length
1279 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // in quotes
1280 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 2)); // kind
1281 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001282 InclusionAbbrev = Stream.EmitAbbrev(Abbrev);
Douglas Gregor796d76a2010-10-20 22:00:55 +00001283 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001284
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001285 for (Preprocessor::macro_iterator I = PP.macro_begin(), E = PP.macro_end();
1286 I != E; ++I) {
Chris Lattner34321bc2009-04-10 21:41:48 +00001287 // FIXME: This emits macros in hash table order, we should do it in a stable
1288 // order so that output is reproducible.
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001289 MacroInfo *MI = I->second;
1290
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001291 // Don't emit builtin macros like __LINE__ to the AST file unless they have
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001292 // been redefined by the header (in which case they are not isBuiltinMacro).
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001293 // Also skip macros from a AST file if we're chaining.
Douglas Gregoreb114da2010-10-01 01:03:07 +00001294
1295 // FIXME: There is a (probably minor) optimization we could do here, if
1296 // the macro comes from the original PCH but the identifier comes from a
1297 // chained PCH, by storing the offset into the original PCH rather than
1298 // writing the macro definition a second time.
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001299 if (MI->isBuiltinMacro() ||
Douglas Gregoreb114da2010-10-01 01:03:07 +00001300 (Chain && I->first->isFromAST() && MI->isFromAST()))
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001301 continue;
1302
Chris Lattnerc523d8e2009-04-11 21:15:38 +00001303 AddIdentifierRef(I->first, Record);
Douglas Gregorc3366a52009-04-21 23:56:24 +00001304 MacroOffsets[I->first] = Stream.GetCurrentBitNo();
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001305 Record.push_back(MI->getDefinitionLoc().getRawEncoding());
1306 Record.push_back(MI->isUsed());
Mike Stump11289f42009-09-09 15:08:12 +00001307
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001308 unsigned Code;
1309 if (MI->isObjectLike()) {
Sebastian Redl539c5062010-08-18 23:57:32 +00001310 Code = PP_MACRO_OBJECT_LIKE;
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001311 } else {
Sebastian Redl539c5062010-08-18 23:57:32 +00001312 Code = PP_MACRO_FUNCTION_LIKE;
Mike Stump11289f42009-09-09 15:08:12 +00001313
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001314 Record.push_back(MI->isC99Varargs());
1315 Record.push_back(MI->isGNUVarargs());
1316 Record.push_back(MI->getNumArgs());
1317 for (MacroInfo::arg_iterator I = MI->arg_begin(), E = MI->arg_end();
1318 I != E; ++I)
Chris Lattnerc523d8e2009-04-11 21:15:38 +00001319 AddIdentifierRef(*I, Record);
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001320 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001321
Douglas Gregoraae92242010-03-19 21:51:54 +00001322 // If we have a detailed preprocessing record, record the macro definition
1323 // ID that corresponds to this macro.
1324 if (PPRec)
1325 Record.push_back(getMacroDefinitionID(PPRec->findMacroDefinition(MI)));
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001326
Douglas Gregor8f45df52009-04-16 22:23:12 +00001327 Stream.EmitRecord(Code, Record);
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001328 Record.clear();
1329
Chris Lattner2199f5b2009-04-10 18:08:30 +00001330 // Emit the tokens array.
1331 for (unsigned TokNo = 0, e = MI->getNumTokens(); TokNo != e; ++TokNo) {
1332 // Note that we know that the preprocessor does not have any annotation
1333 // tokens in it because they are created by the parser, and thus can't be
1334 // in a macro definition.
1335 const Token &Tok = MI->getReplacementToken(TokNo);
Mike Stump11289f42009-09-09 15:08:12 +00001336
Chris Lattner2199f5b2009-04-10 18:08:30 +00001337 Record.push_back(Tok.getLocation().getRawEncoding());
1338 Record.push_back(Tok.getLength());
1339
Chris Lattner2199f5b2009-04-10 18:08:30 +00001340 // FIXME: When reading literal tokens, reconstruct the literal pointer if
1341 // it is needed.
Chris Lattnerc523d8e2009-04-11 21:15:38 +00001342 AddIdentifierRef(Tok.getIdentifierInfo(), Record);
Mike Stump11289f42009-09-09 15:08:12 +00001343
Chris Lattner2199f5b2009-04-10 18:08:30 +00001344 // FIXME: Should translate token kind to a stable encoding.
1345 Record.push_back(Tok.getKind());
1346 // FIXME: Should translate token flags to a stable encoding.
1347 Record.push_back(Tok.getFlags());
Mike Stump11289f42009-09-09 15:08:12 +00001348
Sebastian Redl539c5062010-08-18 23:57:32 +00001349 Stream.EmitRecord(PP_TOKEN, Record);
Chris Lattner2199f5b2009-04-10 18:08:30 +00001350 Record.clear();
1351 }
Douglas Gregorc3366a52009-04-21 23:56:24 +00001352 ++NumMacros;
Chris Lattnerbaa52f42009-04-10 18:00:12 +00001353 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001354
Douglas Gregoraae92242010-03-19 21:51:54 +00001355 // If the preprocessor has a preprocessing record, emit it.
1356 unsigned NumPreprocessingRecords = 0;
1357 if (PPRec) {
Sebastian Redl7abd8d52010-09-27 23:20:01 +00001358 unsigned IndexBase = Chain ? PPRec->getNumPreallocatedEntities() : 0;
Sebastian Redl9609b4f2010-09-27 22:18:47 +00001359 for (PreprocessingRecord::iterator E = PPRec->begin(Chain),
1360 EEnd = PPRec->end(Chain);
Douglas Gregoraae92242010-03-19 21:51:54 +00001361 E != EEnd; ++E) {
1362 Record.clear();
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001363
Douglas Gregoraae92242010-03-19 21:51:54 +00001364 if (MacroInstantiation *MI = dyn_cast<MacroInstantiation>(*E)) {
Sebastian Redl9609b4f2010-09-27 22:18:47 +00001365 Record.push_back(IndexBase + NumPreprocessingRecords++);
Douglas Gregoraae92242010-03-19 21:51:54 +00001366 AddSourceLocation(MI->getSourceRange().getBegin(), Record);
1367 AddSourceLocation(MI->getSourceRange().getEnd(), Record);
1368 AddIdentifierRef(MI->getName(), Record);
1369 Record.push_back(getMacroDefinitionID(MI->getDefinition()));
Sebastian Redl539c5062010-08-18 23:57:32 +00001370 Stream.EmitRecord(PP_MACRO_INSTANTIATION, Record);
Douglas Gregoraae92242010-03-19 21:51:54 +00001371 continue;
1372 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001373
Douglas Gregoraae92242010-03-19 21:51:54 +00001374 if (MacroDefinition *MD = dyn_cast<MacroDefinition>(*E)) {
1375 // Record this macro definition's location.
Sebastian Redl50e26582010-09-15 19:54:06 +00001376 MacroID ID = getMacroDefinitionID(MD);
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001377
Douglas Gregor91096292010-10-02 19:29:26 +00001378 // Don't write the macro definition if it is from another AST file.
1379 if (ID < FirstMacroID)
1380 continue;
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001381
Douglas Gregor91096292010-10-02 19:29:26 +00001382 unsigned Position = ID - FirstMacroID;
1383 if (Position != MacroDefinitionOffsets.size()) {
1384 if (Position > MacroDefinitionOffsets.size())
1385 MacroDefinitionOffsets.resize(Position + 1);
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001386
1387 MacroDefinitionOffsets[Position] = Stream.GetCurrentBitNo();
Douglas Gregoraae92242010-03-19 21:51:54 +00001388 } else
1389 MacroDefinitionOffsets.push_back(Stream.GetCurrentBitNo());
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001390
Sebastian Redl9609b4f2010-09-27 22:18:47 +00001391 Record.push_back(IndexBase + NumPreprocessingRecords++);
Douglas Gregoraae92242010-03-19 21:51:54 +00001392 Record.push_back(ID);
1393 AddSourceLocation(MD->getSourceRange().getBegin(), Record);
1394 AddSourceLocation(MD->getSourceRange().getEnd(), Record);
1395 AddIdentifierRef(MD->getName(), Record);
1396 AddSourceLocation(MD->getLocation(), Record);
Sebastian Redl539c5062010-08-18 23:57:32 +00001397 Stream.EmitRecord(PP_MACRO_DEFINITION, Record);
Douglas Gregoraae92242010-03-19 21:51:54 +00001398 continue;
1399 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001400
Douglas Gregor796d76a2010-10-20 22:00:55 +00001401 if (InclusionDirective *ID = dyn_cast<InclusionDirective>(*E)) {
1402 Record.push_back(PP_INCLUSION_DIRECTIVE);
1403 Record.push_back(IndexBase + NumPreprocessingRecords++);
1404 AddSourceLocation(ID->getSourceRange().getBegin(), Record);
1405 AddSourceLocation(ID->getSourceRange().getEnd(), Record);
1406 Record.push_back(ID->getFileName().size());
1407 Record.push_back(ID->wasInQuotes());
1408 Record.push_back(static_cast<unsigned>(ID->getKind()));
1409 llvm::SmallString<64> Buffer;
1410 Buffer += ID->getFileName();
1411 Buffer += ID->getFile()->getName();
1412 Stream.EmitRecordWithBlob(InclusionAbbrev, Record, Buffer);
1413 continue;
1414 }
Douglas Gregoraae92242010-03-19 21:51:54 +00001415 }
1416 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001417
Douglas Gregor8f45df52009-04-16 22:23:12 +00001418 Stream.ExitBlock();
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001419
Douglas Gregoraae92242010-03-19 21:51:54 +00001420 // Write the offsets table for the preprocessing record.
1421 if (NumPreprocessingRecords > 0) {
1422 // Write the offsets table for identifier IDs.
1423 using namespace llvm;
1424 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001425 Abbrev->Add(BitCodeAbbrevOp(MACRO_DEFINITION_OFFSETS));
Douglas Gregoraae92242010-03-19 21:51:54 +00001426 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // # of records
1427 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // # of macro defs
1428 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
1429 unsigned MacroDefOffsetAbbrev = Stream.EmitAbbrev(Abbrev);
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00001430
Douglas Gregoraae92242010-03-19 21:51:54 +00001431 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +00001432 Record.push_back(MACRO_DEFINITION_OFFSETS);
Douglas Gregoraae92242010-03-19 21:51:54 +00001433 Record.push_back(NumPreprocessingRecords);
1434 Record.push_back(MacroDefinitionOffsets.size());
1435 Stream.EmitRecordWithBlob(MacroDefOffsetAbbrev, Record,
Sebastian Redl3df5a082010-07-30 17:03:48 +00001436 (const char *)data(MacroDefinitionOffsets),
Douglas Gregoraae92242010-03-19 21:51:54 +00001437 MacroDefinitionOffsets.size() * sizeof(uint32_t));
1438 }
Chris Lattnereeffaef2009-04-10 17:15:23 +00001439}
1440
Douglas Gregorc5046832009-04-27 18:38:38 +00001441//===----------------------------------------------------------------------===//
1442// Type Serialization
1443//===----------------------------------------------------------------------===//
Chris Lattnereeffaef2009-04-10 17:15:23 +00001444
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001445/// \brief Write the representation of a type to the AST stream.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001446void ASTWriter::WriteType(QualType T) {
Argyrios Kyrtzidisa7fbbb02010-08-20 16:04:04 +00001447 TypeIdx &Idx = TypeIdxs[T];
Argyrios Kyrtzidisbb5c7eae2010-08-20 16:03:59 +00001448 if (Idx.getIndex() == 0) // we haven't seen this type before.
1449 Idx = TypeIdx(NextTypeID++);
Mike Stump11289f42009-09-09 15:08:12 +00001450
Douglas Gregor9b3932c2010-10-05 18:37:06 +00001451 assert(Idx.getIndex() >= FirstTypeID && "Re-writing a type from a prior AST");
Douglas Gregordc72caa2010-10-04 18:21:45 +00001452
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001453 // Record the offset for this type.
Argyrios Kyrtzidisbb5c7eae2010-08-20 16:03:59 +00001454 unsigned Index = Idx.getIndex() - FirstTypeID;
Sebastian Redl66c5eef2010-07-27 00:17:23 +00001455 if (TypeOffsets.size() == Index)
Douglas Gregor8f45df52009-04-16 22:23:12 +00001456 TypeOffsets.push_back(Stream.GetCurrentBitNo());
Sebastian Redl66c5eef2010-07-27 00:17:23 +00001457 else if (TypeOffsets.size() < Index) {
1458 TypeOffsets.resize(Index + 1);
1459 TypeOffsets[Index] = Stream.GetCurrentBitNo();
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001460 }
1461
1462 RecordData Record;
Mike Stump11289f42009-09-09 15:08:12 +00001463
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001464 // Emit the type's representation.
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001465 ASTTypeWriter W(*this, Record);
John McCall8ccfcb52009-09-24 19:53:00 +00001466
Douglas Gregor1b8fe5b72009-11-16 21:35:15 +00001467 if (T.hasLocalNonFastQualifiers()) {
1468 Qualifiers Qs = T.getLocalQualifiers();
1469 AddTypeRef(T.getLocalUnqualifiedType(), Record);
John McCall8ccfcb52009-09-24 19:53:00 +00001470 Record.push_back(Qs.getAsOpaqueValue());
Sebastian Redl539c5062010-08-18 23:57:32 +00001471 W.Code = TYPE_EXT_QUAL;
John McCall8ccfcb52009-09-24 19:53:00 +00001472 } else {
1473 switch (T->getTypeClass()) {
1474 // For all of the concrete, non-dependent types, call the
1475 // appropriate visitor function.
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001476#define TYPE(Class, Base) \
Mike Stump281d6d72010-01-20 02:03:14 +00001477 case Type::Class: W.Visit##Class##Type(cast<Class##Type>(T)); break;
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001478#define ABSTRACT_TYPE(Class, Base)
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001479#include "clang/AST/TypeNodes.def"
John McCall8ccfcb52009-09-24 19:53:00 +00001480 }
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001481 }
1482
1483 // Emit the serialized record.
Douglas Gregor8f45df52009-04-16 22:23:12 +00001484 Stream.EmitRecord(W.Code, Record);
Douglas Gregorfeb84b02009-04-14 21:18:50 +00001485
1486 // Flush any expressions that were written as part of this type.
Douglas Gregor8f45df52009-04-16 22:23:12 +00001487 FlushStmts();
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001488}
1489
Douglas Gregorc5046832009-04-27 18:38:38 +00001490//===----------------------------------------------------------------------===//
1491// Declaration Serialization
1492//===----------------------------------------------------------------------===//
1493
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001494/// \brief Write the block containing all of the declaration IDs
1495/// lexically declared within the given DeclContext.
1496///
1497/// \returns the offset of the DECL_CONTEXT_LEXICAL block within the
1498/// bistream, or 0 if no block was written.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001499uint64_t ASTWriter::WriteDeclContextLexicalBlock(ASTContext &Context,
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001500 DeclContext *DC) {
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00001501 if (DC->decls_empty())
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001502 return 0;
1503
Douglas Gregor8f45df52009-04-16 22:23:12 +00001504 uint64_t Offset = Stream.GetCurrentBitNo();
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001505 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +00001506 Record.push_back(DECL_CONTEXT_LEXICAL);
Argyrios Kyrtzidis0e88a562010-10-14 20:14:34 +00001507 llvm::SmallVector<KindDeclIDPair, 64> Decls;
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00001508 for (DeclContext::decl_iterator D = DC->decls_begin(), DEnd = DC->decls_end();
1509 D != DEnd; ++D)
Argyrios Kyrtzidis0e88a562010-10-14 20:14:34 +00001510 Decls.push_back(std::make_pair((*D)->getKind(), GetDeclRef(*D)));
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001511
Douglas Gregora57c3ab2009-04-22 22:34:57 +00001512 ++NumLexicalDeclContexts;
Sebastian Redl66c5eef2010-07-27 00:17:23 +00001513 Stream.EmitRecordWithBlob(DeclContextLexicalAbbrev, Record,
Argyrios Kyrtzidis0e88a562010-10-14 20:14:34 +00001514 reinterpret_cast<char*>(Decls.data()),
1515 Decls.size() * sizeof(KindDeclIDPair));
Douglas Gregoref84c4b2009-04-09 22:27:44 +00001516 return Offset;
1517}
1518
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001519void ASTWriter::WriteTypeDeclOffsets() {
Sebastian Redl1ea025b2010-07-16 16:36:56 +00001520 using namespace llvm;
1521 RecordData Record;
1522
1523 // Write the type offsets array
1524 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001525 Abbrev->Add(BitCodeAbbrevOp(TYPE_OFFSET));
Sebastian Redl1ea025b2010-07-16 16:36:56 +00001526 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // # of types
1527 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // types block
1528 unsigned TypeOffsetAbbrev = Stream.EmitAbbrev(Abbrev);
1529 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +00001530 Record.push_back(TYPE_OFFSET);
Sebastian Redl1ea025b2010-07-16 16:36:56 +00001531 Record.push_back(TypeOffsets.size());
1532 Stream.EmitRecordWithBlob(TypeOffsetAbbrev, Record,
Sebastian Redl3df5a082010-07-30 17:03:48 +00001533 (const char *)data(TypeOffsets),
Sebastian Redl1ea025b2010-07-16 16:36:56 +00001534 TypeOffsets.size() * sizeof(TypeOffsets[0]));
1535
1536 // Write the declaration offsets array
1537 Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001538 Abbrev->Add(BitCodeAbbrevOp(DECL_OFFSET));
Sebastian Redl1ea025b2010-07-16 16:36:56 +00001539 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // # of declarations
1540 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // declarations block
1541 unsigned DeclOffsetAbbrev = Stream.EmitAbbrev(Abbrev);
1542 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +00001543 Record.push_back(DECL_OFFSET);
Sebastian Redl1ea025b2010-07-16 16:36:56 +00001544 Record.push_back(DeclOffsets.size());
1545 Stream.EmitRecordWithBlob(DeclOffsetAbbrev, Record,
Sebastian Redl3df5a082010-07-30 17:03:48 +00001546 (const char *)data(DeclOffsets),
Sebastian Redl1ea025b2010-07-16 16:36:56 +00001547 DeclOffsets.size() * sizeof(DeclOffsets[0]));
1548}
1549
Douglas Gregorc5046832009-04-27 18:38:38 +00001550//===----------------------------------------------------------------------===//
1551// Global Method Pool and Selector Serialization
1552//===----------------------------------------------------------------------===//
1553
Douglas Gregore84a9da2009-04-20 20:36:09 +00001554namespace {
Douglas Gregorc78d3462009-04-24 21:10:55 +00001555// Trait used for the on-disk hash table used in the method pool.
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001556class ASTMethodPoolTrait {
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001557 ASTWriter &Writer;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001558
1559public:
1560 typedef Selector key_type;
1561 typedef key_type key_type_ref;
Mike Stump11289f42009-09-09 15:08:12 +00001562
Sebastian Redl834bb972010-08-04 17:20:04 +00001563 struct data_type {
Sebastian Redl539c5062010-08-18 23:57:32 +00001564 SelectorID ID;
Sebastian Redl834bb972010-08-04 17:20:04 +00001565 ObjCMethodList Instance, Factory;
1566 };
Douglas Gregorc78d3462009-04-24 21:10:55 +00001567 typedef const data_type& data_type_ref;
1568
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001569 explicit ASTMethodPoolTrait(ASTWriter &Writer) : Writer(Writer) { }
Mike Stump11289f42009-09-09 15:08:12 +00001570
Douglas Gregorc78d3462009-04-24 21:10:55 +00001571 static unsigned ComputeHash(Selector Sel) {
Argyrios Kyrtzidis4bd97102010-08-20 16:03:52 +00001572 return serialization::ComputeHash(Sel);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001573 }
Mike Stump11289f42009-09-09 15:08:12 +00001574
1575 std::pair<unsigned,unsigned>
Douglas Gregorc78d3462009-04-24 21:10:55 +00001576 EmitKeyDataLength(llvm::raw_ostream& Out, Selector Sel,
1577 data_type_ref Methods) {
1578 unsigned KeyLen = 2 + (Sel.getNumArgs()? Sel.getNumArgs() * 4 : 4);
1579 clang::io::Emit16(Out, KeyLen);
Sebastian Redl834bb972010-08-04 17:20:04 +00001580 unsigned DataLen = 4 + 2 + 2; // 2 bytes for each of the method counts
1581 for (const ObjCMethodList *Method = &Methods.Instance; Method;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001582 Method = Method->Next)
1583 if (Method->Method)
1584 DataLen += 4;
Sebastian Redl834bb972010-08-04 17:20:04 +00001585 for (const ObjCMethodList *Method = &Methods.Factory; Method;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001586 Method = Method->Next)
1587 if (Method->Method)
1588 DataLen += 4;
1589 clang::io::Emit16(Out, DataLen);
1590 return std::make_pair(KeyLen, DataLen);
1591 }
Mike Stump11289f42009-09-09 15:08:12 +00001592
Douglas Gregor95c13f52009-04-25 17:48:32 +00001593 void EmitKey(llvm::raw_ostream& Out, Selector Sel, unsigned) {
Mike Stump11289f42009-09-09 15:08:12 +00001594 uint64_t Start = Out.tell();
Douglas Gregor95c13f52009-04-25 17:48:32 +00001595 assert((Start >> 32) == 0 && "Selector key offset too large");
1596 Writer.SetSelectorOffset(Sel, Start);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001597 unsigned N = Sel.getNumArgs();
1598 clang::io::Emit16(Out, N);
1599 if (N == 0)
1600 N = 1;
1601 for (unsigned I = 0; I != N; ++I)
Mike Stump11289f42009-09-09 15:08:12 +00001602 clang::io::Emit32(Out,
Douglas Gregorc78d3462009-04-24 21:10:55 +00001603 Writer.getIdentifierRef(Sel.getIdentifierInfoForSlot(I)));
1604 }
Mike Stump11289f42009-09-09 15:08:12 +00001605
Douglas Gregorc78d3462009-04-24 21:10:55 +00001606 void EmitData(llvm::raw_ostream& Out, key_type_ref,
Douglas Gregor4647cfa2009-04-24 21:49:02 +00001607 data_type_ref Methods, unsigned DataLen) {
1608 uint64_t Start = Out.tell(); (void)Start;
Sebastian Redl834bb972010-08-04 17:20:04 +00001609 clang::io::Emit32(Out, Methods.ID);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001610 unsigned NumInstanceMethods = 0;
Sebastian Redl834bb972010-08-04 17:20:04 +00001611 for (const ObjCMethodList *Method = &Methods.Instance; Method;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001612 Method = Method->Next)
1613 if (Method->Method)
1614 ++NumInstanceMethods;
1615
1616 unsigned NumFactoryMethods = 0;
Sebastian Redl834bb972010-08-04 17:20:04 +00001617 for (const ObjCMethodList *Method = &Methods.Factory; Method;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001618 Method = Method->Next)
1619 if (Method->Method)
1620 ++NumFactoryMethods;
1621
1622 clang::io::Emit16(Out, NumInstanceMethods);
1623 clang::io::Emit16(Out, NumFactoryMethods);
Sebastian Redl834bb972010-08-04 17:20:04 +00001624 for (const ObjCMethodList *Method = &Methods.Instance; Method;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001625 Method = Method->Next)
1626 if (Method->Method)
1627 clang::io::Emit32(Out, Writer.getDeclID(Method->Method));
Sebastian Redl834bb972010-08-04 17:20:04 +00001628 for (const ObjCMethodList *Method = &Methods.Factory; Method;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001629 Method = Method->Next)
1630 if (Method->Method)
1631 clang::io::Emit32(Out, Writer.getDeclID(Method->Method));
Douglas Gregor4647cfa2009-04-24 21:49:02 +00001632
1633 assert(Out.tell() - Start == DataLen && "Data length is wrong");
Douglas Gregorc78d3462009-04-24 21:10:55 +00001634 }
1635};
1636} // end anonymous namespace
1637
Sebastian Redla19a67f2010-08-03 21:58:15 +00001638/// \brief Write ObjC data: selectors and the method pool.
Douglas Gregorc78d3462009-04-24 21:10:55 +00001639///
1640/// The method pool contains both instance and factory methods, stored
Sebastian Redla19a67f2010-08-03 21:58:15 +00001641/// in an on-disk hash table indexed by the selector. The hash table also
1642/// contains an empty entry for every other selector known to Sema.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001643void ASTWriter::WriteSelectors(Sema &SemaRef) {
Douglas Gregorc78d3462009-04-24 21:10:55 +00001644 using namespace llvm;
1645
Sebastian Redla19a67f2010-08-03 21:58:15 +00001646 // Do we have to do anything at all?
Sebastian Redl834bb972010-08-04 17:20:04 +00001647 if (SemaRef.MethodPool.empty() && SelectorIDs.empty())
Sebastian Redla19a67f2010-08-03 21:58:15 +00001648 return;
Sebastian Redld95a56e2010-08-04 18:21:41 +00001649 unsigned NumTableEntries = 0;
Sebastian Redla19a67f2010-08-03 21:58:15 +00001650 // Create and write out the blob that contains selectors and the method pool.
Douglas Gregorc78d3462009-04-24 21:10:55 +00001651 {
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001652 OnDiskChainedHashTableGenerator<ASTMethodPoolTrait> Generator;
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00001653 ASTMethodPoolTrait Trait(*this);
Mike Stump11289f42009-09-09 15:08:12 +00001654
Sebastian Redla19a67f2010-08-03 21:58:15 +00001655 // Create the on-disk hash table representation. We walk through every
1656 // selector we've seen and look it up in the method pool.
Sebastian Redld95a56e2010-08-04 18:21:41 +00001657 SelectorOffsets.resize(NextSelectorID - FirstSelectorID);
Sebastian Redl539c5062010-08-18 23:57:32 +00001658 for (llvm::DenseMap<Selector, SelectorID>::iterator
Sebastian Redl834bb972010-08-04 17:20:04 +00001659 I = SelectorIDs.begin(), E = SelectorIDs.end();
1660 I != E; ++I) {
1661 Selector S = I->first;
Sebastian Redla19a67f2010-08-03 21:58:15 +00001662 Sema::GlobalMethodPool::iterator F = SemaRef.MethodPool.find(S);
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001663 ASTMethodPoolTrait::data_type Data = {
Sebastian Redl834bb972010-08-04 17:20:04 +00001664 I->second,
1665 ObjCMethodList(),
1666 ObjCMethodList()
1667 };
1668 if (F != SemaRef.MethodPool.end()) {
1669 Data.Instance = F->second.first;
1670 Data.Factory = F->second.second;
1671 }
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001672 // Only write this selector if it's not in an existing AST or something
Sebastian Redld95a56e2010-08-04 18:21:41 +00001673 // changed.
1674 if (Chain && I->second < FirstSelectorID) {
1675 // Selector already exists. Did it change?
1676 bool changed = false;
1677 for (ObjCMethodList *M = &Data.Instance; !changed && M && M->Method;
1678 M = M->Next) {
1679 if (M->Method->getPCHLevel() == 0)
1680 changed = true;
1681 }
1682 for (ObjCMethodList *M = &Data.Factory; !changed && M && M->Method;
1683 M = M->Next) {
1684 if (M->Method->getPCHLevel() == 0)
1685 changed = true;
1686 }
1687 if (!changed)
1688 continue;
Sebastian Redl6e1a2a02010-08-04 21:22:45 +00001689 } else if (Data.Instance.Method || Data.Factory.Method) {
1690 // A new method pool entry.
1691 ++NumTableEntries;
Sebastian Redld95a56e2010-08-04 18:21:41 +00001692 }
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00001693 Generator.insert(S, Data, Trait);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001694 }
1695
Douglas Gregorc78d3462009-04-24 21:10:55 +00001696 // Create the on-disk hash table in a buffer.
Mike Stump11289f42009-09-09 15:08:12 +00001697 llvm::SmallString<4096> MethodPool;
Douglas Gregorc78d3462009-04-24 21:10:55 +00001698 uint32_t BucketOffset;
1699 {
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001700 ASTMethodPoolTrait Trait(*this);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001701 llvm::raw_svector_ostream Out(MethodPool);
1702 // Make sure that no bucket is at offset 0
Douglas Gregor4647cfa2009-04-24 21:49:02 +00001703 clang::io::Emit32(Out, 0);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001704 BucketOffset = Generator.Emit(Out, Trait);
1705 }
1706
1707 // Create a blob abbreviation
1708 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001709 Abbrev->Add(BitCodeAbbrevOp(METHOD_POOL));
Douglas Gregorc78d3462009-04-24 21:10:55 +00001710 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32));
Douglas Gregor95c13f52009-04-25 17:48:32 +00001711 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32));
Douglas Gregorc78d3462009-04-24 21:10:55 +00001712 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
1713 unsigned MethodPoolAbbrev = Stream.EmitAbbrev(Abbrev);
1714
Douglas Gregor95c13f52009-04-25 17:48:32 +00001715 // Write the method pool
Douglas Gregorc78d3462009-04-24 21:10:55 +00001716 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +00001717 Record.push_back(METHOD_POOL);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001718 Record.push_back(BucketOffset);
Sebastian Redld95a56e2010-08-04 18:21:41 +00001719 Record.push_back(NumTableEntries);
Daniel Dunbar8100d012009-08-24 09:31:37 +00001720 Stream.EmitRecordWithBlob(MethodPoolAbbrev, Record, MethodPool.str());
Douglas Gregor95c13f52009-04-25 17:48:32 +00001721
1722 // Create a blob abbreviation for the selector table offsets.
1723 Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001724 Abbrev->Add(BitCodeAbbrevOp(SELECTOR_OFFSETS));
Douglas Gregor95c13f52009-04-25 17:48:32 +00001725 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // index
1726 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
1727 unsigned SelectorOffsetAbbrev = Stream.EmitAbbrev(Abbrev);
1728
1729 // Write the selector offsets table.
1730 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +00001731 Record.push_back(SELECTOR_OFFSETS);
Douglas Gregor95c13f52009-04-25 17:48:32 +00001732 Record.push_back(SelectorOffsets.size());
1733 Stream.EmitRecordWithBlob(SelectorOffsetAbbrev, Record,
Sebastian Redl3df5a082010-07-30 17:03:48 +00001734 (const char *)data(SelectorOffsets),
Douglas Gregor95c13f52009-04-25 17:48:32 +00001735 SelectorOffsets.size() * 4);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001736 }
1737}
1738
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001739/// \brief Write the selectors referenced in @selector expression into AST file.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001740void ASTWriter::WriteReferencedSelectorsPool(Sema &SemaRef) {
Fariborz Jahanianc51609a2010-07-23 19:11:11 +00001741 using namespace llvm;
1742 if (SemaRef.ReferencedSelectors.empty())
1743 return;
Sebastian Redlada023c2010-08-04 20:40:17 +00001744
Fariborz Jahanianc51609a2010-07-23 19:11:11 +00001745 RecordData Record;
Sebastian Redlada023c2010-08-04 20:40:17 +00001746
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001747 // Note: this writes out all references even for a dependent AST. But it is
Sebastian Redl51c79d82010-08-04 22:21:29 +00001748 // very tricky to fix, and given that @selector shouldn't really appear in
1749 // headers, probably not worth it. It's not a correctness issue.
Fariborz Jahanianc51609a2010-07-23 19:11:11 +00001750 for (DenseMap<Selector, SourceLocation>::iterator S =
1751 SemaRef.ReferencedSelectors.begin(),
1752 E = SemaRef.ReferencedSelectors.end(); S != E; ++S) {
1753 Selector Sel = (*S).first;
1754 SourceLocation Loc = (*S).second;
1755 AddSelectorRef(Sel, Record);
1756 AddSourceLocation(Loc, Record);
1757 }
Sebastian Redl539c5062010-08-18 23:57:32 +00001758 Stream.EmitRecord(REFERENCED_SELECTOR_POOL, Record);
Fariborz Jahanianc51609a2010-07-23 19:11:11 +00001759}
1760
Douglas Gregorc5046832009-04-27 18:38:38 +00001761//===----------------------------------------------------------------------===//
1762// Identifier Table Serialization
1763//===----------------------------------------------------------------------===//
1764
Douglas Gregorc78d3462009-04-24 21:10:55 +00001765namespace {
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001766class ASTIdentifierTableTrait {
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001767 ASTWriter &Writer;
Douglas Gregorc3366a52009-04-21 23:56:24 +00001768 Preprocessor &PP;
Douglas Gregore84a9da2009-04-20 20:36:09 +00001769
Douglas Gregor1d583f22009-04-28 21:18:29 +00001770 /// \brief Determines whether this is an "interesting" identifier
1771 /// that needs a full IdentifierInfo structure written into the hash
1772 /// table.
1773 static bool isInterestingIdentifier(const IdentifierInfo *II) {
1774 return II->isPoisoned() ||
1775 II->isExtensionToken() ||
1776 II->hasMacroDefinition() ||
1777 II->getObjCOrBuiltinID() ||
1778 II->getFETokenInfo<void>();
1779 }
1780
Douglas Gregore84a9da2009-04-20 20:36:09 +00001781public:
1782 typedef const IdentifierInfo* key_type;
1783 typedef key_type key_type_ref;
Mike Stump11289f42009-09-09 15:08:12 +00001784
Sebastian Redl539c5062010-08-18 23:57:32 +00001785 typedef IdentID data_type;
Douglas Gregore84a9da2009-04-20 20:36:09 +00001786 typedef data_type data_type_ref;
Mike Stump11289f42009-09-09 15:08:12 +00001787
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001788 ASTIdentifierTableTrait(ASTWriter &Writer, Preprocessor &PP)
Douglas Gregorc3366a52009-04-21 23:56:24 +00001789 : Writer(Writer), PP(PP) { }
Douglas Gregore84a9da2009-04-20 20:36:09 +00001790
1791 static unsigned ComputeHash(const IdentifierInfo* II) {
Daniel Dunbarf8502d52009-10-17 23:52:28 +00001792 return llvm::HashString(II->getName());
Douglas Gregore84a9da2009-04-20 20:36:09 +00001793 }
Mike Stump11289f42009-09-09 15:08:12 +00001794
1795 std::pair<unsigned,unsigned>
1796 EmitKeyDataLength(llvm::raw_ostream& Out, const IdentifierInfo* II,
Sebastian Redl539c5062010-08-18 23:57:32 +00001797 IdentID ID) {
Daniel Dunbar2c422dc92009-10-18 20:26:12 +00001798 unsigned KeyLen = II->getLength() + 1;
Douglas Gregor1d583f22009-04-28 21:18:29 +00001799 unsigned DataLen = 4; // 4 bytes for the persistent ID << 1
1800 if (isInterestingIdentifier(II)) {
Douglas Gregorb9256522009-04-28 21:32:13 +00001801 DataLen += 2; // 2 bytes for builtin ID, flags
Mike Stump11289f42009-09-09 15:08:12 +00001802 if (II->hasMacroDefinition() &&
Douglas Gregor1d583f22009-04-28 21:18:29 +00001803 !PP.getMacroInfo(const_cast<IdentifierInfo *>(II))->isBuiltinMacro())
Douglas Gregorb9256522009-04-28 21:32:13 +00001804 DataLen += 4;
Douglas Gregor1d583f22009-04-28 21:18:29 +00001805 for (IdentifierResolver::iterator D = IdentifierResolver::begin(II),
1806 DEnd = IdentifierResolver::end();
1807 D != DEnd; ++D)
Sebastian Redl539c5062010-08-18 23:57:32 +00001808 DataLen += sizeof(DeclID);
Douglas Gregor1d583f22009-04-28 21:18:29 +00001809 }
Douglas Gregora868bbd2009-04-21 22:25:48 +00001810 clang::io::Emit16(Out, DataLen);
Douglas Gregorab4df582009-04-28 20:01:51 +00001811 // We emit the key length after the data length so that every
1812 // string is preceded by a 16-bit length. This matches the PTH
1813 // format for storing identifiers.
Douglas Gregor5287b4e2009-04-25 21:04:17 +00001814 clang::io::Emit16(Out, KeyLen);
Douglas Gregore84a9da2009-04-20 20:36:09 +00001815 return std::make_pair(KeyLen, DataLen);
1816 }
Mike Stump11289f42009-09-09 15:08:12 +00001817
1818 void EmitKey(llvm::raw_ostream& Out, const IdentifierInfo* II,
Douglas Gregore84a9da2009-04-20 20:36:09 +00001819 unsigned KeyLen) {
1820 // Record the location of the key data. This is used when generating
1821 // the mapping from persistent IDs to strings.
1822 Writer.SetIdentifierOffset(II, Out.tell());
Daniel Dunbar2c422dc92009-10-18 20:26:12 +00001823 Out.write(II->getNameStart(), KeyLen);
Douglas Gregore84a9da2009-04-20 20:36:09 +00001824 }
Mike Stump11289f42009-09-09 15:08:12 +00001825
1826 void EmitData(llvm::raw_ostream& Out, const IdentifierInfo* II,
Sebastian Redl539c5062010-08-18 23:57:32 +00001827 IdentID ID, unsigned) {
Douglas Gregor1d583f22009-04-28 21:18:29 +00001828 if (!isInterestingIdentifier(II)) {
1829 clang::io::Emit32(Out, ID << 1);
1830 return;
1831 }
Douglas Gregorb9256522009-04-28 21:32:13 +00001832
Douglas Gregor1d583f22009-04-28 21:18:29 +00001833 clang::io::Emit32(Out, (ID << 1) | 0x01);
Douglas Gregore84a9da2009-04-20 20:36:09 +00001834 uint32_t Bits = 0;
Mike Stump11289f42009-09-09 15:08:12 +00001835 bool hasMacroDefinition =
1836 II->hasMacroDefinition() &&
Douglas Gregorc3366a52009-04-21 23:56:24 +00001837 !PP.getMacroInfo(const_cast<IdentifierInfo *>(II))->isBuiltinMacro();
Douglas Gregorb9256522009-04-28 21:32:13 +00001838 Bits = (uint32_t)II->getObjCOrBuiltinID();
Daniel Dunbar91b640a2009-12-18 20:58:47 +00001839 Bits = (Bits << 1) | unsigned(hasMacroDefinition);
1840 Bits = (Bits << 1) | unsigned(II->isExtensionToken());
1841 Bits = (Bits << 1) | unsigned(II->isPoisoned());
Argyrios Kyrtzidis3084a612010-08-11 22:55:12 +00001842 Bits = (Bits << 1) | unsigned(II->hasRevertedTokenIDToIdentifier());
Daniel Dunbar91b640a2009-12-18 20:58:47 +00001843 Bits = (Bits << 1) | unsigned(II->isCPlusPlusOperatorKeyword());
Douglas Gregorb9256522009-04-28 21:32:13 +00001844 clang::io::Emit16(Out, Bits);
Douglas Gregore84a9da2009-04-20 20:36:09 +00001845
Douglas Gregorc3366a52009-04-21 23:56:24 +00001846 if (hasMacroDefinition)
Douglas Gregorb9256522009-04-28 21:32:13 +00001847 clang::io::Emit32(Out, Writer.getMacroOffset(II));
Douglas Gregorc3366a52009-04-21 23:56:24 +00001848
Douglas Gregora868bbd2009-04-21 22:25:48 +00001849 // Emit the declaration IDs in reverse order, because the
1850 // IdentifierResolver provides the declarations as they would be
1851 // visible (e.g., the function "stat" would come before the struct
1852 // "stat"), but IdentifierResolver::AddDeclToIdentifierChain()
1853 // adds declarations to the end of the list (so we need to see the
1854 // struct "status" before the function "status").
Sebastian Redlff4a2952010-07-23 23:49:55 +00001855 // Only emit declarations that aren't from a chained PCH, though.
Mike Stump11289f42009-09-09 15:08:12 +00001856 llvm::SmallVector<Decl *, 16> Decls(IdentifierResolver::begin(II),
Douglas Gregora868bbd2009-04-21 22:25:48 +00001857 IdentifierResolver::end());
1858 for (llvm::SmallVector<Decl *, 16>::reverse_iterator D = Decls.rbegin(),
1859 DEnd = Decls.rend();
Douglas Gregore84a9da2009-04-20 20:36:09 +00001860 D != DEnd; ++D)
Sebastian Redl78f51772010-08-02 18:30:12 +00001861 clang::io::Emit32(Out, Writer.getDeclID(*D));
Douglas Gregore84a9da2009-04-20 20:36:09 +00001862 }
1863};
1864} // end anonymous namespace
1865
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001866/// \brief Write the identifier table into the AST file.
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001867///
1868/// The identifier table consists of a blob containing string data
1869/// (the actual identifiers themselves) and a separate "offsets" index
1870/// that maps identifier IDs to locations within the blob.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00001871void ASTWriter::WriteIdentifierTable(Preprocessor &PP) {
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001872 using namespace llvm;
1873
1874 // Create and write out the blob that contains the identifier
1875 // strings.
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001876 {
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001877 OnDiskChainedHashTableGenerator<ASTIdentifierTableTrait> Generator;
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00001878 ASTIdentifierTableTrait Trait(*this, PP);
Mike Stump11289f42009-09-09 15:08:12 +00001879
Douglas Gregore6648fb2009-04-28 20:33:11 +00001880 // Look for any identifiers that were named while processing the
1881 // headers, but are otherwise not needed. We add these to the hash
1882 // table to enable checking of the predefines buffer in the case
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001883 // where the user adds new macro definitions when building the AST
Douglas Gregore6648fb2009-04-28 20:33:11 +00001884 // file.
1885 for (IdentifierTable::iterator ID = PP.getIdentifierTable().begin(),
1886 IDEnd = PP.getIdentifierTable().end();
1887 ID != IDEnd; ++ID)
1888 getIdentifierRef(ID->second);
1889
Sebastian Redlff4a2952010-07-23 23:49:55 +00001890 // Create the on-disk hash table representation. We only store offsets
1891 // for identifiers that appear here for the first time.
1892 IdentifierOffsets.resize(NextIdentID - FirstIdentID);
Sebastian Redl539c5062010-08-18 23:57:32 +00001893 for (llvm::DenseMap<const IdentifierInfo *, IdentID>::iterator
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001894 ID = IdentifierIDs.begin(), IDEnd = IdentifierIDs.end();
1895 ID != IDEnd; ++ID) {
1896 assert(ID->first && "NULL identifier in identifier table");
Sebastian Redld44cd6a2010-08-18 23:57:06 +00001897 if (!Chain || !ID->first->isFromAST())
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00001898 Generator.insert(ID->first, ID->second, Trait);
Douglas Gregore84a9da2009-04-20 20:36:09 +00001899 }
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001900
Douglas Gregore84a9da2009-04-20 20:36:09 +00001901 // Create the on-disk hash table in a buffer.
Mike Stump11289f42009-09-09 15:08:12 +00001902 llvm::SmallString<4096> IdentifierTable;
Douglas Gregora868bbd2009-04-21 22:25:48 +00001903 uint32_t BucketOffset;
Douglas Gregore84a9da2009-04-20 20:36:09 +00001904 {
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00001905 ASTIdentifierTableTrait Trait(*this, PP);
Douglas Gregore84a9da2009-04-20 20:36:09 +00001906 llvm::raw_svector_ostream Out(IdentifierTable);
Douglas Gregorc78d3462009-04-24 21:10:55 +00001907 // Make sure that no bucket is at offset 0
Douglas Gregor4647cfa2009-04-24 21:49:02 +00001908 clang::io::Emit32(Out, 0);
Douglas Gregora868bbd2009-04-21 22:25:48 +00001909 BucketOffset = Generator.Emit(Out, Trait);
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001910 }
1911
1912 // Create a blob abbreviation
1913 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001914 Abbrev->Add(BitCodeAbbrevOp(IDENTIFIER_TABLE));
Douglas Gregora868bbd2009-04-21 22:25:48 +00001915 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32));
Douglas Gregore84a9da2009-04-20 20:36:09 +00001916 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
Douglas Gregor8f45df52009-04-16 22:23:12 +00001917 unsigned IDTableAbbrev = Stream.EmitAbbrev(Abbrev);
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001918
1919 // Write the identifier table
1920 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +00001921 Record.push_back(IDENTIFIER_TABLE);
Douglas Gregora868bbd2009-04-21 22:25:48 +00001922 Record.push_back(BucketOffset);
Daniel Dunbar8100d012009-08-24 09:31:37 +00001923 Stream.EmitRecordWithBlob(IDTableAbbrev, Record, IdentifierTable.str());
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001924 }
1925
1926 // Write the offsets table for identifier IDs.
Douglas Gregor0e149972009-04-25 19:10:14 +00001927 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00001928 Abbrev->Add(BitCodeAbbrevOp(IDENTIFIER_OFFSET));
Douglas Gregor0e149972009-04-25 19:10:14 +00001929 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // # of identifiers
1930 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
1931 unsigned IdentifierOffsetAbbrev = Stream.EmitAbbrev(Abbrev);
1932
1933 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +00001934 Record.push_back(IDENTIFIER_OFFSET);
Douglas Gregor0e149972009-04-25 19:10:14 +00001935 Record.push_back(IdentifierOffsets.size());
1936 Stream.EmitRecordWithBlob(IdentifierOffsetAbbrev, Record,
Sebastian Redl3df5a082010-07-30 17:03:48 +00001937 (const char *)data(IdentifierOffsets),
Douglas Gregor0e149972009-04-25 19:10:14 +00001938 IdentifierOffsets.size() * sizeof(uint32_t));
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00001939}
1940
Douglas Gregorc5046832009-04-27 18:38:38 +00001941//===----------------------------------------------------------------------===//
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00001942// DeclContext's Name Lookup Table Serialization
1943//===----------------------------------------------------------------------===//
1944
1945namespace {
1946// Trait used for the on-disk hash table used in the method pool.
1947class ASTDeclContextNameLookupTrait {
1948 ASTWriter &Writer;
1949
1950public:
1951 typedef DeclarationName key_type;
1952 typedef key_type key_type_ref;
1953
1954 typedef DeclContext::lookup_result data_type;
1955 typedef const data_type& data_type_ref;
1956
1957 explicit ASTDeclContextNameLookupTrait(ASTWriter &Writer) : Writer(Writer) { }
1958
1959 unsigned ComputeHash(DeclarationName Name) {
1960 llvm::FoldingSetNodeID ID;
1961 ID.AddInteger(Name.getNameKind());
1962
1963 switch (Name.getNameKind()) {
1964 case DeclarationName::Identifier:
1965 ID.AddString(Name.getAsIdentifierInfo()->getName());
1966 break;
1967 case DeclarationName::ObjCZeroArgSelector:
1968 case DeclarationName::ObjCOneArgSelector:
1969 case DeclarationName::ObjCMultiArgSelector:
1970 ID.AddInteger(serialization::ComputeHash(Name.getObjCSelector()));
1971 break;
1972 case DeclarationName::CXXConstructorName:
1973 case DeclarationName::CXXDestructorName:
1974 case DeclarationName::CXXConversionFunctionName:
1975 ID.AddInteger(Writer.GetOrCreateTypeID(Name.getCXXNameType()));
1976 break;
1977 case DeclarationName::CXXOperatorName:
1978 ID.AddInteger(Name.getCXXOverloadedOperator());
1979 break;
1980 case DeclarationName::CXXLiteralOperatorName:
1981 ID.AddString(Name.getCXXLiteralIdentifier()->getName());
1982 case DeclarationName::CXXUsingDirective:
1983 break;
1984 }
1985
1986 return ID.ComputeHash();
1987 }
1988
1989 std::pair<unsigned,unsigned>
1990 EmitKeyDataLength(llvm::raw_ostream& Out, DeclarationName Name,
1991 data_type_ref Lookup) {
1992 unsigned KeyLen = 1;
1993 switch (Name.getNameKind()) {
1994 case DeclarationName::Identifier:
1995 case DeclarationName::ObjCZeroArgSelector:
1996 case DeclarationName::ObjCOneArgSelector:
1997 case DeclarationName::ObjCMultiArgSelector:
1998 case DeclarationName::CXXConstructorName:
1999 case DeclarationName::CXXDestructorName:
2000 case DeclarationName::CXXConversionFunctionName:
2001 case DeclarationName::CXXLiteralOperatorName:
2002 KeyLen += 4;
2003 break;
2004 case DeclarationName::CXXOperatorName:
2005 KeyLen += 1;
2006 break;
2007 case DeclarationName::CXXUsingDirective:
2008 break;
2009 }
2010 clang::io::Emit16(Out, KeyLen);
2011
2012 // 2 bytes for num of decls and 4 for each DeclID.
2013 unsigned DataLen = 2 + 4 * (Lookup.second - Lookup.first);
2014 clang::io::Emit16(Out, DataLen);
2015
2016 return std::make_pair(KeyLen, DataLen);
2017 }
2018
2019 void EmitKey(llvm::raw_ostream& Out, DeclarationName Name, unsigned) {
2020 using namespace clang::io;
2021
2022 assert(Name.getNameKind() < 0x100 && "Invalid name kind ?");
2023 Emit8(Out, Name.getNameKind());
2024 switch (Name.getNameKind()) {
2025 case DeclarationName::Identifier:
2026 Emit32(Out, Writer.getIdentifierRef(Name.getAsIdentifierInfo()));
2027 break;
2028 case DeclarationName::ObjCZeroArgSelector:
2029 case DeclarationName::ObjCOneArgSelector:
2030 case DeclarationName::ObjCMultiArgSelector:
2031 Emit32(Out, Writer.getSelectorRef(Name.getObjCSelector()));
2032 break;
2033 case DeclarationName::CXXConstructorName:
2034 case DeclarationName::CXXDestructorName:
2035 case DeclarationName::CXXConversionFunctionName:
2036 Emit32(Out, Writer.getTypeID(Name.getCXXNameType()));
2037 break;
2038 case DeclarationName::CXXOperatorName:
2039 assert(Name.getCXXOverloadedOperator() < 0x100 && "Invalid operator ?");
2040 Emit8(Out, Name.getCXXOverloadedOperator());
2041 break;
2042 case DeclarationName::CXXLiteralOperatorName:
2043 Emit32(Out, Writer.getIdentifierRef(Name.getCXXLiteralIdentifier()));
2044 break;
2045 case DeclarationName::CXXUsingDirective:
2046 break;
2047 }
2048 }
2049
2050 void EmitData(llvm::raw_ostream& Out, key_type_ref,
2051 data_type Lookup, unsigned DataLen) {
2052 uint64_t Start = Out.tell(); (void)Start;
2053 clang::io::Emit16(Out, Lookup.second - Lookup.first);
2054 for (; Lookup.first != Lookup.second; ++Lookup.first)
2055 clang::io::Emit32(Out, Writer.GetDeclRef(*Lookup.first));
2056
2057 assert(Out.tell() - Start == DataLen && "Data length is wrong");
2058 }
2059};
2060} // end anonymous namespace
2061
Argyrios Kyrtzidisba88bfa2010-08-20 16:04:35 +00002062/// \brief Write the block containing all of the declaration IDs
2063/// visible from the given DeclContext.
2064///
2065/// \returns the offset of the DECL_CONTEXT_VISIBLE block within the
Sebastian Redla4071b42010-08-24 00:50:09 +00002066/// bitstream, or 0 if no block was written.
Argyrios Kyrtzidisba88bfa2010-08-20 16:04:35 +00002067uint64_t ASTWriter::WriteDeclContextVisibleBlock(ASTContext &Context,
2068 DeclContext *DC) {
2069 if (DC->getPrimaryContext() != DC)
2070 return 0;
2071
2072 // Since there is no name lookup into functions or methods, don't bother to
2073 // build a visible-declarations table for these entities.
2074 if (DC->isFunctionOrMethod())
2075 return 0;
2076
2077 // If not in C++, we perform name lookup for the translation unit via the
2078 // IdentifierInfo chains, don't bother to build a visible-declarations table.
2079 // FIXME: In C++ we need the visible declarations in order to "see" the
2080 // friend declarations, is there a way to do this without writing the table ?
2081 if (DC->isTranslationUnit() && !Context.getLangOptions().CPlusPlus)
2082 return 0;
2083
2084 // Force the DeclContext to build a its name-lookup table.
Argyrios Kyrtzidisd32ee892010-08-20 23:35:55 +00002085 if (DC->hasExternalVisibleStorage())
2086 DC->MaterializeVisibleDeclsFromExternalStorage();
2087 else
2088 DC->lookup(DeclarationName());
Argyrios Kyrtzidisba88bfa2010-08-20 16:04:35 +00002089
2090 // Serialize the contents of the mapping used for lookup. Note that,
2091 // although we have two very different code paths, the serialized
2092 // representation is the same for both cases: a declaration name,
2093 // followed by a size, followed by references to the visible
2094 // declarations that have that name.
2095 uint64_t Offset = Stream.GetCurrentBitNo();
2096 StoredDeclsMap *Map = static_cast<StoredDeclsMap*>(DC->getLookupPtr());
2097 if (!Map || Map->empty())
2098 return 0;
2099
2100 OnDiskChainedHashTableGenerator<ASTDeclContextNameLookupTrait> Generator;
2101 ASTDeclContextNameLookupTrait Trait(*this);
2102
2103 // Create the on-disk hash table representation.
2104 for (StoredDeclsMap::iterator D = Map->begin(), DEnd = Map->end();
2105 D != DEnd; ++D) {
2106 DeclarationName Name = D->first;
2107 DeclContext::lookup_result Result = D->second.getLookupResult();
2108 Generator.insert(Name, Result, Trait);
2109 }
2110
2111 // Create the on-disk hash table in a buffer.
2112 llvm::SmallString<4096> LookupTable;
2113 uint32_t BucketOffset;
2114 {
2115 llvm::raw_svector_ostream Out(LookupTable);
2116 // Make sure that no bucket is at offset 0
2117 clang::io::Emit32(Out, 0);
2118 BucketOffset = Generator.Emit(Out, Trait);
2119 }
2120
2121 // Write the lookup table
2122 RecordData Record;
2123 Record.push_back(DECL_CONTEXT_VISIBLE);
2124 Record.push_back(BucketOffset);
2125 Stream.EmitRecordWithBlob(DeclContextVisibleLookupAbbrev, Record,
2126 LookupTable.str());
2127
2128 Stream.EmitRecord(DECL_CONTEXT_VISIBLE, Record);
2129 ++NumVisibleDeclContexts;
2130 return Offset;
2131}
2132
Sebastian Redla4071b42010-08-24 00:50:09 +00002133/// \brief Write an UPDATE_VISIBLE block for the given context.
2134///
2135/// UPDATE_VISIBLE blocks contain the declarations that are added to an existing
2136/// DeclContext in a dependent AST file. As such, they only exist for the TU
2137/// (in C++) and for namespaces.
2138void ASTWriter::WriteDeclContextVisibleUpdate(const DeclContext *DC) {
2139 assert((DC->isTranslationUnit() || DC->isNamespace()) &&
2140 "Only TU and namespaces should have visible decl updates.");
2141
2142 // Make the context build its lookup table, but don't make it load external
2143 // decls.
2144 DC->lookup(DeclarationName());
2145
2146 StoredDeclsMap *Map = static_cast<StoredDeclsMap*>(DC->getLookupPtr());
2147 if (!Map || Map->empty())
2148 return;
2149
2150 OnDiskChainedHashTableGenerator<ASTDeclContextNameLookupTrait> Generator;
2151 ASTDeclContextNameLookupTrait Trait(*this);
2152
2153 // Create the hash table.
Sebastian Redla4071b42010-08-24 00:50:09 +00002154 for (StoredDeclsMap::iterator D = Map->begin(), DEnd = Map->end();
2155 D != DEnd; ++D) {
2156 DeclarationName Name = D->first;
2157 DeclContext::lookup_result Result = D->second.getLookupResult();
Sebastian Redl9617e7e2010-08-24 00:50:16 +00002158 // For any name that appears in this table, the results are complete, i.e.
2159 // they overwrite results from previous PCHs. Merging is always a mess.
2160 Generator.insert(Name, Result, Trait);
Sebastian Redla4071b42010-08-24 00:50:09 +00002161 }
2162
2163 // Create the on-disk hash table in a buffer.
2164 llvm::SmallString<4096> LookupTable;
2165 uint32_t BucketOffset;
2166 {
2167 llvm::raw_svector_ostream Out(LookupTable);
2168 // Make sure that no bucket is at offset 0
2169 clang::io::Emit32(Out, 0);
2170 BucketOffset = Generator.Emit(Out, Trait);
2171 }
2172
2173 // Write the lookup table
2174 RecordData Record;
2175 Record.push_back(UPDATE_VISIBLE);
2176 Record.push_back(getDeclID(cast<Decl>(DC)));
2177 Record.push_back(BucketOffset);
2178 Stream.EmitRecordWithBlob(UpdateVisibleAbbrev, Record, LookupTable.str());
2179}
2180
Sebastian Redl401b39a2010-08-24 22:50:24 +00002181/// \brief Write ADDITIONAL_TEMPLATE_SPECIALIZATIONS blocks for all templates
2182/// that have new specializations in the current AST file.
2183void ASTWriter::WriteAdditionalTemplateSpecializations() {
2184 RecordData Record;
2185 for (AdditionalTemplateSpecializationsMap::iterator
2186 I = AdditionalTemplateSpecializations.begin(),
2187 E = AdditionalTemplateSpecializations.end();
2188 I != E; ++I) {
2189 Record.clear();
2190 Record.push_back(I->first);
2191 Record.insert(Record.end(), I->second.begin(), I->second.end());
2192 Stream.EmitRecord(ADDITIONAL_TEMPLATE_SPECIALIZATIONS, Record);
2193 }
2194}
2195
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00002196//===----------------------------------------------------------------------===//
Douglas Gregorc5046832009-04-27 18:38:38 +00002197// General Serialization Routines
2198//===----------------------------------------------------------------------===//
2199
Douglas Gregorbc8a78d52009-04-15 21:30:51 +00002200/// \brief Write a record containing the given attributes.
Argyrios Kyrtzidis9beef8e2010-10-18 19:20:11 +00002201void ASTWriter::WriteAttributes(const AttrVec &Attrs, RecordData &Record) {
2202 Record.push_back(Attrs.size());
Alexis Huntdcfba7b2010-08-18 23:23:40 +00002203 for (AttrVec::const_iterator i = Attrs.begin(), e = Attrs.end(); i != e; ++i){
2204 const Attr * A = *i;
2205 Record.push_back(A->getKind()); // FIXME: stable encoding, target attrs
2206 AddSourceLocation(A->getLocation(), Record);
2207 Record.push_back(A->isInherited());
Douglas Gregorbc8a78d52009-04-15 21:30:51 +00002208
Alexis Huntdcfba7b2010-08-18 23:23:40 +00002209#include "clang/Serialization/AttrPCHWrite.inc"
Daniel Dunbarfc6507e2010-05-27 02:25:39 +00002210
Douglas Gregorbc8a78d52009-04-15 21:30:51 +00002211 }
Douglas Gregorbc8a78d52009-04-15 21:30:51 +00002212}
2213
Benjamin Kramercd495032010-09-02 15:06:24 +00002214void ASTWriter::AddString(llvm::StringRef Str, RecordData &Record) {
Douglas Gregorbc8a78d52009-04-15 21:30:51 +00002215 Record.push_back(Str.size());
2216 Record.insert(Record.end(), Str.begin(), Str.end());
2217}
2218
Douglas Gregore84a9da2009-04-20 20:36:09 +00002219/// \brief Note that the identifier II occurs at the given offset
2220/// within the identifier table.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002221void ASTWriter::SetIdentifierOffset(const IdentifierInfo *II, uint32_t Offset) {
Sebastian Redl539c5062010-08-18 23:57:32 +00002222 IdentID ID = IdentifierIDs[II];
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00002223 // Only store offsets new to this AST file. Other identifier names are looked
Sebastian Redlff4a2952010-07-23 23:49:55 +00002224 // up earlier in the chain and thus don't need an offset.
2225 if (ID >= FirstIdentID)
2226 IdentifierOffsets[ID - FirstIdentID] = Offset;
Douglas Gregore84a9da2009-04-20 20:36:09 +00002227}
2228
Douglas Gregor95c13f52009-04-25 17:48:32 +00002229/// \brief Note that the selector Sel occurs at the given offset
2230/// within the method pool/selector table.
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002231void ASTWriter::SetSelectorOffset(Selector Sel, uint32_t Offset) {
Douglas Gregor95c13f52009-04-25 17:48:32 +00002232 unsigned ID = SelectorIDs[Sel];
2233 assert(ID && "Unknown selector");
Sebastian Redld95a56e2010-08-04 18:21:41 +00002234 // Don't record offsets for selectors that are also available in a different
2235 // file.
2236 if (ID < FirstSelectorID)
2237 return;
2238 SelectorOffsets[ID - FirstSelectorID] = Offset;
Douglas Gregor95c13f52009-04-25 17:48:32 +00002239}
2240
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002241ASTWriter::ASTWriter(llvm::BitstreamWriter &Stream)
Sebastian Redld95a56e2010-08-04 18:21:41 +00002242 : Stream(Stream), Chain(0), FirstDeclID(1), NextDeclID(FirstDeclID),
Sebastian Redl539c5062010-08-18 23:57:32 +00002243 FirstTypeID(NUM_PREDEF_TYPE_IDS), NextTypeID(FirstTypeID),
Sebastian Redld95a56e2010-08-04 18:21:41 +00002244 FirstIdentID(1), NextIdentID(FirstIdentID), FirstSelectorID(1),
Douglas Gregor91096292010-10-02 19:29:26 +00002245 NextSelectorID(FirstSelectorID), FirstMacroID(1), NextMacroID(FirstMacroID),
2246 CollectedStmts(&StmtsToEmit),
Sebastian Redld95a56e2010-08-04 18:21:41 +00002247 NumStatements(0), NumMacros(0), NumLexicalDeclContexts(0),
2248 NumVisibleDeclContexts(0) {
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00002249}
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002250
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002251void ASTWriter::WriteAST(Sema &SemaRef, MemorizeStatCalls *StatCalls,
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00002252 const char *isysroot) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002253 // Emit the file header.
Douglas Gregor8f45df52009-04-16 22:23:12 +00002254 Stream.Emit((unsigned)'C', 8);
2255 Stream.Emit((unsigned)'P', 8);
2256 Stream.Emit((unsigned)'C', 8);
2257 Stream.Emit((unsigned)'H', 8);
Mike Stump11289f42009-09-09 15:08:12 +00002258
Chris Lattner28fa4e62009-04-26 22:26:21 +00002259 WriteBlockInfoBlock();
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002260
Sebastian Redl143413f2010-07-12 22:02:52 +00002261 if (Chain)
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002262 WriteASTChain(SemaRef, StatCalls, isysroot);
Sebastian Redl143413f2010-07-12 22:02:52 +00002263 else
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002264 WriteASTCore(SemaRef, StatCalls, isysroot);
Sebastian Redl143413f2010-07-12 22:02:52 +00002265}
2266
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002267void ASTWriter::WriteASTCore(Sema &SemaRef, MemorizeStatCalls *StatCalls,
Sebastian Redl143413f2010-07-12 22:02:52 +00002268 const char *isysroot) {
2269 using namespace llvm;
2270
2271 ASTContext &Context = SemaRef.Context;
2272 Preprocessor &PP = SemaRef.PP;
2273
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002274 // The translation unit is the first declaration we'll emit.
2275 DeclIDs[Context.getTranslationUnitDecl()] = 1;
Sebastian Redlff4a2952010-07-23 23:49:55 +00002276 ++NextDeclID;
Douglas Gregor12bfa382009-10-17 00:13:19 +00002277 DeclTypesToEmit.push(Context.getTranslationUnitDecl());
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002278
Douglas Gregor4621c6a2009-04-22 18:49:13 +00002279 // Make sure that we emit IdentifierInfos (and any attached
2280 // declarations) for builtins.
2281 {
2282 IdentifierTable &Table = PP.getIdentifierTable();
2283 llvm::SmallVector<const char *, 32> BuiltinNames;
2284 Context.BuiltinInfo.GetBuiltinNames(BuiltinNames,
2285 Context.getLangOptions().NoBuiltin);
2286 for (unsigned I = 0, N = BuiltinNames.size(); I != N; ++I)
2287 getIdentifierRef(&Table.get(BuiltinNames[I]));
2288 }
2289
Chris Lattner0c797362009-09-08 18:19:27 +00002290 // Build a record containing all of the tentative definitions in this file, in
Sebastian Redl35351a92010-01-31 22:27:38 +00002291 // TentativeDefinitions order. Generally, this record will be empty for
Chris Lattner0c797362009-09-08 18:19:27 +00002292 // headers.
Douglas Gregord4df8652009-04-22 22:02:47 +00002293 RecordData TentativeDefinitions;
Sebastian Redl35351a92010-01-31 22:27:38 +00002294 for (unsigned i = 0, e = SemaRef.TentativeDefinitions.size(); i != e; ++i) {
2295 AddDeclRef(SemaRef.TentativeDefinitions[i], TentativeDefinitions);
Chris Lattner0c797362009-09-08 18:19:27 +00002296 }
Douglas Gregord4df8652009-04-22 22:02:47 +00002297
Argyrios Kyrtzidis35672e72010-08-13 18:42:17 +00002298 // Build a record containing all of the file scoped decls in this file.
2299 RecordData UnusedFileScopedDecls;
2300 for (unsigned i=0, e = SemaRef.UnusedFileScopedDecls.size(); i !=e; ++i)
2301 AddDeclRef(SemaRef.UnusedFileScopedDecls[i], UnusedFileScopedDecls);
Sebastian Redl08aca90252010-08-05 18:21:25 +00002302
Argyrios Kyrtzidisee1afa32010-08-05 09:48:08 +00002303 RecordData WeakUndeclaredIdentifiers;
2304 if (!SemaRef.WeakUndeclaredIdentifiers.empty()) {
2305 WeakUndeclaredIdentifiers.push_back(
2306 SemaRef.WeakUndeclaredIdentifiers.size());
2307 for (llvm::DenseMap<IdentifierInfo*,Sema::WeakInfo>::iterator
2308 I = SemaRef.WeakUndeclaredIdentifiers.begin(),
2309 E = SemaRef.WeakUndeclaredIdentifiers.end(); I != E; ++I) {
2310 AddIdentifierRef(I->first, WeakUndeclaredIdentifiers);
2311 AddIdentifierRef(I->second.getAlias(), WeakUndeclaredIdentifiers);
2312 AddSourceLocation(I->second.getLocation(), WeakUndeclaredIdentifiers);
2313 WeakUndeclaredIdentifiers.push_back(I->second.getUsed());
2314 }
2315 }
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +00002316
Douglas Gregoracfc76c2009-04-22 22:18:58 +00002317 // Build a record containing all of the locally-scoped external
2318 // declarations in this header file. Generally, this record will be
2319 // empty.
2320 RecordData LocallyScopedExternalDecls;
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00002321 // FIXME: This is filling in the AST file in densemap order which is
Chris Lattner0c797362009-09-08 18:19:27 +00002322 // nondeterminstic!
Mike Stump11289f42009-09-09 15:08:12 +00002323 for (llvm::DenseMap<DeclarationName, NamedDecl *>::iterator
Douglas Gregoracfc76c2009-04-22 22:18:58 +00002324 TD = SemaRef.LocallyScopedExternalDecls.begin(),
2325 TDEnd = SemaRef.LocallyScopedExternalDecls.end();
2326 TD != TDEnd; ++TD)
2327 AddDeclRef(TD->second, LocallyScopedExternalDecls);
2328
Douglas Gregor61cac2b2009-04-27 20:06:05 +00002329 // Build a record containing all of the ext_vector declarations.
2330 RecordData ExtVectorDecls;
2331 for (unsigned I = 0, N = SemaRef.ExtVectorDecls.size(); I != N; ++I)
2332 AddDeclRef(SemaRef.ExtVectorDecls[I], ExtVectorDecls);
2333
Argyrios Kyrtzidisaf2eac22010-07-06 15:37:04 +00002334 // Build a record containing all of the VTable uses information.
2335 RecordData VTableUses;
Argyrios Kyrtzidisedee67f2010-08-03 17:29:52 +00002336 if (!SemaRef.VTableUses.empty()) {
2337 VTableUses.push_back(SemaRef.VTableUses.size());
2338 for (unsigned I = 0, N = SemaRef.VTableUses.size(); I != N; ++I) {
2339 AddDeclRef(SemaRef.VTableUses[I].first, VTableUses);
2340 AddSourceLocation(SemaRef.VTableUses[I].second, VTableUses);
2341 VTableUses.push_back(SemaRef.VTablesUsed[SemaRef.VTableUses[I].first]);
2342 }
Argyrios Kyrtzidisaf2eac22010-07-06 15:37:04 +00002343 }
2344
2345 // Build a record containing all of dynamic classes declarations.
2346 RecordData DynamicClasses;
2347 for (unsigned I = 0, N = SemaRef.DynamicClasses.size(); I != N; ++I)
2348 AddDeclRef(SemaRef.DynamicClasses[I], DynamicClasses);
2349
Argyrios Kyrtzidis7f76d112010-08-05 09:48:16 +00002350 // Build a record containing all of pending implicit instantiations.
Chandler Carruth54080172010-08-25 08:44:16 +00002351 RecordData PendingInstantiations;
Argyrios Kyrtzidis7f76d112010-08-05 09:48:16 +00002352 for (std::deque<Sema::PendingImplicitInstantiation>::iterator
Chandler Carruth54080172010-08-25 08:44:16 +00002353 I = SemaRef.PendingInstantiations.begin(),
2354 N = SemaRef.PendingInstantiations.end(); I != N; ++I) {
2355 AddDeclRef(I->first, PendingInstantiations);
2356 AddSourceLocation(I->second, PendingInstantiations);
Argyrios Kyrtzidis7f76d112010-08-05 09:48:16 +00002357 }
2358 assert(SemaRef.PendingLocalImplicitInstantiations.empty() &&
2359 "There are local ones at end of translation unit!");
2360
Argyrios Kyrtzidis2d688102010-08-02 07:14:54 +00002361 // Build a record containing some declaration references.
2362 RecordData SemaDeclRefs;
2363 if (SemaRef.StdNamespace || SemaRef.StdBadAlloc) {
2364 AddDeclRef(SemaRef.getStdNamespace(), SemaDeclRefs);
2365 AddDeclRef(SemaRef.getStdBadAlloc(), SemaDeclRefs);
2366 }
2367
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00002368 // Write the remaining AST contents.
Douglas Gregor652d82a2009-04-18 05:55:16 +00002369 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +00002370 Stream.EnterSubblock(AST_BLOCK_ID, 5);
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00002371 WriteMetadata(Context, isysroot);
Sebastian Redl143413f2010-07-12 22:02:52 +00002372 WriteLanguageOptions(Context.getLangOptions());
Douglas Gregor0086a5a2009-07-07 00:12:59 +00002373 if (StatCalls && !isysroot)
Douglas Gregor11cfd942010-07-12 23:48:14 +00002374 WriteStatCache(*StatCalls);
Douglas Gregor0086a5a2009-07-07 00:12:59 +00002375 WriteSourceManagerBlock(Context.getSourceManager(), PP, isysroot);
Steve Naroffc277ad12009-07-18 15:33:26 +00002376 // Write the record of special types.
2377 Record.clear();
Mike Stump11289f42009-09-09 15:08:12 +00002378
Steve Naroffc277ad12009-07-18 15:33:26 +00002379 AddTypeRef(Context.getBuiltinVaListType(), Record);
2380 AddTypeRef(Context.getObjCIdType(), Record);
2381 AddTypeRef(Context.getObjCSelType(), Record);
2382 AddTypeRef(Context.getObjCProtoType(), Record);
2383 AddTypeRef(Context.getObjCClassType(), Record);
2384 AddTypeRef(Context.getRawCFConstantStringType(), Record);
2385 AddTypeRef(Context.getRawObjCFastEnumerationStateType(), Record);
2386 AddTypeRef(Context.getFILEType(), Record);
Mike Stumpa4de80b2009-07-28 02:25:19 +00002387 AddTypeRef(Context.getjmp_bufType(), Record);
2388 AddTypeRef(Context.getsigjmp_bufType(), Record);
Douglas Gregora8eed7d2009-08-21 00:27:50 +00002389 AddTypeRef(Context.ObjCIdRedefinitionType, Record);
2390 AddTypeRef(Context.ObjCClassRedefinitionType, Record);
Mike Stumpd0153282009-10-20 02:12:22 +00002391 AddTypeRef(Context.getRawBlockdescriptorType(), Record);
Mike Stumpe1b19ba2009-10-22 00:49:09 +00002392 AddTypeRef(Context.getRawBlockdescriptorExtendedType(), Record);
Fariborz Jahaniane804c282010-04-23 17:41:07 +00002393 AddTypeRef(Context.ObjCSelRedefinitionType, Record);
2394 AddTypeRef(Context.getRawNSConstantStringType(), Record);
Argyrios Kyrtzidise862cbc2010-07-04 21:44:19 +00002395 Record.push_back(Context.isInt128Installed());
Sebastian Redl539c5062010-08-18 23:57:32 +00002396 Stream.EmitRecord(SPECIAL_TYPES, Record);
Mike Stump11289f42009-09-09 15:08:12 +00002397
Douglas Gregor1970d882009-04-26 03:49:13 +00002398 // Keep writing types and declarations until all types and
2399 // declarations have been written.
Sebastian Redl539c5062010-08-18 23:57:32 +00002400 Stream.EnterSubblock(DECLTYPES_BLOCK_ID, 3);
Douglas Gregor12bfa382009-10-17 00:13:19 +00002401 WriteDeclsBlockAbbrevs();
2402 while (!DeclTypesToEmit.empty()) {
2403 DeclOrType DOT = DeclTypesToEmit.front();
2404 DeclTypesToEmit.pop();
2405 if (DOT.isType())
2406 WriteType(DOT.getType());
2407 else
2408 WriteDecl(Context, DOT.getDecl());
2409 }
2410 Stream.ExitBlock();
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +00002411
Douglas Gregor45053152009-10-17 17:25:45 +00002412 WritePreprocessor(PP);
Sebastian Redla19a67f2010-08-03 21:58:15 +00002413 WriteSelectors(SemaRef);
Fariborz Jahanianc51609a2010-07-23 19:11:11 +00002414 WriteReferencedSelectorsPool(SemaRef);
Douglas Gregorc3366a52009-04-21 23:56:24 +00002415 WriteIdentifierTable(PP);
Douglas Gregor745ed142009-04-25 18:35:21 +00002416
Sebastian Redl1ea025b2010-07-16 16:36:56 +00002417 WriteTypeDeclOffsets();
Douglas Gregor652d82a2009-04-18 05:55:16 +00002418
Douglas Gregord4df8652009-04-22 22:02:47 +00002419 // Write the record containing external, unnamed definitions.
Douglas Gregor1a0d0b92009-04-14 00:24:19 +00002420 if (!ExternalDefinitions.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002421 Stream.EmitRecord(EXTERNAL_DEFINITIONS, ExternalDefinitions);
Douglas Gregord4df8652009-04-22 22:02:47 +00002422
2423 // Write the record containing tentative definitions.
2424 if (!TentativeDefinitions.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002425 Stream.EmitRecord(TENTATIVE_DEFINITIONS, TentativeDefinitions);
Douglas Gregoracfc76c2009-04-22 22:18:58 +00002426
Argyrios Kyrtzidis35672e72010-08-13 18:42:17 +00002427 // Write the record containing unused file scoped decls.
2428 if (!UnusedFileScopedDecls.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002429 Stream.EmitRecord(UNUSED_FILESCOPED_DECLS, UnusedFileScopedDecls);
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +00002430
Argyrios Kyrtzidisee1afa32010-08-05 09:48:08 +00002431 // Write the record containing weak undeclared identifiers.
2432 if (!WeakUndeclaredIdentifiers.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002433 Stream.EmitRecord(WEAK_UNDECLARED_IDENTIFIERS,
Argyrios Kyrtzidisee1afa32010-08-05 09:48:08 +00002434 WeakUndeclaredIdentifiers);
2435
Douglas Gregoracfc76c2009-04-22 22:18:58 +00002436 // Write the record containing locally-scoped external definitions.
2437 if (!LocallyScopedExternalDecls.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002438 Stream.EmitRecord(LOCALLY_SCOPED_EXTERNAL_DECLS,
Douglas Gregoracfc76c2009-04-22 22:18:58 +00002439 LocallyScopedExternalDecls);
Douglas Gregor61cac2b2009-04-27 20:06:05 +00002440
2441 // Write the record containing ext_vector type names.
2442 if (!ExtVectorDecls.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002443 Stream.EmitRecord(EXT_VECTOR_DECLS, ExtVectorDecls);
Mike Stump11289f42009-09-09 15:08:12 +00002444
Argyrios Kyrtzidisaf2eac22010-07-06 15:37:04 +00002445 // Write the record containing VTable uses information.
2446 if (!VTableUses.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002447 Stream.EmitRecord(VTABLE_USES, VTableUses);
Argyrios Kyrtzidisaf2eac22010-07-06 15:37:04 +00002448
2449 // Write the record containing dynamic classes declarations.
2450 if (!DynamicClasses.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002451 Stream.EmitRecord(DYNAMIC_CLASSES, DynamicClasses);
Argyrios Kyrtzidisaf2eac22010-07-06 15:37:04 +00002452
Argyrios Kyrtzidis7f76d112010-08-05 09:48:16 +00002453 // Write the record containing pending implicit instantiations.
Chandler Carruth54080172010-08-25 08:44:16 +00002454 if (!PendingInstantiations.empty())
2455 Stream.EmitRecord(PENDING_IMPLICIT_INSTANTIATIONS, PendingInstantiations);
Argyrios Kyrtzidis7f76d112010-08-05 09:48:16 +00002456
Argyrios Kyrtzidis2d688102010-08-02 07:14:54 +00002457 // Write the record containing declaration references of Sema.
2458 if (!SemaDeclRefs.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002459 Stream.EmitRecord(SEMA_DECL_REFS, SemaDeclRefs);
Argyrios Kyrtzidis2d688102010-08-02 07:14:54 +00002460
Douglas Gregor08f01292009-04-17 22:13:46 +00002461 // Some simple statistics
Douglas Gregor652d82a2009-04-18 05:55:16 +00002462 Record.clear();
Douglas Gregor08f01292009-04-17 22:13:46 +00002463 Record.push_back(NumStatements);
Douglas Gregorc3366a52009-04-21 23:56:24 +00002464 Record.push_back(NumMacros);
Douglas Gregora57c3ab2009-04-22 22:34:57 +00002465 Record.push_back(NumLexicalDeclContexts);
2466 Record.push_back(NumVisibleDeclContexts);
Sebastian Redl539c5062010-08-18 23:57:32 +00002467 Stream.EmitRecord(STATISTICS, Record);
Douglas Gregor8f45df52009-04-16 22:23:12 +00002468 Stream.ExitBlock();
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002469}
2470
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002471void ASTWriter::WriteASTChain(Sema &SemaRef, MemorizeStatCalls *StatCalls,
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00002472 const char *isysroot) {
Sebastian Redl143413f2010-07-12 22:02:52 +00002473 using namespace llvm;
2474
Sebastian Redl07a89a82010-07-30 00:29:29 +00002475 FirstDeclID += Chain->getTotalNumDecls();
2476 FirstTypeID += Chain->getTotalNumTypes();
2477 FirstIdentID += Chain->getTotalNumIdentifiers();
Sebastian Redld95a56e2010-08-04 18:21:41 +00002478 FirstSelectorID += Chain->getTotalNumSelectors();
Douglas Gregor91096292010-10-02 19:29:26 +00002479 FirstMacroID += Chain->getTotalNumMacroDefinitions();
Sebastian Redl07a89a82010-07-30 00:29:29 +00002480 NextDeclID = FirstDeclID;
2481 NextTypeID = FirstTypeID;
2482 NextIdentID = FirstIdentID;
Sebastian Redld95a56e2010-08-04 18:21:41 +00002483 NextSelectorID = FirstSelectorID;
Douglas Gregor91096292010-10-02 19:29:26 +00002484 NextMacroID = FirstMacroID;
Sebastian Redl07a89a82010-07-30 00:29:29 +00002485
Sebastian Redl143413f2010-07-12 22:02:52 +00002486 ASTContext &Context = SemaRef.Context;
2487 Preprocessor &PP = SemaRef.PP;
Sebastian Redl1ea025b2010-07-16 16:36:56 +00002488
Sebastian Redl143413f2010-07-12 22:02:52 +00002489 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +00002490 Stream.EnterSubblock(AST_BLOCK_ID, 5);
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00002491 WriteMetadata(Context, isysroot);
Sebastian Redl1ea025b2010-07-16 16:36:56 +00002492 if (StatCalls && !isysroot)
2493 WriteStatCache(*StatCalls);
2494 // FIXME: Source manager block should only write new stuff, which could be
2495 // done by tracking the largest ID in the chain
2496 WriteSourceManagerBlock(Context.getSourceManager(), PP, isysroot);
Sebastian Redl143413f2010-07-12 22:02:52 +00002497
2498 // The special types are in the chained PCH.
2499
2500 // We don't start with the translation unit, but with its decls that
Sebastian Redld44cd6a2010-08-18 23:57:06 +00002501 // don't come from the chained PCH.
Sebastian Redl143413f2010-07-12 22:02:52 +00002502 const TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
Argyrios Kyrtzidis0e88a562010-10-14 20:14:34 +00002503 llvm::SmallVector<KindDeclIDPair, 64> NewGlobalDecls;
Sebastian Redl66c5eef2010-07-27 00:17:23 +00002504 for (DeclContext::decl_iterator I = TU->noload_decls_begin(),
2505 E = TU->noload_decls_end();
Sebastian Redl143413f2010-07-12 22:02:52 +00002506 I != E; ++I) {
Sebastian Redl4b1f4902010-07-27 18:24:41 +00002507 if ((*I)->getPCHLevel() == 0)
Argyrios Kyrtzidis0e88a562010-10-14 20:14:34 +00002508 NewGlobalDecls.push_back(std::make_pair((*I)->getKind(), GetDeclRef(*I)));
Sebastian Redle7c1fe62010-08-13 00:28:03 +00002509 else if ((*I)->isChangedSinceDeserialization())
2510 (void)GetDeclRef(*I); // Make sure it's written, but don't record it.
Sebastian Redl143413f2010-07-12 22:02:52 +00002511 }
Sebastian Redl66c5eef2010-07-27 00:17:23 +00002512 // We also need to write a lexical updates block for the TU.
Sebastian Redl4b1f4902010-07-27 18:24:41 +00002513 llvm::BitCodeAbbrev *Abv = new llvm::BitCodeAbbrev();
Sebastian Redl539c5062010-08-18 23:57:32 +00002514 Abv->Add(llvm::BitCodeAbbrevOp(TU_UPDATE_LEXICAL));
Sebastian Redl4b1f4902010-07-27 18:24:41 +00002515 Abv->Add(llvm::BitCodeAbbrevOp(llvm::BitCodeAbbrevOp::Blob));
2516 unsigned TuUpdateLexicalAbbrev = Stream.EmitAbbrev(Abv);
2517 Record.clear();
Sebastian Redl539c5062010-08-18 23:57:32 +00002518 Record.push_back(TU_UPDATE_LEXICAL);
Sebastian Redl4b1f4902010-07-27 18:24:41 +00002519 Stream.EmitRecordWithBlob(TuUpdateLexicalAbbrev, Record,
2520 reinterpret_cast<const char*>(NewGlobalDecls.data()),
Argyrios Kyrtzidis0e88a562010-10-14 20:14:34 +00002521 NewGlobalDecls.size() * sizeof(KindDeclIDPair));
Sebastian Redla4071b42010-08-24 00:50:09 +00002522 // And in C++, a visible updates block for the TU.
2523 if (Context.getLangOptions().CPlusPlus) {
2524 Abv = new llvm::BitCodeAbbrev();
2525 Abv->Add(llvm::BitCodeAbbrevOp(UPDATE_VISIBLE));
2526 Abv->Add(llvm::BitCodeAbbrevOp(llvm::BitCodeAbbrevOp::VBR, 6));
2527 Abv->Add(llvm::BitCodeAbbrevOp(llvm::BitCodeAbbrevOp::Fixed, 32));
2528 Abv->Add(llvm::BitCodeAbbrevOp(llvm::BitCodeAbbrevOp::Blob));
2529 UpdateVisibleAbbrev = Stream.EmitAbbrev(Abv);
2530 WriteDeclContextVisibleUpdate(TU);
2531 }
Sebastian Redl143413f2010-07-12 22:02:52 +00002532
Sebastian Redl98912122010-07-27 23:01:28 +00002533 // Build a record containing all of the new tentative definitions in this
2534 // file, in TentativeDefinitions order.
2535 RecordData TentativeDefinitions;
2536 for (unsigned i = 0, e = SemaRef.TentativeDefinitions.size(); i != e; ++i) {
2537 if (SemaRef.TentativeDefinitions[i]->getPCHLevel() == 0)
2538 AddDeclRef(SemaRef.TentativeDefinitions[i], TentativeDefinitions);
2539 }
2540
Argyrios Kyrtzidis35672e72010-08-13 18:42:17 +00002541 // Build a record containing all of the file scoped decls in this file.
2542 RecordData UnusedFileScopedDecls;
2543 for (unsigned i=0, e = SemaRef.UnusedFileScopedDecls.size(); i !=e; ++i) {
2544 if (SemaRef.UnusedFileScopedDecls[i]->getPCHLevel() == 0)
2545 AddDeclRef(SemaRef.UnusedFileScopedDecls[i], UnusedFileScopedDecls);
Sebastian Redl98912122010-07-27 23:01:28 +00002546 }
2547
Sebastian Redl08aca90252010-08-05 18:21:25 +00002548 // We write the entire table, overwriting the tables from the chain.
2549 RecordData WeakUndeclaredIdentifiers;
2550 if (!SemaRef.WeakUndeclaredIdentifiers.empty()) {
2551 WeakUndeclaredIdentifiers.push_back(
2552 SemaRef.WeakUndeclaredIdentifiers.size());
2553 for (llvm::DenseMap<IdentifierInfo*,Sema::WeakInfo>::iterator
2554 I = SemaRef.WeakUndeclaredIdentifiers.begin(),
2555 E = SemaRef.WeakUndeclaredIdentifiers.end(); I != E; ++I) {
2556 AddIdentifierRef(I->first, WeakUndeclaredIdentifiers);
2557 AddIdentifierRef(I->second.getAlias(), WeakUndeclaredIdentifiers);
2558 AddSourceLocation(I->second.getLocation(), WeakUndeclaredIdentifiers);
2559 WeakUndeclaredIdentifiers.push_back(I->second.getUsed());
2560 }
2561 }
2562
Sebastian Redl98912122010-07-27 23:01:28 +00002563 // Build a record containing all of the locally-scoped external
2564 // declarations in this header file. Generally, this record will be
2565 // empty.
2566 RecordData LocallyScopedExternalDecls;
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00002567 // FIXME: This is filling in the AST file in densemap order which is
Sebastian Redl98912122010-07-27 23:01:28 +00002568 // nondeterminstic!
2569 for (llvm::DenseMap<DeclarationName, NamedDecl *>::iterator
2570 TD = SemaRef.LocallyScopedExternalDecls.begin(),
2571 TDEnd = SemaRef.LocallyScopedExternalDecls.end();
2572 TD != TDEnd; ++TD) {
2573 if (TD->second->getPCHLevel() == 0)
2574 AddDeclRef(TD->second, LocallyScopedExternalDecls);
2575 }
2576
2577 // Build a record containing all of the ext_vector declarations.
2578 RecordData ExtVectorDecls;
2579 for (unsigned I = 0, N = SemaRef.ExtVectorDecls.size(); I != N; ++I) {
2580 if (SemaRef.ExtVectorDecls[I]->getPCHLevel() == 0)
2581 AddDeclRef(SemaRef.ExtVectorDecls[I], ExtVectorDecls);
2582 }
2583
Sebastian Redl08aca90252010-08-05 18:21:25 +00002584 // Build a record containing all of the VTable uses information.
2585 // We write everything here, because it's too hard to determine whether
2586 // a use is new to this part.
2587 RecordData VTableUses;
2588 if (!SemaRef.VTableUses.empty()) {
2589 VTableUses.push_back(SemaRef.VTableUses.size());
2590 for (unsigned I = 0, N = SemaRef.VTableUses.size(); I != N; ++I) {
2591 AddDeclRef(SemaRef.VTableUses[I].first, VTableUses);
2592 AddSourceLocation(SemaRef.VTableUses[I].second, VTableUses);
2593 VTableUses.push_back(SemaRef.VTablesUsed[SemaRef.VTableUses[I].first]);
2594 }
2595 }
2596
2597 // Build a record containing all of dynamic classes declarations.
2598 RecordData DynamicClasses;
2599 for (unsigned I = 0, N = SemaRef.DynamicClasses.size(); I != N; ++I)
2600 if (SemaRef.DynamicClasses[I]->getPCHLevel() == 0)
2601 AddDeclRef(SemaRef.DynamicClasses[I], DynamicClasses);
2602
2603 // Build a record containing all of pending implicit instantiations.
Chandler Carruth54080172010-08-25 08:44:16 +00002604 RecordData PendingInstantiations;
Sebastian Redl08aca90252010-08-05 18:21:25 +00002605 for (std::deque<Sema::PendingImplicitInstantiation>::iterator
Chandler Carruth54080172010-08-25 08:44:16 +00002606 I = SemaRef.PendingInstantiations.begin(),
2607 N = SemaRef.PendingInstantiations.end(); I != N; ++I) {
Sebastian Redl08aca90252010-08-05 18:21:25 +00002608 if (I->first->getPCHLevel() == 0) {
Chandler Carruth54080172010-08-25 08:44:16 +00002609 AddDeclRef(I->first, PendingInstantiations);
2610 AddSourceLocation(I->second, PendingInstantiations);
Sebastian Redl08aca90252010-08-05 18:21:25 +00002611 }
2612 }
2613 assert(SemaRef.PendingLocalImplicitInstantiations.empty() &&
2614 "There are local ones at end of translation unit!");
2615
2616 // Build a record containing some declaration references.
2617 // It's not worth the effort to avoid duplication here.
2618 RecordData SemaDeclRefs;
2619 if (SemaRef.StdNamespace || SemaRef.StdBadAlloc) {
2620 AddDeclRef(SemaRef.getStdNamespace(), SemaDeclRefs);
2621 AddDeclRef(SemaRef.getStdBadAlloc(), SemaDeclRefs);
2622 }
2623
Sebastian Redl539c5062010-08-18 23:57:32 +00002624 Stream.EnterSubblock(DECLTYPES_BLOCK_ID, 3);
Sebastian Redl143413f2010-07-12 22:02:52 +00002625 WriteDeclsBlockAbbrevs();
2626 while (!DeclTypesToEmit.empty()) {
2627 DeclOrType DOT = DeclTypesToEmit.front();
2628 DeclTypesToEmit.pop();
2629 if (DOT.isType())
2630 WriteType(DOT.getType());
2631 else
2632 WriteDecl(Context, DOT.getDecl());
2633 }
2634 Stream.ExitBlock();
2635
Sebastian Redl98912122010-07-27 23:01:28 +00002636 WritePreprocessor(PP);
Sebastian Redl51c79d82010-08-04 22:21:29 +00002637 WriteSelectors(SemaRef);
2638 WriteReferencedSelectorsPool(SemaRef);
Sebastian Redlff4a2952010-07-23 23:49:55 +00002639 WriteIdentifierTable(PP);
Sebastian Redl1ea025b2010-07-16 16:36:56 +00002640 WriteTypeDeclOffsets();
Sebastian Redl98912122010-07-27 23:01:28 +00002641
Argyrios Kyrtzidis839bbac2010-08-03 17:30:10 +00002642 /// Build a record containing first declarations from a chained PCH and the
Sebastian Redl42a0f6a2010-08-18 23:56:27 +00002643 /// most recent declarations in this AST that they point to.
Argyrios Kyrtzidis839bbac2010-08-03 17:30:10 +00002644 RecordData FirstLatestDeclIDs;
2645 for (FirstLatestDeclMap::iterator
2646 I = FirstLatestDecls.begin(), E = FirstLatestDecls.end(); I != E; ++I) {
2647 assert(I->first->getPCHLevel() > I->second->getPCHLevel() &&
2648 "Expected first & second to be in different PCHs");
2649 AddDeclRef(I->first, FirstLatestDeclIDs);
2650 AddDeclRef(I->second, FirstLatestDeclIDs);
2651 }
2652 if (!FirstLatestDeclIDs.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002653 Stream.EmitRecord(REDECLS_UPDATE_LATEST, FirstLatestDeclIDs);
Argyrios Kyrtzidis839bbac2010-08-03 17:30:10 +00002654
Sebastian Redl98912122010-07-27 23:01:28 +00002655 // Write the record containing external, unnamed definitions.
2656 if (!ExternalDefinitions.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002657 Stream.EmitRecord(EXTERNAL_DEFINITIONS, ExternalDefinitions);
Sebastian Redl98912122010-07-27 23:01:28 +00002658
2659 // Write the record containing tentative definitions.
2660 if (!TentativeDefinitions.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002661 Stream.EmitRecord(TENTATIVE_DEFINITIONS, TentativeDefinitions);
Sebastian Redl98912122010-07-27 23:01:28 +00002662
Argyrios Kyrtzidis35672e72010-08-13 18:42:17 +00002663 // Write the record containing unused file scoped decls.
2664 if (!UnusedFileScopedDecls.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002665 Stream.EmitRecord(UNUSED_FILESCOPED_DECLS, UnusedFileScopedDecls);
Sebastian Redl98912122010-07-27 23:01:28 +00002666
Sebastian Redl08aca90252010-08-05 18:21:25 +00002667 // Write the record containing weak undeclared identifiers.
2668 if (!WeakUndeclaredIdentifiers.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002669 Stream.EmitRecord(WEAK_UNDECLARED_IDENTIFIERS,
Sebastian Redl08aca90252010-08-05 18:21:25 +00002670 WeakUndeclaredIdentifiers);
2671
Sebastian Redl98912122010-07-27 23:01:28 +00002672 // Write the record containing locally-scoped external definitions.
2673 if (!LocallyScopedExternalDecls.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002674 Stream.EmitRecord(LOCALLY_SCOPED_EXTERNAL_DECLS,
Sebastian Redl98912122010-07-27 23:01:28 +00002675 LocallyScopedExternalDecls);
2676
2677 // Write the record containing ext_vector type names.
2678 if (!ExtVectorDecls.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002679 Stream.EmitRecord(EXT_VECTOR_DECLS, ExtVectorDecls);
Sebastian Redl98912122010-07-27 23:01:28 +00002680
Sebastian Redl08aca90252010-08-05 18:21:25 +00002681 // Write the record containing VTable uses information.
2682 if (!VTableUses.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002683 Stream.EmitRecord(VTABLE_USES, VTableUses);
Sebastian Redl08aca90252010-08-05 18:21:25 +00002684
2685 // Write the record containing dynamic classes declarations.
2686 if (!DynamicClasses.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002687 Stream.EmitRecord(DYNAMIC_CLASSES, DynamicClasses);
Sebastian Redl08aca90252010-08-05 18:21:25 +00002688
2689 // Write the record containing pending implicit instantiations.
Chandler Carruth54080172010-08-25 08:44:16 +00002690 if (!PendingInstantiations.empty())
2691 Stream.EmitRecord(PENDING_IMPLICIT_INSTANTIATIONS, PendingInstantiations);
Sebastian Redl08aca90252010-08-05 18:21:25 +00002692
2693 // Write the record containing declaration references of Sema.
2694 if (!SemaDeclRefs.empty())
Sebastian Redl539c5062010-08-18 23:57:32 +00002695 Stream.EmitRecord(SEMA_DECL_REFS, SemaDeclRefs);
Sebastian Redl98912122010-07-27 23:01:28 +00002696
Sebastian Redla4071b42010-08-24 00:50:09 +00002697 // Write the updates to C++ namespaces.
2698 for (llvm::SmallPtrSet<const NamespaceDecl *, 16>::iterator
2699 I = UpdatedNamespaces.begin(),
2700 E = UpdatedNamespaces.end();
2701 I != E; ++I)
2702 WriteDeclContextVisibleUpdate(*I);
2703
Sebastian Redl401b39a2010-08-24 22:50:24 +00002704 // Write the updates to C++ template specialization lists.
2705 if (!AdditionalTemplateSpecializations.empty())
2706 WriteAdditionalTemplateSpecializations();
2707
Sebastian Redl98912122010-07-27 23:01:28 +00002708 Record.clear();
2709 Record.push_back(NumStatements);
2710 Record.push_back(NumMacros);
2711 Record.push_back(NumLexicalDeclContexts);
2712 Record.push_back(NumVisibleDeclContexts);
Sebastian Redle7c1fe62010-08-13 00:28:03 +00002713 WriteDeclUpdateBlock();
Sebastian Redl539c5062010-08-18 23:57:32 +00002714 Stream.EmitRecord(STATISTICS, Record);
Sebastian Redl143413f2010-07-12 22:02:52 +00002715 Stream.ExitBlock();
2716}
2717
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002718void ASTWriter::WriteDeclUpdateBlock() {
Sebastian Redle7c1fe62010-08-13 00:28:03 +00002719 if (ReplacedDecls.empty())
2720 return;
2721
2722 RecordData Record;
Sebastian Redl539c5062010-08-18 23:57:32 +00002723 for (llvm::SmallVector<std::pair<DeclID, uint64_t>, 16>::iterator
Sebastian Redle7c1fe62010-08-13 00:28:03 +00002724 I = ReplacedDecls.begin(), E = ReplacedDecls.end(); I != E; ++I) {
2725 Record.push_back(I->first);
2726 Record.push_back(I->second);
2727 }
Sebastian Redl539c5062010-08-18 23:57:32 +00002728 Stream.EmitRecord(DECL_REPLACEMENTS, Record);
Sebastian Redle7c1fe62010-08-13 00:28:03 +00002729}
2730
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002731void ASTWriter::AddSourceLocation(SourceLocation Loc, RecordData &Record) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002732 Record.push_back(Loc.getRawEncoding());
2733}
2734
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002735void ASTWriter::AddSourceRange(SourceRange Range, RecordData &Record) {
Chris Lattnerca025db2010-05-07 21:43:38 +00002736 AddSourceLocation(Range.getBegin(), Record);
2737 AddSourceLocation(Range.getEnd(), Record);
2738}
2739
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002740void ASTWriter::AddAPInt(const llvm::APInt &Value, RecordData &Record) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002741 Record.push_back(Value.getBitWidth());
Benjamin Kramer25f9ea62010-09-06 23:43:28 +00002742 const uint64_t *Words = Value.getRawData();
2743 Record.append(Words, Words + Value.getNumWords());
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002744}
2745
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002746void ASTWriter::AddAPSInt(const llvm::APSInt &Value, RecordData &Record) {
Douglas Gregor1daeb692009-04-13 18:14:40 +00002747 Record.push_back(Value.isUnsigned());
2748 AddAPInt(Value, Record);
2749}
2750
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002751void ASTWriter::AddAPFloat(const llvm::APFloat &Value, RecordData &Record) {
Douglas Gregore0a3a512009-04-14 21:55:33 +00002752 AddAPInt(Value.bitcastToAPInt(), Record);
2753}
2754
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002755void ASTWriter::AddIdentifierRef(const IdentifierInfo *II, RecordData &Record) {
Douglas Gregor4621c6a2009-04-22 18:49:13 +00002756 Record.push_back(getIdentifierRef(II));
2757}
2758
Sebastian Redl539c5062010-08-18 23:57:32 +00002759IdentID ASTWriter::getIdentifierRef(const IdentifierInfo *II) {
Douglas Gregor4621c6a2009-04-22 18:49:13 +00002760 if (II == 0)
2761 return 0;
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00002762
Sebastian Redl539c5062010-08-18 23:57:32 +00002763 IdentID &ID = IdentifierIDs[II];
Douglas Gregor3ed42cb2009-04-11 00:14:32 +00002764 if (ID == 0)
Sebastian Redlff4a2952010-07-23 23:49:55 +00002765 ID = NextIdentID++;
Douglas Gregor4621c6a2009-04-22 18:49:13 +00002766 return ID;
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002767}
2768
Sebastian Redl50e26582010-09-15 19:54:06 +00002769MacroID ASTWriter::getMacroDefinitionID(MacroDefinition *MD) {
Douglas Gregoraae92242010-03-19 21:51:54 +00002770 if (MD == 0)
2771 return 0;
Sebastian Redl50e26582010-09-15 19:54:06 +00002772
2773 MacroID &ID = MacroDefinitions[MD];
Douglas Gregoraae92242010-03-19 21:51:54 +00002774 if (ID == 0)
Douglas Gregor91096292010-10-02 19:29:26 +00002775 ID = NextMacroID++;
Douglas Gregoraae92242010-03-19 21:51:54 +00002776 return ID;
2777}
2778
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002779void ASTWriter::AddSelectorRef(const Selector SelRef, RecordData &Record) {
Sebastian Redl834bb972010-08-04 17:20:04 +00002780 Record.push_back(getSelectorRef(SelRef));
2781}
2782
Sebastian Redl539c5062010-08-18 23:57:32 +00002783SelectorID ASTWriter::getSelectorRef(Selector Sel) {
Sebastian Redl834bb972010-08-04 17:20:04 +00002784 if (Sel.getAsOpaquePtr() == 0) {
2785 return 0;
Steve Naroff2ddea052009-04-23 10:39:46 +00002786 }
2787
Sebastian Redl539c5062010-08-18 23:57:32 +00002788 SelectorID &SID = SelectorIDs[Sel];
Sebastian Redld95a56e2010-08-04 18:21:41 +00002789 if (SID == 0 && Chain) {
2790 // This might trigger a ReadSelector callback, which will set the ID for
2791 // this selector.
2792 Chain->LoadSelector(Sel);
2793 }
Steve Naroff2ddea052009-04-23 10:39:46 +00002794 if (SID == 0) {
Sebastian Redld95a56e2010-08-04 18:21:41 +00002795 SID = NextSelectorID++;
Steve Naroff2ddea052009-04-23 10:39:46 +00002796 }
Sebastian Redl834bb972010-08-04 17:20:04 +00002797 return SID;
Steve Naroff2ddea052009-04-23 10:39:46 +00002798}
2799
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002800void ASTWriter::AddCXXTemporary(const CXXTemporary *Temp, RecordData &Record) {
Chris Lattnercba86142010-05-10 00:25:06 +00002801 AddDeclRef(Temp->getDestructor(), Record);
2802}
2803
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002804void ASTWriter::AddTemplateArgumentLocInfo(TemplateArgument::ArgKind Kind,
Argyrios Kyrtzidisae85e242010-06-22 09:54:59 +00002805 const TemplateArgumentLocInfo &Arg,
2806 RecordData &Record) {
2807 switch (Kind) {
John McCall0ad16662009-10-29 08:12:44 +00002808 case TemplateArgument::Expression:
Argyrios Kyrtzidisae85e242010-06-22 09:54:59 +00002809 AddStmt(Arg.getAsExpr());
John McCall0ad16662009-10-29 08:12:44 +00002810 break;
2811 case TemplateArgument::Type:
Argyrios Kyrtzidisae85e242010-06-22 09:54:59 +00002812 AddTypeSourceInfo(Arg.getAsTypeSourceInfo(), Record);
John McCall0ad16662009-10-29 08:12:44 +00002813 break;
Douglas Gregor9167f8b2009-11-11 01:00:40 +00002814 case TemplateArgument::Template:
Argyrios Kyrtzidisddf5f212010-06-28 09:31:42 +00002815 AddSourceRange(Arg.getTemplateQualifierRange(), Record);
2816 AddSourceLocation(Arg.getTemplateNameLoc(), Record);
Douglas Gregor9167f8b2009-11-11 01:00:40 +00002817 break;
John McCall0ad16662009-10-29 08:12:44 +00002818 case TemplateArgument::Null:
2819 case TemplateArgument::Integral:
2820 case TemplateArgument::Declaration:
2821 case TemplateArgument::Pack:
2822 break;
2823 }
2824}
2825
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002826void ASTWriter::AddTemplateArgumentLoc(const TemplateArgumentLoc &Arg,
Argyrios Kyrtzidisae85e242010-06-22 09:54:59 +00002827 RecordData &Record) {
2828 AddTemplateArgument(Arg.getArgument(), Record);
Argyrios Kyrtzidisddf5f212010-06-28 09:31:42 +00002829
2830 if (Arg.getArgument().getKind() == TemplateArgument::Expression) {
2831 bool InfoHasSameExpr
2832 = Arg.getArgument().getAsExpr() == Arg.getLocInfo().getAsExpr();
2833 Record.push_back(InfoHasSameExpr);
2834 if (InfoHasSameExpr)
2835 return; // Avoid storing the same expr twice.
2836 }
Argyrios Kyrtzidisae85e242010-06-22 09:54:59 +00002837 AddTemplateArgumentLocInfo(Arg.getArgument().getKind(), Arg.getLocInfo(),
2838 Record);
2839}
2840
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002841void ASTWriter::AddTypeSourceInfo(TypeSourceInfo *TInfo, RecordData &Record) {
John McCallbcd03502009-12-07 02:54:59 +00002842 if (TInfo == 0) {
John McCall8f115c62009-10-16 21:56:05 +00002843 AddTypeRef(QualType(), Record);
2844 return;
2845 }
2846
John McCallbcd03502009-12-07 02:54:59 +00002847 AddTypeRef(TInfo->getType(), Record);
John McCall8f115c62009-10-16 21:56:05 +00002848 TypeLocWriter TLW(*this, Record);
John McCallbcd03502009-12-07 02:54:59 +00002849 for (TypeLoc TL = TInfo->getTypeLoc(); !TL.isNull(); TL = TL.getNextTypeLoc())
Kovarththanan Rajaratnama9c81a82010-03-14 07:06:50 +00002850 TLW.Visit(TL);
John McCall8f115c62009-10-16 21:56:05 +00002851}
2852
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002853void ASTWriter::AddTypeRef(QualType T, RecordData &Record) {
Argyrios Kyrtzidis9ab44ea2010-08-20 16:04:14 +00002854 Record.push_back(GetOrCreateTypeID(T));
2855}
2856
2857TypeID ASTWriter::GetOrCreateTypeID(QualType T) {
Argyrios Kyrtzidis082e4612010-08-20 16:04:20 +00002858 return MakeTypeID(T,
2859 std::bind1st(std::mem_fun(&ASTWriter::GetOrCreateTypeIdx), this));
2860}
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002861
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00002862TypeID ASTWriter::getTypeID(QualType T) const {
Argyrios Kyrtzidis082e4612010-08-20 16:04:20 +00002863 return MakeTypeID(T,
2864 std::bind1st(std::mem_fun(&ASTWriter::getTypeIdx), this));
Argyrios Kyrtzidise394f2c2010-08-20 16:04:09 +00002865}
2866
2867TypeIdx ASTWriter::GetOrCreateTypeIdx(QualType T) {
2868 if (T.isNull())
2869 return TypeIdx();
2870 assert(!T.getLocalFastQualifiers());
2871
Argyrios Kyrtzidisa7fbbb02010-08-20 16:04:04 +00002872 TypeIdx &Idx = TypeIdxs[T];
Argyrios Kyrtzidisbb5c7eae2010-08-20 16:03:59 +00002873 if (Idx.getIndex() == 0) {
Douglas Gregor1970d882009-04-26 03:49:13 +00002874 // We haven't seen this type before. Assign it a new ID and put it
John McCall8ccfcb52009-09-24 19:53:00 +00002875 // into the queue of types to emit.
Argyrios Kyrtzidisbb5c7eae2010-08-20 16:03:59 +00002876 Idx = TypeIdx(NextTypeID++);
Douglas Gregor12bfa382009-10-17 00:13:19 +00002877 DeclTypesToEmit.push(T);
Douglas Gregor1970d882009-04-26 03:49:13 +00002878 }
Argyrios Kyrtzidise394f2c2010-08-20 16:04:09 +00002879 return Idx;
2880}
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002881
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00002882TypeIdx ASTWriter::getTypeIdx(QualType T) const {
Argyrios Kyrtzidise394f2c2010-08-20 16:04:09 +00002883 if (T.isNull())
2884 return TypeIdx();
2885 assert(!T.getLocalFastQualifiers());
2886
Argyrios Kyrtzidis07347322010-08-20 16:04:27 +00002887 TypeIdxMap::const_iterator I = TypeIdxs.find(T);
2888 assert(I != TypeIdxs.end() && "Type not emitted!");
2889 return I->second;
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002890}
2891
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002892void ASTWriter::AddDeclRef(const Decl *D, RecordData &Record) {
Sebastian Redl66c5eef2010-07-27 00:17:23 +00002893 Record.push_back(GetDeclRef(D));
2894}
2895
Sebastian Redl539c5062010-08-18 23:57:32 +00002896DeclID ASTWriter::GetDeclRef(const Decl *D) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002897 if (D == 0) {
Sebastian Redl66c5eef2010-07-27 00:17:23 +00002898 return 0;
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002899 }
Douglas Gregor9b3932c2010-10-05 18:37:06 +00002900 assert(!(reinterpret_cast<uintptr_t>(D) & 0x01) && "Invalid decl pointer");
Sebastian Redl539c5062010-08-18 23:57:32 +00002901 DeclID &ID = DeclIDs[D];
Mike Stump11289f42009-09-09 15:08:12 +00002902 if (ID == 0) {
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002903 // We haven't seen this declaration before. Give it a new ID and
2904 // enqueue it in the list of declarations to emit.
Sebastian Redlff4a2952010-07-23 23:49:55 +00002905 ID = NextDeclID++;
Douglas Gregor12bfa382009-10-17 00:13:19 +00002906 DeclTypesToEmit.push(const_cast<Decl *>(D));
Sebastian Redle7c1fe62010-08-13 00:28:03 +00002907 } else if (ID < FirstDeclID && D->isChangedSinceDeserialization()) {
2908 // We don't add it to the replacement collection here, because we don't
2909 // have the offset yet.
2910 DeclTypesToEmit.push(const_cast<Decl *>(D));
2911 // Reset the flag, so that we don't add this decl multiple times.
2912 const_cast<Decl *>(D)->setChangedSinceDeserialization(false);
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002913 }
2914
Sebastian Redl66c5eef2010-07-27 00:17:23 +00002915 return ID;
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002916}
2917
Sebastian Redl539c5062010-08-18 23:57:32 +00002918DeclID ASTWriter::getDeclID(const Decl *D) {
Douglas Gregore84a9da2009-04-20 20:36:09 +00002919 if (D == 0)
2920 return 0;
2921
2922 assert(DeclIDs.find(D) != DeclIDs.end() && "Declaration not emitted!");
2923 return DeclIDs[D];
2924}
2925
Sebastian Redl55c0ad52010-08-18 23:56:21 +00002926void ASTWriter::AddDeclarationName(DeclarationName Name, RecordData &Record) {
Chris Lattner258172e2009-04-27 07:35:58 +00002927 // FIXME: Emit a stable enum for NameKind. 0 = Identifier etc.
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002928 Record.push_back(Name.getNameKind());
2929 switch (Name.getNameKind()) {
2930 case DeclarationName::Identifier:
2931 AddIdentifierRef(Name.getAsIdentifierInfo(), Record);
2932 break;
2933
2934 case DeclarationName::ObjCZeroArgSelector:
2935 case DeclarationName::ObjCOneArgSelector:
2936 case DeclarationName::ObjCMultiArgSelector:
Steve Naroff2ddea052009-04-23 10:39:46 +00002937 AddSelectorRef(Name.getObjCSelector(), Record);
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002938 break;
2939
2940 case DeclarationName::CXXConstructorName:
2941 case DeclarationName::CXXDestructorName:
2942 case DeclarationName::CXXConversionFunctionName:
2943 AddTypeRef(Name.getCXXNameType(), Record);
2944 break;
2945
2946 case DeclarationName::CXXOperatorName:
2947 Record.push_back(Name.getCXXOverloadedOperator());
2948 break;
2949
Alexis Hunt3d221f22009-11-29 07:34:05 +00002950 case DeclarationName::CXXLiteralOperatorName:
2951 AddIdentifierRef(Name.getCXXLiteralIdentifier(), Record);
2952 break;
2953
Douglas Gregoref84c4b2009-04-09 22:27:44 +00002954 case DeclarationName::CXXUsingDirective:
2955 // No extra data to emit
2956 break;
2957 }
2958}
Chris Lattnerca025db2010-05-07 21:43:38 +00002959
Argyrios Kyrtzidis434383d2010-10-15 18:21:24 +00002960void ASTWriter::AddDeclarationNameLoc(const DeclarationNameLoc &DNLoc,
2961 DeclarationName Name, RecordData &Record) {
2962 switch (Name.getNameKind()) {
2963 case DeclarationName::CXXConstructorName:
2964 case DeclarationName::CXXDestructorName:
2965 case DeclarationName::CXXConversionFunctionName:
2966 AddTypeSourceInfo(DNLoc.NamedType.TInfo, Record);
2967 break;
2968
2969 case DeclarationName::CXXOperatorName:
2970 AddSourceLocation(
2971 SourceLocation::getFromRawEncoding(DNLoc.CXXOperatorName.BeginOpNameLoc),
2972 Record);
2973 AddSourceLocation(
2974 SourceLocation::getFromRawEncoding(DNLoc.CXXOperatorName.EndOpNameLoc),
2975 Record);
2976 break;
2977
2978 case DeclarationName::CXXLiteralOperatorName:
2979 AddSourceLocation(
2980 SourceLocation::getFromRawEncoding(DNLoc.CXXLiteralOperatorName.OpNameLoc),
2981 Record);
2982 break;
2983
2984 case DeclarationName::Identifier:
2985 case DeclarationName::ObjCZeroArgSelector:
2986 case DeclarationName::ObjCOneArgSelector:
2987 case DeclarationName::ObjCMultiArgSelector:
2988 case DeclarationName::CXXUsingDirective:
2989 break;
2990 }
2991}
2992
2993void ASTWriter::AddDeclarationNameInfo(const DeclarationNameInfo &NameInfo,
2994 RecordData &Record) {
2995 AddDeclarationName(NameInfo.getName(), Record);
2996 AddSourceLocation(NameInfo.getLoc(), Record);
2997 AddDeclarationNameLoc(NameInfo.getInfo(), NameInfo.getName(), Record);
2998}
2999
3000void ASTWriter::AddQualifierInfo(const QualifierInfo &Info,
3001 RecordData &Record) {
3002 AddNestedNameSpecifier(Info.NNS, Record);
3003 AddSourceRange(Info.NNSRange, Record);
3004 Record.push_back(Info.NumTemplParamLists);
3005 for (unsigned i=0, e=Info.NumTemplParamLists; i != e; ++i)
3006 AddTemplateParameterList(Info.TemplParamLists[i], Record);
3007}
3008
Sebastian Redl55c0ad52010-08-18 23:56:21 +00003009void ASTWriter::AddNestedNameSpecifier(NestedNameSpecifier *NNS,
Chris Lattnerca025db2010-05-07 21:43:38 +00003010 RecordData &Record) {
3011 // Nested name specifiers usually aren't too long. I think that 8 would
3012 // typically accomodate the vast majority.
3013 llvm::SmallVector<NestedNameSpecifier *, 8> NestedNames;
3014
3015 // Push each of the NNS's onto a stack for serialization in reverse order.
3016 while (NNS) {
3017 NestedNames.push_back(NNS);
3018 NNS = NNS->getPrefix();
3019 }
3020
3021 Record.push_back(NestedNames.size());
3022 while(!NestedNames.empty()) {
3023 NNS = NestedNames.pop_back_val();
3024 NestedNameSpecifier::SpecifierKind Kind = NNS->getKind();
3025 Record.push_back(Kind);
3026 switch (Kind) {
3027 case NestedNameSpecifier::Identifier:
3028 AddIdentifierRef(NNS->getAsIdentifier(), Record);
3029 break;
3030
3031 case NestedNameSpecifier::Namespace:
3032 AddDeclRef(NNS->getAsNamespace(), Record);
3033 break;
3034
3035 case NestedNameSpecifier::TypeSpec:
3036 case NestedNameSpecifier::TypeSpecWithTemplate:
3037 AddTypeRef(QualType(NNS->getAsType(), 0), Record);
3038 Record.push_back(Kind == NestedNameSpecifier::TypeSpecWithTemplate);
3039 break;
3040
3041 case NestedNameSpecifier::Global:
3042 // Don't need to write an associated value.
3043 break;
3044 }
3045 }
3046}
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +00003047
Sebastian Redl55c0ad52010-08-18 23:56:21 +00003048void ASTWriter::AddTemplateName(TemplateName Name, RecordData &Record) {
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00003049 TemplateName::NameKind Kind = Name.getKind();
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +00003050 Record.push_back(Kind);
3051 switch (Kind) {
3052 case TemplateName::Template:
3053 AddDeclRef(Name.getAsTemplateDecl(), Record);
3054 break;
3055
3056 case TemplateName::OverloadedTemplate: {
3057 OverloadedTemplateStorage *OvT = Name.getAsOverloadedTemplate();
3058 Record.push_back(OvT->size());
3059 for (OverloadedTemplateStorage::iterator I = OvT->begin(), E = OvT->end();
3060 I != E; ++I)
3061 AddDeclRef(*I, Record);
3062 break;
3063 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00003064
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +00003065 case TemplateName::QualifiedTemplate: {
3066 QualifiedTemplateName *QualT = Name.getAsQualifiedTemplateName();
3067 AddNestedNameSpecifier(QualT->getQualifier(), Record);
3068 Record.push_back(QualT->hasTemplateKeyword());
3069 AddDeclRef(QualT->getTemplateDecl(), Record);
3070 break;
3071 }
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00003072
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +00003073 case TemplateName::DependentTemplate: {
3074 DependentTemplateName *DepT = Name.getAsDependentTemplateName();
3075 AddNestedNameSpecifier(DepT->getQualifier(), Record);
3076 Record.push_back(DepT->isIdentifier());
3077 if (DepT->isIdentifier())
3078 AddIdentifierRef(DepT->getIdentifier(), Record);
3079 else
3080 Record.push_back(DepT->getOperator());
3081 break;
3082 }
3083 }
3084}
3085
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00003086void ASTWriter::AddTemplateArgument(const TemplateArgument &Arg,
Argyrios Kyrtzidis106caf922010-06-19 19:28:53 +00003087 RecordData &Record) {
3088 Record.push_back(Arg.getKind());
3089 switch (Arg.getKind()) {
3090 case TemplateArgument::Null:
3091 break;
3092 case TemplateArgument::Type:
3093 AddTypeRef(Arg.getAsType(), Record);
3094 break;
3095 case TemplateArgument::Declaration:
3096 AddDeclRef(Arg.getAsDecl(), Record);
3097 break;
3098 case TemplateArgument::Integral:
3099 AddAPSInt(*Arg.getAsIntegral(), Record);
3100 AddTypeRef(Arg.getIntegralType(), Record);
3101 break;
3102 case TemplateArgument::Template:
3103 AddTemplateName(Arg.getAsTemplate(), Record);
3104 break;
3105 case TemplateArgument::Expression:
3106 AddStmt(Arg.getAsExpr());
3107 break;
3108 case TemplateArgument::Pack:
3109 Record.push_back(Arg.pack_size());
3110 for (TemplateArgument::pack_iterator I=Arg.pack_begin(), E=Arg.pack_end();
3111 I != E; ++I)
3112 AddTemplateArgument(*I, Record);
3113 break;
3114 }
3115}
Argyrios Kyrtzidis818c5db2010-06-23 13:48:30 +00003116
3117void
Sebastian Redl55c0ad52010-08-18 23:56:21 +00003118ASTWriter::AddTemplateParameterList(const TemplateParameterList *TemplateParams,
Argyrios Kyrtzidis818c5db2010-06-23 13:48:30 +00003119 RecordData &Record) {
3120 assert(TemplateParams && "No TemplateParams!");
3121 AddSourceLocation(TemplateParams->getTemplateLoc(), Record);
3122 AddSourceLocation(TemplateParams->getLAngleLoc(), Record);
3123 AddSourceLocation(TemplateParams->getRAngleLoc(), Record);
3124 Record.push_back(TemplateParams->size());
3125 for (TemplateParameterList::const_iterator
3126 P = TemplateParams->begin(), PEnd = TemplateParams->end();
3127 P != PEnd; ++P)
3128 AddDeclRef(*P, Record);
3129}
3130
3131/// \brief Emit a template argument list.
3132void
Sebastian Redl55c0ad52010-08-18 23:56:21 +00003133ASTWriter::AddTemplateArgumentList(const TemplateArgumentList *TemplateArgs,
Argyrios Kyrtzidis818c5db2010-06-23 13:48:30 +00003134 RecordData &Record) {
3135 assert(TemplateArgs && "No TemplateArgs!");
3136 Record.push_back(TemplateArgs->flat_size());
3137 for (int i=0, e = TemplateArgs->flat_size(); i != e; ++i)
3138 AddTemplateArgument(TemplateArgs->get(i), Record);
3139}
Argyrios Kyrtzidis2c2167a2010-07-02 11:55:32 +00003140
3141
3142void
Sebastian Redl55c0ad52010-08-18 23:56:21 +00003143ASTWriter::AddUnresolvedSet(const UnresolvedSetImpl &Set, RecordData &Record) {
Argyrios Kyrtzidis2c2167a2010-07-02 11:55:32 +00003144 Record.push_back(Set.size());
3145 for (UnresolvedSetImpl::const_iterator
3146 I = Set.begin(), E = Set.end(); I != E; ++I) {
3147 AddDeclRef(I.getDecl(), Record);
3148 Record.push_back(I.getAccess());
3149 }
3150}
Argyrios Kyrtzidis3701fcd2010-07-02 23:30:27 +00003151
Sebastian Redl55c0ad52010-08-18 23:56:21 +00003152void ASTWriter::AddCXXBaseSpecifier(const CXXBaseSpecifier &Base,
Argyrios Kyrtzidis3701fcd2010-07-02 23:30:27 +00003153 RecordData &Record) {
3154 Record.push_back(Base.isVirtual());
3155 Record.push_back(Base.isBaseOfClass());
3156 Record.push_back(Base.getAccessSpecifierAsWritten());
Nick Lewycky19b9f952010-07-26 16:56:01 +00003157 AddTypeSourceInfo(Base.getTypeSourceInfo(), Record);
Argyrios Kyrtzidis3701fcd2010-07-02 23:30:27 +00003158 AddSourceRange(Base.getSourceRange(), Record);
3159}
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00003160
Sebastian Redl55c0ad52010-08-18 23:56:21 +00003161void ASTWriter::AddCXXBaseOrMemberInitializers(
Argyrios Kyrtzidis5b6a03f2010-08-09 10:54:12 +00003162 const CXXBaseOrMemberInitializer * const *BaseOrMembers,
3163 unsigned NumBaseOrMembers, RecordData &Record) {
3164 Record.push_back(NumBaseOrMembers);
3165 for (unsigned i=0; i != NumBaseOrMembers; ++i) {
3166 const CXXBaseOrMemberInitializer *Init = BaseOrMembers[i];
3167
3168 Record.push_back(Init->isBaseInitializer());
3169 if (Init->isBaseInitializer()) {
3170 AddTypeSourceInfo(Init->getBaseClassInfo(), Record);
3171 Record.push_back(Init->isBaseVirtual());
3172 } else {
3173 AddDeclRef(Init->getMember(), Record);
3174 }
3175 AddSourceLocation(Init->getMemberLocation(), Record);
3176 AddStmt(Init->getInit());
3177 AddDeclRef(Init->getAnonUnionMember(), Record);
3178 AddSourceLocation(Init->getLParenLoc(), Record);
3179 AddSourceLocation(Init->getRParenLoc(), Record);
3180 Record.push_back(Init->isWritten());
3181 if (Init->isWritten()) {
3182 Record.push_back(Init->getSourceOrder());
3183 } else {
3184 Record.push_back(Init->getNumArrayIndices());
3185 for (unsigned i=0, e=Init->getNumArrayIndices(); i != e; ++i)
3186 AddDeclRef(Init->getArrayIndex(i), Record);
3187 }
3188 }
3189}
3190
Sebastian Redl2c499f62010-08-18 23:56:43 +00003191void ASTWriter::SetReader(ASTReader *Reader) {
Sebastian Redl07a89a82010-07-30 00:29:29 +00003192 assert(Reader && "Cannot remove chain");
3193 assert(FirstDeclID == NextDeclID &&
3194 FirstTypeID == NextTypeID &&
3195 FirstIdentID == NextIdentID &&
Sebastian Redld95a56e2010-08-04 18:21:41 +00003196 FirstSelectorID == NextSelectorID &&
Douglas Gregor91096292010-10-02 19:29:26 +00003197 FirstMacroID == NextMacroID &&
Sebastian Redl07a89a82010-07-30 00:29:29 +00003198 "Setting chain after writing has started.");
3199 Chain = Reader;
3200}
3201
Sebastian Redl539c5062010-08-18 23:57:32 +00003202void ASTWriter::IdentifierRead(IdentID ID, IdentifierInfo *II) {
Sebastian Redlff4a2952010-07-23 23:49:55 +00003203 IdentifierIDs[II] = ID;
3204}
3205
Argyrios Kyrtzidisbb5c7eae2010-08-20 16:03:59 +00003206void ASTWriter::TypeRead(TypeIdx Idx, QualType T) {
Douglas Gregor9b3932c2010-10-05 18:37:06 +00003207 // Always take the highest-numbered type index. This copes with an interesting
3208 // case for chained AST writing where we schedule writing the type and then,
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00003209 // later, deserialize the type from another AST. In this case, we want to
Douglas Gregor9b3932c2010-10-05 18:37:06 +00003210 // keep the higher-numbered entry so that we can properly write it out to
3211 // the AST file.
3212 TypeIdx &StoredIdx = TypeIdxs[T];
3213 if (Idx.getIndex() >= StoredIdx.getIndex())
3214 StoredIdx = Idx;
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00003215}
3216
Sebastian Redl539c5062010-08-18 23:57:32 +00003217void ASTWriter::DeclRead(DeclID ID, const Decl *D) {
Sebastian Redl1ea025b2010-07-16 16:36:56 +00003218 DeclIDs[D] = ID;
Sebastian Redl85b2a6a2010-07-14 23:45:08 +00003219}
Sebastian Redl834bb972010-08-04 17:20:04 +00003220
Sebastian Redl539c5062010-08-18 23:57:32 +00003221void ASTWriter::SelectorRead(SelectorID ID, Selector S) {
Sebastian Redl834bb972010-08-04 17:20:04 +00003222 SelectorIDs[S] = ID;
3223}
Douglas Gregor91096292010-10-02 19:29:26 +00003224
Michael J. Spencer4c0ffa82010-10-21 03:16:25 +00003225void ASTWriter::MacroDefinitionRead(serialization::MacroID ID,
Douglas Gregor91096292010-10-02 19:29:26 +00003226 MacroDefinition *MD) {
3227 MacroDefinitions[MD] = ID;
3228}