//===- ChainedDiagnosticConsumer.h - Chain Diagnostic Clients ---*- C++ -*-===//
 
//
 
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 
// See https://llvm.org/LICENSE.txt for license information.
 
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
//
 
//===----------------------------------------------------------------------===//
 
 
 
#ifndef LLVM_CLANG_FRONTEND_CHAINEDDIAGNOSTICCONSUMER_H
 
#define LLVM_CLANG_FRONTEND_CHAINEDDIAGNOSTICCONSUMER_H
 
 
 
#include "clang/Basic/Diagnostic.h"
 
#include <memory>
 
 
 
namespace clang {
 
class LangOptions;
 
 
 
/// ChainedDiagnosticConsumer - Chain two diagnostic clients so that diagnostics
 
/// go to the first client and then the second. The first diagnostic client
 
/// should be the "primary" client, and will be used for computing whether the
 
/// diagnostics should be included in counts.
 
class ChainedDiagnosticConsumer : public DiagnosticConsumer {
 
  virtual void anchor();
 
  std::unique_ptr<DiagnosticConsumer> OwningPrimary;
 
  DiagnosticConsumer *Primary;
 
  std::unique_ptr<DiagnosticConsumer> Secondary;
 
 
 
public:
 
  ChainedDiagnosticConsumer(std::unique_ptr<DiagnosticConsumer> Primary,
 
                            std::unique_ptr<DiagnosticConsumer> Secondary)
 
      : OwningPrimary(std::move(Primary)), Primary(OwningPrimary.get()),
 
        Secondary(std::move(Secondary)) {}
 
 
 
  /// Construct without taking ownership of \c Primary.
 
  ChainedDiagnosticConsumer(DiagnosticConsumer *Primary,
 
                            std::unique_ptr<DiagnosticConsumer> Secondary)
 
      : Primary(Primary), Secondary(std::move(Secondary)) {}
 
 
 
  void BeginSourceFile(const LangOptions &LO,
 
                       const Preprocessor *PP) override {
 
    Primary->BeginSourceFile(LO, PP);
 
    Secondary->BeginSourceFile(LO, PP);
 
  }
 
 
 
  void EndSourceFile() override {
 
    Secondary->EndSourceFile();
 
    Primary->EndSourceFile();
 
  }
 
 
 
  void finish() override {
 
    Secondary->finish();
 
    Primary->finish();
 
  }
 
 
 
  bool IncludeInDiagnosticCounts() const override {
 
    return Primary->IncludeInDiagnosticCounts();
 
  }
 
 
 
  void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
 
                        const Diagnostic &Info) override {
 
    // Default implementation (Warnings/errors count).
 
    DiagnosticConsumer::HandleDiagnostic(DiagLevel, Info);
 
 
 
    Primary->HandleDiagnostic(DiagLevel, Info);
 
    Secondary->HandleDiagnostic(DiagLevel, Info);
 
  }
 
};
 
 
 
} // end namspace clang
 
 
 
#endif