//===--- ASTPrinter.cpp - Swift Language AST Printer ----------------------===// // // This source file is part of the Swift.org open source project // // Copyright (c) 2014 - 2020 Apple Inc. and the Swift project authors // Licensed under Apache License v2.0 with Runtime Library Exception // // See https://swift.org/LICENSE.txt for license information // See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors // //===----------------------------------------------------------------------===// // // This file implements printing for the Swift ASTs. // //===----------------------------------------------------------------------===// #include "swift/AST/ASTPrinter.h" #include "swift/AST/ASTContext.h" #include "swift/AST/ASTMangler.h" #include "swift/AST/ASTVisitor.h" #include "swift/AST/Attr.h" #include "swift/AST/Builtins.h" #include "swift/AST/ClangModuleLoader.h" #include "swift/AST/Comment.h" #include "swift/AST/Decl.h" #include "swift/AST/ExistentialLayout.h" #include "swift/AST/Expr.h" #include "swift/AST/FileUnit.h" #include "swift/AST/GenericParamList.h" #include "swift/AST/GenericSignature.h" #include "swift/AST/Module.h" #include "swift/AST/NameLookup.h" #include "swift/AST/NameLookupRequests.h" #include "swift/AST/ParameterList.h" #include "swift/AST/PrintOptions.h" #include "swift/AST/ProtocolConformance.h" #include "swift/AST/SILLayout.h" #include "swift/AST/Stmt.h" #include "swift/AST/TypeVisitor.h" #include "swift/AST/TypeWalker.h" #include "swift/AST/Types.h" #include "swift/Basic/Defer.h" #include "swift/Basic/Feature.h" #include "swift/Basic/FixedBitSet.h" #include "swift/Basic/PrimitiveParsing.h" #include "swift/Basic/QuotedString.h" #include "swift/Basic/STLExtras.h" #include "swift/Basic/StringExtras.h" #include "swift/Basic/Unicode.h" #include "swift/ClangImporter/ClangImporterRequests.h" #include "swift/Config.h" #include "swift/Parse/Lexer.h" #include "swift/Strings.h" #include "clang/AST/ASTContext.h" #include "clang/AST/Attr.h" #include "clang/AST/Decl.h" #include "clang/AST/DeclObjC.h" #include "clang/Basic/Module.h" #include "clang/Basic/SourceManager.h" #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/StringSwitch.h" #include "llvm/Support/Compiler.h" #include "llvm/Support/ConvertUTF.h" #include "llvm/Support/SaveAndRestore.h" #include "llvm/Support/raw_ostream.h" #include #include using namespace swift; // Defined here to avoid repeatedly paying the price of template instantiation. const std::function PrintOptions::defaultPrintExtensionContentAsMembers = [] (const ExtensionDecl *) { return false; }; void PrintOptions::setBaseType(Type T) { if (T->is()) return; TransformContext = TypeTransformContext(T); } void PrintOptions::initForSynthesizedExtension(TypeOrExtensionDecl D) { TransformContext = TypeTransformContext(D); } void PrintOptions::clearSynthesizedExtension() { TransformContext.reset(); } static bool isPublicOrUsableFromInline(const ValueDecl *VD) { AccessScope scope = VD->getFormalAccessScope(/*useDC*/nullptr, /*treatUsableFromInlineAsPublic*/true); return scope.isPublic(); } static bool isPublicOrUsableFromInline(Type ty) { // Note the double negative here: we're looking for any referenced decls that // are *not* public-or-usableFromInline. return !ty.findIf([](Type typePart) -> bool { // FIXME: If we have an internal typealias for a non-internal type, we ought // to be able to print it by desugaring. if (auto *aliasTy = dyn_cast(typePart.getPointer())) return !isPublicOrUsableFromInline(aliasTy->getDecl()); if (auto *nominal = typePart->getAnyNominal()) return !isPublicOrUsableFromInline(nominal); return false; }); } static bool isPrespecilizationDeclWithTarget(const ValueDecl *vd) { // Add exported prespecialized symbols. for (auto *attr : vd->getAttrs().getAttributes()) { if (!attr->isExported()) continue; if (auto *targetFun = attr->getTargetFunctionDecl(vd)) return true; } return false; } static bool contributesToParentTypeStorage(const AbstractStorageDecl *ASD) { auto *DC = ASD->getDeclContext()->getAsDecl(); if (!DC) return false; auto *ND = dyn_cast(DC); if (!ND) return false; return !ND->isResilient() && ASD->hasStorage() && !ASD->isStatic(); } PrintOptions PrintOptions::printSwiftInterfaceFile(ModuleDecl *ModuleToPrint, bool preferTypeRepr, bool printFullConvention, bool printSPIs) { PrintOptions result; result.IsForSwiftInterface = true; result.PrintLongAttrsOnSeparateLines = true; result.TypeDefinitions = true; result.PrintIfConfig = false; result.CurrentModule = ModuleToPrint; result.FullyQualifiedTypes = true; result.FullyQualifiedTypesIfAmbiguous = true; result.FullyQualifiedExtendedTypesIfAmbiguous = true; result.UseExportedModuleNames = true; result.AllowNullTypes = false; result.SkipImports = true; result.OmitNameOfInaccessibleProperties = true; result.FunctionDefinitions = true; result.CollapseSingleGetterProperty = false; result.VarInitializers = true; result.EnumRawValues = EnumRawValueMode::PrintObjCOnly; result.OpaqueReturnTypePrinting = OpaqueReturnTypePrintingMode::StableReference; result.PreferTypeRepr = preferTypeRepr; if (printFullConvention) result.PrintFunctionRepresentationAttrs = PrintOptions::FunctionRepresentationMode::Full; result.AlwaysTryPrintParameterLabels = true; result.PrintSPIs = printSPIs; result.PrintExplicitAny = true; result.DesugarExistentialConstraint = true; // We should print __consuming, __owned, etc for the module interface file. result.SkipUnderscoredKeywords = false; // We should provide backward-compatible Swift interfaces when we can. result.PrintCompatibilityFeatureChecks = true; result.FunctionBody = [](const ValueDecl *decl, ASTPrinter &printer) { auto AFD = dyn_cast(decl); if (!AFD) return; if (AFD->getResilienceExpansion() != ResilienceExpansion::Minimal) return; if (!AFD->hasInlinableBodyText()) return; SmallString<128> scratch; printer << " " << AFD->getInlinableBodyText(scratch); }; class ShouldPrintForModuleInterface : public ShouldPrintChecker { bool shouldPrint(const Decl *D, const PrintOptions &options) override { if (!D) return false; // Skip anything that is marked `@_implementationOnly` itself. if (D->getAttrs().hasAttribute()) return false; // Skip SPI decls if `PrintSPIs`. if (!options.PrintSPIs && D->isSPI()) return false; // Skip anything that isn't 'public' or '@usableFromInline' or has a // _specialize attribute with a targetFunction parameter. if (auto *VD = dyn_cast(D)) { if (!isPublicOrUsableFromInline(VD) && !isPrespecilizationDeclWithTarget(VD)) { // We do want to print private stored properties, without their // original names present. if (auto *ASD = dyn_cast(VD)) if (contributesToParentTypeStorage(ASD)) return true; return false; } } // Skip extensions that extend things we wouldn't print. if (auto *ED = dyn_cast(D)) { if (!shouldPrint(ED->getExtendedNominal(), options)) return false; // Skip extensions to implementation-only imported types that have // no public members. auto localModule = ED->getParentModule(); auto nominalModule = ED->getExtendedNominal()->getParentModule(); if (localModule != nominalModule && localModule->isImportedImplementationOnly(nominalModule)) { bool shouldPrintMembers = llvm::any_of( ED->getAllMembers(), [&](const Decl *member) -> bool { return shouldPrint(member, options); }); if (!shouldPrintMembers) return false; } for (const Requirement &req : ED->getGenericRequirements()) { if (!isPublicOrUsableFromInline(req.getFirstType())) return false; switch (req.getKind()) { case RequirementKind::Conformance: case RequirementKind::Superclass: case RequirementKind::SameType: if (!isPublicOrUsableFromInline(req.getSecondType())) return false; break; case RequirementKind::Layout: break; } } } // Skip typealiases that just redeclare generic parameters. if (auto *alias = dyn_cast(D)) { if (alias->isImplicit()) { const Decl *parent = D->getDeclContext()->getAsDecl(); if (auto *genericCtx = parent->getAsGenericContext()) { bool matchesGenericParam = llvm::any_of(genericCtx->getInnermostGenericParamTypes(), [alias](const GenericTypeParamType *param) { return param->getName() == alias->getName(); }); if (matchesGenericParam) return false; } } } // Skip stub constructors. if (auto *ctor = dyn_cast(D)) { if (ctor->hasStubImplementation()) return false; } return ShouldPrintChecker::shouldPrint(D, options); } }; result.CurrentPrintabilityChecker = std::make_shared(); // FIXME: We don't really need 'public' on everything; we could just change // the default to 'public' and mark the 'internal' things. result.PrintAccess = true; result.ExcludeAttrList = { DAK_AccessControl, DAK_SetterAccess, DAK_Lazy, DAK_StaticInitializeObjCMetadata, DAK_RestatedObjCConformance, DAK_NonSendable, }; return result; } TypeTransformContext::TypeTransformContext(Type T) : BaseType(T.getPointer()) { assert(T->mayHaveMembers()); } TypeTransformContext::TypeTransformContext(TypeOrExtensionDecl D) : BaseType(nullptr), Decl(D) { if (auto NTD = Decl.Decl.dyn_cast()) BaseType = NTD->getDeclaredTypeInContext().getPointer(); else { auto *ED = Decl.Decl.get(); BaseType = ED->getDeclaredTypeInContext().getPointer(); } } TypeOrExtensionDecl TypeTransformContext::getDecl() const { return Decl; } DeclContext *TypeTransformContext::getDeclContext() const { return Decl.getAsDecl()->getDeclContext(); } Type TypeTransformContext::getBaseType() const { return Type(BaseType); } bool TypeTransformContext::isPrintingSynthesizedExtension() const { return !Decl.isNull(); } void ASTPrinter::anchor() {} void ASTPrinter::printIndent() { llvm::SmallString<16> Str; for (unsigned i = 0; i != CurrentIndentation; ++i) Str += ' '; printText(Str); } void ASTPrinter::printTextImpl(StringRef Text) { forceNewlines(); printText(Text); } void ASTPrinter::printEscapedStringLiteral(StringRef str) { SmallString<128> encodeBuf; StringRef escaped = Lexer::getEncodedStringSegment(str, encodeBuf, /*isFirstSegment*/true, /*isLastSegment*/true, /*indentToStrip*/~0U /* sentinel */); // FIXME: This is wasteful, but ASTPrinter is an abstract class that doesn't // have a directly-accessible ostream. SmallString<128> escapeBuf; llvm::raw_svector_ostream os(escapeBuf); os << QuotedString(escaped); printTextImpl(escapeBuf.str()); } void ASTPrinter::printTypeRef(Type T, const TypeDecl *RefTo, Identifier Name, PrintNameContext Context) { if (isa(RefTo)) { Context = PrintNameContext::GenericParameter; } else if (T && T->is()) { assert(T->castTo()->getSelfType()->getAnyNominal() && "protocol Self handled as GenericTypeParamDecl"); Context = PrintNameContext::ClassDynamicSelf; } printName(Name, Context); } void ASTPrinter::printModuleRef(ModuleEntity Mod, Identifier Name) { printName(Name); } void ASTPrinter::callPrintDeclPre(const Decl *D, Optional Bracket) { forceNewlines(); if (SynthesizeTarget && isa(D)) printSynthesizedExtensionPre(cast(D), SynthesizeTarget, Bracket); else printDeclPre(D, Bracket); } ASTPrinter &ASTPrinter::operator<<(QuotedString s) { llvm::SmallString<32> Str; llvm::raw_svector_ostream OS(Str); OS << s; printTextImpl(OS.str()); return *this; } ASTPrinter &ASTPrinter::operator<<(unsigned long long N) { llvm::SmallString<32> Str; llvm::raw_svector_ostream OS(Str); OS << N; printTextImpl(OS.str()); return *this; } ASTPrinter &ASTPrinter::operator<<(UUID UU) { llvm::SmallString Str; UU.toString(Str); printTextImpl(Str); return *this; } ASTPrinter &ASTPrinter::operator<<(Identifier name) { return *this << DeclName(name); } ASTPrinter &ASTPrinter::operator<<(DeclBaseName name) { return *this << DeclName(name); } ASTPrinter &ASTPrinter::operator<<(DeclName name) { llvm::SmallString<32> str; llvm::raw_svector_ostream os(str); name.print(os); printTextImpl(os.str()); return *this; } ASTPrinter &ASTPrinter::operator<<(DeclNameRef ref) { llvm::SmallString<32> str; llvm::raw_svector_ostream os(str); ref.print(os); printTextImpl(os.str()); return *this; } llvm::raw_ostream &swift:: operator<<(llvm::raw_ostream &OS, tok keyword) { switch (keyword) { #define KEYWORD(KW) case tok::kw_##KW: OS << #KW; break; #define POUND_KEYWORD(KW) case tok::pound_##KW: OS << "#"#KW; break; #define PUNCTUATOR(PUN, TEXT) case tok::PUN: OS << TEXT; break; #include "swift/Syntax/TokenKinds.def" default: llvm_unreachable("unexpected keyword or punctuator kind"); } return OS; } uint8_t swift::getKeywordLen(tok keyword) { switch (keyword) { #define KEYWORD(KW) case tok::kw_##KW: return StringRef(#KW).size(); #define POUND_KEYWORD(KW) case tok::pound_##KW: return StringRef("#"#KW).size(); #define PUNCTUATOR(PUN, TEXT) case tok::PUN: return StringRef(TEXT).size(); #include "swift/Syntax/TokenKinds.def" default: llvm_unreachable("unexpected keyword or punctuator kind"); } } StringRef swift::getCodePlaceholder() { return "<#code#>"; } ASTPrinter &operator<<(ASTPrinter &printer, tok keyword) { SmallString<16> Buffer; llvm::raw_svector_ostream OS(Buffer); OS << keyword; printer.printKeyword(Buffer.str(), PrintOptions()); return printer; } /// Determine whether to escape the given keyword in the given context. static bool escapeKeywordInContext(StringRef keyword, PrintNameContext context){ bool isKeyword = llvm::StringSwitch(keyword) #define KEYWORD(KW) \ .Case(#KW, true) #include "swift/Syntax/TokenKinds.def" .Default(false); switch (context) { case PrintNameContext::Normal: case PrintNameContext::Attribute: return isKeyword; case PrintNameContext::Keyword: case PrintNameContext::IntroducerKeyword: return false; case PrintNameContext::ClassDynamicSelf: case PrintNameContext::GenericParameter: return isKeyword && keyword != "Self"; case PrintNameContext::TypeMember: return isKeyword || !canBeMemberName(keyword); case PrintNameContext::FunctionParameterExternal: case PrintNameContext::FunctionParameterLocal: case PrintNameContext::TupleElement: return !canBeArgumentLabel(keyword); } llvm_unreachable("Unhandled PrintNameContext in switch."); } void ASTPrinter::printName(Identifier Name, PrintNameContext Context) { callPrintNamePre(Context); if (Name.empty()) { *this << "_"; printNamePost(Context); return; } bool shouldEscapeKeyword = escapeKeywordInContext(Name.str(), Context); if (shouldEscapeKeyword) *this << "`"; *this << Name.str(); if (shouldEscapeKeyword) *this << "`"; printNamePost(Context); } void StreamPrinter::printText(StringRef Text) { OS << Text; } /// Whether we will be printing a TypeLoc by using the TypeRepr printer static bool willUseTypeReprPrinting(TypeLoc tyLoc, Type currentType, const PrintOptions &options) { // Special case for when transforming archetypes if (currentType && tyLoc.getType()) return false; return ((options.PreferTypeRepr && tyLoc.hasLocation()) || (tyLoc.getType().isNull() && tyLoc.getTypeRepr())); } namespace { /// AST pretty-printer. class PrintAST : public ASTVisitor { ASTPrinter &Printer; PrintOptions Options; unsigned IndentLevel = 0; Decl *Current = nullptr; Type CurrentType; void setCurrentType(Type NewCurrentType) { CurrentType = NewCurrentType; assert(CurrentType.isNull() || !CurrentType->hasArchetype() && "CurrentType should be an interface type"); } friend DeclVisitor; /// RAII object that increases the indentation level. class IndentRAII { PrintAST &Self; bool DoIndent; public: IndentRAII(PrintAST &self, bool DoIndent = true) : Self(self), DoIndent(DoIndent) { if (DoIndent) Self.IndentLevel += Self.Options.Indent; } ~IndentRAII() { if (DoIndent) Self.IndentLevel -= Self.Options.Indent; } }; /// Indent the current number of indentation spaces. void indent() { Printer.setIndent(IndentLevel); } /// Record the location of this declaration, which is about to /// be printed, marking the name and signature end locations. template void recordDeclLoc(Decl *decl, const FnTy &NameFn, llvm::function_ref ParamFn = []{}) { Printer.callPrintDeclLoc(decl); NameFn(); Printer.printDeclNameEndLoc(decl); ParamFn(); Printer.printDeclNameOrSignatureEndLoc(decl); } void printSourceRange(CharSourceRange Range, ASTContext &Ctx) { Printer << Ctx.SourceMgr.extractText(Range); } static std::string sanitizeClangDocCommentStyle(StringRef Line) { static StringRef ClangStart = "/*!"; static StringRef SwiftStart = "/**"; auto Pos = Line.find(ClangStart); if (Pos == StringRef::npos) return Line.str(); StringRef Segment[2]; // The text before "/*!" Segment[0] = Line.substr(0, Pos); // The text after "/*!" Segment[1] = Line.substr(Pos).substr(ClangStart.size()); // Only sanitize when "/*!" appears at the start of this line. if (Segment[0].trim().empty()) { return (llvm::Twine(Segment[0]) + SwiftStart + Segment[1]).str(); } return Line.str(); } void printClangDocumentationComment(const clang::Decl *D) { const auto &ClangContext = D->getASTContext(); const clang::RawComment *RC = ClangContext.getRawCommentForAnyRedecl(D); if (!RC) return; bool Invalid; unsigned StartLocCol = ClangContext.getSourceManager().getSpellingColumnNumber( RC->getBeginLoc(), &Invalid); if (Invalid) StartLocCol = 0; unsigned WhitespaceToTrim = StartLocCol ? StartLocCol - 1 : 0; SmallVector Lines; StringRef RawText = RC->getRawText(ClangContext.getSourceManager()).rtrim("\n\r"); trimLeadingWhitespaceFromLines(RawText, WhitespaceToTrim, Lines); bool FirstLine = true; for (auto Line : Lines) { if (FirstLine) Printer << sanitizeClangDocCommentStyle(unicode::sanitizeUTF8(Line)); else Printer << unicode::sanitizeUTF8(Line); Printer.printNewline(); FirstLine = false; } } void printRawComment(RawComment RC) { indent(); SmallVector Lines; for (const auto &SRC : RC.Comments) { Lines.clear(); StringRef RawText = SRC.RawText.rtrim("\n\r"); unsigned WhitespaceToTrim = SRC.ColumnIndent - 1; trimLeadingWhitespaceFromLines(RawText, WhitespaceToTrim, Lines); for (auto Line : Lines) { Printer << Line; Printer.printNewline(); } } } void printSwiftDocumentationComment(const Decl *D) { if (Options.CascadeDocComment) D = getDocCommentProvidingDecl(D); if (!D) return; auto RC = D->getRawComment(); if (RC.isEmpty()) return; printRawComment(RC); } void printDocumentationComment(const Decl *D) { if (!Options.PrintDocumentationComments) return; // Try to print a comment from Clang. auto MaybeClangNode = D->getClangNode(); if (MaybeClangNode) { if (auto *CD = MaybeClangNode.getAsDecl()) printClangDocumentationComment(CD); return; } printSwiftDocumentationComment(D); } void printStaticKeyword(StaticSpellingKind StaticSpelling) { switch (StaticSpelling) { case StaticSpellingKind::None: llvm_unreachable("should not be called for non-static decls"); case StaticSpellingKind::KeywordStatic: Printer << tok::kw_static << " "; break; case StaticSpellingKind::KeywordClass: Printer << tok::kw_class << " "; break; } } void printAccess(AccessLevel access, StringRef suffix = "") { switch (access) { case AccessLevel::Private: Printer << tok::kw_private; break; case AccessLevel::FilePrivate: Printer << tok::kw_fileprivate; break; case AccessLevel::Internal: if (!Options.PrintInternalAccessKeyword) return; Printer << tok::kw_internal; break; case AccessLevel::Public: Printer << tok::kw_public; break; case AccessLevel::Open: Printer.printKeyword("open", Options); break; } Printer << suffix << " "; } void printAccess(const ValueDecl *D) { assert(!llvm::is_contained(Options.ExcludeAttrList, DAK_AccessControl) || llvm::is_contained(Options.ExcludeAttrList, DAK_SetterAccess)); if (!Options.PrintAccess || isa(D->getDeclContext())) return; if (D->getAttrs().hasAttribute() && !llvm::is_contained(Options.ExcludeAttrList, DAK_AccessControl)) return; printAccess(D->getFormalAccess()); bool shouldSkipSetterAccess = llvm::is_contained(Options.ExcludeAttrList, DAK_SetterAccess); if (auto storageDecl = dyn_cast(D)) { if (auto setter = storageDecl->getAccessor(AccessorKind::Set)) { AccessLevel setterAccess = setter->getFormalAccess(); if (setterAccess != D->getFormalAccess() && !shouldSkipSetterAccess) printAccess(setterAccess, "(set)"); } } } void printTypeWithOptions(Type T, const PrintOptions &options) { if (options.TransformContext) { // FIXME: it's not clear exactly what we want to keep from the existing // options, and what we want to discard. PrintOptions FreshOptions; FreshOptions.ExcludeAttrList = options.ExcludeAttrList; FreshOptions.ExclusiveAttrList = options.ExclusiveAttrList; FreshOptions.PrintOptionalAsImplicitlyUnwrapped = options.PrintOptionalAsImplicitlyUnwrapped; FreshOptions.TransformContext = options.TransformContext; FreshOptions.CurrentModule = options.CurrentModule; FreshOptions.FullyQualifiedTypesIfAmbiguous = options.FullyQualifiedTypesIfAmbiguous; T.print(Printer, FreshOptions); return; } T.print(Printer, options); } void printType(Type T) { printTypeWithOptions(T, Options); } void printTransformedTypeWithOptions(Type T, PrintOptions options) { if (CurrentType && Current && CurrentType->mayHaveMembers()) { auto *M = Current->getDeclContext()->getParentModule(); SubstitutionMap subMap; if (auto *NTD = dyn_cast(Current)) subMap = CurrentType->getContextSubstitutionMap(M, NTD); else if (auto *ED = dyn_cast(Current)) subMap = CurrentType->getContextSubstitutionMap(M, ED); else { Decl *subTarget = Current; if (isa(Current)) { auto *DC = Current->getDeclContext(); if (auto *FD = dyn_cast(DC)) subTarget = FD; } subMap = CurrentType->getMemberSubstitutionMap( M, cast(subTarget)); } T = T.subst(subMap, SubstFlags::DesugarMemberTypes); options.TransformContext = TypeTransformContext(CurrentType); } printTypeWithOptions(T, options); } void printTransformedType(Type T) { printTransformedTypeWithOptions(T, Options); } void printTypeLocWithOptions(const TypeLoc &TL, const PrintOptions &options) { if (CurrentType && TL.getType()) { printTransformedTypeWithOptions(TL.getType(), options); return; } // Print a TypeRepr if instructed to do so by options, or if the type // is null. if (willUseTypeReprPrinting(TL, CurrentType, options)) { if (auto repr = TL.getTypeRepr()) repr->print(Printer, options); return; } TL.getType().print(Printer, options); } void printTypeLoc(const TypeLoc &TL) { printTypeLocWithOptions(TL, Options); } void printTypeLocForImplicitlyUnwrappedOptional(TypeLoc TL, bool IUO) { PrintOptions options = Options; options.PrintOptionalAsImplicitlyUnwrapped = IUO; printTypeLocWithOptions(TL, options); } void printContextIfNeeded(const Decl *decl) { if (IndentLevel > 0) return; switch (Options.ShouldQualifyNestedDeclarations) { case PrintOptions::QualifyNestedDeclarations::Never: return; case PrintOptions::QualifyNestedDeclarations::TypesOnly: if (!isa(decl)) return; break; case PrintOptions::QualifyNestedDeclarations::Always: break; } auto *container = dyn_cast(decl->getDeclContext()); if (!container) return; printType(container->getDeclaredInterfaceType()); Printer << "."; } void printAttributes(const Decl *D); void printTypedPattern(const TypedPattern *TP); void printBraceStmt(const BraceStmt *stmt, bool newlineIfEmpty = true); void printAccessorDecl(const AccessorDecl *decl); public: void printPattern(const Pattern *pattern); enum GenericSignatureFlags { PrintParams = 1, PrintRequirements = 2, InnermostOnly = 4, SwapSelfAndDependentMemberType = 8, PrintInherited = 16, }; void printInheritedFromRequirementSignature(ProtocolDecl *proto, Decl *attachingTo); void printWhereClauseFromRequirementSignature(ProtocolDecl *proto, Decl *attachingTo); void printInherited(const Decl *decl); void printGenericSignature(GenericSignature genericSig, unsigned flags); void printGenericSignature(GenericSignature genericSig, unsigned flags, llvm::function_ref filter); void printSingleDepthOfGenericSignature( TypeArrayView genericParams, ArrayRef requirements, unsigned flags, llvm::function_ref filter); void printSingleDepthOfGenericSignature( TypeArrayView genericParams, ArrayRef requirements, bool &isFirstReq, unsigned flags, llvm::function_ref filter); void printRequirement(const Requirement &req); private: bool shouldPrint(const Decl *D, bool Notify = false); bool shouldPrintPattern(const Pattern *P); void printPatternType(const Pattern *P); void printAccessors(const AbstractStorageDecl *ASD); void printSelfAccessKindModifiersIfNeeded(const FuncDecl *FD); void printMembersOfDecl(Decl * NTD, bool needComma = false, bool openBracket = true, bool closeBracket = true); void printMembers(ArrayRef members, bool needComma = false, bool openBracket = true, bool closeBracket = true); void printGenericDeclGenericParams(GenericContext *decl); void printDeclGenericRequirements(GenericContext *decl); void printPrimaryAssociatedTypes(ProtocolDecl *decl); void printBodyIfNecessary(const AbstractFunctionDecl *decl); void printEnumElement(EnumElementDecl *elt); /// \returns true if anything was printed. bool printASTNodes(const ArrayRef &Elements, bool NeedIndent = true); void printOneParameter(const ParamDecl *param, ParameterTypeFlags paramFlags, bool ArgNameIsAPIByDefault); void printParameterList(ParameterList *PL, ArrayRef params, bool isAPINameByDefault); /// Print the function parameters in curried or selector style, /// to match the original function declaration. void printFunctionParameters(AbstractFunctionDecl *AFD); void printArgument(const Argument &arg); void printStmtCondition(StmtCondition stmt); #define DECL(Name,Parent) void visit##Name##Decl(Name##Decl *decl); #define ABSTRACT_DECL(Name, Parent) #define DECL_RANGE(Name,Start,End) #include "swift/AST/DeclNodes.def" #define STMT(Name, Parent) void visit##Name##Stmt(Name##Stmt *stmt); #include "swift/AST/StmtNodes.def" #define EXPR(Name,Parent) void visit##Name##Expr(Name##Expr *expr); #define ABSTRACT_EXPR(Name, Parent) #define DECL_RANGE(Name,Start,End) #include "swift/AST/ExprNodes.def" void printSynthesizedExtension(Type ExtendedType, ExtensionDecl *ExtDecl); void printSynthesizedExtensionImpl(Type ExtendedType, ExtensionDecl *ExtDecl); void printExtension(ExtensionDecl* ExtDecl); void printExtendedTypeName(TypeLoc ExtendedTypeLoc); public: PrintAST(ASTPrinter &Printer, const PrintOptions &Options) : Printer(Printer), Options(Options) { if (Options.TransformContext) { Type CurrentType = Options.TransformContext->getBaseType(); if (CurrentType && CurrentType->hasArchetype()) { // OpenedArchetypeTypes get replaced by a GenericTypeParamType without a // name in mapTypeOutOfContext. The GenericTypeParamType has no children // so we can't use it for TypeTransformContext. // To work around this, replace the OpenedArchetypeType with the type of // the protocol itself. if (auto *Opened = CurrentType->getAs()) { assert(Opened->isRoot()); CurrentType = Opened->getExistentialType(); } CurrentType = CurrentType->mapTypeOutOfContext(); } setCurrentType(CurrentType); } } using ASTVisitor::visit; bool visit(Expr *E) { if (!Options.PrintExprs) { return false; } ASTVisitor::visit(E); return true; } bool visit(Decl *D) { #if SWIFT_BUILD_ONLY_SYNTAXPARSERLIB return false; // not needed for the parser library. #endif bool Synthesize = Options.TransformContext && Options.TransformContext->isPrintingSynthesizedExtension() && isa(D); if (!shouldPrint(D, true) && !Synthesize) return false; Decl *Old = Current; Current = D; SWIFT_DEFER { Current = Old; }; Type OldType = CurrentType; if (CurrentType && (Old != nullptr || Options.PrintAsMember)) { if (auto *NTD = dyn_cast(D)) { auto Subs = CurrentType->getContextSubstitutionMap( Options.CurrentModule, NTD->getDeclContext()); setCurrentType(NTD->getDeclaredInterfaceType().subst(Subs)); } } SWIFT_DEFER { setCurrentType(OldType); }; if (Synthesize) { Printer.setSynthesizedTarget(Options.TransformContext->getDecl()); } // We want to print a newline before doc comments. Swift code already // handles this, but we need to insert it for clang doc comments when not // printing other clang comments. Do it now so the printDeclPre callback // happens after the newline. if (Options.PrintDocumentationComments && !Options.PrintRegularClangComments && D->hasClangNode()) { auto clangNode = D->getClangNode(); auto clangDecl = clangNode.getAsDecl(); if (clangDecl && clangDecl->getASTContext().getRawCommentForAnyRedecl(clangDecl)) { Printer.printNewline(); indent(); } } Printer.callPrintDeclPre(D, Options.BracketOptions); if (Options.PrintCompatibilityFeatureChecks) { printWithCompatibilityFeatureChecks(Printer, Options, D, [&]{ ASTVisitor::visit(D); }); } else { ASTVisitor::visit(D); } if (Synthesize) { Printer.setSynthesizedTarget({}); Printer.printSynthesizedExtensionPost(cast(D), Options.TransformContext->getDecl(), Options.BracketOptions); } else { Printer.callPrintDeclPost(D, Options.BracketOptions); } return true; } }; } // unnamed namespace static StaticSpellingKind getCorrectStaticSpelling(const Decl *D) { if (auto *ASD = dyn_cast(D)) { return ASD->getCorrectStaticSpelling(); } else if (auto *PBD = dyn_cast(D)) { return PBD->getCorrectStaticSpelling(); } else if (auto *FD = dyn_cast(D)) { return FD->getCorrectStaticSpelling(); } else { return StaticSpellingKind::None; } } static bool hasAsyncGetter(const AbstractStorageDecl *ASD) { if (auto getter = ASD->getAccessor(AccessorKind::Get)) { assert(!getter->getAttrs().hasAttribute()); return getter->hasAsync(); } return false; } static bool hasThrowsGetter(const AbstractStorageDecl *ASD) { if (auto getter = ASD->getAccessor(AccessorKind::Get)) { assert(!getter->getAttrs().hasAttribute()); return getter->hasThrows(); } return false; } static bool hasMutatingGetter(const AbstractStorageDecl *ASD) { return ASD->getAccessor(AccessorKind::Get) && ASD->isGetterMutating(); } static bool hasNonMutatingSetter(const AbstractStorageDecl *ASD) { if (!ASD->isSettable(nullptr)) return false; auto setter = ASD->getAccessor(AccessorKind::Set); return setter && setter->isExplicitNonMutating(); } static bool hasLessAccessibleSetter(const AbstractStorageDecl *ASD) { return ASD->getSetterFormalAccess() < ASD->getFormalAccess(); } void PrintAST::printAttributes(const Decl *D) { if (Options.SkipAttributes) return; // Save the current number of exclude attrs to restore once we're done. unsigned originalExcludeAttrCount = Options.ExcludeAttrList.size(); if (Options.PrintImplicitAttrs) { // Don't print a redundant 'final' if we are printing a 'static' decl. if (D->getDeclContext()->getSelfClassDecl() && getCorrectStaticSpelling(D) == StaticSpellingKind::KeywordStatic) { Options.ExcludeAttrList.push_back(DAK_Final); } if (auto vd = dyn_cast(D)) { // Don't print @_hasInitialValue if we're printing an initializer // expression or if the storage is resilient. if (vd->isInitExposedToClients() || vd->isResilient()) Options.ExcludeAttrList.push_back(DAK_HasInitialValue); if (!Options.PrintForSIL) { // Don't print @_hasStorage if the value is simply stored, or the // decl is resilient. if (vd->isResilient() || (vd->getImplInfo().isSimpleStored() && !hasLessAccessibleSetter(vd))) Options.ExcludeAttrList.push_back(DAK_HasStorage); } } // SPI groups if (Options.PrintSPIs && DeclAttribute::canAttributeAppearOnDeclKind( DAK_SPIAccessControl, D->getKind())) { interleave(D->getSPIGroups(), [&](Identifier spiName) { Printer.printAttrName("_spi", true); Printer << "(" << spiName << ") "; }, [&] { Printer << ""; }); Options.ExcludeAttrList.push_back(DAK_SPIAccessControl); } // Don't print any contextual decl modifiers. // We will handle 'mutating' and 'nonmutating' separately. if (isa(D)) { #define EXCLUDE_ATTR(Class) Options.ExcludeAttrList.push_back(DAK_##Class); #define CONTEXTUAL_DECL_ATTR(X, Class, Y, Z) EXCLUDE_ATTR(Class) #define CONTEXTUAL_SIMPLE_DECL_ATTR(X, Class, Y, Z) EXCLUDE_ATTR(Class) #define CONTEXTUAL_DECL_ATTR_ALIAS(X, Class) EXCLUDE_ATTR(Class) #include "swift/AST/Attr.def" } // If the declaration is implicitly @objc, print the attribute now. if (auto VD = dyn_cast(D)) { if (VD->isObjC() && !isa(VD) && !VD->getAttrs().hasAttribute()) { Printer.printAttrName("@objc"); Printer << " "; } } // If the declaration has designated inits that won't be visible to // clients, or if it inherits superclass convenience initializers, // then print those attributes specially. if (auto CD = dyn_cast(D)) { if (CD->inheritsSuperclassInitializers()) { Printer.printAttrName("@_inheritsConvenienceInitializers"); Printer << " "; } if (CD->hasMissingDesignatedInitializers()) { Printer.printAttrName("@_hasMissingDesignatedInitializers"); Printer << " "; } } } // We will handle 'mutating' and 'nonmutating' separately. if (isa(D)) { Options.ExcludeAttrList.push_back(DAK_Mutating); Options.ExcludeAttrList.push_back(DAK_NonMutating); Options.ExcludeAttrList.push_back(DAK_Consuming); } D->getAttrs().print(Printer, Options, D); // Print the implicit 'final' attribute. if (auto VD = dyn_cast(D)) { auto VarD = dyn_cast(D); if (VD->isFinal() && !VD->getAttrs().hasAttribute() && // Don't print a redundant 'final' if printing a 'let' or 'static' decl. !(VarD && VarD->isLet()) && getCorrectStaticSpelling(D) != StaticSpellingKind::KeywordStatic && VD->getKind() != DeclKind::Accessor) { Printer.printAttrName("final"); Printer << " "; } } Options.ExcludeAttrList.resize(originalExcludeAttrCount); } void PrintAST::printTypedPattern(const TypedPattern *TP) { printPattern(TP->getSubPattern()); Printer << ": "; // Make sure to check if the underlying var decl is an implicitly unwrapped // optional. bool isIUO = false; if (auto *named = dyn_cast(TP->getSubPattern())) if (auto decl = named->getDecl()) isIUO = decl->isImplicitlyUnwrappedOptional(); const auto TyLoc = TypeLoc(TP->getTypeRepr(), TP->hasType() ? TP->getType() : Type()); printTypeLocForImplicitlyUnwrappedOptional(TyLoc, isIUO); } /// Determines if we are required to print the name of a property declaration, /// or if we can elide it by printing a '_' instead. static bool mustPrintPropertyName(VarDecl *decl, const PrintOptions &opts) { // If we're not allowed to omit the name, we must print it. if (!opts.OmitNameOfInaccessibleProperties) return true; // If it contributes to the parent's storage, we must print it because clients // need to be able to directly access the storage. // FIXME: We might be able to avoid printing names for some of these // if we serialized references to them using field indices. if (contributesToParentTypeStorage(decl)) return true; // If it's public or @usableFromInline, we must print the name because it's a // visible entry-point. if (isPublicOrUsableFromInline(decl)) return true; // If it has an initial value, we must print the name because it's used in // the mangled name of the initializer expression generator function. // FIXME: We _could_ figure out a way to generate an entry point // for the initializer expression without revealing the name. We just // don't have a mangling for it. if (decl->hasInitialValue()) return true; // If none of those are true, we can elide the name of the variable. return false; } /// Gets the print name context of a given decl, choosing between TypeMember /// and Normal, depending if this decl lives in a nominal type decl. static PrintNameContext getTypeMemberPrintNameContext(const Decl *d) { return d->getDeclContext()->isTypeContext() ? PrintNameContext::TypeMember : PrintNameContext::Normal; } void PrintAST::printPattern(const Pattern *pattern) { switch (pattern->getKind()) { case PatternKind::Any: Printer << "_"; break; case PatternKind::Named: { auto named = cast(pattern); auto decl = named->getDecl(); recordDeclLoc(decl, [&]{ // FIXME: This always returns true now, because of the FIXMEs listed in // mustPrintPropertyName. if (mustPrintPropertyName(decl, Options)) Printer.printName(named->getBoundName(), getTypeMemberPrintNameContext(decl)); else Printer << "_"; }); break; } case PatternKind::Paren: Printer << "("; printPattern(cast(pattern)->getSubPattern()); Printer << ")"; break; case PatternKind::Tuple: { Printer << "("; auto TP = cast(pattern); auto Fields = TP->getElements(); for (unsigned i = 0, e = Fields.size(); i != e; ++i) { const auto &Elt = Fields[i]; if (i != 0) Printer << ", "; printPattern(Elt.getPattern()); } Printer << ")"; break; } case PatternKind::Typed: printTypedPattern(cast(pattern)); break; case PatternKind::Is: { auto isa = cast(pattern); Printer << tok::kw_is << " "; isa->getCastType().print(Printer, Options); break; } case PatternKind::EnumElement: { auto elt = cast(pattern); Printer << "." << elt->getElementDecl()->getBaseName(); if (elt->hasSubPattern()) printPattern(elt->getSubPattern()); break; } case PatternKind::OptionalSome: printPattern(cast(pattern)->getSubPattern()); Printer << '?'; break; case PatternKind::Bool: Printer << (cast(pattern)->getValue() ? tok::kw_true : tok::kw_false); break; case PatternKind::Expr: { auto expr = cast(pattern)->getSubExpr(); visit(expr); break; } case PatternKind::Binding: { auto bPattern = cast(pattern); Printer.printIntroducerKeyword( bPattern->isLet() ? "let" : "var", Options, " "); printPattern(bPattern->getSubPattern()); } } } /// If we can't find the depth of a type, return ErrorDepth. static const unsigned ErrorDepth = ~0U; /// A helper function to return the depth of a type. static unsigned getDepthOfType(Type ty) { unsigned depth = ErrorDepth; auto combineDepth = [&depth](unsigned newDepth) -> bool { // If there is no current depth (depth == ErrorDepth), then assign to // newDepth; otherwise, choose the deeper of the current and new depth. // Since ErrorDepth == ~0U, ErrorDepth + 1 == 0, which is smaller than any // valid depth + 1. depth = std::max(depth+1U, newDepth+1U) - 1U; return false; }; ty.findIf([combineDepth](Type t) -> bool { if (auto paramTy = t->getAs()) return combineDepth(paramTy->getDepth()); if (auto depMemTy = dyn_cast(t->getCanonicalType())) { CanType rootTy; do { rootTy = depMemTy.getBase(); } while ((depMemTy = dyn_cast(rootTy))); if (auto rootParamTy = dyn_cast(rootTy)) return combineDepth(rootParamTy->getDepth()); } return false; }); return depth; } namespace { struct RequirementPrintLocation { /// The Decl where the requirement should be attached (whether inherited or in /// a where clause) Decl *AttachedTo; /// Whether the requirement needs to be in a where clause. bool InWhereClause; }; } // end anonymous namespace /// Heuristically work out a good place for \c req to be printed inside \c /// proto. /// /// This depends only on the protocol so that we make the same decisions for all /// requirements in all associated types, guaranteeing that all of them will be /// printed somewhere. That is, taking an AssociatedTypeDecl as an argument and /// asking "should this requirement be printed on this ATD?" seems more likely /// to result in inconsistencies in what is printed where, versus what this /// function does: asking "where should this requirement be printed?" and then /// callers check if the location is the ATD. static RequirementPrintLocation bestRequirementPrintLocation(ProtocolDecl *proto, const Requirement &req) { auto protoSelf = proto->getProtocolSelfType(); // Returns the most relevant decl within proto connected to outerType (or null // if one doesn't exist), and whether the type is an "direct use", // i.e. outerType itself is Self or Self.T, but not, say, Self.T.U, or // Array. (The first's decl will be proto, while the other three will // be Self.T.) auto findRelevantDeclAndDirectUse = [&](Type outerType) { TypeDecl *relevantDecl = nullptr; Type foundType; (void)outerType.findIf([&](Type t) { if (t->isEqual(protoSelf)) { relevantDecl = proto; foundType = t; return true; } else if (auto DMT = t->getAs()) { auto assocType = DMT->getAssocType(); if (assocType && assocType->getProtocol() == proto) { relevantDecl = assocType; foundType = t; return true; } } // not here, so let's keep looking. return false; }); // If we didn't find anything, relevantDecl and foundType will be null, as // desired. auto directUse = foundType && outerType->isEqual(foundType); return std::make_pair(relevantDecl, directUse); }; Decl *bestDecl; bool inWhereClause; switch (req.getKind()) { case RequirementKind::Layout: case RequirementKind::Conformance: case RequirementKind::Superclass: { auto subject = req.getFirstType(); auto result = findRelevantDeclAndDirectUse(subject); bestDecl = result.first; inWhereClause = !bestDecl || !result.second; break; } case RequirementKind::SameType: { auto lhs = req.getFirstType(); auto rhs = req.getSecondType(); auto lhsResult = findRelevantDeclAndDirectUse(lhs); auto rhsResult = findRelevantDeclAndDirectUse(rhs); // Default to using the left type's decl. bestDecl = lhsResult.first; // But maybe the right type's one is "obviously" better! // e.g. Int == Self.T auto lhsDoesntExist = !lhsResult.first; // e.g. Self.T.U == Self.V should go on V (first two conditions), but // Self.T.U == Self should go on T (third condition). auto rhsBetterDirect = !lhsResult.second && rhsResult.second && rhsResult.first != proto; auto rhsOfSelfToAssoc = lhsResult.first == proto && rhsResult.first; // e.g. Self == Self.T.U if (lhsDoesntExist || rhsBetterDirect || rhsOfSelfToAssoc) bestDecl = rhsResult.first; // Same-type requirements can only occur in where clauses inWhereClause = true; break; } } // Didn't find anything that we think is relevant, so let's default to a where // clause on the protocol. if (!bestDecl) { bestDecl = proto; inWhereClause = true; } return {/*AttachedTo=*/bestDecl, inWhereClause}; } void PrintAST::printInheritedFromRequirementSignature(ProtocolDecl *proto, Decl *attachingTo) { printGenericSignature( GenericSignature::get({proto->getProtocolSelfType()} , proto->getRequirementSignature().getRequirements()), PrintInherited, [&](const Requirement &req) { // Skip the inferred 'Self : AnyObject' constraint if this is an // @objc protocol. if (req.getKind() == RequirementKind::Layout && req.getFirstType()->isEqual(proto->getProtocolSelfType()) && req.getLayoutConstraint()->getKind() == LayoutConstraintKind::Class && proto->isObjC()) { return false; } auto location = bestRequirementPrintLocation(proto, req); return location.AttachedTo == attachingTo && !location.InWhereClause; }); } void PrintAST::printWhereClauseFromRequirementSignature(ProtocolDecl *proto, Decl *attachingTo) { unsigned flags = PrintRequirements; if (isa(attachingTo)) flags |= SwapSelfAndDependentMemberType; printGenericSignature( GenericSignature::get({proto->getProtocolSelfType()} , proto->getRequirementSignature().getRequirements()), flags, [&](const Requirement &req) { auto location = bestRequirementPrintLocation(proto, req); return location.AttachedTo == attachingTo && location.InWhereClause; }); } /// A helper function to return the depth of a requirement. static unsigned getDepthOfRequirement(const Requirement &req) { switch (req.getKind()) { case RequirementKind::Conformance: case RequirementKind::Layout: return getDepthOfType(req.getFirstType()); case RequirementKind::Superclass: case RequirementKind::SameType: { // Return the max valid depth of firstType and secondType. unsigned firstDepth = getDepthOfType(req.getFirstType()); unsigned secondDepth = getDepthOfType(req.getSecondType()); unsigned maxDepth; if (firstDepth == ErrorDepth && secondDepth != ErrorDepth) maxDepth = secondDepth; else if (firstDepth != ErrorDepth && secondDepth == ErrorDepth) maxDepth = firstDepth; else maxDepth = std::max(firstDepth, secondDepth); return maxDepth; } } llvm_unreachable("bad RequirementKind"); } static void getRequirementsAtDepth(GenericSignature genericSig, unsigned depth, SmallVectorImpl &result) { for (auto reqt : genericSig.getRequirements()) { unsigned currentDepth = getDepthOfRequirement(reqt); assert(currentDepth != ErrorDepth); if (currentDepth == depth) result.push_back(reqt); } } void PrintAST::printGenericSignature(GenericSignature genericSig, unsigned flags) { printGenericSignature(genericSig, flags, // print everything [&](const Requirement &) { return true; }); } void PrintAST::printGenericSignature( GenericSignature genericSig, unsigned flags, llvm::function_ref filter) { auto requirements = genericSig.getRequirements(); if (flags & InnermostOnly) { auto genericParams = genericSig.getInnermostGenericParams(); printSingleDepthOfGenericSignature(genericParams, requirements, flags, filter); return; } auto genericParams = genericSig.getGenericParams(); if (!Options.PrintInSILBody) { printSingleDepthOfGenericSignature(genericParams, requirements, flags, filter); return; } // In order to recover the nested GenericParamLists, we divide genericParams // and requirements according to depth. unsigned paramIdx = 0, numParam = genericParams.size(); while (paramIdx < numParam) { unsigned depth = genericParams[paramIdx]->getDepth(); // Move index to genericParams. unsigned lastParamIdx = paramIdx; do { ++lastParamIdx; } while (lastParamIdx < numParam && genericParams[lastParamIdx]->getDepth() == depth); // Collect requirements for this level. SmallVector requirementsAtDepth; getRequirementsAtDepth(genericSig, depth, requirementsAtDepth); printSingleDepthOfGenericSignature( genericParams.slice(paramIdx, lastParamIdx - paramIdx), requirementsAtDepth, flags, filter); paramIdx = lastParamIdx; } } void PrintAST::printSingleDepthOfGenericSignature( TypeArrayView genericParams, ArrayRef requirements, unsigned flags, llvm::function_ref filter) { bool isFirstReq = true; printSingleDepthOfGenericSignature(genericParams, requirements, isFirstReq, flags, filter); } void PrintAST::printSingleDepthOfGenericSignature( TypeArrayView genericParams, ArrayRef requirements, bool &isFirstReq, unsigned flags, llvm::function_ref filter) { bool printParams = (flags & PrintParams); bool printRequirements = (flags & PrintRequirements); printRequirements &= Options.PrintGenericRequirements; bool printInherited = (flags & PrintInherited); bool swapSelfAndDependentMemberType = (flags & SwapSelfAndDependentMemberType); unsigned typeContextDepth = 0; SubstitutionMap subMap; ModuleDecl *M = nullptr; if (CurrentType && Current) { if (!CurrentType->isExistentialType()) { auto *DC = Current->getInnermostDeclContext()->getInnermostTypeContext(); M = DC->getParentModule(); subMap = CurrentType->getContextSubstitutionMap(M, DC); if (!subMap.empty()) { typeContextDepth = subMap.getGenericSignature() .getGenericParams().back()->getDepth() + 1; } } } auto substParam = [&](Type param) -> Type { if (subMap.empty()) return param; return param.subst( [&](SubstitutableType *type) -> Type { if (cast(type)->getDepth() < typeContextDepth) return Type(type).subst(subMap); return type; }, [&](CanType depType, Type substType, ProtocolDecl *proto) { return M->lookupConformance(substType, proto); }); }; /// Separate the explicit generic parameters from the implicit, opaque /// generic parameters. We only print the former. TypeArrayView opaqueGenericParams; for (unsigned index : indices(genericParams)) { auto gpDecl = genericParams[index]->getDecl(); if (!gpDecl) continue; if (gpDecl->isOpaqueType() && gpDecl->isImplicit()) { // We found the first implicit opaque type parameter. Split the // generic parameters array at this position. opaqueGenericParams = genericParams.slice(index); genericParams = genericParams.slice(0, index); break; } } // Determines whether a given type is based on one of the opaque generic // parameters. auto dependsOnOpaque = [&](Type type) { if (opaqueGenericParams.empty()) return false; if (!type->isTypeParameter()) return false; auto rootGP = type->getRootGenericParam(); for (auto opaqueGP : opaqueGenericParams) { if (rootGP->isEqual(opaqueGP)) return true; } return false; }; if (printParams && !genericParams.empty()) { // Print the generic parameters. Printer << "<"; llvm::interleave( genericParams, [&](GenericTypeParamType *param) { if (!subMap.empty()) { printType(substParam(param)); } else if (auto *GP = param->getDecl()) { Printer.callPrintStructurePre(PrintStructureKind::GenericParameter, GP); Printer.printName(GP->getName(), PrintNameContext::GenericParameter); Printer.printStructurePost(PrintStructureKind::GenericParameter, GP); } else { printType(param); } }, [&] { Printer << ", "; }); } if (printRequirements || printInherited) { for (const auto &req : requirements) { if (!filter(req)) continue; auto first = req.getFirstType(); if (dependsOnOpaque(first)) continue; Type second; if (req.getKind() != RequirementKind::Layout) { second = req.getSecondType(); if (dependsOnOpaque(second)) continue; } if (!subMap.empty()) { Type subFirst = substParam(first); if (!subFirst->hasError()) first = subFirst; if (second) { Type subSecond = substParam(second); if (!subSecond->hasError()) second = subSecond; if (!(first->is() || first->isTypeParameter()) && !(second->is() || second->isTypeParameter())) continue; } } if (isFirstReq) { if (printRequirements) Printer << " " << tok::kw_where << " "; else Printer << " : "; isFirstReq = false; } else { Printer << ", "; } // Swap the order of Self == Self.A requirements if requested. if (swapSelfAndDependentMemberType && req.getKind() == RequirementKind::SameType && first->is() && second->is()) std::swap(first, second); if (printInherited) { // We only print the second part of a requirement in the "inherited" // clause. switch (req.getKind()) { case RequirementKind::Layout: req.getLayoutConstraint()->print(Printer, Options); break; case RequirementKind::Conformance: case RequirementKind::Superclass: printType(second); break; case RequirementKind::SameType: llvm_unreachable("same-type constraints belong in the where clause"); break; } } else { Printer.callPrintStructurePre(PrintStructureKind::GenericRequirement); printRequirement(req); Printer.printStructurePost(PrintStructureKind::GenericRequirement); } } } if (printParams && !genericParams.empty()) Printer << ">"; } void PrintAST::printRequirement(const Requirement &req) { printTransformedType(req.getFirstType()); switch (req.getKind()) { case RequirementKind::Layout: Printer << " : "; req.getLayoutConstraint()->print(Printer, Options); return; case RequirementKind::Conformance: case RequirementKind::Superclass: Printer << " : "; break; case RequirementKind::SameType: Printer << " == "; break; } printTransformedType(req.getSecondType()); } bool PrintAST::shouldPrintPattern(const Pattern *P) { return Options.shouldPrint(P); } void PrintAST::printPatternType(const Pattern *P) { if (P->hasType()) { Printer << ": "; printType(P->getType()); } } bool ShouldPrintChecker::shouldPrint(const Pattern *P, const PrintOptions &Options) { bool ShouldPrint = false; P->forEachVariable([&](const VarDecl *VD) { ShouldPrint |= shouldPrint(VD, Options); }); return ShouldPrint; } bool isNonSendableExtension(const Decl *D) { ASTContext &ctx = D->getASTContext(); const ExtensionDecl *ED = dyn_cast(D); if (!ED || !ED->getAttrs().isUnavailable(ctx)) return false; auto nonSendable = ED->getExtendedNominal()->getAttrs().getEffectiveSendableAttr(); if (!isa_and_nonnull(nonSendable)) return false; // GetImplicitSendableRequest::evaluate() creates its extension with the // attribute's AtLoc, so this is a good way to quickly check if the extension // was synthesized for an '@_nonSendable' attribute. return ED->getLocFromSource() == nonSendable->AtLoc; } bool ShouldPrintChecker::shouldPrint(const Decl *D, const PrintOptions &Options) { #if SWIFT_BUILD_ONLY_SYNTAXPARSERLIB return false; // not needed for the parser library. #endif if (auto *ED= dyn_cast(D)) { if (Options.printExtensionContentAsMembers(ED)) return false; } if (Options.SkipMissingMemberPlaceholders && isa(D)) return false; if (Options.SkipDeinit && isa(D)) { return false; } if (Options.SkipImports && isa(D)) { return false; } // Optionally skip these checks for extensions synthesized for '@_nonSendable' if (!Options.AlwaysPrintNonSendableExtensions || !isNonSendableExtension(D)) { if (Options.SkipImplicit && D->isImplicit()) { const auto &IgnoreList = Options.TreatAsExplicitDeclList; if (!llvm::is_contained(IgnoreList, D)) return false; } if (Options.SkipUnavailable && D->getAttrs().isUnavailable(D->getASTContext())) return false; } if (Options.ExplodeEnumCaseDecls) { if (isa(D)) return true; if (isa(D)) return false; } else if (auto *EED = dyn_cast(D)) { // Enum elements are printed as part of the EnumCaseDecl, unless they were // imported without source info. return !EED->getSourceRange().isValid(); } if (auto *ASD = dyn_cast(D)) { if (Options.OmitNameOfInaccessibleProperties && contributesToParentTypeStorage(ASD)) return true; } // Skip declarations that are not accessible. if (auto *VD = dyn_cast(D)) { if (Options.AccessFilter > AccessLevel::Private && VD->getFormalAccess() < Options.AccessFilter) return false; } // Skip clang decls marked with the swift_private attribute. if (Options.SkipSwiftPrivateClangDecls) { if (auto ClangD = D->getClangDecl()) { if (ClangD->hasAttr()) return false; } } if (Options.SkipPrivateStdlibDecls && D->isPrivateStdlibDecl(!Options.SkipUnderscoredStdlibProtocols)) return false; if (Options.SkipEmptyExtensionDecls && isa(D)) { auto Ext = cast(D); // If the extension doesn't add protocols or has no members that we should // print then skip printing it. SmallVector ProtocolsToPrint; getInheritedForPrinting(Ext, Options, ProtocolsToPrint); if (ProtocolsToPrint.empty()) { bool HasMemberToPrint = false; for (auto Member : Ext->getAllMembers()) { if (shouldPrint(Member, Options)) { HasMemberToPrint = true; break; } } if (!HasMemberToPrint) return false; } } // If asked to skip overrides and witnesses, do so. if (Options.SkipOverrides) { if (auto *VD = dyn_cast(D)) { if (VD->getOverriddenDecl()) return false; if (!VD->getSatisfiedProtocolRequirements().empty()) return false; if (auto clangDecl = VD->getClangDecl()) { // If the Clang declaration is from a protocol but was mirrored into // class or extension thereof, treat it as an override. if (isa(clangDecl->getDeclContext()) && VD->getDeclContext()->getSelfClassDecl()) return false; // Check whether Clang considers it an override. if (auto objcMethod = dyn_cast(clangDecl)) { SmallVector overriddenMethods; objcMethod->getOverriddenMethods(overriddenMethods); if (!overriddenMethods.empty()) return false; } else if (auto objcProperty = dyn_cast(clangDecl)) { if (auto getter = objcProperty->getGetterMethodDecl()) { SmallVector overriddenMethods; getter->getOverriddenMethods(overriddenMethods); if (!overriddenMethods.empty()) return false; } } } } } // We need to handle PatternBindingDecl as a special case here because its // attributes can only be retrieved from the inside VarDecls. if (auto *PD = dyn_cast(D)) { auto ShouldPrint = false; for (auto idx : range(PD->getNumPatternEntries())) { ShouldPrint |= shouldPrint(PD->getPattern(idx), Options); if (ShouldPrint) return true; } return false; } if (isa(D)) { return Options.PrintIfConfig; } return true; } bool PrintAST::shouldPrint(const Decl *D, bool Notify) { auto Result = Options.shouldPrint(D); if (!Result && Notify) Printer.callAvoidPrintDeclPost(D); return Result; } void PrintAST::printBraceStmt(const BraceStmt *stmt, bool newlineIfEmpty) { Printer << "{"; if (printASTNodes(stmt->getElements()) || newlineIfEmpty) { Printer.printNewline(); indent(); } Printer << "}"; } void PrintAST::printBodyIfNecessary(const AbstractFunctionDecl *decl) { if (auto BodyFunc = Options.FunctionBody) { BodyFunc(decl, Printer); indent(); return; } if (!Options.FunctionDefinitions || !decl->getBody()) return; Printer << " "; printBraceStmt(decl->getBody(), /*newlineIfEmpty*/!isa(decl)); } void PrintAST::printSelfAccessKindModifiersIfNeeded(const FuncDecl *FD) { if (!Options.PrintSelfAccessKindKeyword) return; const auto *AD = dyn_cast(FD); switch (FD->getSelfAccessKind()) { case SelfAccessKind::Mutating: if ((!AD || AD->isAssumedNonMutating()) && !Options.excludeAttrKind(DAK_Mutating)) Printer.printKeyword("mutating", Options, " "); break; case SelfAccessKind::NonMutating: if (AD && AD->isExplicitNonMutating() && !Options.excludeAttrKind(DAK_NonMutating)) Printer.printKeyword("nonmutating", Options, " "); break; case SelfAccessKind::Consuming: if (!Options.excludeAttrKind(DAK_Consuming)) Printer.printKeyword("__consuming", Options, " "); break; } } void PrintAST::printAccessors(const AbstractStorageDecl *ASD) { if (isa(ASD) && !Options.PrintPropertyAccessors) return; if (isa(ASD) && !Options.PrintSubscriptAccessors) return; auto impl = ASD->getImplInfo(); // AbstractAccessors is suppressed by FunctionDefinitions. bool PrintAbstract = Options.AbstractAccessors && !Options.FunctionDefinitions; // Don't print accessors for trivially stored properties... if (impl.isSimpleStored()) { // ...unless we're printing for SIL, which expects a { get set? } on // trivial properties if (Options.PrintForSIL) { Printer << " { get " << (impl.supportsMutation() ? "set }" : "}"); } // ...or you're private/internal(set), at which point we'll print // @_hasStorage var x: T { get } else if (ASD->isSettable(nullptr) && hasLessAccessibleSetter(ASD)) { if (PrintAbstract) { Printer << " { get }"; } else { Printer << " {"; { IndentRAII indentMore(*this); indent(); Printer.printNewline(); Printer << "get"; } indent(); Printer.printNewline(); Printer << "}"; } } return; } // prints with a space prefixed auto printWithSpace = [&](StringRef word) { Printer << " "; Printer.printKeyword(word, Options); }; const bool asyncGet = hasAsyncGetter(ASD); const bool throwsGet = hasThrowsGetter(ASD); // We sometimes want to print the accessors abstractly // instead of listing out how they're actually implemented. bool inProtocol = isa(ASD->getDeclContext()); if ((inProtocol && !Options.PrintAccessorBodiesInProtocols) || PrintAbstract) { bool settable = ASD->isSettable(nullptr); bool mutatingGetter = hasMutatingGetter(ASD); bool nonmutatingSetter = hasNonMutatingSetter(ASD); // We're about to print something like this: // { mutating? get async? throws? (nonmutating? set)? } // But don't print "{ get set }" if we don't have to. if (!inProtocol && !Options.PrintGetSetOnRWProperties && settable && !mutatingGetter && !nonmutatingSetter && !asyncGet && !throwsGet) { return; } Printer << " {"; if (mutatingGetter) printWithSpace("mutating"); printWithSpace("get"); if (asyncGet) printWithSpace("async"); if (throwsGet) printWithSpace("throws"); if (settable) { if (nonmutatingSetter) printWithSpace("nonmutating"); printWithSpace("set"); } Printer << " }"; return; } // Should we print the 'modify' accessor? auto shouldHideModifyAccessor = [&] { if (impl.getReadWriteImpl() != ReadWriteImplKind::Modify) return true; // Always hide in a protocol. return isa(ASD->getDeclContext()); }; auto isGetSetImpl = [&] { return ((impl.getReadImpl() == ReadImplKind::Stored || impl.getReadImpl() == ReadImplKind::Get) && (impl.getWriteImpl() == WriteImplKind::Stored || impl.getWriteImpl() == WriteImplKind::Set) && (shouldHideModifyAccessor())); }; // Honor !Options.PrintGetSetOnRWProperties in the only remaining // case where we could end up printing { get set }. if ((PrintAbstract || isGetSetImpl()) && !Options.PrintGetSetOnRWProperties && !Options.FunctionDefinitions && !ASD->isGetterMutating() && !ASD->getAccessor(AccessorKind::Set)->isExplicitNonMutating() && !asyncGet && !throwsGet) { return; } // Otherwise, print all the concrete defining accessors. bool PrintAccessorBody = Options.FunctionDefinitions; // Helper to print an accessor. Returns true if the // accessor was present but skipped. auto PrintAccessor = [&](AccessorKind Kind) -> bool { auto *Accessor = ASD->getAccessor(Kind); if (!Accessor || !shouldPrint(Accessor)) return true; if (!PrintAccessorBody) { Printer << " "; printSelfAccessKindModifiersIfNeeded(Accessor); Printer.printKeyword(getAccessorLabel(Accessor->getAccessorKind()), Options); // handle any effects specifiers if (Accessor->getAccessorKind() == AccessorKind::Get) { if (asyncGet) printWithSpace("async"); if (throwsGet) printWithSpace("throws"); } } else { { IndentRAII IndentMore(*this); indent(); visit(Accessor); } indent(); Printer.printNewline(); } return false; }; // Determine if we should print the getter without the 'get { ... }' // block around it. bool isOnlyGetter = impl.getReadImpl() == ReadImplKind::Get && ASD->getAccessor(AccessorKind::Get); bool isGetterMutating = ASD->supportsMutation() || ASD->isGetterMutating(); bool hasEffects = asyncGet || throwsGet; if (isOnlyGetter && !isGetterMutating && !hasEffects && PrintAccessorBody && Options.FunctionBody && Options.CollapseSingleGetterProperty) { Options.FunctionBody(ASD->getAccessor(AccessorKind::Get), Printer); indent(); return; } Printer << " {"; if (PrintAccessorBody) Printer.printNewline(); if (PrintAbstract) { PrintAccessor(AccessorKind::Get); if (ASD->supportsMutation()) PrintAccessor(AccessorKind::Set); } else { switch (impl.getReadImpl()) { case ReadImplKind::Stored: case ReadImplKind::Inherited: break; case ReadImplKind::Get: PrintAccessor(AccessorKind::Get); break; case ReadImplKind::Address: PrintAccessor(AccessorKind::Address); break; case ReadImplKind::Read: PrintAccessor(AccessorKind::Read); break; } switch (impl.getWriteImpl()) { case WriteImplKind::Immutable: break; case WriteImplKind::Stored: llvm_unreachable("simply-stored variable should have been filtered out"); case WriteImplKind::StoredWithObservers: case WriteImplKind::InheritedWithObservers: { PrintAccessor(AccessorKind::Get); PrintAccessor(AccessorKind::Set); break; } case WriteImplKind::Set: PrintAccessor(AccessorKind::Set); if (!shouldHideModifyAccessor()) PrintAccessor(AccessorKind::Modify); break; case WriteImplKind::MutableAddress: PrintAccessor(AccessorKind::MutableAddress); PrintAccessor(AccessorKind::WillSet); PrintAccessor(AccessorKind::DidSet); break; case WriteImplKind::Modify: PrintAccessor(AccessorKind::Modify); break; } } if (!PrintAccessorBody) Printer << " "; Printer << "}"; indent(); } // This provides logic for looking up all members of a namespace. This is // intentionally implemented only in the printer and should *only* be used for // debugging, testing, generating module dumps, etc. (In other words, if you're // trying to get all the members of a namespace in another part of the compiler, // you're probably doing something wrong. This is a very expensive operation, // so we want to do it only when absolutely necessary.) static void addNamespaceMembers(Decl *decl, llvm::SmallVector &members) { auto &ctx = decl->getASTContext(); auto namespaceDecl = cast(decl->getClangDecl()); // This is only to keep track of the members we've already seen. llvm::SmallPtrSet addedMembers; for (auto redecl : namespaceDecl->redecls()) { for (auto member : redecl->decls()) { if (auto classTemplate = dyn_cast(member)) { // Add all specializations to a worklist so we don't accidently mutate // the list of decls we're iterating over. llvm::SmallPtrSet specWorklist; for (auto spec : classTemplate->specializations()) specWorklist.insert(spec); for (auto spec : specWorklist) { if (auto import = ctx.getClangModuleLoader()->importDeclDirectly(spec)) if (addedMembers.insert(import).second) members.push_back(import); } } auto namedDecl = dyn_cast(member); if (!namedDecl) continue; auto name = ctx.getClangModuleLoader()->importName(namedDecl); if (!name) continue; // If we're building libSyntaxParser, #if out the clang importer request // because libSyntaxParser doesn't know about the clang importer. CXXNamespaceMemberLookup lookupRequest({cast(decl), name}); for (auto found : evaluateOrDefault(ctx.evaluator, lookupRequest, {})) { if (addedMembers.insert(found).second) members.push_back(found); } } } } void PrintAST::printMembersOfDecl(Decl *D, bool needComma, bool openBracket, bool closeBracket) { llvm::SmallVector Members; auto AddMembers = [&](IterableDeclContext *idc) { if (Options.PrintCurrentMembersOnly) { for (auto RD : idc->getMembers()) Members.push_back(RD); } else { for (auto RD : idc->getAllMembers()) Members.push_back(RD); } }; if (auto Ext = dyn_cast(D)) { AddMembers(Ext); } else if (auto NTD = dyn_cast(D)) { AddMembers(NTD); for (auto Ext : NTD->getExtensions()) { if (Options.printExtensionContentAsMembers(Ext)) AddMembers(Ext); } if (Options.PrintExtensionFromConformingProtocols) { for (auto Conf : NTD->getAllConformances()) { for (auto Ext : Conf->getProtocol()->getExtensions()) { if (Options.printExtensionContentAsMembers(Ext)) AddMembers(Ext); } } } if (isa_and_nonnull(D->getClangDecl())) addNamespaceMembers(D, Members); } printMembers(Members, needComma, openBracket, closeBracket); } void PrintAST::printMembers(ArrayRef members, bool needComma, bool openBracket, bool closeBracket) { if (openBracket) { Printer << " {"; if (!Options.PrintEmptyMembersOnSameLine || !members.empty()) Printer.printNewline(); } { IndentRAII indentMore(*this); for (auto i = members.begin(), iEnd = members.end(); i != iEnd; ++i) { auto member = *i; if (!shouldPrint(member, true)) continue; if (!member->shouldPrintInContext(Options)) continue; if (Options.EmptyLineBetweenMembers) Printer.printNewline(); indent(); visit(member); if (needComma && std::next(i) != iEnd) Printer << ","; Printer.printNewline(); } } indent(); if (closeBracket) Printer << "}"; } void PrintAST::printGenericDeclGenericParams(GenericContext *decl) { if (decl->isGeneric()) if (auto GenericSig = decl->getGenericSignature()) { Printer.printStructurePre(PrintStructureKind::DeclGenericParameterClause); printGenericSignature(GenericSig, PrintParams | InnermostOnly); Printer.printStructurePost(PrintStructureKind::DeclGenericParameterClause); } } void PrintAST::printDeclGenericRequirements(GenericContext *decl) { const auto genericSig = decl->getGenericSignature(); if (!genericSig) return; // If the declaration is itself non-generic, it might still // carry a contextual where clause. const auto parentSig = decl->getParent()->getGenericSignatureOfContext(); if (parentSig && parentSig->isEqual(genericSig)) return; Printer.printStructurePre(PrintStructureKind::DeclGenericParameterClause); printGenericSignature(genericSig, PrintRequirements, [parentSig](const Requirement &req) { if (parentSig) return !parentSig->isRequirementSatisfied(req); return true; }); Printer.printStructurePost(PrintStructureKind::DeclGenericParameterClause); } void PrintAST::printInherited(const Decl *decl) { if (!Options.PrintInherited) { return; } SmallVector TypesToPrint; getInheritedForPrinting(decl, Options, TypesToPrint); if (TypesToPrint.empty()) return; if (Options.PrintSpaceBeforeInheritance) { Printer << " "; } Printer << ": "; interleave(TypesToPrint, [&](InheritedEntry inherited) { if (inherited.isUnchecked) Printer << "@unchecked "; printTypeLoc(inherited); }, [&]() { Printer << ", "; }); } static void getModuleEntities(const clang::Module *ClangMod, SmallVectorImpl &ModuleEnts) { if (!ClangMod) return; getModuleEntities(ClangMod->Parent, ModuleEnts); ModuleEnts.push_back(ClangMod); } static void getModuleEntities(ImportDecl *Import, SmallVectorImpl &ModuleEnts) { if (auto *ClangMod = Import->getClangModule()) { getModuleEntities(ClangMod, ModuleEnts); return; } auto Mod = Import->getModule(); if (!Mod) return; if (auto *ClangMod = Mod->findUnderlyingClangModule()) { getModuleEntities(ClangMod, ModuleEnts); } else { ModuleEnts.push_back(Mod); } } void PrintAST::visitImportDecl(ImportDecl *decl) { printAttributes(decl); Printer.printIntroducerKeyword("import", Options, " "); switch (decl->getImportKind()) { case ImportKind::Module: break; case ImportKind::Type: Printer << tok::kw_typealias << " "; break; case ImportKind::Struct: Printer << tok::kw_struct << " "; break; case ImportKind::Class: Printer << tok::kw_class << " "; break; case ImportKind::Enum: Printer << tok::kw_enum << " "; break; case ImportKind::Protocol: Printer << tok::kw_protocol << " "; break; case ImportKind::Var: Printer << tok::kw_var << " "; break; case ImportKind::Func: Printer << tok::kw_func << " "; break; } SmallVector ModuleEnts; getModuleEntities(decl, ModuleEnts); ArrayRef Mods = ModuleEnts; llvm::interleave(decl->getImportPath(), [&](const ImportPath::Element &Elem) { if (!Mods.empty()) { // Should print the module real name in case module // aliasing is used (see -module-alias), since that's // the actual binary name. Identifier Name = decl->getASTContext().getRealModuleName(Elem.Item); if (Options.MapCrossImportOverlaysToDeclaringModule) { if (auto *MD = Mods.front().getAsSwiftModule()) { ModuleDecl *Declaring = const_cast(MD) ->getDeclaringModuleIfCrossImportOverlay(); if (Declaring) Name = Declaring->getRealName(); } } Printer.printModuleRef(Mods.front(), Name); Mods = Mods.slice(1); } else { Printer << Elem.Item.str(); } }, [&] { Printer << "."; }); } void PrintAST::printExtendedTypeName(TypeLoc ExtendedTypeLoc) { bool OldFullyQualifiedTypesIfAmbiguous = Options.FullyQualifiedTypesIfAmbiguous; Options.FullyQualifiedTypesIfAmbiguous = Options.FullyQualifiedExtendedTypesIfAmbiguous; SWIFT_DEFER { Options.FullyQualifiedTypesIfAmbiguous = OldFullyQualifiedTypesIfAmbiguous; }; // Strip off generic arguments, if any. auto Ty = ExtendedTypeLoc.getType()->getAnyNominal()->getDeclaredType(); printTypeLoc(TypeLoc(ExtendedTypeLoc.getTypeRepr(), Ty)); } void PrintAST::printSynthesizedExtension(Type ExtendedType, ExtensionDecl *ExtDecl) { if (Options.PrintCompatibilityFeatureChecks && Options.BracketOptions.shouldOpenExtension(ExtDecl) && Options.BracketOptions.shouldCloseExtension(ExtDecl)) { printWithCompatibilityFeatureChecks(Printer, Options, ExtDecl, [&]{ printSynthesizedExtensionImpl(ExtendedType, ExtDecl); }); } else { printSynthesizedExtensionImpl(ExtendedType, ExtDecl); } } void PrintAST::printSynthesizedExtensionImpl(Type ExtendedType, ExtensionDecl *ExtDecl) { auto printRequirementsFrom = [&](ExtensionDecl *ED, bool &IsFirst) { auto Sig = ED->getGenericSignature(); printSingleDepthOfGenericSignature(Sig.getGenericParams(), Sig.getRequirements(), IsFirst, PrintRequirements, [](const Requirement &Req){ return true; }); }; auto printCombinedRequirementsIfNeeded = [&]() -> bool { if (!Options.TransformContext || !Options.TransformContext->isPrintingSynthesizedExtension()) return false; // Combined requirements only needed if the transform context is an enabling // extension of the protocol rather than a nominal (which can't have // constraints of its own). ExtensionDecl *Target = dyn_cast( Options.TransformContext->getDecl().getAsDecl()); if (!Target || Target == ExtDecl) return false; bool IsFirst = true; if (ExtDecl->isConstrainedExtension()) { printRequirementsFrom(ExtDecl, IsFirst); } if (Target->isConstrainedExtension()) { if (auto *NTD = Target->getExtendedNominal()) { // Update the current decl and type transform for Target rather than // ExtDecl. PrintOptions Adjusted = Options; Adjusted.initForSynthesizedExtension(NTD); llvm::SaveAndRestore TempCurrent(Current, NTD); llvm::SaveAndRestore TempOptions(Options, Adjusted); printRequirementsFrom(Target, IsFirst); } } return true; }; if (Options.BracketOptions.shouldOpenExtension(ExtDecl)) { printDocumentationComment(ExtDecl); printAttributes(ExtDecl); Printer.printIntroducerKeyword("extension", Options, " "); printExtendedTypeName(TypeLoc::withoutLoc(ExtendedType)); printInherited(ExtDecl); // We may need to combine requirements from ExtDecl (which has the members // to print) and the TransformContexts' decl if it is an enabling extension // of the base NominalDecl (which can have its own requirements) rather than // base NominalDecl itself (which can't). E.g: // // protocol Foo {} // extension Foo where { ... } // struct Bar {} // extension Bar: Foo where { ... } // // should produce a synthesized extension of Bar with both sets of // requirements: // // extension Bar where { ... } // if (!printCombinedRequirementsIfNeeded()) printDeclGenericRequirements(ExtDecl); } if (Options.TypeDefinitions) { printMembersOfDecl(ExtDecl, false, Options.BracketOptions.shouldOpenExtension(ExtDecl), Options.BracketOptions.shouldCloseExtension(ExtDecl)); } } void PrintAST::printExtension(ExtensionDecl *decl) { if (Options.BracketOptions.shouldOpenExtension(decl)) { printDocumentationComment(decl); printAttributes(decl); Printer.printIntroducerKeyword("extension", Options, " "); recordDeclLoc(decl, [&]{ // We cannot extend sugared types. Type extendedType = decl->getExtendedType(); if (!extendedType) { // Fallback to TypeRepr. printTypeLoc(decl->getExtendedTypeRepr()); return; } if (!extendedType->getAnyNominal()) { // Fallback to the type. This usually means we're trying to print an // UnboundGenericType. printTypeLoc(TypeLoc::withoutLoc(extendedType)); return; } printExtendedTypeName(TypeLoc(decl->getExtendedTypeRepr(), extendedType)); }); printInherited(decl); if (auto genericSig = decl->getGenericSignature()) { auto baseGenericSig = decl->getExtendedNominal()->getGenericSignature(); assert(baseGenericSig && "an extension can't be generic if the base type isn't"); printGenericSignature(genericSig, PrintRequirements, [baseGenericSig](const Requirement &req) -> bool { // Only include constraints that are not satisfied by the base type. return !baseGenericSig->isRequirementSatisfied(req); }); } } if (Options.TypeDefinitions) { printMembersOfDecl(decl, false, Options.BracketOptions.shouldOpenExtension(decl), Options.BracketOptions.shouldCloseExtension(decl)); } } /// Functions to determine which features a particular declaration uses. The /// usesFeatureNNN functions correspond to the features in Features.def. static bool usesFeatureStaticAssert(Decl *decl) { return false; } static bool usesFeatureEffectfulProp(Decl *decl) { if (auto asd = dyn_cast(decl)) return asd->getEffectfulGetAccessor() != nullptr; return false; } static bool usesFeatureAsyncAwait(Decl *decl) { if (auto func = dyn_cast(decl)) { if (func->hasAsync()) return true; } // Check for async functions in the types of declarations. if (auto value = dyn_cast(decl)) { if (Type type = value->getInterfaceType()) { bool hasAsync = type.findIf([](Type type) { if (auto fnType = type->getAs()) { if (fnType->isAsync()) return true; } return false; }); if (hasAsync) return true; } } return false; } static bool usesFeatureMarkerProtocol(Decl *decl) { return false; } static bool usesFeatureActors(Decl *decl) { if (auto classDecl = dyn_cast(decl)) { if (classDecl->isActor()) return true; } if (auto ext = dyn_cast(decl)) { if (auto classDecl = ext->getSelfClassDecl()) if (classDecl->isActor()) return true; } // Check for actors in the types of declarations. if (auto value = dyn_cast(decl)) { if (Type type = value->getInterfaceType()) { bool hasActor = type.findIf([](Type type) { if (auto classDecl = type->getClassOrBoundGenericClass()) { if (classDecl->isActor()) return true; } return false; }); if (hasActor) return true; } } return false; } static bool usesFeatureConcurrentFunctions(Decl *decl) { return false; } static bool usesFeatureActors2(Decl *decl) { return false; } static bool usesFeatureSendable(Decl *decl) { if (auto func = dyn_cast(decl)) { if (func->isSendable()) return true; } // Check for sendable functions in the types of declarations. if (auto value = dyn_cast(decl)) { if (Type type = value->getInterfaceType()) { bool hasSendable = type.findIf([](Type type) { if (auto fnType = type->getAs()) { if (fnType->isSendable()) return true; } return false; }); if (hasSendable) return true; } } return false; } static bool usesFeatureRethrowsProtocol( Decl *decl, SmallPtrSet &checked) { // Make sure we don't recurse. if (!checked.insert(decl).second) return false; // Check an inheritance clause for a marker protocol. auto checkInherited = [&](ArrayRef inherited) -> bool { for (const auto &inheritedEntry : inherited) { if (auto inheritedType = inheritedEntry.getType()) { if (inheritedType->isExistentialType()) { auto layout = inheritedType->getExistentialLayout(); for (ProtocolDecl *proto : layout.getProtocols()) { if (usesFeatureRethrowsProtocol(proto, checked)) return true; } } } } return false; }; if (auto nominal = dyn_cast(decl)) { if (checkInherited(nominal->getInherited())) return true; } if (auto proto = dyn_cast(decl)) { if (proto->getAttrs().hasAttribute()) return true; } if (auto ext = dyn_cast(decl)) { if (auto nominal = ext->getSelfNominalTypeDecl()) if (usesFeatureRethrowsProtocol(nominal, checked)) return true; if (checkInherited(ext->getInherited())) return true; } if (auto genericSig = decl->getInnermostDeclContext() ->getGenericSignatureOfContext()) { for (const auto &req : genericSig.getRequirements()) { if (req.getKind() == RequirementKind::Conformance && usesFeatureRethrowsProtocol(req.getProtocolDecl(), checked)) return true; } } if (auto value = dyn_cast(decl)) { if (Type type = value->getInterfaceType()) { bool hasRethrowsProtocol = type.findIf([&](Type type) { if (auto nominal = type->getAnyNominal()) { if (usesFeatureRethrowsProtocol(nominal, checked)) return true; } return false; }); if (hasRethrowsProtocol) return true; } } return false; } static bool usesFeatureRethrowsProtocol(Decl *decl) { SmallPtrSet checked; return usesFeatureRethrowsProtocol(decl, checked); } static bool usesFeatureGlobalActors(Decl *decl) { if (auto nominal = dyn_cast(decl)) { if (nominal->getAttrs().hasAttribute()) return true; } if (auto ext = dyn_cast(decl)) { if (auto nominal = ext->getExtendedNominal()) if (usesFeatureGlobalActors(nominal)) return true; } return false; } static bool usesBuiltinType(Decl *decl, BuiltinTypeKind kind) { auto typeMatches = [kind](Type type) { return type.findIf([&](Type type) { if (auto builtinTy = type->getAs()) return builtinTy->getBuiltinTypeKind() == kind; return false; }); }; if (auto value = dyn_cast(decl)) { if (Type type = value->getInterfaceType()) { if (typeMatches(type)) return true; } } if (auto patternBinding = dyn_cast(decl)) { for (unsigned idx : range(patternBinding->getNumPatternEntries())) { if (Type type = patternBinding->getPattern(idx)->getType()) if (typeMatches(type)) return true; } } return false; } static bool usesFeatureBuiltinJob(Decl *decl) { return usesBuiltinType(decl, BuiltinTypeKind::BuiltinJob); } static bool usesFeatureBuiltinExecutor(Decl *decl) { return usesBuiltinType(decl, BuiltinTypeKind::BuiltinExecutor); } static bool usesFeatureBuiltinBuildExecutor(Decl *decl) { return false; } static bool usesFeatureBuiltinBuildMainExecutor(Decl *decl) { return false; } static bool usesFeatureBuiltinContinuation(Decl *decl) { return false; } static bool usesFeatureBuiltinHopToActor(Decl *decl) { return false; } static bool usesFeatureBuiltinTaskGroupWithArgument(Decl *decl) { return false; } static bool usesFeatureBuiltinCreateAsyncTaskInGroup(Decl *decl) { return false; } static bool usesFeatureBuiltinMove(Decl *decl) { return false; } static bool usesFeatureBuiltinCopy(Decl *decl) { return false; } static bool usesFeatureSpecializeAttributeWithAvailability(Decl *decl) { if (auto func = dyn_cast(decl)) { for (auto specialize : func->getAttrs().getAttributes()) { if (!specialize->getAvailableAttrs().empty()) return true; } } return false; } static void suppressingFeatureSpecializeAttributeWithAvailability( PrintOptions &options, llvm::function_ref action) { llvm::SaveAndRestore scope( options.PrintSpecializeAttributeWithAvailability, false); action(); } static bool usesFeatureInheritActorContext(Decl *decl) { if (auto func = dyn_cast(decl)) { for (auto param : *func->getParameters()) { if (param->getAttrs().hasAttribute()) return true; } } return false; } static bool usesFeatureImplicitSelfCapture(Decl *decl) { if (auto func = dyn_cast(decl)) { for (auto param : *func->getParameters()) { if (param->getAttrs().hasAttribute()) return true; } } return false; } static bool usesFeatureBuiltinStackAlloc(Decl *decl) { return false; } static bool usesFeatureBuiltinAssumeAlignment(Decl *decl) { return false; } static bool usesFeatureUnsafeInheritExecutor(Decl *decl) { return decl->getAttrs().hasAttribute(); } static void suppressingFeatureUnsafeInheritExecutor(PrintOptions &options, llvm::function_ref action) { unsigned originalExcludeAttrCount = options.ExcludeAttrList.size(); options.ExcludeAttrList.push_back(DAK_UnsafeInheritExecutor); action(); options.ExcludeAttrList.resize(originalExcludeAttrCount); } static bool usesFeaturePrimaryAssociatedTypes2(Decl *decl) { if (auto *protoDecl = dyn_cast(decl)) { if (protoDecl->getPrimaryAssociatedTypes().size() > 0) return true; } return false; } static void suppressingFeaturePrimaryAssociatedTypes2(PrintOptions &options, llvm::function_ref action) { bool originalPrintPrimaryAssociatedTypes = options.PrintPrimaryAssociatedTypes; options.PrintPrimaryAssociatedTypes = false; action(); options.PrintPrimaryAssociatedTypes = originalPrintPrimaryAssociatedTypes; } static bool usesFeatureUnavailableFromAsync(Decl *decl) { return decl->getAttrs().hasAttribute(); } static void suppressingFeatureUnavailableFromAsync(PrintOptions &options, llvm::function_ref action) { unsigned originalExcludeAttrCount = options.ExcludeAttrList.size(); options.ExcludeAttrList.push_back(DAK_UnavailableFromAsync); action(); options.ExcludeAttrList.resize(originalExcludeAttrCount); } static bool usesFeatureNoAsyncAvailability(Decl *decl) { return decl->getAttrs().getNoAsync(decl->getASTContext()) != nullptr; } static bool usesFeatureConciseMagicFile(Decl *decl) { return false; } static bool usesFeatureForwardTrailingClosures(Decl *decl) { return false; } static bool usesFeatureBareSlashRegexLiterals(Decl *decl) { return false; } static bool usesFeatureVariadicGenerics(Decl *decl) { return false; } static bool usesFeatureNamedOpaqueTypes(Decl *decl) { return false; } static bool usesFeatureFlowSensitiveConcurrencyCaptures(Decl *decl) { return false; } static bool usesFeatureMoveOnly(Decl *decl) { return false; } static bool usesFeatureOneWayClosureParameters(Decl *decl) { return false; } static bool usesFeatureTypeWitnessSystemInference(Decl *decl) { return false; } static bool usesFeatureBoundGenericExtensions(Decl *decl) { return false; } static bool usesFeatureDifferentiableProgramming(Decl *decl) { return false; } static bool usesFeatureForwardModeDifferentiation(Decl *decl) { return false; } static bool usesFeatureAdditiveArithmeticDerivedConformances(Decl *decl) { return false; } static void suppressingFeatureNoAsyncAvailability(PrintOptions &options, llvm::function_ref action) { llvm::SaveAndRestore orignalOptions(options); options.SuppressNoAsyncAvailabilityAttr = true; action(); } /// Suppress the printing of a particular feature. static void suppressingFeature(PrintOptions &options, Feature feature, llvm::function_ref action) { switch (feature) { #define LANGUAGE_FEATURE(FeatureName, SENumber, Description, Option) \ case Feature::FeatureName: \ llvm_unreachable("not a suppressible feature"); #define SUPPRESSIBLE_LANGUAGE_FEATURE(FeatureName, SENumber, Description, Option) \ case Feature::FeatureName: \ suppressingFeature##FeatureName(options, action); \ return; #include "swift/Basic/Features.def" } llvm_unreachable("exhaustive switch"); } using BasicFeatureSet = FixedBitSet; class FeatureSet { BasicFeatureSet required; // Stored inverted: index i actually represents // Feature(numFeatures() - i) // // This is the easiest way of letting us iterate from largest to // smallest, i.e. from the newest to the oldest feature, which is // the order in which we need to emit #if clauses. using SuppressibleFeatureSet = FixedBitSet; SuppressibleFeatureSet suppressible; public: class SuppressibleGenerator { SuppressibleFeatureSet::iterator i, e; friend class FeatureSet; SuppressibleGenerator(const SuppressibleFeatureSet &set) : i(set.begin()), e(set.end()) {} public: bool empty() const { return i == e; } Feature next() { return Feature(numFeatures() - *i++); } }; bool empty() const { return required.empty() && suppressible.empty(); } bool hasAnyRequired() const { return !required.empty(); } const BasicFeatureSet &requiredFeatures() const { return required; } bool hasAnySuppressible() const { return !suppressible.empty(); } SuppressibleGenerator generateSuppressibleFeatures() const { return SuppressibleGenerator(suppressible); } enum InsertOrRemove: bool { Insert = true, Remove = false }; void collectRequiredFeature(Feature feature, InsertOrRemove operation) { assert(!isSuppressibleFeature(feature)); required.insertOrRemove(feature, operation == Insert); } void collectSuppressibleFeature(Feature feature, InsertOrRemove operation) { assert(isSuppressibleFeature(feature)); suppressible.insertOrRemove(numFeatures() - size_t(feature), operation == Insert); } /// Go through all the features used by the given declaration and /// either add or remove them to this set. void collectFeaturesUsed(Decl *decl, InsertOrRemove operation) { // Go through each of the features, checking whether the // declaration uses that feature. #define LANGUAGE_FEATURE(FeatureName, SENumber, Description, Option) \ if (usesFeature##FeatureName(decl)) \ collectRequiredFeature(Feature::FeatureName, operation); #define SUPPRESSIBLE_LANGUAGE_FEATURE(FeatureName, SENumber, Description, Option) \ if (usesFeature##FeatureName(decl)) \ collectSuppressibleFeature(Feature::FeatureName, operation); #include "swift/Basic/Features.def" } }; /// Get the set of features that are uniquely used by this declaration, and are /// not part of the enclosing context. static FeatureSet getUniqueFeaturesUsed(Decl *decl) { // Add all the features used by this declaration. FeatureSet features; features.collectFeaturesUsed(decl, FeatureSet::Insert); // Remove all the features used by all enclosing declarations. Decl *enclosingDecl = decl; while (!features.empty()) { // Find the next outermost enclosing declaration. if (auto accessor = dyn_cast(enclosingDecl)) enclosingDecl = accessor->getStorage(); else enclosingDecl = enclosingDecl->getDeclContext()->getAsDecl(); if (!enclosingDecl) break; features.collectFeaturesUsed(enclosingDecl, FeatureSet::Remove); } return features; } static void printCompatibilityCheckIf(ASTPrinter &printer, bool isElseIf, bool includeCompilerCheck, const BasicFeatureSet &features) { assert(!features.empty()); printer << (isElseIf ? "#elseif " : "#if "); if (includeCompilerCheck) printer << "compiler(>=5.3) && "; bool first = true; for (auto feature : features) { if (!first) { printer << " && "; } else { first = false; } printer << "$" << getFeatureName(feature); } printer.printNewline(); } /// Generate a #if ... #elseif ... #endif chain for the given /// suppressible feature checks. static void printWithSuppressibleFeatureChecks(ASTPrinter &printer, PrintOptions &options, bool firstInChain, bool includeCompilerCheck, FeatureSet::SuppressibleGenerator &generator, llvm::function_ref printBody) { // If we've run out of features to check for, enter an `#else`, // print the body one last time, and close the chain with `#endif`. // Note that, if we didn't have any suppressible features at all, // we shouldn't have started this recursion. if (generator.empty()) { printer << "#else"; printer.printNewline(); printBody(); printer.printNewline(); printer << "#endif"; return; } // Otherwise, enter a `#if` or `#elseif` for the next feature. Feature feature = generator.next(); printCompatibilityCheckIf(printer, /*elseif*/ !firstInChain, includeCompilerCheck, {feature}); // Print the body. printBody(); printer.printNewline(); // Start suppressing the feature and recurse to either generate // more `#elseif` clauses or finish off with `#endif`. suppressingFeature(options, feature, [&] { printWithSuppressibleFeatureChecks(printer, options, /*first*/ false, includeCompilerCheck, generator, printBody); }); } /// Generate the appropriate #if block(s) necessary to protect the use /// of compiler-version-dependent features in the given function. /// /// In the most general form, with both required features and multiple /// suppressible features in play, the generated code pattern looks like /// the following (assuming that feature $bar implies feature $baz): /// /// ``` /// #if compiler(>=5.3) && $foo /// #if $bar /// @foo @bar @baz func @test() {} /// #elseif $baz /// @foo @baz func @test() {} /// #else /// @foo func @test() {} /// #endif /// #endif /// ``` void swift::printWithCompatibilityFeatureChecks(ASTPrinter &printer, PrintOptions &options, Decl *decl, llvm::function_ref printBody) { // A single accessor does not get a feature check, // it should go around the whole decl. if (isa(decl)) { printBody(); return; } FeatureSet features = getUniqueFeaturesUsed(decl); if (features.empty()) { printBody(); return; } // Enter a `#if` for the required features, if any. bool hasRequiredFeatures = features.hasAnyRequired(); if (hasRequiredFeatures) { printCompatibilityCheckIf(printer, /*elseif*/ false, /*compiler check*/ true, features.requiredFeatures()); } // Do the recursive suppression logic if we have suppressible // features, or else just print the body. if (features.hasAnySuppressible()) { auto generator = features.generateSuppressibleFeatures(); printWithSuppressibleFeatureChecks(printer, options, /*first*/ true, /*compiler check*/ !hasRequiredFeatures, generator, printBody); } else { printBody(); } // Close the `#if` for the required features. if (hasRequiredFeatures) { printer.printNewline(); printer << "#endif"; } } void PrintAST::visitExtensionDecl(ExtensionDecl *decl) { if (Options.TransformContext && Options.TransformContext->isPrintingSynthesizedExtension()) { auto extendedType = Options.TransformContext->getBaseType(); if (extendedType->hasArchetype()) extendedType = extendedType->mapTypeOutOfContext(); printSynthesizedExtension(extendedType, decl); } else printExtension(decl); } void PrintAST::visitPatternBindingDecl(PatternBindingDecl *decl) { // FIXME: We're not printing proper "{ get set }" annotations in pattern // binding decls. As a hack, scan the decl to find out if any of the // variables are immutable, and if so, we print as 'let'. This allows us to // handle the 'let x = 4' case properly at least. const VarDecl *anyVar = nullptr; for (auto idx : range(decl->getNumPatternEntries())) { decl->getPattern(idx)->forEachVariable([&](VarDecl *V) { anyVar = V; }); if (anyVar) break; } if (anyVar) printDocumentationComment(anyVar); // FIXME: PatternBindingDecls don't have attributes themselves, so just assume // the variables all have the same attributes. This isn't exactly true // after type-checking, but it's close enough for now. if (anyVar) { printAttributes(anyVar); printAccess(anyVar); } if (decl->isStatic()) printStaticKeyword(decl->getCorrectStaticSpelling()); if (anyVar) { Printer << (anyVar->isSettable(anyVar->getDeclContext()) ? "var " : "let "); } else { Printer << "let "; } bool isFirst = true; for (auto idx : range(decl->getNumPatternEntries())) { auto *pattern = decl->getPattern(idx); if (!shouldPrintPattern(pattern)) continue; if (isFirst) isFirst = false; else Printer << ", "; printPattern(pattern); // We also try to print type for named patterns, e.g. var Field = 10; // and tuple patterns, e.g. var (T1, T2) = (10, 10) if (isa(pattern) || isa(pattern)) { printPatternType(pattern); } if (Options.PrintExprs) { if (auto initExpr = decl->getInit(idx)) { Printer << " = "; visit(initExpr); } } else if (Options.VarInitializers) { auto *vd = decl->getAnchoringVarDecl(idx); if (decl->hasInitStringRepresentation(idx) && vd->isInitExposedToClients()) { SmallString<128> scratch; Printer << " = " << decl->getInitStringRepresentation(idx, scratch); } } // If we're just printing a single pattern and it has accessors, // print the accessors here. It is an error to add accessors to a // pattern binding with multiple entries. if (auto var = decl->getSingleVar()) { printAccessors(var); } } } void PrintAST::visitTopLevelCodeDecl(TopLevelCodeDecl *decl) { printASTNodes(decl->getBody()->getElements(), /*NeedIndent=*/false); } void PrintAST::visitIfConfigDecl(IfConfigDecl *ICD) { if (!Options.PrintIfConfig) return; for (auto &Clause : ICD->getClauses()) { if (&Clause == &*ICD->getClauses().begin()) Printer << tok::pound_if << " /* condition */"; // FIXME: print condition else if (Clause.Cond) Printer << tok::pound_elseif << " /* condition */"; // FIXME: print condition else Printer << tok::pound_else; printASTNodes(Clause.Elements); Printer.printNewline(); indent(); } Printer << tok::pound_endif; } void PrintAST::visitPoundDiagnosticDecl(PoundDiagnosticDecl *PDD) { /// TODO: Should we even print #error/#warning? if (PDD->isError()) { Printer << tok::pound_error; } else { Printer << tok::pound_warning; } Printer << "(\"" << PDD->getMessage()->getValue() << "\")"; } void PrintAST::visitOpaqueTypeDecl(OpaqueTypeDecl *decl) { // TODO: If we introduce explicit opaque type decls, print them. assert(decl->getName().empty()); } void PrintAST::visitTypeAliasDecl(TypeAliasDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printAccess(decl); Printer.printIntroducerKeyword("typealias", Options, " "); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName(), getTypeMemberPrintNameContext(decl)); }, [&]{ // Signature printGenericDeclGenericParams(decl); }); bool ShouldPrint = true; Type Ty = decl->getUnderlyingType(); // If the underlying type is private, don't print it. if (Options.SkipPrivateStdlibDecls && Ty && Ty.isPrivateStdlibType()) ShouldPrint = false; if (ShouldPrint) { Printer << " = "; // FIXME: An inferred associated type witness type alias may reference // an opaque type, but OpaqueTypeArchetypes are always canonicalized // so lose type sugar for generic params. Bind the generic signature so // we can map params back into the generic signature and print them // correctly. // // Remove this when we have a way to represent non-canonical archetypes // preserving sugar. llvm::SaveAndRestore setGenericSig( Options.GenericSig, decl->getGenericSignature().getPointer()); printTypeLoc(TypeLoc(decl->getUnderlyingTypeRepr(), Ty)); printDeclGenericRequirements(decl); } } void PrintAST::visitGenericTypeParamDecl(GenericTypeParamDecl *decl) { recordDeclLoc(decl, [&] { Printer.printName(decl->getName(), PrintNameContext::GenericParameter); }); printInherited(decl); } void PrintAST::visitAssociatedTypeDecl(AssociatedTypeDecl *decl) { printDocumentationComment(decl); printAttributes(decl); Printer.printIntroducerKeyword("associatedtype", Options, " "); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName(), PrintNameContext::TypeMember); }); auto proto = decl->getProtocol(); printInheritedFromRequirementSignature(proto, decl); if (decl->hasDefaultDefinitionType()) { Printer << " = "; decl->getDefaultDefinitionType().print(Printer, Options); } // As with protocol's trailing where clauses, use the requirement signature // when available. printWhereClauseFromRequirementSignature(proto, decl); } void PrintAST::visitEnumDecl(EnumDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printAccess(decl); if (Options.PrintOriginalSourceText && decl->getStartLoc().isValid()) { ASTContext &Ctx = decl->getASTContext(); printSourceRange(CharSourceRange(Ctx.SourceMgr, decl->getStartLoc(), decl->getBraces().Start.getAdvancedLoc(-1)), Ctx); } else { Printer.printIntroducerKeyword("enum", Options, " "); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName(), getTypeMemberPrintNameContext(decl)); }, [&]{ // Signature printGenericDeclGenericParams(decl); }); printInherited(decl); printDeclGenericRequirements(decl); } if (Options.TypeDefinitions) { printMembersOfDecl(decl, false, true, Options.BracketOptions.shouldCloseNominal(decl)); } } void PrintAST::visitStructDecl(StructDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printAccess(decl); if (Options.PrintOriginalSourceText && decl->getStartLoc().isValid()) { ASTContext &Ctx = decl->getASTContext(); printSourceRange(CharSourceRange(Ctx.SourceMgr, decl->getStartLoc(), decl->getBraces().Start.getAdvancedLoc(-1)), Ctx); } else { Printer.printIntroducerKeyword("struct", Options, " "); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName(), getTypeMemberPrintNameContext(decl)); }, [&]{ // Signature printGenericDeclGenericParams(decl); }); printInherited(decl); printDeclGenericRequirements(decl); } if (Options.TypeDefinitions) { printMembersOfDecl(decl, false, true, Options.BracketOptions.shouldCloseNominal(decl)); } } void PrintAST::visitClassDecl(ClassDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printAccess(decl); if (Options.PrintOriginalSourceText && decl->getStartLoc().isValid()) { ASTContext &Ctx = decl->getASTContext(); printSourceRange(CharSourceRange(Ctx.SourceMgr, decl->getStartLoc(), decl->getBraces().Start.getAdvancedLoc(-1)), Ctx); } else { Printer.printIntroducerKeyword( decl->isExplicitActor() ? "actor" : "class", Options, " "); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName(), getTypeMemberPrintNameContext(decl)); }, [&]{ // Signature printGenericDeclGenericParams(decl); }); printInherited(decl); printDeclGenericRequirements(decl); } if (Options.TypeDefinitions) { printMembersOfDecl(decl, false, true, Options.BracketOptions.shouldCloseNominal(decl)); } } void PrintAST::printPrimaryAssociatedTypes(ProtocolDecl *decl) { auto primaryAssocTypes = decl->getPrimaryAssociatedTypes(); if (primaryAssocTypes.empty()) return; Printer.printStructurePre(PrintStructureKind::DeclGenericParameterClause); Printer << "<"; llvm::interleave( primaryAssocTypes, [&](AssociatedTypeDecl *assocType) { Printer.callPrintStructurePre(PrintStructureKind::GenericParameter, assocType); Printer.printTypeRef(assocType->getDeclaredInterfaceType(), assocType, assocType->getName(), PrintNameContext::GenericParameter); Printer.printStructurePost(PrintStructureKind::GenericParameter, assocType); }, [&] { Printer << ", "; }); Printer << ">"; Printer.printStructurePost(PrintStructureKind::DeclGenericParameterClause); } void PrintAST::visitProtocolDecl(ProtocolDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printAccess(decl); if (Options.PrintOriginalSourceText && decl->getStartLoc().isValid()) { ASTContext &Ctx = decl->getASTContext(); printSourceRange(CharSourceRange(Ctx.SourceMgr, decl->getStartLoc(), decl->getBraces().Start.getAdvancedLoc(-1)), Ctx); } else { Printer.printIntroducerKeyword("protocol", Options, " "); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName()); }); if (Options.PrintPrimaryAssociatedTypes) { printPrimaryAssociatedTypes(decl); } printInheritedFromRequirementSignature(decl, decl); // The trailing where clause is a syntactic thing, which isn't serialized // (etc.) and thus isn't available for printing things out of // already-compiled SIL modules. The requirement signature is available in // such cases, so let's go with that when we can. printWhereClauseFromRequirementSignature(decl, decl); } if (Options.TypeDefinitions) { printMembersOfDecl(decl, false, true, Options.BracketOptions.shouldCloseNominal(decl)); } } static bool isStructOrClassContext(DeclContext *dc) { auto *nominal = dc->getSelfNominalTypeDecl(); if (nominal == nullptr) return false; return isa(nominal) || isa(nominal); } static bool isEscaping(Type type) { if (auto *funcType = type->getAs()) { if (funcType->getExtInfo().getRepresentation() == FunctionTypeRepresentation::CFunctionPointer) return false; return !funcType->getExtInfo().isNoEscape(); } return false; } static void printParameterFlags(ASTPrinter &printer, const PrintOptions &options, ParameterTypeFlags flags, bool escaping) { if (!options.excludeAttrKind(TAK_autoclosure) && flags.isAutoClosure()) printer.printAttrName("@autoclosure "); if (!options.excludeAttrKind(TAK_noDerivative) && flags.isNoDerivative()) printer.printAttrName("@noDerivative "); switch (flags.getValueOwnership()) { case ValueOwnership::Default: /* nothing */ break; case ValueOwnership::InOut: printer.printKeyword("inout", options, " "); break; case ValueOwnership::Shared: printer.printKeyword("__shared", options, " "); break; case ValueOwnership::Owned: printer.printKeyword("__owned", options, " "); break; } if (flags.isIsolated()) printer.printKeyword("isolated", options, " "); if (!options.excludeAttrKind(TAK_escaping) && escaping) printer.printKeyword("@escaping", options, " "); if (flags.isCompileTimeConst()) printer.printKeyword("_const", options, " "); } void PrintAST::visitVarDecl(VarDecl *decl) { printDocumentationComment(decl); // Print @_hasStorage when the attribute is not already // on, decl has storage and it is on a class. if (Options.PrintForSIL && decl->hasStorage() && isStructOrClassContext(decl->getDeclContext()) && !decl->getAttrs().hasAttribute()) Printer << "@_hasStorage "; printAttributes(decl); printAccess(decl); if (decl->isStatic() && Options.PrintStaticKeyword) printStaticKeyword(decl->getCorrectStaticSpelling()); if (decl->getKind() == DeclKind::Var || Options.PrintParameterSpecifiers) { // Map all non-let specifiers to 'var'. This is not correct, but // SourceKit relies on this for info about parameter decls. Printer.printIntroducerKeyword(decl->isLet() ? "let" : "var", Options, " "); } printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName(), getTypeMemberPrintNameContext(decl)); }); { Printer.printStructurePre(PrintStructureKind::DeclResultTypeClause); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::DeclResultTypeClause); }; auto type = decl->getInterfaceType(); Printer << ": "; TypeLoc tyLoc; if (auto *repr = decl->getTypeReprOrParentPatternTypeRepr()) { tyLoc = TypeLoc(repr, type); } else { tyLoc = TypeLoc::withoutLoc(type); } Printer.printDeclResultTypePre(decl, tyLoc); // HACK: When printing result types for vars with opaque result types, // always print them using the `some` keyword instead of printing // the full stable reference. llvm::SaveAndRestore x(Options.OpaqueReturnTypePrinting, PrintOptions::OpaqueReturnTypePrintingMode::WithOpaqueKeyword); printTypeLocForImplicitlyUnwrappedOptional( tyLoc, decl->isImplicitlyUnwrappedOptional()); } printAccessors(decl); } void PrintAST::visitParamDecl(ParamDecl *decl) { visitVarDecl(decl); } void PrintAST::printOneParameter(const ParamDecl *param, ParameterTypeFlags paramFlags, bool ArgNameIsAPIByDefault) { Printer.callPrintStructurePre(PrintStructureKind::FunctionParameter, param); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::FunctionParameter, param); }; auto printArgName = [&]() { // Print argument name. auto ArgName = param->getArgumentName(); auto BodyName = param->getName(); switch (Options.ArgAndParamPrinting) { case PrintOptions::ArgAndParamPrintingMode::EnumElement: if (ArgName.empty() && BodyName.empty() && !param->hasDefaultExpr()) { // Don't print anything, in the style of a tuple element. return; } // Else, print the argument only. LLVM_FALLTHROUGH; case PrintOptions::ArgAndParamPrintingMode::ArgumentOnly: if (ArgName.empty() && !Options.PrintEmptyArgumentNames) { return; } Printer.printName(ArgName, PrintNameContext::FunctionParameterExternal); if (!ArgNameIsAPIByDefault && !ArgName.empty()) Printer << " _"; break; case PrintOptions::ArgAndParamPrintingMode::MatchSource: if (ArgName == BodyName && ArgNameIsAPIByDefault) { Printer.printName(ArgName, PrintNameContext::FunctionParameterExternal); break; } if (ArgName.empty() && !ArgNameIsAPIByDefault) { Printer.printName(BodyName, PrintNameContext::FunctionParameterLocal); break; } LLVM_FALLTHROUGH; case PrintOptions::ArgAndParamPrintingMode::BothAlways: Printer.printName(ArgName, PrintNameContext::FunctionParameterExternal); Printer << " "; Printer.printName(BodyName, PrintNameContext::FunctionParameterLocal); break; } Printer << ": "; }; printAttributes(param); printArgName(); TypeLoc TheTypeLoc; if (auto *repr = param->getTypeRepr()) { TheTypeLoc = TypeLoc(repr, param->getInterfaceType()); } else { TheTypeLoc = TypeLoc::withoutLoc(param->getInterfaceType()); } // If the parameter is variadic, we will print the "..." after it, but we have // to strip off the added array type. if (param->isVariadic() && TheTypeLoc.getType()) { if (auto *BGT = TheTypeLoc.getType()->getAs()) TheTypeLoc.setType(BGT->getGenericArgs()[0]); } { Printer.printStructurePre(PrintStructureKind::FunctionParameterType); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::FunctionParameterType); }; if (!param->isVariadic() && !willUseTypeReprPrinting(TheTypeLoc, CurrentType, Options)) { auto type = TheTypeLoc.getType(); printParameterFlags(Printer, Options, paramFlags, isEscaping(type)); } printTypeLocForImplicitlyUnwrappedOptional( TheTypeLoc, param->isImplicitlyUnwrappedOptional()); if (param->isVariadic()) Printer << "..."; } if (param->isDefaultArgument() && Options.PrintDefaultArgumentValue) { Printer.callPrintStructurePre(PrintStructureKind::DefaultArgumentClause); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::DefaultArgumentClause); }; SmallString<128> scratch; auto defaultArgStr = param->getDefaultValueStringRepresentation(scratch); assert(!defaultArgStr.empty() && "empty default argument?"); Printer << " = "; switch (param->getDefaultArgumentKind()) { #define MAGIC_IDENTIFIER(NAME, STRING, SYNTAX_KIND) \ case DefaultArgumentKind::NAME: #include "swift/AST/MagicIdentifierKinds.def" Printer.printKeyword(defaultArgStr, Options); break; default: Printer << defaultArgStr; break; } } } void PrintAST::printParameterList(ParameterList *PL, ArrayRef params, bool isAPINameByDefault) { Printer.printStructurePre(PrintStructureKind::FunctionParameterList); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::FunctionParameterList); }; Printer << "("; const unsigned paramSize = params.size(); for (unsigned i = 0, e = PL->size(); i != e; ++i) { if (i > 0) Printer << ", "; auto paramFlags = (i < paramSize) ? params[i].getParameterFlags() : ParameterTypeFlags(); printOneParameter(PL->get(i), paramFlags, isAPINameByDefault); } Printer << ")"; } void PrintAST::printFunctionParameters(AbstractFunctionDecl *AFD) { auto BodyParams = AFD->getParameters(); auto curTy = AFD->getInterfaceType(); // Skip over the implicit 'self'. if (AFD->hasImplicitSelfDecl()) if (auto funTy = curTy->getAs()) curTy = funTy->getResult(); ArrayRef parameterListTypes; if (auto funTy = curTy->getAs()) parameterListTypes = funTy->getParams(); printParameterList(BodyParams, parameterListTypes, AFD->argumentNameIsAPIByDefault()); if (AFD->hasAsync() || AFD->hasThrows()) { Printer.printStructurePre(PrintStructureKind::EffectsSpecifiers); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::EffectsSpecifiers); }; if (AFD->hasAsync()) { Printer << " "; if (AFD->getAttrs().hasAttribute()) Printer.printKeyword("reasync", Options); else Printer.printKeyword("async", Options); } if (AFD->hasThrows()) { if (AFD->getAttrs().hasAttribute()) Printer << " " << tok::kw_rethrows; else Printer << " " << tok::kw_throws; } } } bool PrintAST::printASTNodes(const ArrayRef &Elements, bool NeedIndent) { IndentRAII IndentMore(*this, NeedIndent); bool PrintedSomething = false; for (auto element : Elements) { PrintedSomething = true; Printer.printNewline(); indent(); if (auto decl = element.dyn_cast()) { if (decl->shouldPrintInContext(Options)) visit(decl); } else if (auto stmt = element.dyn_cast()) { visit(stmt); } else { visit(element.get()); } } return PrintedSomething; } void PrintAST::visitAccessorDecl(AccessorDecl *decl) { printDocumentationComment(decl); printAttributes(decl); // Explicitly print 'mutating' and 'nonmutating' if needed. printSelfAccessKindModifiersIfNeeded(decl); switch (auto kind = decl->getAccessorKind()) { case AccessorKind::Get: case AccessorKind::Address: case AccessorKind::Read: case AccessorKind::Modify: case AccessorKind::DidSet: case AccessorKind::MutableAddress: recordDeclLoc(decl, [&]{ Printer << getAccessorLabel(decl->getAccessorKind()); }); break; case AccessorKind::Set: case AccessorKind::WillSet: recordDeclLoc(decl, [&]{ Printer << getAccessorLabel(decl->getAccessorKind()); auto params = decl->getParameters(); if (params->size() != 0 && !params->get(0)->isImplicit()) { auto Name = params->get(0)->getName(); if (!Name.empty()) { Printer << "("; Printer.printName(Name); Printer << ")"; } } }); } // handle effects specifiers before the body if (decl->hasAsync()) Printer << " async"; if (decl->hasThrows()) Printer << " throws"; printBodyIfNecessary(decl); } void PrintAST::visitFuncDecl(FuncDecl *decl) { ASTContext &Ctx = decl->getASTContext(); printDocumentationComment(decl); printAttributes(decl); printAccess(decl); if (Options.PrintOriginalSourceText && decl->getStartLoc().isValid()) { SourceLoc StartLoc = decl->getStartLoc(); SourceLoc EndLoc; if (decl->getResultTypeRepr()) { EndLoc = decl->getResultTypeSourceRange().End; } else { EndLoc = decl->getSignatureSourceRange().End; } CharSourceRange Range = Lexer::getCharSourceRangeFromSourceRange(Ctx.SourceMgr, SourceRange(StartLoc, EndLoc)); printSourceRange(Range, Ctx); } else { if (decl->isStatic() && Options.PrintStaticKeyword) printStaticKeyword(decl->getCorrectStaticSpelling()); printSelfAccessKindModifiersIfNeeded(decl); Printer.printIntroducerKeyword("func", Options, " "); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ // Name if (!decl->hasName()) { Printer << ""; } else { Printer.printName(decl->getBaseIdentifier(), getTypeMemberPrintNameContext(decl)); if (decl->isOperator()) Printer << " "; } }, [&] { // Parameters printGenericDeclGenericParams(decl); printFunctionParameters(decl); }); Type ResultTy = decl->getResultInterfaceType(); if (ResultTy && !ResultTy->isVoid()) { Printer.printStructurePre(PrintStructureKind::DeclResultTypeClause); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::DeclResultTypeClause); }; TypeLoc ResultTyLoc(decl->getResultTypeRepr(), ResultTy); // When printing a protocol requirement with types substituted for a // conforming class, replace occurrences of the 'Self' generic parameter // in the result type with DynamicSelfType, instead of the static // conforming type. auto *proto = dyn_cast(decl->getDeclContext()); if (proto && Options.TransformContext) { auto BaseType = Options.TransformContext->getBaseType(); if (BaseType->getClassOrBoundGenericClass()) { ResultTy = ResultTy.subst( [&](Type t) -> Type { if (t->isEqual(proto->getSelfInterfaceType())) return DynamicSelfType::get(t, Ctx); return t; }, MakeAbstractConformanceForGenericType()); ResultTyLoc = TypeLoc::withoutLoc(ResultTy); } } if (!ResultTyLoc.getTypeRepr()) ResultTyLoc = TypeLoc::withoutLoc(ResultTy); // FIXME: Hacky way to workaround the fact that 'Self' as return // TypeRepr is not getting 'typechecked'. See // \c resolveTopLevelIdentTypeComponent function in TypeCheckType.cpp. if (auto *simId = dyn_cast_or_null(ResultTyLoc.getTypeRepr())) { if (simId->getNameRef().isSimpleName(Ctx.Id_Self)) ResultTyLoc = TypeLoc::withoutLoc(ResultTy); } Printer << " -> "; Printer.printDeclResultTypePre(decl, ResultTyLoc); Printer.callPrintStructurePre(PrintStructureKind::FunctionReturnType); // HACK: When printing result types for funcs with opaque result types, // always print them using the `some` keyword instead of printing // the full stable reference. llvm::SaveAndRestore x(Options.OpaqueReturnTypePrinting, PrintOptions::OpaqueReturnTypePrintingMode::WithOpaqueKeyword); printTypeLocForImplicitlyUnwrappedOptional( ResultTyLoc, decl->isImplicitlyUnwrappedOptional()); Printer.printStructurePost(PrintStructureKind::FunctionReturnType); } printDeclGenericRequirements(decl); } printBodyIfNecessary(decl); // If the function has an opaque result type, print the opaque type decl. if (auto opaqueResult = decl->getOpaqueResultTypeDecl()) { Printer.printNewline(); visit(opaqueResult); } } void PrintAST::printEnumElement(EnumElementDecl *elt) { recordDeclLoc(elt, [&]{ Printer.printName(elt->getBaseIdentifier(), getTypeMemberPrintNameContext(elt)); }); if (auto *PL = elt->getParameterList()) { llvm::SaveAndRestore mode(Options.ArgAndParamPrinting, PrintOptions::ArgAndParamPrintingMode::EnumElement); auto params = ArrayRef(); if (!elt->isInvalid()) { // Walk to the params of the associated values. // (EnumMetaType) -> (AssocValues) -> Enum auto type = elt->getInterfaceType(); params = type->castTo() ->getResult() ->castTo() ->getParams(); } // @escaping is not valid in enum element position, even though the // attribute is implicitly added. Ignore it when printing the parameters. Options.ExcludeAttrList.push_back(TAK_escaping); printParameterList(PL, params, /*isAPINameByDefault*/true); Options.ExcludeAttrList.pop_back(); } switch (Options.EnumRawValues) { case PrintOptions::EnumRawValueMode::Skip: return; case PrintOptions::EnumRawValueMode::PrintObjCOnly: if (!elt->isObjC()) return; break; case PrintOptions::EnumRawValueMode::Print: break; } auto *raw = elt->getStructuralRawValueExpr(); if (!raw || raw->isImplicit()) return; // Print the explicit raw value expression. Printer << " = "; switch (raw->getKind()) { case ExprKind::IntegerLiteral: case ExprKind::FloatLiteral: { auto *numLiteral = cast(raw); Printer.callPrintStructurePre(PrintStructureKind::NumberLiteral); if (numLiteral->isNegative()) Printer << "-"; Printer << numLiteral->getDigitsText(); Printer.printStructurePost(PrintStructureKind::NumberLiteral); break; } case ExprKind::StringLiteral: { Printer.callPrintStructurePre(PrintStructureKind::StringLiteral); llvm::SmallString<32> str; llvm::raw_svector_ostream os(str); os << QuotedString(cast(raw)->getValue()); Printer << str; Printer.printStructurePost(PrintStructureKind::StringLiteral); break; } default: break; // Incorrect raw value; skip it for error recovery. } } void PrintAST::visitEnumCaseDecl(EnumCaseDecl *decl) { auto elems = decl->getElements(); if (!elems.empty()) { // Documentation comments over the case are attached to the enum elements. printDocumentationComment(elems[0]); printAttributes(elems[0]); } Printer.printIntroducerKeyword("case", Options, " "); llvm::interleave(elems.begin(), elems.end(), [&](EnumElementDecl *elt) { printEnumElement(elt); }, [&] { Printer << ", "; }); } void PrintAST::visitEnumElementDecl(EnumElementDecl *decl) { printDocumentationComment(decl); // In cases where there is no parent EnumCaseDecl (such as imported or // deserialized elements), print the element independently. printAttributes(decl); Printer.printIntroducerKeyword("case", Options, " "); printEnumElement(decl); } void PrintAST::visitSubscriptDecl(SubscriptDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printAccess(decl); if (decl->isStatic() && Options.PrintStaticKeyword) printStaticKeyword(decl->getCorrectStaticSpelling()); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer << "subscript"; }, [&] { // Parameters printGenericDeclGenericParams(decl); auto params = ArrayRef(); if (!decl->isInvalid()) { // Walk to the params of the subscript's indices. auto type = decl->getInterfaceType(); params = type->castTo()->getParams(); } printParameterList(decl->getIndices(), params, /*isAPINameByDefault*/false); }); { Printer.printStructurePre(PrintStructureKind::DeclResultTypeClause); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::DeclResultTypeClause); }; Printer << " -> "; TypeLoc elementTy(decl->getElementTypeRepr(), decl->getElementInterfaceType()); Printer.printDeclResultTypePre(decl, elementTy); Printer.callPrintStructurePre(PrintStructureKind::FunctionReturnType); // HACK: When printing result types for subscripts with opaque result types, // always print them using the `some` keyword instead of printing // the full stable reference. llvm::SaveAndRestore x(Options.OpaqueReturnTypePrinting, PrintOptions::OpaqueReturnTypePrintingMode::WithOpaqueKeyword); printTypeLocForImplicitlyUnwrappedOptional( elementTy, decl->isImplicitlyUnwrappedOptional()); Printer.printStructurePost(PrintStructureKind::FunctionReturnType); } printDeclGenericRequirements(decl); printAccessors(decl); } void PrintAST::visitConstructorDecl(ConstructorDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printAccess(decl); if ((decl->getInitKind() == CtorInitializerKind::Convenience || decl->getInitKind() == CtorInitializerKind::ConvenienceFactory) && !decl->getAttrs().hasAttribute()) { // Protocol extension initializers are modeled as convenience initializers, // but they're not written that way in source. Check if we're actually // printing onto a class. bool isClassContext; if (CurrentType) { isClassContext = CurrentType->getClassOrBoundGenericClass() != nullptr; } else { const DeclContext *dc = decl->getDeclContext(); isClassContext = dc->getSelfClassDecl() != nullptr; } if (isClassContext) { Printer.printKeyword("convenience", Options, " "); } else { assert(decl->getDeclContext()->getExtendedProtocolDecl() && "unexpected convenience initializer"); } } else if (decl->getInitKind() == CtorInitializerKind::Factory) { Printer << "/*not inherited*/ "; } printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer << "init"; }, [&] { // Signature if (decl->isFailable()) { if (decl->isImplicitlyUnwrappedOptional()) Printer << "!"; else Printer << "?"; } printGenericDeclGenericParams(decl); printFunctionParameters(decl); }); printDeclGenericRequirements(decl); printBodyIfNecessary(decl); } void PrintAST::visitDestructorDecl(DestructorDecl *decl) { printDocumentationComment(decl); printAttributes(decl); printContextIfNeeded(decl); recordDeclLoc(decl, [&]{ Printer << "deinit"; }); printBodyIfNecessary(decl); } void PrintAST::visitInfixOperatorDecl(InfixOperatorDecl *decl) { Printer.printKeyword("infix", Options, " "); Printer.printIntroducerKeyword("operator", Options, " "); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName()); }); if (auto *group = decl->getPrecedenceGroup()) Printer << " : " << group->getName(); } void PrintAST::visitPrecedenceGroupDecl(PrecedenceGroupDecl *decl) { Printer.printIntroducerKeyword("precedencegroup", Options, " "); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName()); }); Printer << " {"; Printer.printNewline(); { IndentRAII indentMore(*this); if (!decl->isAssociativityImplicit() || !decl->isNonAssociative()) { indent(); Printer.printKeyword("associativity", Options, ": "); switch (decl->getAssociativity()) { case Associativity::None: Printer.printKeyword("none", Options); break; case Associativity::Left: Printer.printKeyword("left", Options); break; case Associativity::Right: Printer.printKeyword("right", Options); break; } Printer.printNewline(); } if (!decl->isAssignmentImplicit() || decl->isAssignment()) { indent(); Printer.printKeyword("assignment", Options, ": "); Printer.printKeyword(decl->isAssignment() ? "true" : "false", Options); Printer.printNewline(); } if (!decl->getHigherThan().empty()) { indent(); Printer.printKeyword("higherThan", Options, ": "); if (!decl->getHigherThan().empty()) { Printer << decl->getHigherThan()[0].Name; for (auto &rel : decl->getHigherThan().slice(1)) Printer << ", " << rel.Name; } Printer.printNewline(); } if (!decl->getLowerThan().empty()) { indent(); Printer.printKeyword("lowerThan", Options, ": "); if (!decl->getLowerThan().empty()) { Printer << decl->getLowerThan()[0].Name; for (auto &rel : decl->getLowerThan().slice(1)) Printer << ", " << rel.Name; } Printer.printNewline(); } } indent(); Printer << "}"; } void PrintAST::visitPrefixOperatorDecl(PrefixOperatorDecl *decl) { Printer.printKeyword("prefix", Options, " "); Printer.printIntroducerKeyword("operator", Options, " "); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName()); }); } void PrintAST::visitPostfixOperatorDecl(PostfixOperatorDecl *decl) { Printer.printKeyword("postfix", Options, " "); Printer.printIntroducerKeyword("operator", Options, " "); recordDeclLoc(decl, [&]{ Printer.printName(decl->getName()); }); } void PrintAST::visitModuleDecl(ModuleDecl *decl) { } void PrintAST::visitMissingMemberDecl(MissingMemberDecl *decl) { Printer << "/* placeholder for "; recordDeclLoc(decl, [&]{ Printer << decl->getName(); }); unsigned numVTableEntries = decl->getNumberOfVTableEntries(); if (numVTableEntries > 0) Printer << " (vtable entries: " << numVTableEntries << ")"; unsigned numFieldOffsetVectorEntries = decl->getNumberOfFieldOffsetVectorEntries(); if (numFieldOffsetVectorEntries > 0) Printer << " (field offsets: " << numFieldOffsetVectorEntries << ")"; Printer << " */"; } void PrintAST::visitIntegerLiteralExpr(IntegerLiteralExpr *expr) { Printer << expr->getDigitsText(); } void PrintAST::visitFloatLiteralExpr(FloatLiteralExpr *expr) { Printer << expr->getDigitsText(); } void PrintAST::visitNilLiteralExpr(NilLiteralExpr *expr) { Printer << "nil"; } void PrintAST::visitStringLiteralExpr(StringLiteralExpr *expr) { Printer << "\"" << expr->getValue() << "\""; } void PrintAST::visitBooleanLiteralExpr(BooleanLiteralExpr *expr) { if (expr->getValue()) { Printer << "true"; } else { Printer << "false"; } } void PrintAST::visitRegexLiteralExpr(RegexLiteralExpr *expr) { Printer << expr->getRegexText(); } void PrintAST::visitErrorExpr(ErrorExpr *expr) { Printer << ""; } void PrintAST::visitIfExpr(IfExpr *expr) { } void PrintAST::visitIsExpr(IsExpr *expr) { } void PrintAST::visitTapExpr(TapExpr *expr) { } void PrintAST::visitTryExpr(TryExpr *expr) { Printer << "try "; visit(expr->getSubExpr()); } void PrintAST::visitCallExpr(CallExpr *expr) { visit(expr->getFn()); Printer << "("; auto args = expr->getArgs()->getOriginalArgs(); bool isFirst = true; // FIXME: handle trailing closures. for (auto arg : *args) { if (!isFirst) { Printer << ", "; } printArgument(arg); isFirst = false; } Printer << ")"; } void PrintAST::printArgument(const Argument &arg) { auto label = arg.getLabel(); if (!label.empty()) { Printer << label.str(); Printer << ": "; } if (arg.isInOut()) { Printer << "&"; } visit(arg.getExpr()); } void PrintAST::visitLoadExpr(LoadExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitTypeExpr(TypeExpr *expr) { if (auto metaType = expr->getType()->castTo()) { // Don't print `.Type` for an expr. printType(metaType->getInstanceType()); } else { printType(expr->getType()); } } void PrintAST::visitArrayExpr(ArrayExpr *expr) { Printer << "["; bool isFirst = true; auto elements = expr->getElements(); for (auto element : elements) { if (!isFirst) { Printer << ", "; } visit(element); isFirst = false; } Printer << "]"; } void PrintAST::visitDictionaryExpr(DictionaryExpr *expr) { Printer << "["; bool isFirst = true; auto elements = expr->getElements(); for (auto element : elements) { auto *tupleExpr = cast(element); if (!isFirst) { Printer << ", "; } visit(tupleExpr->getElement(0)); Printer << ": "; visit(tupleExpr->getElement(1)); isFirst = false; } Printer << "]"; } void PrintAST::visitArrowExpr(ArrowExpr *expr) { } void PrintAST::visitAwaitExpr(AwaitExpr *expr) { Printer << "await "; visit(expr->getSubExpr()); } void PrintAST::visitInOutExpr(InOutExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitParenExpr(ParenExpr *expr) { Printer << "("; visit(expr->getSubExpr()); Printer << ")"; } void PrintAST::visitTupleExpr(TupleExpr *expr) { Printer << "("; bool isFirst = true; auto elements = expr->getElements(); for (auto element : elements) { if (!isFirst) { Printer << ", "; } visit(element); isFirst = false; } Printer << ")"; } void PrintAST::visitPackExpr(PackExpr *expr) { } void PrintAST::visitReifyPackExpr(ReifyPackExpr *expr) { } void PrintAST::visitAssignExpr(AssignExpr *expr) { visit(expr->getDest()); Printer << " = "; visit(expr->getSrc()); } void PrintAST::visitBinaryExpr(BinaryExpr *expr) { visit(expr->getLHS()); Printer << " "; if (auto operatorRef = expr->getFn()->getMemberOperatorRef()) { Printer << operatorRef->getDecl()->getBaseName(); } else if (auto *operatorRef = dyn_cast(expr->getFn())) { Printer << operatorRef->getDecl()->getBaseName(); } Printer << " "; visit(expr->getRHS()); } void PrintAST::visitCoerceExpr(CoerceExpr *expr) { } void PrintAST::visitOneWayExpr(OneWayExpr *expr) { } void PrintAST::visitClosureExpr(ClosureExpr *expr) { } void PrintAST::visitDeclRefExpr(DeclRefExpr *expr) { Printer << expr->getDecl()->getBaseName(); } void PrintAST::visitDotSelfExpr(DotSelfExpr *expr) { visit(expr->getSubExpr()); Printer << ".self"; } void PrintAST::visitErasureExpr(ErasureExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitKeyPathExpr(KeyPathExpr *expr) { } void PrintAST::visitForceTryExpr(ForceTryExpr *expr) { Printer << "try! "; visit(expr->getSubExpr()); } void PrintAST::visitSequenceExpr(SequenceExpr *expr) { } void PrintAST::visitSuperRefExpr(SuperRefExpr *expr) { } void PrintAST::visitMemberRefExpr(MemberRefExpr *expr) { visit(expr->getBase()); Printer << "."; Printer << expr->getMember().getDecl()->getName(); } void PrintAST::visitSubscriptExpr(SubscriptExpr *expr) { } void PrintAST::visitEnumIsCaseExpr(EnumIsCaseExpr *expr) { } void PrintAST::visitForceValueExpr(ForceValueExpr *expr) { } void PrintAST::visitKeyPathDotExpr(KeyPathDotExpr *expr) { } void PrintAST::visitAutoClosureExpr(AutoClosureExpr *expr) { visit(expr->getSingleExpressionBody()); } void PrintAST::visitCaptureListExpr(CaptureListExpr *expr) { } void PrintAST::visitDynamicTypeExpr(DynamicTypeExpr *expr) { } void PrintAST::visitOpaqueValueExpr(OpaqueValueExpr *expr) { } void PrintAST::visitOptionalTryExpr(OptionalTryExpr *expr) { } void PrintAST::visitPrefixUnaryExpr(PrefixUnaryExpr *expr) { } void PrintAST::visitBindOptionalExpr(BindOptionalExpr *expr) { } void PrintAST::visitBridgeToObjCExpr(BridgeToObjCExpr *expr) { } void PrintAST::visitObjCSelectorExpr(ObjCSelectorExpr *expr) { } void PrintAST::visitPostfixUnaryExpr(PostfixUnaryExpr *expr) { } void PrintAST::visitTupleElementExpr(TupleElementExpr *expr) { } void PrintAST::visitDerivedToBaseExpr(DerivedToBaseExpr *expr) { } void PrintAST::visitDotSyntaxCallExpr(DotSyntaxCallExpr *expr) { visit(expr->getBase()); Printer << "."; visit(expr->getFn()); } void PrintAST::visitObjectLiteralExpr(ObjectLiteralExpr *expr) { } void PrintAST::visitUnresolvedDotExpr(UnresolvedDotExpr *expr) { visit(expr->getBase()); Printer << "."; Printer << expr->getName().getBaseName(); } void PrintAST::visitArrayToPointerExpr(ArrayToPointerExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitBridgeFromObjCExpr(BridgeFromObjCExpr *expr) { } void PrintAST::visitCodeCompletionExpr(CodeCompletionExpr *expr) { } void PrintAST::visitInOutToPointerExpr(InOutToPointerExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitLinearFunctionExpr(LinearFunctionExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitDefaultArgumentExpr(DefaultArgumentExpr *expr) { } void PrintAST::visitLazyInitializerExpr(LazyInitializerExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitOpenExistentialExpr(OpenExistentialExpr *expr) { visit(expr->getExistentialValue()); visit(expr->getSubExpr()); } void PrintAST::visitStringToPointerExpr(StringToPointerExpr *expr) { } void PrintAST::visitVarargExpansionExpr(VarargExpansionExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitArchetypeToSuperExpr(ArchetypeToSuperExpr *expr) { } void PrintAST::visitDestructureTupleExpr(DestructureTupleExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitDynamicMemberRefExpr(DynamicMemberRefExpr *expr) { } void PrintAST::visitDynamicSubscriptExpr(DynamicSubscriptExpr *expr) { } void PrintAST::visitPointerToPointerExpr(PointerToPointerExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitUnresolvedMemberExpr(UnresolvedMemberExpr *expr) { } void PrintAST::visitDiscardAssignmentExpr(DiscardAssignmentExpr *expr) { Printer << "_"; } void PrintAST::visitEditorPlaceholderExpr(EditorPlaceholderExpr *expr) { } void PrintAST::visitForcedCheckedCastExpr(ForcedCheckedCastExpr *expr) { visit(expr->getSubExpr()); Printer << " as! "; printType(expr->getCastType()); } void PrintAST::visitConditionalCheckedCastExpr(ConditionalCheckedCastExpr *expr) { visit(expr->getSubExpr()); Printer << " as? "; printType(expr->getCastType()); } void PrintAST::visitOverloadedDeclRefExpr(OverloadedDeclRefExpr *expr) { } void PrintAST::visitUnresolvedDeclRefExpr(UnresolvedDeclRefExpr *expr) { } void PrintAST::visitUnresolvedPatternExpr(UnresolvedPatternExpr *expr) { } void PrintAST::visitAnyHashableErasureExpr(AnyHashableErasureExpr *expr) { } void PrintAST::visitConstructorRefCallExpr(ConstructorRefCallExpr *expr) { if (auto type = expr->getType()) { if (auto *funcType = type->getAs()) { printType(funcType->getResult()); } } } void PrintAST::visitFunctionConversionExpr(FunctionConversionExpr *expr) { } void PrintAST::visitInjectIntoOptionalExpr(InjectIntoOptionalExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitKeyPathApplicationExpr(KeyPathApplicationExpr *expr) { } void PrintAST::visitMetatypeConversionExpr(MetatypeConversionExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitOptionalEvaluationExpr(OptionalEvaluationExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitUnderlyingToOpaqueExpr(UnderlyingToOpaqueExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitUnevaluatedInstanceExpr(UnevaluatedInstanceExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitDotSyntaxBaseIgnoredExpr(DotSyntaxBaseIgnoredExpr *expr) { } void PrintAST::visitUnresolvedSpecializeExpr(UnresolvedSpecializeExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitClassMetatypeToObjectExpr(ClassMetatypeToObjectExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitAppliedPropertyWrapperExpr(AppliedPropertyWrapperExpr *expr) { } void PrintAST::visitDifferentiableFunctionExpr(DifferentiableFunctionExpr *expr) { visit(expr->getSubExpr()); } void PrintAST::visitMagicIdentifierLiteralExpr(MagicIdentifierLiteralExpr *expr) { } void PrintAST::visitForeignObjectConversionExpr(ForeignObjectConversionExpr *expr) { } void PrintAST::visitOtherConstructorDeclRefExpr(OtherConstructorDeclRefExpr *expr) { } void PrintAST::visitRebindSelfInConstructorExpr(RebindSelfInConstructorExpr *expr) { } void PrintAST::visitMakeTemporarilyEscapableExpr(MakeTemporarilyEscapableExpr *expr) { } void PrintAST::visitProtocolMetatypeToObjectExpr(ProtocolMetatypeToObjectExpr *expr) { } void PrintAST::visitUnresolvedTypeConversionExpr(UnresolvedTypeConversionExpr *expr) { } void PrintAST::visitConditionalBridgeFromObjCExpr(ConditionalBridgeFromObjCExpr *expr) { } void PrintAST::visitCovariantReturnConversionExpr(CovariantReturnConversionExpr *expr) { } void PrintAST::visitInterpolatedStringLiteralExpr(InterpolatedStringLiteralExpr *expr) { } void PrintAST::visitCollectionUpcastConversionExpr(CollectionUpcastConversionExpr *expr) { } void PrintAST::visitCovariantFunctionConversionExpr(CovariantFunctionConversionExpr *expr) { } void PrintAST::visitExistentialMetatypeToObjectExpr(ExistentialMetatypeToObjectExpr *expr) { } void PrintAST::visitUnresolvedMemberChainResultExpr(swift::UnresolvedMemberChainResultExpr *expr) { } void PrintAST::visitLinearFunctionExtractOriginalExpr(swift::LinearFunctionExtractOriginalExpr *expr) { } void PrintAST::visitLinearToDifferentiableFunctionExpr(swift::LinearToDifferentiableFunctionExpr *expr) { } void PrintAST::visitPropertyWrapperValuePlaceholderExpr(swift::PropertyWrapperValuePlaceholderExpr *expr) { } void PrintAST::visitDifferentiableFunctionExtractOriginalExpr(swift::DifferentiableFunctionExtractOriginalExpr *expr) { } void PrintAST::visitBraceStmt(BraceStmt *stmt) { printBraceStmt(stmt); } void PrintAST::visitReturnStmt(ReturnStmt *stmt) { if (!stmt->hasResult()) { if (auto *FD = dyn_cast(Current)) { if (auto *Body = FD->getBody()) { if (Body->getLastElement().dyn_cast() == stmt) { // Don't print empty return. return; } } } } Printer << tok::kw_return; if (stmt->hasResult()) { Printer << " "; visit(stmt->getResult()); } } void PrintAST::visitYieldStmt(YieldStmt *stmt) { Printer.printKeyword("yield", Options, " "); bool parens = (stmt->getYields().size() != 1 || stmt->getLParenLoc().isValid()); if (parens) Printer << "("; bool first = true; for (auto yield : stmt->getYields()) { if (first) { first = false; } else { Printer << ", "; } // FIXME: print expression. (void) yield; } if (parens) Printer << ")"; } void PrintAST::visitThrowStmt(ThrowStmt *stmt) { Printer << tok::kw_throw << " "; visit(stmt->getSubExpr()); } void PrintAST::visitPoundAssertStmt(PoundAssertStmt *stmt) { Printer << tok::pound_assert << " "; // FIXME: print expression. } void PrintAST::visitDeferStmt(DeferStmt *stmt) { Printer << tok::kw_defer << " "; visit(stmt->getBodyAsWritten()); } void PrintAST::visitIfStmt(IfStmt *stmt) { Printer << tok::kw_if << " "; printStmtCondition(stmt->getCond()); Printer << " "; visit(stmt->getThenStmt()); if (auto elseStmt = stmt->getElseStmt()) { Printer << " " << tok::kw_else << " "; visit(elseStmt); } } void PrintAST::visitGuardStmt(GuardStmt *stmt) { Printer << tok::kw_guard << " "; printStmtCondition(stmt->getCond()); Printer << " else "; visit(stmt->getBody()); } void PrintAST::visitWhileStmt(WhileStmt *stmt) { Printer << tok::kw_while << " "; printStmtCondition(stmt->getCond()); Printer << " "; visit(stmt->getBody()); } void PrintAST::visitRepeatWhileStmt(RepeatWhileStmt *stmt) { Printer << tok::kw_repeat << " "; visit(stmt->getBody()); Printer << " " << tok::kw_while << " "; visit(stmt->getCond()); } void PrintAST::printStmtCondition(StmtCondition condition) { interleave( condition, [&](StmtConditionElement &elt) { if (auto pattern = elt.getPatternOrNull()) { printPattern(pattern); auto initializer = elt.getInitializer(); if (initializer) { Printer << " = "; visit(initializer); } } else if (auto boolean = elt.getBooleanOrNull()) { visit(boolean); } }, [&] { Printer << ", "; }); } void PrintAST::visitDoStmt(DoStmt *stmt) { Printer << tok::kw_do << " "; visit(stmt->getBody()); } void PrintAST::visitDoCatchStmt(DoCatchStmt *stmt) { Printer << tok::kw_do << " "; visit(stmt->getBody()); for (auto clause : stmt->getCatches()) { visitCaseStmt(clause); } } void PrintAST::visitForEachStmt(ForEachStmt *stmt) { Printer << tok::kw_for << " "; printPattern(stmt->getPattern()); Printer << " " << tok::kw_in << " "; // FIXME: print container Printer << " "; visit(stmt->getBody()); } void PrintAST::visitBreakStmt(BreakStmt *stmt) { Printer << tok::kw_break; } void PrintAST::visitContinueStmt(ContinueStmt *stmt) { Printer << tok::kw_continue; } void PrintAST::visitFallthroughStmt(FallthroughStmt *stmt) { Printer << tok::kw_fallthrough; } void PrintAST::visitSwitchStmt(SwitchStmt *stmt) { Printer << tok::kw_switch << " "; visit(stmt->getSubjectExpr()); Printer << " {"; Printer.printNewline(); for (auto N : stmt->getRawCases()) { if (N.is()) visit(cast(N.get())); else visit(cast(N.get())); Printer.printNewline(); } indent(); Printer << "}"; } void PrintAST::visitCaseStmt(CaseStmt *CS) { if (CS->hasUnknownAttr()) Printer << "@unknown "; if (CS->isDefault()) { Printer << tok::kw_default; } else { auto PrintCaseLabelItem = [&](const CaseLabelItem &CLI) { if (auto *P = CLI.getPattern()) printPattern(P); if (CLI.getGuardExpr()) { Printer << " " << tok::kw_where << " "; // FIXME: print guard expr } }; Printer << tok::kw_case << " "; interleave(CS->getCaseLabelItems(), PrintCaseLabelItem, [&] { Printer << ", "; }); } Printer << ":"; Printer.printNewline(); printASTNodes((cast(CS->getBody())->getElements())); } void PrintAST::visitFailStmt(FailStmt *stmt) { Printer << tok::kw_return << " " << tok::kw_nil; } void Decl::print(raw_ostream &os) const { PrintOptions options; options.FunctionDefinitions = true; options.TypeDefinitions = true; options.VarInitializers = true; // FIXME: Move all places where SIL printing is happening to explicit options. // For example, see \c ProjectionPath::print. options.PreferTypeRepr = false; print(os, options); } void Decl::print(raw_ostream &OS, const PrintOptions &Opts) const { StreamPrinter Printer(OS); print(Printer, Opts); } bool Decl::print(ASTPrinter &Printer, const PrintOptions &Opts) const { PrintAST printer(Printer, Opts); return printer.visit(const_cast(this)); } void Decl::printInherited(ASTPrinter &Printer, const PrintOptions &Opts) const { PrintAST printer(Printer, Opts); printer.printInherited(this); } bool Decl::shouldPrintInContext(const PrintOptions &PO) const { // Skip getters/setters. They are part of the variable or subscript. if (isa(this)) return false; if (PO.ExplodePatternBindingDecls) { if (isa(this)) return true; if (isa(this)) return false; } else { // Try to preserve the PatternBindingDecl structure. // Skip stored variables, unless they came from a Clang module. // Stored variables in Swift source will be picked up by the // PatternBindingDecl. if (auto *VD = dyn_cast(this)) { if (!VD->hasClangNode() && VD->hasStorage()) return false; } // Skip pattern bindings that consist of just one variable with // interesting accessors. if (auto pbd = dyn_cast(this)) { if (pbd->getPatternList().size() == 1) { auto pattern = pbd->getPattern(0)->getSemanticsProvidingPattern(); if (auto named = dyn_cast(pattern)) { if (!named->getDecl()->hasStorage()) return false; } } } } if (isa(this)) { return PO.PrintIfConfig; } // Print everything else. return true; } void Pattern::print(llvm::raw_ostream &OS, const PrintOptions &Options) const { StreamPrinter StreamPrinter(OS); PrintAST Printer(StreamPrinter, Options); Printer.printPattern(this); } //===----------------------------------------------------------------------===// // Type Printing //===----------------------------------------------------------------------===// template void printCType(ASTContext &Ctx, ASTPrinter &Printer, ExtInfo &info) { auto *cml = Ctx.getClangModuleLoader(); SmallString<64> buf; llvm::raw_svector_ostream os(buf); info.getClangTypeInfo().printType(cml, os); Printer << ", cType: " << QuotedString(os.str()); } namespace { class TypePrinter : public TypeVisitor { using super = TypeVisitor; ASTPrinter &Printer; const PrintOptions &Options; Optional> VisibleClangModules; void printGenericArgs(ArrayRef Args) { if (Args.empty()) return; Printer << "<"; interleave(Args, [&](Type Arg) { visit(Arg); }, [&] { Printer << ", "; }); Printer << ">"; } /// Helper function for printing a type that is embedded within a larger type. /// /// This is necessary whenever the inner type may not normally be represented /// as a 'type-simple' production in the type grammar. void printWithParensIfNotSimple(Type T) { if (T.isNull()) { visit(T); return; } bool isSimple = isSimpleUnderPrintOptions(T); if (isSimple) { visit(T); } else { Printer << "("; visit(T); Printer << ")"; } } /// Determine whether the given type has a simple representation /// under the current print options. bool isSimpleUnderPrintOptions(Type T) { if (auto typealias = dyn_cast(T.getPointer())) { if (shouldDesugarTypeAliasType(typealias)) return isSimpleUnderPrintOptions(typealias->getSinglyDesugaredType()); } else if (auto opaque = dyn_cast(T.getPointer())) { if (opaque->isRoot()) { switch (Options.OpaqueReturnTypePrinting) { case PrintOptions::OpaqueReturnTypePrintingMode::StableReference: case PrintOptions::OpaqueReturnTypePrintingMode::Description: return true; case PrintOptions::OpaqueReturnTypePrintingMode::WithOpaqueKeyword: return opaque->getDecl()->hasExplicitGenericParams(); case PrintOptions::OpaqueReturnTypePrintingMode::WithoutOpaqueKeyword: return opaque->getDecl()->hasExplicitGenericParams() || isSimpleUnderPrintOptions(opaque->getExistentialType()); } llvm_unreachable("bad opaque-return-type printing mode"); } } else if (auto existential = dyn_cast(T.getPointer())) { if (!Options.PrintExplicitAny || !existential->shouldPrintWithAny()) return isSimpleUnderPrintOptions(existential->getConstraintType()); } else if (auto existential = dyn_cast(T.getPointer())) { if (!Options.PrintExplicitAny) return isSimpleUnderPrintOptions(existential->getInstanceType()); } return T->hasSimpleTypeRepr(); } /// Computes the map that is cached by `getVisibleClangModules()`. /// Do not call directly. llvm::DenseMap computeVisibleClangModules() { assert(Options.CurrentModule && "CurrentModule needs to be set to determine imported Clang modules"); llvm::DenseMap Result; // For the current module, consider both private and public imports. ModuleDecl::ImportFilter Filter = ModuleDecl::ImportFilterKind::Exported; Filter |= ModuleDecl::ImportFilterKind::Default; Filter |= ModuleDecl::ImportFilterKind::SPIAccessControl; SmallVector Imports; Options.CurrentModule->getImportedModules(Imports, Filter); SmallVector ModulesToProcess; for (const auto &Import : Imports) { ModulesToProcess.push_back(Import.importedModule); } SmallPtrSet Processed; while (!ModulesToProcess.empty()) { ModuleDecl *Mod = ModulesToProcess.back(); ModulesToProcess.pop_back(); if (!Processed.insert(Mod).second) continue; if (const clang::Module *ClangModule = Mod->findUnderlyingClangModule()) Result[ClangModule] = Mod; // For transitive imports, consider only public imports. Imports.clear(); Mod->getImportedModules(Imports, ModuleDecl::ImportFilterKind::Exported); for (const auto &Import : Imports) { ModulesToProcess.push_back(Import.importedModule); } } return Result; } /// Returns all Clang modules that are visible from `Options.CurrentModule`. /// This includes any modules that are imported transitively through public /// (`@_exported`) imports. /// /// The returned map associates each visible Clang module with the /// corresponding Swift module. const llvm::DenseMap & getVisibleClangModules() { if (!VisibleClangModules) { VisibleClangModules = computeVisibleClangModules(); } return *VisibleClangModules; } template void printModuleContext(T *Ty) { FileUnit *File = cast(Ty->getDecl()->getModuleScopeContext()); ModuleDecl *Mod = File->getParentModule(); StringRef ExportedModuleName = File->getExportedModuleName(); // Clang declarations need special treatment: Multiple Clang modules can // contain the same declarations from a textually included header, but not // all of these modules may be visible. We therefore need to make sure we // choose a module that is visible from the current module. This is possible // only if we know what the current module is. const clang::Decl *ClangDecl = Ty->getDecl()->getClangDecl(); if (ClangDecl && Options.CurrentModule) { for (auto *Redecl : ClangDecl->redecls()) { auto *owningModule = Redecl->getOwningModule(); if (!owningModule) continue; clang::Module *ClangModule = owningModule->getTopLevelModule(); if (!ClangModule) continue; if (ModuleDecl *VisibleModule = getVisibleClangModules().lookup(ClangModule)) { Mod = VisibleModule; ExportedModuleName = ClangModule->ExportAsModule; break; } } } if (Options.MapCrossImportOverlaysToDeclaringModule) { if (ModuleDecl *Declaring = Mod->getDeclaringModuleIfCrossImportOverlay()) Mod = Declaring; } // Should use the module real (binary) name here and everywhere else the // module is printed in case module aliasing is used (see -module-alias) Identifier Name = Mod->getRealName(); if (Options.UseExportedModuleNames && !ExportedModuleName.empty()) { Name = Mod->getASTContext().getIdentifier(ExportedModuleName); } if (Options.UseOriginallyDefinedInModuleNames) { Decl *D = Ty->getDecl(); for (auto attr: D->getAttrs().getAttributes()) { Name = Mod->getASTContext() .getIdentifier(const_cast(attr) ->OriginalModuleName); break; } } Printer.printModuleRef(Mod, Name); Printer << "."; } template void printTypeDeclName( T *Ty, PrintNameContext NameContext = PrintNameContext::Normal) { TypeDecl *TD = Ty->getDecl(); Printer.printTypeRef(Ty, TD, TD->getName(), NameContext); } // FIXME: we should have a callback that would tell us // whether it's kosher to print a module name or not bool isLLDBExpressionModule(ModuleDecl *M) { if (!M) return false; return M->getRealName().str().startswith(LLDB_EXPRESSIONS_MODULE_NAME_PREFIX); } bool shouldPrintFullyQualified(TypeBase *T) { if (Options.FullyQualifiedTypes) return true; Decl *D; if (auto *TAT = dyn_cast(T)) D = TAT->getDecl(); else D = T->getAnyGeneric(); // If we cannot find the declaration, be extra careful and print // the type qualified. if (!D) return true; ModuleDecl *M = D->getDeclContext()->getParentModule(); if (M->isBuiltinModule()) return true; if (!Options.FullyQualifiedTypesIfAmbiguous) return false; if (Options.CurrentModule && M == Options.CurrentModule) { return false; } // Don't print qualifiers for types from the standard library. if (M->isStdlibModule() || M->getRealName() == M->getASTContext().Id_ObjectiveC || M->isSystemModule() || isLLDBExpressionModule(M)) return false; // Don't print qualifiers for imported types. if (!Options.QualifyImportedTypes) for (auto File : M->getFiles()) { if (File->getKind() == FileUnitKind::ClangModule || File->getKind() == FileUnitKind::DWARFModule) return false; } return true; } public: TypePrinter(ASTPrinter &Printer, const PrintOptions &PO) : Printer(Printer), Options(PO) {} template void printQualifiedType(T *Ty) { PrintNameContext NameContext = PrintNameContext::Normal; // If we printed a parent type or a module qualification, let the printer // know we're printing a type member so it escapes `Type` and `Protocol`. if (auto parent = Ty->getParent()) { visitParentType(parent); NameContext = PrintNameContext::TypeMember; } else if (shouldPrintFullyQualified(Ty)) { printModuleContext(Ty); NameContext = PrintNameContext::TypeMember; } printTypeDeclName(Ty, NameContext); } void visit(Type T) { #if SWIFT_BUILD_ONLY_SYNTAXPARSERLIB return; // not needed for the parser library. #endif Printer.printTypePre(TypeLoc::withoutLoc(T)); SWIFT_DEFER { Printer.printTypePost(TypeLoc::withoutLoc(T)); }; super::visit(T); } void visitErrorType(ErrorType *T) { if (auto originalType = T->getOriginalType()) { if (Options.PrintInSILBody) Printer << "@error_type "; visit(originalType); } else Printer << "<>"; } void visitUnresolvedType(UnresolvedType *T) { if (Options.PrintTypesForDebugging) Printer << "<>"; else Printer << "_"; } void visitPlaceholderType(PlaceholderType *T) { if (Options.PrintTypesForDebugging) { Printer << "<getOriginator(); if (auto *typeVar = originator.dyn_cast()) { visit(typeVar); } else if (auto *VD = originator.dyn_cast()) { Printer << "decl = "; Printer << VD->getName(); } else if (auto *EE = originator.dyn_cast()) { Printer << "error_expr"; } else if (auto *DMT = originator.dyn_cast()) { visit(DMT); } else { Printer << "placeholder_type_repr"; } Printer << ">>"; } else { Printer << "<>"; } } #ifdef ASTPRINTER_HANDLE_BUILTINTYPE #error "ASTPRINTER_HANDLE_BUILTINTYPE should not be defined?!" #endif #define ASTPRINTER_PRINT_BUILTINTYPE(NAME) \ void visit##NAME(NAME *T) { \ SmallString<32> buffer; \ T->getTypeName(buffer); \ Printer << buffer; \ } ASTPRINTER_PRINT_BUILTINTYPE(BuiltinRawPointerType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinRawUnsafeContinuationType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinJobType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinExecutorType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinDefaultActorStorageType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinNativeObjectType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinBridgeObjectType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinUnsafeValueBufferType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinIntegerLiteralType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinVectorType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinIntegerType) ASTPRINTER_PRINT_BUILTINTYPE(BuiltinFloatType) #undef ASTPRINTER_PRINT_BUILTINTYPE void visitSILTokenType(SILTokenType *T) { Printer << BUILTIN_TYPE_NAME_SILTOKEN; } bool shouldDesugarTypeAliasType(TypeAliasType *T) { return Options.PrintForSIL || Options.PrintTypeAliasUnderlyingType; } void visitTypeAliasType(TypeAliasType *T) { if (shouldDesugarTypeAliasType(T)) { visit(T->getSinglyDesugaredType()); return; } printQualifiedType(T); printGenericArgs(T->getDirectGenericArgs()); } void visitParenType(ParenType *T) { Printer << "("; printParameterFlags(Printer, Options, T->getParameterFlags(), /*escaping*/ false); visit(T->getUnderlyingType()->getInOutObjectType()); Printer << ")"; } void visitPackType(PackType *T) { Printer << "("; auto Fields = T->getElementTypes(); for (unsigned i = 0, e = Fields.size(); i != e; ++i) { if (i) Printer << ", "; Type EltType = Fields[i]; visit(EltType); } Printer << ")"; } void visitPackExpansionType(PackExpansionType *T) { Printer << "("; visit(T->getPatternType()); Printer << "..." << ")"; } void visitTupleType(TupleType *T) { Printer.callPrintStructurePre(PrintStructureKind::TupleType); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::TupleType); }; Printer << "("; auto Fields = T->getElements(); for (unsigned i = 0, e = Fields.size(); i != e; ++i) { if (i) Printer << ", "; const TupleTypeElt &TD = Fields[i]; Type EltType = TD.getRawType(); Printer.callPrintStructurePre(PrintStructureKind::TupleElement); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::TupleElement); }; if (TD.hasName()) { Printer.printName(TD.getName(), PrintNameContext::TupleElement); Printer << ": "; } if (TD.isVararg()) { visit(TD.getVarargBaseTy()); Printer << "..."; } else { printParameterFlags(Printer, Options, TD.getParameterFlags(), /*escaping*/ false); visit(EltType); } } Printer << ")"; } void visitUnboundGenericType(UnboundGenericType *T) { printQualifiedType(T); } void visitBoundGenericType(BoundGenericType *T) { if (Options.SynthesizeSugarOnTypes) { if (T->isArray()) { Printer << "["; visit(T->getGenericArgs()[0]); Printer << "]"; return; } if (T->isDictionary()) { Printer << "["; visit(T->getGenericArgs()[0]); Printer << " : "; visit(T->getGenericArgs()[1]); Printer << "]"; return; } if (T->isOptional()) { printWithParensIfNotSimple(T->getGenericArgs()[0]); Printer << "?"; return; } } printQualifiedType(T); printGenericArgs(T->getGenericArgs()); } void visitParentType(Type T) { /// Don't print the parent type if it's being printed in that type context. if (Options.TransformContext) { if (auto currentType = Options.TransformContext->getBaseType()) { auto printingType = T; if (currentType->hasArchetype()) currentType = currentType->mapTypeOutOfContext(); if (auto errorTy = printingType->getAs()) if (auto origTy = errorTy->getOriginalType()) printingType = origTy; if (printingType->hasArchetype()) printingType = printingType->mapTypeOutOfContext(); if (currentType->isEqual(printingType)) return; } } PrintOptions innerOptions = Options; innerOptions.SynthesizeSugarOnTypes = false; if (auto sugarType = dyn_cast(T.getPointer())) T = sugarType->getImplementationType(); TypePrinter(Printer, innerOptions).printWithParensIfNotSimple(T); Printer << "."; } void visitEnumType(EnumType *T) { printQualifiedType(T); } void visitStructType(StructType *T) { printQualifiedType(T); } void visitClassType(ClassType *T) { printQualifiedType(T); } void visitAnyMetatypeType(AnyMetatypeType *T) { if (T->hasRepresentation()) { switch (T->getRepresentation()) { case MetatypeRepresentation::Thin: Printer << "@thin "; break; case MetatypeRepresentation::Thick: Printer << "@thick "; break; case MetatypeRepresentation::ObjC: Printer << "@objc_metatype "; break; } } Type instanceType = T->getInstanceType(); if (Options.PrintExplicitAny && T->is()) { Printer << "any "; // FIXME: We need to replace nested existential metatypes so that // we don't print duplicate 'any'. This will be unnecessary once // ExistentialMetatypeType is split into ExistentialType(MetatypeType). printWithParensIfNotSimple(instanceType.transform([](Type type) -> Type { if (auto existential = type->getAs()) return MetatypeType::get(existential->getInstanceType()); return type; })); } else if (T->is() && instanceType->is()) { // The 'any' keyword is needed to distinguish between existential // metatypes and singleton metatypes. However, 'any' usually isn't // printed for Any and AnyObject, because it's unnecessary to write // 'any' with these specific constraints. Force printing with 'any' // for the existential instance type in this case. instanceType->getAs()->forcePrintWithAny([&](Type ty) { printWithParensIfNotSimple(ty); }); } else { printWithParensIfNotSimple(instanceType); } // We spell normal metatypes of existential types as .Protocol. if (isa(T) && T->getInstanceType()->isAnyExistentialType() && !Options.PrintExplicitAny) { Printer << ".Protocol"; } else { Printer << ".Type"; } } void visitModuleType(ModuleType *T) { Printer << "module<"; // Should print the module real name in case module aliasing is // used (see -module-alias), since that's the actual binary name. Printer.printModuleRef(T->getModule(), T->getModule()->getRealName()); Printer << ">"; } void visitDynamicSelfType(DynamicSelfType *T) { if (Options.PrintInSILBody) { Printer << "@dynamic_self "; visit(T->getSelfType()); return; } // Try to print as a reference to the static type so that we will get a USR, // in cursor info. auto staticSelfT = T->getSelfType(); if (auto *NTD = staticSelfT->getAnyNominal()) { if (isa(NTD)) { auto Name = T->getASTContext().Id_Self; Printer.printTypeRef(T, NTD, Name); return; } } visit(staticSelfT); } void printFunctionExtInfo(AnyFunctionType *fnType) { if (!fnType->hasExtInfo()) { Printer << "@_NO_EXTINFO "; return; } auto &ctx = fnType->getASTContext(); auto info = fnType->getExtInfo(); if (Options.SkipAttributes) return; if (!Options.excludeAttrKind(TAK_differentiable)) { switch (info.getDifferentiabilityKind()) { case DifferentiabilityKind::Normal: Printer << "@differentiable "; break; case DifferentiabilityKind::Linear: Printer << "@differentiable(_linear) "; break; case DifferentiabilityKind::Forward: Printer << "@differentiable(_forward) "; break; case DifferentiabilityKind::Reverse: Printer << "@differentiable(reverse) "; break; case DifferentiabilityKind::NonDifferentiable: break; } } if (Type globalActor = info.getGlobalActor()) { Printer << "@"; visit(globalActor); Printer << " "; } if (!Options.excludeAttrKind(TAK_Sendable) && info.isSendable()) { Printer.printSimpleAttr("@Sendable") << " "; } SmallString<64> buf; switch (Options.PrintFunctionRepresentationAttrs) { case PrintOptions::FunctionRepresentationMode::None: return; case PrintOptions::FunctionRepresentationMode::Full: case PrintOptions::FunctionRepresentationMode::NameOnly: if (Options.excludeAttrKind(TAK_convention) || info.getSILRepresentation() == SILFunctionType::Representation::Thick) return; bool printClangType = Options.PrintFunctionRepresentationAttrs == PrintOptions::FunctionRepresentationMode::Full; Printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute); Printer.printAttrName("@convention"); Printer << "("; // TODO: coalesce into a single convention attribute. switch (info.getSILRepresentation()) { case SILFunctionType::Representation::Thick: llvm_unreachable("thick is not printed"); case SILFunctionType::Representation::Thin: Printer << "thin"; break; case SILFunctionType::Representation::Block: Printer << "block"; if (printClangType && fnType->hasNonDerivableClangType()) printCType(ctx, Printer, info); break; case SILFunctionType::Representation::CFunctionPointer: Printer << "c"; if (printClangType && fnType->hasNonDerivableClangType()) printCType(ctx, Printer, info); break; case SILFunctionType::Representation::Method: Printer << "method"; break; case SILFunctionType::Representation::CXXMethod: Printer << "cxx_method"; break; case SILFunctionType::Representation::ObjCMethod: Printer << "objc_method"; break; case SILFunctionType::Representation::WitnessMethod: Printer << "witness_method"; break; case SILFunctionType::Representation::Closure: Printer << "closure"; break; } Printer << ")"; Printer.printStructurePost(PrintStructureKind::BuiltinAttribute); Printer << " "; } } void printFunctionExtInfo(SILFunctionType *fnType) { auto &Ctx = fnType->getASTContext(); auto info = fnType->getExtInfo(); auto witnessMethodConformance = fnType->getWitnessMethodConformanceOrInvalid(); if (Options.SkipAttributes) return; if (!Options.excludeAttrKind(TAK_differentiable)) { switch (info.getDifferentiabilityKind()) { case DifferentiabilityKind::Normal: Printer << "@differentiable "; break; case DifferentiabilityKind::Linear: Printer << "@differentiable(_linear) "; break; case DifferentiabilityKind::Forward: Printer << "@differentiable(_forward) "; break; case DifferentiabilityKind::Reverse: Printer << "@differentiable(reverse) "; break; case DifferentiabilityKind::NonDifferentiable: break; } } SmallString<64> buf; switch (Options.PrintFunctionRepresentationAttrs) { case PrintOptions::FunctionRepresentationMode::None: break; case PrintOptions::FunctionRepresentationMode::NameOnly: case PrintOptions::FunctionRepresentationMode::Full: if (Options.excludeAttrKind(TAK_convention) || info.getRepresentation() == SILFunctionType::Representation::Thick) break; bool printClangType = Options.PrintFunctionRepresentationAttrs == PrintOptions::FunctionRepresentationMode::Full; Printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute); Printer.printAttrName("@convention"); Printer << "("; switch (info.getRepresentation()) { case SILFunctionType::Representation::Thick: llvm_unreachable("thick is not printed"); case SILFunctionType::Representation::Thin: Printer << "thin"; break; case SILFunctionType::Representation::Block: Printer << "block"; if (printClangType && fnType->hasNonDerivableClangType()) printCType(Ctx, Printer, info); break; case SILFunctionType::Representation::CFunctionPointer: Printer << "c"; if (printClangType && fnType->hasNonDerivableClangType()) printCType(Ctx, Printer, info); break; case SILFunctionType::Representation::Method: Printer << "method"; break; case SILFunctionType::Representation::CXXMethod: Printer << "cxx_method"; break; case SILFunctionType::Representation::ObjCMethod: Printer << "objc_method"; break; case SILFunctionType::Representation::WitnessMethod: Printer << "witness_method: "; printTypeDeclName( witnessMethodConformance.getRequirement()->getDeclaredType() ->castTo()); break; case SILFunctionType::Representation::Closure: Printer << "closure"; break; } Printer << ")"; Printer.printStructurePost(PrintStructureKind::BuiltinAttribute); Printer << " "; } if (info.isPseudogeneric()) { Printer.printSimpleAttr("@pseudogeneric") << " "; } if (info.isNoEscape()) { Printer.printSimpleAttr("@noescape") << " "; } if (info.isSendable()) { Printer.printSimpleAttr("@Sendable") << " "; } if (info.isAsync()) { Printer.printSimpleAttr("@async") << " "; } } void visitAnyFunctionTypeParams(ArrayRef Params, bool printLabels) { Printer << "("; for (unsigned i = 0, e = Params.size(); i != e; ++i) { if (i) Printer << ", "; const AnyFunctionType::Param &Param = Params[i]; Printer.callPrintStructurePre(PrintStructureKind::FunctionParameter); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::FunctionParameter); }; if ((Options.AlwaysTryPrintParameterLabels || printLabels) && Param.hasLabel()) { // Label printing was requested and we have an external label. Print it // and omit the internal label. Printer.printName(Param.getLabel(), PrintNameContext::FunctionParameterExternal); Printer << ": "; } else if (Options.AlwaysTryPrintParameterLabels && Param.hasInternalLabel() && !Param.getInternalLabel().hasDollarPrefix()) { // We didn't have an external parameter label but were requested to // always try and print parameter labels. // If the internal label is a valid internal parameter label (does not // start with '$'), print the internal label. If we have neither an // external nor a printable internal label, only print the type. Printer << "_ "; Printer.printName(Param.getInternalLabel(), PrintNameContext::FunctionParameterLocal); Printer << ": "; } auto type = Param.getPlainType(); if (Param.isVariadic()) { visit(type); Printer << "..."; } else { printParameterFlags(Printer, Options, Param.getParameterFlags(), isEscaping(type)); visit(type); } } Printer << ")"; } void visitFunctionType(FunctionType *T) { Printer.callPrintStructurePre(PrintStructureKind::FunctionType); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::FunctionType); }; printFunctionExtInfo(T); // If we're stripping argument labels from types, do it when printing. visitAnyFunctionTypeParams(T->getParams(), /*printLabels*/false); if (T->hasExtInfo()) { if (T->isAsync()) { Printer << " "; Printer.printKeyword("async", Options); } if (T->isThrowing()) Printer << " " << tok::kw_throws; } Printer << " -> "; Printer.callPrintStructurePre(PrintStructureKind::FunctionReturnType); T->getResult().print(Printer, Options); Printer.printStructurePost(PrintStructureKind::FunctionReturnType); } void printGenericSignature(GenericSignature genericSig, unsigned flags) { PrintAST(Printer, Options).printGenericSignature(genericSig, flags); } void printSubstitutions(SubstitutionMap subs) { Printer << " <"; interleave(subs.getReplacementTypes(), [&](Type type) { visit(type); }, [&]{ Printer << ", "; }); Printer << ">"; } void visitGenericFunctionType(GenericFunctionType *T) { Printer.callPrintStructurePre(PrintStructureKind::FunctionType); SWIFT_DEFER { Printer.printStructurePost(PrintStructureKind::FunctionType); }; printFunctionExtInfo(T); printGenericSignature(T->getGenericSignature(), PrintAST::PrintParams | PrintAST::PrintRequirements); Printer << " "; visitAnyFunctionTypeParams(T->getParams(), /*printLabels*/true); if (T->hasExtInfo()) { if (T->isAsync()) { Printer << " "; Printer.printKeyword("async", Options); } if (T->isThrowing()) Printer << " " << tok::kw_throws; } Printer << " -> "; Printer.callPrintStructurePre(PrintStructureKind::FunctionReturnType); T->getResult().print(Printer, Options); Printer.printStructurePost(PrintStructureKind::FunctionReturnType); } void printSILCoroutineKind(SILCoroutineKind kind) { switch (kind) { case SILCoroutineKind::None: return; case SILCoroutineKind::YieldOnce: Printer << "@yield_once "; return; case SILCoroutineKind::YieldMany: Printer << "@yield_many "; return; } llvm_unreachable("bad convention"); } void printSILAsyncAttr(bool isAsync) { if (isAsync) { Printer << "@async "; } } void printCalleeConvention(ParameterConvention conv) { switch (conv) { case ParameterConvention::Direct_Unowned: return; case ParameterConvention::Direct_Owned: Printer << "@callee_owned "; return; case ParameterConvention::Direct_Guaranteed: Printer << "@callee_guaranteed "; return; case ParameterConvention::Indirect_In: case ParameterConvention::Indirect_In_Constant: case ParameterConvention::Indirect_Inout: case ParameterConvention::Indirect_InoutAliasable: case ParameterConvention::Indirect_In_Guaranteed: llvm_unreachable("callee convention cannot be indirect"); } llvm_unreachable("bad convention"); } void visitSILFunctionType(SILFunctionType *T) { printSILCoroutineKind(T->getCoroutineKind()); printFunctionExtInfo(T); printCalleeConvention(T->getCalleeConvention()); if (auto sig = T->getInvocationGenericSignature()) { printGenericSignature(sig, PrintAST::PrintParams | PrintAST::PrintRequirements); Printer << " "; } // If this is a substituted function type, then its generic signature is // independent of the enclosing context, and defines the parameters active // in the interface params and results. Unsubstituted types use the existing // environment, which may be a sil decl's generic environment. // // Yeah, this is fiddly. In the end, we probably want all decls to have // substituted types in terms of a generic signature declared on the decl, // which would make this logic more uniform. TypePrinter *sub = this; Optional subBuffer; PrintOptions subOptions = Options; if (auto substitutions = T->getPatternSubstitutions()) { subOptions.GenericSig = nullptr; subBuffer.emplace(Printer, subOptions); sub = &*subBuffer; sub->Printer << "@substituted "; sub->printGenericSignature(substitutions.getGenericSignature(), PrintAST::PrintParams | PrintAST::PrintRequirements); sub->Printer << " "; } // Capture list used here to ensure we don't print anything using `this` // printer, but only the sub-Printer. [T, sub, &subOptions] { sub->Printer << "("; bool first = true; for (auto param : T->getParameters()) { sub->Printer.printSeparator(first, ", "); param.print(sub->Printer, subOptions); } sub->Printer << ") -> "; bool parenthesizeResults = mustParenthesizeResults(T); if (parenthesizeResults) sub->Printer << "("; first = true; for (auto yield : T->getYields()) { sub->Printer.printSeparator(first, ", "); sub->Printer << "@yields "; yield.print(sub->Printer, subOptions); } for (auto result : T->getResults()) { sub->Printer.printSeparator(first, ", "); result.print(sub->Printer, subOptions); } if (T->hasErrorResult()) { // The error result is implicitly @owned; don't print that. assert(T->getErrorResult().getConvention() == ResultConvention::Owned); sub->Printer.printSeparator(first, ", "); sub->Printer << "@error "; T->getErrorResult().getInterfaceType().print(sub->Printer, subOptions); } if (parenthesizeResults) sub->Printer << ")"; }(); // Both the pattern and invocation substitution types are always in // terms of the outer environment. But this wouldn't necessarily be // true with higher-rank polymorphism. if (auto substitutions = T->getPatternSubstitutions()) { Printer << " for"; printSubstitutions(substitutions); } if (auto substitutions = T->getInvocationSubstitutions()) { Printer << " for"; printSubstitutions(substitutions); } } static bool mustParenthesizeResults(SILFunctionType *T) { // If we don't have exactly one result, we must parenthesize. unsigned totalResults = T->getNumYields() + T->getNumResults() + unsigned(T->hasErrorResult()); if (totalResults != 1) return true; // If we have substitutions, we must parenthesize if the single // result is a function type. if (!T->hasPatternSubstitutions() && !T->hasInvocationSubstitutions()) return false; if (T->getNumResults() == 1) return isa(T->getResults()[0].getInterfaceType()); if (T->getNumYields() == 1) return isa(T->getYields()[0].getInterfaceType()); return isa(T->getErrorResult().getInterfaceType()); } void visitSILBlockStorageType(SILBlockStorageType *T) { Printer << "@block_storage "; printWithParensIfNotSimple(T->getCaptureType()); } void visitSILBoxType(SILBoxType *T) { // Print attributes. if (T->getLayout()->capturesGenericEnvironment()) { Printer << "@captures_generics "; } { // A box layout has its own independent generic environment. Don't try // to print it with the environment's generic params. PrintOptions subOptions = Options; subOptions.GenericSig = nullptr; TypePrinter sub(Printer, subOptions); // Capture list used here to ensure we don't print anything using `this` // printer, but only the sub-Printer. [&sub, T]{ if (auto sig = T->getLayout()->getGenericSignature()) { sub.printGenericSignature(sig, PrintAST::PrintParams | PrintAST::PrintRequirements); sub.Printer << " "; } sub.Printer << "{"; interleave(T->getLayout()->getFields(), [&](const SILField &field) { sub.Printer << (field.isMutable() ? " var " : " let "); sub.visit(field.getLoweredType()); }, [&]{ sub.Printer << ","; }); sub.Printer << " }"; }(); } // The arguments to the layout, if any, do come from the outer environment. if (auto subMap = T->getSubstitutions()) { printSubstitutions(subMap); } } void visitSILMoveOnlyType(SILMoveOnlyType *T) { Printer << "@moveOnly "; printWithParensIfNotSimple(T->getInnerType()); } void visitArraySliceType(ArraySliceType *T) { Printer << "["; visit(T->getBaseType()); Printer << "]"; } void visitDictionaryType(DictionaryType *T) { Printer << "["; visit(T->getKeyType()); Printer << " : "; visit(T->getValueType()); Printer << "]"; } void visitOptionalType(OptionalType *T) { auto printAsIUO = Options.PrintOptionalAsImplicitlyUnwrapped; // Printing optionals with a trailing '!' applies only to // top-level optionals, not to any nested within. const_cast(Options).PrintOptionalAsImplicitlyUnwrapped = false; printWithParensIfNotSimple(T->getBaseType()); const_cast(Options).PrintOptionalAsImplicitlyUnwrapped = printAsIUO; if (printAsIUO) Printer << "!"; else Printer << "?"; } void visitVariadicSequenceType(VariadicSequenceType *T) { if (Options.PrintForSIL) { Printer << "["; visit(T->getBaseType()); Printer << "]"; } else { visit(T->getBaseType()); Printer << "..."; } } void visitProtocolType(ProtocolType *T) { printQualifiedType(T); } void visitProtocolCompositionType(ProtocolCompositionType *T) { if (T->getMembers().empty()) { if (T->hasExplicitAnyObject()) Printer << "AnyObject"; else Printer.printKeyword("Any", Options); } else { interleave(T->getMembers(), [&](Type Ty) { visit(Ty); }, [&] { Printer << " & "; }); if (T->hasExplicitAnyObject()) Printer << " & AnyObject"; } } void visitParameterizedProtocolType(ParameterizedProtocolType *T) { visit(T->getBaseType()); Printer << "<"; interleave(T->getArgs(), [&](Type Ty) { visit(Ty); }, [&] { Printer << ", "; }); Printer << ">"; } void visitExistentialType(ExistentialType *T) { if (Options.PrintExplicitAny && T->shouldPrintWithAny()) Printer << "any "; // FIXME: The desugared type is used here only to support // existential types with protocol typealiases in Swift // interfaces. Verifying that the underlying type of a // protocol typealias is a constriant type is fundamentally // circular, so the desugared type should be written in source. if (Options.DesugarExistentialConstraint && !T->isAnyObject()) { visit(T->getConstraintType()->getDesugaredType()); } else { visit(T->getConstraintType()); } } void visitLValueType(LValueType *T) { Printer << "@lvalue "; visit(T->getObjectType()); } void visitInOutType(InOutType *T) { Printer << tok::kw_inout << " "; visit(T->getObjectType()); } void visitOpenedArchetypeType(OpenedArchetypeType *T) { if (auto parent = T->getParent()) { printArchetypeCommon(T); return; } if (Options.PrintForSIL) Printer << "@opened(\"" << T->getOpenedExistentialID() << "\") "; visit(T->getExistentialType()); } void printDependentMember(DependentMemberType *T) { if (auto *const Assoc = T->getAssocType()) { if (Options.ProtocolQualifiedDependentMemberTypes) { Printer << "["; Printer.printName(Assoc->getProtocol()->getName()); Printer << "]"; } Printer.printTypeRef(T, Assoc, T->getName()); } else { Printer.printName(T->getName()); } } void printArchetypeCommon(ArchetypeType *T) { if (Options.AlternativeTypeNames) { auto found = Options.AlternativeTypeNames->find(T->getCanonicalType()); if (found != Options.AlternativeTypeNames->end()) { Printer << found->second.str(); return; } } auto interfaceType = T->getInterfaceType(); if (auto *dependentMember = interfaceType->getAs()) { visitParentType(T->getParent()); printDependentMember(dependentMember); } else { visit(interfaceType); } } void visitPrimaryArchetypeType(PrimaryArchetypeType *T) { printArchetypeCommon(T); } void visitOpaqueTypeArchetypeType(OpaqueTypeArchetypeType *T) { if (auto parent = T->getParent()) { printArchetypeCommon(T); return; } // Try to print a named opaque type. auto printNamedOpaque = [&] { unsigned ordinal = T->getInterfaceType()->castTo()->getIndex(); if (auto genericParam = T->getDecl()->getExplicitGenericParam(ordinal)) { visit(genericParam->getDeclaredInterfaceType()); return true; } return false; }; switch (Options.OpaqueReturnTypePrinting) { case PrintOptions::OpaqueReturnTypePrintingMode::WithOpaqueKeyword: if (printNamedOpaque()) return; Printer.printKeyword("some", Options, /*Suffix=*/" "); LLVM_FALLTHROUGH; case PrintOptions::OpaqueReturnTypePrintingMode::WithoutOpaqueKeyword: { if (printNamedOpaque()) return; auto constraint = T->getExistentialType(); if (auto existential = constraint->getAs()) constraint = existential->getConstraintType(); visit(constraint); return; } case PrintOptions::OpaqueReturnTypePrintingMode::StableReference: { // Print the source of the opaque return type as a mangled name. // We'll use type reconstruction while parsing the attribute to // turn this back into a reference to the naming decl for the opaque // type. Printer << "@_opaqueReturnTypeOf("; OpaqueTypeDecl *decl = T->getDecl(); Printer.printEscapedStringLiteral( decl->getOpaqueReturnTypeIdentifier().str()); Printer << ", " << T->getInterfaceType() ->castTo() ->getIndex(); // The identifier after the closing parenthesis is irrelevant and can be // anything. It just needs to be there for the @_opaqueReturnTypeOf // attribute to apply to, but the attribute alone references the opaque // type. Printer << ") __"; printGenericArgs(T->getSubstitutions().getReplacementTypes()); return; } case PrintOptions::OpaqueReturnTypePrintingMode::Description: { // TODO(opaque): present opaque types with user-facing syntax. we should // probably print this as `some P` and record the fact that we printed that // so that diagnostics can add followup notes. Printer << "(return type of " << T->getDecl()->getNamingDecl()->printRef(); Printer << ')'; if (!T->getSubstitutions().empty()) { Printer << '<'; auto replacements = T->getSubstitutions().getReplacementTypes(); llvm::interleave( replacements.begin(), replacements.end(), [&](Type t) { visit(t); }, [&] { Printer << ", "; }); Printer << '>'; } return; } } } void visitSequenceArchetypeType(SequenceArchetypeType *T) { printArchetypeCommon(T); } void visitGenericTypeParamType(GenericTypeParamType *T) { auto decl = T->getDecl(); if (!decl) { // If we have an alternate name for this type, use it. if (Options.AlternativeTypeNames) { auto found = Options.AlternativeTypeNames->find(T->getCanonicalType()); if (found != Options.AlternativeTypeNames->end()) { Printer << found->second.str(); return; } } // When printing SIL types, use a generic signature to map them from // canonical types to sugared types. if (Options.GenericSig) T = Options.GenericSig->getSugaredType(T); } // Print opaque types as "some ..." if (decl && decl->isOpaqueType()) { // If we have and should print based on the type representation, do so. if (auto opaqueRepr = decl->getOpaqueTypeRepr()) { if (willUseTypeReprPrinting(opaqueRepr, Type(), Options)) { opaqueRepr->print(Printer, Options); return; } } // Print based on the type. Printer << "some "; if (!decl->getConformingProtocols().empty()) { llvm::interleave(decl->getConformingProtocols(), Printer, [&](ProtocolDecl *proto){ if (auto printType = proto->getDeclaredType()) printType->print(Printer, Options); else Printer << proto->getNameStr(); }, " & "); } else { Printer << "Any"; } return; } const auto Name = T->getName(); if (Name.empty()) { Printer << ""; } else if (auto *Decl = T->getDecl()) { Printer.printTypeRef(T, Decl, Name); } else { Printer.printName(Name); } } void visitDependentMemberType(DependentMemberType *T) { visitParentType(T->getBase()); printDependentMember(T); } #define REF_STORAGE(Name, name, ...) \ void visit##Name##StorageType(Name##StorageType *T) { \ if (Options.PrintStorageRepresentationAttrs) \ Printer << "@sil_" #name " "; \ visit(T->getReferentType()); \ } #include "swift/AST/ReferenceStorage.def" void visitTypeVariableType(TypeVariableType *T) { if (Options.PrintTypesForDebugging) { Printer << "$T" << T->getID(); return; } Printer << "_"; } }; } // unnamed namespace void Type::print(raw_ostream &OS, const PrintOptions &PO) const { StreamPrinter Printer(OS); print(Printer, PO); } void Type::print(ASTPrinter &Printer, const PrintOptions &PO) const { if (isNull()) { if (!PO.AllowNullTypes) { // Use report_fatal_error instead of assert to trap in release builds too. llvm::report_fatal_error("Cannot pretty-print a null type"); } Printer << ""; return; } TypePrinter(Printer, PO).visit(*this); } void AnyFunctionType::printParams(ArrayRef Params, raw_ostream &OS, const PrintOptions &PO) { StreamPrinter Printer(OS); printParams(Params, Printer, PO); } void AnyFunctionType::printParams(ArrayRef Params, ASTPrinter &Printer, const PrintOptions &PO) { TypePrinter(Printer, PO).visitAnyFunctionTypeParams(Params, /*printLabels*/true); } std::string AnyFunctionType::getParamListAsString(ArrayRef Params, const PrintOptions &PO) { SmallString<16> Scratch; llvm::raw_svector_ostream OS(Scratch); AnyFunctionType::printParams(Params, OS); return std::string(OS.str()); } void LayoutConstraintInfo::print(raw_ostream &OS, const PrintOptions &PO) const { StreamPrinter Printer(OS); print(Printer, PO); } void LayoutConstraint::print(raw_ostream &OS, const PrintOptions &PO) const { assert(*this); getPointer()->print(OS, PO); } void LayoutConstraintInfo::print(ASTPrinter &Printer, const PrintOptions &PO) const { Printer << getName(); switch (getKind()) { case LayoutConstraintKind::UnknownLayout: case LayoutConstraintKind::RefCountedObject: case LayoutConstraintKind::NativeRefCountedObject: case LayoutConstraintKind::Class: case LayoutConstraintKind::NativeClass: case LayoutConstraintKind::Trivial: return; case LayoutConstraintKind::TrivialOfAtMostSize: case LayoutConstraintKind::TrivialOfExactSize: Printer << "("; Printer << SizeInBits; if (Alignment) Printer << ", " << Alignment; Printer << ")"; break; } } void LayoutConstraint::dump() const { if (!*this) { llvm::errs() << "(null)\n"; return; } getPointer()->print(llvm::errs()); } void GenericSignatureImpl::print(raw_ostream &OS, PrintOptions PO) const { GenericSignature(const_cast(this)).print(OS, PO); } void GenericSignatureImpl::print(ASTPrinter &Printer, PrintOptions PO) const { GenericSignature(const_cast(this)).print(Printer, PO); } void GenericSignature::print(raw_ostream &OS, const PrintOptions &Opts) const { StreamPrinter Printer(OS); print(Printer, Opts); } void GenericSignature::print(ASTPrinter &Printer, const PrintOptions &Opts) const { if (isNull()) { Printer << ""; return; } PrintAST(Printer, Opts).printGenericSignature(*this, PrintAST::PrintParams | PrintAST::PrintRequirements); } void GenericSignature::dump() const { print(llvm::errs()); llvm::errs() << '\n'; } void Requirement::dump() const { dump(llvm::errs()); llvm::errs() << '\n'; } void Requirement::dump(raw_ostream &out) const { switch (getKind()) { case RequirementKind::Conformance: out << "conforms_to: "; break; case RequirementKind::Layout: out << "layout: "; break; case RequirementKind::Superclass: out << "superclass: "; break; case RequirementKind::SameType: out << "same_type: "; break; } PrintOptions opts; opts.ProtocolQualifiedDependentMemberTypes = true; getFirstType().print(out, opts); out << " "; if (getKind() != RequirementKind::Layout && getSecondType()) getSecondType().print(out, opts); else if (getLayoutConstraint()) out << getLayoutConstraint(); } void Requirement::print(raw_ostream &os, const PrintOptions &opts) const { StreamPrinter printer(os); PrintAST(printer, opts).printRequirement(*this); } void Requirement::print(ASTPrinter &printer, const PrintOptions &opts) const { PrintAST(printer, opts).printRequirement(*this); } std::string GenericSignatureImpl::getAsString() const { return GenericSignature(const_cast(this)) .getAsString(); } std::string GenericSignature::getAsString() const { std::string result; llvm::raw_string_ostream out(result); print(out); return out.str(); } static StringRef getStringForParameterConvention(ParameterConvention conv) { switch (conv) { case ParameterConvention::Indirect_In: return "@in "; case ParameterConvention::Indirect_In_Constant: return "@in_constant "; case ParameterConvention::Indirect_In_Guaranteed: return "@in_guaranteed "; case ParameterConvention::Indirect_Inout: return "@inout "; case ParameterConvention::Indirect_InoutAliasable: return "@inout_aliasable "; case ParameterConvention::Direct_Owned: return "@owned "; case ParameterConvention::Direct_Unowned: return ""; case ParameterConvention::Direct_Guaranteed: return "@guaranteed "; } llvm_unreachable("bad parameter convention"); } StringRef swift::getCheckedCastKindName(CheckedCastKind kind) { switch (kind) { case CheckedCastKind::Unresolved: return "unresolved"; case CheckedCastKind::Coercion: return "coercion"; case CheckedCastKind::ValueCast: return "value_cast"; case CheckedCastKind::ArrayDowncast: return "array_downcast"; case CheckedCastKind::DictionaryDowncast: return "dictionary_downcast"; case CheckedCastKind::SetDowncast: return "set_downcast"; case CheckedCastKind::BridgingCoercion: return "bridging_coercion"; } llvm_unreachable("bad checked cast name"); } void SILParameterInfo::dump() const { print(llvm::errs()); llvm::errs() << '\n'; } void SILParameterInfo::print(raw_ostream &OS, const PrintOptions &Opts) const { StreamPrinter Printer(OS); print(Printer, Opts); } void SILParameterInfo::print(ASTPrinter &Printer, const PrintOptions &Opts) const { switch (getDifferentiability()) { case SILParameterDifferentiability::NotDifferentiable: Printer << "@noDerivative "; break; default: break; } Printer << getStringForParameterConvention(getConvention()); getInterfaceType().print(Printer, Opts); } static StringRef getStringForResultConvention(ResultConvention conv) { switch (conv) { case ResultConvention::Indirect: return "@out "; case ResultConvention::Owned: return "@owned "; case ResultConvention::Unowned: return ""; case ResultConvention::UnownedInnerPointer: return "@unowned_inner_pointer "; case ResultConvention::Autoreleased: return "@autoreleased "; } llvm_unreachable("bad result convention"); } void SILResultInfo::dump() const { print(llvm::errs()); llvm::errs() << '\n'; } void SILResultInfo::print(raw_ostream &OS, const PrintOptions &Opts) const { StreamPrinter Printer(OS); print(Printer, Opts); } void SILResultInfo::print(ASTPrinter &Printer, const PrintOptions &Opts) const { switch (getDifferentiability()) { case SILResultDifferentiability::NotDifferentiable: Printer << "@noDerivative "; break; default: break; } Printer << getStringForResultConvention(getConvention()); getInterfaceType().print(Printer, Opts); } std::string Type::getString(const PrintOptions &PO) const { std::string Result; llvm::raw_string_ostream OS(Result); print(OS, PO); return OS.str(); } std::string TypeBase::getString(const PrintOptions &PO) const { std::string Result; llvm::raw_string_ostream OS(Result); print(OS, PO); return OS.str(); } std::string Type::getStringAsComponent(const PrintOptions &PO) const { std::string Result; llvm::raw_string_ostream OS(Result); if (getPointer()->hasSimpleTypeRepr()) { print(OS, PO); } else { OS << "("; print(OS, PO); OS << ")"; } return OS.str(); } std::string TypeBase::getStringAsComponent(const PrintOptions &PO) const { std::string Result; llvm::raw_string_ostream OS(Result); if (hasSimpleTypeRepr()) { print(OS, PO); } else { OS << "("; print(OS, PO); OS << ")"; } return OS.str(); } void TypeBase::dumpPrint() const { print(llvm::errs()); llvm::errs() << '\n'; } void TypeBase::print(raw_ostream &OS, const PrintOptions &PO) const { Type(const_cast(this)).print(OS, PO); } void TypeBase::print(ASTPrinter &Printer, const PrintOptions &PO) const { Type(const_cast(this)).print(Printer, PO); } std::string LayoutConstraint::getString(const PrintOptions &PO) const { std::string Result; llvm::raw_string_ostream OS(Result); print(OS, PO); return OS.str(); } std::string LayoutConstraintInfo::getString(const PrintOptions &PO) const { std::string Result; llvm::raw_string_ostream OS(Result); print(OS, PO); return OS.str(); } void ProtocolConformance::printName(llvm::raw_ostream &os, const PrintOptions &PO) const { if (getKind() == ProtocolConformanceKind::Normal) { if (auto genericSig = getGenericSignature()) { StreamPrinter sPrinter(os); TypePrinter typePrinter(sPrinter, PO); typePrinter .printGenericSignature(genericSig, PrintAST::PrintParams | PrintAST::PrintRequirements); os << ' '; } } getType()->print(os, PO); os << ": "; switch (getKind()) { case ProtocolConformanceKind::Normal: { auto normal = cast(this); os << normal->getProtocol()->getName() << " module " << normal->getDeclContext()->getParentModule()->getRealName(); break; } case ProtocolConformanceKind::Self: { auto self = cast(this); os << self->getProtocol()->getName() << " module " << self->getDeclContext()->getParentModule()->getRealName(); break; } case ProtocolConformanceKind::Specialized: { auto spec = cast(this); os << "specialize <"; interleave(spec->getSubstitutionMap().getReplacementTypes(), [&](Type type) { type.print(os, PO); }, [&] { os << ", "; }); os << "> ("; spec->getGenericConformance()->printName(os, PO); os << ")"; break; } case ProtocolConformanceKind::Inherited: { auto inherited = cast(this); os << "inherit ("; inherited->getInheritedConformance()->printName(os, PO); os << ")"; break; } case ProtocolConformanceKind::Builtin: { auto builtin = cast(this); os << builtin->getProtocol()->getName() << " type " << builtin->getType(); break; } } } void swift::printEnumElementsAsCases( llvm::DenseSet &UnhandledElements, llvm::raw_ostream &OS) { // Sort the missing elements to a vector because set does not guarantee // orders. SmallVector SortedElements; SortedElements.insert(SortedElements.begin(), UnhandledElements.begin(), UnhandledElements.end()); std::sort(SortedElements.begin(), SortedElements.end(), [](EnumElementDecl *LHS, EnumElementDecl *RHS) { return LHS->getNameStr().compare(RHS->getNameStr()) < 0; }); auto printPayloads = [](ParameterList *PL, llvm::raw_ostream &OS) { // If the enum element has no payloads, return. if (!PL) return; OS << "("; // Print each element in the pattern match. for (auto i = PL->begin(); i != PL->end(); ++i) { auto *param = *i; if (param->hasName()) { OS << tok::kw_let << " " << param->getName().str(); } else { OS << "_"; } if (i + 1 != PL->end()) { OS << ", "; } } OS << ")"; }; // Print each enum element name. std::for_each(SortedElements.begin(), SortedElements.end(), [&](EnumElementDecl *EE) { OS << tok::kw_case << " ." << EE->getNameStr(); printPayloads(EE->getParameterList(), OS); OS << ": " << getCodePlaceholder() << "\n"; }); } void swift::getInheritedForPrinting( const Decl *decl, const PrintOptions &options, llvm::SmallVectorImpl &Results) { ArrayRef inherited; if (auto td = dyn_cast(decl)) { inherited = td->getInherited(); } else if (auto ed = dyn_cast(decl)) { inherited = ed->getInherited(); } // Collect explicit inherited types. for (auto entry: inherited) { if (auto ty = entry.getType()) { bool foundUnprintable = ty.findIf([&](Type subTy) { if (auto aliasTy = dyn_cast(subTy.getPointer())) return !options.shouldPrint(aliasTy->getDecl()); if (auto NTD = subTy->getAnyNominal()) { if (!options.shouldPrint(NTD)) return true; } return false; }); if (foundUnprintable) continue; } Results.push_back(entry); } // Collect synthesized conformances. auto &ctx = decl->getASTContext(); llvm::SetVector protocols; llvm::TinyPtrVector uncheckedProtocols; for (auto attr : decl->getAttrs().getAttributes()) { if (auto *proto = ctx.getProtocol(attr->getProtocolKind())) { // The SerialExecutor conformance is only synthesized on the root // actor class, so we can just test resilience immediately. if (proto->isSpecificProtocol(KnownProtocolKind::SerialExecutor) && cast(decl)->isResilient()) continue; if (attr->getProtocolKind() == KnownProtocolKind::RawRepresentable && isa(decl) && cast(decl)->hasRawType()) continue; protocols.insert(proto); if (attr->isUnchecked()) uncheckedProtocols.push_back(proto); } } for (size_t i = 0; i < protocols.size(); i++) { auto proto = protocols[i]; bool isUnchecked = llvm::is_contained(uncheckedProtocols, proto); if (!options.shouldPrint(proto)) { // If private stdlib protocols are skipped and this is a private stdlib // protocol, see if any of its inherited protocols are public. Those // protocols can affect the user-visible behavior of the declaration, and // should be printed. if (options.SkipPrivateStdlibDecls && proto->isPrivateStdlibDecl(!options.SkipUnderscoredStdlibProtocols)) { auto inheritedProtocols = proto->getInheritedProtocols(); protocols.insert(inheritedProtocols.begin(), inheritedProtocols.end()); if (isUnchecked) copy(inheritedProtocols, std::back_inserter(uncheckedProtocols)); } continue; } Results.push_back({TypeLoc::withoutLoc(proto->getDeclaredInterfaceType()), isUnchecked}); } } //===----------------------------------------------------------------------===// // Generic param list printing. //===----------------------------------------------------------------------===// void RequirementRepr::dump() const { print(llvm::errs()); llvm::errs() << "\n"; } void RequirementRepr::print(raw_ostream &out) const { StreamPrinter printer(out); print(printer); } void RequirementRepr::print(ASTPrinter &out) const { auto printLayoutConstraint = [&](const LayoutConstraintLoc &LayoutConstraintLoc) { LayoutConstraintLoc.getLayoutConstraint()->print(out, PrintOptions()); }; switch (getKind()) { case RequirementReprKind::LayoutConstraint: if (auto *repr = getSubjectRepr()) { repr->print(out, PrintOptions()); } out << " : "; printLayoutConstraint(getLayoutConstraintLoc()); break; case RequirementReprKind::TypeConstraint: if (auto *repr = getSubjectRepr()) { repr->print(out, PrintOptions()); } out << " : "; if (auto *repr = getConstraintRepr()) { repr->print(out, PrintOptions()); } break; case RequirementReprKind::SameType: if (auto *repr = getFirstTypeRepr()) { repr->print(out, PrintOptions()); } out << " == "; if (auto *repr = getSecondTypeRepr()) { repr->print(out, PrintOptions()); } break; } } void GenericParamList::dump() const { print(llvm::errs()); llvm::errs() << '\n'; } void GenericParamList::print(raw_ostream &out, const PrintOptions &PO) const { StreamPrinter printer(out); print(printer, PO); } static void printTrailingRequirements(ASTPrinter &Printer, ArrayRef Reqs, bool printWhereKeyword) { if (Reqs.empty()) return; if (printWhereKeyword) Printer << " where "; interleave( Reqs, [&](const RequirementRepr &req) { Printer.callPrintStructurePre(PrintStructureKind::GenericRequirement); req.print(Printer); Printer.printStructurePost(PrintStructureKind::GenericRequirement); }, [&] { Printer << ", "; }); } void GenericParamList::print(ASTPrinter &Printer, const PrintOptions &PO) const { Printer << '<'; interleave( *this, [&](const GenericTypeParamDecl *P) { Printer << P->getName(); if (!P->getInherited().empty()) { Printer << " : "; auto loc = P->getInherited()[0]; if (willUseTypeReprPrinting(loc, nullptr, PO)) { loc.getTypeRepr()->print(Printer, PO); } else { loc.getType()->print(Printer, PO); } } }, [&] { Printer << ", "; }); printTrailingRequirements(Printer, getRequirements(), /*printWhereKeyword*/ true); Printer << '>'; } void TrailingWhereClause::print(llvm::raw_ostream &OS, bool printWhereKeyword) const { StreamPrinter Printer(OS); printTrailingRequirements(Printer, getRequirements(), printWhereKeyword); }