blob: 61663e1ddb68bff2b14592027a3455bca4add3e1 [file] [log] [blame]
//===--- 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 (refsToEmit)
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::makeArrayRef(*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);
}