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