mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
14a7ae56db
This patch removes the duplication between FrontendOptions and TBDGenOptions by parsing the TBDGenOptions up front similarly to IRGen and SIL options.
114 lines
3.2 KiB
C++
114 lines
3.2 KiB
C++
//===--- TBDGenVisitor.h - AST Visitor for TBD generation -----------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 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 defines the visitor that finds all symbols in a swift AST.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#ifndef SWIFT_TBDGEN_TBDGENVISITOR_H
|
|
#define SWIFT_TBDGEN_TBDGENVISITOR_H
|
|
|
|
#include "swift/AST/ASTMangler.h"
|
|
#include "swift/AST/ASTVisitor.h"
|
|
#include "swift/AST/Module.h"
|
|
#include "swift/AST/ParameterList.h"
|
|
#include "swift/Basic/LLVM.h"
|
|
#include "swift/IRGen/Linking.h"
|
|
#include "swift/SIL/FormalLinkage.h"
|
|
#include "swift/SIL/SILDeclRef.h"
|
|
#include "swift/SIL/SILWitnessTable.h"
|
|
#include "swift/SIL/TypeLowering.h"
|
|
#include "llvm/ADT/StringSet.h"
|
|
#include "llvm/ADT/Triple.h"
|
|
|
|
#include "tapi/InterfaceFile.h"
|
|
|
|
using namespace swift::irgen;
|
|
using StringSet = llvm::StringSet<>;
|
|
|
|
namespace swift {
|
|
|
|
struct TBDGenOptions;
|
|
|
|
namespace tbdgen {
|
|
|
|
class TBDGenVisitor : public ASTVisitor<TBDGenVisitor> {
|
|
public:
|
|
tapi::internal::InterfaceFile &Symbols;
|
|
tapi::internal::ArchitectureSet Archs;
|
|
StringSet *StringSymbols;
|
|
|
|
const UniversalLinkageInfo &UniversalLinkInfo;
|
|
ModuleDecl *SwiftModule;
|
|
const TBDGenOptions &Opts;
|
|
|
|
private:
|
|
bool FileHasEntryPoint = false;
|
|
|
|
void addSymbol(StringRef name, tapi::internal::SymbolKind kind =
|
|
tapi::internal::SymbolKind::GlobalSymbol);
|
|
|
|
void addSymbol(SILDeclRef declRef);
|
|
|
|
void addSymbol(LinkEntity entity);
|
|
|
|
void addConformances(DeclContext *DC);
|
|
|
|
void addDispatchThunk(SILDeclRef declRef);
|
|
|
|
public:
|
|
TBDGenVisitor(tapi::internal::InterfaceFile &symbols,
|
|
tapi::internal::ArchitectureSet archs, StringSet *stringSymbols,
|
|
const UniversalLinkageInfo &universalLinkInfo,
|
|
ModuleDecl *swiftModule, const TBDGenOptions &opts)
|
|
: Symbols(symbols), Archs(archs), StringSymbols(stringSymbols),
|
|
UniversalLinkInfo(universalLinkInfo), SwiftModule(swiftModule),
|
|
Opts(opts) {}
|
|
|
|
void setFileHasEntryPoint(bool hasEntryPoint) {
|
|
FileHasEntryPoint = hasEntryPoint;
|
|
|
|
if (hasEntryPoint)
|
|
addSymbol("main");
|
|
}
|
|
|
|
/// \brief Adds the global symbols associated with the first file.
|
|
void addFirstFileSymbols();
|
|
|
|
void visitAbstractFunctionDecl(AbstractFunctionDecl *AFD);
|
|
|
|
void visitAccessorDecl(AccessorDecl *AD);
|
|
|
|
void visitNominalTypeDecl(NominalTypeDecl *NTD);
|
|
|
|
void visitClassDecl(ClassDecl *CD);
|
|
|
|
void visitConstructorDecl(ConstructorDecl *CD);
|
|
|
|
void visitDestructorDecl(DestructorDecl *DD);
|
|
|
|
void visitExtensionDecl(ExtensionDecl *ED);
|
|
|
|
void visitProtocolDecl(ProtocolDecl *PD);
|
|
|
|
void visitAbstractStorageDecl(AbstractStorageDecl *ASD);
|
|
|
|
void visitVarDecl(VarDecl *VD);
|
|
|
|
void visitEnumDecl(EnumDecl *ED);
|
|
|
|
void visitDecl(Decl *D) {}
|
|
};
|
|
} // end namespace tbdgen
|
|
} // end namespace swift
|
|
|
|
#endif
|