DeclPrinter.cpp revision 15de72cf580840c61e5704c2f8a2b56f9d0638e1 
1//===--- DeclPrinter.cpp - Printing implementation for Decl ASTs ----------===//
2//
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//
10// This file implements the Decl::dump method, which pretty print the
11// AST back out to C/Objective-C/C++/Objective-C++ code.
12//
13//===----------------------------------------------------------------------===//
14#include "clang/AST/ASTContext.h"
15#include "clang/AST/DeclVisitor.h"
16#include "clang/AST/Decl.h"
17#include "clang/AST/DeclCXX.h"
18#include "clang/AST/DeclObjC.h"
19#include "clang/AST/Expr.h"
20#include "clang/AST/ExprCXX.h"
21#include "clang/AST/PrettyPrinter.h"
22#include "clang/Basic/Module.h"
23#include "llvm/Support/raw_ostream.h"
24using namespace clang;
25
26namespace {
27  class DeclPrinter : public DeclVisitor<DeclPrinter> {
28    raw_ostream &Out;
29    ASTContext &Context;
30    PrintingPolicy Policy;
31    unsigned Indentation;
32    bool PrintInstantiation;
33
34    raw_ostream& Indent() { return Indent(Indentation); }
35    raw_ostream& Indent(unsigned Indentation);
36    void ProcessDeclGroup(SmallVectorImpl<Decl*>& Decls);
37
38    void Print(AccessSpecifier AS);
39
40  public:
41    DeclPrinter(raw_ostream &Out, ASTContext &Context,
42                const PrintingPolicy &Policy,
43                unsigned Indentation = 0,
44                bool PrintInstantiation = false)
45      : Out(Out), Context(Context), Policy(Policy), Indentation(Indentation),
46        PrintInstantiation(PrintInstantiation) { }
47
48    void VisitDeclContext(DeclContext *DC, bool Indent = true);
49
50    void VisitTranslationUnitDecl(TranslationUnitDecl *D);
51    void VisitTypedefDecl(TypedefDecl *D);
52    void VisitTypeAliasDecl(TypeAliasDecl *D);
53    void VisitEnumDecl(EnumDecl *D);
54    void VisitRecordDecl(RecordDecl *D);
55    void VisitEnumConstantDecl(EnumConstantDecl *D);
56    void VisitFunctionDecl(FunctionDecl *D);
57    void VisitFieldDecl(FieldDecl *D);
58    void VisitVarDecl(VarDecl *D);
59    void VisitLabelDecl(LabelDecl *D);
60    void VisitParmVarDecl(ParmVarDecl *D);
61    void VisitFileScopeAsmDecl(FileScopeAsmDecl *D);
62    void VisitImportDecl(ImportDecl *D);
63    void VisitStaticAssertDecl(StaticAssertDecl *D);
64    void VisitNamespaceDecl(NamespaceDecl *D);
65    void VisitUsingDirectiveDecl(UsingDirectiveDecl *D);
66    void VisitNamespaceAliasDecl(NamespaceAliasDecl *D);
67    void VisitCXXRecordDecl(CXXRecordDecl *D);
68    void VisitLinkageSpecDecl(LinkageSpecDecl *D);
69    void VisitTemplateDecl(const TemplateDecl *D);
70    void VisitFunctionTemplateDecl(FunctionTemplateDecl *D);
71    void VisitClassTemplateDecl(ClassTemplateDecl *D);
72    void VisitObjCMethodDecl(ObjCMethodDecl *D);
73    void VisitObjCClassDecl(ObjCClassDecl *D);
74    void VisitObjCImplementationDecl(ObjCImplementationDecl *D);
75    void VisitObjCInterfaceDecl(ObjCInterfaceDecl *D);
76    void VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D);
77    void VisitObjCProtocolDecl(ObjCProtocolDecl *D);
78    void VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D);
79    void VisitObjCCategoryDecl(ObjCCategoryDecl *D);
80    void VisitObjCCompatibleAliasDecl(ObjCCompatibleAliasDecl *D);
81    void VisitObjCPropertyDecl(ObjCPropertyDecl *D);
82    void VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *D);
83    void VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D);
84    void VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D);
85    void VisitUsingDecl(UsingDecl *D);
86    void VisitUsingShadowDecl(UsingShadowDecl *D);
87
88    void PrintTemplateParameters(const TemplateParameterList *Params,
89                                 const TemplateArgumentList *Args);
90    void prettyPrintAttributes(Decl *D);
91  };
92}
93
94void Decl::print(raw_ostream &Out, unsigned Indentation,
95                 bool PrintInstantiation) const {
96  print(Out, getASTContext().getPrintingPolicy(), Indentation, PrintInstantiation);
97}
98
99void Decl::print(raw_ostream &Out, const PrintingPolicy &Policy,
100                 unsigned Indentation, bool PrintInstantiation) const {
101  DeclPrinter Printer(Out, getASTContext(), Policy, Indentation, PrintInstantiation);
102  Printer.Visit(const_cast<Decl*>(this));
103}
104
105static QualType GetBaseType(QualType T) {
106  // FIXME: This should be on the Type class!
107  QualType BaseType = T;
108  while (!BaseType->isSpecifierType()) {
109    if (isa<TypedefType>(BaseType))
110      break;
111    else if (const PointerType* PTy = BaseType->getAs<PointerType>())
112      BaseType = PTy->getPointeeType();
113    else if (const ArrayType* ATy = dyn_cast<ArrayType>(BaseType))
114      BaseType = ATy->getElementType();
115    else if (const FunctionType* FTy = BaseType->getAs<FunctionType>())
116      BaseType = FTy->getResultType();
117    else if (const VectorType *VTy = BaseType->getAs<VectorType>())
118      BaseType = VTy->getElementType();
119    else
120      llvm_unreachable("Unknown declarator!");
121  }
122  return BaseType;
123}
124
125static QualType getDeclType(Decl* D) {
126  if (TypedefNameDecl* TDD = dyn_cast<TypedefNameDecl>(D))
127    return TDD->getUnderlyingType();
128  if (ValueDecl* VD = dyn_cast<ValueDecl>(D))
129    return VD->getType();
130  return QualType();
131}
132
133void Decl::printGroup(Decl** Begin, unsigned NumDecls,
134                      raw_ostream &Out, const PrintingPolicy &Policy,
135                      unsigned Indentation) {
136  if (NumDecls == 1) {
137    (*Begin)->print(Out, Policy, Indentation);
138    return;
139  }
140
141  Decl** End = Begin + NumDecls;
142  TagDecl* TD = dyn_cast<TagDecl>(*Begin);
143  if (TD)
144    ++Begin;
145
146  PrintingPolicy SubPolicy(Policy);
147  if (TD && TD->isCompleteDefinition()) {
148    TD->print(Out, Policy, Indentation);
149    Out << " ";
150    SubPolicy.SuppressTag = true;
151  }
152
153  bool isFirst = true;
154  for ( ; Begin != End; ++Begin) {
155    if (isFirst) {
156      SubPolicy.SuppressSpecifiers = false;
157      isFirst = false;
158    } else {
159      if (!isFirst) Out << ", ";
160      SubPolicy.SuppressSpecifiers = true;
161    }
162
163    (*Begin)->print(Out, SubPolicy, Indentation);
164  }
165}
166
167void DeclContext::dumpDeclContext() const {
168  // Get the translation unit
169  const DeclContext *DC = this;
170  while (!DC->isTranslationUnit())
171    DC = DC->getParent();
172
173  ASTContext &Ctx = cast<TranslationUnitDecl>(DC)->getASTContext();
174  DeclPrinter Printer(llvm::errs(), Ctx, Ctx.getPrintingPolicy(), 0);
175  Printer.VisitDeclContext(const_cast<DeclContext *>(this), /*Indent=*/false);
176}
177
178void Decl::dump() const {
179  print(llvm::errs());
180}
181
182raw_ostream& DeclPrinter::Indent(unsigned Indentation) {
183  for (unsigned i = 0; i != Indentation; ++i)
184    Out << "  ";
185  return Out;
186}
187
188void DeclPrinter::prettyPrintAttributes(Decl *D) {
189  if (D->hasAttrs()) {
190    AttrVec &Attrs = D->getAttrs();
191    for (AttrVec::const_iterator i=Attrs.begin(), e=Attrs.end(); i!=e; ++i) {
192        Attr *A = *i;
193        A->printPretty(Out, Context);
194    }
195  }
196}
197
198void DeclPrinter::ProcessDeclGroup(SmallVectorImpl<Decl*>& Decls) {
199  this->Indent();
200  Decl::printGroup(Decls.data(), Decls.size(), Out, Policy, Indentation);
201  Out << ";\n";
202  Decls.clear();
203
204}
205
206void DeclPrinter::Print(AccessSpecifier AS) {
207  switch(AS) {
208  case AS_none:      llvm_unreachable("No access specifier!");
209  case AS_public:    Out << "public"; break;
210  case AS_protected: Out << "protected"; break;
211  case AS_private:   Out << "private"; break;
212  }
213}
214
215//----------------------------------------------------------------------------
216// Common C declarations
217//----------------------------------------------------------------------------
218
219void DeclPrinter::VisitDeclContext(DeclContext *DC, bool Indent) {
220  if (Indent)
221    Indentation += Policy.Indentation;
222
223  SmallVector<Decl*, 2> Decls;
224  for (DeclContext::decl_iterator D = DC->decls_begin(), DEnd = DC->decls_end();
225       D != DEnd; ++D) {
226
227    // Don't print ObjCIvarDecls, as they are printed when visiting the
228    // containing ObjCInterfaceDecl.
229    if (isa<ObjCIvarDecl>(*D))
230      continue;
231
232    if (!Policy.Dump) {
233      // Skip over implicit declarations in pretty-printing mode.
234      if (D->isImplicit()) continue;
235      // FIXME: Ugly hack so we don't pretty-print the builtin declaration
236      // of __builtin_va_list or __[u]int128_t.  There should be some other way
237      // to check that.
238      if (NamedDecl *ND = dyn_cast<NamedDecl>(*D)) {
239        if (IdentifierInfo *II = ND->getIdentifier()) {
240          if (II->isStr("__builtin_va_list") ||
241              II->isStr("__int128_t") || II->isStr("__uint128_t"))
242            continue;
243        }
244      }
245    }
246
247    // The next bits of code handles stuff like "struct {int x;} a,b"; we're
248    // forced to merge the declarations because there's no other way to
249    // refer to the struct in question.  This limited merging is safe without
250    // a bunch of other checks because it only merges declarations directly
251    // referring to the tag, not typedefs.
252    //
253    // Check whether the current declaration should be grouped with a previous
254    // unnamed struct.
255    QualType CurDeclType = getDeclType(*D);
256    if (!Decls.empty() && !CurDeclType.isNull()) {
257      QualType BaseType = GetBaseType(CurDeclType);
258      if (!BaseType.isNull() && isa<TagType>(BaseType) &&
259          cast<TagType>(BaseType)->getDecl() == Decls[0]) {
260        Decls.push_back(*D);
261        continue;
262      }
263    }
264
265    // If we have a merged group waiting to be handled, handle it now.
266    if (!Decls.empty())
267      ProcessDeclGroup(Decls);
268
269    // If the current declaration is an unnamed tag type, save it
270    // so we can merge it with the subsequent declaration(s) using it.
271    if (isa<TagDecl>(*D) && !cast<TagDecl>(*D)->getIdentifier()) {
272      Decls.push_back(*D);
273      continue;
274    }
275
276    if (isa<AccessSpecDecl>(*D)) {
277      Indentation -= Policy.Indentation;
278      this->Indent();
279      Print(D->getAccess());
280      Out << ":\n";
281      Indentation += Policy.Indentation;
282      continue;
283    }
284
285    this->Indent();
286    Visit(*D);
287
288    // FIXME: Need to be able to tell the DeclPrinter when
289    const char *Terminator = 0;
290    if (isa<FunctionDecl>(*D) &&
291        cast<FunctionDecl>(*D)->isThisDeclarationADefinition())
292      Terminator = 0;
293    else if (isa<ObjCMethodDecl>(*D) && cast<ObjCMethodDecl>(*D)->getBody())
294      Terminator = 0;
295    else if (isa<NamespaceDecl>(*D) || isa<LinkageSpecDecl>(*D) ||
296             isa<ObjCImplementationDecl>(*D) ||
297             isa<ObjCInterfaceDecl>(*D) ||
298             isa<ObjCProtocolDecl>(*D) ||
299             isa<ObjCCategoryImplDecl>(*D) ||
300             isa<ObjCCategoryDecl>(*D))
301      Terminator = 0;
302    else if (isa<EnumConstantDecl>(*D)) {
303      DeclContext::decl_iterator Next = D;
304      ++Next;
305      if (Next != DEnd)
306        Terminator = ",";
307    } else
308      Terminator = ";";
309
310    if (Terminator)
311      Out << Terminator;
312    Out << "\n";
313  }
314
315  if (!Decls.empty())
316    ProcessDeclGroup(Decls);
317
318  if (Indent)
319    Indentation -= Policy.Indentation;
320}
321
322void DeclPrinter::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
323  VisitDeclContext(D, false);
324}
325
326void DeclPrinter::VisitTypedefDecl(TypedefDecl *D) {
327  std::string S = D->getNameAsString();
328  D->getUnderlyingType().getAsStringInternal(S, Policy);
329  if (!Policy.SuppressSpecifiers) {
330    Out << "typedef ";
331
332    if (D->isModulePrivate())
333      Out << "__module_private__ ";
334  }
335  Out << S;
336  prettyPrintAttributes(D);
337}
338
339void DeclPrinter::VisitTypeAliasDecl(TypeAliasDecl *D) {
340  Out << "using " << D->getNameAsString() << " = "
341      << D->getUnderlyingType().getAsString(Policy);
342}
343
344void DeclPrinter::VisitEnumDecl(EnumDecl *D) {
345  if (!Policy.SuppressSpecifiers && D->isModulePrivate())
346    Out << "__module_private__ ";
347  Out << "enum ";
348  if (D->isScoped()) {
349    if (D->isScopedUsingClassTag())
350      Out << "class ";
351    else
352      Out << "struct ";
353  }
354  Out << *D;
355
356  if (D->isFixed()) {
357    std::string Underlying;
358    D->getIntegerType().getAsStringInternal(Underlying, Policy);
359    Out << " : " << Underlying;
360  }
361
362  if (D->isCompleteDefinition()) {
363    Out << " {\n";
364    VisitDeclContext(D);
365    Indent() << "}";
366  }
367  prettyPrintAttributes(D);
368}
369
370void DeclPrinter::VisitRecordDecl(RecordDecl *D) {
371  if (!Policy.SuppressSpecifiers && D->isModulePrivate())
372    Out << "__module_private__ ";
373  Out << D->getKindName();
374  if (D->getIdentifier())
375    Out << ' ' << *D;
376
377  if (D->isCompleteDefinition()) {
378    Out << " {\n";
379    VisitDeclContext(D);
380    Indent() << "}";
381  }
382}
383
384void DeclPrinter::VisitEnumConstantDecl(EnumConstantDecl *D) {
385  Out << *D;
386  if (Expr *Init = D->getInitExpr()) {
387    Out << " = ";
388    Init->printPretty(Out, Context, 0, Policy, Indentation);
389  }
390}
391
392void DeclPrinter::VisitFunctionDecl(FunctionDecl *D) {
393  if (!Policy.SuppressSpecifiers) {
394    switch (D->getStorageClassAsWritten()) {
395    case SC_None: break;
396    case SC_Extern: Out << "extern "; break;
397    case SC_Static: Out << "static "; break;
398    case SC_PrivateExtern: Out << "__private_extern__ "; break;
399    case SC_Auto: case SC_Register: case SC_OpenCLWorkGroupLocal:
400      llvm_unreachable("invalid for functions");
401    }
402
403    if (D->isInlineSpecified())  Out << "inline ";
404    if (D->isVirtualAsWritten()) Out << "virtual ";
405    if (D->isModulePrivate())    Out << "__module_private__ ";
406  }
407
408  PrintingPolicy SubPolicy(Policy);
409  SubPolicy.SuppressSpecifiers = false;
410  std::string Proto = D->getNameInfo().getAsString();
411
412  QualType Ty = D->getType();
413  while (const ParenType *PT = dyn_cast<ParenType>(Ty)) {
414    Proto = '(' + Proto + ')';
415    Ty = PT->getInnerType();
416  }
417
418  if (isa<FunctionType>(Ty)) {
419    const FunctionType *AFT = Ty->getAs<FunctionType>();
420    const FunctionProtoType *FT = 0;
421    if (D->hasWrittenPrototype())
422      FT = dyn_cast<FunctionProtoType>(AFT);
423
424    Proto += "(";
425    if (FT) {
426      llvm::raw_string_ostream POut(Proto);
427      DeclPrinter ParamPrinter(POut, Context, SubPolicy, Indentation);
428      for (unsigned i = 0, e = D->getNumParams(); i != e; ++i) {
429        if (i) POut << ", ";
430        ParamPrinter.VisitParmVarDecl(D->getParamDecl(i));
431      }
432
433      if (FT->isVariadic()) {
434        if (D->getNumParams()) POut << ", ";
435        POut << "...";
436      }
437    } else if (D->doesThisDeclarationHaveABody() && !D->hasPrototype()) {
438      for (unsigned i = 0, e = D->getNumParams(); i != e; ++i) {
439        if (i)
440          Proto += ", ";
441        Proto += D->getParamDecl(i)->getNameAsString();
442      }
443    }
444
445    Proto += ")";
446
447    if (FT && FT->getTypeQuals()) {
448      unsigned TypeQuals = FT->getTypeQuals();
449      if (TypeQuals & Qualifiers::Const)
450        Proto += " const";
451      if (TypeQuals & Qualifiers::Volatile)
452        Proto += " volatile";
453      if (TypeQuals & Qualifiers::Restrict)
454        Proto += " restrict";
455    }
456
457    if (FT && FT->hasDynamicExceptionSpec()) {
458      Proto += " throw(";
459      if (FT->getExceptionSpecType() == EST_MSAny)
460        Proto += "...";
461      else
462        for (unsigned I = 0, N = FT->getNumExceptions(); I != N; ++I) {
463          if (I)
464            Proto += ", ";
465
466          std::string ExceptionType;
467          FT->getExceptionType(I).getAsStringInternal(ExceptionType, SubPolicy);
468          Proto += ExceptionType;
469        }
470      Proto += ")";
471    } else if (FT && isNoexceptExceptionSpec(FT->getExceptionSpecType())) {
472      Proto += " noexcept";
473      if (FT->getExceptionSpecType() == EST_ComputedNoexcept) {
474        Proto += "(";
475        llvm::raw_string_ostream EOut(Proto);
476        FT->getNoexceptExpr()->printPretty(EOut, Context, 0, SubPolicy,
477                                           Indentation);
478        EOut.flush();
479        Proto += EOut.str();
480        Proto += ")";
481      }
482    }
483
484    if (CXXConstructorDecl *CDecl = dyn_cast<CXXConstructorDecl>(D)) {
485      bool HasInitializerList = false;
486      for (CXXConstructorDecl::init_const_iterator B = CDecl->init_begin(),
487           E = CDecl->init_end();
488           B != E; ++B) {
489        CXXCtorInitializer * BMInitializer = (*B);
490        if (BMInitializer->isInClassMemberInitializer())
491          continue;
492
493        if (!HasInitializerList) {
494          Proto += " : ";
495          Out << Proto;
496          Proto.clear();
497          HasInitializerList = true;
498        } else
499          Out << ", ";
500
501        if (BMInitializer->isAnyMemberInitializer()) {
502          FieldDecl *FD = BMInitializer->getAnyMember();
503          Out << *FD;
504        } else {
505          Out << QualType(BMInitializer->getBaseClass(), 0).getAsString(Policy);
506        }
507
508        Out << "(";
509        if (!BMInitializer->getInit()) {
510          // Nothing to print
511        } else {
512          Expr *Init = BMInitializer->getInit();
513          if (ExprWithCleanups *Tmp = dyn_cast<ExprWithCleanups>(Init))
514            Init = Tmp->getSubExpr();
515
516          Init = Init->IgnoreParens();
517
518          Expr *SimpleInit = 0;
519          Expr **Args = 0;
520          unsigned NumArgs = 0;
521          if (ParenListExpr *ParenList = dyn_cast<ParenListExpr>(Init)) {
522            Args = ParenList->getExprs();
523            NumArgs = ParenList->getNumExprs();
524          } else if (CXXConstructExpr *Construct
525                                        = dyn_cast<CXXConstructExpr>(Init)) {
526            Args = Construct->getArgs();
527            NumArgs = Construct->getNumArgs();
528          } else
529            SimpleInit = Init;
530
531          if (SimpleInit)
532            SimpleInit->printPretty(Out, Context, 0, Policy, Indentation);
533          else {
534            for (unsigned I = 0; I != NumArgs; ++I) {
535              if (isa<CXXDefaultArgExpr>(Args[I]))
536                break;
537
538              if (I)
539                Out << ", ";
540              Args[I]->printPretty(Out, Context, 0, Policy, Indentation);
541            }
542          }
543        }
544        Out << ")";
545      }
546    }
547    else
548      AFT->getResultType().getAsStringInternal(Proto, Policy);
549  } else {
550    Ty.getAsStringInternal(Proto, Policy);
551  }
552
553  Out << Proto;
554  prettyPrintAttributes(D);
555
556  if (D->isPure())
557    Out << " = 0";
558  else if (D->isDeletedAsWritten())
559    Out << " = delete";
560  else if (D->doesThisDeclarationHaveABody()) {
561    if (!D->hasPrototype() && D->getNumParams()) {
562      // This is a K&R function definition, so we need to print the
563      // parameters.
564      Out << '\n';
565      DeclPrinter ParamPrinter(Out, Context, SubPolicy, Indentation);
566      Indentation += Policy.Indentation;
567      for (unsigned i = 0, e = D->getNumParams(); i != e; ++i) {
568        Indent();
569        ParamPrinter.VisitParmVarDecl(D->getParamDecl(i));
570        Out << ";\n";
571      }
572      Indentation -= Policy.Indentation;
573    } else
574      Out << ' ';
575
576    D->getBody()->printPretty(Out, Context, 0, SubPolicy, Indentation);
577    Out << '\n';
578  }
579}
580
581void DeclPrinter::VisitFieldDecl(FieldDecl *D) {
582  if (!Policy.SuppressSpecifiers && D->isMutable())
583    Out << "mutable ";
584  if (!Policy.SuppressSpecifiers && D->isModulePrivate())
585    Out << "__module_private__ ";
586
587  std::string Name = D->getNameAsString();
588  D->getType().getAsStringInternal(Name, Policy);
589  Out << Name;
590
591  if (D->isBitField()) {
592    Out << " : ";
593    D->getBitWidth()->printPretty(Out, Context, 0, Policy, Indentation);
594  }
595
596  Expr *Init = D->getInClassInitializer();
597  if (!Policy.SuppressInitializers && Init) {
598    Out << " = ";
599    Init->printPretty(Out, Context, 0, Policy, Indentation);
600  }
601  prettyPrintAttributes(D);
602}
603
604void DeclPrinter::VisitLabelDecl(LabelDecl *D) {
605  Out << D->getNameAsString() << ":";
606}
607
608
609void DeclPrinter::VisitVarDecl(VarDecl *D) {
610  StorageClass SCAsWritten = D->getStorageClassAsWritten();
611  if (!Policy.SuppressSpecifiers && SCAsWritten != SC_None)
612    Out << VarDecl::getStorageClassSpecifierString(SCAsWritten) << " ";
613
614  if (!Policy.SuppressSpecifiers && D->isThreadSpecified())
615    Out << "__thread ";
616  if (!Policy.SuppressSpecifiers && D->isModulePrivate())
617    Out << "__module_private__ ";
618
619  std::string Name = D->getNameAsString();
620  QualType T = D->getType();
621  if (ParmVarDecl *Parm = dyn_cast<ParmVarDecl>(D))
622    T = Parm->getOriginalType();
623  T.getAsStringInternal(Name, Policy);
624  Out << Name;
625  Expr *Init = D->getInit();
626  if (!Policy.SuppressInitializers && Init) {
627    if (D->hasCXXDirectInitializer())
628      Out << "(";
629    else {
630        CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(Init);
631        if (!CCE || CCE->getConstructor()->isCopyOrMoveConstructor())
632          Out << " = ";
633    }
634    Init->printPretty(Out, Context, 0, Policy, Indentation);
635    if (D->hasCXXDirectInitializer())
636      Out << ")";
637  }
638  prettyPrintAttributes(D);
639}
640
641void DeclPrinter::VisitParmVarDecl(ParmVarDecl *D) {
642  VisitVarDecl(D);
643}
644
645void DeclPrinter::VisitFileScopeAsmDecl(FileScopeAsmDecl *D) {
646  Out << "__asm (";
647  D->getAsmString()->printPretty(Out, Context, 0, Policy, Indentation);
648  Out << ")";
649}
650
651void DeclPrinter::VisitImportDecl(ImportDecl *D) {
652  Out << "__import_module__ " << D->getImportedModule()->getFullModuleName()
653      << ";\n";
654}
655
656void DeclPrinter::VisitStaticAssertDecl(StaticAssertDecl *D) {
657  Out << "static_assert(";
658  D->getAssertExpr()->printPretty(Out, Context, 0, Policy, Indentation);
659  Out << ", ";
660  D->getMessage()->printPretty(Out, Context, 0, Policy, Indentation);
661  Out << ")";
662}
663
664//----------------------------------------------------------------------------
665// C++ declarations
666//----------------------------------------------------------------------------
667void DeclPrinter::VisitNamespaceDecl(NamespaceDecl *D) {
668  Out << "namespace " << *D << " {\n";
669  VisitDeclContext(D);
670  Indent() << "}";
671}
672
673void DeclPrinter::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
674  Out << "using namespace ";
675  if (D->getQualifier())
676    D->getQualifier()->print(Out, Policy);
677  Out << *D->getNominatedNamespaceAsWritten();
678}
679
680void DeclPrinter::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
681  Out << "namespace " << *D << " = ";
682  if (D->getQualifier())
683    D->getQualifier()->print(Out, Policy);
684  Out << *D->getAliasedNamespace();
685}
686
687void DeclPrinter::VisitCXXRecordDecl(CXXRecordDecl *D) {
688  if (!Policy.SuppressSpecifiers && D->isModulePrivate())
689    Out << "__module_private__ ";
690  Out << D->getKindName();
691  if (D->getIdentifier())
692    Out << ' ' << *D;
693
694  if (D->isCompleteDefinition()) {
695    // Print the base classes
696    if (D->getNumBases()) {
697      Out << " : ";
698      for (CXXRecordDecl::base_class_iterator Base = D->bases_begin(),
699             BaseEnd = D->bases_end(); Base != BaseEnd; ++Base) {
700        if (Base != D->bases_begin())
701          Out << ", ";
702
703        if (Base->isVirtual())
704          Out << "virtual ";
705
706        AccessSpecifier AS = Base->getAccessSpecifierAsWritten();
707        if (AS != AS_none)
708          Print(AS);
709        Out << " " << Base->getType().getAsString(Policy);
710
711        if (Base->isPackExpansion())
712          Out << "...";
713      }
714    }
715
716    // Print the class definition
717    // FIXME: Doesn't print access specifiers, e.g., "public:"
718    Out << " {\n";
719    VisitDeclContext(D);
720    Indent() << "}";
721  }
722}
723
724void DeclPrinter::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
725  const char *l;
726  if (D->getLanguage() == LinkageSpecDecl::lang_c)
727    l = "C";
728  else {
729    assert(D->getLanguage() == LinkageSpecDecl::lang_cxx &&
730           "unknown language in linkage specification");
731    l = "C++";
732  }
733
734  Out << "extern \"" << l << "\" ";
735  if (D->hasBraces()) {
736    Out << "{\n";
737    VisitDeclContext(D);
738    Indent() << "}";
739  } else
740    Visit(*D->decls_begin());
741}
742
743void DeclPrinter::PrintTemplateParameters(
744    const TemplateParameterList *Params, const TemplateArgumentList *Args = 0) {
745  assert(Params);
746  assert(!Args || Params->size() == Args->size());
747
748  Out << "template <";
749
750  for (unsigned i = 0, e = Params->size(); i != e; ++i) {
751    if (i != 0)
752      Out << ", ";
753
754    const Decl *Param = Params->getParam(i);
755    if (const TemplateTypeParmDecl *TTP =
756          dyn_cast<TemplateTypeParmDecl>(Param)) {
757
758      if (TTP->wasDeclaredWithTypename())
759        Out << "typename ";
760      else
761        Out << "class ";
762
763      if (TTP->isParameterPack())
764        Out << "... ";
765
766      Out << TTP->getNameAsString();
767
768      if (Args) {
769        Out << " = ";
770        Args->get(i).print(Policy, Out);
771      } else if (TTP->hasDefaultArgument()) {
772        Out << " = ";
773        Out << TTP->getDefaultArgument().getAsString(Policy);
774      };
775    } else if (const NonTypeTemplateParmDecl *NTTP =
776                 dyn_cast<NonTypeTemplateParmDecl>(Param)) {
777      Out << NTTP->getType().getAsString(Policy);
778
779      if (NTTP->isParameterPack() && !isa<PackExpansionType>(NTTP->getType()))
780        Out << "...";
781
782      if (IdentifierInfo *Name = NTTP->getIdentifier()) {
783        Out << ' ';
784        Out << Name->getName();
785      }
786
787      if (Args) {
788        Out << " = ";
789        Args->get(i).print(Policy, Out);
790      } else if (NTTP->hasDefaultArgument()) {
791        Out << " = ";
792        NTTP->getDefaultArgument()->printPretty(Out, Context, 0, Policy,
793                                                Indentation);
794      }
795    } else if (const TemplateTemplateParmDecl *TTPD =
796                 dyn_cast<TemplateTemplateParmDecl>(Param)) {
797      VisitTemplateDecl(TTPD);
798      // FIXME: print the default argument, if present.
799    }
800  }
801
802  Out << "> ";
803}
804
805void DeclPrinter::VisitTemplateDecl(const TemplateDecl *D) {
806  PrintTemplateParameters(D->getTemplateParameters());
807
808  if (const TemplateTemplateParmDecl *TTP =
809        dyn_cast<TemplateTemplateParmDecl>(D)) {
810    Out << "class ";
811    if (TTP->isParameterPack())
812      Out << "...";
813    Out << D->getName();
814  } else {
815    Visit(D->getTemplatedDecl());
816  }
817}
818
819void DeclPrinter::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
820  if (PrintInstantiation) {
821    TemplateParameterList *Params = D->getTemplateParameters();
822    for (FunctionTemplateDecl::spec_iterator I = D->spec_begin(), E = D->spec_end();
823         I != E; ++I) {
824      PrintTemplateParameters(Params, (*I)->getTemplateSpecializationArgs());
825      Visit(*I);
826    }
827  }
828
829  return VisitRedeclarableTemplateDecl(D);
830}
831
832void DeclPrinter::VisitClassTemplateDecl(ClassTemplateDecl *D) {
833  if (PrintInstantiation) {
834    TemplateParameterList *Params = D->getTemplateParameters();
835    for (ClassTemplateDecl::spec_iterator I = D->spec_begin(), E = D->spec_end();
836         I != E; ++I) {
837      PrintTemplateParameters(Params, &(*I)->getTemplateArgs());
838      Visit(*I);
839      Out << '\n';
840    }
841  }
842
843  return VisitRedeclarableTemplateDecl(D);
844}
845
846//----------------------------------------------------------------------------
847// Objective-C declarations
848//----------------------------------------------------------------------------
849
850void DeclPrinter::VisitObjCClassDecl(ObjCClassDecl *D) {
851  Out << "@class " << *D->getForwardInterfaceDecl();
852}
853
854void DeclPrinter::VisitObjCMethodDecl(ObjCMethodDecl *OMD) {
855  if (OMD->isInstanceMethod())
856    Out << "- ";
857  else
858    Out << "+ ";
859  if (!OMD->getResultType().isNull())
860    Out << '(' << OMD->getResultType().getAsString(Policy) << ")";
861
862  std::string name = OMD->getSelector().getAsString();
863  std::string::size_type pos, lastPos = 0;
864  for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(),
865       E = OMD->param_end(); PI != E; ++PI) {
866    // FIXME: selector is missing here!
867    pos = name.find_first_of(':', lastPos);
868    Out << " " << name.substr(lastPos, pos - lastPos);
869    Out << ":(" << (*PI)->getType().getAsString(Policy) << ')' << **PI;
870    lastPos = pos + 1;
871  }
872
873  if (OMD->param_begin() == OMD->param_end())
874    Out << " " << name;
875
876  if (OMD->isVariadic())
877      Out << ", ...";
878
879  if (OMD->getBody()) {
880    Out << ' ';
881    OMD->getBody()->printPretty(Out, Context, 0, Policy);
882    Out << '\n';
883  }
884}
885
886void DeclPrinter::VisitObjCImplementationDecl(ObjCImplementationDecl *OID) {
887  std::string I = OID->getNameAsString();
888  ObjCInterfaceDecl *SID = OID->getSuperClass();
889
890  if (SID)
891    Out << "@implementation " << I << " : " << *SID;
892  else
893    Out << "@implementation " << I;
894  Out << "\n";
895  VisitDeclContext(OID, false);
896  Out << "@end";
897}
898
899void DeclPrinter::VisitObjCInterfaceDecl(ObjCInterfaceDecl *OID) {
900  std::string I = OID->getNameAsString();
901  ObjCInterfaceDecl *SID = OID->getSuperClass();
902
903  if (SID)
904    Out << "@interface " << I << " : " << *SID;
905  else
906    Out << "@interface " << I;
907
908  // Protocols?
909  const ObjCList<ObjCProtocolDecl> &Protocols = OID->getReferencedProtocols();
910  if (!Protocols.empty()) {
911    for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
912         E = Protocols.end(); I != E; ++I)
913      Out << (I == Protocols.begin() ? '<' : ',') << **I;
914  }
915
916  if (!Protocols.empty())
917    Out << "> ";
918
919  if (OID->ivar_size() > 0) {
920    Out << "{\n";
921    Indentation += Policy.Indentation;
922    for (ObjCInterfaceDecl::ivar_iterator I = OID->ivar_begin(),
923         E = OID->ivar_end(); I != E; ++I) {
924      Indent() << (*I)->getType().getAsString(Policy) << ' ' << **I << ";\n";
925    }
926    Indentation -= Policy.Indentation;
927    Out << "}\n";
928  }
929
930  VisitDeclContext(OID, false);
931  Out << "@end";
932  // FIXME: implement the rest...
933}
934
935void DeclPrinter::VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D) {
936  Out << "@protocol ";
937  for (ObjCForwardProtocolDecl::protocol_iterator I = D->protocol_begin(),
938         E = D->protocol_end();
939       I != E; ++I) {
940    if (I != D->protocol_begin()) Out << ", ";
941    Out << **I;
942  }
943}
944
945void DeclPrinter::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
946  Out << "@protocol " << *PID << '\n';
947  VisitDeclContext(PID, false);
948  Out << "@end";
949}
950
951void DeclPrinter::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *PID) {
952  Out << "@implementation " << *PID->getClassInterface() << '(' << *PID <<")\n";
953
954  VisitDeclContext(PID, false);
955  Out << "@end";
956  // FIXME: implement the rest...
957}
958
959void DeclPrinter::VisitObjCCategoryDecl(ObjCCategoryDecl *PID) {
960  Out << "@interface " << *PID->getClassInterface() << '(' << *PID << ")\n";
961  VisitDeclContext(PID, false);
962  Out << "@end";
963
964  // FIXME: implement the rest...
965}
966
967void DeclPrinter::VisitObjCCompatibleAliasDecl(ObjCCompatibleAliasDecl *AID) {
968  Out << "@compatibility_alias " << *AID
969      << ' ' << *AID->getClassInterface() << ";\n";
970}
971
972/// PrintObjCPropertyDecl - print a property declaration.
973///
974void DeclPrinter::VisitObjCPropertyDecl(ObjCPropertyDecl *PDecl) {
975  if (PDecl->getPropertyImplementation() == ObjCPropertyDecl::Required)
976    Out << "@required\n";
977  else if (PDecl->getPropertyImplementation() == ObjCPropertyDecl::Optional)
978    Out << "@optional\n";
979
980  Out << "@property";
981  if (PDecl->getPropertyAttributes() != ObjCPropertyDecl::OBJC_PR_noattr) {
982    bool first = true;
983    Out << " (";
984    if (PDecl->getPropertyAttributes() &
985        ObjCPropertyDecl::OBJC_PR_readonly) {
986      Out << (first ? ' ' : ',') << "readonly";
987      first = false;
988    }
989
990    if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_getter) {
991      Out << (first ? ' ' : ',') << "getter = "
992          << PDecl->getGetterName().getAsString();
993      first = false;
994    }
995    if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_setter) {
996      Out << (first ? ' ' : ',') << "setter = "
997          << PDecl->getSetterName().getAsString();
998      first = false;
999    }
1000
1001    if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_assign) {
1002      Out << (first ? ' ' : ',') << "assign";
1003      first = false;
1004    }
1005
1006    if (PDecl->getPropertyAttributes() &
1007        ObjCPropertyDecl::OBJC_PR_readwrite) {
1008      Out << (first ? ' ' : ',') << "readwrite";
1009      first = false;
1010    }
1011
1012    if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_retain) {
1013      Out << (first ? ' ' : ',') << "retain";
1014      first = false;
1015    }
1016
1017    if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_strong) {
1018      Out << (first ? ' ' : ',') << "strong";
1019      first = false;
1020    }
1021
1022    if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_copy) {
1023      Out << (first ? ' ' : ',') << "copy";
1024      first = false;
1025    }
1026
1027    if (PDecl->getPropertyAttributes() &
1028        ObjCPropertyDecl::OBJC_PR_nonatomic) {
1029      Out << (first ? ' ' : ',') << "nonatomic";
1030      first = false;
1031    }
1032    if (PDecl->getPropertyAttributes() &
1033        ObjCPropertyDecl::OBJC_PR_atomic) {
1034      Out << (first ? ' ' : ',') << "atomic";
1035      first = false;
1036    }
1037
1038    (void) first; // Silence dead store warning due to idiomatic code.
1039    Out << " )";
1040  }
1041  Out << ' ' << PDecl->getType().getAsString(Policy) << ' ' << *PDecl;
1042}
1043
1044void DeclPrinter::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PID) {
1045  if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Synthesize)
1046    Out << "@synthesize ";
1047  else
1048    Out << "@dynamic ";
1049  Out << *PID->getPropertyDecl();
1050  if (PID->getPropertyIvarDecl())
1051    Out << '=' << *PID->getPropertyIvarDecl();
1052}
1053
1054void DeclPrinter::VisitUsingDecl(UsingDecl *D) {
1055  Out << "using ";
1056  D->getQualifier()->print(Out, Policy);
1057  Out << *D;
1058}
1059
1060void
1061DeclPrinter::VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D) {
1062  Out << "using typename ";
1063  D->getQualifier()->print(Out, Policy);
1064  Out << D->getDeclName();
1065}
1066
1067void DeclPrinter::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1068  Out << "using ";
1069  D->getQualifier()->print(Out, Policy);
1070  Out << D->getDeclName();
1071}
1072
1073void DeclPrinter::VisitUsingShadowDecl(UsingShadowDecl *D) {
1074  // ignore
1075}
1076