1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
//===--- 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);
}
|
