tools/llvm-pdbdump/LinePrinter.h

   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
  32   raw_ostream &getStream() { return OS; }
  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>
  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> CompilandFilters;
  52   std::list<Regex> TypeFilters;
  53   std::list<Regex> SymbolFilters;
  54 };
  55
  56 template <class T>
  57 inline raw_ostream &operator<<(LinePrinter &Printer, const T &Item) {
  58   Printer.getStream() << Item;
  59   return Printer.getStream();
  60 }
  61
  62 enum class PDB_ColorItem {
  63   None,
  64   Address,
  65   Type,
  66   Keyword,
  67   Offset,
  68   Identifier,
  69   Path,
  70   SectionHeader,
  71   LiteralValue,
  72   Register,
  73 };
  74
  75 class WithColor {
  76 public:
  77   WithColor(LinePrinter &P, PDB_ColorItem C);
  78   ~WithColor();
  79
  80   raw_ostream &get() { return OS; }
  81
  82 private:
  83   void translateColor(PDB_ColorItem C, raw_ostream::Colors &Color,
  84                       bool &Bold) const;
  85   raw_ostream &OS;
  86 };
  87 }
  88
  89 #endif