mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-07 05:56:05 +00:00
a72d7eea54
This patch improves the design of the IncrementalParser and Interpreter classes. Now the incremental parser is only responsible for building the partial translation unit declaration and the AST, while the Interpreter fills in the lower level llvm::Module and other JIT-related infrastructure. Finally the Interpreter class now orchestrates the AST and the LLVM IR with the IncrementalParser and IncrementalExecutor classes. The design improvement allows us to rework some of the logic that extracts an interpreter value into the clang::Value object. The new implementation simplifies use-cases which are used for out-of-process execution by allowing interpreter to be inherited or customized with an clang::ASTConsumer. This change will enable completing the pretty printing work which is in llvm/llvm-project#84769
418 lines
16 KiB
C++
418 lines
16 KiB
C++
//===- MultiplexConsumer.cpp - AST Consumer for PCH Generation --*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines the MultiplexConsumer class. It also declares and defines
|
|
// MultiplexASTDeserializationListener and MultiplexASTMutationListener, which
|
|
// are implementation details of MultiplexConsumer.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Frontend/MultiplexConsumer.h"
|
|
#include "clang/AST/ASTMutationListener.h"
|
|
#include "clang/AST/DeclGroup.h"
|
|
|
|
using namespace clang;
|
|
|
|
namespace clang {
|
|
|
|
class NamespaceDecl;
|
|
class TranslationUnitDecl;
|
|
|
|
MultiplexASTDeserializationListener::MultiplexASTDeserializationListener(
|
|
const std::vector<ASTDeserializationListener*>& L)
|
|
: Listeners(L) {
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::ReaderInitialized(
|
|
ASTReader *Reader) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->ReaderInitialized(Reader);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::IdentifierRead(
|
|
serialization::IdentifierID ID, IdentifierInfo *II) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->IdentifierRead(ID, II);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::MacroRead(
|
|
serialization::MacroID ID, MacroInfo *MI) {
|
|
for (auto &Listener : Listeners)
|
|
Listener->MacroRead(ID, MI);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::TypeRead(
|
|
serialization::TypeIdx Idx, QualType T) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->TypeRead(Idx, T);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::DeclRead(GlobalDeclID ID,
|
|
const Decl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->DeclRead(ID, D);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::PredefinedDeclBuilt(PredefinedDeclIDs ID, const Decl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->PredefinedDeclBuilt(ID, D);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::SelectorRead(
|
|
serialization::SelectorID ID, Selector Sel) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->SelectorRead(ID, Sel);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::MacroDefinitionRead(
|
|
serialization::PreprocessedEntityID ID, MacroDefinitionRecord *MD) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->MacroDefinitionRead(ID, MD);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::ModuleRead(
|
|
serialization::SubmoduleID ID, Module *Mod) {
|
|
for (auto &Listener : Listeners)
|
|
Listener->ModuleRead(ID, Mod);
|
|
}
|
|
|
|
void MultiplexASTDeserializationListener::ModuleImportRead(
|
|
serialization::SubmoduleID ID, SourceLocation ImportLoc) {
|
|
for (auto &Listener : Listeners)
|
|
Listener->ModuleImportRead(ID, ImportLoc);
|
|
}
|
|
|
|
// This ASTMutationListener forwards its notifications to a set of
|
|
// child listeners.
|
|
class MultiplexASTMutationListener : public ASTMutationListener {
|
|
public:
|
|
// Does NOT take ownership of the elements in L.
|
|
MultiplexASTMutationListener(ArrayRef<ASTMutationListener*> L);
|
|
void CompletedTagDefinition(const TagDecl *D) override;
|
|
void AddedVisibleDecl(const DeclContext *DC, const Decl *D) override;
|
|
void AddedCXXImplicitMember(const CXXRecordDecl *RD, const Decl *D) override;
|
|
void AddedCXXTemplateSpecialization(const ClassTemplateDecl *TD,
|
|
const ClassTemplateSpecializationDecl *D) override;
|
|
void AddedCXXTemplateSpecialization(const VarTemplateDecl *TD,
|
|
const VarTemplateSpecializationDecl *D) override;
|
|
void AddedCXXTemplateSpecialization(const FunctionTemplateDecl *TD,
|
|
const FunctionDecl *D) override;
|
|
void ResolvedExceptionSpec(const FunctionDecl *FD) override;
|
|
void DeducedReturnType(const FunctionDecl *FD, QualType ReturnType) override;
|
|
void ResolvedOperatorDelete(const CXXDestructorDecl *DD,
|
|
const FunctionDecl *Delete,
|
|
Expr *ThisArg) override;
|
|
void CompletedImplicitDefinition(const FunctionDecl *D) override;
|
|
void InstantiationRequested(const ValueDecl *D) override;
|
|
void VariableDefinitionInstantiated(const VarDecl *D) override;
|
|
void FunctionDefinitionInstantiated(const FunctionDecl *D) override;
|
|
void DefaultArgumentInstantiated(const ParmVarDecl *D) override;
|
|
void DefaultMemberInitializerInstantiated(const FieldDecl *D) override;
|
|
void AddedObjCCategoryToInterface(const ObjCCategoryDecl *CatD,
|
|
const ObjCInterfaceDecl *IFD) override;
|
|
void DeclarationMarkedUsed(const Decl *D) override;
|
|
void DeclarationMarkedOpenMPThreadPrivate(const Decl *D) override;
|
|
void DeclarationMarkedOpenMPAllocate(const Decl *D, const Attr *A) override;
|
|
void DeclarationMarkedOpenMPDeclareTarget(const Decl *D,
|
|
const Attr *Attr) override;
|
|
void RedefinedHiddenDefinition(const NamedDecl *D, Module *M) override;
|
|
void AddedAttributeToRecord(const Attr *Attr,
|
|
const RecordDecl *Record) override;
|
|
void EnteringModulePurview() override;
|
|
void AddedManglingNumber(const Decl *D, unsigned) override;
|
|
void AddedStaticLocalNumbers(const Decl *D, unsigned) override;
|
|
void AddedAnonymousNamespace(const TranslationUnitDecl *,
|
|
NamespaceDecl *AnonNamespace) override;
|
|
|
|
private:
|
|
std::vector<ASTMutationListener*> Listeners;
|
|
};
|
|
|
|
MultiplexASTMutationListener::MultiplexASTMutationListener(
|
|
ArrayRef<ASTMutationListener*> L)
|
|
: Listeners(L.begin(), L.end()) {
|
|
}
|
|
|
|
void MultiplexASTMutationListener::CompletedTagDefinition(const TagDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->CompletedTagDefinition(D);
|
|
}
|
|
|
|
void MultiplexASTMutationListener::AddedVisibleDecl(
|
|
const DeclContext *DC, const Decl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->AddedVisibleDecl(DC, D);
|
|
}
|
|
|
|
void MultiplexASTMutationListener::AddedCXXImplicitMember(
|
|
const CXXRecordDecl *RD, const Decl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->AddedCXXImplicitMember(RD, D);
|
|
}
|
|
void MultiplexASTMutationListener::AddedCXXTemplateSpecialization(
|
|
const ClassTemplateDecl *TD, const ClassTemplateSpecializationDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->AddedCXXTemplateSpecialization(TD, D);
|
|
}
|
|
void MultiplexASTMutationListener::AddedCXXTemplateSpecialization(
|
|
const VarTemplateDecl *TD, const VarTemplateSpecializationDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->AddedCXXTemplateSpecialization(TD, D);
|
|
}
|
|
void MultiplexASTMutationListener::AddedCXXTemplateSpecialization(
|
|
const FunctionTemplateDecl *TD, const FunctionDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->AddedCXXTemplateSpecialization(TD, D);
|
|
}
|
|
void MultiplexASTMutationListener::ResolvedExceptionSpec(
|
|
const FunctionDecl *FD) {
|
|
for (auto &Listener : Listeners)
|
|
Listener->ResolvedExceptionSpec(FD);
|
|
}
|
|
void MultiplexASTMutationListener::DeducedReturnType(const FunctionDecl *FD,
|
|
QualType ReturnType) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->DeducedReturnType(FD, ReturnType);
|
|
}
|
|
void MultiplexASTMutationListener::ResolvedOperatorDelete(
|
|
const CXXDestructorDecl *DD, const FunctionDecl *Delete, Expr *ThisArg) {
|
|
for (auto *L : Listeners)
|
|
L->ResolvedOperatorDelete(DD, Delete, ThisArg);
|
|
}
|
|
void MultiplexASTMutationListener::CompletedImplicitDefinition(
|
|
const FunctionDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->CompletedImplicitDefinition(D);
|
|
}
|
|
void MultiplexASTMutationListener::InstantiationRequested(const ValueDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->InstantiationRequested(D);
|
|
}
|
|
void MultiplexASTMutationListener::VariableDefinitionInstantiated(
|
|
const VarDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->VariableDefinitionInstantiated(D);
|
|
}
|
|
void MultiplexASTMutationListener::FunctionDefinitionInstantiated(
|
|
const FunctionDecl *D) {
|
|
for (auto &Listener : Listeners)
|
|
Listener->FunctionDefinitionInstantiated(D);
|
|
}
|
|
void MultiplexASTMutationListener::DefaultArgumentInstantiated(
|
|
const ParmVarDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->DefaultArgumentInstantiated(D);
|
|
}
|
|
void MultiplexASTMutationListener::DefaultMemberInitializerInstantiated(
|
|
const FieldDecl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->DefaultMemberInitializerInstantiated(D);
|
|
}
|
|
void MultiplexASTMutationListener::AddedObjCCategoryToInterface(
|
|
const ObjCCategoryDecl *CatD,
|
|
const ObjCInterfaceDecl *IFD) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->AddedObjCCategoryToInterface(CatD, IFD);
|
|
}
|
|
void MultiplexASTMutationListener::DeclarationMarkedUsed(const Decl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->DeclarationMarkedUsed(D);
|
|
}
|
|
void MultiplexASTMutationListener::DeclarationMarkedOpenMPThreadPrivate(
|
|
const Decl *D) {
|
|
for (size_t i = 0, e = Listeners.size(); i != e; ++i)
|
|
Listeners[i]->DeclarationMarkedOpenMPThreadPrivate(D);
|
|
}
|
|
void MultiplexASTMutationListener::DeclarationMarkedOpenMPAllocate(
|
|
const Decl *D, const Attr *A) {
|
|
for (ASTMutationListener *L : Listeners)
|
|
L->DeclarationMarkedOpenMPAllocate(D, A);
|
|
}
|
|
void MultiplexASTMutationListener::DeclarationMarkedOpenMPDeclareTarget(
|
|
const Decl *D, const Attr *Attr) {
|
|
for (auto *L : Listeners)
|
|
L->DeclarationMarkedOpenMPDeclareTarget(D, Attr);
|
|
}
|
|
void MultiplexASTMutationListener::RedefinedHiddenDefinition(const NamedDecl *D,
|
|
Module *M) {
|
|
for (auto *L : Listeners)
|
|
L->RedefinedHiddenDefinition(D, M);
|
|
}
|
|
|
|
void MultiplexASTMutationListener::AddedAttributeToRecord(
|
|
const Attr *Attr,
|
|
const RecordDecl *Record) {
|
|
for (auto *L : Listeners)
|
|
L->AddedAttributeToRecord(Attr, Record);
|
|
}
|
|
|
|
void MultiplexASTMutationListener::EnteringModulePurview() {
|
|
for (auto *L : Listeners)
|
|
L->EnteringModulePurview();
|
|
}
|
|
|
|
void MultiplexASTMutationListener::AddedManglingNumber(const Decl *D,
|
|
unsigned Number) {
|
|
for (auto *L : Listeners)
|
|
L->AddedManglingNumber(D, Number);
|
|
}
|
|
void MultiplexASTMutationListener::AddedStaticLocalNumbers(const Decl *D,
|
|
unsigned Number) {
|
|
for (auto *L : Listeners)
|
|
L->AddedStaticLocalNumbers(D, Number);
|
|
}
|
|
void MultiplexASTMutationListener::AddedAnonymousNamespace(
|
|
const TranslationUnitDecl *TU, NamespaceDecl *AnonNamespace) {
|
|
for (auto *L : Listeners)
|
|
L->AddedAnonymousNamespace(TU, AnonNamespace);
|
|
}
|
|
|
|
} // end namespace clang
|
|
|
|
MultiplexConsumer::MultiplexConsumer(
|
|
std::vector<std::unique_ptr<ASTConsumer>> C)
|
|
: Consumers(std::move(C)) {
|
|
// Collect the mutation listeners and deserialization listeners of all
|
|
// children, and create a multiplex listener each if so.
|
|
std::vector<ASTMutationListener *> mutationListeners;
|
|
std::vector<ASTDeserializationListener*> serializationListeners;
|
|
for (auto &Consumer : Consumers) {
|
|
if (auto *mutationListener = Consumer->GetASTMutationListener())
|
|
mutationListeners.push_back(mutationListener);
|
|
if (auto *serializationListener = Consumer->GetASTDeserializationListener())
|
|
serializationListeners.push_back(serializationListener);
|
|
}
|
|
if (!mutationListeners.empty()) {
|
|
MutationListener =
|
|
std::make_unique<MultiplexASTMutationListener>(mutationListeners);
|
|
}
|
|
if (!serializationListeners.empty()) {
|
|
DeserializationListener =
|
|
std::make_unique<MultiplexASTDeserializationListener>(
|
|
serializationListeners);
|
|
}
|
|
}
|
|
|
|
MultiplexConsumer::MultiplexConsumer(std::unique_ptr<ASTConsumer> C)
|
|
: MultiplexConsumer([](std::unique_ptr<ASTConsumer> Consumer) {
|
|
std::vector<std::unique_ptr<ASTConsumer>> Consumers;
|
|
Consumers.push_back(std::move(Consumer));
|
|
return Consumers;
|
|
}(std::move(C))) {}
|
|
|
|
MultiplexConsumer::~MultiplexConsumer() {}
|
|
|
|
void MultiplexConsumer::Initialize(ASTContext &Context) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->Initialize(Context);
|
|
}
|
|
|
|
bool MultiplexConsumer::HandleTopLevelDecl(DeclGroupRef D) {
|
|
bool Continue = true;
|
|
for (auto &Consumer : Consumers)
|
|
Continue = Continue && Consumer->HandleTopLevelDecl(D);
|
|
return Continue;
|
|
}
|
|
|
|
void MultiplexConsumer::HandleInlineFunctionDefinition(FunctionDecl *D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleInlineFunctionDefinition(D);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleCXXStaticMemberVarInstantiation(VarDecl *VD) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleCXXStaticMemberVarInstantiation(VD);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleInterestingDecl(DeclGroupRef D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleInterestingDecl(D);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleTranslationUnit(ASTContext &Ctx) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleTranslationUnit(Ctx);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleTagDeclDefinition(TagDecl *D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleTagDeclDefinition(D);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleTagDeclRequiredDefinition(const TagDecl *D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleTagDeclRequiredDefinition(D);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleCXXImplicitFunctionInstantiation(FunctionDecl *D){
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleCXXImplicitFunctionInstantiation(D);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleTopLevelDeclInObjCContainer(DeclGroupRef D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleTopLevelDeclInObjCContainer(D);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleImplicitImportDecl(ImportDecl *D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleImplicitImportDecl(D);
|
|
}
|
|
|
|
void MultiplexConsumer::CompleteTentativeDefinition(VarDecl *D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->CompleteTentativeDefinition(D);
|
|
}
|
|
|
|
void MultiplexConsumer::CompleteExternalDeclaration(DeclaratorDecl *D) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->CompleteExternalDeclaration(D);
|
|
}
|
|
|
|
void MultiplexConsumer::AssignInheritanceModel(CXXRecordDecl *RD) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->AssignInheritanceModel(RD);
|
|
}
|
|
|
|
void MultiplexConsumer::HandleVTable(CXXRecordDecl *RD) {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->HandleVTable(RD);
|
|
}
|
|
|
|
ASTMutationListener *MultiplexConsumer::GetASTMutationListener() {
|
|
return MutationListener.get();
|
|
}
|
|
|
|
ASTDeserializationListener *MultiplexConsumer::GetASTDeserializationListener() {
|
|
return DeserializationListener.get();
|
|
}
|
|
|
|
void MultiplexConsumer::PrintStats() {
|
|
for (auto &Consumer : Consumers)
|
|
Consumer->PrintStats();
|
|
}
|
|
|
|
bool MultiplexConsumer::shouldSkipFunctionBody(Decl *D) {
|
|
bool Skip = true;
|
|
for (auto &Consumer : Consumers)
|
|
Skip = Skip && Consumer->shouldSkipFunctionBody(D);
|
|
return Skip;
|
|
}
|
|
|
|
void MultiplexConsumer::InitializeSema(Sema &S) {
|
|
for (auto &Consumer : Consumers)
|
|
if (SemaConsumer *SC = dyn_cast<SemaConsumer>(Consumer.get()))
|
|
SC->InitializeSema(S);
|
|
}
|
|
|
|
void MultiplexConsumer::ForgetSema() {
|
|
for (auto &Consumer : Consumers)
|
|
if (SemaConsumer *SC = dyn_cast<SemaConsumer>(Consumer.get()))
|
|
SC->ForgetSema();
|
|
}
|