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(LangOpts: LO, PP);
42 Secondary->BeginSourceFile(LangOpts: 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