1//===-- MultiplexConsumer.h - AST Consumer for PCH Generation ---*- C++ -*-===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10//  This file declares the MultiplexConsumer class, which can be used to
11//  multiplex ASTConsumer and SemaConsumer messages to many consumers.
12//
13//===----------------------------------------------------------------------===//
14
15#ifndef CLANG_FRONTEND_MULTIPLEXCONSUMER_H
16#define CLANG_FRONTEND_MULTIPLEXCONSUMER_H
17
18#include "clang/Basic/LLVM.h"
19#include "clang/Sema/SemaConsumer.h"
20#include <memory>
21#include <vector>
22
23namespace clang {
24
25class MultiplexASTMutationListener;
26class MultiplexASTDeserializationListener;
27
28// Has a list of ASTConsumers and calls each of them. Owns its children.
29class MultiplexConsumer : public SemaConsumer {
30public:
31  // Takes ownership of the pointers in C.
32  MultiplexConsumer(ArrayRef<ASTConsumer*> C);
33  ~MultiplexConsumer();
34
35  // ASTConsumer
36  void Initialize(ASTContext &Context) override;
37  void HandleCXXStaticMemberVarInstantiation(VarDecl *VD) override;
38  bool HandleTopLevelDecl(DeclGroupRef D) override;
39  void HandleInlineMethodDefinition(CXXMethodDecl *D) override;
40  void HandleInterestingDecl(DeclGroupRef D) override;
41  void HandleTranslationUnit(ASTContext &Ctx) override;
42  void HandleTagDeclDefinition(TagDecl *D) override;
43  void HandleCXXImplicitFunctionInstantiation(FunctionDecl *D) override;
44  void HandleTopLevelDeclInObjCContainer(DeclGroupRef D) override;
45  void CompleteTentativeDefinition(VarDecl *D) override;
46  void HandleVTable(CXXRecordDecl *RD, bool DefinitionRequired) override;
47  ASTMutationListener *GetASTMutationListener() override;
48  ASTDeserializationListener *GetASTDeserializationListener() override;
49  void PrintStats() override;
50
51  // SemaConsumer
52  void InitializeSema(Sema &S) override;
53  void ForgetSema() override;
54
55private:
56  std::vector<ASTConsumer*> Consumers;  // Owns these.
57  std::unique_ptr<MultiplexASTMutationListener> MutationListener;
58  std::unique_ptr<MultiplexASTDeserializationListener> DeserializationListener;
59};
60
61}  // end namespace clang
62
63#endif
64