This commit is contained in:
Stepan Ponomarev 2022-11-02 13:01:02 +01:00
parent e74bc99746
commit 54fa95f013

View File

@ -1026,8 +1026,7 @@ struct count_fractional_digits<Num, Den, N, false> {
// Format subseconds which are given as an integer type with an appropriate // Format subseconds which are given as an integer type with an appropriate
// number of digits. // number of digits.
template <typename Char, typename OutputIt, typename Duration> template <typename Char, typename OutputIt, typename Duration>
void write_fractional_seconds(OutputIt& out, Duration d) { void write_fractional_seconds(OutputIt& out, Duration d, int precision = -1) {
FMT_ASSERT(!std::is_floating_point<typename Duration::rep>::value, "");
constexpr auto num_fractional_digits = constexpr auto num_fractional_digits =
count_fractional_digits<Duration::period::num, count_fractional_digits<Duration::period::num,
Duration::period::den>::value; Duration::period::den>::value;
@ -1036,59 +1035,37 @@ void write_fractional_seconds(OutputIt& out, Duration d) {
typename std::common_type<typename Duration::rep, typename std::common_type<typename Duration::rep,
std::chrono::seconds::rep>::type, std::chrono::seconds::rep>::type,
std::ratio<1, detail::pow10(num_fractional_digits)>>; std::ratio<1, detail::pow10(num_fractional_digits)>>;
if (std::ratio_less<typename subsecond_precision::period,
std::chrono::seconds::period>::value) {
*out++ = '.';
auto fractional =
detail::abs(d) - std::chrono::duration_cast<std::chrono::seconds>(d);
auto subseconds =
std::chrono::treat_as_floating_point<
typename subsecond_precision::rep>::value
? fractional.count()
: std::chrono::duration_cast<subsecond_precision>(fractional)
.count();
uint32_or_64_or_128_t<long long> n =
to_unsigned(to_nonnegative_int(subseconds, max_value<long long>()));
int num_digits = detail::count_digits(n);
if (num_fractional_digits > num_digits)
out = std::fill_n(out, num_fractional_digits - num_digits, '0');
out = format_decimal<Char>(out, n, num_digits).end;
}
}
template <typename Char, typename OutputIt, typename Duration> const auto fractional =
void write_fractional_seconds(OutputIt& out, Duration d, int precision) { detail::abs(d) - std::chrono::duration_cast<std::chrono::seconds>(d);
constexpr auto num_fractional_digits = uint32_or_64_or_128_t<long long> subseconds =
count_fractional_digits<Duration::period::num, std::chrono::treat_as_floating_point<
Duration::period::den>::value; typename subsecond_precision::rep>::value
? fractional.count()
: std::chrono::duration_cast<subsecond_precision>(fractional).count();
const int num_digits = detail::count_digits(subseconds);
using subsecond_precision = std::chrono::duration< int leading_zeroes = std::max(0, num_fractional_digits - num_digits);
typename std::common_type<typename Duration::rep, if (precision < 0) {
std::chrono::seconds::rep>::type, FMT_ASSERT(!std::is_floating_point<typename Duration::rep>::value, "");
std::ratio<1, detail::pow10(num_fractional_digits)>>; if (std::ratio_less<typename subsecond_precision::period,
if (precision > 0) { std::chrono::seconds::period>::value) {
*out++ = '.';
out = std::fill_n(out, leading_zeroes, '0');
out = format_decimal<Char>(out, subseconds, num_digits).end;
}
} else {
*out++ = '.'; *out++ = '.';
auto fractional = leading_zeroes = std::min(leading_zeroes, precision);
detail::abs(d) - std::chrono::duration_cast<std::chrono::seconds>(d); out = std::fill_n(out, leading_zeroes, '0');
auto subseconds = int remaining = precision - leading_zeroes;
std::chrono::treat_as_floating_point<
typename subsecond_precision::rep>::value
? fractional.count()
: std::chrono::duration_cast<subsecond_precision>(fractional)
.count();
uint32_or_64_or_128_t<long long> n =
to_unsigned(to_nonnegative_int(subseconds, max_value<long long>()));
int num_digits = detail::count_digits(n);
int zeroes =
std::min(std::max(0, num_fractional_digits - num_digits), precision);
if (num_fractional_digits > num_digits) out = std::fill_n(out, zeroes, '0');
int remaining = precision - zeroes;
if (remaining < num_digits) { if (remaining < num_digits) {
n /= to_unsigned(detail::pow10(to_unsigned(num_digits - remaining))); subseconds /=
out = format_decimal<Char>(out, n, remaining).end; to_unsigned(detail::pow10(to_unsigned(num_digits - remaining)));
out = format_decimal<Char>(out, subseconds, remaining).end;
return; return;
} }
out = format_decimal<Char>(out, n, num_digits).end; out = format_decimal<Char>(out, subseconds, num_digits).end;
remaining -= num_digits; remaining -= num_digits;
out = std::fill_n(out, remaining, '0'); out = std::fill_n(out, remaining, '0');
} }
@ -1097,48 +1074,27 @@ void write_fractional_seconds(OutputIt& out, Duration d, int precision) {
// Format subseconds which are given as a floating point type with an appropiate // Format subseconds which are given as a floating point type with an appropiate
// number of digits. We cannot pass the Duration here, as we explicitly need to // number of digits. We cannot pass the Duration here, as we explicitly need to
// pass the Rep value in the chrono_formatter. // pass the Rep value in the chrono_formatter.
template <typename Duration>
void write_floating_seconds(memory_buffer& buf, Duration duration) {
using Rep = typename Duration::rep;
FMT_ASSERT(std::is_floating_point<Rep>::value, "");
auto num_fractional_digits =
count_fractional_digits<Duration::period::num,
Duration::period::den>::value;
// For non-integer values, we ensure at least 6 digits to get microsecond
// precision.
auto val = duration.count();
if (num_fractional_digits < 6 &&
static_cast<Rep>(std::round(val)) != val)
num_fractional_digits = 6;
format_to(
std::back_inserter(buf), runtime("{:.{}f}"),
std::fmod(val *
static_cast<Rep>(Duration::period::num) /
static_cast<Rep>(Duration::period::den),
static_cast<Rep>(60)),
num_fractional_digits);
}
template <typename Duration> template <typename Duration>
void write_floating_seconds(memory_buffer& buf, Duration duration, void write_floating_seconds(memory_buffer& buf, Duration duration,
int precision) { int num_fractional_digits = -1) {
using Rep = typename Duration::rep; using rep = typename Duration::rep;
if (precision < 0) { FMT_ASSERT(std::is_floating_point<rep>::value, "");
write_floating_seconds(buf, duration);
return; auto val = duration.count();
if (num_fractional_digits < 0) {
num_fractional_digits =
count_fractional_digits<Duration::period::num,
Duration::period::den>::value;
if (num_fractional_digits < 6 && static_cast<rep>(std::round(val)) != val)
num_fractional_digits = 6;
} }
FMT_ASSERT(std::is_floating_point<Rep>::value, ""); format_to(std::back_inserter(buf), runtime("{:.{}f}"),
auto val = duration.count(); std::fmod(val * static_cast<rep>(Duration::period::num) /
static_cast<rep>(Duration::period::den),
format_to( static_cast<rep>(60)),
std::back_inserter(buf), runtime("{:.{}f}"), num_fractional_digits);
std::fmod(val *
static_cast<Rep>(Duration::period::num) /
static_cast<Rep>(Duration::period::den),
static_cast<Rep>(60)),
precision);
} }
template <typename OutputIt, typename Char, template <typename OutputIt, typename Char,
@ -1578,8 +1534,7 @@ class tm_writer {
}; };
struct chrono_format_checker : null_chrono_spec_handler<chrono_format_checker> { struct chrono_format_checker : null_chrono_spec_handler<chrono_format_checker> {
bool is_floating_point = false; bool has_precision_integral = false;
bool has_precision = false;
FMT_NORETURN void unsupported() { FMT_THROW(format_error("no date")); } FMT_NORETURN void unsupported() { FMT_THROW(format_error("no date")); }
@ -1594,7 +1549,7 @@ struct chrono_format_checker : null_chrono_spec_handler<chrono_format_checker> {
FMT_CONSTEXPR void on_iso_time() {} FMT_CONSTEXPR void on_iso_time() {}
FMT_CONSTEXPR void on_am_pm() {} FMT_CONSTEXPR void on_am_pm() {}
FMT_CONSTEXPR void on_duration_value() const { FMT_CONSTEXPR void on_duration_value() const {
if (has_precision && !is_floating_point) { if (has_precision_integral) {
FMT_THROW(format_error("precision not allowed for this argument type")); FMT_THROW(format_error("precision not allowed for this argument type"));
} }
} }
@ -2073,9 +2028,8 @@ struct formatter<std::chrono::duration<Rep, Period>, Char> {
begin = detail::parse_width(begin, end, handler); begin = detail::parse_width(begin, end, handler);
if (begin == end) return {begin, begin}; if (begin == end) return {begin, begin};
auto checker = detail::chrono_format_checker(); auto checker = detail::chrono_format_checker();
checker.is_floating_point = std::is_floating_point<Rep>::value;
if (*begin == '.') { if (*begin == '.') {
checker.has_precision = true; checker.has_precision_integral = !std::is_floating_point<Rep>::value;
begin = detail::parse_precision(begin, end, handler); begin = detail::parse_precision(begin, end, handler);
} }
if (begin != end && *begin == 'L') { if (begin != end && *begin == 'L') {