aboutsummaryrefslogtreecommitdiffstats
path: root/tools/llvm-pdbdump/LinePrinter.cpp
blob: 6bbc403f5caf9aac7ef6b90daded4f06f1445df2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
//===- LinePrinter.cpp ------------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include "LinePrinter.h"

#include "llvm-pdbdump.h"

#include "llvm/Support/Regex.h"

#include <algorithm>

using namespace llvm;

LinePrinter::LinePrinter(int Indent, llvm::raw_ostream &Stream)
    : OS(Stream), IndentSpaces(Indent), CurrentIndent(0) {
  SetFilters(TypeFilters, opts::ExcludeTypes.begin(), opts::ExcludeTypes.end());
  SetFilters(SymbolFilters, opts::ExcludeSymbols.begin(),
             opts::ExcludeSymbols.end());
  SetFilters(CompilandFilters, opts::ExcludeCompilands.begin(),
             opts::ExcludeCompilands.end());
}

void LinePrinter::Indent() { CurrentIndent += IndentSpaces; }

void LinePrinter::Unindent() {
  CurrentIndent = std::max(0, CurrentIndent - IndentSpaces);
}

void LinePrinter::NewLine() {
  OS << "\n";
  OS.indent(CurrentIndent);
}

bool LinePrinter::IsTypeExcluded(llvm::StringRef TypeName) {
  if (TypeName.empty())
    return false;

  for (auto &Expr : TypeFilters) {
    if (Expr.match(TypeName))
      return true;
  }
  return false;
}

bool LinePrinter::IsSymbolExcluded(llvm::StringRef SymbolName) {
  if (SymbolName.empty())
    return false;

  for (auto &Expr : SymbolFilters) {
    if (Expr.match(SymbolName))
      return true;
  }
  return false;
}

bool LinePrinter::IsCompilandExcluded(llvm::StringRef CompilandName) {
  if (CompilandName.empty())
    return false;

  for (auto &Expr : CompilandFilters) {
    if (Expr.match(CompilandName))
      return true;
  }
  return false;
}

WithColor::WithColor(LinePrinter &P, PDB_ColorItem C) : OS(P.OS) {
  if (C == PDB_ColorItem::None)
    OS.resetColor();
  else {
    raw_ostream::Colors Color;
    bool Bold;
    translateColor(C, Color, Bold);
    OS.changeColor(Color, Bold);
  }
}

WithColor::~WithColor() { OS.resetColor(); }

void WithColor::translateColor(PDB_ColorItem C, raw_ostream::Colors &Color,
                               bool &Bold) const {
  switch (C) {
  case PDB_ColorItem::Address:
    Color = raw_ostream::YELLOW;
    Bold = true;
    return;
  case PDB_ColorItem::Keyword:
    Color = raw_ostream::MAGENTA;
    Bold = true;
    return;
  case PDB_ColorItem::Register:
  case PDB_ColorItem::Offset:
    Color = raw_ostream::YELLOW;
    Bold = false;
    return;
  case PDB_ColorItem::Type:
    Color = raw_ostream::CYAN;
    Bold = true;
    return;
  case PDB_ColorItem::Identifier:
    Color = raw_ostream::CYAN;
    Bold = false;
    return;
  case PDB_ColorItem::Path:
    Color = raw_ostream::CYAN;
    Bold = false;
    return;
  case PDB_ColorItem::SectionHeader:
    Color = raw_ostream::RED;
    Bold = true;
    return;
  case PDB_ColorItem::LiteralValue:
    Color = raw_ostream::GREEN;
    Bold = true;
  default:
    return;
  }
}