1 //===- LinePrinter.h ------------------------------------------ *- C++ --*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #ifndef LLVM_TOOLS_LLVMPDBDUMP_LINEPRINTER_H
11 #define LLVM_TOOLS_LLVMPDBDUMP_LINEPRINTER_H
12 
13 #include "llvm/ADT/StringRef.h"
14 #include "llvm/ADT/Twine.h"
15 #include "llvm/Support/raw_ostream.h"
16 #include "llvm/Support/Regex.h"
17 
18 #include <list>
19 
20 namespace llvm {
21 
22 class LinePrinter {
23   friend class WithColor;
24 
25 public:
26   LinePrinter(int Indent, raw_ostream &Stream);
27 
28   void Indent();
29   void Unindent();
30   void NewLine();
31 
getStream()32   raw_ostream &getStream() { return OS; }
getIndentLevel()33   int getIndentLevel() const { return CurrentIndent; }
34 
35   bool IsTypeExcluded(llvm::StringRef TypeName);
36   bool IsSymbolExcluded(llvm::StringRef SymbolName);
37   bool IsCompilandExcluded(llvm::StringRef CompilandName);
38 
39 private:
40   template <typename Iter>
SetFilters(std::list<Regex> & List,Iter Begin,Iter End)41   void SetFilters(std::list<Regex> &List, Iter Begin, Iter End) {
42     List.clear();
43     for (; Begin != End; ++Begin)
44       List.emplace_back(StringRef(*Begin));
45   }
46 
47   raw_ostream &OS;
48   int IndentSpaces;
49   int CurrentIndent;
50 
51   std::list<Regex> ExcludeCompilandFilters;
52   std::list<Regex> ExcludeTypeFilters;
53   std::list<Regex> ExcludeSymbolFilters;
54 
55   std::list<Regex> IncludeCompilandFilters;
56   std::list<Regex> IncludeTypeFilters;
57   std::list<Regex> IncludeSymbolFilters;
58 };
59 
60 template <class T>
61 inline raw_ostream &operator<<(LinePrinter &Printer, const T &Item) {
62   Printer.getStream() << Item;
63   return Printer.getStream();
64 }
65 
66 enum class PDB_ColorItem {
67   None,
68   Address,
69   Type,
70   Keyword,
71   Offset,
72   Identifier,
73   Path,
74   SectionHeader,
75   LiteralValue,
76   Register,
77 };
78 
79 class WithColor {
80 public:
81   WithColor(LinePrinter &P, PDB_ColorItem C);
82   ~WithColor();
83 
get()84   raw_ostream &get() { return OS; }
85 
86 private:
87   void applyColor(PDB_ColorItem C);
88   raw_ostream &OS;
89 };
90 }
91 
92 #endif
93