fmt/test/custom-formatter-test.cc

83 lines
2.4 KiB
C++
Raw Normal View History

/*
Custom argument formatter tests
Copyright (c) 2016, Victor Zverovich
All rights reserved.
For the license information refer to format.h.
*/
#include "fmt/printf.h"
#include "gtest-extra.h"
using fmt::PrintfArgFormatter;
// A custom argument formatter that doesn't print `-` for floating-point values
// rounded to 0.
class CustomArgFormatter : public fmt::ArgFormatter<char> {
public:
CustomArgFormatter(fmt::buffer &buf, fmt::basic_context<char> &ctx,
2017-01-28 12:51:35 +00:00
fmt::format_specs &s)
: fmt::ArgFormatter<char>(buf, ctx, s) {}
using fmt::ArgFormatter<char>::operator();
2016-11-20 16:47:24 +00:00
void operator()(double value) {
if (round(value * pow(10, spec().precision())) == 0)
value = 0;
fmt::ArgFormatter<char>::operator()(value);
}
};
// A custom argument formatter that doesn't print `-` for floating-point values
// rounded to 0.
class CustomPrintfArgFormatter : public PrintfArgFormatter<char> {
public:
CustomPrintfArgFormatter(fmt::buffer &buf, fmt::format_specs &spec)
: PrintfArgFormatter<char>(buf, spec) {}
using PrintfArgFormatter<char>::operator();
void operator()(double value) {
if (round(value * pow(10, spec().precision())) == 0)
value = 0;
PrintfArgFormatter<char>::operator()(value);
}
};
2017-02-05 14:54:03 +00:00
std::string custom_vformat(fmt::CStringRef format_str, fmt::args args) {
fmt::internal::MemoryBuffer<char> buffer;
// Pass custom argument formatter as a template arg to vwrite.
fmt::vformat_to<CustomArgFormatter>(buffer, format_str, args);
return std::string(buffer.data(), buffer.size());
}
2016-08-27 00:23:13 +00:00
template <typename... Args>
std::string custom_format(const char *format_str, const Args & ... args) {
2017-02-05 14:41:39 +00:00
auto va = fmt::make_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-11-07 00:11:24 +00:00
typedef fmt::printf_context<char, CustomPrintfArgFormatter>
CustomPrintfFormatter;
std::string custom_vsprintf(
const char* format_str,
fmt::basic_args<CustomPrintfFormatter> args) {
fmt::internal::MemoryBuffer<char> buffer;
2016-11-07 00:11:24 +00:00
CustomPrintfFormatter formatter(format_str, args);
formatter.format(buffer);
return std::string(buffer.data(), buffer.size());
}
2016-08-27 00:23:13 +00:00
template <typename... Args>
std::string custom_sprintf(const char *format_str, const Args & ... args) {
2017-02-05 14:41:39 +00:00
auto va = fmt::make_args<CustomPrintfFormatter>(args...);
2016-08-27 14:57:48 +00:00
return custom_vsprintf(format_str, va);
2016-08-27 00:23:13 +00:00
}
TEST(CustomFormatterTest, Format) {
EXPECT_EQ("0.00", custom_format("{:.2f}", -.00001));
EXPECT_EQ("0.00", custom_sprintf("%.2f", -.00001));
}