260 lines
		
	
	
		
			9.7 KiB
		
	
	
	
		
			C
		
	
	
	
		
		
			
		
	
	
			260 lines
		
	
	
		
			9.7 KiB
		
	
	
	
		
			C
		
	
	
	
|  | // Formatting library for C++ - optional wchar_t and exotic character support
 | ||
|  | //
 | ||
|  | // Copyright (c) 2012 - present, Victor Zverovich
 | ||
|  | // All rights reserved.
 | ||
|  | //
 | ||
|  | // For the license information refer to format.h.
 | ||
|  | 
 | ||
|  | #ifndef FMT_XCHAR_H_
 | ||
|  | #define FMT_XCHAR_H_
 | ||
|  | 
 | ||
|  | #include <cwchar>
 | ||
|  | 
 | ||
|  | #include "format.h"
 | ||
|  | 
 | ||
|  | #ifndef FMT_STATIC_THOUSANDS_SEPARATOR
 | ||
|  | #  include <locale>
 | ||
|  | #endif
 | ||
|  | 
 | ||
|  | FMT_BEGIN_NAMESPACE | ||
|  | namespace detail { | ||
|  | 
 | ||
|  | template <typename T> | ||
|  | using is_exotic_char = bool_constant<!std::is_same<T, char>::value>; | ||
|  | 
 | ||
|  | inline auto write_loc(std::back_insert_iterator<detail::buffer<wchar_t>> out, | ||
|  |                       loc_value value, const format_specs<wchar_t>& specs, | ||
|  |                       locale_ref loc) -> bool { | ||
|  | #ifndef FMT_STATIC_THOUSANDS_SEPARATOR
 | ||
|  |   auto& numpunct = | ||
|  |       std::use_facet<std::numpunct<wchar_t>>(loc.get<std::locale>()); | ||
|  |   auto separator = std::wstring(); | ||
|  |   auto grouping = numpunct.grouping(); | ||
|  |   if (!grouping.empty()) separator = std::wstring(1, numpunct.thousands_sep()); | ||
|  |   return value.visit(loc_writer<wchar_t>{out, specs, separator, grouping, {}}); | ||
|  | #endif
 | ||
|  |   return false; | ||
|  | } | ||
|  | }  // namespace detail
 | ||
|  | 
 | ||
|  | FMT_BEGIN_EXPORT | ||
|  | 
 | ||
|  | using wstring_view = basic_string_view<wchar_t>; | ||
|  | using wformat_parse_context = basic_format_parse_context<wchar_t>; | ||
|  | using wformat_context = buffer_context<wchar_t>; | ||
|  | using wformat_args = basic_format_args<wformat_context>; | ||
|  | using wmemory_buffer = basic_memory_buffer<wchar_t>; | ||
|  | 
 | ||
|  | #if FMT_GCC_VERSION && FMT_GCC_VERSION < 409
 | ||
|  | // Workaround broken conversion on older gcc.
 | ||
|  | template <typename... Args> using wformat_string = wstring_view; | ||
|  | inline auto runtime(wstring_view s) -> wstring_view { return s; } | ||
|  | #else
 | ||
|  | template <typename... Args> | ||
|  | using wformat_string = basic_format_string<wchar_t, type_identity_t<Args>...>; | ||
|  | inline auto runtime(wstring_view s) -> runtime_format_string<wchar_t> { | ||
|  |   return {{s}}; | ||
|  | } | ||
|  | #endif
 | ||
|  | 
 | ||
|  | template <> struct is_char<wchar_t> : std::true_type {}; | ||
|  | template <> struct is_char<detail::char8_type> : std::true_type {}; | ||
|  | template <> struct is_char<char16_t> : std::true_type {}; | ||
|  | template <> struct is_char<char32_t> : std::true_type {}; | ||
|  | 
 | ||
|  | template <typename... T> | ||
|  | constexpr auto make_wformat_args(const T&... args) | ||
|  |     -> format_arg_store<wformat_context, T...> { | ||
|  |   return {args...}; | ||
|  | } | ||
|  | 
 | ||
|  | inline namespace literals { | ||
|  | #if FMT_USE_USER_DEFINED_LITERALS && !FMT_USE_NONTYPE_TEMPLATE_ARGS
 | ||
|  | constexpr auto operator""_a(const wchar_t* s, size_t) | ||
|  |     -> detail::udl_arg<wchar_t> { | ||
|  |   return {s}; | ||
|  | } | ||
|  | #endif
 | ||
|  | }  // namespace literals
 | ||
|  | 
 | ||
|  | template <typename It, typename Sentinel> | ||
|  | auto join(It begin, Sentinel end, wstring_view sep) | ||
|  |     -> join_view<It, Sentinel, wchar_t> { | ||
|  |   return {begin, end, sep}; | ||
|  | } | ||
|  | 
 | ||
|  | template <typename Range> | ||
|  | auto join(Range&& range, wstring_view sep) | ||
|  |     -> join_view<detail::iterator_t<Range>, detail::sentinel_t<Range>, | ||
|  |                  wchar_t> { | ||
|  |   return join(std::begin(range), std::end(range), sep); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename T> | ||
|  | auto join(std::initializer_list<T> list, wstring_view sep) | ||
|  |     -> join_view<const T*, const T*, wchar_t> { | ||
|  |   return join(std::begin(list), std::end(list), sep); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename Char, FMT_ENABLE_IF(!std::is_same<Char, char>::value)> | ||
|  | auto vformat(basic_string_view<Char> format_str, | ||
|  |              basic_format_args<buffer_context<type_identity_t<Char>>> args) | ||
|  |     -> std::basic_string<Char> { | ||
|  |   auto buf = basic_memory_buffer<Char>(); | ||
|  |   detail::vformat_to(buf, format_str, args); | ||
|  |   return to_string(buf); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename... T> | ||
|  | auto format(wformat_string<T...> fmt, T&&... args) -> std::wstring { | ||
|  |   return vformat(fmt::wstring_view(fmt), fmt::make_wformat_args(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | // Pass char_t as a default template parameter instead of using
 | ||
|  | // std::basic_string<char_t<S>> to reduce the symbol size.
 | ||
|  | template <typename S, typename... T, typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(!std::is_same<Char, char>::value && | ||
|  |                         !std::is_same<Char, wchar_t>::value)> | ||
|  | auto format(const S& format_str, T&&... args) -> std::basic_string<Char> { | ||
|  |   return vformat(detail::to_string_view(format_str), | ||
|  |                  fmt::make_format_args<buffer_context<Char>>(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename Locale, typename S, typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(detail::is_locale<Locale>::value&& | ||
|  |                             detail::is_exotic_char<Char>::value)> | ||
|  | inline auto vformat( | ||
|  |     const Locale& loc, const S& format_str, | ||
|  |     basic_format_args<buffer_context<type_identity_t<Char>>> args) | ||
|  |     -> std::basic_string<Char> { | ||
|  |   return detail::vformat(loc, detail::to_string_view(format_str), args); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename Locale, typename S, typename... T, typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(detail::is_locale<Locale>::value&& | ||
|  |                             detail::is_exotic_char<Char>::value)> | ||
|  | inline auto format(const Locale& loc, const S& format_str, T&&... args) | ||
|  |     -> std::basic_string<Char> { | ||
|  |   return detail::vformat(loc, detail::to_string_view(format_str), | ||
|  |                          fmt::make_format_args<buffer_context<Char>>(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename OutputIt, typename S, typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(detail::is_output_iterator<OutputIt, Char>::value&& | ||
|  |                             detail::is_exotic_char<Char>::value)> | ||
|  | auto vformat_to(OutputIt out, const S& format_str, | ||
|  |                 basic_format_args<buffer_context<type_identity_t<Char>>> args) | ||
|  |     -> OutputIt { | ||
|  |   auto&& buf = detail::get_buffer<Char>(out); | ||
|  |   detail::vformat_to(buf, detail::to_string_view(format_str), args); | ||
|  |   return detail::get_iterator(buf, out); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename OutputIt, typename S, typename... T, | ||
|  |           typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(detail::is_output_iterator<OutputIt, Char>::value&& | ||
|  |                             detail::is_exotic_char<Char>::value)> | ||
|  | inline auto format_to(OutputIt out, const S& fmt, T&&... args) -> OutputIt { | ||
|  |   return vformat_to(out, detail::to_string_view(fmt), | ||
|  |                     fmt::make_format_args<buffer_context<Char>>(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename Locale, typename S, typename OutputIt, typename... Args, | ||
|  |           typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(detail::is_output_iterator<OutputIt, Char>::value&& | ||
|  |                             detail::is_locale<Locale>::value&& | ||
|  |                                 detail::is_exotic_char<Char>::value)> | ||
|  | inline auto vformat_to( | ||
|  |     OutputIt out, const Locale& loc, const S& format_str, | ||
|  |     basic_format_args<buffer_context<type_identity_t<Char>>> args) -> OutputIt { | ||
|  |   auto&& buf = detail::get_buffer<Char>(out); | ||
|  |   vformat_to(buf, detail::to_string_view(format_str), args, | ||
|  |              detail::locale_ref(loc)); | ||
|  |   return detail::get_iterator(buf, out); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename OutputIt, typename Locale, typename S, typename... T, | ||
|  |           typename Char = char_t<S>, | ||
|  |           bool enable = detail::is_output_iterator<OutputIt, Char>::value && | ||
|  |                         detail::is_locale<Locale>::value && | ||
|  |                         detail::is_exotic_char<Char>::value> | ||
|  | inline auto format_to(OutputIt out, const Locale& loc, const S& format_str, | ||
|  |                       T&&... args) -> | ||
|  |     typename std::enable_if<enable, OutputIt>::type { | ||
|  |   return vformat_to(out, loc, detail::to_string_view(format_str), | ||
|  |                     fmt::make_format_args<buffer_context<Char>>(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename OutputIt, typename Char, typename... Args, | ||
|  |           FMT_ENABLE_IF(detail::is_output_iterator<OutputIt, Char>::value&& | ||
|  |                             detail::is_exotic_char<Char>::value)> | ||
|  | inline auto vformat_to_n( | ||
|  |     OutputIt out, size_t n, basic_string_view<Char> format_str, | ||
|  |     basic_format_args<buffer_context<type_identity_t<Char>>> args) | ||
|  |     -> format_to_n_result<OutputIt> { | ||
|  |   using traits = detail::fixed_buffer_traits; | ||
|  |   auto buf = detail::iterator_buffer<OutputIt, Char, traits>(out, n); | ||
|  |   detail::vformat_to(buf, format_str, args); | ||
|  |   return {buf.out(), buf.count()}; | ||
|  | } | ||
|  | 
 | ||
|  | template <typename OutputIt, typename S, typename... T, | ||
|  |           typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(detail::is_output_iterator<OutputIt, Char>::value&& | ||
|  |                             detail::is_exotic_char<Char>::value)> | ||
|  | inline auto format_to_n(OutputIt out, size_t n, const S& fmt, T&&... args) | ||
|  |     -> format_to_n_result<OutputIt> { | ||
|  |   return vformat_to_n(out, n, detail::to_string_view(fmt), | ||
|  |                       fmt::make_format_args<buffer_context<Char>>(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename S, typename... T, typename Char = char_t<S>, | ||
|  |           FMT_ENABLE_IF(detail::is_exotic_char<Char>::value)> | ||
|  | inline auto formatted_size(const S& fmt, T&&... args) -> size_t { | ||
|  |   auto buf = detail::counting_buffer<Char>(); | ||
|  |   detail::vformat_to(buf, detail::to_string_view(fmt), | ||
|  |                      fmt::make_format_args<buffer_context<Char>>(args...)); | ||
|  |   return buf.count(); | ||
|  | } | ||
|  | 
 | ||
|  | inline void vprint(std::FILE* f, wstring_view fmt, wformat_args args) { | ||
|  |   auto buf = wmemory_buffer(); | ||
|  |   detail::vformat_to(buf, fmt, args); | ||
|  |   buf.push_back(L'\0'); | ||
|  |   if (std::fputws(buf.data(), f) == -1) | ||
|  |     FMT_THROW(system_error(errno, FMT_STRING("cannot write to file"))); | ||
|  | } | ||
|  | 
 | ||
|  | inline void vprint(wstring_view fmt, wformat_args args) { | ||
|  |   vprint(stdout, fmt, args); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename... T> | ||
|  | void print(std::FILE* f, wformat_string<T...> fmt, T&&... args) { | ||
|  |   return vprint(f, wstring_view(fmt), fmt::make_wformat_args(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename... T> void print(wformat_string<T...> fmt, T&&... args) { | ||
|  |   return vprint(wstring_view(fmt), fmt::make_wformat_args(args...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename... T> | ||
|  | void println(std::FILE* f, wformat_string<T...> fmt, T&&... args) { | ||
|  |   return print(f, L"{}\n", fmt::format(fmt, std::forward<T>(args)...)); | ||
|  | } | ||
|  | 
 | ||
|  | template <typename... T> void println(wformat_string<T...> fmt, T&&... args) { | ||
|  |   return print(L"{}\n", fmt::format(fmt, std::forward<T>(args)...)); | ||
|  | } | ||
|  | 
 | ||
|  | /**
 | ||
|  |   Converts *value* to ``std::wstring`` using the default format for type *T*. | ||
|  |  */ | ||
|  | template <typename T> inline auto to_wstring(const T& value) -> std::wstring { | ||
|  |   return format(FMT_STRING(L"{}"), value); | ||
|  | } | ||
|  | FMT_END_EXPORT | ||
|  | FMT_END_NAMESPACE | ||
|  | 
 | ||
|  | #endif  // FMT_XCHAR_H_
 |