MultiplexConsumer.h 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. #pragma once
  2. #ifdef __GNUC__
  3. #pragma GCC diagnostic push
  4. #pragma GCC diagnostic ignored "-Wunused-parameter"
  5. #endif
  6. //===-- MultiplexConsumer.h - AST Consumer for PCH Generation ---*- C++ -*-===//
  7. //
  8. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  9. // See https://llvm.org/LICENSE.txt for license information.
  10. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  11. //
  12. //===----------------------------------------------------------------------===//
  13. //
  14. // This file declares the MultiplexConsumer class, which can be used to
  15. // multiplex ASTConsumer and SemaConsumer messages to many consumers.
  16. //
  17. //===----------------------------------------------------------------------===//
  18. #ifndef LLVM_CLANG_FRONTEND_MULTIPLEXCONSUMER_H
  19. #define LLVM_CLANG_FRONTEND_MULTIPLEXCONSUMER_H
  20. #include "clang/Basic/LLVM.h"
  21. #include "clang/Sema/SemaConsumer.h"
  22. #include "clang/Serialization/ASTDeserializationListener.h"
  23. #include <memory>
  24. #include <vector>
  25. namespace clang {
  26. class MultiplexASTMutationListener;
  27. // This ASTDeserializationListener forwards its notifications to a set of
  28. // child listeners.
  29. class MultiplexASTDeserializationListener : public ASTDeserializationListener {
  30. public:
  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. private:
  46. std::vector<ASTDeserializationListener *> Listeners;
  47. };
  48. // Has a list of ASTConsumers and calls each of them. Owns its children.
  49. class MultiplexConsumer : public SemaConsumer {
  50. public:
  51. // Takes ownership of the pointers in C.
  52. MultiplexConsumer(std::vector<std::unique_ptr<ASTConsumer>> C);
  53. ~MultiplexConsumer() override;
  54. // ASTConsumer
  55. void Initialize(ASTContext &Context) override;
  56. void HandleCXXStaticMemberVarInstantiation(VarDecl *VD) override;
  57. bool HandleTopLevelDecl(DeclGroupRef D) override;
  58. void HandleInlineFunctionDefinition(FunctionDecl *D) override;
  59. void HandleInterestingDecl(DeclGroupRef D) override;
  60. void HandleTranslationUnit(ASTContext &Ctx) override;
  61. void HandleTagDeclDefinition(TagDecl *D) override;
  62. void HandleTagDeclRequiredDefinition(const TagDecl *D) override;
  63. void HandleCXXImplicitFunctionInstantiation(FunctionDecl *D) override;
  64. void HandleTopLevelDeclInObjCContainer(DeclGroupRef D) override;
  65. void HandleImplicitImportDecl(ImportDecl *D) override;
  66. void CompleteTentativeDefinition(VarDecl *D) override;
  67. void CompleteExternalDeclaration(VarDecl *D) override;
  68. void AssignInheritanceModel(CXXRecordDecl *RD) override;
  69. void HandleVTable(CXXRecordDecl *RD) override;
  70. ASTMutationListener *GetASTMutationListener() override;
  71. ASTDeserializationListener *GetASTDeserializationListener() override;
  72. void PrintStats() override;
  73. bool shouldSkipFunctionBody(Decl *D) override;
  74. // SemaConsumer
  75. void InitializeSema(Sema &S) override;
  76. void ForgetSema() override;
  77. private:
  78. std::vector<std::unique_ptr<ASTConsumer>> Consumers; // Owns these.
  79. std::unique_ptr<MultiplexASTMutationListener> MutationListener;
  80. std::unique_ptr<MultiplexASTDeserializationListener> DeserializationListener;
  81. };
  82. } // end namespace clang
  83. #endif
  84. #ifdef __GNUC__
  85. #pragma GCC diagnostic pop
  86. #endif