pub fn format_number(value: f64) -> String {
if !value.is_finite() {
return format!("{value}");
}
if value == 0.0 {
return "0".to_string();
}
let rounded = value.round();
let near_integer = (value - rounded).abs() <= f64::EPSILON * value.abs().max(1.0) * 4.0;
if near_integer && rounded.abs() < 1e15 {
return format!("{}", rounded as i64);
}
truncate_significant(&format!("{value}"), 16)
}
fn truncate_significant(text: &str, max_digits: usize) -> String {
let (mantissa, exponent) = match text.find('e').or_else(|| text.find('E')) {
Some(index) => {
let exponent: i32 = text[index + 1..].parse().unwrap_or(0);
(&text[..index], exponent)
}
None => (text, 0),
};
let (sign, mantissa) = mantissa
.strip_prefix('-')
.map_or(("", mantissa), |rest| ("-", rest));
let digits: Vec<char> = mantissa.chars().filter(|c| c.is_ascii_digit()).collect();
let before_dot = mantissa.find('.').unwrap_or(mantissa.len());
let point = before_dot as i32 + exponent;
let first_nonzero = digits
.iter()
.position(|c| *c != '0')
.unwrap_or(digits.len());
let mut digits = digits;
if digits.len() - first_nonzero > max_digits {
digits.truncate(first_nonzero + max_digits);
}
let mut out = String::from(sign);
if point <= 0 {
out.push_str("0.");
for _ in 0..(-point) {
out.push('0');
}
for c in &digits {
out.push(*c);
}
} else if point as usize >= digits.len() {
for c in &digits {
out.push(*c);
}
for _ in 0..(point as usize - digits.len()) {
out.push('0');
}
} else {
for (index, c) in digits.iter().enumerate() {
if index == point as usize {
out.push('.');
}
out.push(*c);
}
}
out
}
#[cfg(test)]
mod tests {
use super::format_number;
#[test]
fn integers_print_without_decimals() {
assert_eq!(format_number(0.0), "0");
assert_eq!(format_number(100.0), "100");
assert_eq!(format_number(-3.0), "-3");
}
#[test]
fn floats_match_reference_precision() {
assert_eq!(format_number(1.8106601717798212), "1.810660171779821");
assert_eq!(format_number(-1.2803300858899105), "-1.280330085889910");
assert_eq!(format_number(0.016), "0.016");
assert_eq!(format_number(0.125), "0.125");
assert_eq!(format_number(1.5), "1.5");
}
}