visi_core/core/engine/
result_data.rs1use serde::{Deserialize, Serialize};
2
3#[derive(Debug, Clone, Serialize, Deserialize)]
17pub enum ResultData {
18 None,
20 Boolean(bool),
22 Integer(i64),
24 Float(f64),
27 String(String),
29 List(Vec<ResultData>),
32 Dict(Vec<(ResultData, ResultData)>),
34 Error(String),
36}
37
38impl std::fmt::Display for ResultData {
39 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
40 match self {
41 ResultData::None => write!(f, ""),
42 ResultData::Boolean(b) => write!(f, "{}", if *b { "TRUE" } else { "FALSE" }),
43 ResultData::Integer(i) => write!(f, "{}", i),
44 ResultData::Float(fl) => write!(f, "{}", format_excel_number(*fl)),
45 ResultData::String(s) => write!(f, "{}", s),
46 ResultData::List(l) => {
47 let items: Vec<String> = l.iter().map(|i| i.to_string()).collect();
48 write!(f, "[{}]", items.join(", "))
49 }
50 ResultData::Dict(d) => {
51 let items: Vec<String> = d.iter().map(|(k, v)| format!("{}: {}", k, v)).collect();
52 write!(f, "{{ {} }}", items.join(", "))
53 }
54 ResultData::Error(e) => write!(f, "Error: {}", e),
55 }
56 }
57}
58
59fn round_digits_half_up(digits: &str, exp: i32, keep: usize) -> (String, i32) {
64 if digits.len() <= keep {
65 return (digits.to_string(), exp);
66 }
67 let mut kept: Vec<u8> = digits.as_bytes()[..keep].to_vec();
68 let round_up = digits.as_bytes()[keep] >= b'5';
69 let mut exp = exp;
70 if round_up {
71 let mut i = keep;
72 loop {
73 if i == 0 {
74 kept.insert(0, b'1');
75 kept.pop();
76 exp += 1;
77 break;
78 }
79 i -= 1;
80 if kept[i] == b'9' {
81 kept[i] = b'0';
82 } else {
83 kept[i] += 1;
84 break;
85 }
86 }
87 }
88 let mut out = String::from_utf8(kept).expect("ascii digits");
89 while out.len() > 1 && out.ends_with('0') {
90 out.pop();
91 }
92 (out, exp)
93}
94
95pub(crate) const EXCEL_ERROR_CODES: &[&str] = &[
100 "#NULL!", "#DIV/0!", "#VALUE!", "#REF!", "#NAME?", "#NUM!", "#N/A", "#CALC!", "#SPILL!",
101];
102
103pub(crate) fn is_excel_error_code(src: &str) -> bool {
111 EXCEL_ERROR_CODES
112 .iter()
113 .any(|e| src.eq_ignore_ascii_case(e))
114}
115
116pub(crate) fn format_excel_number(f: f64) -> String {
117 if f == 0.0 {
118 return "0".to_string();
119 }
120 if f.is_nan() || f.is_infinite() {
121 return "#NUM!".to_string();
122 }
123
124 let sci = format!("{:.14e}", f);
125 let (mantissa, exp_str) = sci.split_once('e').expect("{:e} always emits an exponent");
126 let exp: i32 = exp_str.parse().expect("{:e} emits an integer exponent");
127 let sign = if f < 0.0 { "-" } else { "" };
128 let digits: String = mantissa.chars().filter(|c| c.is_ascii_digit()).collect();
129 let digits = digits.trim_end_matches('0');
130 let digits = if digits.is_empty() { "0" } else { digits };
131
132 let decimal_len = if exp >= 0 {
133 let int_digits = (exp + 1) as usize;
134 let frac_digits = digits.len().saturating_sub(int_digits);
135 int_digits + usize::from(frac_digits > 0) + frac_digits
136 } else {
137 2 + (-exp - 1) as usize + digits.len()
138 };
139
140 if decimal_len <= 20 {
141 if exp >= 0 {
142 let int_digits = (exp + 1) as usize;
143 let mut out = String::from(sign);
144 if digits.len() <= int_digits {
145 out.push_str(digits);
146 out.push_str(&"0".repeat(int_digits - digits.len()));
147 } else {
148 out.push_str(&digits[..int_digits]);
149 out.push('.');
150 out.push_str(&digits[int_digits..]);
151 }
152 out
153 } else {
154 format!("{}0.{}{}", sign, "0".repeat((-exp - 1) as usize), digits)
155 }
156 } else {
157 let suffix_budget_len = if exp.abs() >= 99 {
158 5
159 } else {
160 format!("E{:+03}", exp).len()
161 };
162 let frac_digits = 18usize.saturating_sub(suffix_budget_len).min(14);
163
164 let (rounded_digits, exp) = round_digits_half_up(digits, exp, frac_digits + 1);
165 let mut mantissa = String::from(sign);
166 mantissa.push_str(&rounded_digits[..1]);
167 if rounded_digits.len() > 1 {
168 mantissa.push('.');
169 mantissa.push_str(&rounded_digits[1..]);
170 }
171 format!("{}E{:+03}", mantissa, exp)
172 }
173}