87 lines
2.6 KiB
C++
87 lines
2.6 KiB
C++
/*
|
|
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::BasicPrintfArgFormatter;
|
|
|
|
// A custom argument formatter that doesn't print `-` for floating-point values
|
|
// rounded to 0.
|
|
class CustomArgFormatter
|
|
: public fmt::BasicArgFormatter<CustomArgFormatter, char> {
|
|
public:
|
|
CustomArgFormatter(fmt::BasicFormatter<char, CustomArgFormatter> &f,
|
|
fmt::FormatSpec &s, const char *fmt)
|
|
: fmt::BasicArgFormatter<CustomArgFormatter, char>(f, s, fmt) {}
|
|
|
|
void visit_double(double value) {
|
|
if (round(value * pow(10, spec().precision())) == 0)
|
|
value = 0;
|
|
fmt::BasicArgFormatter<CustomArgFormatter, char>::visit_double(value);
|
|
}
|
|
};
|
|
|
|
// A custom argument formatter that doesn't print `-` for floating-point values
|
|
// rounded to 0.
|
|
class CustomPrintfArgFormatter :
|
|
public BasicPrintfArgFormatter<CustomPrintfArgFormatter, char> {
|
|
public:
|
|
typedef BasicPrintfArgFormatter<CustomPrintfArgFormatter, char> Base;
|
|
|
|
CustomPrintfArgFormatter(fmt::BasicWriter<char> &w, fmt::FormatSpec &spec)
|
|
: Base(w, spec) {}
|
|
|
|
void visit_double(double value) {
|
|
if (round(value * pow(10, spec().precision())) == 0)
|
|
value = 0;
|
|
Base::visit_double(value);
|
|
}
|
|
};
|
|
|
|
typedef fmt::BasicFormatter<char, CustomArgFormatter> CustomFormatter;
|
|
|
|
std::string custom_vformat(const char *format_str,
|
|
fmt::basic_format_args<CustomFormatter> args) {
|
|
fmt::MemoryWriter writer;
|
|
// Pass custom argument formatter as a template arg to BasicFormatter.
|
|
CustomFormatter formatter(args, writer);
|
|
formatter.format(format_str);
|
|
return writer.str();
|
|
}
|
|
|
|
template <typename... Args>
|
|
std::string custom_format(const char *format_str, const Args & ... args) {
|
|
auto va = fmt::make_format_args<fmt::BasicFormatter<char>>(args...);
|
|
return custom_vformat(format_str, va);
|
|
}
|
|
|
|
typedef fmt::PrintfFormatter<char, CustomPrintfArgFormatter>
|
|
CustomPrintfFormatter;
|
|
|
|
std::string custom_vsprintf(
|
|
const char* format_str,
|
|
fmt::basic_format_args<CustomPrintfFormatter> args) {
|
|
fmt::MemoryWriter writer;
|
|
CustomPrintfFormatter formatter(args, writer);
|
|
formatter.format(format_str);
|
|
return writer.str();
|
|
}
|
|
|
|
template <typename... Args>
|
|
std::string custom_sprintf(const char *format_str, const Args & ... args) {
|
|
auto va = fmt::make_format_args<fmt::BasicFormatter<char>>(args...);
|
|
return custom_vsprintf(format_str, va);
|
|
}
|
|
|
|
TEST(CustomFormatterTest, Format) {
|
|
EXPECT_EQ("0.00", custom_format("{:.2f}", -.00001));
|
|
EXPECT_EQ("0.00", custom_sprintf("%.2f", -.00001));
|
|
}
|