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: ChainedDiagnosticConsumer(std::unique_ptr<DiagnosticConsumer> Primary,std::unique_ptr<DiagnosticConsumer> Secondary)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. ChainedDiagnosticConsumer(DiagnosticConsumer * Primary,std::unique_ptr<DiagnosticConsumer> Secondary)35 ChainedDiagnosticConsumer(DiagnosticConsumer *Primary, 36 std::unique_ptr<DiagnosticConsumer> Secondary) 37 : Primary(Primary), Secondary(std::move(Secondary)) {} 38 BeginSourceFile(const LangOptions & LO,const Preprocessor * PP)39 void BeginSourceFile(const LangOptions &LO, 40 const Preprocessor *PP) override { 41 Primary->BeginSourceFile(LO, PP); 42 Secondary->BeginSourceFile(LO, PP); 43 } 44 EndSourceFile()45 void EndSourceFile() override { 46 Secondary->EndSourceFile(); 47 Primary->EndSourceFile(); 48 } 49 finish()50 void finish() override { 51 Secondary->finish(); 52 Primary->finish(); 53 } 54 IncludeInDiagnosticCounts()55 bool IncludeInDiagnosticCounts() const override { 56 return Primary->IncludeInDiagnosticCounts(); 57 } 58 HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,const Diagnostic & Info)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 72