Fix wide print overload (#1609)

fmtlib · Mar 28, 2020 · 33ceb46 · 33ceb46
1 parent 2864e84
commit 33ceb46
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/include/fmt/core.h b/include/fmt/core.h
@@ -326,8 +326,8 @@ FMT_CONSTEXPR typename std::make_unsigned<Int>::type to_unsigned(Int value) {
 
 constexpr unsigned char micro[] = "\u00B5";
 
-constexpr bool is_utf8() {
-  return FMT_UNICODE ||
+template <typename Char> constexpr bool is_unicode() {
+  return FMT_UNICODE || sizeof(Char) != 1 ||
          (sizeof(micro) == 3 && micro[0] == 0xC2 && micro[1] == 0xB5);
 }
 
@@ -1659,11 +1659,12 @@ typename buffer_context<Char>::iterator vformat_to(
     buffer<Char>& buf, basic_string_view<Char> format_str,
     basic_format_args<buffer_context<type_identity_t<Char>>> args);
 
-FMT_API void vprint_mojibake(std::FILE*, string_view, format_args);
+template <typename Ctx>
+inline void vprint_mojibake(std::FILE*,
+                            basic_string_view<typename Ctx::char_type>,
+                            basic_format_args<Ctx>) {}
 
-#ifndef _WIN32
-inline void vprint_mojibake(std::FILE*, string_view, format_args) {}
-#endif
+FMT_API void vprint_mojibake(std::FILE*, string_view, format_args);
 }  // namespace internal
 
 /**
@@ -1751,10 +1752,9 @@ FMT_API void vprint(std::FILE*, string_view, format_args);
     fmt::print(stderr, "Don't {}!", "panic");
   \endrst
  */
-template <typename S, typename... Args,
-          FMT_ENABLE_IF(internal::is_string<S>::value)>
+template <typename S, typename... Args, typename Char = char_t<S>>
 inline void print(std::FILE* f, const S& format_str, Args&&... args) {
-  return internal::is_utf8()
+  return internal::is_unicode<Char>()
              ? vprint(f, to_string_view(format_str),
                       internal::make_args_checked<Args...>(format_str, args...))
              : internal::vprint_mojibake(
@@ -1773,10 +1773,9 @@ inline void print(std::FILE* f, const S& format_str, Args&&... args) {
     fmt::print("Elapsed time: {0:.2f} seconds", 1.23);
   \endrst
  */
-template <typename S, typename... Args,
-          FMT_ENABLE_IF(internal::is_string<S>::value)>
+template <typename S, typename... Args, typename Char = char_t<S>>
 inline void print(const S& format_str, Args&&... args) {
-  return internal::is_utf8()
+  return internal::is_unicode<Char>()
              ? vprint(to_string_view(format_str),
                       internal::make_args_checked<Args...>(format_str, args...))
              : internal::vprint_mojibake(

diff --git a/test/format-test.cc b/test/format-test.cc
@@ -1756,6 +1756,7 @@ TEST(FormatTest, Print) {
   EXPECT_WRITE(stdout, fmt::print("Don't {}!", "panic"), "Don't panic!");
   EXPECT_WRITE(stderr, fmt::print(stderr, "Don't {}!", "panic"),
                "Don't panic!");
+  EXPECT_WRITE(stdout, fmt::print(L"test"), "test");
 #endif
 }