2018-03-04 17:16:51 +00:00
|
|
|
// Formatting library for C++ - custom argument formatter tests
|
|
|
|
//
|
|
|
|
// Copyright (c) 2012 - present, Victor Zverovich
|
|
|
|
// All rights reserved.
|
|
|
|
//
|
|
|
|
// For the license information refer to format.h.
|
|
|
|
|
2019-05-07 19:09:11 +00:00
|
|
|
#ifndef _CRT_SECURE_NO_WARNINGS
|
2019-05-30 14:01:31 +00:00
|
|
|
#define _CRT_SECURE_NO_WARNINGS
|
2019-05-07 19:09:11 +00:00
|
|
|
#endif
|
|
|
|
|
2018-03-04 17:16:51 +00:00
|
|
|
#include "fmt/format.h"
|
2016-06-07 23:23:32 +00:00
|
|
|
#include "gtest-extra.h"
|
|
|
|
|
2018-07-04 20:17:03 +00:00
|
|
|
// MSVC 2013 is known to be broken.
|
|
|
|
#if !FMT_MSC_VER || FMT_MSC_VER > 1800
|
|
|
|
|
2016-06-07 23:23:32 +00:00
|
|
|
// A custom argument formatter that doesn't print `-` for floating-point values
|
|
|
|
// rounded to 0.
|
2019-01-13 02:27:38 +00:00
|
|
|
class custom_arg_formatter
|
2020-07-04 13:41:24 +00:00
|
|
|
: public fmt::detail::arg_formatter<fmt::format_context::iterator, char> {
|
2016-06-07 23:23:32 +00:00
|
|
|
public:
|
2020-07-04 13:41:24 +00:00
|
|
|
using base = fmt::detail::arg_formatter<fmt::format_context::iterator, char>;
|
2018-01-14 15:19:23 +00:00
|
|
|
|
2019-01-13 02:27:38 +00:00
|
|
|
custom_arg_formatter(fmt::format_context& ctx,
|
2019-02-10 03:34:42 +00:00
|
|
|
fmt::format_parse_context* parse_ctx,
|
2020-06-06 15:51:17 +00:00
|
|
|
fmt::format_specs* s = nullptr,
|
|
|
|
const char* = nullptr)
|
2019-02-10 03:34:42 +00:00
|
|
|
: base(ctx, parse_ctx, s) {}
|
2016-06-07 23:23:32 +00:00
|
|
|
|
2018-01-14 15:19:23 +00:00
|
|
|
using base::operator();
|
2016-11-20 16:47:24 +00:00
|
|
|
|
2018-03-30 18:20:12 +00:00
|
|
|
iterator operator()(double value) {
|
2018-10-17 16:15:29 +00:00
|
|
|
// Comparing a float to 0.0 is safe.
|
2019-07-07 13:39:20 +00:00
|
|
|
if (round(value * pow(10, specs()->precision)) == 0.0) value = 0;
|
2018-03-30 18:20:12 +00:00
|
|
|
return base::operator()(value);
|
2016-06-07 23:23:32 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2018-07-04 20:17:03 +00:00
|
|
|
std::string custom_vformat(fmt::string_view format_str, fmt::format_args args) {
|
2017-02-18 17:13:12 +00:00
|
|
|
fmt::memory_buffer buffer;
|
2020-06-27 00:24:03 +00:00
|
|
|
fmt::detail::buffer<char>& base = buffer;
|
2017-02-14 21:29:47 +00:00
|
|
|
// Pass custom argument formatter as a template arg to vwrite.
|
2020-05-29 23:51:45 +00:00
|
|
|
fmt::vformat_to<custom_arg_formatter>(std::back_inserter(base), format_str,
|
|
|
|
args);
|
2017-02-14 21:29:47 +00:00
|
|
|
return std::string(buffer.data(), buffer.size());
|
2016-06-07 23:23:32 +00:00
|
|
|
}
|
|
|
|
|
2016-08-27 00:23:13 +00:00
|
|
|
template <typename... Args>
|
2019-01-13 02:27:38 +00:00
|
|
|
std::string custom_format(const char* format_str, const Args&... args) {
|
2018-04-08 14:21:26 +00:00
|
|
|
auto va = fmt::make_format_args(args...);
|
2016-08-27 14:57:48 +00:00
|
|
|
return custom_vformat(format_str, va);
|
2016-08-27 00:23:13 +00:00
|
|
|
}
|
|
|
|
|
2016-06-07 23:23:32 +00:00
|
|
|
TEST(CustomFormatterTest, Format) {
|
|
|
|
EXPECT_EQ("0.00", custom_format("{:.2f}", -.00001));
|
|
|
|
}
|
2018-07-04 20:17:03 +00:00
|
|
|
#endif
|