contrib/llvm-project/clang/include/clang/Frontend/ChainedDiagnosticConsumer.h

   1 //===- ChainedDiagnosticConsumer.h - Chain Diagnostic Clients ---*- 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 #ifndef LLVM_CLANG_FRONTEND_CHAINEDDIAGNOSTICCONSUMER_H
  10 #define LLVM_CLANG_FRONTEND_CHAINEDDIAGNOSTICCONSUMER_H
  11
  12 #include "clang/Basic/Diagnostic.h"
  13 #include <memory>
  14
  15 namespace clang {
  16 class LangOptions;
  17
  18 /// ChainedDiagnosticConsumer - Chain two diagnostic clients so that diagnostics
  19 /// go to the first client and then the second. The first diagnostic client
  20 /// should be the "primary" client, and will be used for computing whether the
  21 /// diagnostics should be included in counts.
  22 class ChainedDiagnosticConsumer : public DiagnosticConsumer {
  23   virtual void anchor();
  24   std::unique_ptr<DiagnosticConsumer> OwningPrimary;
  25   DiagnosticConsumer *Primary;
  26   std::unique_ptr<DiagnosticConsumer> Secondary;
  27
  28 public:
  29   ChainedDiagnosticConsumer(std::unique_ptr<DiagnosticConsumer> Primary,
  30                             std::unique_ptr<DiagnosticConsumer> Secondary)
  31       : OwningPrimary(std::move(Primary)), Primary(OwningPrimary.get()),
  32         Secondary(std::move(Secondary)) {}
  33
  34   /// Construct without taking ownership of \c Primary.
  35   ChainedDiagnosticConsumer(DiagnosticConsumer *Primary,
  36                             std::unique_ptr<DiagnosticConsumer> Secondary)
  37       : Primary(Primary), Secondary(std::move(Secondary)) {}
  38
  39   void BeginSourceFile(const LangOptions &LO,
  40                        const Preprocessor *PP) override {
  41     Primary->BeginSourceFile(LO, PP);
  42     Secondary->BeginSourceFile(LO, PP);
  43   }
  44
  45   void EndSourceFile() override {
  46     Secondary->EndSourceFile();
  47     Primary->EndSourceFile();
  48   }
  49
  50   void finish() override {
  51     Secondary->finish();
  52     Primary->finish();
  53   }
  54
  55   bool IncludeInDiagnosticCounts() const override {
  56     return Primary->IncludeInDiagnosticCounts();
  57   }
  58
  59   void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
  60                         const Diagnostic &Info) override {
  61     // Default implementation (Warnings/errors count).
  62     DiagnosticConsumer::HandleDiagnostic(DiagLevel, Info);
  63
  64     Primary->HandleDiagnostic(DiagLevel, Info);
  65     Secondary->HandleDiagnostic(DiagLevel, Info);
  66   }
  67 };
  68
  69 } // end namspace clang
  70
  71 #endif