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