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 15namespace clang { 16class 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. 22class ChainedDiagnosticConsumer : public DiagnosticConsumer { 23 virtual void anchor(); 24 std::unique_ptr<DiagnosticConsumer> OwningPrimary; 25 DiagnosticConsumer *Primary; 26 std::unique_ptr<DiagnosticConsumer> Secondary; 27 28public: 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 72