//===- 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/SmallString.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 class FormatAdapter : public detail::format_adapter { protected: explicit FormatAdapter(T &&Item) : Item(std::forward(Item)) {} T Item; }; namespace detail { template class AlignAdapter final : public FormatAdapter { AlignStyle Where; size_t Amount; char Fill; public: AlignAdapter(T &&Item, AlignStyle Where, size_t Amount, char Fill) : FormatAdapter(std::forward(Item)), Where(Where), Amount(Amount), Fill(Fill) {} void format(llvm::raw_ostream &Stream, StringRef Style) { auto Adapter = detail::build_format_adapter(std::forward(this->Item)); FmtAlign(Adapter, Where, Amount, Fill).format(Stream, Style); } }; template class PadAdapter final : public FormatAdapter { size_t Left; size_t Right; public: PadAdapter(T &&Item, size_t Left, size_t Right) : FormatAdapter(std::forward(Item)), Left(Left), Right(Right) {} void format(llvm::raw_ostream &Stream, StringRef Style) { auto Adapter = detail::build_format_adapter(std::forward(this->Item)); Stream.indent(Left); Adapter.format(Stream, Style); Stream.indent(Right); } }; template class RepeatAdapter final : public FormatAdapter { size_t Count; public: RepeatAdapter(T &&Item, size_t Count) : FormatAdapter(std::forward(Item)), Count(Count) {} void format(llvm::raw_ostream &Stream, StringRef Style) { auto Adapter = detail::build_format_adapter(std::forward(this->Item)); for (size_t I = 0; I < Count; ++I) { Adapter.format(Stream, Style); } } }; class ErrorAdapter : public FormatAdapter { public: ErrorAdapter(Error &&Item) : FormatAdapter(std::move(Item)) {} ErrorAdapter(ErrorAdapter &&) = default; ~ErrorAdapter() { consumeError(std::move(Item)); } void format(llvm::raw_ostream &Stream, StringRef Style) { Stream << Item; } }; } template detail::AlignAdapter fmt_align(T &&Item, AlignStyle Where, size_t Amount, char Fill = ' ') { return detail::AlignAdapter(std::forward(Item), Where, Amount, Fill); } template detail::PadAdapter fmt_pad(T &&Item, size_t Left, size_t Right) { return detail::PadAdapter(std::forward(Item), Left, Right); } template detail::RepeatAdapter fmt_repeat(T &&Item, size_t Count) { return detail::RepeatAdapter(std::forward(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