123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699 |
- #ifndef FMT_STD_H_
- #define FMT_STD_H_
- #include "format.h"
- #include "ostream.h"
- #ifndef FMT_MODULE
- # include <atomic>
- # include <bitset>
- # include <complex>
- # include <cstdlib>
- # include <exception>
- # include <memory>
- # include <thread>
- # include <type_traits>
- # include <typeinfo>
- # include <utility>
- # include <vector>
- # if FMT_CPLUSPLUS >= 201703L
- # if FMT_HAS_INCLUDE(<filesystem>)
- # include <filesystem>
- # endif
- # if FMT_HAS_INCLUDE(<variant>)
- # include <variant>
- # endif
- # if FMT_HAS_INCLUDE(<optional>)
- # include <optional>
- # endif
- # endif
- # if FMT_CPLUSPLUS > 201703L && FMT_HAS_INCLUDE(<source_location>)
- # include <source_location>
- # endif
- # if FMT_CPLUSPLUS > 202002L && FMT_HAS_INCLUDE(<expected>)
- # include <expected>
- # endif
- #endif
- #if FMT_HAS_INCLUDE(<version>)
- # include <version>
- #endif
- #if FMT_HAS_INCLUDE(<cxxabi.h>) || defined(__GLIBCXX__)
- # include <cxxabi.h>
- # ifndef __GABIXX_CXXABI_H__
- # define FMT_HAS_ABI_CXA_DEMANGLE
- # endif
- #endif
- #ifndef FMT_CPP_LIB_FILESYSTEM
- # ifdef __cpp_lib_filesystem
- # define FMT_CPP_LIB_FILESYSTEM __cpp_lib_filesystem
- # else
- # define FMT_CPP_LIB_FILESYSTEM 0
- # endif
- #endif
- #ifndef FMT_CPP_LIB_VARIANT
- # ifdef __cpp_lib_variant
- # define FMT_CPP_LIB_VARIANT __cpp_lib_variant
- # else
- # define FMT_CPP_LIB_VARIANT 0
- # endif
- #endif
- #if FMT_CPP_LIB_FILESYSTEM
- FMT_BEGIN_NAMESPACE
- namespace detail {
- template <typename Char, typename PathChar>
- auto get_path_string(const std::filesystem::path& p,
- const std::basic_string<PathChar>& native) {
- if constexpr (std::is_same_v<Char, char> && std::is_same_v<PathChar, wchar_t>)
- return to_utf8<wchar_t>(native, to_utf8_error_policy::replace);
- else
- return p.string<Char>();
- }
- template <typename Char, typename PathChar>
- void write_escaped_path(basic_memory_buffer<Char>& quoted,
- const std::filesystem::path& p,
- const std::basic_string<PathChar>& native) {
- if constexpr (std::is_same_v<Char, char> &&
- std::is_same_v<PathChar, wchar_t>) {
- auto buf = basic_memory_buffer<wchar_t>();
- write_escaped_string<wchar_t>(std::back_inserter(buf), native);
- bool valid = to_utf8<wchar_t>::convert(quoted, {buf.data(), buf.size()});
- FMT_ASSERT(valid, "invalid utf16");
- } else if constexpr (std::is_same_v<Char, PathChar>) {
- write_escaped_string<std::filesystem::path::value_type>(
- std::back_inserter(quoted), native);
- } else {
- write_escaped_string<Char>(std::back_inserter(quoted), p.string<Char>());
- }
- }
- }
- FMT_EXPORT
- template <typename Char> struct formatter<std::filesystem::path, Char> {
- private:
- format_specs specs_;
- detail::arg_ref<Char> width_ref_;
- bool debug_ = false;
- char path_type_ = 0;
- public:
- FMT_CONSTEXPR void set_debug_format(bool set = true) { debug_ = set; }
- template <typename ParseContext> FMT_CONSTEXPR auto parse(ParseContext& ctx) {
- auto it = ctx.begin(), end = ctx.end();
- if (it == end) return it;
- it = detail::parse_align(it, end, specs_);
- if (it == end) return it;
- it = detail::parse_dynamic_spec(it, end, specs_.width, width_ref_, ctx);
- if (it != end && *it == '?') {
- debug_ = true;
- ++it;
- }
- if (it != end && (*it == 'g')) path_type_ = detail::to_ascii(*it++);
- return it;
- }
- template <typename FormatContext>
- auto format(const std::filesystem::path& p, FormatContext& ctx) const {
- auto specs = specs_;
- auto path_string =
- !path_type_ ? p.native()
- : p.generic_string<std::filesystem::path::value_type>();
- detail::handle_dynamic_spec<detail::width_checker>(specs.width, width_ref_,
- ctx);
- if (!debug_) {
- auto s = detail::get_path_string<Char>(p, path_string);
- return detail::write(ctx.out(), basic_string_view<Char>(s), specs);
- }
- auto quoted = basic_memory_buffer<Char>();
- detail::write_escaped_path(quoted, p, path_string);
- return detail::write(ctx.out(),
- basic_string_view<Char>(quoted.data(), quoted.size()),
- specs);
- }
- };
- class path : public std::filesystem::path {
- public:
- auto display_string() const -> std::string {
- const std::filesystem::path& base = *this;
- return fmt::format(FMT_STRING("{}"), base);
- }
- auto system_string() const -> std::string { return string(); }
- auto generic_display_string() const -> std::string {
- const std::filesystem::path& base = *this;
- return fmt::format(FMT_STRING("{:g}"), base);
- }
- auto generic_system_string() const -> std::string { return generic_string(); }
- };
- FMT_END_NAMESPACE
- #endif
- FMT_BEGIN_NAMESPACE
- FMT_EXPORT
- template <std::size_t N, typename Char>
- struct formatter<std::bitset<N>, Char> : nested_formatter<string_view> {
- private:
-
- struct writer {
- const std::bitset<N>& bs;
- template <typename OutputIt>
- FMT_CONSTEXPR auto operator()(OutputIt out) -> OutputIt {
- for (auto pos = N; pos > 0; --pos) {
- out = detail::write<Char>(out, bs[pos - 1] ? Char('1') : Char('0'));
- }
- return out;
- }
- };
- public:
- template <typename FormatContext>
- auto format(const std::bitset<N>& bs, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- return write_padded(ctx, writer{bs});
- }
- };
- FMT_EXPORT
- template <typename Char>
- struct formatter<std::thread::id, Char> : basic_ostream_formatter<Char> {};
- FMT_END_NAMESPACE
- #ifdef __cpp_lib_optional
- FMT_BEGIN_NAMESPACE
- FMT_EXPORT
- template <typename T, typename Char>
- struct formatter<std::optional<T>, Char,
- std::enable_if_t<is_formattable<T, Char>::value>> {
- private:
- formatter<T, Char> underlying_;
- static constexpr basic_string_view<Char> optional =
- detail::string_literal<Char, 'o', 'p', 't', 'i', 'o', 'n', 'a', 'l',
- '('>{};
- static constexpr basic_string_view<Char> none =
- detail::string_literal<Char, 'n', 'o', 'n', 'e'>{};
- template <class U>
- FMT_CONSTEXPR static auto maybe_set_debug_format(U& u, bool set)
- -> decltype(u.set_debug_format(set)) {
- u.set_debug_format(set);
- }
- template <class U>
- FMT_CONSTEXPR static void maybe_set_debug_format(U&, ...) {}
- public:
- template <typename ParseContext> FMT_CONSTEXPR auto parse(ParseContext& ctx) {
- maybe_set_debug_format(underlying_, true);
- return underlying_.parse(ctx);
- }
- template <typename FormatContext>
- auto format(const std::optional<T>& opt, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- if (!opt) return detail::write<Char>(ctx.out(), none);
- auto out = ctx.out();
- out = detail::write<Char>(out, optional);
- ctx.advance_to(out);
- out = underlying_.format(*opt, ctx);
- return detail::write(out, ')');
- }
- };
- FMT_END_NAMESPACE
- #endif
- #if defined(__cpp_lib_expected) || FMT_CPP_LIB_VARIANT
- FMT_BEGIN_NAMESPACE
- namespace detail {
- template <typename Char, typename OutputIt, typename T>
- auto write_escaped_alternative(OutputIt out, const T& v) -> OutputIt {
- if constexpr (has_to_string_view<T>::value)
- return write_escaped_string<Char>(out, detail::to_string_view(v));
- if constexpr (std::is_same_v<T, Char>) return write_escaped_char(out, v);
- return write<Char>(out, v);
- }
- }
- FMT_END_NAMESPACE
- #endif
- #ifdef __cpp_lib_expected
- FMT_BEGIN_NAMESPACE
- FMT_EXPORT
- template <typename T, typename E, typename Char>
- struct formatter<std::expected<T, E>, Char,
- std::enable_if_t<is_formattable<T, Char>::value &&
- is_formattable<E, Char>::value>> {
- template <typename ParseContext>
- FMT_CONSTEXPR auto parse(ParseContext& ctx) -> decltype(ctx.begin()) {
- return ctx.begin();
- }
- template <typename FormatContext>
- auto format(const std::expected<T, E>& value, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- auto out = ctx.out();
- if (value.has_value()) {
- out = detail::write<Char>(out, "expected(");
- out = detail::write_escaped_alternative<Char>(out, *value);
- } else {
- out = detail::write<Char>(out, "unexpected(");
- out = detail::write_escaped_alternative<Char>(out, value.error());
- }
- *out++ = ')';
- return out;
- }
- };
- FMT_END_NAMESPACE
- #endif
- #ifdef __cpp_lib_source_location
- FMT_BEGIN_NAMESPACE
- FMT_EXPORT
- template <> struct formatter<std::source_location> {
- template <typename ParseContext> FMT_CONSTEXPR auto parse(ParseContext& ctx) {
- return ctx.begin();
- }
- template <typename FormatContext>
- auto format(const std::source_location& loc, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- auto out = ctx.out();
- out = detail::write(out, loc.file_name());
- out = detail::write(out, ':');
- out = detail::write<char>(out, loc.line());
- out = detail::write(out, ':');
- out = detail::write<char>(out, loc.column());
- out = detail::write(out, ": ");
- out = detail::write(out, loc.function_name());
- return out;
- }
- };
- FMT_END_NAMESPACE
- #endif
- #if FMT_CPP_LIB_VARIANT
- FMT_BEGIN_NAMESPACE
- namespace detail {
- template <typename T>
- using variant_index_sequence =
- std::make_index_sequence<std::variant_size<T>::value>;
- template <typename> struct is_variant_like_ : std::false_type {};
- template <typename... Types>
- struct is_variant_like_<std::variant<Types...>> : std::true_type {};
- template <typename T, typename C> class is_variant_formattable_ {
- template <std::size_t... Is>
- static std::conjunction<
- is_formattable<std::variant_alternative_t<Is, T>, C>...>
- check(std::index_sequence<Is...>);
- public:
- static constexpr const bool value =
- decltype(check(variant_index_sequence<T>{}))::value;
- };
- }
- template <typename T> struct is_variant_like {
- static constexpr const bool value = detail::is_variant_like_<T>::value;
- };
- template <typename T, typename C> struct is_variant_formattable {
- static constexpr const bool value =
- detail::is_variant_formattable_<T, C>::value;
- };
- FMT_EXPORT
- template <typename Char> struct formatter<std::monostate, Char> {
- template <typename ParseContext>
- FMT_CONSTEXPR auto parse(ParseContext& ctx) -> decltype(ctx.begin()) {
- return ctx.begin();
- }
- template <typename FormatContext>
- auto format(const std::monostate&, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- return detail::write<Char>(ctx.out(), "monostate");
- }
- };
- FMT_EXPORT
- template <typename Variant, typename Char>
- struct formatter<
- Variant, Char,
- std::enable_if_t<std::conjunction_v<
- is_variant_like<Variant>, is_variant_formattable<Variant, Char>>>> {
- template <typename ParseContext>
- FMT_CONSTEXPR auto parse(ParseContext& ctx) -> decltype(ctx.begin()) {
- return ctx.begin();
- }
- template <typename FormatContext>
- auto format(const Variant& value, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- auto out = ctx.out();
- out = detail::write<Char>(out, "variant(");
- FMT_TRY {
- std::visit(
- [&](const auto& v) {
- out = detail::write_escaped_alternative<Char>(out, v);
- },
- value);
- }
- FMT_CATCH(const std::bad_variant_access&) {
- detail::write<Char>(out, "valueless by exception");
- }
- *out++ = ')';
- return out;
- }
- };
- FMT_END_NAMESPACE
- #endif
- FMT_BEGIN_NAMESPACE
- FMT_EXPORT
- template <typename Char> struct formatter<std::error_code, Char> {
- template <typename ParseContext>
- FMT_CONSTEXPR auto parse(ParseContext& ctx) -> decltype(ctx.begin()) {
- return ctx.begin();
- }
- template <typename FormatContext>
- FMT_CONSTEXPR auto format(const std::error_code& ec, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- auto out = ctx.out();
- out = detail::write_bytes<Char>(out, ec.category().name(), format_specs());
- out = detail::write<Char>(out, Char(':'));
- out = detail::write<Char>(out, ec.value());
- return out;
- }
- };
- #if FMT_USE_RTTI
- namespace detail {
- template <typename Char, typename OutputIt>
- auto write_demangled_name(OutputIt out, const std::type_info& ti) -> OutputIt {
- # ifdef FMT_HAS_ABI_CXA_DEMANGLE
- int status = 0;
- std::size_t size = 0;
- std::unique_ptr<char, void (*)(void*)> demangled_name_ptr(
- abi::__cxa_demangle(ti.name(), nullptr, &size, &status), &std::free);
- string_view demangled_name_view;
- if (demangled_name_ptr) {
- demangled_name_view = demangled_name_ptr.get();
-
-
-
-
-
-
-
- if (demangled_name_view.starts_with("std::")) {
- char* begin = demangled_name_ptr.get();
- char* to = begin + 5;
- for (char *from = to, *end = begin + demangled_name_view.size();
- from < end;) {
-
- if (from[0] == '_' && from[1] == '_') {
- char* next = from + 1;
- while (next < end && *next != ':') next++;
- if (next[0] == ':' && next[1] == ':') {
- from = next + 2;
- continue;
- }
- }
- *to++ = *from++;
- }
- demangled_name_view = {begin, detail::to_unsigned(to - begin)};
- }
- } else {
- demangled_name_view = string_view(ti.name());
- }
- return detail::write_bytes<Char>(out, demangled_name_view);
- # elif FMT_MSC_VERSION
- const string_view demangled_name(ti.name());
- for (std::size_t i = 0; i < demangled_name.size(); ++i) {
- auto sub = demangled_name;
- sub.remove_prefix(i);
- if (sub.starts_with("enum ")) {
- i += 4;
- continue;
- }
- if (sub.starts_with("class ") || sub.starts_with("union ")) {
- i += 5;
- continue;
- }
- if (sub.starts_with("struct ")) {
- i += 6;
- continue;
- }
- if (*sub.begin() != ' ') *out++ = *sub.begin();
- }
- return out;
- # else
- return detail::write_bytes<Char>(out, string_view(ti.name()));
- # endif
- }
- }
- FMT_EXPORT
- template <typename Char>
- struct formatter<std::type_info, Char
- > {
- public:
- FMT_CONSTEXPR auto parse(basic_format_parse_context<Char>& ctx)
- -> decltype(ctx.begin()) {
- return ctx.begin();
- }
- template <typename Context>
- auto format(const std::type_info& ti, Context& ctx) const
- -> decltype(ctx.out()) {
- return detail::write_demangled_name<Char>(ctx.out(), ti);
- }
- };
- #endif
- FMT_EXPORT
- template <typename T, typename Char>
- struct formatter<
- T, Char,
- typename std::enable_if<std::is_base_of<std::exception, T>::value>::type> {
- private:
- bool with_typename_ = false;
- public:
- FMT_CONSTEXPR auto parse(basic_format_parse_context<Char>& ctx)
- -> decltype(ctx.begin()) {
- auto it = ctx.begin();
- auto end = ctx.end();
- if (it == end || *it == '}') return it;
- if (*it == 't') {
- ++it;
- with_typename_ = FMT_USE_RTTI != 0;
- }
- return it;
- }
- template <typename Context>
- auto format(const std::exception& ex, Context& ctx) const
- -> decltype(ctx.out()) {
- auto out = ctx.out();
- #if FMT_USE_RTTI
- if (with_typename_) {
- out = detail::write_demangled_name<Char>(out, typeid(ex));
- *out++ = ':';
- *out++ = ' ';
- }
- #endif
- return detail::write_bytes<Char>(out, string_view(ex.what()));
- }
- };
- namespace detail {
- template <typename T, typename Enable = void>
- struct has_flip : std::false_type {};
- template <typename T>
- struct has_flip<T, void_t<decltype(std::declval<T>().flip())>>
- : std::true_type {};
- template <typename T> struct is_bit_reference_like {
- static constexpr const bool value =
- std::is_convertible<T, bool>::value &&
- std::is_nothrow_assignable<T, bool>::value && has_flip<T>::value;
- };
- #ifdef _LIBCPP_VERSION
- template <typename C>
- struct is_bit_reference_like<std::__bit_const_reference<C>> {
- static constexpr const bool value = true;
- };
- #endif
- }
- FMT_EXPORT
- template <typename BitRef, typename Char>
- struct formatter<BitRef, Char,
- enable_if_t<detail::is_bit_reference_like<BitRef>::value>>
- : formatter<bool, Char> {
- template <typename FormatContext>
- FMT_CONSTEXPR auto format(const BitRef& v, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- return formatter<bool, Char>::format(v, ctx);
- }
- };
- template <typename T, typename Deleter>
- auto ptr(const std::unique_ptr<T, Deleter>& p) -> const void* {
- return p.get();
- }
- template <typename T> auto ptr(const std::shared_ptr<T>& p) -> const void* {
- return p.get();
- }
- FMT_EXPORT
- template <typename T, typename Char>
- struct formatter<std::atomic<T>, Char,
- enable_if_t<is_formattable<T, Char>::value>>
- : formatter<T, Char> {
- template <typename FormatContext>
- auto format(const std::atomic<T>& v, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- return formatter<T, Char>::format(v.load(), ctx);
- }
- };
- #ifdef __cpp_lib_atomic_flag_test
- FMT_EXPORT
- template <typename Char>
- struct formatter<std::atomic_flag, Char> : formatter<bool, Char> {
- template <typename FormatContext>
- auto format(const std::atomic_flag& v, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- return formatter<bool, Char>::format(v.test(), ctx);
- }
- };
- #endif
- FMT_EXPORT
- template <typename T, typename Char> struct formatter<std::complex<T>, Char> {
- private:
- detail::dynamic_format_specs<Char> specs_;
- template <typename FormatContext, typename OutputIt>
- FMT_CONSTEXPR auto do_format(const std::complex<T>& c,
- detail::dynamic_format_specs<Char>& specs,
- FormatContext& ctx, OutputIt out) const
- -> OutputIt {
- if (c.real() != 0) {
- *out++ = Char('(');
- out = detail::write<Char>(out, c.real(), specs, ctx.locale());
- specs.sign = sign::plus;
- out = detail::write<Char>(out, c.imag(), specs, ctx.locale());
- if (!detail::isfinite(c.imag())) *out++ = Char(' ');
- *out++ = Char('i');
- *out++ = Char(')');
- return out;
- }
- out = detail::write<Char>(out, c.imag(), specs, ctx.locale());
- if (!detail::isfinite(c.imag())) *out++ = Char(' ');
- *out++ = Char('i');
- return out;
- }
- public:
- FMT_CONSTEXPR auto parse(basic_format_parse_context<Char>& ctx)
- -> decltype(ctx.begin()) {
- if (ctx.begin() == ctx.end() || *ctx.begin() == '}') return ctx.begin();
- return parse_format_specs(ctx.begin(), ctx.end(), specs_, ctx,
- detail::type_constant<T, Char>::value);
- }
- template <typename FormatContext>
- auto format(const std::complex<T>& c, FormatContext& ctx) const
- -> decltype(ctx.out()) {
- auto specs = specs_;
- if (specs.width_ref.kind != detail::arg_id_kind::none ||
- specs.precision_ref.kind != detail::arg_id_kind::none) {
- detail::handle_dynamic_spec<detail::width_checker>(specs.width,
- specs.width_ref, ctx);
- detail::handle_dynamic_spec<detail::precision_checker>(
- specs.precision, specs.precision_ref, ctx);
- }
- if (specs.width == 0) return do_format(c, specs, ctx, ctx.out());
- auto buf = basic_memory_buffer<Char>();
- auto outer_specs = format_specs();
- outer_specs.width = specs.width;
- outer_specs.fill = specs.fill;
- outer_specs.align = specs.align;
- specs.width = 0;
- specs.fill = {};
- specs.align = align::none;
- do_format(c, specs, ctx, basic_appender<Char>(buf));
- return detail::write<Char>(ctx.out(),
- basic_string_view<Char>(buf.data(), buf.size()),
- outer_specs);
- }
- };
- FMT_END_NAMESPACE
- #endif
|