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
|
//===--- SILGenRequests.cpp - Requests for SIL 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/SILGenRequests.h"
#include "swift/AST/ASTContext.h"
#include "swift/AST/Module.h"
#include "swift/AST/FileUnit.h"
#include "swift/AST/SourceFile.h"
#include "swift/SIL/SILModule.h"
#include "swift/Subsystems.h"
using namespace swift;
namespace swift {
// Implement the SILGen type zone (zone 12).
#define SWIFT_TYPEID_ZONE SILGen
#define SWIFT_TYPEID_HEADER "swift/AST/SILGenTypeIDZone.def"
#include "swift/Basic/ImplementTypeIDZone.h"
#undef SWIFT_TYPEID_ZONE
#undef SWIFT_TYPEID_HEADER
} // end namespace swift
void swift::simple_display(llvm::raw_ostream &out,
const ASTLoweringDescriptor &desc) {
auto *MD = desc.context.dyn_cast<ModuleDecl *>();
auto *unit = desc.context.dyn_cast<FileUnit *>();
if (MD) {
out << "Lowering AST to SIL for module " << MD->getName();
} else {
assert(unit);
out << "Lowering AST to SIL for file ";
simple_display(out, unit);
}
}
SourceLoc swift::extractNearestSourceLoc(const ASTLoweringDescriptor &desc) {
return SourceLoc();
}
evaluator::DependencySource ASTLoweringRequest::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.context.dyn_cast<ModuleDecl *>()) {
return nullptr;
}
// If we have a single source file, it's the source of dependencies.
return dyn_cast<SourceFile>(desc.context.get<FileUnit *>());
}
ArrayRef<FileUnit *> ASTLoweringDescriptor::getFilesToEmit() const {
// If we have a specific set of SILDeclRefs to emit, we don't emit any whole
// files.
if (SourcesToEmit)
return {};
if (auto *mod = context.dyn_cast<ModuleDecl *>())
return mod->getFiles();
// For a single file, we can form an ArrayRef that points at its storage in
// the union.
return llvm::ArrayRef(*context.getAddrOfPtr1());
}
SourceFile *ASTLoweringDescriptor::getSourceFileToParse() const {
#ifndef NDEBUG
auto sfCount = llvm::count_if(getFilesToEmit(), [](FileUnit *file) {
return isa<SourceFile>(file);
});
auto silFileCount = llvm::count_if(getFilesToEmit(), [](FileUnit *file) {
auto *SF = dyn_cast<SourceFile>(file);
return SF && SF->Kind == SourceFileKind::SIL;
});
assert(silFileCount == 0 || (silFileCount == 1 && sfCount == 1) &&
"Cannot currently mix a .sil file with other SourceFiles");
#endif
for (auto *file : getFilesToEmit()) {
// Skip other kinds of files.
auto *SF = dyn_cast<SourceFile>(file);
if (!SF)
continue;
// Given the above precondition that a .sil file isn't mixed with other
// SourceFiles, we can return a SIL file if we have it, or return nullptr.
if (SF->Kind == SourceFileKind::SIL) {
return SF;
} else {
return nullptr;
}
}
return nullptr;
}
// Define request evaluation functions for each of the SILGen requests.
static AbstractRequestFunction *silGenRequestFunctions[] = {
#define SWIFT_REQUEST(Zone, Name, Sig, Caching, LocOptions) \
reinterpret_cast<AbstractRequestFunction *>(&Name::evaluateRequest),
#include "swift/AST/SILGenTypeIDZone.def"
#undef SWIFT_REQUEST
};
void swift::registerSILGenRequestFunctions(Evaluator &evaluator) {
evaluator.registerRequestFunctions(Zone::SILGen,
silGenRequestFunctions);
}
|
