123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- #pragma once
- #ifdef __GNUC__
- #pragma GCC diagnostic push
- #pragma GCC diagnostic ignored "-Wunused-parameter"
- #endif
- //===- FormatAdapters.h - Formatters for common LLVM types -----*- C++ -*-===//
- //
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
- // See https://llvm.org/LICENSE.txt for license information.
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- //
- //===----------------------------------------------------------------------===//
- #ifndef LLVM_SUPPORT_FORMATADAPTERS_H
- #define LLVM_SUPPORT_FORMATADAPTERS_H
- #include "llvm/ADT/StringRef.h"
- #include "llvm/Support/Error.h"
- #include "llvm/Support/FormatCommon.h"
- #include "llvm/Support/FormatVariadicDetails.h"
- #include "llvm/Support/raw_ostream.h"
- namespace llvm {
- template <typename T> class FormatAdapter : public detail::format_adapter {
- protected:
- explicit FormatAdapter(T &&Item) : Item(std::forward<T>(Item)) {}
- T Item;
- };
- namespace detail {
- template <typename T> class AlignAdapter final : public FormatAdapter<T> {
- AlignStyle Where;
- size_t Amount;
- char Fill;
- public:
- AlignAdapter(T &&Item, AlignStyle Where, size_t Amount, char Fill)
- : FormatAdapter<T>(std::forward<T>(Item)), Where(Where), Amount(Amount),
- Fill(Fill) {}
- void format(llvm::raw_ostream &Stream, StringRef Style) override {
- auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));
- FmtAlign(Adapter, Where, Amount, Fill).format(Stream, Style);
- }
- };
- template <typename T> class PadAdapter final : public FormatAdapter<T> {
- size_t Left;
- size_t Right;
- public:
- PadAdapter(T &&Item, size_t Left, size_t Right)
- : FormatAdapter<T>(std::forward<T>(Item)), Left(Left), Right(Right) {}
- void format(llvm::raw_ostream &Stream, StringRef Style) override {
- auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));
- Stream.indent(Left);
- Adapter.format(Stream, Style);
- Stream.indent(Right);
- }
- };
- template <typename T> class RepeatAdapter final : public FormatAdapter<T> {
- size_t Count;
- public:
- RepeatAdapter(T &&Item, size_t Count)
- : FormatAdapter<T>(std::forward<T>(Item)), Count(Count) {}
- void format(llvm::raw_ostream &Stream, StringRef Style) override {
- auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));
- for (size_t I = 0; I < Count; ++I) {
- Adapter.format(Stream, Style);
- }
- }
- };
- class ErrorAdapter : public FormatAdapter<Error> {
- public:
- ErrorAdapter(Error &&Item) : FormatAdapter(std::move(Item)) {}
- ErrorAdapter(ErrorAdapter &&) = default;
- ~ErrorAdapter() { consumeError(std::move(Item)); }
- void format(llvm::raw_ostream &Stream, StringRef Style) override {
- Stream << Item;
- }
- };
- }
- template <typename T>
- detail::AlignAdapter<T> fmt_align(T &&Item, AlignStyle Where, size_t Amount,
- char Fill = ' ') {
- return detail::AlignAdapter<T>(std::forward<T>(Item), Where, Amount, Fill);
- }
- template <typename T>
- detail::PadAdapter<T> fmt_pad(T &&Item, size_t Left, size_t Right) {
- return detail::PadAdapter<T>(std::forward<T>(Item), Left, Right);
- }
- template <typename T>
- detail::RepeatAdapter<T> fmt_repeat(T &&Item, size_t Count) {
- return detail::RepeatAdapter<T>(std::forward<T>(Item), Count);
- }
- // llvm::Error values must be consumed before being destroyed.
- // Wrapping an error in fmt_consume explicitly indicates that the formatv_object
- // should take ownership and consume it.
- inline detail::ErrorAdapter fmt_consume(Error &&Item) {
- return detail::ErrorAdapter(std::move(Item));
- }
- }
- #endif
- #ifdef __GNUC__
- #pragma GCC diagnostic pop
- #endif
|