1 //===- ChainedDiagnosticConsumer.h - Chain Diagnostic Clients ---*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #ifndef LLVM_CLANG_FRONTEND_CHAINEDDIAGNOSTICCONSUMER_H
11 #define LLVM_CLANG_FRONTEND_CHAINEDDIAGNOSTICCONSUMER_H
13 #include "clang/Basic/Diagnostic.h"
19 /// ChainedDiagnosticConsumer - Chain two diagnostic clients so that diagnostics
20 /// go to the first client and then the second. The first diagnostic client
21 /// should be the "primary" client, and will be used for computing whether the
22 /// diagnostics should be included in counts.
23 class ChainedDiagnosticConsumer : public DiagnosticConsumer {
24 virtual void anchor();
25 std::unique_ptr<DiagnosticConsumer> OwningPrimary;
26 DiagnosticConsumer *Primary;
27 std::unique_ptr<DiagnosticConsumer> Secondary;
30 ChainedDiagnosticConsumer(std::unique_ptr<DiagnosticConsumer> Primary,
31 std::unique_ptr<DiagnosticConsumer> Secondary)
32 : OwningPrimary(std::move(Primary)), Primary(OwningPrimary.get()),
33 Secondary(std::move(Secondary)) {}
35 /// \brief Construct without taking ownership of \c Primary.
36 ChainedDiagnosticConsumer(DiagnosticConsumer *Primary,
37 std::unique_ptr<DiagnosticConsumer> Secondary)
38 : Primary(Primary), Secondary(std::move(Secondary)) {}
40 void BeginSourceFile(const LangOptions &LO,
41 const Preprocessor *PP) override {
42 Primary->BeginSourceFile(LO, PP);
43 Secondary->BeginSourceFile(LO, PP);
46 void EndSourceFile() override {
47 Secondary->EndSourceFile();
48 Primary->EndSourceFile();
51 void finish() override {
56 bool IncludeInDiagnosticCounts() const override {
57 return Primary->IncludeInDiagnosticCounts();
60 void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
61 const Diagnostic &Info) override {
62 // Default implementation (Warnings/errors count).
63 DiagnosticConsumer::HandleDiagnostic(DiagLevel, Info);
65 Primary->HandleDiagnostic(DiagLevel, Info);
66 Secondary->HandleDiagnostic(DiagLevel, Info);
70 } // end namspace clang