From 664c0ee77cfda93d6a774a280e09affa6b448e5a Mon Sep 17 00:00:00 2001
From: Mark Thom <markjordanthom@gmail.com>
Date: Mon, 7 Mar 2022 19:44:31 -0700
Subject: [PATCH] fix float formatting (#1331)

---
 src/heap_print.rs           | 36 ++++++++++++++++++++----------------
 src/machine/system_calls.rs |  2 +-
 2 files changed, 21 insertions(+), 17 deletions(-)

diff --git a/src/heap_print.rs b/src/heap_print.rs
index a0231848..b7ee8552 100644
--- a/src/heap_print.rs
+++ b/src/heap_print.rs
@@ -484,6 +484,24 @@ macro_rules! print_char {
     };
 }
 
+pub fn fmt_float(mut fl: f64) -> String {
+    if OrderedFloat(fl) == -0f64 {
+        fl = 0f64;
+    }
+
+    if fl.fract() == 0f64 {
+        if fl.abs() >= 1.0e16 {
+            format!("{:.1e}", fl.trunc())
+        } else {
+            format!("{:.1}", fl.trunc())
+        }
+    } else if 0f64 < fl.fract().abs() && fl.fract().abs() <= 1.0e-16 {
+        format!("{0:.15e}", fl)
+    } else {
+        format!("{}", fl)
+    }
+}
+
 #[derive(Debug)]
 pub struct HCPrinter<'a, Outputter> {
     outputter: Outputter,
@@ -935,22 +953,8 @@ impl<'a, Outputter: HCValueOutputter> HCPrinter<'a, Outputter> {
 
         match n {
             NumberFocus::Unfocused(n) => match n {
-                Number::Float(OrderedFloat(mut fl)) => {
-                    if OrderedFloat(fl) == -0f64 {
-                        fl = 0f64;
-                    }
-
-                    let output_str = if fl.fract() == 0f64 {
-                        if fl.abs() >= 1.0e16 {
-                            format!("{:.1e}", fl.trunc())
-                        } else {
-                            format!("{:.1}", fl.trunc())
-                        }
-                    } else if 0f64 < fl.fract().abs() && fl.fract().abs() <= 1.0e-16 {
-                        format!("{:>1e}", fl)
-                    } else {
-                        format!("{}", fl)
-                    };
+                Number::Float(OrderedFloat(fl)) => {
+                    let output_str = fmt_float(fl);
 
                     push_space_if_amb!(self, &output_str, {
                         append_str!(self, &output_str);
diff --git a/src/machine/system_calls.rs b/src/machine/system_calls.rs
index b8bd9063..8dc8d4e8 100644
--- a/src/machine/system_calls.rs
+++ b/src/machine/system_calls.rs
@@ -1818,7 +1818,7 @@ impl Machine {
 
         let string = match Number::try_from(n) {
             Ok(Number::Float(OrderedFloat(n))) => {
-                format!("{0:<20?}", n)
+                fmt_float(n)
             }
             Ok(Number::Fixnum(n)) => n.get_num().to_string(),
             Ok(Number::Integer(n)) => n.to_string(),
-- 
2.54.0