Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 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: | ||
| 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 |