148 lines
3.5 KiB
C++
148 lines
3.5 KiB
C++
|
|
#include "DiffPrinter.h"
|
|
|
|
#include "llvm/Support/FormatAdapters.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::pdb;
|
|
|
|
namespace {
|
|
struct Colorize {
|
|
Colorize(raw_ostream &OS, DiffResult Result) : OS(OS) {
|
|
if (!OS.has_colors())
|
|
return;
|
|
switch (Result) {
|
|
case DiffResult::IDENTICAL:
|
|
OS.changeColor(raw_ostream::Colors::GREEN, false);
|
|
break;
|
|
case DiffResult::EQUIVALENT:
|
|
OS.changeColor(raw_ostream::Colors::YELLOW, true);
|
|
break;
|
|
default:
|
|
OS.changeColor(raw_ostream::Colors::RED, false);
|
|
break;
|
|
}
|
|
}
|
|
|
|
~Colorize() {
|
|
if (OS.has_colors())
|
|
OS.resetColor();
|
|
}
|
|
|
|
raw_ostream &OS;
|
|
};
|
|
}
|
|
|
|
DiffPrinter::DiffPrinter(uint32_t Indent, StringRef Header,
|
|
uint32_t PropertyWidth, uint32_t FieldWidth,
|
|
bool Result, bool Fields, raw_ostream &Stream)
|
|
: PrintResult(Result), PrintValues(Fields), Indent(Indent),
|
|
PropertyWidth(PropertyWidth), FieldWidth(FieldWidth), OS(Stream) {
|
|
printHeaderRow();
|
|
printFullRow(Header);
|
|
}
|
|
|
|
DiffPrinter::~DiffPrinter() {}
|
|
|
|
uint32_t DiffPrinter::tableWidth() const {
|
|
// `|`
|
|
uint32_t W = 1;
|
|
|
|
// `<width>|`
|
|
W += PropertyWidth + 1;
|
|
|
|
if (PrintResult) {
|
|
// ` I |`
|
|
W += 4;
|
|
}
|
|
|
|
if (PrintValues) {
|
|
// `<width>|<width>|`
|
|
W += 2 * (FieldWidth + 1);
|
|
}
|
|
return W;
|
|
}
|
|
|
|
void DiffPrinter::printFullRow(StringRef Text) {
|
|
newLine();
|
|
printValue(Text, DiffResult::UNSPECIFIED, AlignStyle::Center,
|
|
tableWidth() - 2, true);
|
|
printSeparatorRow();
|
|
}
|
|
|
|
void DiffPrinter::printSeparatorRow() {
|
|
newLine();
|
|
OS << formatv("{0}", fmt_repeat('-', PropertyWidth));
|
|
if (PrintResult) {
|
|
OS << '+';
|
|
OS << formatv("{0}", fmt_repeat('-', 3));
|
|
}
|
|
if (PrintValues) {
|
|
OS << '+';
|
|
OS << formatv("{0}", fmt_repeat('-', FieldWidth));
|
|
OS << '+';
|
|
OS << formatv("{0}", fmt_repeat('-', FieldWidth));
|
|
}
|
|
OS << '|';
|
|
}
|
|
|
|
void DiffPrinter::printHeaderRow() {
|
|
newLine('-');
|
|
OS << formatv("{0}", fmt_repeat('-', tableWidth() - 1));
|
|
}
|
|
|
|
void DiffPrinter::newLine(char InitialChar) {
|
|
OS << "\n";
|
|
OS.indent(Indent) << InitialChar;
|
|
}
|
|
|
|
void DiffPrinter::printExplicit(StringRef Property, DiffResult C,
|
|
StringRef Left, StringRef Right) {
|
|
newLine();
|
|
printValue(Property, DiffResult::UNSPECIFIED, AlignStyle::Right,
|
|
PropertyWidth, true);
|
|
printResult(C);
|
|
printValue(Left, C, AlignStyle::Center, FieldWidth, false);
|
|
printValue(Right, C, AlignStyle::Center, FieldWidth, false);
|
|
printSeparatorRow();
|
|
}
|
|
|
|
void DiffPrinter::printResult(DiffResult Result) {
|
|
if (!PrintResult)
|
|
return;
|
|
switch (Result) {
|
|
case DiffResult::DIFFERENT:
|
|
printValue("D", Result, AlignStyle::Center, 3, true);
|
|
break;
|
|
case DiffResult::EQUIVALENT:
|
|
printValue("E", Result, AlignStyle::Center, 3, true);
|
|
break;
|
|
case DiffResult::IDENTICAL:
|
|
printValue("I", Result, AlignStyle::Center, 3, true);
|
|
break;
|
|
case DiffResult::UNSPECIFIED:
|
|
printValue(" ", Result, AlignStyle::Center, 3, true);
|
|
break;
|
|
}
|
|
}
|
|
|
|
void DiffPrinter::printValue(StringRef Value, DiffResult C, AlignStyle Style,
|
|
uint32_t Width, bool Force) {
|
|
if (!Force && !PrintValues)
|
|
return;
|
|
|
|
if (Style == AlignStyle::Right)
|
|
--Width;
|
|
|
|
std::string FormattedItem =
|
|
formatv("{0}", fmt_align(Value, Style, Width)).str();
|
|
if (C != DiffResult::UNSPECIFIED) {
|
|
Colorize Color(OS, C);
|
|
OS << FormattedItem;
|
|
} else
|
|
OS << FormattedItem;
|
|
if (Style == AlignStyle::Right)
|
|
OS << ' ';
|
|
OS << '|';
|
|
}
|