forked from OSchip/llvm-project
60 lines
1.7 KiB
C++
60 lines
1.7 KiB
C++
//===- RenderingSupport.h - output stream rendering support functions ----===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_COV_RENDERINGSUPPORT_H
|
|
#define LLVM_COV_RENDERINGSUPPORT_H
|
|
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <utility>
|
|
|
|
namespace llvm {
|
|
|
|
/// \brief A helper class that resets the output stream's color if needed
|
|
/// when destroyed.
|
|
class ColoredRawOstream {
|
|
ColoredRawOstream(const ColoredRawOstream &OS) LLVM_DELETED_FUNCTION;
|
|
|
|
public:
|
|
raw_ostream &OS;
|
|
bool IsColorUsed;
|
|
|
|
ColoredRawOstream(raw_ostream &OS, bool IsColorUsed)
|
|
: OS(OS), IsColorUsed(IsColorUsed) {}
|
|
|
|
ColoredRawOstream(ColoredRawOstream &&Other)
|
|
: OS(Other.OS), IsColorUsed(Other.IsColorUsed) {
|
|
// Reset the other IsColorUsed so that the other object won't reset the
|
|
// color when destroyed.
|
|
Other.IsColorUsed = false;
|
|
}
|
|
|
|
~ColoredRawOstream() {
|
|
if (IsColorUsed)
|
|
OS.resetColor();
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
inline raw_ostream &operator<<(const ColoredRawOstream &OS, T &&Value) {
|
|
return OS.OS << std::forward<T>(Value);
|
|
}
|
|
|
|
/// \brief Change the color of the output stream if the `IsColorUsed` flag
|
|
/// is true. Returns an object that resets the color when destroyed.
|
|
inline ColoredRawOstream colored_ostream(raw_ostream &OS,
|
|
raw_ostream::Colors Color,
|
|
bool IsColorUsed = false) {
|
|
if (IsColorUsed)
|
|
OS.changeColor(Color);
|
|
return ColoredRawOstream(OS, IsColorUsed);
|
|
}
|
|
}
|
|
|
|
#endif // LLVM_COV_RENDERINGSUPPORT_H
|