add fix-ostream-support-in-sprintf.patch
Signed-off-by: Kefu Chai <tchaikov@gmail.com>
This commit is contained in:
parent
87706d77ed
commit
3a6666b683
211
fix-ostream-support-in-sprintf.patch
Normal file
211
fix-ostream-support-in-sprintf.patch
Normal file
@ -0,0 +1,211 @@
|
||||
From e99809f29da1002e8b9246e9278084ad231174e5 Mon Sep 17 00:00:00 2001
|
||||
From: Victor Zverovich <viz@fb.com>
|
||||
Date: Sun, 12 Apr 2020 07:38:54 -0700
|
||||
Subject: [PATCH] Fix ostream support in sprintf (#1631)
|
||||
|
||||
---
|
||||
include/fmt/core.h | 12 +++++++-----
|
||||
include/fmt/ostream.h | 33 ++++++++++++++++++++++++++++-----
|
||||
include/fmt/printf.h | 9 +++++++--
|
||||
test/core-test.cc | 3 ++-
|
||||
test/format-test.cc | 5 +++--
|
||||
test/printf-test.cc | 4 +---
|
||||
6 files changed, 48 insertions(+), 18 deletions(-)
|
||||
|
||||
diff --git a/include/fmt/core.h b/include/fmt/core.h
|
||||
index cab3dfe0b..1a75100ad 100644
|
||||
--- a/include/fmt/core.h
|
||||
+++ b/include/fmt/core.h
|
||||
@@ -827,7 +827,8 @@ template <typename Char> struct string_value {
|
||||
template <typename Context> struct custom_value {
|
||||
using parse_context = basic_format_parse_context<typename Context::char_type>;
|
||||
const void* value;
|
||||
- void (*format)(const void* arg, parse_context& parse_ctx, Context& ctx);
|
||||
+ void (*format)(const void* arg,
|
||||
+ typename Context::parse_context_type& parse_ctx, Context& ctx);
|
||||
};
|
||||
|
||||
// A formatting argument value.
|
||||
@@ -887,9 +888,9 @@ template <typename Context> class value {
|
||||
private:
|
||||
// Formats an argument of a custom type, such as a user-defined class.
|
||||
template <typename T, typename Formatter>
|
||||
- static void format_custom_arg(
|
||||
- const void* arg, basic_format_parse_context<char_type>& parse_ctx,
|
||||
- Context& ctx) {
|
||||
+ static void format_custom_arg(const void* arg,
|
||||
+ typename Context::parse_context_type& parse_ctx,
|
||||
+ Context& ctx) {
|
||||
Formatter f;
|
||||
parse_ctx.advance_to(f.parse(parse_ctx));
|
||||
ctx.advance_to(f.format(*static_cast<const T*>(arg), ctx));
|
||||
@@ -1066,7 +1067,7 @@ template <typename Context> class basic_format_arg {
|
||||
public:
|
||||
explicit handle(internal::custom_value<Context> custom) : custom_(custom) {}
|
||||
|
||||
- void format(basic_format_parse_context<char_type>& parse_ctx,
|
||||
+ void format(typename Context::parse_context_type& parse_ctx,
|
||||
Context& ctx) const {
|
||||
custom_.format(custom_.value, parse_ctx, ctx);
|
||||
}
|
||||
@@ -1277,6 +1278,7 @@ template <typename OutputIt, typename Char> class basic_format_context {
|
||||
public:
|
||||
using iterator = OutputIt;
|
||||
using format_arg = basic_format_arg<basic_format_context>;
|
||||
+ using parse_context_type = basic_format_parse_context<Char>;
|
||||
template <typename T> using formatter_type = formatter<T, char_type>;
|
||||
|
||||
basic_format_context(const basic_format_context&) = delete;
|
||||
diff --git a/include/fmt/ostream.h b/include/fmt/ostream.h
|
||||
index c4831533d..4526f5238 100644
|
||||
--- a/include/fmt/ostream.h
|
||||
+++ b/include/fmt/ostream.h
|
||||
@@ -9,9 +9,14 @@
|
||||
#define FMT_OSTREAM_H_
|
||||
|
||||
#include <ostream>
|
||||
+
|
||||
#include "format.h"
|
||||
|
||||
FMT_BEGIN_NAMESPACE
|
||||
+
|
||||
+template <typename CHar> class basic_printf_parse_context;
|
||||
+template <typename OutputIt, typename Char> class basic_printf_context;
|
||||
+
|
||||
namespace internal {
|
||||
|
||||
template <class Char> class formatbuf : public std::basic_streambuf<Char> {
|
||||
@@ -93,9 +98,9 @@ void format_value(buffer<Char>& buf, const T& value,
|
||||
locale_ref loc = locale_ref()) {
|
||||
formatbuf<Char> format_buf(buf);
|
||||
std::basic_ostream<Char> output(&format_buf);
|
||||
- #if !defined(FMT_STATIC_THOUSANDS_SEPARATOR)
|
||||
+#if !defined(FMT_STATIC_THOUSANDS_SEPARATOR)
|
||||
if (loc) output.imbue(loc.get<std::locale>());
|
||||
- #endif
|
||||
+#endif
|
||||
output.exceptions(std::ios_base::failbit | std::ios_base::badbit);
|
||||
output << value;
|
||||
buf.resize(buf.size());
|
||||
@@ -104,14 +109,32 @@ void format_value(buffer<Char>& buf, const T& value,
|
||||
// Formats an object of type T that has an overloaded ostream operator<<.
|
||||
template <typename T, typename Char>
|
||||
struct fallback_formatter<T, Char, enable_if_t<is_streamable<T, Char>::value>>
|
||||
- : formatter<basic_string_view<Char>, Char> {
|
||||
- template <typename Context>
|
||||
- auto format(const T& value, Context& ctx) -> decltype(ctx.out()) {
|
||||
+ : private formatter<basic_string_view<Char>, Char> {
|
||||
+ auto parse(basic_format_parse_context<Char>& ctx) -> decltype(ctx.begin()) {
|
||||
+ return formatter<basic_string_view<Char>, Char>::parse(ctx);
|
||||
+ }
|
||||
+ template <typename ParseCtx,
|
||||
+ FMT_ENABLE_IF(std::is_same<
|
||||
+ ParseCtx, basic_printf_parse_context<Char>>::value)>
|
||||
+ auto parse(ParseCtx& ctx) -> decltype(ctx.begin()) {
|
||||
+ return ctx.begin();
|
||||
+ }
|
||||
+
|
||||
+ template <typename OutputIt>
|
||||
+ auto format(const T& value, basic_format_context<OutputIt, Char>& ctx)
|
||||
+ -> OutputIt {
|
||||
basic_memory_buffer<Char> buffer;
|
||||
format_value(buffer, value, ctx.locale());
|
||||
basic_string_view<Char> str(buffer.data(), buffer.size());
|
||||
return formatter<basic_string_view<Char>, Char>::format(str, ctx);
|
||||
}
|
||||
+ template <typename OutputIt>
|
||||
+ auto format(const T& value, basic_printf_context<OutputIt, Char>& ctx)
|
||||
+ -> OutputIt {
|
||||
+ basic_memory_buffer<Char> buffer;
|
||||
+ format_value(buffer, value, ctx.locale());
|
||||
+ return std::copy(buffer.begin(), buffer.end(), ctx.out());
|
||||
+ }
|
||||
};
|
||||
} // namespace internal
|
||||
|
||||
diff --git a/include/fmt/printf.h b/include/fmt/printf.h
|
||||
index bfbaa0479..9c7b85ac5 100644
|
||||
--- a/include/fmt/printf.h
|
||||
+++ b/include/fmt/printf.h
|
||||
@@ -189,6 +189,10 @@ using internal::vprintf;
|
||||
|
||||
template <typename Range> class printf_arg_formatter;
|
||||
|
||||
+template <typename Char>
|
||||
+class basic_printf_parse_context : public basic_format_parse_context<Char> {
|
||||
+ using basic_format_parse_context<Char>::basic_format_parse_context;
|
||||
+};
|
||||
template <typename OutputIt, typename Char> class basic_printf_context;
|
||||
|
||||
/**
|
||||
@@ -324,6 +328,7 @@ template <typename OutputIt, typename Char> class basic_printf_context {
|
||||
using char_type = Char;
|
||||
using iterator = OutputIt;
|
||||
using format_arg = basic_format_arg<basic_printf_context>;
|
||||
+ using parse_context_type = basic_printf_parse_context<Char>;
|
||||
template <typename T> using formatter_type = printf_formatter<T>;
|
||||
|
||||
private:
|
||||
@@ -331,7 +336,7 @@ template <typename OutputIt, typename Char> class basic_printf_context {
|
||||
|
||||
OutputIt out_;
|
||||
basic_format_args<basic_printf_context> args_;
|
||||
- basic_format_parse_context<Char> parse_ctx_;
|
||||
+ parse_context_type parse_ctx_;
|
||||
|
||||
static void parse_flags(format_specs& specs, const Char*& it,
|
||||
const Char* end);
|
||||
@@ -362,7 +367,7 @@ template <typename OutputIt, typename Char> class basic_printf_context {
|
||||
|
||||
format_arg arg(int id) const { return args_.get(id); }
|
||||
|
||||
- basic_format_parse_context<Char>& parse_context() { return parse_ctx_; }
|
||||
+ parse_context_type& parse_context() { return parse_ctx_; }
|
||||
|
||||
FMT_CONSTEXPR void on_error(const char* message) {
|
||||
parse_ctx_.on_error(message);
|
||||
diff --git a/test/core-test.cc b/test/core-test.cc
|
||||
index f19d0423b..735fcde26 100644
|
||||
--- a/test/core-test.cc
|
||||
+++ b/test/core-test.cc
|
||||
@@ -210,7 +210,8 @@ TEST(ArgTest, FormatArgs) {
|
||||
}
|
||||
|
||||
struct custom_context {
|
||||
- typedef char char_type;
|
||||
+ using char_type = char;
|
||||
+ using parse_context_type = fmt::format_parse_context;
|
||||
|
||||
template <typename T> struct formatter_type {
|
||||
template <typename ParseContext>
|
||||
diff --git a/test/format-test.cc b/test/format-test.cc
|
||||
index 6a873185c..e2bd98d27 100644
|
||||
--- a/test/format-test.cc
|
||||
+++ b/test/format-test.cc
|
||||
@@ -2260,8 +2260,9 @@ struct test_parse_context {
|
||||
};
|
||||
|
||||
struct test_context {
|
||||
- typedef char char_type;
|
||||
- typedef fmt::basic_format_arg<test_context> format_arg;
|
||||
+ using char_type = char;
|
||||
+ using format_arg = fmt::basic_format_arg<test_context>;
|
||||
+ using parse_context_type = fmt::format_parse_context;
|
||||
|
||||
template <typename T> struct formatter_type {
|
||||
typedef fmt::formatter<T, char_type> type;
|
||||
diff --git a/test/printf-test.cc b/test/printf-test.cc
|
||||
index 545e02aab..70b1238da 100644
|
||||
--- a/test/printf-test.cc
|
||||
+++ b/test/printf-test.cc
|
||||
@@ -508,9 +508,7 @@ TEST(PrintfTest, PrintfError) {
|
||||
TEST(PrintfTest, WideString) { EXPECT_EQ(L"abc", fmt::sprintf(L"%s", L"abc")); }
|
||||
|
||||
TEST(PrintfTest, PrintfCustom) {
|
||||
- // The test is disabled for now because it requires decoupling
|
||||
- // fallback_formatter::format from format_context.
|
||||
- //EXPECT_EQ("abc", test_sprintf("%s", TestString("abc")));
|
||||
+ EXPECT_EQ("abc", test_sprintf("%s", TestString("abc")));
|
||||
}
|
||||
|
||||
TEST(PrintfTest, OStream) {
|
Loading…
Reference in New Issue
Block a user