mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
//===--- 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/Basic/Assertions.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 (isa<ModuleDecl *>(desc.context)) {
|
|
return nullptr;
|
|
}
|
|
|
|
// If we have a single source file, it's the source of dependencies.
|
|
return dyn_cast<SourceFile>(cast<FileUnit *>(desc.context));
|
|
}
|
|
|
|
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);
|
|
}
|