FormatAdapters.h revision 321369 
1//===- FormatAdapters.h - Formatters for common LLVM types -----*- 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_SUPPORT_FORMATADAPTERS_H
11#define LLVM_SUPPORT_FORMATADAPTERS_H
12
13#include "llvm/ADT/SmallString.h"
14#include "llvm/ADT/StringRef.h"
15#include "llvm/Support/FormatCommon.h"
16#include "llvm/Support/FormatVariadicDetails.h"
17#include "llvm/Support/raw_ostream.h"
18
19namespace llvm {
20template <typename T> class FormatAdapter : public detail::format_adapter {
21protected:
22  explicit FormatAdapter(T &&Item) : Item(Item) {}
23
24  T Item;
25};
26
27namespace detail {
28template <typename T> class AlignAdapter final : public FormatAdapter<T> {
29  AlignStyle Where;
30  size_t Amount;
31  char Fill;
32
33public:
34  AlignAdapter(T &&Item, AlignStyle Where, size_t Amount, char Fill)
35      : FormatAdapter<T>(std::forward<T>(Item)), Where(Where), Amount(Amount),
36        Fill(Fill) {}
37
38  void format(llvm::raw_ostream &Stream, StringRef Style) {
39    auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));
40    FmtAlign(Adapter, Where, Amount, Fill).format(Stream, Style);
41  }
42};
43
44template <typename T> class PadAdapter final : public FormatAdapter<T> {
45  size_t Left;
46  size_t Right;
47
48public:
49  PadAdapter(T &&Item, size_t Left, size_t Right)
50      : FormatAdapter<T>(std::forward<T>(Item)), Left(Left), Right(Right) {}
51
52  void format(llvm::raw_ostream &Stream, StringRef Style) {
53    auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));
54    Stream.indent(Left);
55    Adapter.format(Stream, Style);
56    Stream.indent(Right);
57  }
58};
59
60template <typename T> class RepeatAdapter final : public FormatAdapter<T> {
61  size_t Count;
62
63public:
64  RepeatAdapter(T &&Item, size_t Count)
65      : FormatAdapter<T>(std::forward<T>(Item)), Count(Count) {}
66
67  void format(llvm::raw_ostream &Stream, StringRef Style) {
68    auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));
69    for (size_t I = 0; I < Count; ++I) {
70      Adapter.format(Stream, Style);
71    }
72  }
73};
74}
75
76template <typename T>
77detail::AlignAdapter<T> fmt_align(T &&Item, AlignStyle Where, size_t Amount,
78                                  char Fill = ' ') {
79  return detail::AlignAdapter<T>(std::forward<T>(Item), Where, Amount, Fill);
80}
81
82template <typename T>
83detail::PadAdapter<T> fmt_pad(T &&Item, size_t Left, size_t Right) {
84  return detail::PadAdapter<T>(std::forward<T>(Item), Left, Right);
85}
86
87template <typename T>
88detail::RepeatAdapter<T> fmt_repeat(T &&Item, size_t Count) {
89  return detail::RepeatAdapter<T>(std::forward<T>(Item), Count);
90}
91}
92
93#endif
94