xref: /aosp_15_r20/external/clang/lib/AST/DeclPrinter.cpp (revision 67e74705e28f6214e480b399dd47ea732279e315)
1*67e74705SXin Li //===--- DeclPrinter.cpp - Printing implementation for Decl ASTs ----------===//
2*67e74705SXin Li //
3*67e74705SXin Li //                     The LLVM Compiler Infrastructure
4*67e74705SXin Li //
5*67e74705SXin Li // This file is distributed under the University of Illinois Open Source
6*67e74705SXin Li // License. See LICENSE.TXT for details.
7*67e74705SXin Li //
8*67e74705SXin Li //===----------------------------------------------------------------------===//
9*67e74705SXin Li //
10*67e74705SXin Li // This file implements the Decl::print method, which pretty prints the
11*67e74705SXin Li // AST back out to C/Objective-C/C++/Objective-C++ code.
12*67e74705SXin Li //
13*67e74705SXin Li //===----------------------------------------------------------------------===//
14*67e74705SXin Li #include "clang/AST/ASTContext.h"
15*67e74705SXin Li #include "clang/AST/Attr.h"
16*67e74705SXin Li #include "clang/AST/Decl.h"
17*67e74705SXin Li #include "clang/AST/DeclCXX.h"
18*67e74705SXin Li #include "clang/AST/DeclObjC.h"
19*67e74705SXin Li #include "clang/AST/DeclVisitor.h"
20*67e74705SXin Li #include "clang/AST/Expr.h"
21*67e74705SXin Li #include "clang/AST/ExprCXX.h"
22*67e74705SXin Li #include "clang/AST/PrettyPrinter.h"
23*67e74705SXin Li #include "clang/Basic/Module.h"
24*67e74705SXin Li #include "llvm/Support/raw_ostream.h"
25*67e74705SXin Li using namespace clang;
26*67e74705SXin Li 
27*67e74705SXin Li namespace {
28*67e74705SXin Li   class DeclPrinter : public DeclVisitor<DeclPrinter> {
29*67e74705SXin Li     raw_ostream &Out;
30*67e74705SXin Li     PrintingPolicy Policy;
31*67e74705SXin Li     unsigned Indentation;
32*67e74705SXin Li     bool PrintInstantiation;
33*67e74705SXin Li 
Indent()34*67e74705SXin Li     raw_ostream& Indent() { return Indent(Indentation); }
35*67e74705SXin Li     raw_ostream& Indent(unsigned Indentation);
36*67e74705SXin Li     void ProcessDeclGroup(SmallVectorImpl<Decl*>& Decls);
37*67e74705SXin Li 
38*67e74705SXin Li     void Print(AccessSpecifier AS);
39*67e74705SXin Li 
40*67e74705SXin Li     /// Print an Objective-C method type in parentheses.
41*67e74705SXin Li     ///
42*67e74705SXin Li     /// \param Quals The Objective-C declaration qualifiers.
43*67e74705SXin Li     /// \param T The type to print.
44*67e74705SXin Li     void PrintObjCMethodType(ASTContext &Ctx, Decl::ObjCDeclQualifier Quals,
45*67e74705SXin Li                              QualType T);
46*67e74705SXin Li 
47*67e74705SXin Li     void PrintObjCTypeParams(ObjCTypeParamList *Params);
48*67e74705SXin Li 
49*67e74705SXin Li   public:
DeclPrinter(raw_ostream & Out,const PrintingPolicy & Policy,unsigned Indentation=0,bool PrintInstantiation=false)50*67e74705SXin Li     DeclPrinter(raw_ostream &Out, const PrintingPolicy &Policy,
51*67e74705SXin Li                 unsigned Indentation = 0, bool PrintInstantiation = false)
52*67e74705SXin Li       : Out(Out), Policy(Policy), Indentation(Indentation),
53*67e74705SXin Li         PrintInstantiation(PrintInstantiation) { }
54*67e74705SXin Li 
55*67e74705SXin Li     void VisitDeclContext(DeclContext *DC, bool Indent = true);
56*67e74705SXin Li 
57*67e74705SXin Li     void VisitTranslationUnitDecl(TranslationUnitDecl *D);
58*67e74705SXin Li     void VisitTypedefDecl(TypedefDecl *D);
59*67e74705SXin Li     void VisitTypeAliasDecl(TypeAliasDecl *D);
60*67e74705SXin Li     void VisitEnumDecl(EnumDecl *D);
61*67e74705SXin Li     void VisitRecordDecl(RecordDecl *D);
62*67e74705SXin Li     void VisitEnumConstantDecl(EnumConstantDecl *D);
63*67e74705SXin Li     void VisitEmptyDecl(EmptyDecl *D);
64*67e74705SXin Li     void VisitFunctionDecl(FunctionDecl *D);
65*67e74705SXin Li     void VisitFriendDecl(FriendDecl *D);
66*67e74705SXin Li     void VisitFieldDecl(FieldDecl *D);
67*67e74705SXin Li     void VisitVarDecl(VarDecl *D);
68*67e74705SXin Li     void VisitLabelDecl(LabelDecl *D);
69*67e74705SXin Li     void VisitParmVarDecl(ParmVarDecl *D);
70*67e74705SXin Li     void VisitFileScopeAsmDecl(FileScopeAsmDecl *D);
71*67e74705SXin Li     void VisitImportDecl(ImportDecl *D);
72*67e74705SXin Li     void VisitStaticAssertDecl(StaticAssertDecl *D);
73*67e74705SXin Li     void VisitNamespaceDecl(NamespaceDecl *D);
74*67e74705SXin Li     void VisitUsingDirectiveDecl(UsingDirectiveDecl *D);
75*67e74705SXin Li     void VisitNamespaceAliasDecl(NamespaceAliasDecl *D);
76*67e74705SXin Li     void VisitCXXRecordDecl(CXXRecordDecl *D);
77*67e74705SXin Li     void VisitLinkageSpecDecl(LinkageSpecDecl *D);
78*67e74705SXin Li     void VisitTemplateDecl(const TemplateDecl *D);
79*67e74705SXin Li     void VisitFunctionTemplateDecl(FunctionTemplateDecl *D);
80*67e74705SXin Li     void VisitClassTemplateDecl(ClassTemplateDecl *D);
81*67e74705SXin Li     void VisitObjCMethodDecl(ObjCMethodDecl *D);
82*67e74705SXin Li     void VisitObjCImplementationDecl(ObjCImplementationDecl *D);
83*67e74705SXin Li     void VisitObjCInterfaceDecl(ObjCInterfaceDecl *D);
84*67e74705SXin Li     void VisitObjCProtocolDecl(ObjCProtocolDecl *D);
85*67e74705SXin Li     void VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D);
86*67e74705SXin Li     void VisitObjCCategoryDecl(ObjCCategoryDecl *D);
87*67e74705SXin Li     void VisitObjCCompatibleAliasDecl(ObjCCompatibleAliasDecl *D);
88*67e74705SXin Li     void VisitObjCPropertyDecl(ObjCPropertyDecl *D);
89*67e74705SXin Li     void VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *D);
90*67e74705SXin Li     void VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D);
91*67e74705SXin Li     void VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D);
92*67e74705SXin Li     void VisitUsingDecl(UsingDecl *D);
93*67e74705SXin Li     void VisitUsingShadowDecl(UsingShadowDecl *D);
94*67e74705SXin Li     void VisitOMPThreadPrivateDecl(OMPThreadPrivateDecl *D);
95*67e74705SXin Li     void VisitOMPDeclareReductionDecl(OMPDeclareReductionDecl *D);
96*67e74705SXin Li     void VisitOMPCapturedExprDecl(OMPCapturedExprDecl *D);
97*67e74705SXin Li 
98*67e74705SXin Li     void PrintTemplateParameters(const TemplateParameterList *Params,
99*67e74705SXin Li                                  const TemplateArgumentList *Args = nullptr);
100*67e74705SXin Li     void prettyPrintAttributes(Decl *D);
101*67e74705SXin Li     void prettyPrintPragmas(Decl *D);
102*67e74705SXin Li     void printDeclType(QualType T, StringRef DeclName, bool Pack = false);
103*67e74705SXin Li   };
104*67e74705SXin Li }
105*67e74705SXin Li 
print(raw_ostream & Out,unsigned Indentation,bool PrintInstantiation) const106*67e74705SXin Li void Decl::print(raw_ostream &Out, unsigned Indentation,
107*67e74705SXin Li                  bool PrintInstantiation) const {
108*67e74705SXin Li   print(Out, getASTContext().getPrintingPolicy(), Indentation, PrintInstantiation);
109*67e74705SXin Li }
110*67e74705SXin Li 
print(raw_ostream & Out,const PrintingPolicy & Policy,unsigned Indentation,bool PrintInstantiation) const111*67e74705SXin Li void Decl::print(raw_ostream &Out, const PrintingPolicy &Policy,
112*67e74705SXin Li                  unsigned Indentation, bool PrintInstantiation) const {
113*67e74705SXin Li   DeclPrinter Printer(Out, Policy, Indentation, PrintInstantiation);
114*67e74705SXin Li   Printer.Visit(const_cast<Decl*>(this));
115*67e74705SXin Li }
116*67e74705SXin Li 
GetBaseType(QualType T)117*67e74705SXin Li static QualType GetBaseType(QualType T) {
118*67e74705SXin Li   // FIXME: This should be on the Type class!
119*67e74705SXin Li   QualType BaseType = T;
120*67e74705SXin Li   while (!BaseType->isSpecifierType()) {
121*67e74705SXin Li     if (isa<TypedefType>(BaseType))
122*67e74705SXin Li       break;
123*67e74705SXin Li     else if (const PointerType* PTy = BaseType->getAs<PointerType>())
124*67e74705SXin Li       BaseType = PTy->getPointeeType();
125*67e74705SXin Li     else if (const BlockPointerType *BPy = BaseType->getAs<BlockPointerType>())
126*67e74705SXin Li       BaseType = BPy->getPointeeType();
127*67e74705SXin Li     else if (const ArrayType* ATy = dyn_cast<ArrayType>(BaseType))
128*67e74705SXin Li       BaseType = ATy->getElementType();
129*67e74705SXin Li     else if (const FunctionType* FTy = BaseType->getAs<FunctionType>())
130*67e74705SXin Li       BaseType = FTy->getReturnType();
131*67e74705SXin Li     else if (const VectorType *VTy = BaseType->getAs<VectorType>())
132*67e74705SXin Li       BaseType = VTy->getElementType();
133*67e74705SXin Li     else if (const ReferenceType *RTy = BaseType->getAs<ReferenceType>())
134*67e74705SXin Li       BaseType = RTy->getPointeeType();
135*67e74705SXin Li     else if (const AutoType *ATy = BaseType->getAs<AutoType>())
136*67e74705SXin Li       BaseType = ATy->getDeducedType();
137*67e74705SXin Li     else
138*67e74705SXin Li       llvm_unreachable("Unknown declarator!");
139*67e74705SXin Li   }
140*67e74705SXin Li   return BaseType;
141*67e74705SXin Li }
142*67e74705SXin Li 
getDeclType(Decl * D)143*67e74705SXin Li static QualType getDeclType(Decl* D) {
144*67e74705SXin Li   if (TypedefNameDecl* TDD = dyn_cast<TypedefNameDecl>(D))
145*67e74705SXin Li     return TDD->getUnderlyingType();
146*67e74705SXin Li   if (ValueDecl* VD = dyn_cast<ValueDecl>(D))
147*67e74705SXin Li     return VD->getType();
148*67e74705SXin Li   return QualType();
149*67e74705SXin Li }
150*67e74705SXin Li 
printGroup(Decl ** Begin,unsigned NumDecls,raw_ostream & Out,const PrintingPolicy & Policy,unsigned Indentation)151*67e74705SXin Li void Decl::printGroup(Decl** Begin, unsigned NumDecls,
152*67e74705SXin Li                       raw_ostream &Out, const PrintingPolicy &Policy,
153*67e74705SXin Li                       unsigned Indentation) {
154*67e74705SXin Li   if (NumDecls == 1) {
155*67e74705SXin Li     (*Begin)->print(Out, Policy, Indentation);
156*67e74705SXin Li     return;
157*67e74705SXin Li   }
158*67e74705SXin Li 
159*67e74705SXin Li   Decl** End = Begin + NumDecls;
160*67e74705SXin Li   TagDecl* TD = dyn_cast<TagDecl>(*Begin);
161*67e74705SXin Li   if (TD)
162*67e74705SXin Li     ++Begin;
163*67e74705SXin Li 
164*67e74705SXin Li   PrintingPolicy SubPolicy(Policy);
165*67e74705SXin Li 
166*67e74705SXin Li   bool isFirst = true;
167*67e74705SXin Li   for ( ; Begin != End; ++Begin) {
168*67e74705SXin Li     if (isFirst) {
169*67e74705SXin Li       if(TD)
170*67e74705SXin Li         SubPolicy.IncludeTagDefinition = true;
171*67e74705SXin Li       SubPolicy.SuppressSpecifiers = false;
172*67e74705SXin Li       isFirst = false;
173*67e74705SXin Li     } else {
174*67e74705SXin Li       if (!isFirst) Out << ", ";
175*67e74705SXin Li       SubPolicy.IncludeTagDefinition = false;
176*67e74705SXin Li       SubPolicy.SuppressSpecifiers = true;
177*67e74705SXin Li     }
178*67e74705SXin Li 
179*67e74705SXin Li     (*Begin)->print(Out, SubPolicy, Indentation);
180*67e74705SXin Li   }
181*67e74705SXin Li }
182*67e74705SXin Li 
dumpDeclContext() const183*67e74705SXin Li LLVM_DUMP_METHOD void DeclContext::dumpDeclContext() const {
184*67e74705SXin Li   // Get the translation unit
185*67e74705SXin Li   const DeclContext *DC = this;
186*67e74705SXin Li   while (!DC->isTranslationUnit())
187*67e74705SXin Li     DC = DC->getParent();
188*67e74705SXin Li 
189*67e74705SXin Li   ASTContext &Ctx = cast<TranslationUnitDecl>(DC)->getASTContext();
190*67e74705SXin Li   DeclPrinter Printer(llvm::errs(), Ctx.getPrintingPolicy(), 0);
191*67e74705SXin Li   Printer.VisitDeclContext(const_cast<DeclContext *>(this), /*Indent=*/false);
192*67e74705SXin Li }
193*67e74705SXin Li 
Indent(unsigned Indentation)194*67e74705SXin Li raw_ostream& DeclPrinter::Indent(unsigned Indentation) {
195*67e74705SXin Li   for (unsigned i = 0; i != Indentation; ++i)
196*67e74705SXin Li     Out << "  ";
197*67e74705SXin Li   return Out;
198*67e74705SXin Li }
199*67e74705SXin Li 
prettyPrintAttributes(Decl * D)200*67e74705SXin Li void DeclPrinter::prettyPrintAttributes(Decl *D) {
201*67e74705SXin Li   if (Policy.PolishForDeclaration)
202*67e74705SXin Li     return;
203*67e74705SXin Li 
204*67e74705SXin Li   if (D->hasAttrs()) {
205*67e74705SXin Li     AttrVec &Attrs = D->getAttrs();
206*67e74705SXin Li     for (auto *A : Attrs) {
207*67e74705SXin Li       switch (A->getKind()) {
208*67e74705SXin Li #define ATTR(X)
209*67e74705SXin Li #define PRAGMA_SPELLING_ATTR(X) case attr::X:
210*67e74705SXin Li #include "clang/Basic/AttrList.inc"
211*67e74705SXin Li         break;
212*67e74705SXin Li       default:
213*67e74705SXin Li         A->printPretty(Out, Policy);
214*67e74705SXin Li         break;
215*67e74705SXin Li       }
216*67e74705SXin Li     }
217*67e74705SXin Li   }
218*67e74705SXin Li }
219*67e74705SXin Li 
prettyPrintPragmas(Decl * D)220*67e74705SXin Li void DeclPrinter::prettyPrintPragmas(Decl *D) {
221*67e74705SXin Li   if (Policy.PolishForDeclaration)
222*67e74705SXin Li     return;
223*67e74705SXin Li 
224*67e74705SXin Li   if (D->hasAttrs()) {
225*67e74705SXin Li     AttrVec &Attrs = D->getAttrs();
226*67e74705SXin Li     for (auto *A : Attrs) {
227*67e74705SXin Li       switch (A->getKind()) {
228*67e74705SXin Li #define ATTR(X)
229*67e74705SXin Li #define PRAGMA_SPELLING_ATTR(X) case attr::X:
230*67e74705SXin Li #include "clang/Basic/AttrList.inc"
231*67e74705SXin Li         A->printPretty(Out, Policy);
232*67e74705SXin Li         Indent();
233*67e74705SXin Li         break;
234*67e74705SXin Li       default:
235*67e74705SXin Li         break;
236*67e74705SXin Li       }
237*67e74705SXin Li     }
238*67e74705SXin Li   }
239*67e74705SXin Li }
240*67e74705SXin Li 
printDeclType(QualType T,StringRef DeclName,bool Pack)241*67e74705SXin Li void DeclPrinter::printDeclType(QualType T, StringRef DeclName, bool Pack) {
242*67e74705SXin Li   // Normally, a PackExpansionType is written as T[3]... (for instance, as a
243*67e74705SXin Li   // template argument), but if it is the type of a declaration, the ellipsis
244*67e74705SXin Li   // is placed before the name being declared.
245*67e74705SXin Li   if (auto *PET = T->getAs<PackExpansionType>()) {
246*67e74705SXin Li     Pack = true;
247*67e74705SXin Li     T = PET->getPattern();
248*67e74705SXin Li   }
249*67e74705SXin Li   T.print(Out, Policy, (Pack ? "..." : "") + DeclName, Indentation);
250*67e74705SXin Li }
251*67e74705SXin Li 
ProcessDeclGroup(SmallVectorImpl<Decl * > & Decls)252*67e74705SXin Li void DeclPrinter::ProcessDeclGroup(SmallVectorImpl<Decl*>& Decls) {
253*67e74705SXin Li   this->Indent();
254*67e74705SXin Li   Decl::printGroup(Decls.data(), Decls.size(), Out, Policy, Indentation);
255*67e74705SXin Li   Out << ";\n";
256*67e74705SXin Li   Decls.clear();
257*67e74705SXin Li 
258*67e74705SXin Li }
259*67e74705SXin Li 
Print(AccessSpecifier AS)260*67e74705SXin Li void DeclPrinter::Print(AccessSpecifier AS) {
261*67e74705SXin Li   switch(AS) {
262*67e74705SXin Li   case AS_none:      llvm_unreachable("No access specifier!");
263*67e74705SXin Li   case AS_public:    Out << "public"; break;
264*67e74705SXin Li   case AS_protected: Out << "protected"; break;
265*67e74705SXin Li   case AS_private:   Out << "private"; break;
266*67e74705SXin Li   }
267*67e74705SXin Li }
268*67e74705SXin Li 
269*67e74705SXin Li //----------------------------------------------------------------------------
270*67e74705SXin Li // Common C declarations
271*67e74705SXin Li //----------------------------------------------------------------------------
272*67e74705SXin Li 
VisitDeclContext(DeclContext * DC,bool Indent)273*67e74705SXin Li void DeclPrinter::VisitDeclContext(DeclContext *DC, bool Indent) {
274*67e74705SXin Li   if (Policy.TerseOutput)
275*67e74705SXin Li     return;
276*67e74705SXin Li 
277*67e74705SXin Li   if (Indent)
278*67e74705SXin Li     Indentation += Policy.Indentation;
279*67e74705SXin Li 
280*67e74705SXin Li   SmallVector<Decl*, 2> Decls;
281*67e74705SXin Li   for (DeclContext::decl_iterator D = DC->decls_begin(), DEnd = DC->decls_end();
282*67e74705SXin Li        D != DEnd; ++D) {
283*67e74705SXin Li 
284*67e74705SXin Li     // Don't print ObjCIvarDecls, as they are printed when visiting the
285*67e74705SXin Li     // containing ObjCInterfaceDecl.
286*67e74705SXin Li     if (isa<ObjCIvarDecl>(*D))
287*67e74705SXin Li       continue;
288*67e74705SXin Li 
289*67e74705SXin Li     // Skip over implicit declarations in pretty-printing mode.
290*67e74705SXin Li     if (D->isImplicit())
291*67e74705SXin Li       continue;
292*67e74705SXin Li 
293*67e74705SXin Li     // The next bits of code handles stuff like "struct {int x;} a,b"; we're
294*67e74705SXin Li     // forced to merge the declarations because there's no other way to
295*67e74705SXin Li     // refer to the struct in question.  This limited merging is safe without
296*67e74705SXin Li     // a bunch of other checks because it only merges declarations directly
297*67e74705SXin Li     // referring to the tag, not typedefs.
298*67e74705SXin Li     //
299*67e74705SXin Li     // Check whether the current declaration should be grouped with a previous
300*67e74705SXin Li     // unnamed struct.
301*67e74705SXin Li     QualType CurDeclType = getDeclType(*D);
302*67e74705SXin Li     if (!Decls.empty() && !CurDeclType.isNull()) {
303*67e74705SXin Li       QualType BaseType = GetBaseType(CurDeclType);
304*67e74705SXin Li       if (!BaseType.isNull() && isa<ElaboratedType>(BaseType))
305*67e74705SXin Li         BaseType = cast<ElaboratedType>(BaseType)->getNamedType();
306*67e74705SXin Li       if (!BaseType.isNull() && isa<TagType>(BaseType) &&
307*67e74705SXin Li           cast<TagType>(BaseType)->getDecl() == Decls[0]) {
308*67e74705SXin Li         Decls.push_back(*D);
309*67e74705SXin Li         continue;
310*67e74705SXin Li       }
311*67e74705SXin Li     }
312*67e74705SXin Li 
313*67e74705SXin Li     // If we have a merged group waiting to be handled, handle it now.
314*67e74705SXin Li     if (!Decls.empty())
315*67e74705SXin Li       ProcessDeclGroup(Decls);
316*67e74705SXin Li 
317*67e74705SXin Li     // If the current declaration is an unnamed tag type, save it
318*67e74705SXin Li     // so we can merge it with the subsequent declaration(s) using it.
319*67e74705SXin Li     if (isa<TagDecl>(*D) && !cast<TagDecl>(*D)->getIdentifier()) {
320*67e74705SXin Li       Decls.push_back(*D);
321*67e74705SXin Li       continue;
322*67e74705SXin Li     }
323*67e74705SXin Li 
324*67e74705SXin Li     if (isa<AccessSpecDecl>(*D)) {
325*67e74705SXin Li       Indentation -= Policy.Indentation;
326*67e74705SXin Li       this->Indent();
327*67e74705SXin Li       Print(D->getAccess());
328*67e74705SXin Li       Out << ":\n";
329*67e74705SXin Li       Indentation += Policy.Indentation;
330*67e74705SXin Li       continue;
331*67e74705SXin Li     }
332*67e74705SXin Li 
333*67e74705SXin Li     this->Indent();
334*67e74705SXin Li     Visit(*D);
335*67e74705SXin Li 
336*67e74705SXin Li     // FIXME: Need to be able to tell the DeclPrinter when
337*67e74705SXin Li     const char *Terminator = nullptr;
338*67e74705SXin Li     if (isa<OMPThreadPrivateDecl>(*D) || isa<OMPDeclareReductionDecl>(*D))
339*67e74705SXin Li       Terminator = nullptr;
340*67e74705SXin Li     else if (isa<FunctionDecl>(*D) &&
341*67e74705SXin Li              cast<FunctionDecl>(*D)->isThisDeclarationADefinition())
342*67e74705SXin Li       Terminator = nullptr;
343*67e74705SXin Li     else if (isa<ObjCMethodDecl>(*D) && cast<ObjCMethodDecl>(*D)->getBody())
344*67e74705SXin Li       Terminator = nullptr;
345*67e74705SXin Li     else if (isa<NamespaceDecl>(*D) || isa<LinkageSpecDecl>(*D) ||
346*67e74705SXin Li              isa<ObjCImplementationDecl>(*D) ||
347*67e74705SXin Li              isa<ObjCInterfaceDecl>(*D) ||
348*67e74705SXin Li              isa<ObjCProtocolDecl>(*D) ||
349*67e74705SXin Li              isa<ObjCCategoryImplDecl>(*D) ||
350*67e74705SXin Li              isa<ObjCCategoryDecl>(*D))
351*67e74705SXin Li       Terminator = nullptr;
352*67e74705SXin Li     else if (isa<EnumConstantDecl>(*D)) {
353*67e74705SXin Li       DeclContext::decl_iterator Next = D;
354*67e74705SXin Li       ++Next;
355*67e74705SXin Li       if (Next != DEnd)
356*67e74705SXin Li         Terminator = ",";
357*67e74705SXin Li     } else
358*67e74705SXin Li       Terminator = ";";
359*67e74705SXin Li 
360*67e74705SXin Li     if (Terminator)
361*67e74705SXin Li       Out << Terminator;
362*67e74705SXin Li     Out << "\n";
363*67e74705SXin Li 
364*67e74705SXin Li     // Declare target attribute is special one, natural spelling for the pragma
365*67e74705SXin Li     // assumes "ending" construct so print it here.
366*67e74705SXin Li     if (D->hasAttr<OMPDeclareTargetDeclAttr>())
367*67e74705SXin Li       Out << "#pragma omp end declare target\n";
368*67e74705SXin Li   }
369*67e74705SXin Li 
370*67e74705SXin Li   if (!Decls.empty())
371*67e74705SXin Li     ProcessDeclGroup(Decls);
372*67e74705SXin Li 
373*67e74705SXin Li   if (Indent)
374*67e74705SXin Li     Indentation -= Policy.Indentation;
375*67e74705SXin Li }
376*67e74705SXin Li 
VisitTranslationUnitDecl(TranslationUnitDecl * D)377*67e74705SXin Li void DeclPrinter::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
378*67e74705SXin Li   VisitDeclContext(D, false);
379*67e74705SXin Li }
380*67e74705SXin Li 
VisitTypedefDecl(TypedefDecl * D)381*67e74705SXin Li void DeclPrinter::VisitTypedefDecl(TypedefDecl *D) {
382*67e74705SXin Li   if (!Policy.SuppressSpecifiers) {
383*67e74705SXin Li     Out << "typedef ";
384*67e74705SXin Li 
385*67e74705SXin Li     if (D->isModulePrivate())
386*67e74705SXin Li       Out << "__module_private__ ";
387*67e74705SXin Li   }
388*67e74705SXin Li   QualType Ty = D->getTypeSourceInfo()->getType();
389*67e74705SXin Li   Ty.print(Out, Policy, D->getName(), Indentation);
390*67e74705SXin Li   prettyPrintAttributes(D);
391*67e74705SXin Li }
392*67e74705SXin Li 
VisitTypeAliasDecl(TypeAliasDecl * D)393*67e74705SXin Li void DeclPrinter::VisitTypeAliasDecl(TypeAliasDecl *D) {
394*67e74705SXin Li   Out << "using " << *D;
395*67e74705SXin Li   prettyPrintAttributes(D);
396*67e74705SXin Li   Out << " = " << D->getTypeSourceInfo()->getType().getAsString(Policy);
397*67e74705SXin Li }
398*67e74705SXin Li 
VisitEnumDecl(EnumDecl * D)399*67e74705SXin Li void DeclPrinter::VisitEnumDecl(EnumDecl *D) {
400*67e74705SXin Li   if (!Policy.SuppressSpecifiers && D->isModulePrivate())
401*67e74705SXin Li     Out << "__module_private__ ";
402*67e74705SXin Li   Out << "enum ";
403*67e74705SXin Li   if (D->isScoped()) {
404*67e74705SXin Li     if (D->isScopedUsingClassTag())
405*67e74705SXin Li       Out << "class ";
406*67e74705SXin Li     else
407*67e74705SXin Li       Out << "struct ";
408*67e74705SXin Li   }
409*67e74705SXin Li   Out << *D;
410*67e74705SXin Li 
411*67e74705SXin Li   if (D->isFixed())
412*67e74705SXin Li     Out << " : " << D->getIntegerType().stream(Policy);
413*67e74705SXin Li 
414*67e74705SXin Li   if (D->isCompleteDefinition()) {
415*67e74705SXin Li     Out << " {\n";
416*67e74705SXin Li     VisitDeclContext(D);
417*67e74705SXin Li     Indent() << "}";
418*67e74705SXin Li   }
419*67e74705SXin Li   prettyPrintAttributes(D);
420*67e74705SXin Li }
421*67e74705SXin Li 
VisitRecordDecl(RecordDecl * D)422*67e74705SXin Li void DeclPrinter::VisitRecordDecl(RecordDecl *D) {
423*67e74705SXin Li   if (!Policy.SuppressSpecifiers && D->isModulePrivate())
424*67e74705SXin Li     Out << "__module_private__ ";
425*67e74705SXin Li   Out << D->getKindName();
426*67e74705SXin Li 
427*67e74705SXin Li   prettyPrintAttributes(D);
428*67e74705SXin Li 
429*67e74705SXin Li   if (D->getIdentifier())
430*67e74705SXin Li     Out << ' ' << *D;
431*67e74705SXin Li 
432*67e74705SXin Li   if (D->isCompleteDefinition()) {
433*67e74705SXin Li     Out << " {\n";
434*67e74705SXin Li     VisitDeclContext(D);
435*67e74705SXin Li     Indent() << "}";
436*67e74705SXin Li   }
437*67e74705SXin Li }
438*67e74705SXin Li 
VisitEnumConstantDecl(EnumConstantDecl * D)439*67e74705SXin Li void DeclPrinter::VisitEnumConstantDecl(EnumConstantDecl *D) {
440*67e74705SXin Li   Out << *D;
441*67e74705SXin Li   if (Expr *Init = D->getInitExpr()) {
442*67e74705SXin Li     Out << " = ";
443*67e74705SXin Li     Init->printPretty(Out, nullptr, Policy, Indentation);
444*67e74705SXin Li   }
445*67e74705SXin Li }
446*67e74705SXin Li 
VisitFunctionDecl(FunctionDecl * D)447*67e74705SXin Li void DeclPrinter::VisitFunctionDecl(FunctionDecl *D) {
448*67e74705SXin Li   if (!D->getDescribedFunctionTemplate() &&
449*67e74705SXin Li       !D->isFunctionTemplateSpecialization())
450*67e74705SXin Li     prettyPrintPragmas(D);
451*67e74705SXin Li 
452*67e74705SXin Li   CXXConstructorDecl *CDecl = dyn_cast<CXXConstructorDecl>(D);
453*67e74705SXin Li   CXXConversionDecl *ConversionDecl = dyn_cast<CXXConversionDecl>(D);
454*67e74705SXin Li   if (!Policy.SuppressSpecifiers) {
455*67e74705SXin Li     switch (D->getStorageClass()) {
456*67e74705SXin Li     case SC_None: break;
457*67e74705SXin Li     case SC_Extern: Out << "extern "; break;
458*67e74705SXin Li     case SC_Static: Out << "static "; break;
459*67e74705SXin Li     case SC_PrivateExtern: Out << "__private_extern__ "; break;
460*67e74705SXin Li     case SC_Auto: case SC_Register:
461*67e74705SXin Li       llvm_unreachable("invalid for functions");
462*67e74705SXin Li     }
463*67e74705SXin Li 
464*67e74705SXin Li     if (D->isInlineSpecified())  Out << "inline ";
465*67e74705SXin Li     if (D->isVirtualAsWritten()) Out << "virtual ";
466*67e74705SXin Li     if (D->isModulePrivate())    Out << "__module_private__ ";
467*67e74705SXin Li     if (D->isConstexpr() && !D->isExplicitlyDefaulted()) Out << "constexpr ";
468*67e74705SXin Li     if ((CDecl && CDecl->isExplicitSpecified()) ||
469*67e74705SXin Li         (ConversionDecl && ConversionDecl->isExplicit()))
470*67e74705SXin Li       Out << "explicit ";
471*67e74705SXin Li   }
472*67e74705SXin Li 
473*67e74705SXin Li   PrintingPolicy SubPolicy(Policy);
474*67e74705SXin Li   SubPolicy.SuppressSpecifiers = false;
475*67e74705SXin Li   std::string Proto = D->getNameInfo().getAsString();
476*67e74705SXin Li 
477*67e74705SXin Li   QualType Ty = D->getType();
478*67e74705SXin Li   while (const ParenType *PT = dyn_cast<ParenType>(Ty)) {
479*67e74705SXin Li     Proto = '(' + Proto + ')';
480*67e74705SXin Li     Ty = PT->getInnerType();
481*67e74705SXin Li   }
482*67e74705SXin Li 
483*67e74705SXin Li   if (const FunctionType *AFT = Ty->getAs<FunctionType>()) {
484*67e74705SXin Li     const FunctionProtoType *FT = nullptr;
485*67e74705SXin Li     if (D->hasWrittenPrototype())
486*67e74705SXin Li       FT = dyn_cast<FunctionProtoType>(AFT);
487*67e74705SXin Li 
488*67e74705SXin Li     Proto += "(";
489*67e74705SXin Li     if (FT) {
490*67e74705SXin Li       llvm::raw_string_ostream POut(Proto);
491*67e74705SXin Li       DeclPrinter ParamPrinter(POut, SubPolicy, Indentation);
492*67e74705SXin Li       for (unsigned i = 0, e = D->getNumParams(); i != e; ++i) {
493*67e74705SXin Li         if (i) POut << ", ";
494*67e74705SXin Li         ParamPrinter.VisitParmVarDecl(D->getParamDecl(i));
495*67e74705SXin Li       }
496*67e74705SXin Li 
497*67e74705SXin Li       if (FT->isVariadic()) {
498*67e74705SXin Li         if (D->getNumParams()) POut << ", ";
499*67e74705SXin Li         POut << "...";
500*67e74705SXin Li       }
501*67e74705SXin Li     } else if (D->doesThisDeclarationHaveABody() && !D->hasPrototype()) {
502*67e74705SXin Li       for (unsigned i = 0, e = D->getNumParams(); i != e; ++i) {
503*67e74705SXin Li         if (i)
504*67e74705SXin Li           Proto += ", ";
505*67e74705SXin Li         Proto += D->getParamDecl(i)->getNameAsString();
506*67e74705SXin Li       }
507*67e74705SXin Li     }
508*67e74705SXin Li 
509*67e74705SXin Li     Proto += ")";
510*67e74705SXin Li 
511*67e74705SXin Li     if (FT) {
512*67e74705SXin Li       if (FT->isConst())
513*67e74705SXin Li         Proto += " const";
514*67e74705SXin Li       if (FT->isVolatile())
515*67e74705SXin Li         Proto += " volatile";
516*67e74705SXin Li       if (FT->isRestrict())
517*67e74705SXin Li         Proto += " restrict";
518*67e74705SXin Li 
519*67e74705SXin Li       switch (FT->getRefQualifier()) {
520*67e74705SXin Li       case RQ_None:
521*67e74705SXin Li         break;
522*67e74705SXin Li       case RQ_LValue:
523*67e74705SXin Li         Proto += " &";
524*67e74705SXin Li         break;
525*67e74705SXin Li       case RQ_RValue:
526*67e74705SXin Li         Proto += " &&";
527*67e74705SXin Li         break;
528*67e74705SXin Li       }
529*67e74705SXin Li     }
530*67e74705SXin Li 
531*67e74705SXin Li     if (FT && FT->hasDynamicExceptionSpec()) {
532*67e74705SXin Li       Proto += " throw(";
533*67e74705SXin Li       if (FT->getExceptionSpecType() == EST_MSAny)
534*67e74705SXin Li         Proto += "...";
535*67e74705SXin Li       else
536*67e74705SXin Li         for (unsigned I = 0, N = FT->getNumExceptions(); I != N; ++I) {
537*67e74705SXin Li           if (I)
538*67e74705SXin Li             Proto += ", ";
539*67e74705SXin Li 
540*67e74705SXin Li           Proto += FT->getExceptionType(I).getAsString(SubPolicy);
541*67e74705SXin Li         }
542*67e74705SXin Li       Proto += ")";
543*67e74705SXin Li     } else if (FT && isNoexceptExceptionSpec(FT->getExceptionSpecType())) {
544*67e74705SXin Li       Proto += " noexcept";
545*67e74705SXin Li       if (FT->getExceptionSpecType() == EST_ComputedNoexcept) {
546*67e74705SXin Li         Proto += "(";
547*67e74705SXin Li         llvm::raw_string_ostream EOut(Proto);
548*67e74705SXin Li         FT->getNoexceptExpr()->printPretty(EOut, nullptr, SubPolicy,
549*67e74705SXin Li                                            Indentation);
550*67e74705SXin Li         EOut.flush();
551*67e74705SXin Li         Proto += EOut.str();
552*67e74705SXin Li         Proto += ")";
553*67e74705SXin Li       }
554*67e74705SXin Li     }
555*67e74705SXin Li 
556*67e74705SXin Li     if (CDecl) {
557*67e74705SXin Li       bool HasInitializerList = false;
558*67e74705SXin Li       for (const auto *BMInitializer : CDecl->inits()) {
559*67e74705SXin Li         if (BMInitializer->isInClassMemberInitializer())
560*67e74705SXin Li           continue;
561*67e74705SXin Li 
562*67e74705SXin Li         if (!HasInitializerList) {
563*67e74705SXin Li           Proto += " : ";
564*67e74705SXin Li           Out << Proto;
565*67e74705SXin Li           Proto.clear();
566*67e74705SXin Li           HasInitializerList = true;
567*67e74705SXin Li         } else
568*67e74705SXin Li           Out << ", ";
569*67e74705SXin Li 
570*67e74705SXin Li         if (BMInitializer->isAnyMemberInitializer()) {
571*67e74705SXin Li           FieldDecl *FD = BMInitializer->getAnyMember();
572*67e74705SXin Li           Out << *FD;
573*67e74705SXin Li         } else {
574*67e74705SXin Li           Out << QualType(BMInitializer->getBaseClass(), 0).getAsString(Policy);
575*67e74705SXin Li         }
576*67e74705SXin Li 
577*67e74705SXin Li         Out << "(";
578*67e74705SXin Li         if (!BMInitializer->getInit()) {
579*67e74705SXin Li           // Nothing to print
580*67e74705SXin Li         } else {
581*67e74705SXin Li           Expr *Init = BMInitializer->getInit();
582*67e74705SXin Li           if (ExprWithCleanups *Tmp = dyn_cast<ExprWithCleanups>(Init))
583*67e74705SXin Li             Init = Tmp->getSubExpr();
584*67e74705SXin Li 
585*67e74705SXin Li           Init = Init->IgnoreParens();
586*67e74705SXin Li 
587*67e74705SXin Li           Expr *SimpleInit = nullptr;
588*67e74705SXin Li           Expr **Args = nullptr;
589*67e74705SXin Li           unsigned NumArgs = 0;
590*67e74705SXin Li           if (ParenListExpr *ParenList = dyn_cast<ParenListExpr>(Init)) {
591*67e74705SXin Li             Args = ParenList->getExprs();
592*67e74705SXin Li             NumArgs = ParenList->getNumExprs();
593*67e74705SXin Li           } else if (CXXConstructExpr *Construct
594*67e74705SXin Li                                         = dyn_cast<CXXConstructExpr>(Init)) {
595*67e74705SXin Li             Args = Construct->getArgs();
596*67e74705SXin Li             NumArgs = Construct->getNumArgs();
597*67e74705SXin Li           } else
598*67e74705SXin Li             SimpleInit = Init;
599*67e74705SXin Li 
600*67e74705SXin Li           if (SimpleInit)
601*67e74705SXin Li             SimpleInit->printPretty(Out, nullptr, Policy, Indentation);
602*67e74705SXin Li           else {
603*67e74705SXin Li             for (unsigned I = 0; I != NumArgs; ++I) {
604*67e74705SXin Li               assert(Args[I] != nullptr && "Expected non-null Expr");
605*67e74705SXin Li               if (isa<CXXDefaultArgExpr>(Args[I]))
606*67e74705SXin Li                 break;
607*67e74705SXin Li 
608*67e74705SXin Li               if (I)
609*67e74705SXin Li                 Out << ", ";
610*67e74705SXin Li               Args[I]->printPretty(Out, nullptr, Policy, Indentation);
611*67e74705SXin Li             }
612*67e74705SXin Li           }
613*67e74705SXin Li         }
614*67e74705SXin Li         Out << ")";
615*67e74705SXin Li         if (BMInitializer->isPackExpansion())
616*67e74705SXin Li           Out << "...";
617*67e74705SXin Li       }
618*67e74705SXin Li     } else if (!ConversionDecl && !isa<CXXDestructorDecl>(D)) {
619*67e74705SXin Li       if (FT && FT->hasTrailingReturn()) {
620*67e74705SXin Li         Out << "auto " << Proto << " -> ";
621*67e74705SXin Li         Proto.clear();
622*67e74705SXin Li       }
623*67e74705SXin Li       AFT->getReturnType().print(Out, Policy, Proto);
624*67e74705SXin Li       Proto.clear();
625*67e74705SXin Li     }
626*67e74705SXin Li     Out << Proto;
627*67e74705SXin Li   } else {
628*67e74705SXin Li     Ty.print(Out, Policy, Proto);
629*67e74705SXin Li   }
630*67e74705SXin Li 
631*67e74705SXin Li   prettyPrintAttributes(D);
632*67e74705SXin Li 
633*67e74705SXin Li   if (D->isPure())
634*67e74705SXin Li     Out << " = 0";
635*67e74705SXin Li   else if (D->isDeletedAsWritten())
636*67e74705SXin Li     Out << " = delete";
637*67e74705SXin Li   else if (D->isExplicitlyDefaulted())
638*67e74705SXin Li     Out << " = default";
639*67e74705SXin Li   else if (D->doesThisDeclarationHaveABody() && !Policy.TerseOutput) {
640*67e74705SXin Li     if (!D->hasPrototype() && D->getNumParams()) {
641*67e74705SXin Li       // This is a K&R function definition, so we need to print the
642*67e74705SXin Li       // parameters.
643*67e74705SXin Li       Out << '\n';
644*67e74705SXin Li       DeclPrinter ParamPrinter(Out, SubPolicy, Indentation);
645*67e74705SXin Li       Indentation += Policy.Indentation;
646*67e74705SXin Li       for (unsigned i = 0, e = D->getNumParams(); i != e; ++i) {
647*67e74705SXin Li         Indent();
648*67e74705SXin Li         ParamPrinter.VisitParmVarDecl(D->getParamDecl(i));
649*67e74705SXin Li         Out << ";\n";
650*67e74705SXin Li       }
651*67e74705SXin Li       Indentation -= Policy.Indentation;
652*67e74705SXin Li     } else
653*67e74705SXin Li       Out << ' ';
654*67e74705SXin Li 
655*67e74705SXin Li     if (D->getBody())
656*67e74705SXin Li       D->getBody()->printPretty(Out, nullptr, SubPolicy, Indentation);
657*67e74705SXin Li     Out << '\n';
658*67e74705SXin Li   }
659*67e74705SXin Li }
660*67e74705SXin Li 
VisitFriendDecl(FriendDecl * D)661*67e74705SXin Li void DeclPrinter::VisitFriendDecl(FriendDecl *D) {
662*67e74705SXin Li   if (TypeSourceInfo *TSI = D->getFriendType()) {
663*67e74705SXin Li     unsigned NumTPLists = D->getFriendTypeNumTemplateParameterLists();
664*67e74705SXin Li     for (unsigned i = 0; i < NumTPLists; ++i)
665*67e74705SXin Li       PrintTemplateParameters(D->getFriendTypeTemplateParameterList(i));
666*67e74705SXin Li     Out << "friend ";
667*67e74705SXin Li     Out << " " << TSI->getType().getAsString(Policy);
668*67e74705SXin Li   }
669*67e74705SXin Li   else if (FunctionDecl *FD =
670*67e74705SXin Li       dyn_cast<FunctionDecl>(D->getFriendDecl())) {
671*67e74705SXin Li     Out << "friend ";
672*67e74705SXin Li     VisitFunctionDecl(FD);
673*67e74705SXin Li   }
674*67e74705SXin Li   else if (FunctionTemplateDecl *FTD =
675*67e74705SXin Li            dyn_cast<FunctionTemplateDecl>(D->getFriendDecl())) {
676*67e74705SXin Li     Out << "friend ";
677*67e74705SXin Li     VisitFunctionTemplateDecl(FTD);
678*67e74705SXin Li   }
679*67e74705SXin Li   else if (ClassTemplateDecl *CTD =
680*67e74705SXin Li            dyn_cast<ClassTemplateDecl>(D->getFriendDecl())) {
681*67e74705SXin Li     Out << "friend ";
682*67e74705SXin Li     VisitRedeclarableTemplateDecl(CTD);
683*67e74705SXin Li   }
684*67e74705SXin Li }
685*67e74705SXin Li 
VisitFieldDecl(FieldDecl * D)686*67e74705SXin Li void DeclPrinter::VisitFieldDecl(FieldDecl *D) {
687*67e74705SXin Li   // FIXME: add printing of pragma attributes if required.
688*67e74705SXin Li   if (!Policy.SuppressSpecifiers && D->isMutable())
689*67e74705SXin Li     Out << "mutable ";
690*67e74705SXin Li   if (!Policy.SuppressSpecifiers && D->isModulePrivate())
691*67e74705SXin Li     Out << "__module_private__ ";
692*67e74705SXin Li 
693*67e74705SXin Li   Out << D->getASTContext().getUnqualifiedObjCPointerType(D->getType()).
694*67e74705SXin Li          stream(Policy, D->getName(), Indentation);
695*67e74705SXin Li 
696*67e74705SXin Li   if (D->isBitField()) {
697*67e74705SXin Li     Out << " : ";
698*67e74705SXin Li     D->getBitWidth()->printPretty(Out, nullptr, Policy, Indentation);
699*67e74705SXin Li   }
700*67e74705SXin Li 
701*67e74705SXin Li   Expr *Init = D->getInClassInitializer();
702*67e74705SXin Li   if (!Policy.SuppressInitializers && Init) {
703*67e74705SXin Li     if (D->getInClassInitStyle() == ICIS_ListInit)
704*67e74705SXin Li       Out << " ";
705*67e74705SXin Li     else
706*67e74705SXin Li       Out << " = ";
707*67e74705SXin Li     Init->printPretty(Out, nullptr, Policy, Indentation);
708*67e74705SXin Li   }
709*67e74705SXin Li   prettyPrintAttributes(D);
710*67e74705SXin Li }
711*67e74705SXin Li 
VisitLabelDecl(LabelDecl * D)712*67e74705SXin Li void DeclPrinter::VisitLabelDecl(LabelDecl *D) {
713*67e74705SXin Li   Out << *D << ":";
714*67e74705SXin Li }
715*67e74705SXin Li 
VisitVarDecl(VarDecl * D)716*67e74705SXin Li void DeclPrinter::VisitVarDecl(VarDecl *D) {
717*67e74705SXin Li   prettyPrintPragmas(D);
718*67e74705SXin Li 
719*67e74705SXin Li   QualType T = D->getTypeSourceInfo()
720*67e74705SXin Li     ? D->getTypeSourceInfo()->getType()
721*67e74705SXin Li     : D->getASTContext().getUnqualifiedObjCPointerType(D->getType());
722*67e74705SXin Li 
723*67e74705SXin Li   if (!Policy.SuppressSpecifiers) {
724*67e74705SXin Li     StorageClass SC = D->getStorageClass();
725*67e74705SXin Li     if (SC != SC_None)
726*67e74705SXin Li       Out << VarDecl::getStorageClassSpecifierString(SC) << " ";
727*67e74705SXin Li 
728*67e74705SXin Li     switch (D->getTSCSpec()) {
729*67e74705SXin Li     case TSCS_unspecified:
730*67e74705SXin Li       break;
731*67e74705SXin Li     case TSCS___thread:
732*67e74705SXin Li       Out << "__thread ";
733*67e74705SXin Li       break;
734*67e74705SXin Li     case TSCS__Thread_local:
735*67e74705SXin Li       Out << "_Thread_local ";
736*67e74705SXin Li       break;
737*67e74705SXin Li     case TSCS_thread_local:
738*67e74705SXin Li       Out << "thread_local ";
739*67e74705SXin Li       break;
740*67e74705SXin Li     }
741*67e74705SXin Li 
742*67e74705SXin Li     if (D->isModulePrivate())
743*67e74705SXin Li       Out << "__module_private__ ";
744*67e74705SXin Li 
745*67e74705SXin Li     if (D->isConstexpr()) {
746*67e74705SXin Li       Out << "constexpr ";
747*67e74705SXin Li       T.removeLocalConst();
748*67e74705SXin Li     }
749*67e74705SXin Li   }
750*67e74705SXin Li 
751*67e74705SXin Li   printDeclType(T, D->getName());
752*67e74705SXin Li   Expr *Init = D->getInit();
753*67e74705SXin Li   if (!Policy.SuppressInitializers && Init) {
754*67e74705SXin Li     bool ImplicitInit = false;
755*67e74705SXin Li     if (CXXConstructExpr *Construct =
756*67e74705SXin Li             dyn_cast<CXXConstructExpr>(Init->IgnoreImplicit())) {
757*67e74705SXin Li       if (D->getInitStyle() == VarDecl::CallInit &&
758*67e74705SXin Li           !Construct->isListInitialization()) {
759*67e74705SXin Li         ImplicitInit = Construct->getNumArgs() == 0 ||
760*67e74705SXin Li           Construct->getArg(0)->isDefaultArgument();
761*67e74705SXin Li       }
762*67e74705SXin Li     }
763*67e74705SXin Li     if (!ImplicitInit) {
764*67e74705SXin Li       if ((D->getInitStyle() == VarDecl::CallInit) && !isa<ParenListExpr>(Init))
765*67e74705SXin Li         Out << "(";
766*67e74705SXin Li       else if (D->getInitStyle() == VarDecl::CInit) {
767*67e74705SXin Li         Out << " = ";
768*67e74705SXin Li       }
769*67e74705SXin Li       PrintingPolicy SubPolicy(Policy);
770*67e74705SXin Li       SubPolicy.SuppressSpecifiers = false;
771*67e74705SXin Li       SubPolicy.IncludeTagDefinition = false;
772*67e74705SXin Li       Init->printPretty(Out, nullptr, SubPolicy, Indentation);
773*67e74705SXin Li       if ((D->getInitStyle() == VarDecl::CallInit) && !isa<ParenListExpr>(Init))
774*67e74705SXin Li         Out << ")";
775*67e74705SXin Li     }
776*67e74705SXin Li   }
777*67e74705SXin Li   prettyPrintAttributes(D);
778*67e74705SXin Li }
779*67e74705SXin Li 
VisitParmVarDecl(ParmVarDecl * D)780*67e74705SXin Li void DeclPrinter::VisitParmVarDecl(ParmVarDecl *D) {
781*67e74705SXin Li   VisitVarDecl(D);
782*67e74705SXin Li }
783*67e74705SXin Li 
VisitFileScopeAsmDecl(FileScopeAsmDecl * D)784*67e74705SXin Li void DeclPrinter::VisitFileScopeAsmDecl(FileScopeAsmDecl *D) {
785*67e74705SXin Li   Out << "__asm (";
786*67e74705SXin Li   D->getAsmString()->printPretty(Out, nullptr, Policy, Indentation);
787*67e74705SXin Li   Out << ")";
788*67e74705SXin Li }
789*67e74705SXin Li 
VisitImportDecl(ImportDecl * D)790*67e74705SXin Li void DeclPrinter::VisitImportDecl(ImportDecl *D) {
791*67e74705SXin Li   Out << "@import " << D->getImportedModule()->getFullModuleName()
792*67e74705SXin Li       << ";\n";
793*67e74705SXin Li }
794*67e74705SXin Li 
VisitStaticAssertDecl(StaticAssertDecl * D)795*67e74705SXin Li void DeclPrinter::VisitStaticAssertDecl(StaticAssertDecl *D) {
796*67e74705SXin Li   Out << "static_assert(";
797*67e74705SXin Li   D->getAssertExpr()->printPretty(Out, nullptr, Policy, Indentation);
798*67e74705SXin Li   if (StringLiteral *SL = D->getMessage()) {
799*67e74705SXin Li     Out << ", ";
800*67e74705SXin Li     SL->printPretty(Out, nullptr, Policy, Indentation);
801*67e74705SXin Li   }
802*67e74705SXin Li   Out << ")";
803*67e74705SXin Li }
804*67e74705SXin Li 
805*67e74705SXin Li //----------------------------------------------------------------------------
806*67e74705SXin Li // C++ declarations
807*67e74705SXin Li //----------------------------------------------------------------------------
VisitNamespaceDecl(NamespaceDecl * D)808*67e74705SXin Li void DeclPrinter::VisitNamespaceDecl(NamespaceDecl *D) {
809*67e74705SXin Li   if (D->isInline())
810*67e74705SXin Li     Out << "inline ";
811*67e74705SXin Li   Out << "namespace " << *D << " {\n";
812*67e74705SXin Li   VisitDeclContext(D);
813*67e74705SXin Li   Indent() << "}";
814*67e74705SXin Li }
815*67e74705SXin Li 
VisitUsingDirectiveDecl(UsingDirectiveDecl * D)816*67e74705SXin Li void DeclPrinter::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
817*67e74705SXin Li   Out << "using namespace ";
818*67e74705SXin Li   if (D->getQualifier())
819*67e74705SXin Li     D->getQualifier()->print(Out, Policy);
820*67e74705SXin Li   Out << *D->getNominatedNamespaceAsWritten();
821*67e74705SXin Li }
822*67e74705SXin Li 
VisitNamespaceAliasDecl(NamespaceAliasDecl * D)823*67e74705SXin Li void DeclPrinter::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
824*67e74705SXin Li   Out << "namespace " << *D << " = ";
825*67e74705SXin Li   if (D->getQualifier())
826*67e74705SXin Li     D->getQualifier()->print(Out, Policy);
827*67e74705SXin Li   Out << *D->getAliasedNamespace();
828*67e74705SXin Li }
829*67e74705SXin Li 
VisitEmptyDecl(EmptyDecl * D)830*67e74705SXin Li void DeclPrinter::VisitEmptyDecl(EmptyDecl *D) {
831*67e74705SXin Li   prettyPrintAttributes(D);
832*67e74705SXin Li }
833*67e74705SXin Li 
VisitCXXRecordDecl(CXXRecordDecl * D)834*67e74705SXin Li void DeclPrinter::VisitCXXRecordDecl(CXXRecordDecl *D) {
835*67e74705SXin Li   // FIXME: add printing of pragma attributes if required.
836*67e74705SXin Li   if (!Policy.SuppressSpecifiers && D->isModulePrivate())
837*67e74705SXin Li     Out << "__module_private__ ";
838*67e74705SXin Li   Out << D->getKindName();
839*67e74705SXin Li 
840*67e74705SXin Li   prettyPrintAttributes(D);
841*67e74705SXin Li 
842*67e74705SXin Li   if (D->getIdentifier())
843*67e74705SXin Li     Out << ' ' << *D;
844*67e74705SXin Li 
845*67e74705SXin Li   if (D->isCompleteDefinition()) {
846*67e74705SXin Li     // Print the base classes
847*67e74705SXin Li     if (D->getNumBases()) {
848*67e74705SXin Li       Out << " : ";
849*67e74705SXin Li       for (CXXRecordDecl::base_class_iterator Base = D->bases_begin(),
850*67e74705SXin Li              BaseEnd = D->bases_end(); Base != BaseEnd; ++Base) {
851*67e74705SXin Li         if (Base != D->bases_begin())
852*67e74705SXin Li           Out << ", ";
853*67e74705SXin Li 
854*67e74705SXin Li         if (Base->isVirtual())
855*67e74705SXin Li           Out << "virtual ";
856*67e74705SXin Li 
857*67e74705SXin Li         AccessSpecifier AS = Base->getAccessSpecifierAsWritten();
858*67e74705SXin Li         if (AS != AS_none) {
859*67e74705SXin Li           Print(AS);
860*67e74705SXin Li           Out << " ";
861*67e74705SXin Li         }
862*67e74705SXin Li         Out << Base->getType().getAsString(Policy);
863*67e74705SXin Li 
864*67e74705SXin Li         if (Base->isPackExpansion())
865*67e74705SXin Li           Out << "...";
866*67e74705SXin Li       }
867*67e74705SXin Li     }
868*67e74705SXin Li 
869*67e74705SXin Li     // Print the class definition
870*67e74705SXin Li     // FIXME: Doesn't print access specifiers, e.g., "public:"
871*67e74705SXin Li     Out << " {\n";
872*67e74705SXin Li     VisitDeclContext(D);
873*67e74705SXin Li     Indent() << "}";
874*67e74705SXin Li   }
875*67e74705SXin Li }
876*67e74705SXin Li 
VisitLinkageSpecDecl(LinkageSpecDecl * D)877*67e74705SXin Li void DeclPrinter::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
878*67e74705SXin Li   const char *l;
879*67e74705SXin Li   if (D->getLanguage() == LinkageSpecDecl::lang_c)
880*67e74705SXin Li     l = "C";
881*67e74705SXin Li   else {
882*67e74705SXin Li     assert(D->getLanguage() == LinkageSpecDecl::lang_cxx &&
883*67e74705SXin Li            "unknown language in linkage specification");
884*67e74705SXin Li     l = "C++";
885*67e74705SXin Li   }
886*67e74705SXin Li 
887*67e74705SXin Li   Out << "extern \"" << l << "\" ";
888*67e74705SXin Li   if (D->hasBraces()) {
889*67e74705SXin Li     Out << "{\n";
890*67e74705SXin Li     VisitDeclContext(D);
891*67e74705SXin Li     Indent() << "}";
892*67e74705SXin Li   } else
893*67e74705SXin Li     Visit(*D->decls_begin());
894*67e74705SXin Li }
895*67e74705SXin Li 
PrintTemplateParameters(const TemplateParameterList * Params,const TemplateArgumentList * Args)896*67e74705SXin Li void DeclPrinter::PrintTemplateParameters(const TemplateParameterList *Params,
897*67e74705SXin Li                                           const TemplateArgumentList *Args) {
898*67e74705SXin Li   assert(Params);
899*67e74705SXin Li   assert(!Args || Params->size() == Args->size());
900*67e74705SXin Li 
901*67e74705SXin Li   Out << "template <";
902*67e74705SXin Li 
903*67e74705SXin Li   for (unsigned i = 0, e = Params->size(); i != e; ++i) {
904*67e74705SXin Li     if (i != 0)
905*67e74705SXin Li       Out << ", ";
906*67e74705SXin Li 
907*67e74705SXin Li     const Decl *Param = Params->getParam(i);
908*67e74705SXin Li     if (const TemplateTypeParmDecl *TTP =
909*67e74705SXin Li           dyn_cast<TemplateTypeParmDecl>(Param)) {
910*67e74705SXin Li 
911*67e74705SXin Li       if (TTP->wasDeclaredWithTypename())
912*67e74705SXin Li         Out << "typename ";
913*67e74705SXin Li       else
914*67e74705SXin Li         Out << "class ";
915*67e74705SXin Li 
916*67e74705SXin Li       if (TTP->isParameterPack())
917*67e74705SXin Li         Out << "...";
918*67e74705SXin Li 
919*67e74705SXin Li       Out << *TTP;
920*67e74705SXin Li 
921*67e74705SXin Li       if (Args) {
922*67e74705SXin Li         Out << " = ";
923*67e74705SXin Li         Args->get(i).print(Policy, Out);
924*67e74705SXin Li       } else if (TTP->hasDefaultArgument()) {
925*67e74705SXin Li         Out << " = ";
926*67e74705SXin Li         Out << TTP->getDefaultArgument().getAsString(Policy);
927*67e74705SXin Li       };
928*67e74705SXin Li     } else if (const NonTypeTemplateParmDecl *NTTP =
929*67e74705SXin Li                  dyn_cast<NonTypeTemplateParmDecl>(Param)) {
930*67e74705SXin Li       StringRef Name;
931*67e74705SXin Li       if (IdentifierInfo *II = NTTP->getIdentifier())
932*67e74705SXin Li         Name = II->getName();
933*67e74705SXin Li       printDeclType(NTTP->getType(), Name, NTTP->isParameterPack());
934*67e74705SXin Li 
935*67e74705SXin Li       if (Args) {
936*67e74705SXin Li         Out << " = ";
937*67e74705SXin Li         Args->get(i).print(Policy, Out);
938*67e74705SXin Li       } else if (NTTP->hasDefaultArgument()) {
939*67e74705SXin Li         Out << " = ";
940*67e74705SXin Li         NTTP->getDefaultArgument()->printPretty(Out, nullptr, Policy,
941*67e74705SXin Li                                                 Indentation);
942*67e74705SXin Li       }
943*67e74705SXin Li     } else if (const TemplateTemplateParmDecl *TTPD =
944*67e74705SXin Li                  dyn_cast<TemplateTemplateParmDecl>(Param)) {
945*67e74705SXin Li       VisitTemplateDecl(TTPD);
946*67e74705SXin Li       // FIXME: print the default argument, if present.
947*67e74705SXin Li     }
948*67e74705SXin Li   }
949*67e74705SXin Li 
950*67e74705SXin Li   Out << "> ";
951*67e74705SXin Li }
952*67e74705SXin Li 
VisitTemplateDecl(const TemplateDecl * D)953*67e74705SXin Li void DeclPrinter::VisitTemplateDecl(const TemplateDecl *D) {
954*67e74705SXin Li   PrintTemplateParameters(D->getTemplateParameters());
955*67e74705SXin Li 
956*67e74705SXin Li   if (const TemplateTemplateParmDecl *TTP =
957*67e74705SXin Li         dyn_cast<TemplateTemplateParmDecl>(D)) {
958*67e74705SXin Li     Out << "class ";
959*67e74705SXin Li     if (TTP->isParameterPack())
960*67e74705SXin Li       Out << "...";
961*67e74705SXin Li     Out << D->getName();
962*67e74705SXin Li   } else {
963*67e74705SXin Li     Visit(D->getTemplatedDecl());
964*67e74705SXin Li   }
965*67e74705SXin Li }
966*67e74705SXin Li 
VisitFunctionTemplateDecl(FunctionTemplateDecl * D)967*67e74705SXin Li void DeclPrinter::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
968*67e74705SXin Li   if (PrintInstantiation) {
969*67e74705SXin Li     TemplateParameterList *Params = D->getTemplateParameters();
970*67e74705SXin Li     for (auto *I : D->specializations()) {
971*67e74705SXin Li       prettyPrintPragmas(I);
972*67e74705SXin Li       PrintTemplateParameters(Params, I->getTemplateSpecializationArgs());
973*67e74705SXin Li       Visit(I);
974*67e74705SXin Li     }
975*67e74705SXin Li   }
976*67e74705SXin Li 
977*67e74705SXin Li   prettyPrintPragmas(D->getTemplatedDecl());
978*67e74705SXin Li   return VisitRedeclarableTemplateDecl(D);
979*67e74705SXin Li }
980*67e74705SXin Li 
VisitClassTemplateDecl(ClassTemplateDecl * D)981*67e74705SXin Li void DeclPrinter::VisitClassTemplateDecl(ClassTemplateDecl *D) {
982*67e74705SXin Li   if (PrintInstantiation) {
983*67e74705SXin Li     TemplateParameterList *Params = D->getTemplateParameters();
984*67e74705SXin Li     for (auto *I : D->specializations()) {
985*67e74705SXin Li       PrintTemplateParameters(Params, &I->getTemplateArgs());
986*67e74705SXin Li       Visit(I);
987*67e74705SXin Li       Out << '\n';
988*67e74705SXin Li     }
989*67e74705SXin Li   }
990*67e74705SXin Li 
991*67e74705SXin Li   return VisitRedeclarableTemplateDecl(D);
992*67e74705SXin Li }
993*67e74705SXin Li 
994*67e74705SXin Li //----------------------------------------------------------------------------
995*67e74705SXin Li // Objective-C declarations
996*67e74705SXin Li //----------------------------------------------------------------------------
997*67e74705SXin Li 
PrintObjCMethodType(ASTContext & Ctx,Decl::ObjCDeclQualifier Quals,QualType T)998*67e74705SXin Li void DeclPrinter::PrintObjCMethodType(ASTContext &Ctx,
999*67e74705SXin Li                                       Decl::ObjCDeclQualifier Quals,
1000*67e74705SXin Li                                       QualType T) {
1001*67e74705SXin Li   Out << '(';
1002*67e74705SXin Li   if (Quals & Decl::ObjCDeclQualifier::OBJC_TQ_In)
1003*67e74705SXin Li     Out << "in ";
1004*67e74705SXin Li   if (Quals & Decl::ObjCDeclQualifier::OBJC_TQ_Inout)
1005*67e74705SXin Li     Out << "inout ";
1006*67e74705SXin Li   if (Quals & Decl::ObjCDeclQualifier::OBJC_TQ_Out)
1007*67e74705SXin Li     Out << "out ";
1008*67e74705SXin Li   if (Quals & Decl::ObjCDeclQualifier::OBJC_TQ_Bycopy)
1009*67e74705SXin Li     Out << "bycopy ";
1010*67e74705SXin Li   if (Quals & Decl::ObjCDeclQualifier::OBJC_TQ_Byref)
1011*67e74705SXin Li     Out << "byref ";
1012*67e74705SXin Li   if (Quals & Decl::ObjCDeclQualifier::OBJC_TQ_Oneway)
1013*67e74705SXin Li     Out << "oneway ";
1014*67e74705SXin Li   if (Quals & Decl::ObjCDeclQualifier::OBJC_TQ_CSNullability) {
1015*67e74705SXin Li     if (auto nullability = AttributedType::stripOuterNullability(T))
1016*67e74705SXin Li       Out << getNullabilitySpelling(*nullability, true) << ' ';
1017*67e74705SXin Li   }
1018*67e74705SXin Li 
1019*67e74705SXin Li   Out << Ctx.getUnqualifiedObjCPointerType(T).getAsString(Policy);
1020*67e74705SXin Li   Out << ')';
1021*67e74705SXin Li }
1022*67e74705SXin Li 
PrintObjCTypeParams(ObjCTypeParamList * Params)1023*67e74705SXin Li void DeclPrinter::PrintObjCTypeParams(ObjCTypeParamList *Params) {
1024*67e74705SXin Li   Out << "<";
1025*67e74705SXin Li   unsigned First = true;
1026*67e74705SXin Li   for (auto *Param : *Params) {
1027*67e74705SXin Li     if (First) {
1028*67e74705SXin Li       First = false;
1029*67e74705SXin Li     } else {
1030*67e74705SXin Li       Out << ", ";
1031*67e74705SXin Li     }
1032*67e74705SXin Li 
1033*67e74705SXin Li     switch (Param->getVariance()) {
1034*67e74705SXin Li     case ObjCTypeParamVariance::Invariant:
1035*67e74705SXin Li       break;
1036*67e74705SXin Li 
1037*67e74705SXin Li     case ObjCTypeParamVariance::Covariant:
1038*67e74705SXin Li       Out << "__covariant ";
1039*67e74705SXin Li       break;
1040*67e74705SXin Li 
1041*67e74705SXin Li     case ObjCTypeParamVariance::Contravariant:
1042*67e74705SXin Li       Out << "__contravariant ";
1043*67e74705SXin Li       break;
1044*67e74705SXin Li     }
1045*67e74705SXin Li 
1046*67e74705SXin Li     Out << Param->getDeclName().getAsString();
1047*67e74705SXin Li 
1048*67e74705SXin Li     if (Param->hasExplicitBound()) {
1049*67e74705SXin Li       Out << " : " << Param->getUnderlyingType().getAsString(Policy);
1050*67e74705SXin Li     }
1051*67e74705SXin Li   }
1052*67e74705SXin Li   Out << ">";
1053*67e74705SXin Li }
1054*67e74705SXin Li 
VisitObjCMethodDecl(ObjCMethodDecl * OMD)1055*67e74705SXin Li void DeclPrinter::VisitObjCMethodDecl(ObjCMethodDecl *OMD) {
1056*67e74705SXin Li   if (OMD->isInstanceMethod())
1057*67e74705SXin Li     Out << "- ";
1058*67e74705SXin Li   else
1059*67e74705SXin Li     Out << "+ ";
1060*67e74705SXin Li   if (!OMD->getReturnType().isNull()) {
1061*67e74705SXin Li     PrintObjCMethodType(OMD->getASTContext(), OMD->getObjCDeclQualifier(),
1062*67e74705SXin Li                         OMD->getReturnType());
1063*67e74705SXin Li   }
1064*67e74705SXin Li 
1065*67e74705SXin Li   std::string name = OMD->getSelector().getAsString();
1066*67e74705SXin Li   std::string::size_type pos, lastPos = 0;
1067*67e74705SXin Li   for (const auto *PI : OMD->parameters()) {
1068*67e74705SXin Li     // FIXME: selector is missing here!
1069*67e74705SXin Li     pos = name.find_first_of(':', lastPos);
1070*67e74705SXin Li     Out << " " << name.substr(lastPos, pos - lastPos) << ':';
1071*67e74705SXin Li     PrintObjCMethodType(OMD->getASTContext(),
1072*67e74705SXin Li                         PI->getObjCDeclQualifier(),
1073*67e74705SXin Li                         PI->getType());
1074*67e74705SXin Li     Out << *PI;
1075*67e74705SXin Li     lastPos = pos + 1;
1076*67e74705SXin Li   }
1077*67e74705SXin Li 
1078*67e74705SXin Li   if (OMD->param_begin() == OMD->param_end())
1079*67e74705SXin Li     Out << " " << name;
1080*67e74705SXin Li 
1081*67e74705SXin Li   if (OMD->isVariadic())
1082*67e74705SXin Li       Out << ", ...";
1083*67e74705SXin Li 
1084*67e74705SXin Li   prettyPrintAttributes(OMD);
1085*67e74705SXin Li 
1086*67e74705SXin Li   if (OMD->getBody() && !Policy.TerseOutput) {
1087*67e74705SXin Li     Out << ' ';
1088*67e74705SXin Li     OMD->getBody()->printPretty(Out, nullptr, Policy);
1089*67e74705SXin Li   }
1090*67e74705SXin Li   else if (Policy.PolishForDeclaration)
1091*67e74705SXin Li     Out << ';';
1092*67e74705SXin Li }
1093*67e74705SXin Li 
VisitObjCImplementationDecl(ObjCImplementationDecl * OID)1094*67e74705SXin Li void DeclPrinter::VisitObjCImplementationDecl(ObjCImplementationDecl *OID) {
1095*67e74705SXin Li   std::string I = OID->getNameAsString();
1096*67e74705SXin Li   ObjCInterfaceDecl *SID = OID->getSuperClass();
1097*67e74705SXin Li 
1098*67e74705SXin Li   bool eolnOut = false;
1099*67e74705SXin Li   if (SID)
1100*67e74705SXin Li     Out << "@implementation " << I << " : " << *SID;
1101*67e74705SXin Li   else
1102*67e74705SXin Li     Out << "@implementation " << I;
1103*67e74705SXin Li 
1104*67e74705SXin Li   if (OID->ivar_size() > 0) {
1105*67e74705SXin Li     Out << "{\n";
1106*67e74705SXin Li     eolnOut = true;
1107*67e74705SXin Li     Indentation += Policy.Indentation;
1108*67e74705SXin Li     for (const auto *I : OID->ivars()) {
1109*67e74705SXin Li       Indent() << I->getASTContext().getUnqualifiedObjCPointerType(I->getType()).
1110*67e74705SXin Li                     getAsString(Policy) << ' ' << *I << ";\n";
1111*67e74705SXin Li     }
1112*67e74705SXin Li     Indentation -= Policy.Indentation;
1113*67e74705SXin Li     Out << "}\n";
1114*67e74705SXin Li   }
1115*67e74705SXin Li   else if (SID || (OID->decls_begin() != OID->decls_end())) {
1116*67e74705SXin Li     Out << "\n";
1117*67e74705SXin Li     eolnOut = true;
1118*67e74705SXin Li   }
1119*67e74705SXin Li   VisitDeclContext(OID, false);
1120*67e74705SXin Li   if (!eolnOut)
1121*67e74705SXin Li     Out << "\n";
1122*67e74705SXin Li   Out << "@end";
1123*67e74705SXin Li }
1124*67e74705SXin Li 
VisitObjCInterfaceDecl(ObjCInterfaceDecl * OID)1125*67e74705SXin Li void DeclPrinter::VisitObjCInterfaceDecl(ObjCInterfaceDecl *OID) {
1126*67e74705SXin Li   std::string I = OID->getNameAsString();
1127*67e74705SXin Li   ObjCInterfaceDecl *SID = OID->getSuperClass();
1128*67e74705SXin Li 
1129*67e74705SXin Li   if (!OID->isThisDeclarationADefinition()) {
1130*67e74705SXin Li     Out << "@class " << I;
1131*67e74705SXin Li 
1132*67e74705SXin Li     if (auto TypeParams = OID->getTypeParamListAsWritten()) {
1133*67e74705SXin Li       PrintObjCTypeParams(TypeParams);
1134*67e74705SXin Li     }
1135*67e74705SXin Li 
1136*67e74705SXin Li     Out << ";";
1137*67e74705SXin Li     return;
1138*67e74705SXin Li   }
1139*67e74705SXin Li   bool eolnOut = false;
1140*67e74705SXin Li   Out << "@interface " << I;
1141*67e74705SXin Li 
1142*67e74705SXin Li   if (auto TypeParams = OID->getTypeParamListAsWritten()) {
1143*67e74705SXin Li     PrintObjCTypeParams(TypeParams);
1144*67e74705SXin Li   }
1145*67e74705SXin Li 
1146*67e74705SXin Li   if (SID)
1147*67e74705SXin Li     Out << " : " << QualType(OID->getSuperClassType(), 0).getAsString(Policy);
1148*67e74705SXin Li 
1149*67e74705SXin Li   // Protocols?
1150*67e74705SXin Li   const ObjCList<ObjCProtocolDecl> &Protocols = OID->getReferencedProtocols();
1151*67e74705SXin Li   if (!Protocols.empty()) {
1152*67e74705SXin Li     for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
1153*67e74705SXin Li          E = Protocols.end(); I != E; ++I)
1154*67e74705SXin Li       Out << (I == Protocols.begin() ? '<' : ',') << **I;
1155*67e74705SXin Li     Out << "> ";
1156*67e74705SXin Li   }
1157*67e74705SXin Li 
1158*67e74705SXin Li   if (OID->ivar_size() > 0) {
1159*67e74705SXin Li     Out << "{\n";
1160*67e74705SXin Li     eolnOut = true;
1161*67e74705SXin Li     Indentation += Policy.Indentation;
1162*67e74705SXin Li     for (const auto *I : OID->ivars()) {
1163*67e74705SXin Li       Indent() << I->getASTContext()
1164*67e74705SXin Li                       .getUnqualifiedObjCPointerType(I->getType())
1165*67e74705SXin Li                       .getAsString(Policy) << ' ' << *I << ";\n";
1166*67e74705SXin Li     }
1167*67e74705SXin Li     Indentation -= Policy.Indentation;
1168*67e74705SXin Li     Out << "}\n";
1169*67e74705SXin Li   }
1170*67e74705SXin Li   else if (SID || (OID->decls_begin() != OID->decls_end())) {
1171*67e74705SXin Li     Out << "\n";
1172*67e74705SXin Li     eolnOut = true;
1173*67e74705SXin Li   }
1174*67e74705SXin Li 
1175*67e74705SXin Li   VisitDeclContext(OID, false);
1176*67e74705SXin Li   if (!eolnOut)
1177*67e74705SXin Li     Out << "\n";
1178*67e74705SXin Li   Out << "@end";
1179*67e74705SXin Li   // FIXME: implement the rest...
1180*67e74705SXin Li }
1181*67e74705SXin Li 
VisitObjCProtocolDecl(ObjCProtocolDecl * PID)1182*67e74705SXin Li void DeclPrinter::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
1183*67e74705SXin Li   if (!PID->isThisDeclarationADefinition()) {
1184*67e74705SXin Li     Out << "@protocol " << *PID << ";\n";
1185*67e74705SXin Li     return;
1186*67e74705SXin Li   }
1187*67e74705SXin Li   // Protocols?
1188*67e74705SXin Li   const ObjCList<ObjCProtocolDecl> &Protocols = PID->getReferencedProtocols();
1189*67e74705SXin Li   if (!Protocols.empty()) {
1190*67e74705SXin Li     Out << "@protocol " << *PID;
1191*67e74705SXin Li     for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
1192*67e74705SXin Li          E = Protocols.end(); I != E; ++I)
1193*67e74705SXin Li       Out << (I == Protocols.begin() ? '<' : ',') << **I;
1194*67e74705SXin Li     Out << ">\n";
1195*67e74705SXin Li   } else
1196*67e74705SXin Li     Out << "@protocol " << *PID << '\n';
1197*67e74705SXin Li   VisitDeclContext(PID, false);
1198*67e74705SXin Li   Out << "@end";
1199*67e74705SXin Li }
1200*67e74705SXin Li 
VisitObjCCategoryImplDecl(ObjCCategoryImplDecl * PID)1201*67e74705SXin Li void DeclPrinter::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *PID) {
1202*67e74705SXin Li   Out << "@implementation " << *PID->getClassInterface() << '(' << *PID <<")\n";
1203*67e74705SXin Li 
1204*67e74705SXin Li   VisitDeclContext(PID, false);
1205*67e74705SXin Li   Out << "@end";
1206*67e74705SXin Li   // FIXME: implement the rest...
1207*67e74705SXin Li }
1208*67e74705SXin Li 
VisitObjCCategoryDecl(ObjCCategoryDecl * PID)1209*67e74705SXin Li void DeclPrinter::VisitObjCCategoryDecl(ObjCCategoryDecl *PID) {
1210*67e74705SXin Li   Out << "@interface " << *PID->getClassInterface();
1211*67e74705SXin Li   if (auto TypeParams = PID->getTypeParamList()) {
1212*67e74705SXin Li     PrintObjCTypeParams(TypeParams);
1213*67e74705SXin Li   }
1214*67e74705SXin Li   Out << "(" << *PID << ")\n";
1215*67e74705SXin Li   if (PID->ivar_size() > 0) {
1216*67e74705SXin Li     Out << "{\n";
1217*67e74705SXin Li     Indentation += Policy.Indentation;
1218*67e74705SXin Li     for (const auto *I : PID->ivars())
1219*67e74705SXin Li       Indent() << I->getASTContext().getUnqualifiedObjCPointerType(I->getType()).
1220*67e74705SXin Li                     getAsString(Policy) << ' ' << *I << ";\n";
1221*67e74705SXin Li     Indentation -= Policy.Indentation;
1222*67e74705SXin Li     Out << "}\n";
1223*67e74705SXin Li   }
1224*67e74705SXin Li 
1225*67e74705SXin Li   VisitDeclContext(PID, false);
1226*67e74705SXin Li   Out << "@end";
1227*67e74705SXin Li 
1228*67e74705SXin Li   // FIXME: implement the rest...
1229*67e74705SXin Li }
1230*67e74705SXin Li 
VisitObjCCompatibleAliasDecl(ObjCCompatibleAliasDecl * AID)1231*67e74705SXin Li void DeclPrinter::VisitObjCCompatibleAliasDecl(ObjCCompatibleAliasDecl *AID) {
1232*67e74705SXin Li   Out << "@compatibility_alias " << *AID
1233*67e74705SXin Li       << ' ' << *AID->getClassInterface() << ";\n";
1234*67e74705SXin Li }
1235*67e74705SXin Li 
1236*67e74705SXin Li /// PrintObjCPropertyDecl - print a property declaration.
1237*67e74705SXin Li ///
VisitObjCPropertyDecl(ObjCPropertyDecl * PDecl)1238*67e74705SXin Li void DeclPrinter::VisitObjCPropertyDecl(ObjCPropertyDecl *PDecl) {
1239*67e74705SXin Li   if (PDecl->getPropertyImplementation() == ObjCPropertyDecl::Required)
1240*67e74705SXin Li     Out << "@required\n";
1241*67e74705SXin Li   else if (PDecl->getPropertyImplementation() == ObjCPropertyDecl::Optional)
1242*67e74705SXin Li     Out << "@optional\n";
1243*67e74705SXin Li 
1244*67e74705SXin Li   QualType T = PDecl->getType();
1245*67e74705SXin Li 
1246*67e74705SXin Li   Out << "@property";
1247*67e74705SXin Li   if (PDecl->getPropertyAttributes() != ObjCPropertyDecl::OBJC_PR_noattr) {
1248*67e74705SXin Li     bool first = true;
1249*67e74705SXin Li     Out << " (";
1250*67e74705SXin Li     if (PDecl->getPropertyAttributes() &
1251*67e74705SXin Li         ObjCPropertyDecl::OBJC_PR_readonly) {
1252*67e74705SXin Li       Out << (first ? ' ' : ',') << "readonly";
1253*67e74705SXin Li       first = false;
1254*67e74705SXin Li     }
1255*67e74705SXin Li 
1256*67e74705SXin Li     if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_getter) {
1257*67e74705SXin Li       Out << (first ? ' ' : ',') << "getter = ";
1258*67e74705SXin Li       PDecl->getGetterName().print(Out);
1259*67e74705SXin Li       first = false;
1260*67e74705SXin Li     }
1261*67e74705SXin Li     if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_setter) {
1262*67e74705SXin Li       Out << (first ? ' ' : ',') << "setter = ";
1263*67e74705SXin Li       PDecl->getSetterName().print(Out);
1264*67e74705SXin Li       first = false;
1265*67e74705SXin Li     }
1266*67e74705SXin Li 
1267*67e74705SXin Li     if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_assign) {
1268*67e74705SXin Li       Out << (first ? ' ' : ',') << "assign";
1269*67e74705SXin Li       first = false;
1270*67e74705SXin Li     }
1271*67e74705SXin Li 
1272*67e74705SXin Li     if (PDecl->getPropertyAttributes() &
1273*67e74705SXin Li         ObjCPropertyDecl::OBJC_PR_readwrite) {
1274*67e74705SXin Li       Out << (first ? ' ' : ',') << "readwrite";
1275*67e74705SXin Li       first = false;
1276*67e74705SXin Li     }
1277*67e74705SXin Li 
1278*67e74705SXin Li     if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_retain) {
1279*67e74705SXin Li       Out << (first ? ' ' : ',') << "retain";
1280*67e74705SXin Li       first = false;
1281*67e74705SXin Li     }
1282*67e74705SXin Li 
1283*67e74705SXin Li     if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_strong) {
1284*67e74705SXin Li       Out << (first ? ' ' : ',') << "strong";
1285*67e74705SXin Li       first = false;
1286*67e74705SXin Li     }
1287*67e74705SXin Li 
1288*67e74705SXin Li     if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_copy) {
1289*67e74705SXin Li       Out << (first ? ' ' : ',') << "copy";
1290*67e74705SXin Li       first = false;
1291*67e74705SXin Li     }
1292*67e74705SXin Li 
1293*67e74705SXin Li     if (PDecl->getPropertyAttributes() &
1294*67e74705SXin Li         ObjCPropertyDecl::OBJC_PR_nonatomic) {
1295*67e74705SXin Li       Out << (first ? ' ' : ',') << "nonatomic";
1296*67e74705SXin Li       first = false;
1297*67e74705SXin Li     }
1298*67e74705SXin Li     if (PDecl->getPropertyAttributes() &
1299*67e74705SXin Li         ObjCPropertyDecl::OBJC_PR_atomic) {
1300*67e74705SXin Li       Out << (first ? ' ' : ',') << "atomic";
1301*67e74705SXin Li       first = false;
1302*67e74705SXin Li     }
1303*67e74705SXin Li 
1304*67e74705SXin Li     if (PDecl->getPropertyAttributes() &
1305*67e74705SXin Li         ObjCPropertyDecl::OBJC_PR_nullability) {
1306*67e74705SXin Li       if (auto nullability = AttributedType::stripOuterNullability(T)) {
1307*67e74705SXin Li         if (*nullability == NullabilityKind::Unspecified &&
1308*67e74705SXin Li             (PDecl->getPropertyAttributes() &
1309*67e74705SXin Li                ObjCPropertyDecl::OBJC_PR_null_resettable)) {
1310*67e74705SXin Li           Out << (first ? ' ' : ',') << "null_resettable";
1311*67e74705SXin Li         } else {
1312*67e74705SXin Li           Out << (first ? ' ' : ',')
1313*67e74705SXin Li               << getNullabilitySpelling(*nullability, true);
1314*67e74705SXin Li         }
1315*67e74705SXin Li         first = false;
1316*67e74705SXin Li       }
1317*67e74705SXin Li     }
1318*67e74705SXin Li 
1319*67e74705SXin Li     if (PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_class) {
1320*67e74705SXin Li       Out << (first ? ' ' : ',') << "class";
1321*67e74705SXin Li       first = false;
1322*67e74705SXin Li     }
1323*67e74705SXin Li 
1324*67e74705SXin Li     (void) first; // Silence dead store warning due to idiomatic code.
1325*67e74705SXin Li     Out << " )";
1326*67e74705SXin Li   }
1327*67e74705SXin Li   Out << ' ' << PDecl->getASTContext().getUnqualifiedObjCPointerType(T).
1328*67e74705SXin Li                   getAsString(Policy) << ' ' << *PDecl;
1329*67e74705SXin Li   if (Policy.PolishForDeclaration)
1330*67e74705SXin Li     Out << ';';
1331*67e74705SXin Li }
1332*67e74705SXin Li 
VisitObjCPropertyImplDecl(ObjCPropertyImplDecl * PID)1333*67e74705SXin Li void DeclPrinter::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PID) {
1334*67e74705SXin Li   if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Synthesize)
1335*67e74705SXin Li     Out << "@synthesize ";
1336*67e74705SXin Li   else
1337*67e74705SXin Li     Out << "@dynamic ";
1338*67e74705SXin Li   Out << *PID->getPropertyDecl();
1339*67e74705SXin Li   if (PID->getPropertyIvarDecl())
1340*67e74705SXin Li     Out << '=' << *PID->getPropertyIvarDecl();
1341*67e74705SXin Li }
1342*67e74705SXin Li 
VisitUsingDecl(UsingDecl * D)1343*67e74705SXin Li void DeclPrinter::VisitUsingDecl(UsingDecl *D) {
1344*67e74705SXin Li   if (!D->isAccessDeclaration())
1345*67e74705SXin Li     Out << "using ";
1346*67e74705SXin Li   if (D->hasTypename())
1347*67e74705SXin Li     Out << "typename ";
1348*67e74705SXin Li   D->getQualifier()->print(Out, Policy);
1349*67e74705SXin Li   Out << *D;
1350*67e74705SXin Li }
1351*67e74705SXin Li 
1352*67e74705SXin Li void
VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl * D)1353*67e74705SXin Li DeclPrinter::VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D) {
1354*67e74705SXin Li   Out << "using typename ";
1355*67e74705SXin Li   D->getQualifier()->print(Out, Policy);
1356*67e74705SXin Li   Out << D->getDeclName();
1357*67e74705SXin Li }
1358*67e74705SXin Li 
VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl * D)1359*67e74705SXin Li void DeclPrinter::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1360*67e74705SXin Li   if (!D->isAccessDeclaration())
1361*67e74705SXin Li     Out << "using ";
1362*67e74705SXin Li   D->getQualifier()->print(Out, Policy);
1363*67e74705SXin Li   Out << D->getDeclName();
1364*67e74705SXin Li }
1365*67e74705SXin Li 
VisitUsingShadowDecl(UsingShadowDecl * D)1366*67e74705SXin Li void DeclPrinter::VisitUsingShadowDecl(UsingShadowDecl *D) {
1367*67e74705SXin Li   // ignore
1368*67e74705SXin Li }
1369*67e74705SXin Li 
VisitOMPThreadPrivateDecl(OMPThreadPrivateDecl * D)1370*67e74705SXin Li void DeclPrinter::VisitOMPThreadPrivateDecl(OMPThreadPrivateDecl *D) {
1371*67e74705SXin Li   Out << "#pragma omp threadprivate";
1372*67e74705SXin Li   if (!D->varlist_empty()) {
1373*67e74705SXin Li     for (OMPThreadPrivateDecl::varlist_iterator I = D->varlist_begin(),
1374*67e74705SXin Li                                                 E = D->varlist_end();
1375*67e74705SXin Li                                                 I != E; ++I) {
1376*67e74705SXin Li       Out << (I == D->varlist_begin() ? '(' : ',');
1377*67e74705SXin Li       NamedDecl *ND = cast<NamedDecl>(cast<DeclRefExpr>(*I)->getDecl());
1378*67e74705SXin Li       ND->printQualifiedName(Out);
1379*67e74705SXin Li     }
1380*67e74705SXin Li     Out << ")";
1381*67e74705SXin Li   }
1382*67e74705SXin Li }
1383*67e74705SXin Li 
VisitOMPDeclareReductionDecl(OMPDeclareReductionDecl * D)1384*67e74705SXin Li void DeclPrinter::VisitOMPDeclareReductionDecl(OMPDeclareReductionDecl *D) {
1385*67e74705SXin Li   if (!D->isInvalidDecl()) {
1386*67e74705SXin Li     Out << "#pragma omp declare reduction (";
1387*67e74705SXin Li     if (D->getDeclName().getNameKind() == DeclarationName::CXXOperatorName) {
1388*67e74705SXin Li       static const char *const OperatorNames[NUM_OVERLOADED_OPERATORS] = {
1389*67e74705SXin Li           nullptr,
1390*67e74705SXin Li #define OVERLOADED_OPERATOR(Name, Spelling, Token, Unary, Binary, MemberOnly)  \
1391*67e74705SXin Li           Spelling,
1392*67e74705SXin Li #include "clang/Basic/OperatorKinds.def"
1393*67e74705SXin Li       };
1394*67e74705SXin Li       const char *OpName =
1395*67e74705SXin Li           OperatorNames[D->getDeclName().getCXXOverloadedOperator()];
1396*67e74705SXin Li       assert(OpName && "not an overloaded operator");
1397*67e74705SXin Li       Out << OpName;
1398*67e74705SXin Li     } else {
1399*67e74705SXin Li       assert(D->getDeclName().isIdentifier());
1400*67e74705SXin Li       D->printName(Out);
1401*67e74705SXin Li     }
1402*67e74705SXin Li     Out << " : ";
1403*67e74705SXin Li     D->getType().print(Out, Policy);
1404*67e74705SXin Li     Out << " : ";
1405*67e74705SXin Li     D->getCombiner()->printPretty(Out, nullptr, Policy, 0);
1406*67e74705SXin Li     Out << ")";
1407*67e74705SXin Li     if (auto *Init = D->getInitializer()) {
1408*67e74705SXin Li       Out << " initializer(";
1409*67e74705SXin Li       Init->printPretty(Out, nullptr, Policy, 0);
1410*67e74705SXin Li       Out << ")";
1411*67e74705SXin Li     }
1412*67e74705SXin Li   }
1413*67e74705SXin Li }
1414*67e74705SXin Li 
VisitOMPCapturedExprDecl(OMPCapturedExprDecl * D)1415*67e74705SXin Li void DeclPrinter::VisitOMPCapturedExprDecl(OMPCapturedExprDecl *D) {
1416*67e74705SXin Li   D->getInit()->printPretty(Out, nullptr, Policy, Indentation);
1417*67e74705SXin Li }
1418*67e74705SXin Li 
1419