mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
0d117c1345
See #59144 for more on why this is a bad idea. Patch out the synthesized file unit accessor to only clear the source cache, then patch up all the places that were assuming they could iterate over the module's file list and see synthesized files. rdar://94164512
119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
//===--- IRGenRequests.cpp - Requests for LLVM IR Generation --------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/AST/IRGenRequests.h"
|
|
#include "swift/AST/ASTContext.h"
|
|
#include "swift/AST/FileUnit.h"
|
|
#include "swift/AST/Module.h"
|
|
#include "swift/AST/SourceFile.h"
|
|
#include "swift/SIL/SILModule.h"
|
|
#include "swift/AST/TBDGenRequests.h"
|
|
#include "swift/Subsystems.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/ExecutionEngine/Orc/ThreadSafeModule.h"
|
|
|
|
using namespace swift;
|
|
|
|
namespace swift {
|
|
// Implement the IRGen type zone (zone 20).
|
|
#define SWIFT_TYPEID_ZONE IRGen
|
|
#define SWIFT_TYPEID_HEADER "swift/AST/IRGenTypeIDZone.def"
|
|
#include "swift/Basic/ImplementTypeIDZone.h"
|
|
#undef SWIFT_TYPEID_ZONE
|
|
#undef SWIFT_TYPEID_HEADER
|
|
} // end namespace swift
|
|
|
|
llvm::orc::ThreadSafeModule GeneratedModule::intoThreadSafeContext() && {
|
|
return {std::move(Module), std::move(Context)};
|
|
}
|
|
|
|
void swift::simple_display(llvm::raw_ostream &out,
|
|
const IRGenDescriptor &desc) {
|
|
auto *MD = desc.Ctx.dyn_cast<ModuleDecl *>();
|
|
if (MD) {
|
|
out << "IR Generation for module " << MD->getName();
|
|
} else {
|
|
auto *file = desc.Ctx.get<FileUnit *>();
|
|
out << "IR Generation for file ";
|
|
simple_display(out, file);
|
|
}
|
|
}
|
|
|
|
SourceLoc swift::extractNearestSourceLoc(const IRGenDescriptor &desc) {
|
|
return SourceLoc();
|
|
}
|
|
|
|
TinyPtrVector<FileUnit *> IRGenDescriptor::getFilesToEmit() const {
|
|
// If we've been asked to emit a specific set of symbols, we don't emit any
|
|
// whole files.
|
|
if (SymbolsToEmit)
|
|
return {};
|
|
|
|
// For a whole module, we emit IR for all files.
|
|
if (auto *mod = Ctx.dyn_cast<ModuleDecl *>())
|
|
return TinyPtrVector<FileUnit *>(mod->getFiles());
|
|
|
|
// For a primary file, we emit IR for both it and potentially its
|
|
// SynthesizedFileUnit.
|
|
auto *primary = Ctx.get<FileUnit *>();
|
|
TinyPtrVector<FileUnit *> files;
|
|
files.push_back(primary);
|
|
|
|
return files;
|
|
}
|
|
|
|
ModuleDecl *IRGenDescriptor::getParentModule() const {
|
|
if (auto *file = Ctx.dyn_cast<FileUnit *>())
|
|
return file->getParentModule();
|
|
return Ctx.get<ModuleDecl *>();
|
|
}
|
|
|
|
TBDGenDescriptor IRGenDescriptor::getTBDGenDescriptor() const {
|
|
if (auto *file = Ctx.dyn_cast<FileUnit *>()) {
|
|
return TBDGenDescriptor::forFile(file, TBDOpts);
|
|
} else {
|
|
auto *M = Ctx.get<ModuleDecl *>();
|
|
return TBDGenDescriptor::forModule(M, TBDOpts);
|
|
}
|
|
}
|
|
|
|
std::vector<std::string> IRGenDescriptor::getLinkerDirectives() const {
|
|
auto desc = getTBDGenDescriptor();
|
|
desc.getOptions().LinkerDirectivesOnly = true;
|
|
return getPublicSymbols(std::move(desc));
|
|
}
|
|
|
|
evaluator::DependencySource IRGenRequest::readDependencySource(
|
|
const evaluator::DependencyRecorder &e) const {
|
|
auto &desc = std::get<0>(getStorage());
|
|
|
|
// We don't track dependencies in whole-module mode.
|
|
if (auto *mod = desc.Ctx.dyn_cast<ModuleDecl *>()) {
|
|
return nullptr;
|
|
}
|
|
|
|
auto *primary = desc.Ctx.get<FileUnit *>();
|
|
return dyn_cast<SourceFile>(primary);
|
|
}
|
|
|
|
// Define request evaluation functions for each of the IRGen requests.
|
|
static AbstractRequestFunction *irGenRequestFunctions[] = {
|
|
#define SWIFT_REQUEST(Zone, Name, Sig, Caching, LocOptions) \
|
|
reinterpret_cast<AbstractRequestFunction *>(&Name::evaluateRequest),
|
|
#include "swift/AST/IRGenTypeIDZone.def"
|
|
#undef SWIFT_REQUEST
|
|
};
|
|
|
|
void swift::registerIRGenRequestFunctions(Evaluator &evaluator) {
|
|
evaluator.registerRequestFunctions(Zone::IRGen, irGenRequestFunctions);
|
|
}
|