1 files changed, 145 insertions, 0 deletions
diff --git a/vendor/plotters/src/data/float.rs b/vendor/plotters/src/data/float.rs
new file mode 100644
index 000000000..febd330ea
--- /dev/null
+++ b/vendor/plotters/src/data/float.rs
@@ -0,0 +1,145 @@
+// The code that is related to float number handling
+fn find_minimal_repr(n: f64, eps: f64) -> (f64, usize) {
+    if eps >= 1.0 {
+        return (n, 0);
+    }
+    if n - n.floor() < eps {
+        (n.floor(), 0)
+    } else if n.ceil() - n < eps {
+        (n.ceil(), 0)
+    } else {
+        let (rem, pre) = find_minimal_repr((n - n.floor()) * 10.0, eps * 10.0);
+        (n.floor() + rem / 10.0, pre + 1)
+    }
+}
+
+#[allow(clippy::never_loop)]
+fn float_to_string(n: f64, max_precision: usize, min_decimal: usize) -> String {
+    let (mut result, mut count) = loop {
+        let (sign, n) = if n < 0.0 { ("-", -n) } else { ("", n) };
+        let int_part = n.floor();
+
+        let dec_part =
+            ((n.abs() - int_part.abs()) * (10.0f64).powi(max_precision as i32)).round() as u64;
+
+        if dec_part == 0 || max_precision == 0 {
+            break (format!("{}{:.0}", sign, int_part), 0);
+        }
+
+        let mut leading = "".to_string();
+        let mut dec_result = format!("{}", dec_part);
+
+        for _ in 0..(max_precision - dec_result.len()) {
+            leading.push('0');
+        }
+
+        while let Some(c) = dec_result.pop() {
+            if c != '0' {
+                dec_result.push(c);
+                break;
+            }
+        }
+
+        break (
+            format!("{}{:.0}.{}{}", sign, int_part, leading, dec_result),
+            leading.len() + dec_result.len(),
+        );
+    };
+
+    if count == 0 && min_decimal > 0 {
+        result.push('.');
+    }
+
+    while count < min_decimal {
+        result.push('0');
+        count += 1;
+    }
+    result
+}
+
+/// Handles printing of floating point numbers
+pub struct FloatPrettyPrinter {
+    /// Whether scientific notation is allowed
+    pub allow_scientific: bool,
+    /// Minimum allowed number of decimal digits
+    pub min_decimal: i32,
+    /// Maximum allowed number of decimal digits
+    pub max_decimal: i32,
+}
+
+impl FloatPrettyPrinter {
+    /// Handles printing of floating point numbers
+    pub fn print(&self, n: f64) -> String {
+        let (tn, p) = find_minimal_repr(n, (10f64).powi(-self.max_decimal));
+        let d_repr = float_to_string(tn, p, self.min_decimal as usize);
+        if !self.allow_scientific {
+            d_repr
+        } else {
+            if n == 0.0 {
+                return "0".to_string();
+            }
+
+            let mut idx = n.abs().log10().floor();
+            let mut exp = (10.0f64).powf(idx);
+
+            if n.abs() / exp + 1e-5 >= 10.0 {
+                idx += 1.0;
+                exp *= 10.0;
+            }
+
+            if idx.abs() < 3.0 {
+                return d_repr;
+            }
+
+            let (sn, sp) = find_minimal_repr(n / exp, 1e-5);
+            let s_repr = format!(
+                "{}e{}",
+                float_to_string(sn, sp, self.min_decimal as usize),
+                float_to_string(idx, 0, 0)
+            );
+            if s_repr.len() + 1 < d_repr.len() || (tn == 0.0 && n != 0.0) {
+                s_repr
+            } else {
+                d_repr
+            }
+        }
+    }
+}
+
+/// The function that pretty prints the floating number
+/// Since rust doesn't have anything that can format a float with out appearance, so we just
+/// implement a float pretty printing function, which finds the shortest representation of a
+/// floating point number within the allowed error range.
+///
+/// - `n`: The float number to pretty-print
+/// - `allow_sn`: Should we use scientific notation when possible
+/// - **returns**: The pretty printed string
+pub fn pretty_print_float(n: f64, allow_sn: bool) -> String {
+    (FloatPrettyPrinter {
+        allow_scientific: allow_sn,
+        min_decimal: 0,
+        max_decimal: 10,
+    })
+    .print(n)
+}
+
+#[cfg(test)]
+mod test {
+    use super::*;
+    #[test]
+    fn test_pretty_printing() {
+        assert_eq!(pretty_print_float(0.99999999999999999999, false), "1");
+        assert_eq!(pretty_print_float(0.9999, false), "0.9999");
+        assert_eq!(
+            pretty_print_float(-1e-5 - 0.00000000000000001, true),
+            "-1e-5"
+        );
+        assert_eq!(
+            pretty_print_float(-1e-5 - 0.00000000000000001, false),
+            "-0.00001"
+        );
+        assert_eq!(pretty_print_float(1e100, true), "1e100");
+        assert_eq!(pretty_print_float(1234567890f64, true), "1234567890");
+        assert_eq!(pretty_print_float(1000000001f64, true), "1e9");
+    }
+}