1//===-- MultiplexConsumer.h - AST Consumer for PCH Generation ---*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file declares the MultiplexConsumer class, which can be used to
10// multiplex ASTConsumer and SemaConsumer messages to many consumers.
11//
12//===----------------------------------------------------------------------===//
13
14#ifndef LLVM_CLANG_FRONTEND_MULTIPLEXCONSUMER_H
15#define LLVM_CLANG_FRONTEND_MULTIPLEXCONSUMER_H
16
17#include "clang/Basic/LLVM.h"
18#include "clang/Sema/SemaConsumer.h"
19#include "clang/Serialization/ASTDeserializationListener.h"
20#include <memory>
21#include <vector>
22
23namespace clang {
24
25class MultiplexASTMutationListener;
26
27// This ASTDeserializationListener forwards its notifications to a set of
28// child listeners.
29class MultiplexASTDeserializationListener : public ASTDeserializationListener {
30public:
31 // Does NOT take ownership of the elements in L.
32 MultiplexASTDeserializationListener(
33 const std::vector<ASTDeserializationListener *> &L);
34 void ReaderInitialized(ASTReader *Reader) override;
35 void IdentifierRead(serialization::IdentID ID, IdentifierInfo *II) override;
36 void MacroRead(serialization::MacroID ID, MacroInfo *MI) override;
37 void TypeRead(serialization::TypeIdx Idx, QualType T) override;
38 void DeclRead(serialization::DeclID ID, const Decl *D) override;
39 void SelectorRead(serialization::SelectorID iD, Selector Sel) override;
40 void MacroDefinitionRead(serialization::PreprocessedEntityID,
41 MacroDefinitionRecord *MD) override;
42 void ModuleRead(serialization::SubmoduleID ID, Module *Mod) override;
43 void ModuleImportRead(serialization::SubmoduleID ID,
44 SourceLocation ImportLoc) override;
45
46private:
47 std::vector<ASTDeserializationListener *> Listeners;
48};
49
50// Has a list of ASTConsumers and calls each of them. Owns its children.
51class MultiplexConsumer : public SemaConsumer {
52public:
53 // Takes ownership of the pointers in C.
54 MultiplexConsumer(std::vector<std::unique_ptr<ASTConsumer>> C);
55 ~MultiplexConsumer() override;
56
57 // ASTConsumer
58 void Initialize(ASTContext &Context) override;
59 void HandleCXXStaticMemberVarInstantiation(VarDecl *VD) override;
60 bool HandleTopLevelDecl(DeclGroupRef D) override;
61 void HandleInlineFunctionDefinition(FunctionDecl *D) override;
62 void HandleInterestingDecl(DeclGroupRef D) override;
63 void HandleTranslationUnit(ASTContext &Ctx) override;
64 void HandleTagDeclDefinition(TagDecl *D) override;
65 void HandleTagDeclRequiredDefinition(const TagDecl *D) override;
66 void HandleCXXImplicitFunctionInstantiation(FunctionDecl *D) override;
67 void HandleTopLevelDeclInObjCContainer(DeclGroupRef D) override;
68 void HandleImplicitImportDecl(ImportDecl *D) override;
69 void CompleteTentativeDefinition(VarDecl *D) override;
70 void CompleteExternalDeclaration(VarDecl *D) override;
71 void AssignInheritanceModel(CXXRecordDecl *RD) override;
72 void HandleVTable(CXXRecordDecl *RD) override;
73 ASTMutationListener *GetASTMutationListener() override;
74 ASTDeserializationListener *GetASTDeserializationListener() override;
75 void PrintStats() override;
76 bool shouldSkipFunctionBody(Decl *D) override;
77
78 // SemaConsumer
79 void InitializeSema(Sema &S) override;
80 void ForgetSema() override;
81
82private:
83 std::vector<std::unique_ptr<ASTConsumer>> Consumers; // Owns these.
84 std::unique_ptr<MultiplexASTMutationListener> MutationListener;
85 std::unique_ptr<MultiplexASTDeserializationListener> DeserializationListener;
86};
87
88} // end namespace clang
89
90#endif
91

source code of clang/include/clang/Frontend/MultiplexConsumer.h