Home | History | Annotate | Download | only in llvm-pdbdump 
      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.push_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
     90