Compare commits
149 Commits
feature/br
...
feature/ni
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
96a5482e01 | ||
|
|
ffe5d1a158 | ||
|
|
402a13bd00 | ||
|
|
a345d7c9ac | ||
|
|
3fd55bb5c9 | ||
|
|
bab24a5207 | ||
|
|
5288665f70 | ||
|
|
ba75ffcf4f | ||
|
|
b5c977d3aa | ||
|
|
4029441cea | ||
|
|
cd47c609a0 | ||
|
|
ae6acdcdd5 | ||
|
|
c196db2115 | ||
|
|
a3c201e4e4 | ||
|
|
126e62957a | ||
|
|
294a651ae5 | ||
|
|
6f8a1e0da6 | ||
|
|
205ba6ee2d | ||
|
|
547b331773 | ||
|
|
f96612cf23 | ||
|
|
745435b950 | ||
|
|
4ca996cd3f | ||
|
|
3fbb91c414 | ||
|
|
93c9c42607 | ||
|
|
11edc2378e | ||
|
|
962e70c834 | ||
|
|
f803dad0a3 | ||
|
|
19580fc1ad | ||
|
|
e760b2d08e | ||
|
|
0e6ded7154 | ||
|
|
db26403432 | ||
|
|
9193479cce | ||
|
|
f814a75ae5 | ||
|
|
c8da5efb5f | ||
|
|
522e734395 | ||
|
|
2a7d59e512 | ||
|
|
c4142d4bf8 | ||
|
|
885d344b5b | ||
|
|
bed6f007cd | ||
|
|
dbd1b2df60 | ||
|
|
db552047c8 | ||
|
|
bcbacdb0a3 | ||
|
|
d0f37854d9 | ||
|
|
99b03f70c3 | ||
|
|
3e1605a494 | ||
|
|
d6aad08e73 | ||
|
|
8597d14a4e | ||
|
|
01b19b9c35 | ||
|
|
4649a0c78c | ||
|
|
cd0baf5ba7 | ||
|
|
167d169f1a | ||
|
|
080574b112 | ||
|
|
6056b8f122 | ||
|
|
e61b15655a | ||
|
|
6822505602 | ||
|
|
25f7891343 | ||
|
|
bdd0af0a39 | ||
|
|
261924396d | ||
|
|
67ef3bcf87 | ||
|
|
9aac285964 | ||
|
|
ba40c3c673 | ||
|
|
cc01556387 | ||
|
|
35323df20e | ||
|
|
19c115b32f | ||
|
|
6b60b339d6 | ||
|
|
41c8d88b80 | ||
|
|
73e5c305cc | ||
|
|
774b447c84 | ||
|
|
23b7333572 | ||
|
|
ef47c26c50 | ||
|
|
5cc61b0de4 | ||
|
|
42e8d44454 | ||
|
|
f840806f94 | ||
|
|
4a21d4b03a | ||
|
|
4cf162eb82 | ||
|
|
2cab93be18 | ||
|
|
fd34e46689 | ||
|
|
3bb49d1e8f | ||
|
|
1391f196b5 | ||
|
|
3db094c956 | ||
|
|
50941cb6ef | ||
|
|
150b516863 | ||
|
|
dc49afa2c3 | ||
|
|
acb90fbb9d | ||
|
|
7676efca44 | ||
|
|
8e15c623dd | ||
|
|
eb76d8dd23 | ||
|
|
1053d00d22 | ||
|
|
5ff4774c5a | ||
|
|
7e966baa0d | ||
|
|
c52c05aa8e | ||
|
|
129959137d | ||
|
|
4d5af45711 | ||
|
|
471f32f92a | ||
|
|
7b5427196d | ||
|
|
66b7586730 | ||
|
|
630f0e1baf | ||
|
|
bc9fefcb70 | ||
|
|
3d970acc34 | ||
|
|
e0e566db76 | ||
|
|
e3fc1d229a | ||
|
|
78d1f6b4a4 | ||
|
|
45ee1c35fe | ||
|
|
671cfff619 | ||
|
|
7e2fcec4a3 | ||
|
|
12342da649 | ||
|
|
4e9d7611a8 | ||
|
|
e0339f641b | ||
|
|
aa953e1ece | ||
|
|
cbf75c059b | ||
|
|
b2744efeb5 | ||
|
|
ef6849e822 | ||
|
|
aa4dd598b1 | ||
|
|
8b3bd7943e | ||
|
|
a1d1b64b76 | ||
|
|
5094a7fe4d | ||
|
|
c283fd7b60 | ||
|
|
36beccd4ae | ||
|
|
a252f9c626 | ||
|
|
f8bd03d92c | ||
|
|
e44a2e8c3e | ||
|
|
4217c1455b | ||
|
|
d8b3ba0dae | ||
|
|
95a7782f22 | ||
|
|
087211ebc3 | ||
|
|
46d766c85c | ||
|
|
2a14ee73c4 | ||
|
|
401c7c4289 | ||
|
|
3246137545 | ||
|
|
b1f45511d0 | ||
|
|
4b93174261 | ||
|
|
3111a74530 | ||
|
|
ae3fcaf9e9 | ||
|
|
dd78db3d2b | ||
|
|
acf334074f | ||
|
|
e48810d91b | ||
|
|
18db1cf052 | ||
|
|
ed40f79324 | ||
|
|
10ee95c48f | ||
|
|
741a223f3d | ||
|
|
ba139d1b6c | ||
|
|
e0306cb161 | ||
|
|
cea1f67cd0 | ||
|
|
4a3eef5a81 | ||
|
|
91299e3c0b | ||
|
|
1b38d79b81 | ||
|
|
a2d11a42cc | ||
|
|
480a2d1769 | ||
|
|
f30f6864e2 |
20
Cargo.lock
generated
20
Cargo.lock
generated
@@ -43,6 +43,15 @@ dependencies = [
|
|||||||
"libc",
|
"libc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "approx"
|
||||||
|
version = "0.5.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "cab112f0a86d568ea0e627cc1d6be74a1e9cd55214684db5561995f6dad897c6"
|
||||||
|
dependencies = [
|
||||||
|
"num-traits",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "arrayvec"
|
name = "arrayvec"
|
||||||
version = "0.7.6"
|
version = "0.7.6"
|
||||||
@@ -443,6 +452,7 @@ dependencies = [
|
|||||||
"ryu",
|
"ryu",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
|
"statrs",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -965,6 +975,16 @@ version = "0.3.11"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "38b58827f4464d87d377d175e90bf58eb00fd8716ff0a62f80356b5e61555d0d"
|
checksum = "38b58827f4464d87d377d175e90bf58eb00fd8716ff0a62f80356b5e61555d0d"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "statrs"
|
||||||
|
version = "0.18.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "2a3fe7c28c6512e766b0874335db33c94ad7b8f9054228ae1c2abd47ce7d335e"
|
||||||
|
dependencies = [
|
||||||
|
"approx",
|
||||||
|
"num-traits",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "subtle"
|
name = "subtle"
|
||||||
version = "2.5.0"
|
version = "2.5.0"
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ regex = { version = "1.0", optional = true}
|
|||||||
regex-lite = { version = "0.1.6", optional = true}
|
regex-lite = { version = "0.1.6", optional = true}
|
||||||
bitcode = "0.6.3"
|
bitcode = "0.6.3"
|
||||||
csv = "1.3.0"
|
csv = "1.3.0"
|
||||||
|
statrs = { version = "0.18.0", default-features = false, features = [] }
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
default = ["use_regex_full"]
|
default = ["use_regex_full"]
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
use ironcalc_base::{types::CellType, Model};
|
use ironcalc_base::{types::CellType, Model};
|
||||||
|
|
||||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
let mut model = Model::new_empty("formulas-and-errors", "en", "UTC")?;
|
let mut model = Model::new_empty("formulas-and-errors", "en", "UTC", "en")?;
|
||||||
// A1
|
// A1
|
||||||
model.set_user_input(0, 1, 1, "1".to_string())?;
|
model.set_user_input(0, 1, 1, "1".to_string())?;
|
||||||
// A2
|
// A2
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
use ironcalc_base::{cell::CellValue, Model};
|
use ironcalc_base::{cell::CellValue, Model};
|
||||||
|
|
||||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
let mut model = Model::new_empty("hello-world", "en", "UTC")?;
|
let mut model = Model::new_empty("hello-world", "en", "UTC", "en")?;
|
||||||
// A1
|
// A1
|
||||||
model.set_user_input(0, 1, 1, "Hello".to_string())?;
|
model.set_user_input(0, 1, 1, "Hello".to_string())?;
|
||||||
// B1
|
// B1
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
||||||
use crate::expressions::parser::stringify::{to_string, to_string_displaced, DisplaceData};
|
use crate::expressions::parser::stringify::{
|
||||||
|
to_localized_string, to_string_displaced, DisplaceData,
|
||||||
|
};
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
use crate::model::Model;
|
use crate::model::Model;
|
||||||
|
|
||||||
@@ -8,7 +10,7 @@ use crate::model::Model;
|
|||||||
// In IronCalc, if one of the edges of the range is deleted will replace the edge with #REF!
|
// In IronCalc, if one of the edges of the range is deleted will replace the edge with #REF!
|
||||||
// I feel this is unimportant for now.
|
// I feel this is unimportant for now.
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
fn shift_cell_formula(
|
fn shift_cell_formula(
|
||||||
&mut self,
|
&mut self,
|
||||||
sheet: u32,
|
sheet: u32,
|
||||||
@@ -29,7 +31,7 @@ impl Model {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
// FIXME: This is not a very performant way if the formula has changed :S.
|
// FIXME: This is not a very performant way if the formula has changed :S.
|
||||||
let formula = to_string(node, &cell_reference);
|
let formula = to_localized_string(node, &cell_reference, self.locale, self.language);
|
||||||
let formula_displaced = to_string_displaced(node, &cell_reference, displace_data);
|
let formula_displaced = to_string_displaced(node, &cell_reference, displace_data);
|
||||||
if formula != formula_displaced {
|
if formula != formula_displaced {
|
||||||
self.update_cell_with_formula(sheet, row, column, format!("={formula_displaced}"))?;
|
self.update_cell_with_formula(sheet, row, column, format!("={formula_displaced}"))?;
|
||||||
@@ -108,7 +110,9 @@ impl Model {
|
|||||||
// FIXME: we need some user_input getter instead of get_text
|
// FIXME: we need some user_input getter instead of get_text
|
||||||
let formula_or_value = self
|
let formula_or_value = self
|
||||||
.get_cell_formula(sheet, source_row, source_column)?
|
.get_cell_formula(sheet, source_row, source_column)?
|
||||||
.unwrap_or_else(|| source_cell.get_text(&self.workbook.shared_strings, &self.language));
|
.unwrap_or_else(|| {
|
||||||
|
source_cell.get_localized_text(&self.workbook.shared_strings, self.language)
|
||||||
|
});
|
||||||
self.set_user_input(sheet, target_row, target_column, formula_or_value)?;
|
self.set_user_input(sheet, target_row, target_column, formula_or_value)?;
|
||||||
self.workbook
|
self.workbook
|
||||||
.worksheet_mut(sheet)?
|
.worksheet_mut(sheet)?
|
||||||
@@ -490,9 +494,11 @@ impl Model {
|
|||||||
.cell(r.row, column)
|
.cell(r.row, column)
|
||||||
.ok_or("Expected Cell to exist")?;
|
.ok_or("Expected Cell to exist")?;
|
||||||
let style_idx = cell.get_style();
|
let style_idx = cell.get_style();
|
||||||
let formula_or_value = self
|
let formula_or_value =
|
||||||
.get_cell_formula(sheet, r.row, column)?
|
self.get_cell_formula(sheet, r.row, column)?
|
||||||
.unwrap_or_else(|| cell.get_text(&self.workbook.shared_strings, &self.language));
|
.unwrap_or_else(|| {
|
||||||
|
cell.get_localized_text(&self.workbook.shared_strings, self.language)
|
||||||
|
});
|
||||||
original_cells.push((r.row, formula_or_value, style_idx));
|
original_cells.push((r.row, formula_or_value, style_idx));
|
||||||
self.cell_clear_all(sheet, r.row, column)?;
|
self.cell_clear_all(sheet, r.row, column)?;
|
||||||
}
|
}
|
||||||
@@ -577,9 +583,9 @@ impl Model {
|
|||||||
.cell(row, *c)
|
.cell(row, *c)
|
||||||
.ok_or("Expected Cell to exist")?;
|
.ok_or("Expected Cell to exist")?;
|
||||||
let style_idx = cell.get_style();
|
let style_idx = cell.get_style();
|
||||||
let formula_or_value = self
|
let formula_or_value = self.get_cell_formula(sheet, row, *c)?.unwrap_or_else(|| {
|
||||||
.get_cell_formula(sheet, row, *c)?
|
cell.get_localized_text(&self.workbook.shared_strings, self.language)
|
||||||
.unwrap_or_else(|| cell.get_text(&self.workbook.shared_strings, &self.language));
|
});
|
||||||
original_cells.push((*c, formula_or_value, style_idx));
|
original_cells.push((*c, formula_or_value, style_idx));
|
||||||
self.cell_clear_all(sheet, row, *c)?;
|
self.cell_clear_all(sheet, row, *c)?;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,7 +22,7 @@ fn to_f64(value: &ArrayNode) -> Result<f64, Error> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
/// Applies `op` element‐wise for arrays/numbers.
|
/// Applies `op` element‐wise for arrays/numbers.
|
||||||
pub(crate) fn handle_arithmetic(
|
pub(crate) fn handle_arithmetic(
|
||||||
&mut self,
|
&mut self,
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ use crate::{
|
|||||||
token::Error,
|
token::Error,
|
||||||
types::CellReferenceIndex,
|
types::CellReferenceIndex,
|
||||||
},
|
},
|
||||||
|
formatter::format::parse_formatted_number,
|
||||||
model::Model,
|
model::Model,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -13,7 +14,32 @@ pub(crate) enum NumberOrArray {
|
|||||||
Array(Vec<Vec<ArrayNode>>),
|
Array(Vec<Vec<ArrayNode>>),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn cast_number(&self, s: &str) -> Option<f64> {
|
||||||
|
match s.trim().parse::<f64>() {
|
||||||
|
Ok(f) => Some(f),
|
||||||
|
_ => {
|
||||||
|
let currency = &self.locale.currency.symbol;
|
||||||
|
let mut currencies = vec!["$", "€"];
|
||||||
|
if !currencies.iter().any(|e| *e == currency) {
|
||||||
|
currencies.push(currency);
|
||||||
|
}
|
||||||
|
let (decimal_separator, group_separator) =
|
||||||
|
if self.locale.numbers.symbols.decimal == "," {
|
||||||
|
(b',', b'.')
|
||||||
|
} else {
|
||||||
|
(b'.', b',')
|
||||||
|
};
|
||||||
|
// Try to parse as a formatted number (e.g., dates, currencies, percentages)
|
||||||
|
if let Ok((v, _number_format)) =
|
||||||
|
parse_formatted_number(s, ¤cies, decimal_separator, group_separator)
|
||||||
|
{
|
||||||
|
return Some(v);
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
pub(crate) fn get_number_or_array(
|
pub(crate) fn get_number_or_array(
|
||||||
&mut self,
|
&mut self,
|
||||||
node: &Node,
|
node: &Node,
|
||||||
@@ -21,9 +47,9 @@ impl Model {
|
|||||||
) -> Result<NumberOrArray, CalcResult> {
|
) -> Result<NumberOrArray, CalcResult> {
|
||||||
match self.evaluate_node_in_context(node, cell) {
|
match self.evaluate_node_in_context(node, cell) {
|
||||||
CalcResult::Number(f) => Ok(NumberOrArray::Number(f)),
|
CalcResult::Number(f) => Ok(NumberOrArray::Number(f)),
|
||||||
CalcResult::String(s) => match s.parse::<f64>() {
|
CalcResult::String(s) => match self.cast_number(&s) {
|
||||||
Ok(f) => Ok(NumberOrArray::Number(f)),
|
Some(f) => Ok(NumberOrArray::Number(f)),
|
||||||
_ => Err(CalcResult::new_error(
|
None => Err(CalcResult::new_error(
|
||||||
Error::VALUE,
|
Error::VALUE,
|
||||||
cell,
|
cell,
|
||||||
"Expecting number".to_string(),
|
"Expecting number".to_string(),
|
||||||
@@ -89,16 +115,16 @@ impl Model {
|
|||||||
self.cast_to_number(result, cell)
|
self.cast_to_number(result, cell)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn cast_to_number(
|
pub(crate) fn cast_to_number(
|
||||||
&mut self,
|
&mut self,
|
||||||
result: CalcResult,
|
result: CalcResult,
|
||||||
cell: CellReferenceIndex,
|
cell: CellReferenceIndex,
|
||||||
) -> Result<f64, CalcResult> {
|
) -> Result<f64, CalcResult> {
|
||||||
match result {
|
match result {
|
||||||
CalcResult::Number(f) => Ok(f),
|
CalcResult::Number(f) => Ok(f),
|
||||||
CalcResult::String(s) => match s.parse::<f64>() {
|
CalcResult::String(s) => match self.cast_number(&s) {
|
||||||
Ok(f) => Ok(f),
|
Some(f) => Ok(f),
|
||||||
_ => Err(CalcResult::new_error(
|
None => Err(CalcResult::new_error(
|
||||||
Error::VALUE,
|
Error::VALUE,
|
||||||
cell,
|
cell,
|
||||||
"Expecting number".to_string(),
|
"Expecting number".to_string(),
|
||||||
|
|||||||
@@ -122,11 +122,17 @@ impl Cell {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_text(&self, shared_strings: &[String], language: &Language) -> String {
|
pub fn get_localized_text(&self, shared_strings: &[String], language: &Language) -> String {
|
||||||
match self.value(shared_strings, language) {
|
match self.value(shared_strings, language) {
|
||||||
CellValue::None => "".to_string(),
|
CellValue::None => "".to_string(),
|
||||||
CellValue::String(v) => v,
|
CellValue::String(v) => v,
|
||||||
CellValue::Boolean(v) => v.to_string().to_uppercase(),
|
CellValue::Boolean(v) => {
|
||||||
|
if v {
|
||||||
|
language.booleans.r#true.to_string()
|
||||||
|
} else {
|
||||||
|
language.booleans.r#false.to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
CellValue::Number(v) => to_excel_precision_str(v),
|
CellValue::Number(v) => to_excel_precision_str(v),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -171,7 +177,13 @@ impl Cell {
|
|||||||
match self.value(shared_strings, language) {
|
match self.value(shared_strings, language) {
|
||||||
CellValue::None => "".to_string(),
|
CellValue::None => "".to_string(),
|
||||||
CellValue::String(value) => value,
|
CellValue::String(value) => value,
|
||||||
CellValue::Boolean(value) => value.to_string().to_uppercase(),
|
CellValue::Boolean(value) => {
|
||||||
|
if value {
|
||||||
|
language.booleans.r#true.to_string()
|
||||||
|
} else {
|
||||||
|
language.booleans.r#false.to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
CellValue::Number(value) => format_number(value),
|
CellValue::Number(value) => format_number(value),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,6 +12,9 @@ pub(crate) const DEFAULT_WINDOW_WIDTH: i64 = 800;
|
|||||||
pub(crate) const LAST_COLUMN: i32 = 16_384;
|
pub(crate) const LAST_COLUMN: i32 = 16_384;
|
||||||
pub(crate) const LAST_ROW: i32 = 1_048_576;
|
pub(crate) const LAST_ROW: i32 = 1_048_576;
|
||||||
|
|
||||||
|
// Excel uses 15 significant digits of precision for all numeric calculations.
|
||||||
|
pub(crate) const EXCEL_PRECISION: usize = 15;
|
||||||
|
|
||||||
// 693_594 is computed as:
|
// 693_594 is computed as:
|
||||||
// NaiveDate::from_ymd(1900, 1, 1).num_days_from_ce() - 2
|
// NaiveDate::from_ymd(1900, 1, 1).num_days_from_ce() - 2
|
||||||
// The 2 days offset is because of Excel 1900 bug
|
// The 2 days offset is because of Excel 1900 bug
|
||||||
|
|||||||
@@ -79,19 +79,24 @@ pub enum LexerMode {
|
|||||||
|
|
||||||
/// Tokenize an input
|
/// Tokenize an input
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct Lexer {
|
pub struct Lexer<'a> {
|
||||||
position: usize,
|
position: usize,
|
||||||
next_token_position: Option<usize>,
|
next_token_position: Option<usize>,
|
||||||
len: usize,
|
len: usize,
|
||||||
chars: Vec<char>,
|
chars: Vec<char>,
|
||||||
mode: LexerMode,
|
mode: LexerMode,
|
||||||
locale: Locale,
|
locale: &'a Locale,
|
||||||
language: Language,
|
language: &'a Language,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Lexer {
|
impl<'a> Lexer<'a> {
|
||||||
/// Creates a new `Lexer` that returns the tokens of a formula.
|
/// Creates a new `Lexer` that returns the tokens of a formula.
|
||||||
pub fn new(formula: &str, mode: LexerMode, locale: &Locale, language: &Language) -> Lexer {
|
pub fn new(
|
||||||
|
formula: &str,
|
||||||
|
mode: LexerMode,
|
||||||
|
locale: &'a Locale,
|
||||||
|
language: &'a Language,
|
||||||
|
) -> Lexer<'a> {
|
||||||
let chars: Vec<char> = formula.chars().collect();
|
let chars: Vec<char> = formula.chars().collect();
|
||||||
let len = chars.len();
|
let len = chars.len();
|
||||||
Lexer {
|
Lexer {
|
||||||
@@ -100,8 +105,8 @@ impl Lexer {
|
|||||||
next_token_position: None,
|
next_token_position: None,
|
||||||
len,
|
len,
|
||||||
mode,
|
mode,
|
||||||
locale: locale.clone(),
|
locale,
|
||||||
language: language.clone(),
|
language,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -110,6 +115,16 @@ impl Lexer {
|
|||||||
self.mode = mode;
|
self.mode = mode;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sets the locale
|
||||||
|
pub fn set_locale(&mut self, locale: &'a Locale) {
|
||||||
|
self.locale = locale;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the language
|
||||||
|
pub fn set_language(&mut self, language: &'a Language) {
|
||||||
|
self.language = language;
|
||||||
|
}
|
||||||
|
|
||||||
// FIXME: I don't think we should have `is_a1_mode` and `get_formula`.
|
// FIXME: I don't think we should have `is_a1_mode` and `get_formula`.
|
||||||
// The caller already knows those two
|
// The caller already knows those two
|
||||||
|
|
||||||
@@ -188,6 +203,7 @@ impl Lexer {
|
|||||||
':' => TokenType::Colon,
|
':' => TokenType::Colon,
|
||||||
';' => TokenType::Semicolon,
|
';' => TokenType::Semicolon,
|
||||||
'@' => TokenType::At,
|
'@' => TokenType::At,
|
||||||
|
'\\' => TokenType::Backslash,
|
||||||
',' => {
|
',' => {
|
||||||
if self.locale.numbers.symbols.decimal == "," {
|
if self.locale.numbers.symbols.decimal == "," {
|
||||||
match self.consume_number(',') {
|
match self.consume_number(',') {
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ use crate::expressions::{token::TokenType, utils::column_to_number};
|
|||||||
use super::Lexer;
|
use super::Lexer;
|
||||||
use super::{ParsedRange, ParsedReference, Result};
|
use super::{ParsedRange, ParsedReference, Result};
|
||||||
|
|
||||||
impl Lexer {
|
impl<'a> Lexer<'a> {
|
||||||
/// Consumes a reference in A1 style like:
|
/// Consumes a reference in A1 style like:
|
||||||
/// AS23, $AS23, AS$23, $AS$23, R12
|
/// AS23, $AS23, AS$23, $AS$23, R12
|
||||||
/// Or returns an error
|
/// Or returns an error
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ use crate::expressions::token::{TableReference, TableSpecifier};
|
|||||||
use super::Result;
|
use super::Result;
|
||||||
use super::{Lexer, LexerError};
|
use super::{Lexer, LexerError};
|
||||||
|
|
||||||
impl Lexer {
|
impl<'a> Lexer<'a> {
|
||||||
fn consume_table_specifier(&mut self) -> Result<Option<TableSpecifier>> {
|
fn consume_table_specifier(&mut self) -> Result<Option<TableSpecifier>> {
|
||||||
if self.peek_char() == Some('#') {
|
if self.peek_char() == Some('#') {
|
||||||
// It's a specifier
|
// It's a specifier
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ use crate::expressions::{
|
|||||||
types::ParsedReference,
|
types::ParsedReference,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn new_lexer(formula: &str, a1_mode: bool) -> Lexer {
|
fn new_lexer(formula: &str, a1_mode: bool) -> Lexer<'_> {
|
||||||
let locale = get_locale("en").unwrap();
|
let locale = get_locale("en").unwrap();
|
||||||
let language = get_language("en").unwrap();
|
let language = get_language("en").unwrap();
|
||||||
let mode = if a1_mode {
|
let mode = if a1_mode {
|
||||||
@@ -655,7 +655,9 @@ fn test_comma() {
|
|||||||
|
|
||||||
// Used for testing locales where the comma is the decimal separator
|
// Used for testing locales where the comma is the decimal separator
|
||||||
let mut lx = new_lexer("12,34", false);
|
let mut lx = new_lexer("12,34", false);
|
||||||
lx.locale.numbers.symbols.decimal = ",".to_string();
|
let locale = get_locale("de").unwrap();
|
||||||
|
lx.locale = locale;
|
||||||
|
|
||||||
assert_eq!(lx.next_token(), Number(12.34));
|
assert_eq!(lx.next_token(), Number(12.34));
|
||||||
assert_eq!(lx.next_token(), EOF);
|
assert_eq!(lx.next_token(), EOF);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ use crate::expressions::{
|
|||||||
use crate::language::get_language;
|
use crate::language::get_language;
|
||||||
use crate::locale::get_locale;
|
use crate::locale::get_locale;
|
||||||
|
|
||||||
fn new_lexer(formula: &str) -> Lexer {
|
fn new_lexer(formula: &str) -> Lexer<'_> {
|
||||||
let locale = get_locale("en").unwrap();
|
let locale = get_locale("en").unwrap();
|
||||||
let language = get_language("en").unwrap();
|
let language = get_language("en").unwrap();
|
||||||
Lexer::new(formula, LexerMode::A1, locale, language)
|
Lexer::new(formula, LexerMode::A1, locale, language)
|
||||||
|
|||||||
@@ -6,11 +6,11 @@ use crate::{
|
|||||||
token::{Error, TokenType},
|
token::{Error, TokenType},
|
||||||
},
|
},
|
||||||
language::get_language,
|
language::get_language,
|
||||||
locale::get_locale,
|
locale::get_default_locale,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn new_language_lexer(formula: &str, language: &str) -> Lexer {
|
fn new_language_lexer<'a>(formula: &str, language: &str) -> Lexer<'a> {
|
||||||
let locale = get_locale("en").unwrap();
|
let locale = get_default_locale();
|
||||||
let language = get_language(language).unwrap();
|
let language = get_language(language).unwrap();
|
||||||
Lexer::new(formula, LexerMode::A1, locale, language)
|
Lexer::new(formula, LexerMode::A1, locale, language)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ use crate::{
|
|||||||
locale::get_locale,
|
locale::get_locale,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn new_language_lexer(formula: &str, locale: &str, language: &str) -> Lexer {
|
fn new_language_lexer<'a>(formula: &str, locale: &str, language: &str) -> Lexer<'a> {
|
||||||
let locale = get_locale(locale).unwrap();
|
let locale = get_locale(locale).unwrap();
|
||||||
let language = get_language(language).unwrap();
|
let language = get_language(language).unwrap();
|
||||||
Lexer::new(formula, LexerMode::A1, locale, language)
|
Lexer::new(formula, LexerMode::A1, locale, language)
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ use crate::expressions::{
|
|||||||
use crate::language::get_language;
|
use crate::language::get_language;
|
||||||
use crate::locale::get_locale;
|
use crate::locale::get_locale;
|
||||||
|
|
||||||
fn new_lexer(formula: &str) -> Lexer {
|
fn new_lexer(formula: &str) -> Lexer<'_> {
|
||||||
let locale = get_locale("en").unwrap();
|
let locale = get_locale("en").unwrap();
|
||||||
let language = get_language("en").unwrap();
|
let language = get_language("en").unwrap();
|
||||||
Lexer::new(formula, LexerMode::A1, locale, language)
|
Lexer::new(formula, LexerMode::A1, locale, language)
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ use crate::expressions::{
|
|||||||
use crate::language::get_language;
|
use crate::language::get_language;
|
||||||
use crate::locale::get_locale;
|
use crate::locale::get_locale;
|
||||||
|
|
||||||
fn new_lexer(formula: &str) -> Lexer {
|
fn new_lexer(formula: &str) -> Lexer<'_> {
|
||||||
let locale = get_locale("en").unwrap();
|
let locale = get_locale("en").unwrap();
|
||||||
let language = get_language("en").unwrap();
|
let language = get_language("en").unwrap();
|
||||||
Lexer::new(formula, LexerMode::A1, locale, language)
|
Lexer::new(formula, LexerMode::A1, locale, language)
|
||||||
|
|||||||
@@ -31,8 +31,12 @@ f_args => e (',' e)*
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::functions::Function;
|
use crate::functions::Function;
|
||||||
|
use crate::language::get_default_language;
|
||||||
use crate::language::get_language;
|
use crate::language::get_language;
|
||||||
|
use crate::language::Language;
|
||||||
|
use crate::locale::get_default_locale;
|
||||||
use crate::locale::get_locale;
|
use crate::locale::get_locale;
|
||||||
|
use crate::locale::Locale;
|
||||||
use crate::types::Table;
|
use crate::types::Table;
|
||||||
|
|
||||||
use super::lexer;
|
use super::lexer;
|
||||||
@@ -202,28 +206,35 @@ pub enum Node {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct Parser {
|
pub struct Parser<'a> {
|
||||||
lexer: lexer::Lexer,
|
lexer: lexer::Lexer<'a>,
|
||||||
worksheets: Vec<String>,
|
worksheets: Vec<String>,
|
||||||
defined_names: Vec<DefinedNameS>,
|
defined_names: Vec<DefinedNameS>,
|
||||||
context: CellReferenceRC,
|
context: CellReferenceRC,
|
||||||
tables: HashMap<String, Table>,
|
tables: HashMap<String, Table>,
|
||||||
|
locale: &'a Locale,
|
||||||
|
language: &'a Language,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Parser {
|
pub fn new_parser_english<'a>(
|
||||||
|
worksheets: Vec<String>,
|
||||||
|
defined_names: Vec<DefinedNameS>,
|
||||||
|
tables: HashMap<String, Table>,
|
||||||
|
) -> Parser<'a> {
|
||||||
|
let locale = get_default_locale();
|
||||||
|
let language = get_default_language();
|
||||||
|
Parser::new(worksheets, defined_names, tables, locale, language)
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> Parser<'a> {
|
||||||
pub fn new(
|
pub fn new(
|
||||||
worksheets: Vec<String>,
|
worksheets: Vec<String>,
|
||||||
defined_names: Vec<DefinedNameS>,
|
defined_names: Vec<DefinedNameS>,
|
||||||
tables: HashMap<String, Table>,
|
tables: HashMap<String, Table>,
|
||||||
) -> Parser {
|
locale: &'a Locale,
|
||||||
let lexer = lexer::Lexer::new(
|
language: &'a Language,
|
||||||
"",
|
) -> Parser<'a> {
|
||||||
lexer::LexerMode::A1,
|
let lexer = lexer::Lexer::new("", lexer::LexerMode::A1, locale, language);
|
||||||
#[allow(clippy::expect_used)]
|
|
||||||
get_locale("en").expect(""),
|
|
||||||
#[allow(clippy::expect_used)]
|
|
||||||
get_language("en").expect(""),
|
|
||||||
);
|
|
||||||
let context = CellReferenceRC {
|
let context = CellReferenceRC {
|
||||||
sheet: worksheets.first().map_or("", |v| v).to_string(),
|
sheet: worksheets.first().map_or("", |v| v).to_string(),
|
||||||
column: 1,
|
column: 1,
|
||||||
@@ -235,12 +246,24 @@ impl Parser {
|
|||||||
defined_names,
|
defined_names,
|
||||||
context,
|
context,
|
||||||
tables,
|
tables,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn set_lexer_mode(&mut self, mode: lexer::LexerMode) {
|
pub fn set_lexer_mode(&mut self, mode: lexer::LexerMode) {
|
||||||
self.lexer.set_lexer_mode(mode)
|
self.lexer.set_lexer_mode(mode)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn set_locale(&mut self, locale: &'a Locale) {
|
||||||
|
self.locale = locale;
|
||||||
|
self.lexer.set_locale(locale);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_language(&mut self, language: &'a Language) {
|
||||||
|
self.language = language;
|
||||||
|
self.lexer.set_language(language);
|
||||||
|
}
|
||||||
|
|
||||||
pub fn set_worksheets_and_names(
|
pub fn set_worksheets_and_names(
|
||||||
&mut self,
|
&mut self,
|
||||||
worksheets: Vec<String>,
|
worksheets: Vec<String>,
|
||||||
@@ -256,6 +279,27 @@ impl Parser {
|
|||||||
self.parse_expr()
|
self.parse_expr()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Returns the token used to separate arguments in functions and arrays
|
||||||
|
// If the locale decimal separator is '.', then it is a comma ','
|
||||||
|
// Otherwise, it is a semicolon ';'
|
||||||
|
fn get_argument_separator_token(&self) -> TokenType {
|
||||||
|
if self.locale.numbers.symbols.decimal == "." {
|
||||||
|
TokenType::Comma
|
||||||
|
} else {
|
||||||
|
TokenType::Semicolon
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns the token used to separate columns in arrays
|
||||||
|
// If the locale decimal separator is '.', then it is a semicolon ';'
|
||||||
|
fn get_column_separator_token(&self) -> TokenType {
|
||||||
|
if self.locale.numbers.symbols.decimal == "." {
|
||||||
|
TokenType::Semicolon
|
||||||
|
} else {
|
||||||
|
TokenType::Backslash
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn get_sheet_index_by_name(&self, name: &str) -> Option<u32> {
|
fn get_sheet_index_by_name(&self, name: &str) -> Option<u32> {
|
||||||
let worksheets = &self.worksheets;
|
let worksheets = &self.worksheets;
|
||||||
for (i, sheet) in worksheets.iter().enumerate() {
|
for (i, sheet) in worksheets.iter().enumerate() {
|
||||||
@@ -464,6 +508,7 @@ impl Parser {
|
|||||||
|
|
||||||
fn parse_array_row(&mut self) -> Result<Vec<ArrayNode>, Node> {
|
fn parse_array_row(&mut self) -> Result<Vec<ArrayNode>, Node> {
|
||||||
let mut row = Vec::new();
|
let mut row = Vec::new();
|
||||||
|
let column_separator_token = self.get_argument_separator_token();
|
||||||
// and array can only have numbers, string or booleans
|
// and array can only have numbers, string or booleans
|
||||||
// otherwise it is a syntax error
|
// otherwise it is a syntax error
|
||||||
let first_element = match self.parse_expr() {
|
let first_element = match self.parse_expr() {
|
||||||
@@ -471,6 +516,20 @@ impl Parser {
|
|||||||
Node::NumberKind(s) => ArrayNode::Number(s),
|
Node::NumberKind(s) => ArrayNode::Number(s),
|
||||||
Node::StringKind(s) => ArrayNode::String(s),
|
Node::StringKind(s) => ArrayNode::String(s),
|
||||||
Node::ErrorKind(kind) => ArrayNode::Error(kind),
|
Node::ErrorKind(kind) => ArrayNode::Error(kind),
|
||||||
|
Node::UnaryKind {
|
||||||
|
kind: OpUnary::Minus,
|
||||||
|
right,
|
||||||
|
} => {
|
||||||
|
if let Node::NumberKind(n) = *right {
|
||||||
|
ArrayNode::Number(-n)
|
||||||
|
} else {
|
||||||
|
return Err(Node::ParseErrorKind {
|
||||||
|
formula: self.lexer.get_formula(),
|
||||||
|
message: "Invalid value in array".to_string(),
|
||||||
|
position: self.lexer.get_position() as usize,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
error @ Node::ParseErrorKind { .. } => return Err(error),
|
error @ Node::ParseErrorKind { .. } => return Err(error),
|
||||||
_ => {
|
_ => {
|
||||||
return Err(Node::ParseErrorKind {
|
return Err(Node::ParseErrorKind {
|
||||||
@@ -482,14 +541,27 @@ impl Parser {
|
|||||||
};
|
};
|
||||||
row.push(first_element);
|
row.push(first_element);
|
||||||
let mut next_token = self.lexer.peek_token();
|
let mut next_token = self.lexer.peek_token();
|
||||||
// FIXME: this is not respecting the locale
|
while next_token == column_separator_token {
|
||||||
while next_token == TokenType::Comma {
|
|
||||||
self.lexer.advance_token();
|
self.lexer.advance_token();
|
||||||
let value = match self.parse_expr() {
|
let value = match self.parse_expr() {
|
||||||
Node::BooleanKind(s) => ArrayNode::Boolean(s),
|
Node::BooleanKind(s) => ArrayNode::Boolean(s),
|
||||||
Node::NumberKind(s) => ArrayNode::Number(s),
|
Node::NumberKind(s) => ArrayNode::Number(s),
|
||||||
Node::StringKind(s) => ArrayNode::String(s),
|
Node::StringKind(s) => ArrayNode::String(s),
|
||||||
Node::ErrorKind(kind) => ArrayNode::Error(kind),
|
Node::ErrorKind(kind) => ArrayNode::Error(kind),
|
||||||
|
Node::UnaryKind {
|
||||||
|
kind: OpUnary::Minus,
|
||||||
|
right,
|
||||||
|
} => {
|
||||||
|
if let Node::NumberKind(n) = *right {
|
||||||
|
ArrayNode::Number(-n)
|
||||||
|
} else {
|
||||||
|
return Err(Node::ParseErrorKind {
|
||||||
|
formula: self.lexer.get_formula(),
|
||||||
|
message: "Invalid value in array".to_string(),
|
||||||
|
position: self.lexer.get_position() as usize,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
error @ Node::ParseErrorKind { .. } => return Err(error),
|
error @ Node::ParseErrorKind { .. } => return Err(error),
|
||||||
_ => {
|
_ => {
|
||||||
return Err(Node::ParseErrorKind {
|
return Err(Node::ParseErrorKind {
|
||||||
@@ -527,6 +599,7 @@ impl Parser {
|
|||||||
TokenType::String(s) => Node::StringKind(s),
|
TokenType::String(s) => Node::StringKind(s),
|
||||||
TokenType::LeftBrace => {
|
TokenType::LeftBrace => {
|
||||||
// It's an array. It's a collection of rows all of the same dimension
|
// It's an array. It's a collection of rows all of the same dimension
|
||||||
|
let column_separator_token = self.get_column_separator_token();
|
||||||
|
|
||||||
let first_row = match self.parse_array_row() {
|
let first_row = match self.parse_array_row() {
|
||||||
Ok(s) => s,
|
Ok(s) => s,
|
||||||
@@ -536,9 +609,8 @@ impl Parser {
|
|||||||
|
|
||||||
let mut matrix = Vec::new();
|
let mut matrix = Vec::new();
|
||||||
matrix.push(first_row);
|
matrix.push(first_row);
|
||||||
// FIXME: this is not respecting the locale
|
|
||||||
let mut next_token = self.lexer.peek_token();
|
let mut next_token = self.lexer.peek_token();
|
||||||
while next_token == TokenType::Semicolon {
|
while next_token == column_separator_token {
|
||||||
self.lexer.advance_token();
|
self.lexer.advance_token();
|
||||||
let row = match self.parse_array_row() {
|
let row = match self.parse_array_row() {
|
||||||
Ok(s) => s,
|
Ok(s) => s,
|
||||||
@@ -687,12 +759,7 @@ impl Parser {
|
|||||||
message: err.message,
|
message: err.message,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
if let Some(function_kind) = Function::get_function(&name) {
|
// We should do this *only* importing functions from xlsx
|
||||||
return Node::FunctionKind {
|
|
||||||
kind: function_kind,
|
|
||||||
args,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
if &name == "_xlfn.SINGLE" {
|
if &name == "_xlfn.SINGLE" {
|
||||||
if args.len() != 1 {
|
if args.len() != 1 {
|
||||||
return Node::ParseErrorKind {
|
return Node::ParseErrorKind {
|
||||||
@@ -707,6 +774,17 @@ impl Parser {
|
|||||||
child: Box::new(args[0].clone()),
|
child: Box::new(args[0].clone()),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
// We should do this *only* importing functions from xlsx
|
||||||
|
if let Some(function_kind) = self
|
||||||
|
.language
|
||||||
|
.functions
|
||||||
|
.lookup(name.trim_start_matches("_xlfn."))
|
||||||
|
{
|
||||||
|
return Node::FunctionKind {
|
||||||
|
kind: function_kind,
|
||||||
|
args,
|
||||||
|
};
|
||||||
|
}
|
||||||
return Node::InvalidFunctionKind { name, args };
|
return Node::InvalidFunctionKind { name, args };
|
||||||
}
|
}
|
||||||
let context = &self.context;
|
let context = &self.context;
|
||||||
@@ -821,6 +899,7 @@ impl Parser {
|
|||||||
| TokenType::RightBracket
|
| TokenType::RightBracket
|
||||||
| TokenType::Colon
|
| TokenType::Colon
|
||||||
| TokenType::Semicolon
|
| TokenType::Semicolon
|
||||||
|
| TokenType::Backslash
|
||||||
| TokenType::RightBrace
|
| TokenType::RightBrace
|
||||||
| TokenType::Comma
|
| TokenType::Comma
|
||||||
| TokenType::Bang
|
| TokenType::Bang
|
||||||
@@ -1020,12 +1099,13 @@ impl Parser {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn parse_function_args(&mut self) -> Result<Vec<Node>, Node> {
|
fn parse_function_args(&mut self) -> Result<Vec<Node>, Node> {
|
||||||
|
let arg_separator_token = &self.get_argument_separator_token();
|
||||||
let mut args: Vec<Node> = Vec::new();
|
let mut args: Vec<Node> = Vec::new();
|
||||||
let mut next_token = self.lexer.peek_token();
|
let mut next_token = self.lexer.peek_token();
|
||||||
if next_token == TokenType::RightParenthesis {
|
if next_token == TokenType::RightParenthesis {
|
||||||
return Ok(args);
|
return Ok(args);
|
||||||
}
|
}
|
||||||
if self.lexer.peek_token() == TokenType::Comma {
|
if &self.lexer.peek_token() == arg_separator_token {
|
||||||
args.push(Node::EmptyArgKind);
|
args.push(Node::EmptyArgKind);
|
||||||
} else {
|
} else {
|
||||||
let t = self.parse_expr();
|
let t = self.parse_expr();
|
||||||
@@ -1035,11 +1115,11 @@ impl Parser {
|
|||||||
args.push(t);
|
args.push(t);
|
||||||
}
|
}
|
||||||
next_token = self.lexer.peek_token();
|
next_token = self.lexer.peek_token();
|
||||||
while next_token == TokenType::Comma {
|
while &next_token == arg_separator_token {
|
||||||
self.lexer.advance_token();
|
self.lexer.advance_token();
|
||||||
if self.lexer.peek_token() == TokenType::Comma {
|
if &self.lexer.peek_token() == arg_separator_token {
|
||||||
args.push(Node::EmptyArgKind);
|
args.push(Node::EmptyArgKind);
|
||||||
next_token = TokenType::Comma;
|
next_token = arg_separator_token.clone();
|
||||||
} else if self.lexer.peek_token() == TokenType::RightParenthesis {
|
} else if self.lexer.peek_token() == TokenType::RightParenthesis {
|
||||||
args.push(Node::EmptyArgKind);
|
args.push(Node::EmptyArgKind);
|
||||||
return Ok(args);
|
return Ok(args);
|
||||||
|
|||||||
@@ -5,6 +5,8 @@ use super::{
|
|||||||
use crate::{
|
use crate::{
|
||||||
constants::{LAST_COLUMN, LAST_ROW},
|
constants::{LAST_COLUMN, LAST_ROW},
|
||||||
expressions::token::OpUnary,
|
expressions::token::OpUnary,
|
||||||
|
language::Language,
|
||||||
|
locale::Locale,
|
||||||
};
|
};
|
||||||
use crate::{
|
use crate::{
|
||||||
expressions::types::{Area, CellReferenceRC},
|
expressions::types::{Area, CellReferenceRC},
|
||||||
@@ -38,38 +40,78 @@ pub(crate) struct MoveContext<'a> {
|
|||||||
/// We are moving a formula in (row, column) to (row+row_delta, column + column_delta).
|
/// We are moving a formula in (row, column) to (row+row_delta, column + column_delta).
|
||||||
/// All references that do not point to a cell in area will be left untouched.
|
/// All references that do not point to a cell in area will be left untouched.
|
||||||
/// All references that point to a cell in area will be displaced
|
/// All references that point to a cell in area will be displaced
|
||||||
pub(crate) fn move_formula(node: &Node, move_context: &MoveContext) -> String {
|
pub(crate) fn move_formula(
|
||||||
to_string_moved(node, move_context)
|
node: &Node,
|
||||||
|
move_context: &MoveContext,
|
||||||
|
locale: &Locale,
|
||||||
|
language: &Language,
|
||||||
|
) -> String {
|
||||||
|
to_string_moved(node, move_context, locale, language)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn move_function(name: &str, args: &Vec<Node>, move_context: &MoveContext) -> String {
|
fn move_function(
|
||||||
|
name: &str,
|
||||||
|
args: &Vec<Node>,
|
||||||
|
move_context: &MoveContext,
|
||||||
|
locale: &Locale,
|
||||||
|
language: &Language,
|
||||||
|
) -> String {
|
||||||
let mut first = true;
|
let mut first = true;
|
||||||
let mut arguments = "".to_string();
|
let mut arguments = "".to_string();
|
||||||
for el in args {
|
for el in args {
|
||||||
if !first {
|
if !first {
|
||||||
arguments = format!("{},{}", arguments, to_string_moved(el, move_context));
|
arguments = format!(
|
||||||
|
"{},{}",
|
||||||
|
arguments,
|
||||||
|
to_string_moved(el, move_context, locale, language)
|
||||||
|
);
|
||||||
} else {
|
} else {
|
||||||
first = false;
|
first = false;
|
||||||
arguments = to_string_moved(el, move_context);
|
arguments = to_string_moved(el, move_context, locale, language);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
format!("{name}({arguments})")
|
format!("{name}({arguments})")
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn to_string_array_node(node: &ArrayNode) -> String {
|
fn format_number_locale(number: f64, locale: &Locale) -> String {
|
||||||
|
let s = to_excel_precision_str(number);
|
||||||
|
let decimal = &locale.numbers.symbols.decimal;
|
||||||
|
if decimal == "." {
|
||||||
|
s
|
||||||
|
} else {
|
||||||
|
s.replace('.', decimal)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn to_string_array_node(
|
||||||
|
node: &ArrayNode,
|
||||||
|
locale: &Locale,
|
||||||
|
language: &Language,
|
||||||
|
) -> String {
|
||||||
match node {
|
match node {
|
||||||
ArrayNode::Boolean(value) => format!("{value}").to_ascii_uppercase(),
|
ArrayNode::Boolean(value) => {
|
||||||
ArrayNode::Number(number) => to_excel_precision_str(*number),
|
if *value {
|
||||||
|
language.booleans.r#true.to_ascii_uppercase()
|
||||||
|
} else {
|
||||||
|
language.booleans.r#false.to_ascii_uppercase()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ArrayNode::Number(number) => format_number_locale(*number, locale),
|
||||||
ArrayNode::String(value) => format!("\"{value}\""),
|
ArrayNode::String(value) => format!("\"{value}\""),
|
||||||
ArrayNode::Error(kind) => format!("{kind}"),
|
ArrayNode::Error(kind) => format!("{kind}"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn to_string_moved(node: &Node, move_context: &MoveContext) -> String {
|
fn to_string_moved(
|
||||||
|
node: &Node,
|
||||||
|
move_context: &MoveContext,
|
||||||
|
locale: &Locale,
|
||||||
|
language: &Language,
|
||||||
|
) -> String {
|
||||||
use self::Node::*;
|
use self::Node::*;
|
||||||
match node {
|
match node {
|
||||||
BooleanKind(value) => format!("{value}").to_ascii_uppercase(),
|
BooleanKind(value) => format!("{value}").to_ascii_uppercase(),
|
||||||
NumberKind(number) => to_excel_precision_str(*number),
|
NumberKind(number) => format_number_locale(*number, locale),
|
||||||
StringKind(value) => format!("\"{value}\""),
|
StringKind(value) => format!("\"{value}\""),
|
||||||
ReferenceKind {
|
ReferenceKind {
|
||||||
sheet_name,
|
sheet_name,
|
||||||
@@ -329,55 +371,81 @@ fn to_string_moved(node: &Node, move_context: &MoveContext) -> String {
|
|||||||
}
|
}
|
||||||
OpRangeKind { left, right } => format!(
|
OpRangeKind { left, right } => format!(
|
||||||
"{}:{}",
|
"{}:{}",
|
||||||
to_string_moved(left, move_context),
|
to_string_moved(left, move_context, locale, language),
|
||||||
to_string_moved(right, move_context),
|
to_string_moved(right, move_context, locale, language),
|
||||||
),
|
),
|
||||||
OpConcatenateKind { left, right } => format!(
|
OpConcatenateKind { left, right } => format!(
|
||||||
"{}&{}",
|
"{}&{}",
|
||||||
to_string_moved(left, move_context),
|
to_string_moved(left, move_context, locale, language),
|
||||||
to_string_moved(right, move_context),
|
to_string_moved(right, move_context, locale, language),
|
||||||
),
|
),
|
||||||
OpSumKind { kind, left, right } => format!(
|
OpSumKind { kind, left, right } => format!(
|
||||||
"{}{}{}",
|
"{}{}{}",
|
||||||
to_string_moved(left, move_context),
|
to_string_moved(left, move_context, locale, language),
|
||||||
kind,
|
kind,
|
||||||
to_string_moved(right, move_context),
|
to_string_moved(right, move_context, locale, language),
|
||||||
),
|
),
|
||||||
OpProductKind { kind, left, right } => {
|
OpProductKind { kind, left, right } => {
|
||||||
let x = match **left {
|
let x = match **left {
|
||||||
OpSumKind { .. } => format!("({})", to_string_moved(left, move_context)),
|
OpSumKind { .. } => format!(
|
||||||
CompareKind { .. } => format!("({})", to_string_moved(left, move_context)),
|
"({})",
|
||||||
_ => to_string_moved(left, move_context),
|
to_string_moved(left, move_context, locale, language)
|
||||||
|
),
|
||||||
|
CompareKind { .. } => format!(
|
||||||
|
"({})",
|
||||||
|
to_string_moved(left, move_context, locale, language)
|
||||||
|
),
|
||||||
|
_ => to_string_moved(left, move_context, locale, language),
|
||||||
};
|
};
|
||||||
let y = match **right {
|
let y = match **right {
|
||||||
OpSumKind { .. } => format!("({})", to_string_moved(right, move_context)),
|
OpSumKind { .. } => format!(
|
||||||
CompareKind { .. } => format!("({})", to_string_moved(right, move_context)),
|
"({})",
|
||||||
OpProductKind { .. } => format!("({})", to_string_moved(right, move_context)),
|
to_string_moved(right, move_context, locale, language)
|
||||||
|
),
|
||||||
|
CompareKind { .. } => format!(
|
||||||
|
"({})",
|
||||||
|
to_string_moved(right, move_context, locale, language)
|
||||||
|
),
|
||||||
|
OpProductKind { .. } => format!(
|
||||||
|
"({})",
|
||||||
|
to_string_moved(right, move_context, locale, language)
|
||||||
|
),
|
||||||
UnaryKind { .. } => {
|
UnaryKind { .. } => {
|
||||||
format!("({})", to_string_moved(right, move_context))
|
format!(
|
||||||
|
"({})",
|
||||||
|
to_string_moved(right, move_context, locale, language)
|
||||||
|
)
|
||||||
}
|
}
|
||||||
_ => to_string_moved(right, move_context),
|
_ => to_string_moved(right, move_context, locale, language),
|
||||||
};
|
};
|
||||||
format!("{x}{kind}{y}")
|
format!("{x}{kind}{y}")
|
||||||
}
|
}
|
||||||
OpPowerKind { left, right } => format!(
|
OpPowerKind { left, right } => format!(
|
||||||
"{}^{}",
|
"{}^{}",
|
||||||
to_string_moved(left, move_context),
|
to_string_moved(left, move_context, locale, language),
|
||||||
to_string_moved(right, move_context),
|
to_string_moved(right, move_context, locale, language),
|
||||||
),
|
),
|
||||||
InvalidFunctionKind { name, args } => move_function(name, args, move_context),
|
InvalidFunctionKind { name, args } => {
|
||||||
|
move_function(name, args, move_context, locale, language)
|
||||||
|
}
|
||||||
FunctionKind { kind, args } => {
|
FunctionKind { kind, args } => {
|
||||||
let name = &kind.to_string();
|
let name = &kind.to_localized_name(language);
|
||||||
move_function(name, args, move_context)
|
move_function(name, args, move_context, locale, language)
|
||||||
}
|
}
|
||||||
ArrayKind(args) => {
|
ArrayKind(args) => {
|
||||||
let mut first_row = true;
|
let mut first_row = true;
|
||||||
let mut matrix_string = String::new();
|
let mut matrix_string = String::new();
|
||||||
|
|
||||||
// Each element in `args` is assumed to be one "row" (itself a `Vec<T>`).
|
// Each element in `args` is assumed to be one "row" (itself a `Vec<T>`).
|
||||||
|
let row_separator = if locale.numbers.symbols.decimal == "." {
|
||||||
|
';'
|
||||||
|
} else {
|
||||||
|
'/'
|
||||||
|
};
|
||||||
|
let col_separator = if row_separator == ';' { ',' } else { ';' };
|
||||||
for row in args {
|
for row in args {
|
||||||
if !first_row {
|
if !first_row {
|
||||||
matrix_string.push(',');
|
matrix_string.push(col_separator);
|
||||||
} else {
|
} else {
|
||||||
first_row = false;
|
first_row = false;
|
||||||
}
|
}
|
||||||
@@ -387,13 +455,13 @@ fn to_string_moved(node: &Node, move_context: &MoveContext) -> String {
|
|||||||
let mut row_string = String::new();
|
let mut row_string = String::new();
|
||||||
for el in row {
|
for el in row {
|
||||||
if !first_col {
|
if !first_col {
|
||||||
row_string.push(',');
|
row_string.push(row_separator);
|
||||||
} else {
|
} else {
|
||||||
first_col = false;
|
first_col = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Reuse your existing element-stringification function
|
// Reuse your existing element-stringification function
|
||||||
row_string.push_str(&to_string_array_node(el));
|
row_string.push_str(&to_string_array_node(el, locale, language));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Enclose the row in braces
|
// Enclose the row in braces
|
||||||
@@ -410,13 +478,19 @@ fn to_string_moved(node: &Node, move_context: &MoveContext) -> String {
|
|||||||
WrongVariableKind(name) => name.to_string(),
|
WrongVariableKind(name) => name.to_string(),
|
||||||
CompareKind { kind, left, right } => format!(
|
CompareKind { kind, left, right } => format!(
|
||||||
"{}{}{}",
|
"{}{}{}",
|
||||||
to_string_moved(left, move_context),
|
to_string_moved(left, move_context, locale, language),
|
||||||
kind,
|
kind,
|
||||||
to_string_moved(right, move_context),
|
to_string_moved(right, move_context, locale, language),
|
||||||
),
|
),
|
||||||
UnaryKind { kind, right } => match kind {
|
UnaryKind { kind, right } => match kind {
|
||||||
OpUnary::Minus => format!("-{}", to_string_moved(right, move_context)),
|
OpUnary::Minus => format!(
|
||||||
OpUnary::Percentage => format!("{}%", to_string_moved(right, move_context)),
|
"-{}",
|
||||||
|
to_string_moved(right, move_context, locale, language)
|
||||||
|
),
|
||||||
|
OpUnary::Percentage => format!(
|
||||||
|
"{}%",
|
||||||
|
to_string_moved(right, move_context, locale, language)
|
||||||
|
),
|
||||||
},
|
},
|
||||||
ErrorKind(kind) => format!("{kind}"),
|
ErrorKind(kind) => format!("{kind}"),
|
||||||
ParseErrorKind {
|
ParseErrorKind {
|
||||||
@@ -429,7 +503,10 @@ fn to_string_moved(node: &Node, move_context: &MoveContext) -> String {
|
|||||||
automatic: _,
|
automatic: _,
|
||||||
child,
|
child,
|
||||||
} => {
|
} => {
|
||||||
format!("@{}", to_string_moved(child, move_context))
|
format!(
|
||||||
|
"@{}",
|
||||||
|
to_string_moved(child, move_context, locale, language)
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -711,6 +711,7 @@ fn get_function_args_signature(kind: &Function, arg_count: usize) -> Vec<Signatu
|
|||||||
Function::Value => args_signature_scalars(arg_count, 1, 0),
|
Function::Value => args_signature_scalars(arg_count, 1, 0),
|
||||||
Function::Valuetotext => args_signature_scalars(arg_count, 1, 1),
|
Function::Valuetotext => args_signature_scalars(arg_count, 1, 1),
|
||||||
Function::Average => vec![Signature::Vector; arg_count],
|
Function::Average => vec![Signature::Vector; arg_count],
|
||||||
|
Function::Avedev => vec![Signature::Vector; arg_count],
|
||||||
Function::Averagea => vec![Signature::Vector; arg_count],
|
Function::Averagea => vec![Signature::Vector; arg_count],
|
||||||
Function::Averageif => args_signature_sumif(arg_count),
|
Function::Averageif => args_signature_sumif(arg_count),
|
||||||
Function::Averageifs => vec![Signature::Vector; arg_count],
|
Function::Averageifs => vec![Signature::Vector; arg_count],
|
||||||
@@ -871,11 +872,141 @@ fn get_function_args_signature(kind: &Function, arg_count: usize) -> Vec<Signatu
|
|||||||
Function::Combin => args_signature_scalars(arg_count, 2, 0),
|
Function::Combin => args_signature_scalars(arg_count, 2, 0),
|
||||||
Function::Combina => args_signature_scalars(arg_count, 2, 0),
|
Function::Combina => args_signature_scalars(arg_count, 2, 0),
|
||||||
Function::Sumsq => vec![Signature::Vector; arg_count],
|
Function::Sumsq => vec![Signature::Vector; arg_count],
|
||||||
|
|
||||||
Function::N => args_signature_scalars(arg_count, 1, 0),
|
Function::N => args_signature_scalars(arg_count, 1, 0),
|
||||||
Function::Sheets => args_signature_scalars(arg_count, 0, 1),
|
Function::Sheets => args_signature_scalars(arg_count, 0, 1),
|
||||||
Function::Cell => args_signature_scalars(arg_count, 1, 1),
|
Function::Cell => args_signature_scalars(arg_count, 1, 1),
|
||||||
Function::Info => args_signature_scalars(arg_count, 1, 1),
|
Function::Info => args_signature_scalars(arg_count, 1, 1),
|
||||||
|
Function::Daverage => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dcount => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dget => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dmax => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dmin => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dsum => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dcounta => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dproduct => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dstdev => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dvar => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dvarp => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::Dstdevp => vec![Signature::Vector, Signature::Scalar, Signature::Vector],
|
||||||
|
Function::BetaDist => args_signature_scalars(arg_count, 4, 2),
|
||||||
|
Function::BetaInv => args_signature_scalars(arg_count, 3, 2),
|
||||||
|
Function::BinomDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::BinomDistRange => args_signature_scalars(arg_count, 3, 1),
|
||||||
|
Function::BinomInv => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::ChisqDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::ChisqDistRT => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::ChisqInv => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::ChisqInvRT => args_signature_scalars(arg_count, 2, 0),
|
||||||
|
Function::ChisqTest => {
|
||||||
|
if arg_count == 2 {
|
||||||
|
vec![Signature::Vector, Signature::Vector]
|
||||||
|
} else {
|
||||||
|
vec![Signature::Error; arg_count]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Function::ConfidenceNorm => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::ConfidenceT => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::CovarianceP => {
|
||||||
|
if arg_count == 2 {
|
||||||
|
vec![Signature::Vector, Signature::Vector]
|
||||||
|
} else {
|
||||||
|
vec![Signature::Error; arg_count]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Function::CovarianceS => {
|
||||||
|
if arg_count == 2 {
|
||||||
|
vec![Signature::Vector, Signature::Vector]
|
||||||
|
} else {
|
||||||
|
vec![Signature::Error; arg_count]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Function::Devsq => vec![Signature::Vector; arg_count],
|
||||||
|
Function::ExponDist => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::FDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::FDistRT => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::FInv => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::FInvRT => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::FTest => vec![Signature::Vector; 2],
|
||||||
|
Function::Fisher => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::FisherInv => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::Gamma => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::GammaDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::GammaInv => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::GammaLn => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::GammaLnPrecise => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::HypGeomDist => args_signature_scalars(arg_count, 5, 0),
|
||||||
|
Function::LogNormDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::LogNormInv => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::NegbinomDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::NormDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::NormInv => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::NormSdist => args_signature_scalars(arg_count, 2, 0),
|
||||||
|
Function::NormSInv => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::Pearson => {
|
||||||
|
if arg_count == 2 {
|
||||||
|
vec![Signature::Vector, Signature::Vector]
|
||||||
|
} else {
|
||||||
|
vec![Signature::Error; arg_count]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Function::Phi => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::PoissonDist => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::Standardize => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::StDevP => vec![Signature::Vector; arg_count],
|
||||||
|
Function::StDevS => vec![Signature::Vector; arg_count],
|
||||||
|
Function::Stdeva => vec![Signature::Vector; arg_count],
|
||||||
|
Function::Stdevpa => vec![Signature::Vector; arg_count],
|
||||||
|
Function::TDist => args_signature_scalars(arg_count, 3, 0),
|
||||||
|
Function::TDist2T => args_signature_scalars(arg_count, 2, 0),
|
||||||
|
Function::TDistRT => args_signature_scalars(arg_count, 2, 0),
|
||||||
|
Function::TInv => args_signature_scalars(arg_count, 2, 0),
|
||||||
|
Function::TInv2T => args_signature_scalars(arg_count, 2, 0),
|
||||||
|
Function::TTest => {
|
||||||
|
if arg_count == 4 {
|
||||||
|
vec![
|
||||||
|
Signature::Vector,
|
||||||
|
Signature::Vector,
|
||||||
|
Signature::Scalar,
|
||||||
|
Signature::Scalar,
|
||||||
|
]
|
||||||
|
} else {
|
||||||
|
vec![Signature::Error; arg_count]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Function::VarP => vec![Signature::Vector; arg_count],
|
||||||
|
Function::VarS => vec![Signature::Vector; arg_count],
|
||||||
|
Function::VarpA => vec![Signature::Vector; arg_count],
|
||||||
|
Function::VarA => vec![Signature::Vector; arg_count],
|
||||||
|
Function::WeibullDist => args_signature_scalars(arg_count, 4, 0),
|
||||||
|
Function::ZTest => {
|
||||||
|
if arg_count == 2 {
|
||||||
|
vec![Signature::Vector, Signature::Scalar]
|
||||||
|
} else if arg_count == 3 {
|
||||||
|
vec![Signature::Vector, Signature::Scalar, Signature::Scalar]
|
||||||
|
} else {
|
||||||
|
vec![Signature::Error; arg_count]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Function::Sumx2my2 => vec![Signature::Vector; 2],
|
||||||
|
Function::Sumx2py2 => vec![Signature::Vector; 2],
|
||||||
|
Function::Sumxmy2 => vec![Signature::Vector; 2],
|
||||||
|
Function::Correl => vec![Signature::Vector; 2],
|
||||||
|
Function::Rsq => vec![Signature::Vector; 2],
|
||||||
|
Function::Intercept => vec![Signature::Vector; 2],
|
||||||
|
Function::Slope => vec![Signature::Vector; 2],
|
||||||
|
Function::Steyx => vec![Signature::Vector; 2],
|
||||||
|
Function::Gauss => args_signature_scalars(arg_count, 1, 0),
|
||||||
|
Function::Harmean => vec![Signature::Vector; arg_count],
|
||||||
|
Function::Kurt => vec![Signature::Vector; arg_count],
|
||||||
|
Function::Large => vec![Signature::Vector, Signature::Scalar],
|
||||||
|
Function::MaxA => vec![Signature::Vector; arg_count],
|
||||||
|
Function::Median => vec![Signature::Vector; arg_count],
|
||||||
|
Function::MinA => vec![Signature::Vector; arg_count],
|
||||||
|
Function::RankAvg => vec![Signature::Scalar, Signature::Vector, Signature::Scalar],
|
||||||
|
Function::RankEq => vec![Signature::Scalar, Signature::Vector, Signature::Scalar],
|
||||||
|
Function::Skew => vec![Signature::Vector; arg_count],
|
||||||
|
Function::SkewP => vec![Signature::Vector; arg_count],
|
||||||
|
Function::Small => vec![Signature::Vector, Signature::Scalar],
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -901,7 +1032,7 @@ fn static_analysis_on_function(kind: &Function, args: &[Node]) -> StaticResult {
|
|||||||
Function::Atan => scalar_arguments(args),
|
Function::Atan => scalar_arguments(args),
|
||||||
Function::Atan2 => scalar_arguments(args),
|
Function::Atan2 => scalar_arguments(args),
|
||||||
Function::Atanh => scalar_arguments(args),
|
Function::Atanh => scalar_arguments(args),
|
||||||
Function::Choose => scalar_arguments(args), // static_analysis_choose(args, cell),
|
Function::Choose => scalar_arguments(args),
|
||||||
Function::Column => not_implemented(args),
|
Function::Column => not_implemented(args),
|
||||||
Function::Columns => not_implemented(args),
|
Function::Columns => not_implemented(args),
|
||||||
Function::Cos => scalar_arguments(args),
|
Function::Cos => scalar_arguments(args),
|
||||||
@@ -948,7 +1079,6 @@ fn static_analysis_on_function(kind: &Function, args: &[Node]) -> StaticResult {
|
|||||||
Function::Lookup => not_implemented(args),
|
Function::Lookup => not_implemented(args),
|
||||||
Function::Match => not_implemented(args),
|
Function::Match => not_implemented(args),
|
||||||
Function::Offset => static_analysis_offset(args),
|
Function::Offset => static_analysis_offset(args),
|
||||||
// FIXME: Row could return an array
|
|
||||||
Function::Row => StaticResult::Scalar,
|
Function::Row => StaticResult::Scalar,
|
||||||
Function::Rows => not_implemented(args),
|
Function::Rows => not_implemented(args),
|
||||||
Function::Vlookup => not_implemented(args),
|
Function::Vlookup => not_implemented(args),
|
||||||
@@ -977,6 +1107,7 @@ fn static_analysis_on_function(kind: &Function, args: &[Node]) -> StaticResult {
|
|||||||
Function::Valuetotext => not_implemented(args),
|
Function::Valuetotext => not_implemented(args),
|
||||||
Function::Average => not_implemented(args),
|
Function::Average => not_implemented(args),
|
||||||
Function::Averagea => not_implemented(args),
|
Function::Averagea => not_implemented(args),
|
||||||
|
Function::Avedev => not_implemented(args),
|
||||||
Function::Averageif => not_implemented(args),
|
Function::Averageif => not_implemented(args),
|
||||||
Function::Averageifs => not_implemented(args),
|
Function::Averageifs => not_implemented(args),
|
||||||
Function::Count => not_implemented(args),
|
Function::Count => not_implemented(args),
|
||||||
@@ -1139,5 +1270,93 @@ fn static_analysis_on_function(kind: &Function, args: &[Node]) -> StaticResult {
|
|||||||
Function::Sheets => scalar_arguments(args),
|
Function::Sheets => scalar_arguments(args),
|
||||||
Function::Cell => scalar_arguments(args),
|
Function::Cell => scalar_arguments(args),
|
||||||
Function::Info => scalar_arguments(args),
|
Function::Info => scalar_arguments(args),
|
||||||
|
Function::Dget => not_implemented(args),
|
||||||
|
Function::Dmax => not_implemented(args),
|
||||||
|
Function::Dmin => not_implemented(args),
|
||||||
|
Function::Dcount => not_implemented(args),
|
||||||
|
Function::Daverage => not_implemented(args),
|
||||||
|
Function::Dsum => not_implemented(args),
|
||||||
|
Function::Dcounta => not_implemented(args),
|
||||||
|
Function::Dproduct => not_implemented(args),
|
||||||
|
Function::Dstdev => not_implemented(args),
|
||||||
|
Function::Dvar => not_implemented(args),
|
||||||
|
Function::Dvarp => not_implemented(args),
|
||||||
|
Function::Dstdevp => not_implemented(args),
|
||||||
|
Function::BetaDist => StaticResult::Scalar,
|
||||||
|
Function::BetaInv => StaticResult::Scalar,
|
||||||
|
Function::BinomDist => StaticResult::Scalar,
|
||||||
|
Function::BinomDistRange => StaticResult::Scalar,
|
||||||
|
Function::BinomInv => StaticResult::Scalar,
|
||||||
|
Function::ChisqDist => StaticResult::Scalar,
|
||||||
|
Function::ChisqDistRT => StaticResult::Scalar,
|
||||||
|
Function::ChisqInv => StaticResult::Scalar,
|
||||||
|
Function::ChisqInvRT => StaticResult::Scalar,
|
||||||
|
Function::ChisqTest => StaticResult::Scalar,
|
||||||
|
Function::ConfidenceNorm => StaticResult::Scalar,
|
||||||
|
Function::ConfidenceT => StaticResult::Scalar,
|
||||||
|
Function::CovarianceP => StaticResult::Scalar,
|
||||||
|
Function::CovarianceS => StaticResult::Scalar,
|
||||||
|
Function::Devsq => StaticResult::Scalar,
|
||||||
|
Function::ExponDist => StaticResult::Scalar,
|
||||||
|
Function::FDist => StaticResult::Scalar,
|
||||||
|
Function::FDistRT => StaticResult::Scalar,
|
||||||
|
Function::FInv => StaticResult::Scalar,
|
||||||
|
Function::FInvRT => StaticResult::Scalar,
|
||||||
|
Function::FTest => StaticResult::Scalar,
|
||||||
|
Function::Fisher => StaticResult::Scalar,
|
||||||
|
Function::FisherInv => StaticResult::Scalar,
|
||||||
|
Function::Gamma => StaticResult::Scalar,
|
||||||
|
Function::GammaDist => StaticResult::Scalar,
|
||||||
|
Function::GammaInv => StaticResult::Scalar,
|
||||||
|
Function::GammaLn => StaticResult::Scalar,
|
||||||
|
Function::GammaLnPrecise => StaticResult::Scalar,
|
||||||
|
Function::HypGeomDist => StaticResult::Scalar,
|
||||||
|
Function::LogNormDist => StaticResult::Scalar,
|
||||||
|
Function::LogNormInv => StaticResult::Scalar,
|
||||||
|
Function::NegbinomDist => StaticResult::Scalar,
|
||||||
|
Function::NormDist => StaticResult::Scalar,
|
||||||
|
Function::NormInv => StaticResult::Scalar,
|
||||||
|
Function::NormSdist => StaticResult::Scalar,
|
||||||
|
Function::NormSInv => StaticResult::Scalar,
|
||||||
|
Function::Pearson => StaticResult::Scalar,
|
||||||
|
Function::Phi => StaticResult::Scalar,
|
||||||
|
Function::PoissonDist => StaticResult::Scalar,
|
||||||
|
Function::Standardize => StaticResult::Scalar,
|
||||||
|
Function::StDevP => StaticResult::Scalar,
|
||||||
|
Function::StDevS => StaticResult::Scalar,
|
||||||
|
Function::Stdeva => StaticResult::Scalar,
|
||||||
|
Function::Stdevpa => StaticResult::Scalar,
|
||||||
|
Function::TDist => StaticResult::Scalar,
|
||||||
|
Function::TDist2T => StaticResult::Scalar,
|
||||||
|
Function::TDistRT => StaticResult::Scalar,
|
||||||
|
Function::TInv => StaticResult::Scalar,
|
||||||
|
Function::TInv2T => StaticResult::Scalar,
|
||||||
|
Function::TTest => StaticResult::Scalar,
|
||||||
|
Function::VarP => StaticResult::Scalar,
|
||||||
|
Function::VarS => StaticResult::Scalar,
|
||||||
|
Function::VarpA => StaticResult::Scalar,
|
||||||
|
Function::VarA => StaticResult::Scalar,
|
||||||
|
Function::WeibullDist => StaticResult::Scalar,
|
||||||
|
Function::ZTest => StaticResult::Scalar,
|
||||||
|
Function::Sumx2my2 => StaticResult::Scalar,
|
||||||
|
Function::Sumx2py2 => StaticResult::Scalar,
|
||||||
|
Function::Sumxmy2 => StaticResult::Scalar,
|
||||||
|
Function::Correl => StaticResult::Scalar,
|
||||||
|
Function::Rsq => StaticResult::Scalar,
|
||||||
|
Function::Intercept => StaticResult::Scalar,
|
||||||
|
Function::Slope => StaticResult::Scalar,
|
||||||
|
Function::Steyx => StaticResult::Scalar,
|
||||||
|
Function::Gauss => StaticResult::Scalar,
|
||||||
|
Function::Harmean => StaticResult::Scalar,
|
||||||
|
Function::Kurt => StaticResult::Scalar,
|
||||||
|
Function::Large => StaticResult::Scalar,
|
||||||
|
Function::MaxA => StaticResult::Scalar,
|
||||||
|
Function::Median => StaticResult::Scalar,
|
||||||
|
Function::MinA => StaticResult::Scalar,
|
||||||
|
Function::RankAvg => StaticResult::Scalar,
|
||||||
|
Function::RankEq => StaticResult::Scalar,
|
||||||
|
Function::Skew => StaticResult::Scalar,
|
||||||
|
Function::SkewP => StaticResult::Scalar,
|
||||||
|
Function::Small => StaticResult::Scalar,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,6 +3,8 @@ use crate::constants::{LAST_COLUMN, LAST_ROW};
|
|||||||
use crate::expressions::parser::move_formula::to_string_array_node;
|
use crate::expressions::parser::move_formula::to_string_array_node;
|
||||||
use crate::expressions::parser::static_analysis::add_implicit_intersection;
|
use crate::expressions::parser::static_analysis::add_implicit_intersection;
|
||||||
use crate::expressions::token::{OpSum, OpUnary};
|
use crate::expressions::token::{OpSum, OpUnary};
|
||||||
|
use crate::language::{get_language, Language};
|
||||||
|
use crate::locale::{get_locale, Locale};
|
||||||
use crate::{expressions::types::CellReferenceRC, number_format::to_excel_precision_str};
|
use crate::{expressions::types::CellReferenceRC, number_format::to_excel_precision_str};
|
||||||
|
|
||||||
pub enum DisplaceData {
|
pub enum DisplaceData {
|
||||||
@@ -43,17 +45,44 @@ pub enum DisplaceData {
|
|||||||
|
|
||||||
/// This is the internal mode in IronCalc
|
/// This is the internal mode in IronCalc
|
||||||
pub fn to_rc_format(node: &Node) -> String {
|
pub fn to_rc_format(node: &Node) -> String {
|
||||||
stringify(node, None, &DisplaceData::None, false)
|
#[allow(clippy::expect_used)]
|
||||||
|
let locale = get_locale("en").expect("");
|
||||||
|
#[allow(clippy::expect_used)]
|
||||||
|
let language = get_language("en").expect("");
|
||||||
|
stringify(node, None, &DisplaceData::None, false, locale, language)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// This is the mode used to display the formula in the UI
|
/// This is the mode used to display the formula in the UI
|
||||||
pub fn to_string(node: &Node, context: &CellReferenceRC) -> String {
|
pub fn to_localized_string(
|
||||||
stringify(node, Some(context), &DisplaceData::None, false)
|
node: &Node,
|
||||||
|
context: &CellReferenceRC,
|
||||||
|
locale: &Locale,
|
||||||
|
language: &Language,
|
||||||
|
) -> String {
|
||||||
|
stringify(
|
||||||
|
node,
|
||||||
|
Some(context),
|
||||||
|
&DisplaceData::None,
|
||||||
|
false,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// This is the mode used to export the formula to Excel
|
/// This is the mode used to export the formula to Excel
|
||||||
pub fn to_excel_string(node: &Node, context: &CellReferenceRC) -> String {
|
pub fn to_excel_string(node: &Node, context: &CellReferenceRC) -> String {
|
||||||
stringify(node, Some(context), &DisplaceData::None, true)
|
#[allow(clippy::expect_used)]
|
||||||
|
let locale = get_locale("en").expect("");
|
||||||
|
#[allow(clippy::expect_used)]
|
||||||
|
let language = get_language("en").expect("");
|
||||||
|
stringify(
|
||||||
|
node,
|
||||||
|
Some(context),
|
||||||
|
&DisplaceData::None,
|
||||||
|
true,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn to_string_displaced(
|
pub fn to_string_displaced(
|
||||||
@@ -61,7 +90,11 @@ pub fn to_string_displaced(
|
|||||||
context: &CellReferenceRC,
|
context: &CellReferenceRC,
|
||||||
displace_data: &DisplaceData,
|
displace_data: &DisplaceData,
|
||||||
) -> String {
|
) -> String {
|
||||||
stringify(node, Some(context), displace_data, false)
|
#[allow(clippy::expect_used)]
|
||||||
|
let locale = get_locale("en").expect("");
|
||||||
|
#[allow(clippy::expect_used)]
|
||||||
|
let language = get_language("en").expect("");
|
||||||
|
stringify(node, Some(context), displace_data, false, locale, language)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Converts a local reference to a string applying some displacement if needed.
|
/// Converts a local reference to a string applying some displacement if needed.
|
||||||
@@ -273,19 +306,41 @@ fn format_function(
|
|||||||
context: Option<&CellReferenceRC>,
|
context: Option<&CellReferenceRC>,
|
||||||
displace_data: &DisplaceData,
|
displace_data: &DisplaceData,
|
||||||
export_to_excel: bool,
|
export_to_excel: bool,
|
||||||
|
locale: &Locale,
|
||||||
|
language: &Language,
|
||||||
) -> String {
|
) -> String {
|
||||||
let mut first = true;
|
let mut first = true;
|
||||||
let mut arguments = "".to_string();
|
let mut arguments = "".to_string();
|
||||||
|
let arg_separator = if locale.numbers.symbols.decimal == "." {
|
||||||
|
','
|
||||||
|
} else {
|
||||||
|
';'
|
||||||
|
};
|
||||||
for el in args {
|
for el in args {
|
||||||
if !first {
|
if !first {
|
||||||
arguments = format!(
|
arguments = format!(
|
||||||
"{},{}",
|
"{}{}{}",
|
||||||
arguments,
|
arguments,
|
||||||
stringify(el, context, displace_data, export_to_excel)
|
arg_separator,
|
||||||
|
stringify(
|
||||||
|
el,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
);
|
);
|
||||||
} else {
|
} else {
|
||||||
first = false;
|
first = false;
|
||||||
arguments = stringify(el, context, displace_data, export_to_excel);
|
arguments = stringify(
|
||||||
|
el,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
format!("{name}({arguments})")
|
format!("{name}({arguments})")
|
||||||
@@ -321,11 +376,26 @@ fn stringify(
|
|||||||
context: Option<&CellReferenceRC>,
|
context: Option<&CellReferenceRC>,
|
||||||
displace_data: &DisplaceData,
|
displace_data: &DisplaceData,
|
||||||
export_to_excel: bool,
|
export_to_excel: bool,
|
||||||
|
locale: &Locale,
|
||||||
|
language: &Language,
|
||||||
) -> String {
|
) -> String {
|
||||||
use self::Node::*;
|
use self::Node::*;
|
||||||
match node {
|
match node {
|
||||||
BooleanKind(value) => format!("{value}").to_ascii_uppercase(),
|
BooleanKind(value) => {
|
||||||
NumberKind(number) => to_excel_precision_str(*number),
|
if *value {
|
||||||
|
language.booleans.r#true.to_string()
|
||||||
|
} else {
|
||||||
|
language.booleans.r#false.to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
NumberKind(number) => {
|
||||||
|
let s = to_excel_precision_str(*number);
|
||||||
|
if locale.numbers.symbols.decimal == "." {
|
||||||
|
s
|
||||||
|
} else {
|
||||||
|
s.replace(".", &locale.numbers.symbols.decimal)
|
||||||
|
}
|
||||||
|
}
|
||||||
StringKind(value) => format!("\"{value}\""),
|
StringKind(value) => format!("\"{value}\""),
|
||||||
WrongReferenceKind {
|
WrongReferenceKind {
|
||||||
sheet_name,
|
sheet_name,
|
||||||
@@ -469,32 +539,95 @@ fn stringify(
|
|||||||
}
|
}
|
||||||
OpRangeKind { left, right } => format!(
|
OpRangeKind { left, right } => format!(
|
||||||
"{}:{}",
|
"{}:{}",
|
||||||
stringify(left, context, displace_data, export_to_excel),
|
stringify(
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
),
|
||||||
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
),
|
),
|
||||||
OpConcatenateKind { left, right } => format!(
|
OpConcatenateKind { left, right } => format!(
|
||||||
"{}&{}",
|
"{}&{}",
|
||||||
stringify(left, context, displace_data, export_to_excel),
|
stringify(
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
),
|
||||||
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
),
|
),
|
||||||
CompareKind { kind, left, right } => format!(
|
CompareKind { kind, left, right } => format!(
|
||||||
"{}{}{}",
|
"{}{}{}",
|
||||||
stringify(left, context, displace_data, export_to_excel),
|
stringify(
|
||||||
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
),
|
||||||
kind,
|
kind,
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
),
|
),
|
||||||
OpSumKind { kind, left, right } => {
|
OpSumKind { kind, left, right } => {
|
||||||
let left_str = stringify(left, context, displace_data, export_to_excel);
|
let left_str = stringify(
|
||||||
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
);
|
||||||
// if kind is minus then we need parentheses in the right side if they are OpSumKind or CompareKind
|
// if kind is minus then we need parentheses in the right side if they are OpSumKind or CompareKind
|
||||||
let right_str = if (matches!(kind, OpSum::Minus) && matches!(**right, OpSumKind { .. }))
|
let right_str = if (matches!(kind, OpSum::Minus) && matches!(**right, OpSumKind { .. }))
|
||||||
| matches!(**right, CompareKind { .. })
|
| matches!(**right, CompareKind { .. })
|
||||||
{
|
{
|
||||||
format!(
|
format!(
|
||||||
"({})",
|
"({})",
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
)
|
)
|
||||||
} else {
|
} else {
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
)
|
||||||
};
|
};
|
||||||
|
|
||||||
format!("{left_str}{kind}{right_str}")
|
format!("{left_str}{kind}{right_str}")
|
||||||
@@ -503,16 +636,44 @@ fn stringify(
|
|||||||
let x = match **left {
|
let x = match **left {
|
||||||
OpSumKind { .. } | CompareKind { .. } => format!(
|
OpSumKind { .. } | CompareKind { .. } => format!(
|
||||||
"({})",
|
"({})",
|
||||||
stringify(left, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
|
),
|
||||||
|
_ => stringify(
|
||||||
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
),
|
),
|
||||||
_ => stringify(left, context, displace_data, export_to_excel),
|
|
||||||
};
|
};
|
||||||
let y = match **right {
|
let y = match **right {
|
||||||
OpSumKind { .. } | CompareKind { .. } | OpProductKind { .. } => format!(
|
OpSumKind { .. } | CompareKind { .. } | OpProductKind { .. } => format!(
|
||||||
"({})",
|
"({})",
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
|
),
|
||||||
|
_ => stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
),
|
),
|
||||||
_ => stringify(right, context, displace_data, export_to_excel),
|
|
||||||
};
|
};
|
||||||
format!("{x}{kind}{y}")
|
format!("{x}{kind}{y}")
|
||||||
}
|
}
|
||||||
@@ -528,7 +689,14 @@ fn stringify(
|
|||||||
| DefinedNameKind(_)
|
| DefinedNameKind(_)
|
||||||
| TableNameKind(_)
|
| TableNameKind(_)
|
||||||
| WrongVariableKind(_)
|
| WrongVariableKind(_)
|
||||||
| WrongRangeKind { .. } => stringify(left, context, displace_data, export_to_excel),
|
| WrongRangeKind { .. } => stringify(
|
||||||
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
),
|
||||||
OpRangeKind { .. }
|
OpRangeKind { .. }
|
||||||
| OpConcatenateKind { .. }
|
| OpConcatenateKind { .. }
|
||||||
| OpProductKind { .. }
|
| OpProductKind { .. }
|
||||||
@@ -543,7 +711,14 @@ fn stringify(
|
|||||||
| ImplicitIntersection { .. }
|
| ImplicitIntersection { .. }
|
||||||
| EmptyArgKind => format!(
|
| EmptyArgKind => format!(
|
||||||
"({})",
|
"({})",
|
||||||
stringify(left, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
left,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
),
|
),
|
||||||
};
|
};
|
||||||
let y = match **right {
|
let y = match **right {
|
||||||
@@ -556,9 +731,14 @@ fn stringify(
|
|||||||
| DefinedNameKind(_)
|
| DefinedNameKind(_)
|
||||||
| TableNameKind(_)
|
| TableNameKind(_)
|
||||||
| WrongVariableKind(_)
|
| WrongVariableKind(_)
|
||||||
| WrongRangeKind { .. } => {
|
| WrongRangeKind { .. } => stringify(
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
right,
|
||||||
}
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
),
|
||||||
OpRangeKind { .. }
|
OpRangeKind { .. }
|
||||||
| OpConcatenateKind { .. }
|
| OpConcatenateKind { .. }
|
||||||
| OpProductKind { .. }
|
| OpProductKind { .. }
|
||||||
@@ -574,29 +754,56 @@ fn stringify(
|
|||||||
| ImplicitIntersection { .. }
|
| ImplicitIntersection { .. }
|
||||||
| EmptyArgKind => format!(
|
| EmptyArgKind => format!(
|
||||||
"({})",
|
"({})",
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
),
|
),
|
||||||
};
|
};
|
||||||
format!("{x}^{y}")
|
format!("{x}^{y}")
|
||||||
}
|
}
|
||||||
InvalidFunctionKind { name, args } => {
|
InvalidFunctionKind { name, args } => format_function(
|
||||||
format_function(name, args, context, displace_data, export_to_excel)
|
&name.to_ascii_lowercase(),
|
||||||
}
|
args,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
),
|
||||||
FunctionKind { kind, args } => {
|
FunctionKind { kind, args } => {
|
||||||
let name = if export_to_excel {
|
let name = if export_to_excel {
|
||||||
kind.to_xlsx_string()
|
kind.to_xlsx_string()
|
||||||
} else {
|
} else {
|
||||||
kind.to_string()
|
kind.to_localized_name(language)
|
||||||
};
|
};
|
||||||
format_function(&name, args, context, displace_data, export_to_excel)
|
format_function(
|
||||||
|
&name,
|
||||||
|
args,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
)
|
||||||
}
|
}
|
||||||
ArrayKind(args) => {
|
ArrayKind(args) => {
|
||||||
let mut first_row = true;
|
let mut first_row = true;
|
||||||
let mut matrix_string = String::new();
|
let mut matrix_string = String::new();
|
||||||
|
let row_separator = if locale.numbers.symbols.decimal == "." {
|
||||||
|
';'
|
||||||
|
} else {
|
||||||
|
'/'
|
||||||
|
};
|
||||||
|
let col_separator = if row_separator == ';' { ',' } else { ';' };
|
||||||
|
|
||||||
for row in args {
|
for row in args {
|
||||||
if !first_row {
|
if !first_row {
|
||||||
matrix_string.push(';');
|
matrix_string.push(row_separator);
|
||||||
} else {
|
} else {
|
||||||
first_row = false;
|
first_row = false;
|
||||||
}
|
}
|
||||||
@@ -604,11 +811,11 @@ fn stringify(
|
|||||||
let mut row_string = String::new();
|
let mut row_string = String::new();
|
||||||
for el in row {
|
for el in row {
|
||||||
if !first_column {
|
if !first_column {
|
||||||
row_string.push(',');
|
row_string.push(col_separator);
|
||||||
} else {
|
} else {
|
||||||
first_column = false;
|
first_column = false;
|
||||||
}
|
}
|
||||||
row_string.push_str(&to_string_array_node(el));
|
row_string.push_str(&to_string_array_node(el, locale, language));
|
||||||
}
|
}
|
||||||
matrix_string.push_str(&row_string);
|
matrix_string.push_str(&row_string);
|
||||||
}
|
}
|
||||||
@@ -647,19 +854,40 @@ fn stringify(
|
|||||||
if needs_parentheses {
|
if needs_parentheses {
|
||||||
format!(
|
format!(
|
||||||
"-({})",
|
"-({})",
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
)
|
)
|
||||||
} else {
|
} else {
|
||||||
format!(
|
format!(
|
||||||
"-{}",
|
"-{}",
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
OpUnary::Percentage => {
|
OpUnary::Percentage => {
|
||||||
format!(
|
format!(
|
||||||
"{}%",
|
"{}%",
|
||||||
stringify(right, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
right,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
@@ -680,17 +908,38 @@ fn stringify(
|
|||||||
|
|
||||||
add_implicit_intersection(&mut new_node, true);
|
add_implicit_intersection(&mut new_node, true);
|
||||||
if matches!(&new_node, Node::ImplicitIntersection { .. }) {
|
if matches!(&new_node, Node::ImplicitIntersection { .. }) {
|
||||||
return stringify(child, context, displace_data, export_to_excel);
|
return stringify(
|
||||||
|
child,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language,
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
return format!(
|
return format!(
|
||||||
"_xlfn.SINGLE({})",
|
"_xlfn.SINGLE({})",
|
||||||
stringify(child, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
child,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
format!(
|
format!(
|
||||||
"@{}",
|
"@{}",
|
||||||
stringify(child, context, displace_data, export_to_excel)
|
stringify(
|
||||||
|
child,
|
||||||
|
context,
|
||||||
|
displace_data,
|
||||||
|
export_to_excel,
|
||||||
|
locale,
|
||||||
|
language
|
||||||
|
)
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,7 +4,10 @@ mod test_general;
|
|||||||
mod test_implicit_intersection;
|
mod test_implicit_intersection;
|
||||||
mod test_issue_155;
|
mod test_issue_155;
|
||||||
mod test_issue_483;
|
mod test_issue_483;
|
||||||
|
mod test_languages;
|
||||||
|
mod test_locales;
|
||||||
mod test_move_formula;
|
mod test_move_formula;
|
||||||
mod test_ranges;
|
mod test_ranges;
|
||||||
mod test_stringify;
|
mod test_stringify;
|
||||||
mod test_tables;
|
mod test_tables;
|
||||||
|
mod utils;
|
||||||
|
|||||||
@@ -2,8 +2,8 @@ use std::collections::HashMap;
|
|||||||
|
|
||||||
use crate::expressions::{
|
use crate::expressions::{
|
||||||
parser::{
|
parser::{
|
||||||
stringify::{to_excel_string, to_string},
|
stringify::to_excel_string,
|
||||||
Parser,
|
tests::utils::{new_parser, to_english_localized_string},
|
||||||
},
|
},
|
||||||
types::CellReferenceRC,
|
types::CellReferenceRC,
|
||||||
};
|
};
|
||||||
@@ -13,7 +13,7 @@ use crate::expressions::parser::static_analysis::add_implicit_intersection;
|
|||||||
#[test]
|
#[test]
|
||||||
fn simple_test() {
|
fn simple_test() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -72,7 +72,7 @@ fn simple_test() {
|
|||||||
for (formula, expected) in cases {
|
for (formula, expected) in cases {
|
||||||
let mut t = parser.parse(formula, &cell_reference);
|
let mut t = parser.parse(formula, &cell_reference);
|
||||||
add_implicit_intersection(&mut t, true);
|
add_implicit_intersection(&mut t, true);
|
||||||
let r = to_string(&t, &cell_reference);
|
let r = to_english_localized_string(&t, &cell_reference);
|
||||||
assert_eq!(r, expected);
|
assert_eq!(r, expected);
|
||||||
let excel_formula = to_excel_string(&t, &cell_reference);
|
let excel_formula = to_excel_string(&t, &cell_reference);
|
||||||
assert_eq!(excel_formula, formula);
|
assert_eq!(excel_formula, formula);
|
||||||
|
|||||||
@@ -2,14 +2,15 @@
|
|||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::expressions::parser::stringify::{to_rc_format, to_string};
|
use crate::expressions::parser::stringify::to_rc_format;
|
||||||
use crate::expressions::parser::{ArrayNode, Node, Parser};
|
use crate::expressions::parser::tests::utils::{new_parser, to_english_localized_string};
|
||||||
|
use crate::expressions::parser::{ArrayNode, Node};
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn simple_horizontal() {
|
fn simple_horizontal() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -28,13 +29,16 @@ fn simple_horizontal() {
|
|||||||
);
|
);
|
||||||
|
|
||||||
assert_eq!(to_rc_format(&horizontal), "{1,2,3}");
|
assert_eq!(to_rc_format(&horizontal), "{1,2,3}");
|
||||||
assert_eq!(to_string(&horizontal, &cell_reference), "{1,2,3}");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&horizontal, &cell_reference),
|
||||||
|
"{1,2,3}"
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn simple_vertical() {
|
fn simple_vertical() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -52,13 +56,16 @@ fn simple_vertical() {
|
|||||||
])
|
])
|
||||||
);
|
);
|
||||||
assert_eq!(to_rc_format(&vertical), "{1;2;3}");
|
assert_eq!(to_rc_format(&vertical), "{1;2;3}");
|
||||||
assert_eq!(to_string(&vertical, &cell_reference), "{1;2;3}");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&vertical, &cell_reference),
|
||||||
|
"{1;2;3}"
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn simple_matrix() {
|
fn simple_matrix() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -88,5 +95,8 @@ fn simple_matrix() {
|
|||||||
])
|
])
|
||||||
);
|
);
|
||||||
assert_eq!(to_rc_format(&matrix), "{1,2,3;4,5,6;7,8,9}");
|
assert_eq!(to_rc_format(&matrix), "{1,2,3;4,5,6;7,8,9}");
|
||||||
assert_eq!(to_string(&matrix, &cell_reference), "{1,2,3;4,5,6;7,8,9}");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&matrix, &cell_reference),
|
||||||
|
"{1,2,3;4,5,6;7,8,9}"
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,12 +3,12 @@
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::expressions::lexer::LexerMode;
|
use crate::expressions::lexer::LexerMode;
|
||||||
use crate::expressions::parser::stringify::{
|
use crate::expressions::parser::stringify::{to_rc_format, to_string_displaced, DisplaceData};
|
||||||
to_rc_format, to_string, to_string_displaced, DisplaceData,
|
use crate::expressions::parser::Node;
|
||||||
};
|
|
||||||
use crate::expressions::parser::{Node, Parser};
|
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
|
|
||||||
|
use crate::expressions::parser::tests::utils::{new_parser, to_english_localized_string};
|
||||||
|
|
||||||
struct Formula<'a> {
|
struct Formula<'a> {
|
||||||
initial: &'a str,
|
initial: &'a str,
|
||||||
expected: &'a str,
|
expected: &'a str,
|
||||||
@@ -17,7 +17,7 @@ struct Formula<'a> {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_reference() {
|
fn test_parser_reference() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -32,7 +32,7 @@ fn test_parser_reference() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_absolute_column() {
|
fn test_parser_absolute_column() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -47,7 +47,7 @@ fn test_parser_absolute_column() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_absolute_row_col() {
|
fn test_parser_absolute_row_col() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -62,7 +62,7 @@ fn test_parser_absolute_row_col() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_absolute_row_col_1() {
|
fn test_parser_absolute_row_col_1() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -77,7 +77,7 @@ fn test_parser_absolute_row_col_1() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_simple_formula() {
|
fn test_parser_simple_formula() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -93,7 +93,7 @@ fn test_parser_simple_formula() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_boolean() {
|
fn test_parser_boolean() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -109,7 +109,7 @@ fn test_parser_boolean() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_bad_formula() {
|
fn test_parser_bad_formula() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -138,7 +138,7 @@ fn test_parser_bad_formula() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_bad_formula_1() {
|
fn test_parser_bad_formula_1() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -167,7 +167,7 @@ fn test_parser_bad_formula_1() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_bad_formula_2() {
|
fn test_parser_bad_formula_2() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -196,7 +196,7 @@ fn test_parser_bad_formula_2() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_bad_formula_3() {
|
fn test_parser_bad_formula_3() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -225,7 +225,7 @@ fn test_parser_bad_formula_3() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_formulas() {
|
fn test_parser_formulas() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let formulas = vec![
|
let formulas = vec![
|
||||||
Formula {
|
Formula {
|
||||||
@@ -266,14 +266,17 @@ fn test_parser_formulas() {
|
|||||||
},
|
},
|
||||||
);
|
);
|
||||||
assert_eq!(to_rc_format(&t), formula.expected);
|
assert_eq!(to_rc_format(&t), formula.expected);
|
||||||
assert_eq!(to_string(&t, &cell_reference), formula.initial);
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
formula.initial
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_parser_r1c1_formulas() {
|
fn test_parser_r1c1_formulas() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
parser.set_lexer_mode(LexerMode::R1C1);
|
parser.set_lexer_mode(LexerMode::R1C1);
|
||||||
|
|
||||||
let formulas = vec![
|
let formulas = vec![
|
||||||
@@ -330,7 +333,10 @@ fn test_parser_r1c1_formulas() {
|
|||||||
column: 1,
|
column: 1,
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
assert_eq!(to_string(&t, &cell_reference), formula.expected);
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
formula.expected
|
||||||
|
);
|
||||||
assert_eq!(to_rc_format(&t), formula.initial);
|
assert_eq!(to_rc_format(&t), formula.initial);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -338,7 +344,7 @@ fn test_parser_r1c1_formulas() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_quotes() {
|
fn test_parser_quotes() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Second Sheet".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Second Sheet".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -354,7 +360,7 @@ fn test_parser_quotes() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_escape_quotes() {
|
fn test_parser_escape_quotes() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Second '2' Sheet".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Second '2' Sheet".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -370,7 +376,7 @@ fn test_parser_escape_quotes() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_parenthesis() {
|
fn test_parser_parenthesis() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Second2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Second2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -386,7 +392,7 @@ fn test_parser_parenthesis() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_excel_xlfn() {
|
fn test_parser_excel_xlfn() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Second2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Second2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -407,7 +413,7 @@ fn test_to_string_displaced() {
|
|||||||
column: 1,
|
column: 1,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let node = parser.parse("C3", context);
|
let node = parser.parse("C3", context);
|
||||||
let displace_data = DisplaceData::Column {
|
let displace_data = DisplaceData::Column {
|
||||||
@@ -427,7 +433,7 @@ fn test_to_string_displaced_full_ranges() {
|
|||||||
column: 1,
|
column: 1,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let node = parser.parse("SUM(3:3)", context);
|
let node = parser.parse("SUM(3:3)", context);
|
||||||
let displace_data = DisplaceData::Column {
|
let displace_data = DisplaceData::Column {
|
||||||
@@ -460,7 +466,7 @@ fn test_to_string_displaced_too_low() {
|
|||||||
column: 1,
|
column: 1,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let node = parser.parse("C3", context);
|
let node = parser.parse("C3", context);
|
||||||
let displace_data = DisplaceData::Column {
|
let displace_data = DisplaceData::Column {
|
||||||
@@ -480,7 +486,7 @@ fn test_to_string_displaced_too_high() {
|
|||||||
column: 1,
|
column: 1,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let node = parser.parse("C3", context);
|
let node = parser.parse("C3", context);
|
||||||
let displace_data = DisplaceData::Column {
|
let displace_data = DisplaceData::Column {
|
||||||
|
|||||||
@@ -1,13 +1,15 @@
|
|||||||
#![allow(clippy::panic)]
|
#![allow(clippy::panic)]
|
||||||
|
|
||||||
use crate::expressions::parser::{Node, Parser};
|
use crate::expressions::parser::Node;
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
use crate::expressions::parser::tests::utils::new_parser;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn simple() {
|
fn simple() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!B3
|
// Reference cell is Sheet1!B3
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -40,7 +42,7 @@ fn simple() {
|
|||||||
#[test]
|
#[test]
|
||||||
fn simple_add() {
|
fn simple_add() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!B3
|
// Reference cell is Sheet1!B3
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
|
|||||||
@@ -2,14 +2,13 @@
|
|||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::expressions::parser::stringify::to_string;
|
use crate::expressions::parser::tests::utils::{new_parser, to_english_localized_string};
|
||||||
use crate::expressions::parser::Parser;
|
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn issue_155_parser() {
|
fn issue_155_parser() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -18,13 +17,13 @@ fn issue_155_parser() {
|
|||||||
column: 2,
|
column: 2,
|
||||||
};
|
};
|
||||||
let t = parser.parse("A$1:A2", &cell_reference);
|
let t = parser.parse("A$1:A2", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "A$1:A2");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "A$1:A2");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn issue_155_parser_case_2() {
|
fn issue_155_parser_case_2() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -33,13 +32,13 @@ fn issue_155_parser_case_2() {
|
|||||||
column: 20,
|
column: 20,
|
||||||
};
|
};
|
||||||
let t = parser.parse("C$1:D2", &cell_reference);
|
let t = parser.parse("C$1:D2", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "C$1:D2");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "C$1:D2");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn issue_155_parser_only_row() {
|
fn issue_155_parser_only_row() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -49,13 +48,13 @@ fn issue_155_parser_only_row() {
|
|||||||
};
|
};
|
||||||
// This is tricky, I am not sure what to do in these cases
|
// This is tricky, I am not sure what to do in these cases
|
||||||
let t = parser.parse("A$2:B1", &cell_reference);
|
let t = parser.parse("A$2:B1", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "A1:B$2");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "A1:B$2");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn issue_155_parser_only_column() {
|
fn issue_155_parser_only_column() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -65,5 +64,5 @@ fn issue_155_parser_only_column() {
|
|||||||
};
|
};
|
||||||
// This is tricky, I am not sure what to do in these cases
|
// This is tricky, I am not sure what to do in these cases
|
||||||
let t = parser.parse("D1:$A3", &cell_reference);
|
let t = parser.parse("D1:$A3", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "$A1:D3");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "$A1:D3");
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,14 +2,14 @@
|
|||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::expressions::parser::stringify::to_string;
|
use crate::expressions::parser::tests::utils::{new_parser, to_english_localized_string};
|
||||||
use crate::expressions::parser::{Node, Parser};
|
use crate::expressions::parser::Node;
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn issue_483_parser() {
|
fn issue_483_parser() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -19,9 +19,12 @@ fn issue_483_parser() {
|
|||||||
};
|
};
|
||||||
let t = parser.parse("-(A1^1.22)", &cell_reference);
|
let t = parser.parse("-(A1^1.22)", &cell_reference);
|
||||||
assert!(matches!(t, Node::UnaryKind { .. }));
|
assert!(matches!(t, Node::UnaryKind { .. }));
|
||||||
assert_eq!(to_string(&t, &cell_reference), "-(A1^1.22)");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
"-(A1^1.22)"
|
||||||
|
);
|
||||||
|
|
||||||
let t = parser.parse("-A1^1.22", &cell_reference);
|
let t = parser.parse("-A1^1.22", &cell_reference);
|
||||||
assert!(matches!(t, Node::OpPowerKind { .. }));
|
assert!(matches!(t, Node::OpPowerKind { .. }));
|
||||||
assert_eq!(to_string(&t, &cell_reference), "-A1^1.22");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "-A1^1.22");
|
||||||
}
|
}
|
||||||
|
|||||||
58
base/src/expressions/parser/tests/test_languages.rs
Normal file
58
base/src/expressions/parser/tests/test_languages.rs
Normal file
@@ -0,0 +1,58 @@
|
|||||||
|
#![allow(clippy::unwrap_used)]
|
||||||
|
|
||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
use crate::expressions::parser::{DefinedNameS, Node, Parser};
|
||||||
|
use crate::expressions::types::CellReferenceRC;
|
||||||
|
|
||||||
|
use crate::expressions::parser::stringify::to_localized_string;
|
||||||
|
use crate::functions::Function;
|
||||||
|
use crate::language::get_language;
|
||||||
|
use crate::locale::get_locale;
|
||||||
|
use crate::types::Table;
|
||||||
|
|
||||||
|
pub fn to_string(t: &Node, cell_reference: &CellReferenceRC) -> String {
|
||||||
|
let locale = get_locale("en").unwrap();
|
||||||
|
let language = get_language("es").unwrap();
|
||||||
|
to_localized_string(t, cell_reference, locale, language)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn new_parser<'a>(
|
||||||
|
worksheets: Vec<String>,
|
||||||
|
defined_names: Vec<DefinedNameS>,
|
||||||
|
tables: HashMap<String, Table>,
|
||||||
|
) -> Parser<'a> {
|
||||||
|
let locale = get_locale("en").unwrap();
|
||||||
|
let language = get_language("es").unwrap();
|
||||||
|
Parser::new(worksheets, defined_names, tables, locale, language)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn simple_language() {
|
||||||
|
let worksheets = vec!["Sheet1".to_string(), "Second Sheet".to_string()];
|
||||||
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
// Reference cell is Sheet1!A1
|
||||||
|
let cell_reference = CellReferenceRC {
|
||||||
|
sheet: "Sheet1".to_string(),
|
||||||
|
row: 1,
|
||||||
|
column: 1,
|
||||||
|
};
|
||||||
|
let t = parser.parse("FALSO", &cell_reference);
|
||||||
|
assert!(matches!(t, Node::BooleanKind(false)));
|
||||||
|
|
||||||
|
let t = parser.parse("VERDADERO", &cell_reference);
|
||||||
|
assert!(matches!(t, Node::BooleanKind(true)));
|
||||||
|
|
||||||
|
let t = parser.parse("TRUE()", &cell_reference);
|
||||||
|
assert!(matches!(t, Node::InvalidFunctionKind { ref name, args: _} if name == "TRUE"));
|
||||||
|
|
||||||
|
let t = parser.parse("VERDADERO()", &cell_reference);
|
||||||
|
assert!(matches!(
|
||||||
|
t,
|
||||||
|
Node::FunctionKind {
|
||||||
|
kind: Function::True,
|
||||||
|
args: _
|
||||||
|
}
|
||||||
|
));
|
||||||
|
assert_eq!(to_string(&t, &cell_reference), "VERDADERO()".to_string());
|
||||||
|
}
|
||||||
1
base/src/expressions/parser/tests/test_locales.rs
Normal file
1
base/src/expressions/parser/tests/test_locales.rs
Normal file
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
@@ -1,8 +1,17 @@
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::expressions::parser::move_formula::{move_formula, MoveContext};
|
use crate::expressions::parser::move_formula::{move_formula as mf, MoveContext};
|
||||||
use crate::expressions::parser::Parser;
|
use crate::expressions::parser::tests::utils::new_parser;
|
||||||
|
use crate::expressions::parser::Node;
|
||||||
use crate::expressions::types::{Area, CellReferenceRC};
|
use crate::expressions::types::{Area, CellReferenceRC};
|
||||||
|
use crate::language::get_default_language;
|
||||||
|
use crate::locale::get_default_locale;
|
||||||
|
|
||||||
|
fn move_formula(node: &Node, context: &MoveContext) -> String {
|
||||||
|
let locale = get_default_locale();
|
||||||
|
let language = get_default_language();
|
||||||
|
mf(node, context, locale, language)
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_move_formula() {
|
fn test_move_formula() {
|
||||||
@@ -15,7 +24,7 @@ fn test_move_formula() {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Area is C2:F6
|
// Area is C2:F6
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
@@ -102,7 +111,7 @@ fn test_move_formula_context_offset() {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Area is C2:F6
|
// Area is C2:F6
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
@@ -140,7 +149,7 @@ fn test_move_formula_area_limits() {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Area is C2:F6
|
// Area is C2:F6
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
@@ -195,7 +204,7 @@ fn test_move_formula_ranges() {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
sheet: 0,
|
sheet: 0,
|
||||||
@@ -318,7 +327,7 @@ fn test_move_formula_wrong_reference() {
|
|||||||
height: 5,
|
height: 5,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Wrong formulas will NOT be displaced
|
// Wrong formulas will NOT be displaced
|
||||||
let node = parser.parse("Sheet3!AB31", context);
|
let node = parser.parse("Sheet3!AB31", context);
|
||||||
@@ -377,7 +386,7 @@ fn test_move_formula_misc() {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Area is C2:F6
|
// Area is C2:F6
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
@@ -445,7 +454,7 @@ fn test_move_formula_another_sheet() {
|
|||||||
};
|
};
|
||||||
// we add two sheets and we cut/paste from Sheet1 to Sheet2
|
// we add two sheets and we cut/paste from Sheet1 to Sheet2
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Area is C2:F6
|
// Area is C2:F6
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
@@ -487,7 +496,7 @@ fn move_formula_implicit_intersetion() {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Area is C2:F6
|
// Area is C2:F6
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
@@ -524,7 +533,7 @@ fn move_formula_implicit_intersetion_with_ranges() {
|
|||||||
column,
|
column,
|
||||||
};
|
};
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Area is C2:F6
|
// Area is C2:F6
|
||||||
let area = &Area {
|
let area = &Area {
|
||||||
|
|||||||
@@ -2,8 +2,8 @@ use std::collections::HashMap;
|
|||||||
|
|
||||||
use crate::expressions::lexer::LexerMode;
|
use crate::expressions::lexer::LexerMode;
|
||||||
|
|
||||||
use crate::expressions::parser::stringify::{to_rc_format, to_string};
|
use crate::expressions::parser::stringify::to_rc_format;
|
||||||
use crate::expressions::parser::Parser;
|
use crate::expressions::parser::tests::utils::{new_parser, to_english_localized_string};
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
|
|
||||||
struct Formula<'a> {
|
struct Formula<'a> {
|
||||||
@@ -14,7 +14,7 @@ struct Formula<'a> {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_parser_formulas_with_full_ranges() {
|
fn test_parser_formulas_with_full_ranges() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Second Sheet".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Second Sheet".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let formulas = vec![
|
let formulas = vec![
|
||||||
Formula {
|
Formula {
|
||||||
@@ -59,7 +59,10 @@ fn test_parser_formulas_with_full_ranges() {
|
|||||||
},
|
},
|
||||||
);
|
);
|
||||||
assert_eq!(to_rc_format(&t), formula.formula_r1c1);
|
assert_eq!(to_rc_format(&t), formula.formula_r1c1);
|
||||||
assert_eq!(to_string(&t, &cell_reference), formula.formula_a1);
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
formula.formula_a1
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Now the inverse
|
// Now the inverse
|
||||||
@@ -74,14 +77,17 @@ fn test_parser_formulas_with_full_ranges() {
|
|||||||
},
|
},
|
||||||
);
|
);
|
||||||
assert_eq!(to_rc_format(&t), formula.formula_r1c1);
|
assert_eq!(to_rc_format(&t), formula.formula_r1c1);
|
||||||
assert_eq!(to_string(&t, &cell_reference), formula.formula_a1);
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
formula.formula_a1
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_range_inverse_order() {
|
fn test_range_inverse_order() {
|
||||||
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
let worksheets = vec!["Sheet1".to_string(), "Sheet2".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -96,7 +102,7 @@ fn test_range_inverse_order() {
|
|||||||
&cell_reference,
|
&cell_reference,
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
to_string(&t, &cell_reference),
|
to_english_localized_string(&t, &cell_reference),
|
||||||
"SUM(C2:D4)*SUM(Sheet2!C4:D20)*SUM($C4:D$20)".to_string()
|
"SUM(C2:D4)*SUM(Sheet2!C4:D20)*SUM($C4:D$20)".to_string()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,14 +2,13 @@
|
|||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::expressions::parser::stringify::to_string;
|
use crate::expressions::parser::tests::utils::{new_parser, to_english_localized_string};
|
||||||
use crate::expressions::parser::Parser;
|
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn exp_order() {
|
fn exp_order() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
// Reference cell is Sheet1!A1
|
// Reference cell is Sheet1!A1
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -18,25 +17,34 @@ fn exp_order() {
|
|||||||
column: 1,
|
column: 1,
|
||||||
};
|
};
|
||||||
let t = parser.parse("(1 + 2)^3 + 4", &cell_reference);
|
let t = parser.parse("(1 + 2)^3 + 4", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "(1+2)^3+4");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
"(1+2)^3+4"
|
||||||
|
);
|
||||||
|
|
||||||
let t = parser.parse("(C5 + 3)^R4", &cell_reference);
|
let t = parser.parse("(C5 + 3)^R4", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "(C5+3)^R4");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
"(C5+3)^R4"
|
||||||
|
);
|
||||||
|
|
||||||
let t = parser.parse("(C5 + 3)^(R4*6)", &cell_reference);
|
let t = parser.parse("(C5 + 3)^(R4*6)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "(C5+3)^(R4*6)");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
"(C5+3)^(R4*6)"
|
||||||
|
);
|
||||||
|
|
||||||
let t = parser.parse("(C5)^(R4)", &cell_reference);
|
let t = parser.parse("(C5)^(R4)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "C5^R4");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "C5^R4");
|
||||||
|
|
||||||
let t = parser.parse("(5)^(4)", &cell_reference);
|
let t = parser.parse("(5)^(4)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "5^4");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "5^4");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn correct_parenthesis() {
|
fn correct_parenthesis() {
|
||||||
let worksheets = vec!["Sheet1".to_string()];
|
let worksheets = vec!["Sheet1".to_string()];
|
||||||
let mut parser = Parser::new(worksheets, vec![], HashMap::new());
|
let mut parser = new_parser(worksheets, vec![], HashMap::new());
|
||||||
|
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
sheet: "Sheet1".to_string(),
|
sheet: "Sheet1".to_string(),
|
||||||
@@ -45,26 +53,29 @@ fn correct_parenthesis() {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let t = parser.parse("-(1 + 1)", &cell_reference);
|
let t = parser.parse("-(1 + 1)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "-(1+1)");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "-(1+1)");
|
||||||
|
|
||||||
let t = parser.parse("1 - (3 + 4)", &cell_reference);
|
let t = parser.parse("1 - (3 + 4)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "1-(3+4)");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "1-(3+4)");
|
||||||
|
|
||||||
let t = parser.parse("-(1.05*(0.0284 + 0.0046) - 0.0284)", &cell_reference);
|
let t = parser.parse("-(1.05*(0.0284 + 0.0046) - 0.0284)", &cell_reference);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
to_string(&t, &cell_reference),
|
to_english_localized_string(&t, &cell_reference),
|
||||||
"-(1.05*(0.0284+0.0046)-0.0284)"
|
"-(1.05*(0.0284+0.0046)-0.0284)"
|
||||||
);
|
);
|
||||||
|
|
||||||
let t = parser.parse("1 + (3+5)", &cell_reference);
|
let t = parser.parse("1 + (3+5)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "1+3+5");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "1+3+5");
|
||||||
|
|
||||||
let t = parser.parse("1 - (3+5)", &cell_reference);
|
let t = parser.parse("1 - (3+5)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "1-(3+5)");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "1-(3+5)");
|
||||||
|
|
||||||
let t = parser.parse("(1 - 3) - (3+5)", &cell_reference);
|
let t = parser.parse("(1 - 3) - (3+5)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "1-3-(3+5)");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
"1-3-(3+5)"
|
||||||
|
);
|
||||||
|
|
||||||
let t = parser.parse("1 + (3<5)", &cell_reference);
|
let t = parser.parse("1 + (3<5)", &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "1+(3<5)");
|
assert_eq!(to_english_localized_string(&t, &cell_reference), "1+(3<5)");
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,8 +2,7 @@
|
|||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use crate::expressions::parser::stringify::to_string;
|
use crate::expressions::parser::tests::utils::{new_parser, to_english_localized_string};
|
||||||
use crate::expressions::parser::Parser;
|
|
||||||
use crate::expressions::types::CellReferenceRC;
|
use crate::expressions::types::CellReferenceRC;
|
||||||
use crate::expressions::utils::{number_to_column, parse_reference_a1};
|
use crate::expressions::utils::{number_to_column, parse_reference_a1};
|
||||||
use crate::types::{Table, TableColumn, TableStyleInfo};
|
use crate::types::{Table, TableColumn, TableStyleInfo};
|
||||||
@@ -62,7 +61,7 @@ fn simple_table() {
|
|||||||
let row_count = 3;
|
let row_count = 3;
|
||||||
let tables = create_test_table("tblIncome", &column_names, "A1", row_count);
|
let tables = create_test_table("tblIncome", &column_names, "A1", row_count);
|
||||||
|
|
||||||
let mut parser = Parser::new(worksheets, vec![], tables);
|
let mut parser = new_parser(worksheets, vec![], tables);
|
||||||
// Reference cell is 'Sheet One'!F2
|
// Reference cell is 'Sheet One'!F2
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
sheet: "Sheet One".to_string(),
|
sheet: "Sheet One".to_string(),
|
||||||
@@ -72,7 +71,10 @@ fn simple_table() {
|
|||||||
|
|
||||||
let formula = "SUM(tblIncome[[#This Row],[Jan]:[Dec]])";
|
let formula = "SUM(tblIncome[[#This Row],[Jan]:[Dec]])";
|
||||||
let t = parser.parse(formula, &cell_reference);
|
let t = parser.parse(formula, &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "SUM($A$2:$E$2)");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
"SUM($A$2:$E$2)"
|
||||||
|
);
|
||||||
|
|
||||||
// Cell A3
|
// Cell A3
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -82,7 +84,10 @@ fn simple_table() {
|
|||||||
};
|
};
|
||||||
let formula = "SUBTOTAL(109, tblIncome[Jan])";
|
let formula = "SUBTOTAL(109, tblIncome[Jan])";
|
||||||
let t = parser.parse(formula, &cell_reference);
|
let t = parser.parse(formula, &cell_reference);
|
||||||
assert_eq!(to_string(&t, &cell_reference), "SUBTOTAL(109,$A$2:$A$3)");
|
assert_eq!(
|
||||||
|
to_english_localized_string(&t, &cell_reference),
|
||||||
|
"SUBTOTAL(109,$A$2:$A$3)"
|
||||||
|
);
|
||||||
|
|
||||||
// Cell A3 in 'Second Sheet'
|
// Cell A3 in 'Second Sheet'
|
||||||
let cell_reference = CellReferenceRC {
|
let cell_reference = CellReferenceRC {
|
||||||
@@ -93,7 +98,7 @@ fn simple_table() {
|
|||||||
let formula = "SUBTOTAL(109, tblIncome[Jan])";
|
let formula = "SUBTOTAL(109, tblIncome[Jan])";
|
||||||
let t = parser.parse(formula, &cell_reference);
|
let t = parser.parse(formula, &cell_reference);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
to_string(&t, &cell_reference),
|
to_english_localized_string(&t, &cell_reference),
|
||||||
"SUBTOTAL(109,'Sheet One'!$A$2:$A$3)"
|
"SUBTOTAL(109,'Sheet One'!$A$2:$A$3)"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
29
base/src/expressions/parser/tests/utils.rs
Normal file
29
base/src/expressions/parser/tests/utils.rs
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
expressions::{
|
||||||
|
parser::{DefinedNameS, Node, Parser},
|
||||||
|
types::CellReferenceRC,
|
||||||
|
},
|
||||||
|
language::get_default_language,
|
||||||
|
locale::get_default_locale,
|
||||||
|
types::Table,
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::expressions::parser::stringify::to_localized_string;
|
||||||
|
|
||||||
|
pub fn to_english_localized_string(t: &Node, cell_reference: &CellReferenceRC) -> String {
|
||||||
|
let locale = get_default_locale();
|
||||||
|
let language = get_default_language();
|
||||||
|
to_localized_string(t, cell_reference, locale, language)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn new_parser<'a>(
|
||||||
|
worksheets: Vec<String>,
|
||||||
|
defined_names: Vec<DefinedNameS>,
|
||||||
|
tables: HashMap<String, Table>,
|
||||||
|
) -> Parser<'a> {
|
||||||
|
let locale = get_default_locale();
|
||||||
|
let language = get_default_language();
|
||||||
|
Parser::new(worksheets, defined_names, tables, locale, language)
|
||||||
|
}
|
||||||
@@ -241,6 +241,7 @@ pub enum TokenType {
|
|||||||
Percent, // %
|
Percent, // %
|
||||||
And, // &
|
And, // &
|
||||||
At, // @
|
At, // @
|
||||||
|
Backslash, // \
|
||||||
Reference {
|
Reference {
|
||||||
sheet: Option<String>,
|
sheet: Option<String>,
|
||||||
row: i32,
|
row: i32,
|
||||||
|
|||||||
@@ -211,15 +211,19 @@ pub fn parse_reference_a1(r: &str) -> Option<ParsedReference> {
|
|||||||
pub fn is_valid_identifier(name: &str) -> bool {
|
pub fn is_valid_identifier(name: &str) -> bool {
|
||||||
// https://support.microsoft.com/en-us/office/names-in-formulas-fc2935f9-115d-4bef-a370-3aa8bb4c91f1
|
// https://support.microsoft.com/en-us/office/names-in-formulas-fc2935f9-115d-4bef-a370-3aa8bb4c91f1
|
||||||
// https://github.com/MartinTrummer/excel-names/
|
// https://github.com/MartinTrummer/excel-names/
|
||||||
// NOTE: We are being much more restrictive than Excel.
|
|
||||||
// In particular we do not support non ascii characters.
|
|
||||||
let upper = name.to_ascii_uppercase();
|
let upper = name.to_ascii_uppercase();
|
||||||
let bytes = upper.as_bytes();
|
// length of chars
|
||||||
let len = bytes.len();
|
let len = upper.chars().count();
|
||||||
|
|
||||||
|
let mut chars = upper.chars();
|
||||||
|
|
||||||
if len > 255 || len == 0 {
|
if len > 255 || len == 0 {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
let first = bytes[0] as char;
|
let first = match chars.next() {
|
||||||
|
Some(ch) => ch,
|
||||||
|
None => return false,
|
||||||
|
};
|
||||||
// The first character of a name must be a letter, an underscore character (_), or a backslash (\).
|
// The first character of a name must be a letter, an underscore character (_), or a backslash (\).
|
||||||
if !(first.is_ascii_alphabetic() || first == '_' || first == '\\') {
|
if !(first.is_ascii_alphabetic() || first == '_' || first == '\\') {
|
||||||
return false;
|
return false;
|
||||||
@@ -237,20 +241,10 @@ pub fn is_valid_identifier(name: &str) -> bool {
|
|||||||
if parse_reference_r1c1(name).is_some() {
|
if parse_reference_r1c1(name).is_some() {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
let mut i = 1;
|
for ch in chars {
|
||||||
while i < len {
|
if !(ch.is_alphanumeric() || ch == '_' || ch == '.') {
|
||||||
let ch = bytes[i] as char;
|
return false;
|
||||||
match ch {
|
|
||||||
'a'..='z' => {}
|
|
||||||
'A'..='Z' => {}
|
|
||||||
'0'..='9' => {}
|
|
||||||
'_' => {}
|
|
||||||
'.' => {}
|
|
||||||
_ => {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
i += 1;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
true
|
true
|
||||||
|
|||||||
@@ -196,6 +196,7 @@ fn test_names() {
|
|||||||
assert!(is_valid_identifier("_."));
|
assert!(is_valid_identifier("_."));
|
||||||
assert!(is_valid_identifier("_1"));
|
assert!(is_valid_identifier("_1"));
|
||||||
assert!(is_valid_identifier("\\."));
|
assert!(is_valid_identifier("\\."));
|
||||||
|
assert!(is_valid_identifier("truñe"));
|
||||||
|
|
||||||
// invalid
|
// invalid
|
||||||
assert!(!is_valid_identifier("true"));
|
assert!(!is_valid_identifier("true"));
|
||||||
@@ -209,7 +210,6 @@ fn test_names() {
|
|||||||
assert!(!is_valid_identifier("1true"));
|
assert!(!is_valid_identifier("1true"));
|
||||||
|
|
||||||
assert!(!is_valid_identifier("test€"));
|
assert!(!is_valid_identifier("test€"));
|
||||||
assert!(!is_valid_identifier("truñe"));
|
|
||||||
assert!(!is_valid_identifier("tr&ue"));
|
assert!(!is_valid_identifier("tr&ue"));
|
||||||
|
|
||||||
assert!(!is_valid_identifier("LOG10"));
|
assert!(!is_valid_identifier("LOG10"));
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ pub struct Formatted {
|
|||||||
|
|
||||||
/// Returns the vector of chars of the fractional part of a *positive* number:
|
/// Returns the vector of chars of the fractional part of a *positive* number:
|
||||||
/// 3.1415926 ==> ['1', '4', '1', '5', '9', '2', '6']
|
/// 3.1415926 ==> ['1', '4', '1', '5', '9', '2', '6']
|
||||||
fn get_fract_part(value: f64, precision: i32) -> Vec<char> {
|
fn get_fract_part(value: f64, precision: i32, int_len: usize) -> Vec<char> {
|
||||||
let b = format!("{:.1$}", value.fract(), precision as usize)
|
let b = format!("{:.1$}", value.fract(), precision as usize)
|
||||||
.chars()
|
.chars()
|
||||||
.collect::<Vec<char>>();
|
.collect::<Vec<char>>();
|
||||||
@@ -30,6 +30,12 @@ fn get_fract_part(value: f64, precision: i32) -> Vec<char> {
|
|||||||
if last_non_zero < 2 {
|
if last_non_zero < 2 {
|
||||||
return vec![];
|
return vec![];
|
||||||
}
|
}
|
||||||
|
let max_len = if int_len > 15 {
|
||||||
|
2_usize
|
||||||
|
} else {
|
||||||
|
15_usize - int_len + 1
|
||||||
|
};
|
||||||
|
let last_non_zero = usize::min(last_non_zero, max_len + 1);
|
||||||
b[2..last_non_zero].to_vec()
|
b[2..last_non_zero].to_vec()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -122,6 +128,9 @@ pub fn format_number(value_original: f64, format: &str, locale: &Locale) -> Form
|
|||||||
if (1.0e-8..1.0e+11).contains(&value_abs) {
|
if (1.0e-8..1.0e+11).contains(&value_abs) {
|
||||||
let mut text = format!("{value:.9}");
|
let mut text = format!("{value:.9}");
|
||||||
text = text.trim_end_matches('0').trim_end_matches('.').to_string();
|
text = text.trim_end_matches('0').trim_end_matches('.').to_string();
|
||||||
|
if locale.numbers.symbols.decimal != "." {
|
||||||
|
text = text.replace('.', &locale.numbers.symbols.decimal.to_string());
|
||||||
|
}
|
||||||
Formatted {
|
Formatted {
|
||||||
text,
|
text,
|
||||||
color: None,
|
color: None,
|
||||||
@@ -139,13 +148,17 @@ pub fn format_number(value_original: f64, format: &str, locale: &Locale) -> Form
|
|||||||
value /= 10.0_f64.powf(exponent);
|
value /= 10.0_f64.powf(exponent);
|
||||||
let sign = if exponent < 0.0 { '-' } else { '+' };
|
let sign = if exponent < 0.0 { '-' } else { '+' };
|
||||||
let s = format!("{value:.5}");
|
let s = format!("{value:.5}");
|
||||||
|
let mut text = format!(
|
||||||
|
"{}E{}{:02}",
|
||||||
|
s.trim_end_matches('0').trim_end_matches('.'),
|
||||||
|
sign,
|
||||||
|
exponent.abs()
|
||||||
|
);
|
||||||
|
if locale.numbers.symbols.decimal != "." {
|
||||||
|
text = text.replace('.', &locale.numbers.symbols.decimal.to_string());
|
||||||
|
}
|
||||||
Formatted {
|
Formatted {
|
||||||
text: format!(
|
text,
|
||||||
"{}E{}{:02}",
|
|
||||||
s.trim_end_matches('0').trim_end_matches('.'),
|
|
||||||
sign,
|
|
||||||
exponent.abs()
|
|
||||||
),
|
|
||||||
color: None,
|
color: None,
|
||||||
error: None,
|
error: None,
|
||||||
}
|
}
|
||||||
@@ -423,7 +436,7 @@ pub fn format_number(value_original: f64, format: &str, locale: &Locale) -> Form
|
|||||||
if value_abs as i64 == 0 {
|
if value_abs as i64 == 0 {
|
||||||
int_part = vec![];
|
int_part = vec![];
|
||||||
}
|
}
|
||||||
let fract_part = get_fract_part(value_abs, p.precision);
|
let fract_part = get_fract_part(value_abs, p.precision, int_part.len());
|
||||||
// ln is the number of digits of the integer part of the value
|
// ln is the number of digits of the integer part of the value
|
||||||
let ln = int_part.len() as i32;
|
let ln = int_part.len() as i32;
|
||||||
// digit count is the number of digit tokens ('0', '?' and '#') to the left of the decimal point
|
// digit count is the number of digit tokens ('0', '?' and '#') to the left of the decimal point
|
||||||
@@ -744,15 +757,17 @@ fn parse_date(value: &str) -> Result<(i32, String), String> {
|
|||||||
/// "30.34%" => (0.3034, "0.00%")
|
/// "30.34%" => (0.3034, "0.00%")
|
||||||
/// 100€ => (100, "100€")
|
/// 100€ => (100, "100€")
|
||||||
pub(crate) fn parse_formatted_number(
|
pub(crate) fn parse_formatted_number(
|
||||||
value: &str,
|
original: &str,
|
||||||
currencies: &[&str],
|
currencies: &[&str],
|
||||||
|
decimal_separator: u8,
|
||||||
|
group_separator: u8,
|
||||||
) -> Result<(f64, Option<String>), String> {
|
) -> Result<(f64, Option<String>), String> {
|
||||||
let value = value.trim();
|
let value = original.trim();
|
||||||
let scientific_format = "0.00E+00";
|
let scientific_format = "0.00E+00";
|
||||||
|
|
||||||
// Check if it is a percentage
|
// Check if it is a percentage
|
||||||
if let Some(p) = value.strip_suffix('%') {
|
if let Some(p) = value.strip_suffix('%') {
|
||||||
let (f, options) = parse_number(p.trim())?;
|
let (f, options) = parse_number(p.trim(), decimal_separator, group_separator)?;
|
||||||
if options.is_scientific {
|
if options.is_scientific {
|
||||||
return Ok((f / 100.0, Some(scientific_format.to_string())));
|
return Ok((f / 100.0, Some(scientific_format.to_string())));
|
||||||
}
|
}
|
||||||
@@ -768,7 +783,7 @@ pub(crate) fn parse_formatted_number(
|
|||||||
// check if it is a currency in currencies
|
// check if it is a currency in currencies
|
||||||
for currency in currencies {
|
for currency in currencies {
|
||||||
if let Some(p) = value.strip_prefix(&format!("-{currency}")) {
|
if let Some(p) = value.strip_prefix(&format!("-{currency}")) {
|
||||||
let (f, options) = parse_number(p.trim())?;
|
let (f, options) = parse_number(p.trim(), decimal_separator, group_separator)?;
|
||||||
if options.is_scientific {
|
if options.is_scientific {
|
||||||
return Ok((f, Some(scientific_format.to_string())));
|
return Ok((f, Some(scientific_format.to_string())));
|
||||||
}
|
}
|
||||||
@@ -777,7 +792,7 @@ pub(crate) fn parse_formatted_number(
|
|||||||
}
|
}
|
||||||
return Ok((-f, Some(format!("{currency}#,##0"))));
|
return Ok((-f, Some(format!("{currency}#,##0"))));
|
||||||
} else if let Some(p) = value.strip_prefix(currency) {
|
} else if let Some(p) = value.strip_prefix(currency) {
|
||||||
let (f, options) = parse_number(p.trim())?;
|
let (f, options) = parse_number(p.trim(), decimal_separator, group_separator)?;
|
||||||
if options.is_scientific {
|
if options.is_scientific {
|
||||||
return Ok((f, Some(scientific_format.to_string())));
|
return Ok((f, Some(scientific_format.to_string())));
|
||||||
}
|
}
|
||||||
@@ -786,7 +801,7 @@ pub(crate) fn parse_formatted_number(
|
|||||||
}
|
}
|
||||||
return Ok((f, Some(format!("{currency}#,##0"))));
|
return Ok((f, Some(format!("{currency}#,##0"))));
|
||||||
} else if let Some(p) = value.strip_suffix(currency) {
|
} else if let Some(p) = value.strip_suffix(currency) {
|
||||||
let (f, options) = parse_number(p.trim())?;
|
let (f, options) = parse_number(p.trim(), decimal_separator, group_separator)?;
|
||||||
if options.is_scientific {
|
if options.is_scientific {
|
||||||
return Ok((f, Some(scientific_format.to_string())));
|
return Ok((f, Some(scientific_format.to_string())));
|
||||||
}
|
}
|
||||||
@@ -799,12 +814,13 @@ pub(crate) fn parse_formatted_number(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Ok((serial_number, format)) = parse_date(value) {
|
// check if it is a date. NOTE: we don't trim the original here
|
||||||
|
if let Ok((serial_number, format)) = parse_date(original) {
|
||||||
return Ok((serial_number as f64, Some(format)));
|
return Ok((serial_number as f64, Some(format)));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lastly we check if it is a number
|
// Lastly we check if it is a number
|
||||||
let (f, options) = parse_number(value)?;
|
let (f, options) = parse_number(value, decimal_separator, group_separator)?;
|
||||||
if options.is_scientific {
|
if options.is_scientific {
|
||||||
return Ok((f, Some(scientific_format.to_string())));
|
return Ok((f, Some(scientific_format.to_string())));
|
||||||
}
|
}
|
||||||
@@ -827,7 +843,11 @@ struct NumberOptions {
|
|||||||
|
|
||||||
// tries to parse 'value' as a number.
|
// tries to parse 'value' as a number.
|
||||||
// If it is a number it either uses commas as thousands separator or it does not
|
// If it is a number it either uses commas as thousands separator or it does not
|
||||||
fn parse_number(value: &str) -> Result<(f64, NumberOptions), String> {
|
fn parse_number(
|
||||||
|
value: &str,
|
||||||
|
decimal_separator: u8,
|
||||||
|
group_separator: u8,
|
||||||
|
) -> Result<(f64, NumberOptions), String> {
|
||||||
let mut position = 0;
|
let mut position = 0;
|
||||||
let bytes = value.as_bytes();
|
let bytes = value.as_bytes();
|
||||||
let len = bytes.len();
|
let len = bytes.len();
|
||||||
@@ -835,8 +855,6 @@ fn parse_number(value: &str) -> Result<(f64, NumberOptions), String> {
|
|||||||
return Err("Cannot parse number".to_string());
|
return Err("Cannot parse number".to_string());
|
||||||
}
|
}
|
||||||
let mut chars = String::from("");
|
let mut chars = String::from("");
|
||||||
let decimal_separator = b'.';
|
|
||||||
let group_separator = b',';
|
|
||||||
let mut group_separator_index = Vec::new();
|
let mut group_separator_index = Vec::new();
|
||||||
// get the sign
|
// get the sign
|
||||||
let sign = if bytes[0] == b'-' {
|
let sign = if bytes[0] == b'-' {
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
#![allow(clippy::unwrap_used)]
|
#![allow(clippy::unwrap_used)]
|
||||||
|
#![allow(clippy::expect_used)]
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
formatter::format::format_number,
|
formatter::format::format_number,
|
||||||
@@ -202,3 +203,9 @@ fn test_date() {
|
|||||||
"Sat-September-12"
|
"Sat-September-12"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_german_locale() {
|
||||||
|
let locale = get_locale("de").expect("");
|
||||||
|
assert_eq!(format_number(1234.56, "General", locale).text, "1234,56");
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,9 +1,13 @@
|
|||||||
#![allow(clippy::unwrap_used)]
|
#![allow(clippy::unwrap_used)]
|
||||||
|
|
||||||
use crate::formatter::format::parse_formatted_number as parse;
|
use crate::formatter::format::parse_formatted_number;
|
||||||
|
|
||||||
const PARSE_ERROR_MSG: &str = "Could not parse number";
|
const PARSE_ERROR_MSG: &str = "Could not parse number";
|
||||||
|
|
||||||
|
fn parse(input: &str, currencies: &[&str]) -> Result<(f64, Option<String>), String> {
|
||||||
|
parse_formatted_number(input, currencies, b'.', b',')
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn numbers() {
|
fn numbers() {
|
||||||
// whole numbers
|
// whole numbers
|
||||||
|
|||||||
@@ -137,7 +137,7 @@ pub(crate) fn binary_search_descending_or_greater<T: Ord>(target: &T, array: &[T
|
|||||||
Some((n - r - 1) as i32)
|
Some((n - r - 1) as i32)
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
/// Returns an array with the list of cell values in the range
|
/// Returns an array with the list of cell values in the range
|
||||||
pub(crate) fn prepare_array(
|
pub(crate) fn prepare_array(
|
||||||
&mut self,
|
&mut self,
|
||||||
|
|||||||
946
base/src/functions/database.rs
Normal file
946
base/src/functions/database.rs
Normal file
@@ -0,0 +1,946 @@
|
|||||||
|
use chrono::Datelike;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult,
|
||||||
|
expressions::{parser::Node, token::Error, types::CellReferenceIndex},
|
||||||
|
formatter::dates::date_to_serial_number,
|
||||||
|
Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::util::{compare_values, from_wildcard_to_regex, result_matches_regex};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// =DAVERAGE(database, field, criteria)
|
||||||
|
pub(crate) fn fn_daverage(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = match self.resolve_db_field_column(db_left, db_right, &args[1], cell) {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0f64;
|
||||||
|
let mut count = 0usize;
|
||||||
|
|
||||||
|
let mut row = db_left.row + 1; // skip header
|
||||||
|
while row <= db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
});
|
||||||
|
if let CalcResult::Number(n) = v {
|
||||||
|
if n.is_finite() {
|
||||||
|
sum += n;
|
||||||
|
count += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
row += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "No numeric values matched criteria".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(sum / count as f64)
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DCOUNT(database, field, criteria)
|
||||||
|
// Counts numeric entries in the field for rows that match criteria
|
||||||
|
pub(crate) fn fn_dcount(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = match self.resolve_db_field_column(db_left, db_right, &args[1], cell) {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut count = 0usize;
|
||||||
|
let mut row = db_left.row + 1; // skip header
|
||||||
|
while row <= db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
});
|
||||||
|
if matches!(v, CalcResult::Number(_)) {
|
||||||
|
count += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
row += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(count as f64)
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DGET(database, field, criteria)
|
||||||
|
// Returns the (single) field value for the unique matching row
|
||||||
|
pub(crate) fn fn_dget(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = match self.resolve_db_field_column(db_left, db_right, &args[1], cell) {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut result: Option<CalcResult> = None;
|
||||||
|
let mut matches = 0usize;
|
||||||
|
|
||||||
|
let mut row = db_left.row + 1;
|
||||||
|
while row <= db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
matches += 1;
|
||||||
|
if matches > 1 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "More than one matching record".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
result = Some(self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
row += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
match (matches, result) {
|
||||||
|
(0, _) | (_, None) => CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No matching record".to_string(),
|
||||||
|
},
|
||||||
|
(_, Some(v)) => v,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DMAX(database, field, criteria)
|
||||||
|
pub(crate) fn fn_dmax(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
self.db_extreme(args, cell, true)
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DMIN(database, field, criteria)
|
||||||
|
pub(crate) fn fn_dmin(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
self.db_extreme(args, cell, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DSUM(database, field, criteria)
|
||||||
|
pub(crate) fn fn_dsum(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = match self.resolve_db_field_column(db_left, db_right, &args[1], cell) {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
|
||||||
|
// skip header
|
||||||
|
let mut row = db_left.row + 1;
|
||||||
|
while row <= db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
});
|
||||||
|
if let CalcResult::Number(n) = v {
|
||||||
|
if n.is_finite() {
|
||||||
|
sum += n;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
row += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(sum)
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DCOUNTA(database, field, criteria)
|
||||||
|
// Counts non-empty entries (any type) in the field for rows that match criteria
|
||||||
|
pub(crate) fn fn_dcounta(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = match self.resolve_db_field_column(db_left, db_right, &args[1], cell) {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut count = 0;
|
||||||
|
for row in (db_left.row + 1)..=db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
});
|
||||||
|
if !matches!(v, CalcResult::EmptyCell | CalcResult::EmptyArg) {
|
||||||
|
count += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(count as f64)
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DPRODUCT(database, field, criteria)
|
||||||
|
pub(crate) fn fn_dproduct(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = match self.resolve_db_field_column(db_left, db_right, &args[1], cell) {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut product = 1.0f64;
|
||||||
|
let mut has_numeric = false;
|
||||||
|
|
||||||
|
let mut row = db_left.row + 1; // skip header
|
||||||
|
while row <= db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
});
|
||||||
|
if let CalcResult::Number(n) = v {
|
||||||
|
if n.is_finite() {
|
||||||
|
product *= n;
|
||||||
|
has_numeric = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
row += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Excel returns 0 when no rows / no numeric values match for DPRODUCT
|
||||||
|
if has_numeric {
|
||||||
|
CalcResult::Number(product)
|
||||||
|
} else {
|
||||||
|
CalcResult::Number(0.0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Small internal helper for DSTDEV / DVAR
|
||||||
|
// Collects sum, sum of squares, and count of numeric values in the field
|
||||||
|
// for rows that match the criteria.
|
||||||
|
fn db_numeric_stats(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> Result<(f64, f64, usize), CalcResult> {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return Err(CalcResult::new_args_number_error(cell));
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return Err(e),
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = self.resolve_db_field_column(db_left, db_right, &args[1], cell)?;
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return Err(e),
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return Err(CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0f64;
|
||||||
|
let mut sumsq = 0.0f64;
|
||||||
|
let mut count = 0usize;
|
||||||
|
|
||||||
|
let mut row = db_left.row + 1; // skip header
|
||||||
|
while row <= db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
});
|
||||||
|
if let CalcResult::Number(n) = v {
|
||||||
|
if n.is_finite() {
|
||||||
|
sum += n;
|
||||||
|
sumsq += n * n;
|
||||||
|
count += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
row += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok((sum, sumsq, count))
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DSTDEV(database, field, criteria)
|
||||||
|
// Sample standard deviation of matching numeric values
|
||||||
|
pub(crate) fn fn_dstdev(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (sum, sumsq, count) = match self.db_numeric_stats(args, cell) {
|
||||||
|
Ok(stats) => stats,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel behaviour: #DIV/0! if 0 or 1 numeric values match
|
||||||
|
if count < 2 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "Not enough numeric values matched criteria".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let var = (sumsq - (sum * sum) / n) / (n - 1.0);
|
||||||
|
let var = if var < 0.0 { 0.0 } else { var };
|
||||||
|
CalcResult::Number(var.sqrt())
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DVAR(database, field, criteria)
|
||||||
|
// Sample variance of matching numeric values
|
||||||
|
pub(crate) fn fn_dvar(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (sum, sumsq, count) = match self.db_numeric_stats(args, cell) {
|
||||||
|
Ok(stats) => stats,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel behaviour: #DIV/0! if 0 or 1 numeric values match
|
||||||
|
if count < 2 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "Not enough numeric values matched criteria".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let var = (sumsq - (sum * sum) / n) / (n - 1.0);
|
||||||
|
let var = if var < 0.0 { 0.0 } else { var };
|
||||||
|
CalcResult::Number(var)
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DSTDEVP(database, field, criteria)
|
||||||
|
// Population standard deviation of matching numeric values
|
||||||
|
pub(crate) fn fn_dstdevp(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (sum, sumsq, count) = match self.db_numeric_stats(args, cell) {
|
||||||
|
Ok(stats) => stats,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel behaviour: #DIV/0! if no numeric values match
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "No numeric values matched criteria".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let var = (sumsq - (sum * sum) / n) / n;
|
||||||
|
let var = if var < 0.0 { 0.0 } else { var };
|
||||||
|
CalcResult::Number(var.sqrt())
|
||||||
|
}
|
||||||
|
|
||||||
|
// =DVARP(database, field, criteria)
|
||||||
|
// Population variance of matching numeric values
|
||||||
|
pub(crate) fn fn_dvarp(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (sum, sumsq, count) = match self.db_numeric_stats(args, cell) {
|
||||||
|
Ok(stats) => stats,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel behaviour: #DIV/0! if no numeric values match
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "No numeric values matched criteria".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let var = (sumsq - (sum * sum) / n) / n;
|
||||||
|
let var = if var < 0.0 { 0.0 } else { var };
|
||||||
|
CalcResult::Number(var)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Resolve the "field" (2nd arg) to an absolute column index (i32) within the sheet.
|
||||||
|
/// Field can be a number (1-based index) or a header name (case-insensitive).
|
||||||
|
/// Returns the absolute column index, not a 1-based offset within the database range.
|
||||||
|
fn resolve_db_field_column(
|
||||||
|
&mut self,
|
||||||
|
db_left: CellReferenceIndex,
|
||||||
|
db_right: CellReferenceIndex,
|
||||||
|
field_arg: &Node,
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> Result<i32, CalcResult> {
|
||||||
|
let field_column_name = match self.evaluate_node_in_context(field_arg, cell) {
|
||||||
|
CalcResult::String(s) => s.to_lowercase(),
|
||||||
|
CalcResult::Number(index) => {
|
||||||
|
let index = index.floor() as i32;
|
||||||
|
if index < 1 || db_left.column + index - 1 > db_right.column {
|
||||||
|
return Err(CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "Field index out of range".to_string(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return Ok(db_left.column + index - 1);
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(b) => {
|
||||||
|
return if b {
|
||||||
|
Ok(db_left.column)
|
||||||
|
} else {
|
||||||
|
// Index 0 is out of range
|
||||||
|
Err(CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid field specifier".to_string(),
|
||||||
|
})
|
||||||
|
};
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => {
|
||||||
|
return Err(error);
|
||||||
|
}
|
||||||
|
CalcResult::Range { .. } => {
|
||||||
|
return Err(CalcResult::Error {
|
||||||
|
error: Error::NIMPL,
|
||||||
|
origin: cell,
|
||||||
|
message: "Arrays not supported yet".to_string(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
CalcResult::EmptyCell | CalcResult::EmptyArg => "".to_string(),
|
||||||
|
CalcResult::Array(_) => {
|
||||||
|
return Err(CalcResult::Error {
|
||||||
|
error: Error::NIMPL,
|
||||||
|
origin: cell,
|
||||||
|
message: "Arrays not supported yet".to_string(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// We search in the database a column whose header matches field_column_name
|
||||||
|
for column in db_left.column..=db_right.column {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row: db_left.row,
|
||||||
|
column,
|
||||||
|
});
|
||||||
|
match &v {
|
||||||
|
CalcResult::String(s) => {
|
||||||
|
if s.to_lowercase() == field_column_name {
|
||||||
|
return Ok(column);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Number(n) => {
|
||||||
|
if field_column_name == n.to_string() {
|
||||||
|
return Ok(column);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(b) => {
|
||||||
|
if field_column_name == b.to_string() {
|
||||||
|
return Ok(column);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Error { .. }
|
||||||
|
| CalcResult::Range { .. }
|
||||||
|
| CalcResult::EmptyCell
|
||||||
|
| CalcResult::EmptyArg
|
||||||
|
| CalcResult::Array(_) => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Err(CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "Field header not found".to_string(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Check whether a database row matches the criteria range.
|
||||||
|
/// Criteria logic: OR across criteria rows; AND across columns within a row.
|
||||||
|
fn db_row_matches_criteria(
|
||||||
|
&mut self,
|
||||||
|
db_left: CellReferenceIndex,
|
||||||
|
db_right: CellReferenceIndex,
|
||||||
|
row: i32,
|
||||||
|
criteria: (CellReferenceIndex, CellReferenceIndex),
|
||||||
|
) -> bool {
|
||||||
|
let (c_left, c_right) = criteria;
|
||||||
|
|
||||||
|
// Read criteria headers (first row of criteria range)
|
||||||
|
// Map header name (lowercased) -> db column (if exists)
|
||||||
|
let mut crit_cols: Vec<i32> = Vec::new();
|
||||||
|
let mut header_count = 0;
|
||||||
|
// We cover the criteria table:
|
||||||
|
// headerA | headerB | ...
|
||||||
|
// critA1 | critA2 | ...
|
||||||
|
// critB1 | critB2 | ...
|
||||||
|
// ...
|
||||||
|
for column in c_left.column..=c_right.column {
|
||||||
|
let cell = CellReferenceIndex {
|
||||||
|
sheet: c_left.sheet,
|
||||||
|
row: c_left.row,
|
||||||
|
column,
|
||||||
|
};
|
||||||
|
let criteria_header = self.evaluate_cell(cell);
|
||||||
|
if let Ok(s) = self.cast_to_string(criteria_header, cell) {
|
||||||
|
// Non-empty string header. If the header is non string we skip it
|
||||||
|
header_count += 1;
|
||||||
|
let wanted = s.to_lowercase();
|
||||||
|
|
||||||
|
// Find corresponding Database column
|
||||||
|
let mut found = false;
|
||||||
|
for db_column in db_left.column..=db_right.column {
|
||||||
|
let db_header = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row: db_left.row,
|
||||||
|
column: db_column,
|
||||||
|
});
|
||||||
|
if let Ok(hs) = self.cast_to_string(db_header, cell) {
|
||||||
|
if hs.to_lowercase() == wanted {
|
||||||
|
crit_cols.push(db_column);
|
||||||
|
found = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
// that means the criteria column has no matching DB column
|
||||||
|
// If the criteria condition is empty then we remove this condition
|
||||||
|
// otherwise this condition can never be satisfied
|
||||||
|
// We evaluate all criteria rows to see if any is non-empty
|
||||||
|
let mut has_non_empty = false;
|
||||||
|
for r in (c_left.row + 1)..=c_right.row {
|
||||||
|
let ccell = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: c_left.sheet,
|
||||||
|
row: r,
|
||||||
|
column,
|
||||||
|
});
|
||||||
|
if !matches!(ccell, CalcResult::EmptyCell | CalcResult::EmptyArg) {
|
||||||
|
has_non_empty = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if has_non_empty {
|
||||||
|
// This criteria column can never be satisfied
|
||||||
|
header_count -= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
if c_right.row <= c_left.row {
|
||||||
|
// If no criteria rows (only headers), everything matches
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if header_count == 0 {
|
||||||
|
// If there are not "String" headers, nothing matches
|
||||||
|
// NB: There might be String headers that do not match any DB columns,
|
||||||
|
// in that case everything matches.
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Evaluate each criteria row (OR)
|
||||||
|
for r in (c_left.row + 1)..=c_right.row {
|
||||||
|
// AND across columns for this criteria row
|
||||||
|
let mut and_ok = true;
|
||||||
|
|
||||||
|
for (offset, db_col) in crit_cols.iter().enumerate() {
|
||||||
|
// Criteria cell
|
||||||
|
let ccell = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: c_left.sheet,
|
||||||
|
row: r,
|
||||||
|
column: c_left.column + offset as i32,
|
||||||
|
});
|
||||||
|
|
||||||
|
// Empty criteria cell -> ignored
|
||||||
|
if matches!(ccell, CalcResult::EmptyCell | CalcResult::EmptyArg) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Database value for this row/column
|
||||||
|
let db_val = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: *db_col,
|
||||||
|
});
|
||||||
|
|
||||||
|
if !self.criteria_cell_matches(&db_val, &ccell) {
|
||||||
|
and_ok = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if and_ok {
|
||||||
|
// This criteria row satisfied (OR)
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// none matched
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Implements Excel-like criteria matching for a single value.
|
||||||
|
/// Supports prefixes: <>, >=, <=, >, <, = ; wildcards * and ? for string equals.
|
||||||
|
fn criteria_cell_matches(&self, db_val: &CalcResult, crit_cell: &CalcResult) -> bool {
|
||||||
|
// Convert the criteria cell to a string for operator parsing if possible,
|
||||||
|
// otherwise fall back to equality via compare_values.
|
||||||
|
|
||||||
|
let mut criteria = match crit_cell {
|
||||||
|
CalcResult::String(s) => s.trim().to_string(),
|
||||||
|
CalcResult::Number(n) => {
|
||||||
|
// treat as equality with number
|
||||||
|
return match db_val {
|
||||||
|
CalcResult::Number(v) => (*v - *n).abs() <= f64::EPSILON,
|
||||||
|
_ => false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(b) => {
|
||||||
|
// check equality with boolean
|
||||||
|
return match db_val {
|
||||||
|
CalcResult::Boolean(v) => *v == *b,
|
||||||
|
_ => false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
CalcResult::EmptyCell | CalcResult::EmptyArg => "".to_string(),
|
||||||
|
CalcResult::Error { .. } => return false,
|
||||||
|
CalcResult::Range { .. } | CalcResult::Array(_) => return false,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Detect operator prefix
|
||||||
|
let mut op = "="; // default equality (with wildcard semantics for strings)
|
||||||
|
let prefixes = ["<>", ">=", "<=", ">", "<", "="];
|
||||||
|
for p in prefixes.iter() {
|
||||||
|
if criteria.starts_with(p) {
|
||||||
|
op = p;
|
||||||
|
criteria = criteria[p.len()..].trim().to_string();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Is it a number?
|
||||||
|
let rhs_num = criteria.parse::<f64>().ok();
|
||||||
|
|
||||||
|
// Is it a date?
|
||||||
|
// FIXME: We should parse dates according to locale settings
|
||||||
|
let rhs_date = criteria.parse::<chrono::NaiveDate>().ok();
|
||||||
|
|
||||||
|
match op {
|
||||||
|
">" | ">=" | "<" | "<=" => {
|
||||||
|
if let Some(d) = rhs_date {
|
||||||
|
// date comparison
|
||||||
|
let serial = match date_to_serial_number(d.day(), d.month(), d.year()) {
|
||||||
|
Ok(sn) => sn as f64,
|
||||||
|
Err(_) => return false,
|
||||||
|
};
|
||||||
|
|
||||||
|
if let CalcResult::Number(n) = db_val {
|
||||||
|
match op {
|
||||||
|
">" => *n > serial,
|
||||||
|
">=" => *n >= serial,
|
||||||
|
"<" => *n < serial,
|
||||||
|
"<=" => *n <= serial,
|
||||||
|
_ => false,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
} else if let Some(t) = rhs_num {
|
||||||
|
// numeric comparison
|
||||||
|
if let CalcResult::Number(n) = db_val {
|
||||||
|
match op {
|
||||||
|
">" => *n > t,
|
||||||
|
">=" => *n >= t,
|
||||||
|
"<" => *n < t,
|
||||||
|
"<=" => *n <= t,
|
||||||
|
_ => false,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// string comparison (case-insensitive) using compare_values semantics
|
||||||
|
let rhs = CalcResult::String(criteria.to_lowercase());
|
||||||
|
let lhs = match db_val {
|
||||||
|
CalcResult::String(s) => CalcResult::String(s.to_lowercase()),
|
||||||
|
x => x.clone(),
|
||||||
|
};
|
||||||
|
let c = compare_values(&lhs, &rhs);
|
||||||
|
match op {
|
||||||
|
">" => c > 0,
|
||||||
|
">=" => c >= 0,
|
||||||
|
"<" => c < 0,
|
||||||
|
"<=" => c <= 0,
|
||||||
|
_ => false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"<>" => {
|
||||||
|
// not equal (with wildcard semantics for strings)
|
||||||
|
// If rhs has wildcards and db_val is string, do regex; else use compare_values != 0
|
||||||
|
if let CalcResult::String(s) = db_val {
|
||||||
|
if criteria.contains('*') || criteria.contains('?') {
|
||||||
|
if let Ok(re) = from_wildcard_to_regex(&criteria.to_lowercase(), true) {
|
||||||
|
return !result_matches_regex(
|
||||||
|
&CalcResult::String(s.to_lowercase()),
|
||||||
|
&re,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let rhs = if let Some(n) = rhs_num {
|
||||||
|
CalcResult::Number(n)
|
||||||
|
} else {
|
||||||
|
CalcResult::String(criteria.to_lowercase())
|
||||||
|
};
|
||||||
|
let lhs = match db_val {
|
||||||
|
CalcResult::String(s) => CalcResult::String(s.to_lowercase()),
|
||||||
|
x => x.clone(),
|
||||||
|
};
|
||||||
|
compare_values(&lhs, &rhs) != 0
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// equality. For strings, support wildcards (*, ?)
|
||||||
|
if let Some(n) = rhs_num {
|
||||||
|
// numeric equals
|
||||||
|
if let CalcResult::Number(m) = db_val {
|
||||||
|
(*m - n).abs() <= f64::EPSILON
|
||||||
|
} else {
|
||||||
|
compare_values(db_val, &CalcResult::Number(n)) == 0
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// textual/boolean equals (case-insensitive), wildcard-enabled for strings
|
||||||
|
if let CalcResult::String(s) = db_val {
|
||||||
|
if criteria.contains('*') || criteria.contains('?') {
|
||||||
|
if let Ok(re) = from_wildcard_to_regex(&criteria.to_lowercase(), true) {
|
||||||
|
return result_matches_regex(
|
||||||
|
&CalcResult::String(s.to_lowercase()),
|
||||||
|
&re,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// This is weird but we only need to check if "starts with" for equality
|
||||||
|
return s.to_lowercase().starts_with(&criteria.to_lowercase());
|
||||||
|
}
|
||||||
|
// Fallback: compare_values equality
|
||||||
|
compare_values(db_val, &CalcResult::String(criteria.to_lowercase())) == 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Shared implementation for DMAX/DMIN
|
||||||
|
fn db_extreme(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
want_max: bool,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (db_left, db_right) = match self.get_reference(&args[0], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_col = match self.resolve_db_field_column(db_left, db_right, &args[1], cell) {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let criteria = match self.get_reference(&args[2], cell) {
|
||||||
|
Ok(r) => (r.left, r.right),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if db_right.row <= db_left.row {
|
||||||
|
// no data rows
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "No data rows in database".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut best: Option<f64> = None;
|
||||||
|
|
||||||
|
let mut row = db_left.row + 1;
|
||||||
|
while row <= db_right.row {
|
||||||
|
if self.db_row_matches_criteria(db_left, db_right, row, criteria) {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: db_left.sheet,
|
||||||
|
row,
|
||||||
|
column: field_col,
|
||||||
|
});
|
||||||
|
if let CalcResult::Number(value) = v {
|
||||||
|
if value.is_finite() {
|
||||||
|
best = Some(match best {
|
||||||
|
None => value,
|
||||||
|
Some(cur) => {
|
||||||
|
if want_max {
|
||||||
|
value.max(cur)
|
||||||
|
} else {
|
||||||
|
value.min(cur)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
row += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
match best {
|
||||||
|
Some(v) => CalcResult::Number(v),
|
||||||
|
None => CalcResult::Number(0.0),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -4,10 +4,31 @@ use chrono::Months;
|
|||||||
use chrono::NaiveDateTime;
|
use chrono::NaiveDateTime;
|
||||||
use chrono::NaiveTime;
|
use chrono::NaiveTime;
|
||||||
use chrono::Timelike;
|
use chrono::Timelike;
|
||||||
|
use chrono_tz::Tz;
|
||||||
|
|
||||||
const SECONDS_PER_DAY: i32 = 86_400;
|
const SECONDS_PER_DAY: i32 = 86_400;
|
||||||
const SECONDS_PER_DAY_F64: f64 = SECONDS_PER_DAY as f64;
|
const SECONDS_PER_DAY_F64: f64 = SECONDS_PER_DAY as f64;
|
||||||
|
|
||||||
|
fn is_leap_year(year: i32) -> bool {
|
||||||
|
(year % 4 == 0) && (year % 100 != 0 || year % 400 == 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn is_feb_29_between_dates(start: chrono::NaiveDate, end: chrono::NaiveDate) -> bool {
|
||||||
|
let start_year = start.year();
|
||||||
|
let end_year = end.year();
|
||||||
|
|
||||||
|
for year in start_year..=end_year {
|
||||||
|
if is_leap_year(year)
|
||||||
|
&& (year < end_year
|
||||||
|
|| (year == end_year && end.month() > 2)
|
||||||
|
&& (year > start_year || (year == start_year && start.month() <= 2)))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
// ---------------------------------------------------------------------------
|
// ---------------------------------------------------------------------------
|
||||||
// Helper macros to eliminate boilerplate in date/time component extraction
|
// Helper macros to eliminate boilerplate in date/time component extraction
|
||||||
// functions (DAY, MONTH, YEAR, HOUR, MINUTE, SECOND).
|
// functions (DAY, MONTH, YEAR, HOUR, MINUTE, SECOND).
|
||||||
@@ -449,7 +470,7 @@ fn parse_datevalue_text(value: &str) -> Result<i32, String> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
fn get_date_serial(
|
fn get_date_serial(
|
||||||
&mut self,
|
&mut self,
|
||||||
node: &Node,
|
node: &Node,
|
||||||
@@ -750,12 +771,12 @@ impl Model {
|
|||||||
Ok(values)
|
Ok(values)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns the current date/time as an Excel serial number in the model's configured timezone.
|
// Returns the current date/time as an Excel serial number in the given timezone.
|
||||||
// Used by TODAY() and NOW().
|
// Used by TODAY() and NOW().
|
||||||
fn current_excel_serial(&self) -> Option<f64> {
|
pub(crate) fn current_excel_serial_with_timezone(&self, tz: Tz) -> Option<f64> {
|
||||||
let seconds = get_milliseconds_since_epoch() / 1000;
|
let seconds = get_milliseconds_since_epoch() / 1000;
|
||||||
DateTime::from_timestamp(seconds, 0).map(|dt| {
|
DateTime::from_timestamp(seconds, 0).map(|dt| {
|
||||||
let local_time = dt.with_timezone(&self.tz);
|
let local_time = dt.with_timezone(&tz);
|
||||||
let days_from_1900 = local_time.num_days_from_ce() - EXCEL_DATE_BASE;
|
let days_from_1900 = local_time.num_days_from_ce() - EXCEL_DATE_BASE;
|
||||||
let fraction = (local_time.num_seconds_from_midnight() as f64) / (60.0 * 60.0 * 24.0);
|
let fraction = (local_time.num_seconds_from_midnight() as f64) / (60.0 * 60.0 * 24.0);
|
||||||
days_from_1900 as f64 + fraction
|
days_from_1900 as f64 + fraction
|
||||||
@@ -958,7 +979,7 @@ impl Model {
|
|||||||
message: "Wrong number of arguments".to_string(),
|
message: "Wrong number of arguments".to_string(),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
match self.current_excel_serial() {
|
match self.current_excel_serial_with_timezone(self.tz) {
|
||||||
Some(serial) => CalcResult::Number(serial.floor()),
|
Some(serial) => CalcResult::Number(serial.floor()),
|
||||||
None => CalcResult::Error {
|
None => CalcResult::Error {
|
||||||
error: Error::ERROR,
|
error: Error::ERROR,
|
||||||
@@ -969,14 +990,35 @@ impl Model {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn fn_now(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_now(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if !args.is_empty() {
|
if args.len() > 1 {
|
||||||
return CalcResult::Error {
|
return CalcResult::Error {
|
||||||
error: Error::ERROR,
|
error: Error::ERROR,
|
||||||
origin: cell,
|
origin: cell,
|
||||||
message: "Wrong number of arguments".to_string(),
|
message: "Wrong number of arguments".to_string(),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
match self.current_excel_serial() {
|
let tz = match args.first() {
|
||||||
|
Some(arg0) => {
|
||||||
|
// Parse timezone argument
|
||||||
|
let tz_str = match self.get_string(arg0, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let tz: Tz = match tz_str.parse() {
|
||||||
|
Ok(tz) => tz,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: format!("Invalid timezone: {}", &tz_str),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
tz
|
||||||
|
}
|
||||||
|
None => self.tz,
|
||||||
|
};
|
||||||
|
match self.current_excel_serial_with_timezone(tz) {
|
||||||
Some(serial) => CalcResult::Number(serial),
|
Some(serial) => CalcResult::Number(serial),
|
||||||
None => CalcResult::Error {
|
None => CalcResult::Error {
|
||||||
error: Error::ERROR,
|
error: Error::ERROR,
|
||||||
@@ -1567,18 +1609,44 @@ impl Model {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
1 => {
|
1 => {
|
||||||
let year_days = if start_date.year() == end_date.year() {
|
// Procedure E
|
||||||
if (start_date.year() % 4 == 0 && start_date.year() % 100 != 0)
|
|
||||||
|| start_date.year() % 400 == 0
|
let start_year = start_date.year();
|
||||||
{
|
let end_year = end_date.year();
|
||||||
366.0
|
|
||||||
} else {
|
let step_a = start_year != end_year;
|
||||||
365.0
|
let step_b = start_year + 1 != end_year;
|
||||||
|
let step_c = start_date.month() < end_date.month();
|
||||||
|
let step_d = start_date.month() == end_date.month();
|
||||||
|
let step_e = start_date.day() <= end_date.day();
|
||||||
|
let step_f = step_a && (step_b || step_c || (step_d && step_e));
|
||||||
|
if step_f {
|
||||||
|
// 7.
|
||||||
|
// return average of days in year between start_year and end_year, inclusive
|
||||||
|
let mut total_days = 0;
|
||||||
|
for year in start_year..=end_year {
|
||||||
|
if is_leap_year(year) {
|
||||||
|
total_days += 366;
|
||||||
|
} else {
|
||||||
|
total_days += 365;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
days / (total_days as f64 / (end_year - start_year + 1) as f64)
|
||||||
|
} else if step_a && is_leap_year(start_year) {
|
||||||
|
// 8.
|
||||||
|
days / 366.0
|
||||||
|
} else if is_feb_29_between_dates(start_date, end_date) {
|
||||||
|
// 9. If a February 29 occurs between date1 and date2 then return 366
|
||||||
|
days / 366.0
|
||||||
|
} else if end_date.month() == 2 && end_date.day() == 29 {
|
||||||
|
// 10. If date2 is February 29 then return 366
|
||||||
|
days / 366.0
|
||||||
|
} else if !step_a && is_leap_year(start_year) {
|
||||||
|
days / 366.0
|
||||||
} else {
|
} else {
|
||||||
365.0
|
// 11.
|
||||||
};
|
days / 365.0
|
||||||
days / year_days
|
}
|
||||||
}
|
}
|
||||||
2 => days / 360.0,
|
2 => days / 360.0,
|
||||||
3 => days / 365.0,
|
3 => days / 365.0,
|
||||||
@@ -1595,6 +1663,34 @@ impl Model {
|
|||||||
}
|
}
|
||||||
_ => return CalcResult::new_error(Error::NUM, cell, "Invalid basis".to_string()),
|
_ => return CalcResult::new_error(Error::NUM, cell, "Invalid basis".to_string()),
|
||||||
};
|
};
|
||||||
CalcResult::Number(result)
|
CalcResult::Number(result.abs())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
#![allow(clippy::unwrap_used)]
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_is_leap_year() {
|
||||||
|
assert!(is_leap_year(2000));
|
||||||
|
assert!(!is_leap_year(1900));
|
||||||
|
assert!(is_leap_year(2004));
|
||||||
|
assert!(!is_leap_year(2001));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_is_feb_29_between_dates() {
|
||||||
|
let d1 = chrono::NaiveDate::from_ymd_opt(2020, 2, 28).unwrap();
|
||||||
|
let d2 = chrono::NaiveDate::from_ymd_opt(2020, 3, 1).unwrap();
|
||||||
|
assert!(is_feb_29_between_dates(d1, d2));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_is_feb_29_between_dates_false() {
|
||||||
|
let d1 = chrono::NaiveDate::from_ymd_opt(2021, 2, 28).unwrap();
|
||||||
|
let d2 = chrono::NaiveDate::from_ymd_opt(2021, 3, 1).unwrap();
|
||||||
|
assert!(!is_feb_29_between_dates(d1, d2));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,16 +1,18 @@
|
|||||||
|
use statrs::function::erf::{erf, erfc};
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
calc_result::CalcResult,
|
calc_result::CalcResult,
|
||||||
expressions::{parser::Node, token::Error, types::CellReferenceIndex},
|
expressions::{parser::Node, token::Error, types::CellReferenceIndex},
|
||||||
model::Model,
|
model::Model,
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::transcendental::{bessel_i, bessel_j, bessel_k, bessel_y, erf};
|
use super::transcendental::{bessel_i, bessel_j, bessel_k, bessel_y};
|
||||||
// https://root.cern/doc/v610/TMath_8cxx_source.html
|
// https://root.cern/doc/v610/TMath_8cxx_source.html
|
||||||
|
|
||||||
// Notice that the parameters for Bessel functions in Excel and here have inverted order
|
// Notice that the parameters for Bessel functions in Excel and here have inverted order
|
||||||
// EXCEL_BESSEL(x, n) => bessel(n, x)
|
// EXCEL_BESSEL(x, n) => bessel(n, x)
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
pub(crate) fn fn_besseli(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_besseli(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.len() != 2 {
|
if args.len() != 2 {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
@@ -160,7 +162,7 @@ impl Model {
|
|||||||
Ok(f) => f,
|
Ok(f) => f,
|
||||||
Err(s) => return s,
|
Err(s) => return s,
|
||||||
};
|
};
|
||||||
CalcResult::Number(1.0 - erf(x))
|
CalcResult::Number(erfc(x))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn fn_erfcprecise(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_erfcprecise(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
@@ -171,6 +173,6 @@ impl Model {
|
|||||||
Ok(f) => f,
|
Ok(f) => f,
|
||||||
Err(s) => return s,
|
Err(s) => return s,
|
||||||
};
|
};
|
||||||
CalcResult::Number(1.0 - erf(x))
|
CalcResult::Number(erfc(x))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ use crate::{
|
|||||||
// 2^48-1
|
// 2^48-1
|
||||||
const MAX: f64 = 281474976710655.0;
|
const MAX: f64 = 281474976710655.0;
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
// BITAND( number1, number2)
|
// BITAND( number1, number2)
|
||||||
pub(crate) fn fn_bitand(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_bitand(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.len() != 2 {
|
if args.len() != 2 {
|
||||||
|
|||||||
@@ -182,7 +182,7 @@ fn parse_complex_number(s: &str) -> Result<(f64, f64, Suffix), String> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
fn get_complex_number(
|
fn get_complex_number(
|
||||||
&mut self,
|
&mut self,
|
||||||
node: &Node,
|
node: &Node,
|
||||||
|
|||||||
@@ -41,7 +41,7 @@ fn convert_temperature(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
// CONVERT(number, from_unit, to_unit)
|
// CONVERT(number, from_unit, to_unit)
|
||||||
pub(crate) fn fn_convert(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_convert(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.len() != 3 {
|
if args.len() != 3 {
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ use crate::{
|
|||||||
number_format::to_precision,
|
number_format::to_precision,
|
||||||
};
|
};
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
// DELTA(number1, [number2])
|
// DELTA(number1, [number2])
|
||||||
pub(crate) fn fn_delta(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_delta(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
let arg_count = args.len();
|
let arg_count = args.len();
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ fn from_binary_to_decimal(value: f64) -> Result<i64, String> {
|
|||||||
Ok(result)
|
Ok(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
// BIN2DEC(number)
|
// BIN2DEC(number)
|
||||||
pub(crate) fn fn_bin2dec(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_bin2dec(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.len() != 1 {
|
if args.len() != 1 {
|
||||||
|
|||||||
@@ -1,53 +0,0 @@
|
|||||||
pub(crate) fn erf(x: f64) -> f64 {
|
|
||||||
let cof = vec![
|
|
||||||
-1.3026537197817094,
|
|
||||||
6.419_697_923_564_902e-1,
|
|
||||||
1.9476473204185836e-2,
|
|
||||||
-9.561_514_786_808_63e-3,
|
|
||||||
-9.46595344482036e-4,
|
|
||||||
3.66839497852761e-4,
|
|
||||||
4.2523324806907e-5,
|
|
||||||
-2.0278578112534e-5,
|
|
||||||
-1.624290004647e-6,
|
|
||||||
1.303655835580e-6,
|
|
||||||
1.5626441722e-8,
|
|
||||||
-8.5238095915e-8,
|
|
||||||
6.529054439e-9,
|
|
||||||
5.059343495e-9,
|
|
||||||
-9.91364156e-10,
|
|
||||||
-2.27365122e-10,
|
|
||||||
9.6467911e-11,
|
|
||||||
2.394038e-12,
|
|
||||||
-6.886027e-12,
|
|
||||||
8.94487e-13,
|
|
||||||
3.13092e-13,
|
|
||||||
-1.12708e-13,
|
|
||||||
3.81e-16,
|
|
||||||
7.106e-15,
|
|
||||||
-1.523e-15,
|
|
||||||
-9.4e-17,
|
|
||||||
1.21e-16,
|
|
||||||
-2.8e-17,
|
|
||||||
];
|
|
||||||
|
|
||||||
let mut d = 0.0;
|
|
||||||
let mut dd = 0.0;
|
|
||||||
|
|
||||||
let x_abs = x.abs();
|
|
||||||
|
|
||||||
let t = 2.0 / (2.0 + x_abs);
|
|
||||||
let ty = 4.0 * t - 2.0;
|
|
||||||
|
|
||||||
for j in (1..=cof.len() - 1).rev() {
|
|
||||||
let tmp = d;
|
|
||||||
d = ty * d - dd + cof[j];
|
|
||||||
dd = tmp;
|
|
||||||
}
|
|
||||||
|
|
||||||
let res = t * f64::exp(-x_abs * x_abs + 0.5 * (cof[0] + ty * d) - dd);
|
|
||||||
if x < 0.0 {
|
|
||||||
res - 1.0
|
|
||||||
} else {
|
|
||||||
1.0 - res
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -4,7 +4,6 @@ mod bessel_j1_y1;
|
|||||||
mod bessel_jn_yn;
|
mod bessel_jn_yn;
|
||||||
mod bessel_k;
|
mod bessel_k;
|
||||||
mod bessel_util;
|
mod bessel_util;
|
||||||
mod erf;
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test_bessel;
|
mod test_bessel;
|
||||||
@@ -13,4 +12,3 @@ pub(crate) use bessel_i::bessel_i;
|
|||||||
pub(crate) use bessel_jn_yn::jn as bessel_j;
|
pub(crate) use bessel_jn_yn::jn as bessel_j;
|
||||||
pub(crate) use bessel_jn_yn::yn as bessel_y;
|
pub(crate) use bessel_jn_yn::yn as bessel_y;
|
||||||
pub(crate) use bessel_k::bessel_k;
|
pub(crate) use bessel_k::bessel_k;
|
||||||
pub(crate) use erf::erf;
|
|
||||||
|
|||||||
@@ -191,7 +191,7 @@ fn compute_ppmt(
|
|||||||
// All, except for rate are easily solvable in terms of the others.
|
// All, except for rate are easily solvable in terms of the others.
|
||||||
// In these formulas the payment (pmt) is normally negative
|
// In these formulas the payment (pmt) is normally negative
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
fn get_array_of_numbers_generic(
|
fn get_array_of_numbers_generic(
|
||||||
&mut self,
|
&mut self,
|
||||||
arg: &Node,
|
arg: &Node,
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ use crate::{
|
|||||||
model::{Model, ParsedDefinedName},
|
model::{Model, ParsedDefinedName},
|
||||||
};
|
};
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
pub(crate) fn fn_isnumber(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_isnumber(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.len() == 1 {
|
if args.len() == 1 {
|
||||||
match self.evaluate_node_in_context(&args[0], cell) {
|
match self.evaluate_node_in_context(&args[0], cell) {
|
||||||
@@ -373,6 +373,8 @@ impl Model {
|
|||||||
CalcResult::Number(sheet_count)
|
CalcResult::Number(sheet_count)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// INFO(info_type, [reference])
|
||||||
|
/// NB: In Excel "info_type" is localized. Here it is always in English.
|
||||||
pub(crate) fn fn_cell(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_cell(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
let arg_count = args.len();
|
let arg_count = args.len();
|
||||||
if arg_count == 0 || arg_count > 2 {
|
if arg_count == 0 || arg_count > 2 {
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ use crate::{
|
|||||||
|
|
||||||
use super::util::compare_values;
|
use super::util::compare_values;
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
pub(crate) fn fn_true(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_true(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.is_empty() {
|
if args.is_empty() {
|
||||||
CalcResult::Boolean(true)
|
CalcResult::Boolean(true)
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ use crate::{
|
|||||||
|
|
||||||
use super::util::{compare_values, from_wildcard_to_regex, result_matches_regex, values_are_equal};
|
use super::util::{compare_values, from_wildcard_to_regex, result_matches_regex, values_are_equal};
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
pub(crate) fn fn_index(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_index(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
let row_num;
|
let row_num;
|
||||||
let col_num;
|
let col_num;
|
||||||
@@ -698,7 +698,7 @@ impl Model {
|
|||||||
let parsed_reference = ParsedReference::parse_reference_formula(
|
let parsed_reference = ParsedReference::parse_reference_formula(
|
||||||
Some(cell.sheet),
|
Some(cell.sheet),
|
||||||
&s,
|
&s,
|
||||||
&self.locale,
|
self.locale,
|
||||||
|name| self.get_sheet_index_by_name(name),
|
|name| self.get_sheet_index_by_name(name),
|
||||||
);
|
);
|
||||||
|
|
||||||
@@ -839,6 +839,10 @@ impl Model {
|
|||||||
CalcResult::Range { left, right }
|
CalcResult::Range { left, right }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// FORMULATEXT(reference)
|
||||||
|
// Returns a formula as a string. Two differences with Excel:
|
||||||
|
// - It returns the formula in English
|
||||||
|
// - It formats the formula without spaces between elements
|
||||||
pub(crate) fn fn_formulatext(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_formulatext(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.len() != 1 {
|
if args.len() != 1 {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
@@ -860,7 +864,7 @@ impl Model {
|
|||||||
message: "argument must be a reference to a single cell".to_string(),
|
message: "argument must be a reference to a single cell".to_string(),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
if let Ok(Some(f)) = self.get_cell_formula(left.sheet, left.row, left.column) {
|
if let Ok(Some(f)) = self.get_english_cell_formula(left.sheet, left.row, left.column) {
|
||||||
CalcResult::String(f)
|
CalcResult::String(f)
|
||||||
} else {
|
} else {
|
||||||
CalcResult::Error {
|
CalcResult::Error {
|
||||||
|
|||||||
@@ -68,14 +68,14 @@ macro_rules! single_number_fn {
|
|||||||
},
|
},
|
||||||
// If String, parse to f64 then apply or #VALUE! error
|
// If String, parse to f64 then apply or #VALUE! error
|
||||||
ArrayNode::String(s) => {
|
ArrayNode::String(s) => {
|
||||||
let node = match s.parse::<f64>() {
|
let node = match self.cast_number(&s) {
|
||||||
Ok(f) => match $op(f) {
|
Some(f) => match $op(f) {
|
||||||
Ok(x) => ArrayNode::Number(x),
|
Ok(x) => ArrayNode::Number(x),
|
||||||
Err(Error::DIV) => ArrayNode::Error(Error::DIV),
|
Err(Error::DIV) => ArrayNode::Error(Error::DIV),
|
||||||
Err(Error::VALUE) => ArrayNode::Error(Error::VALUE),
|
Err(Error::VALUE) => ArrayNode::Error(Error::VALUE),
|
||||||
Err(e) => ArrayNode::Error(e),
|
Err(e) => ArrayNode::Error(e),
|
||||||
},
|
},
|
||||||
Err(_) => ArrayNode::Error(Error::VALUE),
|
None => ArrayNode::Error(Error::VALUE),
|
||||||
};
|
};
|
||||||
data_row.push(node);
|
data_row.push(node);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
use crate::cast::NumberOrArray;
|
use crate::cast::NumberOrArray;
|
||||||
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
use crate::constants::{EXCEL_PRECISION, LAST_COLUMN, LAST_ROW};
|
||||||
use crate::expressions::parser::ArrayNode;
|
use crate::expressions::parser::ArrayNode;
|
||||||
use crate::expressions::types::CellReferenceIndex;
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
use crate::functions::math_util::{from_roman, to_roman_with_form};
|
use crate::functions::math_util::{from_roman, to_roman_with_form};
|
||||||
use crate::number_format::to_precision;
|
use crate::number_format::{to_excel_precision, to_precision};
|
||||||
use crate::single_number_fn;
|
use crate::single_number_fn;
|
||||||
use crate::{
|
use crate::{
|
||||||
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
@@ -47,7 +47,7 @@ pub fn random() -> f64 {
|
|||||||
Math::random()
|
Math::random()
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
pub(crate) fn fn_min(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_min(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
let mut result = f64::NAN;
|
let mut result = f64::NAN;
|
||||||
for arg in args {
|
for arg in args {
|
||||||
@@ -934,11 +934,11 @@ impl Model {
|
|||||||
if args.len() != 2 {
|
if args.len() != 2 {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
}
|
}
|
||||||
let value = match self.get_number(&args[0], cell) {
|
let value = match self.get_number_no_bools(&args[0], cell) {
|
||||||
Ok(f) => f,
|
Ok(f) => f,
|
||||||
Err(s) => return s,
|
Err(s) => return s,
|
||||||
};
|
};
|
||||||
let divisor = match self.get_number(&args[1], cell) {
|
let divisor = match self.get_number_no_bools(&args[1], cell) {
|
||||||
Ok(f) => f,
|
Ok(f) => f,
|
||||||
Err(s) => return s,
|
Err(s) => return s,
|
||||||
};
|
};
|
||||||
@@ -984,7 +984,9 @@ impl Model {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
let result = f64::floor(value / significance) * significance;
|
// Apply Excel precision to the ratio to handle floating-point rounding errors
|
||||||
|
let ratio = to_excel_precision(value / significance, EXCEL_PRECISION);
|
||||||
|
let result = f64::floor(ratio) * significance;
|
||||||
CalcResult::Number(result)
|
CalcResult::Number(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1022,7 +1024,7 @@ impl Model {
|
|||||||
cell: CellReferenceIndex,
|
cell: CellReferenceIndex,
|
||||||
) -> CalcResult {
|
) -> CalcResult {
|
||||||
let arg_count = args.len();
|
let arg_count = args.len();
|
||||||
if arg_count > 3 {
|
if !(1..=3).contains(&arg_count) {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
}
|
}
|
||||||
let value = match self.get_number(&args[0], cell) {
|
let value = match self.get_number(&args[0], cell) {
|
||||||
@@ -1063,7 +1065,7 @@ impl Model {
|
|||||||
cell: CellReferenceIndex,
|
cell: CellReferenceIndex,
|
||||||
) -> CalcResult {
|
) -> CalcResult {
|
||||||
let arg_count = args.len();
|
let arg_count = args.len();
|
||||||
if arg_count > 2 {
|
if !(1..=2).contains(&arg_count) {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
}
|
}
|
||||||
let value = match self.get_number(&args[0], cell) {
|
let value = match self.get_number(&args[0], cell) {
|
||||||
@@ -1093,7 +1095,7 @@ impl Model {
|
|||||||
|
|
||||||
pub(crate) fn fn_floor_math(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_floor_math(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
let arg_count = args.len();
|
let arg_count = args.len();
|
||||||
if arg_count > 3 {
|
if !(1..=3).contains(&arg_count) {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
}
|
}
|
||||||
let value = match self.get_number(&args[0], cell) {
|
let value = match self.get_number(&args[0], cell) {
|
||||||
@@ -1121,10 +1123,14 @@ impl Model {
|
|||||||
}
|
}
|
||||||
let significance = significance.abs();
|
let significance = significance.abs();
|
||||||
if value < 0.0 && mode != 0.0 {
|
if value < 0.0 && mode != 0.0 {
|
||||||
let result = f64::ceil(value / significance) * significance;
|
// Apply Excel precision to handle floating-point rounding errors
|
||||||
|
let ratio = to_excel_precision(value / significance, EXCEL_PRECISION);
|
||||||
|
let result = f64::ceil(ratio) * significance;
|
||||||
CalcResult::Number(result)
|
CalcResult::Number(result)
|
||||||
} else {
|
} else {
|
||||||
let result = f64::floor(value / significance) * significance;
|
// Apply Excel precision to handle floating-point rounding errors
|
||||||
|
let ratio = to_excel_precision(value / significance, EXCEL_PRECISION);
|
||||||
|
let result = f64::floor(ratio) * significance;
|
||||||
CalcResult::Number(result)
|
CalcResult::Number(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1135,7 +1141,7 @@ impl Model {
|
|||||||
cell: CellReferenceIndex,
|
cell: CellReferenceIndex,
|
||||||
) -> CalcResult {
|
) -> CalcResult {
|
||||||
let arg_count = args.len();
|
let arg_count = args.len();
|
||||||
if arg_count > 2 {
|
if !(1..=2).contains(&arg_count) {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
}
|
}
|
||||||
let value = match self.get_number(&args[0], cell) {
|
let value = match self.get_number(&args[0], cell) {
|
||||||
@@ -1154,7 +1160,9 @@ impl Model {
|
|||||||
return CalcResult::Number(0.0);
|
return CalcResult::Number(0.0);
|
||||||
}
|
}
|
||||||
|
|
||||||
let result = f64::floor(value / significance) * significance;
|
// Apply Excel precision to handle floating-point rounding errors
|
||||||
|
let ratio = to_excel_precision(value / significance, EXCEL_PRECISION);
|
||||||
|
let result = f64::floor(ratio) * significance;
|
||||||
CalcResult::Number(result)
|
CalcResult::Number(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1163,11 +1171,34 @@ impl Model {
|
|||||||
if args.len() != 2 {
|
if args.len() != 2 {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
}
|
}
|
||||||
let value = match self.get_number(&args[0], cell) {
|
|
||||||
|
let value = self.evaluate_node_in_context(&args[0], cell);
|
||||||
|
|
||||||
|
let multiple = self.evaluate_node_in_context(&args[1], cell);
|
||||||
|
|
||||||
|
// if either is empty => #N/A
|
||||||
|
if matches!(value, CalcResult::EmptyArg) || matches!(multiple, CalcResult::EmptyArg) {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NA,
|
||||||
|
origin: cell,
|
||||||
|
message: "Bad argument for MROUND".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Booleans are not cast
|
||||||
|
if matches!(value, CalcResult::Boolean(_)) {
|
||||||
|
return CalcResult::new_error(Error::VALUE, cell, "Expecting number".to_string());
|
||||||
|
}
|
||||||
|
|
||||||
|
if matches!(multiple, CalcResult::Boolean(_)) {
|
||||||
|
return CalcResult::new_error(Error::VALUE, cell, "Expecting number".to_string());
|
||||||
|
}
|
||||||
|
|
||||||
|
let value = match self.cast_to_number(value, cell) {
|
||||||
Ok(f) => f,
|
Ok(f) => f,
|
||||||
Err(s) => return s,
|
Err(s) => return s,
|
||||||
};
|
};
|
||||||
let multiple = match self.get_number(&args[1], cell) {
|
let multiple = match self.cast_to_number(multiple, cell) {
|
||||||
Ok(f) => f,
|
Ok(f) => f,
|
||||||
Err(s) => return s,
|
Err(s) => return s,
|
||||||
};
|
};
|
||||||
@@ -1186,7 +1217,7 @@ impl Model {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn fn_trunc(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_trunc(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
if args.len() > 2 {
|
if !(1..=2).contains(&args.len()) {
|
||||||
return CalcResult::new_args_number_error(cell);
|
return CalcResult::new_args_number_error(cell);
|
||||||
}
|
}
|
||||||
let value = match self.get_number(&args[0], cell) {
|
let value = match self.get_number(&args[0], cell) {
|
||||||
@@ -1210,11 +1241,15 @@ impl Model {
|
|||||||
if !(-15.0..=15.0).contains(&num_digits) {
|
if !(-15.0..=15.0).contains(&num_digits) {
|
||||||
return CalcResult::Number(value);
|
return CalcResult::Number(value);
|
||||||
}
|
}
|
||||||
CalcResult::Number(if value >= 0.0 {
|
let v = if value >= 0.0 {
|
||||||
f64::floor(value * 10f64.powf(num_digits)) / 10f64.powf(num_digits)
|
f64::floor(value * 10f64.powf(num_digits)) / 10f64.powf(num_digits)
|
||||||
} else {
|
} else {
|
||||||
f64::ceil(value * 10f64.powf(num_digits)) / 10f64.powf(num_digits)
|
f64::ceil(value * 10f64.powf(num_digits)) / 10f64.powf(num_digits)
|
||||||
})
|
};
|
||||||
|
if value.is_finite() && v.is_infinite() {
|
||||||
|
return CalcResult::Number(value);
|
||||||
|
}
|
||||||
|
CalcResult::Number(v)
|
||||||
}
|
}
|
||||||
|
|
||||||
single_number_fn!(fn_log10, |f| if f <= 0.0 {
|
single_number_fn!(fn_log10, |f| if f <= 0.0 {
|
||||||
@@ -1250,13 +1285,19 @@ impl Model {
|
|||||||
} else {
|
} else {
|
||||||
Ok((f * PI).sqrt())
|
Ok((f * PI).sqrt())
|
||||||
});
|
});
|
||||||
single_number_fn!(fn_acot, |f| if f == 0.0 {
|
|
||||||
Err(Error::DIV)
|
single_number_fn!(fn_acot, |f| {
|
||||||
} else {
|
let v = f64::atan(1.0 / f);
|
||||||
Ok(f64::atan(1.0 / f))
|
if f >= 0.0 {
|
||||||
|
Ok(v)
|
||||||
|
} else {
|
||||||
|
// To be compatible with Excel we need a different branch
|
||||||
|
// when f < 0
|
||||||
|
Ok(v + PI)
|
||||||
|
}
|
||||||
});
|
});
|
||||||
single_number_fn!(fn_acoth, |f: f64| if f.abs() == 1.0 {
|
single_number_fn!(fn_acoth, |f: f64| if f.abs() == 1.0 {
|
||||||
Err(Error::DIV)
|
Err(Error::NUM)
|
||||||
} else {
|
} else {
|
||||||
Ok(0.5 * (f64::ln((f + 1.0) / (f - 1.0))))
|
Ok(0.5 * (f64::ln((f + 1.0) / (f - 1.0))))
|
||||||
});
|
});
|
||||||
@@ -1265,8 +1306,11 @@ impl Model {
|
|||||||
} else {
|
} else {
|
||||||
Ok(f64::cos(f) / f64::sin(f))
|
Ok(f64::cos(f) / f64::sin(f))
|
||||||
});
|
});
|
||||||
single_number_fn!(fn_coth, |f| if f == 0.0 {
|
single_number_fn!(fn_coth, |f: f64| if f == 0.0 {
|
||||||
Err(Error::DIV)
|
Err(Error::DIV)
|
||||||
|
} else if f.abs() > 20.0 {
|
||||||
|
// for values > 20.0 this is exact in f64
|
||||||
|
Ok(f.signum())
|
||||||
} else {
|
} else {
|
||||||
Ok(f64::cosh(f) / f64::sinh(f))
|
Ok(f64::cosh(f) / f64::sinh(f))
|
||||||
});
|
});
|
||||||
@@ -1280,16 +1324,8 @@ impl Model {
|
|||||||
} else {
|
} else {
|
||||||
Ok(1.0 / f64::sinh(f))
|
Ok(1.0 / f64::sinh(f))
|
||||||
});
|
});
|
||||||
single_number_fn!(fn_sec, |f| if f == 0.0 {
|
single_number_fn!(fn_sec, |f| Ok(1.0 / f64::cos(f)));
|
||||||
Err(Error::DIV)
|
single_number_fn!(fn_sech, |f| Ok(1.0 / f64::cosh(f)));
|
||||||
} else {
|
|
||||||
Ok(1.0 / f64::cos(f))
|
|
||||||
});
|
|
||||||
single_number_fn!(fn_sech, |f| if f == 0.0 {
|
|
||||||
Err(Error::DIV)
|
|
||||||
} else {
|
|
||||||
Ok(1.0 / f64::cosh(f))
|
|
||||||
});
|
|
||||||
single_number_fn!(fn_exp, |f: f64| Ok(f64::exp(f)));
|
single_number_fn!(fn_exp, |f: f64| Ok(f64::exp(f)));
|
||||||
single_number_fn!(fn_fact, |x: f64| {
|
single_number_fn!(fn_fact, |x: f64| {
|
||||||
let x = x.floor();
|
let x = x.floor();
|
||||||
@@ -1320,7 +1356,13 @@ impl Model {
|
|||||||
}
|
}
|
||||||
Ok(acc)
|
Ok(acc)
|
||||||
});
|
});
|
||||||
single_number_fn!(fn_sign, |f| Ok(f64::signum(f)));
|
single_number_fn!(fn_sign, |f| {
|
||||||
|
if f == 0.0 {
|
||||||
|
Ok(0.0)
|
||||||
|
} else {
|
||||||
|
Ok(f64::signum(f))
|
||||||
|
}
|
||||||
|
});
|
||||||
single_number_fn!(fn_degrees, |f| Ok(f * (180.0 / PI)));
|
single_number_fn!(fn_degrees, |f| Ok(f * (180.0 / PI)));
|
||||||
single_number_fn!(fn_radians, |f| Ok(f * (PI / 180.0)));
|
single_number_fn!(fn_radians, |f| Ok(f * (PI / 180.0)));
|
||||||
single_number_fn!(fn_odd, |f| {
|
single_number_fn!(fn_odd, |f| {
|
||||||
|
|||||||
230
base/src/functions/mathematical_sum.rs
Normal file
230
base/src/functions/mathematical_sum.rs
Normal file
@@ -0,0 +1,230 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
type TwoMatricesResult = (i32, i32, Vec<Option<f64>>, Vec<Option<f64>>);
|
||||||
|
|
||||||
|
// Helper to check if two shapes are the same or compatible 1D shapes
|
||||||
|
fn is_same_shape_or_1d(rows1: i32, cols1: i32, rows2: i32, cols2: i32) -> bool {
|
||||||
|
(rows1 == rows2 && cols1 == cols2)
|
||||||
|
|| (rows1 == 1 && cols2 == 1 && cols1 == rows2)
|
||||||
|
|| (rows2 == 1 && cols1 == 1 && cols2 == rows1)
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// SUMX2MY2(array_x, array_y) - Returns the sum of the difference of squares
|
||||||
|
pub(crate) fn fn_sumx2my2(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let result = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(s) => return s,
|
||||||
|
};
|
||||||
|
|
||||||
|
let (_, _, values_left, values_right) = result;
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
for (x_opt, y_opt) in values_left.into_iter().zip(values_right.into_iter()) {
|
||||||
|
let x = x_opt.unwrap_or(0.0);
|
||||||
|
let y = y_opt.unwrap_or(0.0);
|
||||||
|
sum += x * x - y * y;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(sum)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SUMX2PY2(array_x, array_y) - Returns the sum of the sum of squares
|
||||||
|
pub(crate) fn fn_sumx2py2(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let result = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(s) => return s,
|
||||||
|
};
|
||||||
|
|
||||||
|
let (_rows, _cols, values_left, values_right) = result;
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
for (x_opt, y_opt) in values_left.into_iter().zip(values_right.into_iter()) {
|
||||||
|
let x = x_opt.unwrap_or(0.0);
|
||||||
|
let y = y_opt.unwrap_or(0.0);
|
||||||
|
sum += x * x + y * y;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(sum)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SUMXMY2(array_x, array_y) - Returns the sum of squares of differences
|
||||||
|
pub(crate) fn fn_sumxmy2(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let result = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(s) => return s,
|
||||||
|
};
|
||||||
|
|
||||||
|
let (_, _, values_left, values_right) = result;
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
for (x_opt, y_opt) in values_left.into_iter().zip(values_right.into_iter()) {
|
||||||
|
let x = x_opt.unwrap_or(0.0);
|
||||||
|
let y = y_opt.unwrap_or(0.0);
|
||||||
|
let diff = x - y;
|
||||||
|
sum += diff * diff;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(sum)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper function to extract and validate two matrices (ranges or arrays) with compatible shapes.
|
||||||
|
// Returns (rows, cols, values_left, values_right) or an error.
|
||||||
|
pub(crate) fn fn_get_two_matrices(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> Result<TwoMatricesResult, CalcResult> {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return Err(CalcResult::new_args_number_error(cell));
|
||||||
|
}
|
||||||
|
let x_range = self.evaluate_node_in_context(&args[0], cell);
|
||||||
|
let y_range = self.evaluate_node_in_context(&args[1], cell);
|
||||||
|
|
||||||
|
let result = match (x_range, y_range) {
|
||||||
|
(
|
||||||
|
CalcResult::Range {
|
||||||
|
left: l1,
|
||||||
|
right: r1,
|
||||||
|
},
|
||||||
|
CalcResult::Range {
|
||||||
|
left: l2,
|
||||||
|
right: r2,
|
||||||
|
},
|
||||||
|
) => {
|
||||||
|
if l1.sheet != l2.sheet {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let rows1 = r1.row - l1.row + 1;
|
||||||
|
let cols1 = r1.column - l1.column + 1;
|
||||||
|
let rows2 = r2.row - l2.row + 1;
|
||||||
|
let cols2 = r2.column - l2.column + 1;
|
||||||
|
if !is_same_shape_or_1d(rows1, cols1, rows2, cols2) {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges must be of the same shape".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let values_left = self.values_from_range(l1, r1)?;
|
||||||
|
let values_right = self.values_from_range(l2, r2)?;
|
||||||
|
(rows1, cols1, values_left, values_right)
|
||||||
|
}
|
||||||
|
(
|
||||||
|
CalcResult::Array(left),
|
||||||
|
CalcResult::Range {
|
||||||
|
left: l2,
|
||||||
|
right: r2,
|
||||||
|
},
|
||||||
|
) => {
|
||||||
|
let rows2 = r2.row - l2.row + 1;
|
||||||
|
let cols2 = r2.column - l2.column + 1;
|
||||||
|
|
||||||
|
let rows1 = left.len() as i32;
|
||||||
|
let cols1 = if rows1 > 0 { left[0].len() as i32 } else { 0 };
|
||||||
|
if !is_same_shape_or_1d(rows1, cols1, rows2, cols2) {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Array and range must be of the same shape".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let values_left = match self.values_from_array(left) {
|
||||||
|
Err(error) => {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in first array: {:?}", error),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
Ok(v) => v,
|
||||||
|
};
|
||||||
|
let values_right = self.values_from_range(l2, r2)?;
|
||||||
|
(rows2, cols2, values_left, values_right)
|
||||||
|
}
|
||||||
|
(
|
||||||
|
CalcResult::Range {
|
||||||
|
left: l1,
|
||||||
|
right: r1,
|
||||||
|
},
|
||||||
|
CalcResult::Array(right),
|
||||||
|
) => {
|
||||||
|
let rows1 = r1.row - l1.row + 1;
|
||||||
|
let cols1 = r1.column - l1.column + 1;
|
||||||
|
|
||||||
|
let rows2 = right.len() as i32;
|
||||||
|
let cols2 = if rows2 > 0 { right[0].len() as i32 } else { 0 };
|
||||||
|
if !is_same_shape_or_1d(rows1, cols1, rows2, cols2) {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Range and array must be of the same shape".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let values_left = self.values_from_range(l1, r1)?;
|
||||||
|
let values_right = match self.values_from_array(right) {
|
||||||
|
Err(error) => {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in second array: {:?}", error),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
Ok(v) => v,
|
||||||
|
};
|
||||||
|
(rows1, cols1, values_left, values_right)
|
||||||
|
}
|
||||||
|
(CalcResult::Array(left), CalcResult::Array(right)) => {
|
||||||
|
let rows1 = left.len() as i32;
|
||||||
|
let rows2 = right.len() as i32;
|
||||||
|
let cols1 = if rows1 > 0 { left[0].len() as i32 } else { 0 };
|
||||||
|
let cols2 = if rows2 > 0 { right[0].len() as i32 } else { 0 };
|
||||||
|
|
||||||
|
if !is_same_shape_or_1d(rows1, cols1, rows2, cols2) {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Arrays must be of the same shape".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let values_left = match self.values_from_array(left) {
|
||||||
|
Err(error) => {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in first array: {:?}", error),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
Ok(v) => v,
|
||||||
|
};
|
||||||
|
let values_right = match self.values_from_array(right) {
|
||||||
|
Err(error) => {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in second array: {:?}", error),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
Ok(v) => v,
|
||||||
|
};
|
||||||
|
(rows1, cols1, values_left, values_right)
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Both arguments must be ranges or arrays".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -1,733 +0,0 @@
|
|||||||
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
|
||||||
use crate::expressions::types::CellReferenceIndex;
|
|
||||||
use crate::{
|
|
||||||
calc_result::{CalcResult, Range},
|
|
||||||
expressions::parser::Node,
|
|
||||||
expressions::token::Error,
|
|
||||||
model::Model,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::util::build_criteria;
|
|
||||||
|
|
||||||
impl Model {
|
|
||||||
pub(crate) fn fn_average(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
if args.is_empty() {
|
|
||||||
return CalcResult::new_args_number_error(cell);
|
|
||||||
}
|
|
||||||
let mut count = 0.0;
|
|
||||||
let mut sum = 0.0;
|
|
||||||
for arg in args {
|
|
||||||
match self.evaluate_node_in_context(arg, cell) {
|
|
||||||
CalcResult::Number(value) => {
|
|
||||||
count += 1.0;
|
|
||||||
sum += value;
|
|
||||||
}
|
|
||||||
CalcResult::Boolean(b) => {
|
|
||||||
if let Node::ReferenceKind { .. } = arg {
|
|
||||||
} else {
|
|
||||||
sum += if b { 1.0 } else { 0.0 };
|
|
||||||
count += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::Range { left, right } => {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
for row in left.row..(right.row + 1) {
|
|
||||||
for column in left.column..(right.column + 1) {
|
|
||||||
match self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: left.sheet,
|
|
||||||
row,
|
|
||||||
column,
|
|
||||||
}) {
|
|
||||||
CalcResult::Number(value) => {
|
|
||||||
count += 1.0;
|
|
||||||
sum += value;
|
|
||||||
}
|
|
||||||
error @ CalcResult::Error { .. } => return error,
|
|
||||||
CalcResult::Range { .. } => {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::ERROR,
|
|
||||||
cell,
|
|
||||||
"Unexpected Range".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
error @ CalcResult::Error { .. } => return error,
|
|
||||||
CalcResult::String(s) => {
|
|
||||||
if let Node::ReferenceKind { .. } = arg {
|
|
||||||
// Do nothing
|
|
||||||
} else if let Ok(t) = s.parse::<f64>() {
|
|
||||||
sum += t;
|
|
||||||
count += 1.0;
|
|
||||||
} else {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::VALUE,
|
|
||||||
origin: cell,
|
|
||||||
message: "Argument cannot be cast into number".to_string(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
// Ignore everything else
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
if count == 0.0 {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::DIV,
|
|
||||||
origin: cell,
|
|
||||||
message: "Division by Zero".to_string(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
CalcResult::Number(sum / count)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_averagea(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
if args.is_empty() {
|
|
||||||
return CalcResult::new_args_number_error(cell);
|
|
||||||
}
|
|
||||||
let mut count = 0.0;
|
|
||||||
let mut sum = 0.0;
|
|
||||||
for arg in args {
|
|
||||||
match self.evaluate_node_in_context(arg, cell) {
|
|
||||||
CalcResult::Range { left, right } => {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
for row in left.row..(right.row + 1) {
|
|
||||||
for column in left.column..(right.column + 1) {
|
|
||||||
match self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: left.sheet,
|
|
||||||
row,
|
|
||||||
column,
|
|
||||||
}) {
|
|
||||||
CalcResult::String(_) => count += 1.0,
|
|
||||||
CalcResult::Number(value) => {
|
|
||||||
count += 1.0;
|
|
||||||
sum += value;
|
|
||||||
}
|
|
||||||
CalcResult::Boolean(b) => {
|
|
||||||
if b {
|
|
||||||
sum += 1.0;
|
|
||||||
}
|
|
||||||
count += 1.0;
|
|
||||||
}
|
|
||||||
error @ CalcResult::Error { .. } => return error,
|
|
||||||
CalcResult::Range { .. } => {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::ERROR,
|
|
||||||
cell,
|
|
||||||
"Unexpected Range".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
CalcResult::EmptyCell | CalcResult::EmptyArg => {}
|
|
||||||
CalcResult::Array(_) => {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::NIMPL,
|
|
||||||
origin: cell,
|
|
||||||
message: "Arrays not supported yet".to_string(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::Number(value) => {
|
|
||||||
count += 1.0;
|
|
||||||
sum += value;
|
|
||||||
}
|
|
||||||
CalcResult::String(s) => {
|
|
||||||
if let Node::ReferenceKind { .. } = arg {
|
|
||||||
// Do nothing
|
|
||||||
count += 1.0;
|
|
||||||
} else if let Ok(t) = s.parse::<f64>() {
|
|
||||||
sum += t;
|
|
||||||
count += 1.0;
|
|
||||||
} else {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::VALUE,
|
|
||||||
origin: cell,
|
|
||||||
message: "Argument cannot be cast into number".to_string(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::Boolean(b) => {
|
|
||||||
count += 1.0;
|
|
||||||
if b {
|
|
||||||
sum += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
error @ CalcResult::Error { .. } => return error,
|
|
||||||
CalcResult::EmptyCell | CalcResult::EmptyArg => {}
|
|
||||||
CalcResult::Array(_) => {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::NIMPL,
|
|
||||||
origin: cell,
|
|
||||||
message: "Arrays not supported yet".to_string(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
if count == 0.0 {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::DIV,
|
|
||||||
origin: cell,
|
|
||||||
message: "Division by Zero".to_string(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
CalcResult::Number(sum / count)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_count(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
if args.is_empty() {
|
|
||||||
return CalcResult::new_args_number_error(cell);
|
|
||||||
}
|
|
||||||
let mut result = 0.0;
|
|
||||||
for arg in args {
|
|
||||||
match self.evaluate_node_in_context(arg, cell) {
|
|
||||||
CalcResult::Number(_) => {
|
|
||||||
result += 1.0;
|
|
||||||
}
|
|
||||||
CalcResult::Boolean(_) => {
|
|
||||||
if !matches!(arg, Node::ReferenceKind { .. }) {
|
|
||||||
result += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::String(s) => {
|
|
||||||
if !matches!(arg, Node::ReferenceKind { .. }) && s.parse::<f64>().is_ok() {
|
|
||||||
result += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::Range { left, right } => {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
for row in left.row..(right.row + 1) {
|
|
||||||
for column in left.column..(right.column + 1) {
|
|
||||||
if let CalcResult::Number(_) = self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: left.sheet,
|
|
||||||
row,
|
|
||||||
column,
|
|
||||||
}) {
|
|
||||||
result += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
// Ignore everything else
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
CalcResult::Number(result)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_counta(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
if args.is_empty() {
|
|
||||||
return CalcResult::new_args_number_error(cell);
|
|
||||||
}
|
|
||||||
let mut result = 0.0;
|
|
||||||
for arg in args {
|
|
||||||
match self.evaluate_node_in_context(arg, cell) {
|
|
||||||
CalcResult::EmptyCell | CalcResult::EmptyArg => {}
|
|
||||||
CalcResult::Range { left, right } => {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
for row in left.row..(right.row + 1) {
|
|
||||||
for column in left.column..(right.column + 1) {
|
|
||||||
match self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: left.sheet,
|
|
||||||
row,
|
|
||||||
column,
|
|
||||||
}) {
|
|
||||||
CalcResult::EmptyCell | CalcResult::EmptyArg => {}
|
|
||||||
_ => {
|
|
||||||
result += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
result += 1.0;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
CalcResult::Number(result)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_countblank(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
// COUNTBLANK requires only one argument
|
|
||||||
if args.len() != 1 {
|
|
||||||
return CalcResult::new_args_number_error(cell);
|
|
||||||
}
|
|
||||||
let mut result = 0.0;
|
|
||||||
for arg in args {
|
|
||||||
match self.evaluate_node_in_context(arg, cell) {
|
|
||||||
CalcResult::EmptyCell | CalcResult::EmptyArg => result += 1.0,
|
|
||||||
CalcResult::String(s) => {
|
|
||||||
if s.is_empty() {
|
|
||||||
result += 1.0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::Range { left, right } => {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
for row in left.row..(right.row + 1) {
|
|
||||||
for column in left.column..(right.column + 1) {
|
|
||||||
match self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: left.sheet,
|
|
||||||
row,
|
|
||||||
column,
|
|
||||||
}) {
|
|
||||||
CalcResult::EmptyCell | CalcResult::EmptyArg => result += 1.0,
|
|
||||||
CalcResult::String(s) => {
|
|
||||||
if s.is_empty() {
|
|
||||||
result += 1.0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
CalcResult::Number(result)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_countif(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
if args.len() == 2 {
|
|
||||||
let arguments = vec![args[0].clone(), args[1].clone()];
|
|
||||||
self.fn_countifs(&arguments, cell)
|
|
||||||
} else {
|
|
||||||
CalcResult::new_args_number_error(cell)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// AVERAGEIF(criteria_range, criteria, [average_range])
|
|
||||||
/// if average_rage is missing then criteria_range will be used
|
|
||||||
pub(crate) fn fn_averageif(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
if args.len() == 2 {
|
|
||||||
let arguments = vec![args[0].clone(), args[0].clone(), args[1].clone()];
|
|
||||||
self.fn_averageifs(&arguments, cell)
|
|
||||||
} else if args.len() == 3 {
|
|
||||||
let arguments = vec![args[2].clone(), args[0].clone(), args[1].clone()];
|
|
||||||
self.fn_averageifs(&arguments, cell)
|
|
||||||
} else {
|
|
||||||
CalcResult::new_args_number_error(cell)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: This function shares a lot of code with apply_ifs. Can we merge them?
|
|
||||||
pub(crate) fn fn_countifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
let args_count = args.len();
|
|
||||||
if args_count < 2 || !args_count.is_multiple_of(2) {
|
|
||||||
return CalcResult::new_args_number_error(cell);
|
|
||||||
}
|
|
||||||
|
|
||||||
let case_count = args_count / 2;
|
|
||||||
// NB: this is a beautiful example of the borrow checker
|
|
||||||
// The order of these two definitions cannot be swapped.
|
|
||||||
let mut criteria = Vec::new();
|
|
||||||
let mut fn_criteria = Vec::new();
|
|
||||||
let ranges = &mut Vec::new();
|
|
||||||
for case_index in 0..case_count {
|
|
||||||
let criterion = self.evaluate_node_in_context(&args[case_index * 2 + 1], cell);
|
|
||||||
criteria.push(criterion);
|
|
||||||
// NB: We cannot do:
|
|
||||||
// fn_criteria.push(build_criteria(&criterion));
|
|
||||||
// because criterion doesn't live long enough
|
|
||||||
let result = self.evaluate_node_in_context(&args[case_index * 2], cell);
|
|
||||||
if result.is_error() {
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
if let CalcResult::Range { left, right } = result {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
// TODO test ranges are of the same size as sum_range
|
|
||||||
ranges.push(Range { left, right });
|
|
||||||
} else {
|
|
||||||
return CalcResult::new_error(Error::VALUE, cell, "Expected a range".to_string());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for criterion in criteria.iter() {
|
|
||||||
fn_criteria.push(build_criteria(criterion));
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut total = 0.0;
|
|
||||||
let first_range = &ranges[0];
|
|
||||||
let left_row = first_range.left.row;
|
|
||||||
let left_column = first_range.left.column;
|
|
||||||
let right_row = first_range.right.row;
|
|
||||||
let right_column = first_range.right.column;
|
|
||||||
|
|
||||||
let dimension = match self.workbook.worksheet(first_range.left.sheet) {
|
|
||||||
Ok(s) => s.dimension(),
|
|
||||||
Err(_) => {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::ERROR,
|
|
||||||
cell,
|
|
||||||
format!("Invalid worksheet index: '{}'", first_range.left.sheet),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let max_row = dimension.max_row;
|
|
||||||
let max_column = dimension.max_column;
|
|
||||||
|
|
||||||
let open_row = left_row == 1 && right_row == LAST_ROW;
|
|
||||||
let open_column = left_column == 1 && right_column == LAST_COLUMN;
|
|
||||||
|
|
||||||
for row in left_row..right_row + 1 {
|
|
||||||
if open_row && row > max_row {
|
|
||||||
// If the row is larger than the max row in the sheet then all cells are empty.
|
|
||||||
// We compute it only once
|
|
||||||
let mut is_true = true;
|
|
||||||
for fn_criterion in fn_criteria.iter() {
|
|
||||||
if !fn_criterion(&CalcResult::EmptyCell) {
|
|
||||||
is_true = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if is_true {
|
|
||||||
total += ((LAST_ROW - max_row) * (right_column - left_column + 1)) as f64;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
for column in left_column..right_column + 1 {
|
|
||||||
if open_column && column > max_column {
|
|
||||||
// If the column is larger than the max column in the sheet then all cells are empty.
|
|
||||||
// We compute it only once
|
|
||||||
let mut is_true = true;
|
|
||||||
for fn_criterion in fn_criteria.iter() {
|
|
||||||
if !fn_criterion(&CalcResult::EmptyCell) {
|
|
||||||
is_true = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if is_true {
|
|
||||||
total += (LAST_COLUMN - max_column) as f64;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
let mut is_true = true;
|
|
||||||
for case_index in 0..case_count {
|
|
||||||
// We check if value in range n meets criterion n
|
|
||||||
let range = &ranges[case_index];
|
|
||||||
let fn_criterion = &fn_criteria[case_index];
|
|
||||||
let value = self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: range.left.sheet,
|
|
||||||
row: range.left.row + row - first_range.left.row,
|
|
||||||
column: range.left.column + column - first_range.left.column,
|
|
||||||
});
|
|
||||||
if !fn_criterion(&value) {
|
|
||||||
is_true = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if is_true {
|
|
||||||
total += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::Number(total)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn apply_ifs<F>(
|
|
||||||
&mut self,
|
|
||||||
args: &[Node],
|
|
||||||
cell: CellReferenceIndex,
|
|
||||||
mut apply: F,
|
|
||||||
) -> Result<(), CalcResult>
|
|
||||||
where
|
|
||||||
F: FnMut(f64),
|
|
||||||
{
|
|
||||||
let args_count = args.len();
|
|
||||||
if args_count < 3 || args_count.is_multiple_of(2) {
|
|
||||||
return Err(CalcResult::new_args_number_error(cell));
|
|
||||||
}
|
|
||||||
let arg_0 = self.evaluate_node_in_context(&args[0], cell);
|
|
||||||
if arg_0.is_error() {
|
|
||||||
return Err(arg_0);
|
|
||||||
}
|
|
||||||
let sum_range = if let CalcResult::Range { left, right } = arg_0 {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return Err(CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
Range { left, right }
|
|
||||||
} else {
|
|
||||||
return Err(CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Expected a range".to_string(),
|
|
||||||
));
|
|
||||||
};
|
|
||||||
|
|
||||||
let case_count = (args_count - 1) / 2;
|
|
||||||
// NB: this is a beautiful example of the borrow checker
|
|
||||||
// The order of these two definitions cannot be swapped.
|
|
||||||
let mut criteria = Vec::new();
|
|
||||||
let mut fn_criteria = Vec::new();
|
|
||||||
let ranges = &mut Vec::new();
|
|
||||||
for case_index in 1..=case_count {
|
|
||||||
let criterion = self.evaluate_node_in_context(&args[case_index * 2], cell);
|
|
||||||
// NB: criterion might be an error. That's ok
|
|
||||||
criteria.push(criterion);
|
|
||||||
// NB: We cannot do:
|
|
||||||
// fn_criteria.push(build_criteria(&criterion));
|
|
||||||
// because criterion doesn't live long enough
|
|
||||||
let result = self.evaluate_node_in_context(&args[case_index * 2 - 1], cell);
|
|
||||||
if result.is_error() {
|
|
||||||
return Err(result);
|
|
||||||
}
|
|
||||||
if let CalcResult::Range { left, right } = result {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return Err(CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
// TODO test ranges are of the same size as sum_range
|
|
||||||
ranges.push(Range { left, right });
|
|
||||||
} else {
|
|
||||||
return Err(CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Expected a range".to_string(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for criterion in criteria.iter() {
|
|
||||||
fn_criteria.push(build_criteria(criterion));
|
|
||||||
}
|
|
||||||
|
|
||||||
let left_row = sum_range.left.row;
|
|
||||||
let left_column = sum_range.left.column;
|
|
||||||
let mut right_row = sum_range.right.row;
|
|
||||||
let mut right_column = sum_range.right.column;
|
|
||||||
|
|
||||||
if left_row == 1 && right_row == LAST_ROW {
|
|
||||||
right_row = match self.workbook.worksheet(sum_range.left.sheet) {
|
|
||||||
Ok(s) => s.dimension().max_row,
|
|
||||||
Err(_) => {
|
|
||||||
return Err(CalcResult::new_error(
|
|
||||||
Error::ERROR,
|
|
||||||
cell,
|
|
||||||
format!("Invalid worksheet index: '{}'", sum_range.left.sheet),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
if left_column == 1 && right_column == LAST_COLUMN {
|
|
||||||
right_column = match self.workbook.worksheet(sum_range.left.sheet) {
|
|
||||||
Ok(s) => s.dimension().max_column,
|
|
||||||
Err(_) => {
|
|
||||||
return Err(CalcResult::new_error(
|
|
||||||
Error::ERROR,
|
|
||||||
cell,
|
|
||||||
format!("Invalid worksheet index: '{}'", sum_range.left.sheet),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
for row in left_row..right_row + 1 {
|
|
||||||
for column in left_column..right_column + 1 {
|
|
||||||
let mut is_true = true;
|
|
||||||
for case_index in 0..case_count {
|
|
||||||
// We check if value in range n meets criterion n
|
|
||||||
let range = &ranges[case_index];
|
|
||||||
let fn_criterion = &fn_criteria[case_index];
|
|
||||||
let value = self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: range.left.sheet,
|
|
||||||
row: range.left.row + row - sum_range.left.row,
|
|
||||||
column: range.left.column + column - sum_range.left.column,
|
|
||||||
});
|
|
||||||
if !fn_criterion(&value) {
|
|
||||||
is_true = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if is_true {
|
|
||||||
let v = self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: sum_range.left.sheet,
|
|
||||||
row,
|
|
||||||
column,
|
|
||||||
});
|
|
||||||
match v {
|
|
||||||
CalcResult::Number(n) => apply(n),
|
|
||||||
CalcResult::Error { .. } => return Err(v),
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_averageifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
let mut total = 0.0;
|
|
||||||
let mut count = 0.0;
|
|
||||||
|
|
||||||
let average = |value: f64| {
|
|
||||||
total += value;
|
|
||||||
count += 1.0;
|
|
||||||
};
|
|
||||||
if let Err(e) = self.apply_ifs(args, cell, average) {
|
|
||||||
return e;
|
|
||||||
}
|
|
||||||
|
|
||||||
if count == 0.0 {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::DIV,
|
|
||||||
origin: cell,
|
|
||||||
message: "division by 0".to_string(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
CalcResult::Number(total / count)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_minifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
let mut min = f64::INFINITY;
|
|
||||||
let apply_min = |value: f64| min = value.min(min);
|
|
||||||
if let Err(e) = self.apply_ifs(args, cell, apply_min) {
|
|
||||||
return e;
|
|
||||||
}
|
|
||||||
|
|
||||||
if min.is_infinite() {
|
|
||||||
min = 0.0;
|
|
||||||
}
|
|
||||||
CalcResult::Number(min)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_maxifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
let mut max = -f64::INFINITY;
|
|
||||||
let apply_max = |value: f64| max = value.max(max);
|
|
||||||
if let Err(e) = self.apply_ifs(args, cell, apply_max) {
|
|
||||||
return e;
|
|
||||||
}
|
|
||||||
if max.is_infinite() {
|
|
||||||
max = 0.0;
|
|
||||||
}
|
|
||||||
CalcResult::Number(max)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn fn_geomean(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
|
||||||
if args.is_empty() {
|
|
||||||
return CalcResult::new_args_number_error(cell);
|
|
||||||
}
|
|
||||||
let mut count = 0.0;
|
|
||||||
let mut product = 1.0;
|
|
||||||
for arg in args {
|
|
||||||
match self.evaluate_node_in_context(arg, cell) {
|
|
||||||
CalcResult::Number(value) => {
|
|
||||||
count += 1.0;
|
|
||||||
product *= value;
|
|
||||||
}
|
|
||||||
CalcResult::Boolean(b) => {
|
|
||||||
if let Node::ReferenceKind { .. } = arg {
|
|
||||||
} else {
|
|
||||||
product *= if b { 1.0 } else { 0.0 };
|
|
||||||
count += 1.0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CalcResult::Range { left, right } => {
|
|
||||||
if left.sheet != right.sheet {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::VALUE,
|
|
||||||
cell,
|
|
||||||
"Ranges are in different sheets".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
for row in left.row..(right.row + 1) {
|
|
||||||
for column in left.column..(right.column + 1) {
|
|
||||||
match self.evaluate_cell(CellReferenceIndex {
|
|
||||||
sheet: left.sheet,
|
|
||||||
row,
|
|
||||||
column,
|
|
||||||
}) {
|
|
||||||
CalcResult::Number(value) => {
|
|
||||||
count += 1.0;
|
|
||||||
product *= value;
|
|
||||||
}
|
|
||||||
error @ CalcResult::Error { .. } => return error,
|
|
||||||
CalcResult::Range { .. } => {
|
|
||||||
return CalcResult::new_error(
|
|
||||||
Error::ERROR,
|
|
||||||
cell,
|
|
||||||
"Unexpected Range".to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
error @ CalcResult::Error { .. } => return error,
|
|
||||||
CalcResult::String(s) => {
|
|
||||||
if let Node::ReferenceKind { .. } = arg {
|
|
||||||
// Do nothing
|
|
||||||
} else if let Ok(t) = s.parse::<f64>() {
|
|
||||||
product *= t;
|
|
||||||
count += 1.0;
|
|
||||||
} else {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::VALUE,
|
|
||||||
origin: cell,
|
|
||||||
message: "Argument cannot be cast into number".to_string(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
// Ignore everything else
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
if count == 0.0 {
|
|
||||||
return CalcResult::Error {
|
|
||||||
error: Error::DIV,
|
|
||||||
origin: cell,
|
|
||||||
message: "Division by Zero".to_string(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
CalcResult::Number(product.powf(1.0 / count))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
213
base/src/functions/statistical/beta.rs
Normal file
213
base/src/functions/statistical/beta.rs
Normal file
@@ -0,0 +1,213 @@
|
|||||||
|
use statrs::distribution::{Beta, Continuous, ContinuousCDF};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// BETA.DIST(x, alpha, beta, cumulative, [A], [B])
|
||||||
|
pub(crate) fn fn_beta_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let arg_count = args.len();
|
||||||
|
if !(4..=6).contains(&arg_count) {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let alpha = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let beta_param = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// cumulative argument: interpret like Excel
|
||||||
|
let cumulative = match self.evaluate_node_in_context(&args[3], cell) {
|
||||||
|
CalcResult::Boolean(b) => b,
|
||||||
|
CalcResult::Number(n) => n != 0.0,
|
||||||
|
CalcResult::String(s) => {
|
||||||
|
let up = s.to_ascii_uppercase();
|
||||||
|
if up == "TRUE" {
|
||||||
|
true
|
||||||
|
} else if up == "FALSE" {
|
||||||
|
false
|
||||||
|
} else {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "cumulative must be TRUE/FALSE or numeric".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid cumulative argument".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Optional A, B
|
||||||
|
let a = if arg_count >= 5 {
|
||||||
|
match self.get_number_no_bools(&args[4], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
0.0
|
||||||
|
};
|
||||||
|
|
||||||
|
let b = if arg_count >= 6 {
|
||||||
|
match self.get_number_no_bools(&args[5], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
1.0
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel: alpha <= 0 or beta <= 0 → #NUM!
|
||||||
|
if alpha <= 0.0 || beta_param <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"alpha and beta must be > 0 in BETA.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Excel: if x < A, x > B, or A = B → #NUM!
|
||||||
|
if b == a || x < a || x > b {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"x must be between A and B and A < B in BETA.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Transform to standard Beta(0,1)
|
||||||
|
let width = b - a;
|
||||||
|
let t = (x - a) / width;
|
||||||
|
|
||||||
|
let dist = match Beta::new(alpha, beta_param) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Beta distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative {
|
||||||
|
dist.cdf(t)
|
||||||
|
} else {
|
||||||
|
// general-interval beta pdf: f_X(x) = f_T(t) / (B - A), t=(x-A)/(B-A)
|
||||||
|
dist.pdf(t) / width
|
||||||
|
};
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for BETA.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_beta_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let arg_count = args.len();
|
||||||
|
if !(3..=5).contains(&arg_count) {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let alpha = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let beta_param = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let a = if arg_count >= 4 {
|
||||||
|
match self.get_number_no_bools(&args[3], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
0.0
|
||||||
|
};
|
||||||
|
|
||||||
|
let b = if arg_count >= 5 {
|
||||||
|
match self.get_number_no_bools(&args[4], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
1.0
|
||||||
|
};
|
||||||
|
|
||||||
|
if alpha <= 0.0 || beta_param <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"alpha and beta must be > 0 in BETA.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// probability <= 0 or probability > 1 → #NUM!
|
||||||
|
if p <= 0.0 || p > 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"probability must be in (0,1] in BETA.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if b <= a {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"A must be < B in BETA.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match Beta::new(alpha, beta_param) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Beta distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let t = dist.inverse_cdf(p);
|
||||||
|
if t.is_nan() || t.is_infinite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for BETA.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Map back from [0,1] to [A,B]
|
||||||
|
let x = a + t * (b - a);
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
}
|
||||||
311
base/src/functions/statistical/binom.rs
Normal file
311
base/src/functions/statistical/binom.rs
Normal file
@@ -0,0 +1,311 @@
|
|||||||
|
use statrs::distribution::{Binomial, Discrete, DiscreteCDF};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_binom_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
// number_s
|
||||||
|
let number_s = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// trials
|
||||||
|
let trials = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// probability_s
|
||||||
|
let p = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// cumulative (logical)
|
||||||
|
let cumulative = match self.get_boolean(&args[3], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Domain checks
|
||||||
|
if trials < 0.0
|
||||||
|
|| number_s < 0.0
|
||||||
|
|| number_s > trials
|
||||||
|
|| p.is_nan()
|
||||||
|
|| !(0.0..=1.0).contains(&p)
|
||||||
|
{
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for BINOM.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Limit to u64
|
||||||
|
if trials > u64::MAX as f64 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Number of trials too large".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = trials as u64;
|
||||||
|
let k = number_s as u64;
|
||||||
|
|
||||||
|
let dist = match Binomial::new(p, n) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for binomial distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let prob = if cumulative { dist.cdf(k) } else { dist.pmf(k) };
|
||||||
|
|
||||||
|
if prob.is_nan() || prob.is_infinite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for BINOM.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(prob)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_binom_dist_range(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() < 3 || args.len() > 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
// trials
|
||||||
|
let trials = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// probability_s
|
||||||
|
let p = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// number_s (lower)
|
||||||
|
let number_s = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// number_s2 (upper, optional)
|
||||||
|
let number_s2 = if args.len() == 4 {
|
||||||
|
match self.get_number_no_bools(&args[3], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
number_s
|
||||||
|
};
|
||||||
|
|
||||||
|
if trials < 0.0
|
||||||
|
|| number_s < 0.0
|
||||||
|
|| number_s2 < 0.0
|
||||||
|
|| number_s > number_s2
|
||||||
|
|| number_s2 > trials
|
||||||
|
|| p.is_nan()
|
||||||
|
|| !(0.0..=1.0).contains(&p)
|
||||||
|
{
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for BINOM.DIST.RANGE".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if trials > u64::MAX as f64 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Number of trials too large".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = trials as u64;
|
||||||
|
let lower = number_s as u64;
|
||||||
|
let upper = number_s2 as u64;
|
||||||
|
|
||||||
|
let dist = match Binomial::new(p, n) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for binomial distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let prob = if lower == 0 {
|
||||||
|
dist.cdf(upper)
|
||||||
|
} else {
|
||||||
|
let cdf_upper = dist.cdf(upper);
|
||||||
|
let cdf_below_lower = dist.cdf(lower - 1);
|
||||||
|
cdf_upper - cdf_below_lower
|
||||||
|
};
|
||||||
|
|
||||||
|
if prob.is_nan() || prob.is_infinite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for BINOM.DIST.RANGE".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(prob)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_binom_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
// trials
|
||||||
|
let trials = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// probability_s
|
||||||
|
let p = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// alpha
|
||||||
|
let alpha = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if trials < 0.0
|
||||||
|
|| trials > u64::MAX as f64
|
||||||
|
|| p.is_nan()
|
||||||
|
|| !(0.0..=1.0).contains(&p)
|
||||||
|
|| alpha.is_nan()
|
||||||
|
|| !(0.0..=1.0).contains(&alpha)
|
||||||
|
{
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for BINOM.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = trials as u64;
|
||||||
|
|
||||||
|
let dist = match Binomial::new(p, n) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for binomial distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// DiscreteCDF::inverse_cdf returns u64 for binomial
|
||||||
|
let k = statrs::distribution::DiscreteCDF::inverse_cdf(&dist, alpha);
|
||||||
|
|
||||||
|
CalcResult::Number(k as f64)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_negbinom_dist(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
use statrs::distribution::{Discrete, DiscreteCDF, NegativeBinomial};
|
||||||
|
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let number_f = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let number_s = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let probability_s = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let cumulative = match self.get_boolean(&args[3], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if number_f < 0.0 || number_s < 1.0 || !(0.0..=1.0).contains(&probability_s) {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for NEGBINOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Guard against absurdly large failures that won't fit in u64
|
||||||
|
if number_f > (u64::MAX as f64) {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for NEGBINOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match NegativeBinomial::new(number_s, probability_s) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for NEGBINOM.DIST".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let f_u = number_f as u64;
|
||||||
|
let result = if cumulative {
|
||||||
|
dist.cdf(f_u)
|
||||||
|
} else {
|
||||||
|
dist.pmf(f_u)
|
||||||
|
};
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for NEGBINOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
397
base/src/functions/statistical/chisq.rs
Normal file
397
base/src/functions/statistical/chisq.rs
Normal file
@@ -0,0 +1,397 @@
|
|||||||
|
use statrs::distribution::{ChiSquared, Continuous, ContinuousCDF};
|
||||||
|
|
||||||
|
use crate::expressions::parser::ArrayNode;
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// CHISQ.DIST(x, deg_freedom, cumulative)
|
||||||
|
pub(crate) fn fn_chisq_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[2], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"x must be >= 0 in CHISQ.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if df < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in CHISQ.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match ChiSquared::new(df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Chi-squared distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(x) } else { dist.pdf(x) };
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for CHISQ.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CHISQ.DIST.RT(x, deg_freedom)
|
||||||
|
pub(crate) fn fn_chisq_dist_rt(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df_raw = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = df_raw.trunc();
|
||||||
|
|
||||||
|
if x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"x must be >= 0 in CHISQ.DIST.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if df < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in CHISQ.DIST.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match ChiSquared::new(df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Chi-squared distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Right-tail probability: P(X > x).
|
||||||
|
// Use sf(x) directly for better numerical properties than 1 - cdf(x).
|
||||||
|
let result = dist.sf(x);
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() || result < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for CHISQ.DIST.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CHISQ.INV(probability, deg_freedom)
|
||||||
|
pub(crate) fn fn_chisq_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// if probability < 0 or > 1 → #NUM!
|
||||||
|
if !(0.0..=1.0).contains(&p) {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"probability must be in [0,1] in CHISQ.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if df < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in CHISQ.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match ChiSquared::new(df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Chi-squared distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let x = dist.inverse_cdf(p);
|
||||||
|
|
||||||
|
if x.is_nan() || x.is_infinite() || x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for CHISQ.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CHISQ.INV.RT(probability, deg_freedom)
|
||||||
|
pub(crate) fn fn_chisq_inv_rt(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df_raw = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = df_raw.trunc();
|
||||||
|
|
||||||
|
// if probability < 0 or > 1 → #NUM!
|
||||||
|
if !(0.0..=1.0).contains(&p) {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"probability must be in [0,1] in CHISQ.INV.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if df < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in CHISQ.INV.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match ChiSquared::new(df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Chi-squared distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Right-tail inverse: p = P(X > x) = SF(x) = 1 - CDF(x)
|
||||||
|
// So x = inverse_cdf(1 - p).
|
||||||
|
let x = dist.inverse_cdf(1.0 - p);
|
||||||
|
|
||||||
|
if x.is_nan() || x.is_infinite() || x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for CHISQ.INV.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn values_from_range(
|
||||||
|
&mut self,
|
||||||
|
left: CellReferenceIndex,
|
||||||
|
right: CellReferenceIndex,
|
||||||
|
) -> Result<Vec<Option<f64>>, CalcResult> {
|
||||||
|
let mut values = Vec::new();
|
||||||
|
for row_offset in 0..=(right.row - left.row) {
|
||||||
|
for col_offset in 0..=(right.column - left.column) {
|
||||||
|
let cell_ref = CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row: left.row + row_offset,
|
||||||
|
column: left.column + col_offset,
|
||||||
|
};
|
||||||
|
let cell_value = self.evaluate_cell(cell_ref);
|
||||||
|
match cell_value {
|
||||||
|
CalcResult::Number(v) => {
|
||||||
|
values.push(Some(v));
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return Err(error),
|
||||||
|
_ => {
|
||||||
|
values.push(None);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(values)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn values_from_array(
|
||||||
|
&mut self,
|
||||||
|
array: Vec<Vec<ArrayNode>>,
|
||||||
|
) -> Result<Vec<Option<f64>>, Error> {
|
||||||
|
let mut values = Vec::new();
|
||||||
|
for row in array {
|
||||||
|
for item in row {
|
||||||
|
match item {
|
||||||
|
ArrayNode::Number(f) => {
|
||||||
|
values.push(Some(f));
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return Err(error);
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
values.push(None);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(values)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CHISQ.TEST(actual_range, expected_range)
|
||||||
|
pub(crate) fn fn_chisq_test(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (width, height, values_left, values_right) = match self.fn_get_two_matrices(args, cell)
|
||||||
|
{
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(r) => return r,
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut values = Vec::with_capacity(values_left.len());
|
||||||
|
|
||||||
|
// Now we have:
|
||||||
|
// - values: flattened (observed, expected)
|
||||||
|
// - width, height: shape
|
||||||
|
for i in 0..values_left.len() {
|
||||||
|
match (values_left[i], values_right[i]) {
|
||||||
|
(Some(v1), Some(v2)) => {
|
||||||
|
values.push((v1, v2));
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
values.push((1.0, 1.0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if width == 0 || height == 0 || values.len() < 2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"CHISQ.TEST requires at least two data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut chi2 = 0.0;
|
||||||
|
for (obs, exp) in &values {
|
||||||
|
if *obs < 0.0 || *exp < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Negative value in CHISQ.TEST data".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if *exp == 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Zero expected value in CHISQ.TEST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let diff = obs - exp;
|
||||||
|
chi2 += (diff * diff) / exp;
|
||||||
|
}
|
||||||
|
|
||||||
|
if chi2 < 0.0 && chi2 > -1e-12 {
|
||||||
|
chi2 = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
let total = width * height;
|
||||||
|
if total <= 1 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"CHISQ.TEST degrees of freedom is zero".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let df = if width > 1 && height > 1 {
|
||||||
|
(width - 1) * (height - 1)
|
||||||
|
} else {
|
||||||
|
total - 1
|
||||||
|
};
|
||||||
|
|
||||||
|
let dist = match ChiSquared::new(df as f64) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid degrees of freedom in CHISQ.TEST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut p = 1.0 - dist.cdf(chi2);
|
||||||
|
|
||||||
|
// clamp tiny fp noise
|
||||||
|
if p < 0.0 && p > -1e-15 {
|
||||||
|
p = 0.0;
|
||||||
|
}
|
||||||
|
if p > 1.0 && p < 1.0 + 1e-15 {
|
||||||
|
p = 1.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(p)
|
||||||
|
}
|
||||||
|
}
|
||||||
227
base/src/functions/statistical/correl.rs
Normal file
227
base/src/functions/statistical/correl.rs
Normal file
@@ -0,0 +1,227 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// CORREL(array1, array2) - Returns the correlation coefficient of two data sets
|
||||||
|
pub(crate) fn fn_correl(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (_, _, values_left, values_right) = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut n = 0.0;
|
||||||
|
let mut sum_x = 0.0;
|
||||||
|
let mut sum_y = 0.0;
|
||||||
|
let mut sum_x2 = 0.0;
|
||||||
|
let mut sum_y2 = 0.0;
|
||||||
|
let mut sum_xy = 0.0;
|
||||||
|
|
||||||
|
for (x_opt, y_opt) in values_left.into_iter().zip(values_right.into_iter()) {
|
||||||
|
if let (Some(x), Some(y)) = (x_opt, y_opt) {
|
||||||
|
n += 1.0;
|
||||||
|
sum_x += x;
|
||||||
|
sum_y += y;
|
||||||
|
sum_x2 += x * x;
|
||||||
|
sum_y2 += y * y;
|
||||||
|
sum_xy += x * y;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Need at least 2 valid pairs
|
||||||
|
if n < 2.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"CORREL requires at least two numeric data points in each range".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let num = n * sum_xy - sum_x * sum_y;
|
||||||
|
let denom_x = n * sum_x2 - sum_x * sum_x;
|
||||||
|
let denom_y = n * sum_y2 - sum_y * sum_y;
|
||||||
|
let denom = (denom_x * denom_y).sqrt();
|
||||||
|
|
||||||
|
if denom == 0.0 || !denom.is_finite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Division by zero in CORREL".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let r = num / denom;
|
||||||
|
CalcResult::Number(r)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SLOPE(known_y's, known_x's) - Returns the slope of the linear regression line
|
||||||
|
pub(crate) fn fn_slope(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (_rows, _cols, values_y, values_x) = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut n = 0.0;
|
||||||
|
let mut sum_x = 0.0;
|
||||||
|
let mut sum_y = 0.0;
|
||||||
|
let mut sum_x2 = 0.0;
|
||||||
|
let mut sum_xy = 0.0;
|
||||||
|
|
||||||
|
let len = values_y.len().min(values_x.len());
|
||||||
|
for i in 0..len {
|
||||||
|
if let (Some(y), Some(x)) = (values_y[i], values_x[i]) {
|
||||||
|
n += 1.0;
|
||||||
|
sum_x += x;
|
||||||
|
sum_y += y;
|
||||||
|
sum_x2 += x * x;
|
||||||
|
sum_xy += x * y;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if n < 2.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"SLOPE requires at least two numeric data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let denom = n * sum_x2 - sum_x * sum_x;
|
||||||
|
if denom == 0.0 || !denom.is_finite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Division by zero in SLOPE".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let num = n * sum_xy - sum_x * sum_y;
|
||||||
|
let slope = num / denom;
|
||||||
|
|
||||||
|
CalcResult::Number(slope)
|
||||||
|
}
|
||||||
|
|
||||||
|
// INTERCEPT(known_y's, known_x's) - Returns the y-intercept of the linear regression line
|
||||||
|
pub(crate) fn fn_intercept(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (_rows, _cols, values_y, values_x) = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut n = 0.0;
|
||||||
|
let mut sum_x = 0.0;
|
||||||
|
let mut sum_y = 0.0;
|
||||||
|
let mut sum_x2 = 0.0;
|
||||||
|
let mut sum_xy = 0.0;
|
||||||
|
|
||||||
|
let len = values_y.len().min(values_x.len());
|
||||||
|
for i in 0..len {
|
||||||
|
if let (Some(y), Some(x)) = (values_y[i], values_x[i]) {
|
||||||
|
n += 1.0;
|
||||||
|
sum_x += x;
|
||||||
|
sum_y += y;
|
||||||
|
sum_x2 += x * x;
|
||||||
|
sum_xy += x * y;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if n < 2.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"INTERCEPT requires at least two numeric data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let denom = n * sum_x2 - sum_x * sum_x;
|
||||||
|
if denom == 0.0 || !denom.is_finite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Division by zero in INTERCEPT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let num = n * sum_xy - sum_x * sum_y;
|
||||||
|
let slope = num / denom;
|
||||||
|
let intercept = (sum_y - slope * sum_x) / n;
|
||||||
|
|
||||||
|
CalcResult::Number(intercept)
|
||||||
|
}
|
||||||
|
|
||||||
|
// STEYX(known_y's, known_x's) - Returns the standard error of the predicted y-values
|
||||||
|
pub(crate) fn fn_steyx(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (_rows, _cols, values_y, values_x) = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut n = 0.0;
|
||||||
|
let mut sum_x = 0.0;
|
||||||
|
let mut sum_y = 0.0;
|
||||||
|
let mut sum_x2 = 0.0;
|
||||||
|
let mut sum_xy = 0.0;
|
||||||
|
|
||||||
|
// We need the actual pairs again later for residuals
|
||||||
|
let mut pairs: Vec<(f64, f64)> = Vec::new();
|
||||||
|
|
||||||
|
let len = values_y.len().min(values_x.len());
|
||||||
|
for i in 0..len {
|
||||||
|
if let (Some(y), Some(x)) = (values_y[i], values_x[i]) {
|
||||||
|
n += 1.0;
|
||||||
|
sum_x += x;
|
||||||
|
sum_y += y;
|
||||||
|
sum_x2 += x * x;
|
||||||
|
sum_xy += x * y;
|
||||||
|
pairs.push((x, y));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Need at least 3 points for STEYX (n - 2 in denominator)
|
||||||
|
if n < 3.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"STEYX requires at least three numeric data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let denom = n * sum_x2 - sum_x * sum_x;
|
||||||
|
if denom == 0.0 || !denom.is_finite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Division by zero in STEYX".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let num = n * sum_xy - sum_x * sum_y;
|
||||||
|
let slope = num / denom;
|
||||||
|
let intercept = (sum_y - slope * sum_x) / n;
|
||||||
|
|
||||||
|
// Sum of squared residuals: Σ (y - ŷ)^2, ŷ = intercept + slope * x
|
||||||
|
let mut sse = 0.0;
|
||||||
|
for (x, y) in pairs {
|
||||||
|
let y_hat = intercept + slope * x;
|
||||||
|
let diff = y - y_hat;
|
||||||
|
sse += diff * diff;
|
||||||
|
}
|
||||||
|
|
||||||
|
let dof = n - 2.0;
|
||||||
|
if dof <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"STEYX has non-positive degrees of freedom".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let sey = (sse / dof).sqrt();
|
||||||
|
if !sey.is_finite() {
|
||||||
|
return CalcResult::new_error(Error::DIV, cell, "Numerical error in STEYX".to_string());
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(sey)
|
||||||
|
}
|
||||||
|
}
|
||||||
1071
base/src/functions/statistical/count_and_average.rs
Normal file
1071
base/src/functions/statistical/count_and_average.rs
Normal file
File diff suppressed because it is too large
Load Diff
264
base/src/functions/statistical/covariance.rs
Normal file
264
base/src/functions/statistical/covariance.rs
Normal file
@@ -0,0 +1,264 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_covariance_p(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let values1_opts = match self.evaluate_node_in_context(&args[0], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in first array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"First argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let values2_opts = match self.evaluate_node_in_context(&args[1], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in second array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Second argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Same number of cells
|
||||||
|
if values1_opts.len() != values2_opts.len() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NA,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.P requires arrays of the same size".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Count numeric data points in each array (ignoring text/booleans/empty)
|
||||||
|
let count1 = values1_opts.iter().filter(|v| v.is_some()).count();
|
||||||
|
let count2 = values2_opts.iter().filter(|v| v.is_some()).count();
|
||||||
|
|
||||||
|
if count1 == 0 || count2 == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.P requires at least one numeric value in each array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if count1 != count2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NA,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.P arrays must have the same number of numeric data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build paired numeric vectors, position by position
|
||||||
|
let mut xs: Vec<f64> = Vec::with_capacity(count1);
|
||||||
|
let mut ys: Vec<f64> = Vec::with_capacity(count2);
|
||||||
|
|
||||||
|
for (v1_opt, v2_opt) in values1_opts.into_iter().zip(values2_opts.into_iter()) {
|
||||||
|
if let (Some(x), Some(y)) = (v1_opt, v2_opt) {
|
||||||
|
xs.push(x);
|
||||||
|
ys.push(y);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = xs.len();
|
||||||
|
if n == 0 {
|
||||||
|
// Should be impossible given the checks above, but guard anyway
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.P has no paired numeric data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n_f = n as f64;
|
||||||
|
|
||||||
|
let mut sum_x = 0.0;
|
||||||
|
let mut sum_y = 0.0;
|
||||||
|
for i in 0..n {
|
||||||
|
sum_x += xs[i];
|
||||||
|
sum_y += ys[i];
|
||||||
|
}
|
||||||
|
let mean_x = sum_x / n_f;
|
||||||
|
let mean_y = sum_y / n_f;
|
||||||
|
|
||||||
|
let mut sum_prod = 0.0;
|
||||||
|
for i in 0..n {
|
||||||
|
let dx = xs[i] - mean_x;
|
||||||
|
let dy = ys[i] - mean_y;
|
||||||
|
sum_prod += dx * dy;
|
||||||
|
}
|
||||||
|
|
||||||
|
let cov = sum_prod / n_f;
|
||||||
|
CalcResult::Number(cov)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_covariance_s(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let values1_opts = match self.evaluate_node_in_context(&args[0], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in first array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"First argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let values2_opts = match self.evaluate_node_in_context(&args[1], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in second array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Second argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Same number of cells
|
||||||
|
if values1_opts.len() != values2_opts.len() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NA,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.S requires arrays of the same size".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Count numeric data points in each array (ignoring text/booleans/empty)
|
||||||
|
let count1 = values1_opts.iter().filter(|v| v.is_some()).count();
|
||||||
|
let count2 = values2_opts.iter().filter(|v| v.is_some()).count();
|
||||||
|
|
||||||
|
if count1 == 0 || count2 == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.S requires numeric values in each array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if count1 != count2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NA,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.S arrays must have the same number of numeric data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build paired numeric vectors
|
||||||
|
let mut xs: Vec<f64> = Vec::with_capacity(count1);
|
||||||
|
let mut ys: Vec<f64> = Vec::with_capacity(count2);
|
||||||
|
|
||||||
|
for (v1_opt, v2_opt) in values1_opts.into_iter().zip(values2_opts.into_iter()) {
|
||||||
|
if let (Some(x), Some(y)) = (v1_opt, v2_opt) {
|
||||||
|
xs.push(x);
|
||||||
|
ys.push(y);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = xs.len();
|
||||||
|
if n < 2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"COVARIANCE.S requires at least two paired data points".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n_f = n as f64;
|
||||||
|
|
||||||
|
let mut sum_x = 0.0;
|
||||||
|
let mut sum_y = 0.0;
|
||||||
|
for i in 0..n {
|
||||||
|
sum_x += xs[i];
|
||||||
|
sum_y += ys[i];
|
||||||
|
}
|
||||||
|
let mean_x = sum_x / n_f;
|
||||||
|
let mean_y = sum_y / n_f;
|
||||||
|
|
||||||
|
let mut sum_prod = 0.0;
|
||||||
|
for i in 0..n {
|
||||||
|
let dx = xs[i] - mean_x;
|
||||||
|
let dy = ys[i] - mean_y;
|
||||||
|
sum_prod += dx * dy;
|
||||||
|
}
|
||||||
|
|
||||||
|
let cov = sum_prod / (n_f - 1.0);
|
||||||
|
|
||||||
|
CalcResult::Number(cov)
|
||||||
|
}
|
||||||
|
}
|
||||||
135
base/src/functions/statistical/devsq.rs
Normal file
135
base/src/functions/statistical/devsq.rs
Normal file
@@ -0,0 +1,135 @@
|
|||||||
|
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
||||||
|
use crate::expressions::parser::ArrayNode;
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// DEVSQ(number1, [number2], ...)
|
||||||
|
pub(crate) fn fn_devsq(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
// tiny helper so we don't repeat ourselves
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// We ignore booleans and strings
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// We ignore booleans and strings
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// We ignore booleans and strings
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
if count == 0 {
|
||||||
|
// No numeric data at all
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"DEVSQ with no numeric data".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut result = sumsq - (sum * sum) / n;
|
||||||
|
|
||||||
|
// Numerical noise can make result slightly negative when it should be 0
|
||||||
|
if result < 0.0 && result > -1e-12 {
|
||||||
|
result = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
54
base/src/functions/statistical/exponential.rs
Normal file
54
base/src/functions/statistical/exponential.rs
Normal file
@@ -0,0 +1,54 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_expon_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
// EXPON.DIST(x, lambda, cumulative)
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let lambda = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[2], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x < 0.0 || lambda <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for EXPON.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let result = if cumulative {
|
||||||
|
// CDF
|
||||||
|
1.0 - (-lambda * x).exp()
|
||||||
|
} else {
|
||||||
|
// PDF
|
||||||
|
lambda * (-lambda * x).exp()
|
||||||
|
};
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for EXPON.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
418
base/src/functions/statistical/fisher.rs
Normal file
418
base/src/functions/statistical/fisher.rs
Normal file
@@ -0,0 +1,418 @@
|
|||||||
|
use statrs::distribution::{Continuous, ContinuousCDF, FisherSnedecor};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::functions::statistical::t_dist::sample_var;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// FISHER(x) = 0.5 * ln((1 + x) / (1 - x))
|
||||||
|
pub(crate) fn fn_fisher(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 1 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x <= -1.0 || x >= 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "x must be between -1 and 1 (exclusive) in FISHER".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let ratio = (1.0 + x) / (1.0 - x);
|
||||||
|
let result = 0.5 * ratio.ln();
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for FISHER".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// FISHERINV(y) = (e^(2y) - 1) / (e^(2y) + 1) = tanh(y)
|
||||||
|
pub(crate) fn fn_fisher_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 1 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let y = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Use tanh directly to avoid overflow from exp(2y)
|
||||||
|
let result = y.tanh();
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for FISHERINV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// F.DIST(x, deg_freedom1, deg_freedom2, cumulative)
|
||||||
|
pub(crate) fn fn_f_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df1 = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let df2 = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[3], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel domain checks
|
||||||
|
if x < 0.0 {
|
||||||
|
return CalcResult::new_error(Error::NUM, cell, "x must be >= 0 in F.DIST".to_string());
|
||||||
|
}
|
||||||
|
if df1 < 1.0 || df2 < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in F.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match FisherSnedecor::new(df1, df2) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for F distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(x) } else { dist.pdf(x) };
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for F.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_f_dist_rt(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
// F.DIST.RT(x, deg_freedom1, deg_freedom2)
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df1 = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let df2 = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"x must be >= 0 in F.DIST.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if df1 < 1.0 || df2 < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in F.DIST.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match FisherSnedecor::new(df1, df2) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for F distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Right-tail probability: P(F > x) = 1 - CDF(x)
|
||||||
|
let result = 1.0 - dist.cdf(x);
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() || result < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for F.DIST.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// F.INV(probability, deg_freedom1, deg_freedom2)
|
||||||
|
pub(crate) fn fn_f_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let df1 = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let df2 = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// probability < 0 or > 1 → #NUM!
|
||||||
|
if !(0.0..=1.0).contains(&p) {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"probability must be in [0,1] in F.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if df1 < 1.0 || df2 < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in F.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match FisherSnedecor::new(df1, df2) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for F distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let x = dist.inverse_cdf(p);
|
||||||
|
if x.is_nan() || x.is_infinite() || x < 0.0 {
|
||||||
|
return CalcResult::new_error(Error::NUM, cell, "Invalid result for F.INV".to_string());
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// F.INV.RT(probability, deg_freedom1, deg_freedom2)
|
||||||
|
pub(crate) fn fn_f_inv_rt(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let df1 = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let df2 = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if p <= 0.0 || p > 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"probability must be in (0,1] in F.INV.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if df1 < 1.0 || df2 < 1.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"degrees of freedom must be >= 1 in F.INV.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match FisherSnedecor::new(df1, df2) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for F distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// p is right-tail: p = P(F > x) = 1 - CDF(x)
|
||||||
|
let x = dist.inverse_cdf(1.0 - p);
|
||||||
|
if x.is_nan() || x.is_infinite() || x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for F.INV.RT".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// F.TEST(array1, array2)
|
||||||
|
pub(crate) fn fn_f_test(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let values1_opts = match self.evaluate_node_in_context(&args[0], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in first array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"First argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Get second sample as Vec<Option<f64>>
|
||||||
|
let values2_opts = match self.evaluate_node_in_context(&args[1], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in second array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Second argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let values1: Vec<f64> = values1_opts.into_iter().flatten().collect();
|
||||||
|
let values2: Vec<f64> = values2_opts.into_iter().flatten().collect();
|
||||||
|
|
||||||
|
let n1 = values1.len();
|
||||||
|
let n2 = values2.len();
|
||||||
|
|
||||||
|
// If fewer than 2 numeric values in either sample -> #DIV/0!
|
||||||
|
if n1 < 2 || n2 < 2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"F.TEST requires at least two numeric values in each sample".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let v1 = sample_var(&values1);
|
||||||
|
let v2 = sample_var(&values2);
|
||||||
|
|
||||||
|
if v1 <= 0.0 || v2 <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Variance of one sample is zero in F.TEST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// F ratio: larger variance / smaller variance
|
||||||
|
let mut f = v1 / v2;
|
||||||
|
let mut df1 = (n1 - 1) as f64;
|
||||||
|
let mut df2 = (n2 - 1) as f64;
|
||||||
|
|
||||||
|
if f < 1.0 {
|
||||||
|
f = 1.0 / f;
|
||||||
|
std::mem::swap(&mut df1, &mut df2);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match FisherSnedecor::new(df1, df2) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for F distribution in F.TEST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// One-tailed right-tail probability
|
||||||
|
let tail = 1.0 - dist.cdf(f);
|
||||||
|
// F.TEST is two-tailed: p = 2 * tail (with F >= 1)
|
||||||
|
let mut p = 2.0 * tail;
|
||||||
|
|
||||||
|
// Clamp tiny FP noise
|
||||||
|
if p < 0.0 && p > -1e-15 {
|
||||||
|
p = 0.0;
|
||||||
|
}
|
||||||
|
if p > 1.0 && p < 1.0 + 1e-15 {
|
||||||
|
p = 1.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(p)
|
||||||
|
}
|
||||||
|
}
|
||||||
194
base/src/functions/statistical/gamma.rs
Normal file
194
base/src/functions/statistical/gamma.rs
Normal file
@@ -0,0 +1,194 @@
|
|||||||
|
use statrs::distribution::{Continuous, ContinuousCDF, Gamma};
|
||||||
|
use statrs::function::gamma::{gamma, ln_gamma};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_gamma(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 1 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(s) => return s,
|
||||||
|
};
|
||||||
|
if x < 0.0 && x.floor() == x {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for Gamma function".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
let result = gamma(x);
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for Gamma function".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_gamma_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
// GAMMA.DIST(x, alpha, beta, cumulative)
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let alpha = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let beta_scale = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[3], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"x must be >= 0 in GAMMA.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if alpha <= 0.0 || beta_scale <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"alpha and beta must be > 0 in GAMMA.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let rate = 1.0 / beta_scale;
|
||||||
|
|
||||||
|
let dist = match Gamma::new(alpha, rate) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Gamma distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(x) } else { dist.pdf(x) };
|
||||||
|
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for GAMMA.DIST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_gamma_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
// GAMMA.INV(probability, alpha, beta)
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let alpha = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let beta_scale = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if !(0.0..=1.0).contains(&p) {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"probability must be in [0,1] in GAMMA.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if alpha <= 0.0 || beta_scale <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"alpha and beta must be > 0 in GAMMA.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let rate = 1.0 / beta_scale;
|
||||||
|
|
||||||
|
let dist = match Gamma::new(alpha, rate) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Gamma distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let x = dist.inverse_cdf(p);
|
||||||
|
if x.is_nan() || x.is_infinite() || x < 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid result for GAMMA.INV".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_gamma_ln(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 1 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(s) => return s,
|
||||||
|
};
|
||||||
|
if x < 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for Gamma function".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
let result = ln_gamma(x);
|
||||||
|
if result.is_nan() || result.is_infinite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for Gamma Ln function".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_gamma_ln_precise(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
self.fn_gamma_ln(args, cell)
|
||||||
|
}
|
||||||
|
}
|
||||||
39
base/src/functions/statistical/gauss.rs
Normal file
39
base/src/functions/statistical/gauss.rs
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
use statrs::distribution::{ContinuousCDF, Normal};
|
||||||
|
|
||||||
|
use crate::expressions::token::Error;
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{calc_result::CalcResult, expressions::parser::Node, model::Model};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_gauss(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 1 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
let z = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(s) => return s,
|
||||||
|
};
|
||||||
|
let dist = match Normal::new(0.0, 1.0) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::ERROR,
|
||||||
|
origin: cell,
|
||||||
|
message: "Failed to construct standard normal distribution".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = dist.cdf(z) - 0.5;
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for GAUSS".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
87
base/src/functions/statistical/geomean.rs
Normal file
87
base/src/functions/statistical/geomean.rs
Normal file
@@ -0,0 +1,87 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_geomean(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
let mut count = 0.0;
|
||||||
|
let mut product = 1.0;
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
count += 1.0;
|
||||||
|
product *= value;
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(b) => {
|
||||||
|
if let Node::ReferenceKind { .. } = arg {
|
||||||
|
} else {
|
||||||
|
product *= if b { 1.0 } else { 0.0 };
|
||||||
|
count += 1.0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
for row in left.row..(right.row + 1) {
|
||||||
|
for column in left.column..(right.column + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
count += 1.0;
|
||||||
|
product *= value;
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
CalcResult::Range { .. } => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
"Unexpected Range".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
CalcResult::String(s) => {
|
||||||
|
if let Node::ReferenceKind { .. } = arg {
|
||||||
|
// Do nothing
|
||||||
|
} else if let Ok(t) = s.parse::<f64>() {
|
||||||
|
product *= t;
|
||||||
|
count += 1.0;
|
||||||
|
} else {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: "Argument cannot be cast into number".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// Ignore everything else
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if count == 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "Division by Zero".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
CalcResult::Number(product.powf(1.0 / count))
|
||||||
|
}
|
||||||
|
}
|
||||||
108
base/src/functions/statistical/hypegeom.rs
Normal file
108
base/src/functions/statistical/hypegeom.rs
Normal file
@@ -0,0 +1,108 @@
|
|||||||
|
use statrs::distribution::{Discrete, DiscreteCDF, Hypergeometric};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// =HYPGEOM.DIST(sample_s, number_sample, population_s, number_pop, cumulative)
|
||||||
|
pub(crate) fn fn_hyp_geom_dist(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 5 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
// sample_s (number of successes in the sample)
|
||||||
|
let sample_s = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// number_sample (sample size)
|
||||||
|
let number_sample = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// population_s (number of successes in the population)
|
||||||
|
let population_s = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// number_pop (population size)
|
||||||
|
let number_pop = match self.get_number_no_bools(&args[3], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[4], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if sample_s < 0.0 || sample_s > f64::min(number_sample, population_s) {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for HYPGEOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
if sample_s < f64::max(0.0, number_sample + population_s - number_pop) {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for HYPGEOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
if number_sample <= 0.0 || number_sample > number_pop {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for HYPGEOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
if population_s <= 0.0 || population_s > number_pop {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for HYPGEOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let n_pop = number_pop as u64;
|
||||||
|
let k_pop = population_s as u64;
|
||||||
|
let n_sample = number_sample as u64;
|
||||||
|
let k = sample_s as u64;
|
||||||
|
|
||||||
|
let dist = match Hypergeometric::new(n_pop, k_pop, n_sample) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for hypergeometric distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let prob = if cumulative { dist.cdf(k) } else { dist.pmf(k) };
|
||||||
|
|
||||||
|
if !prob.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for HYPGEOM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(prob)
|
||||||
|
}
|
||||||
|
}
|
||||||
337
base/src/functions/statistical/if_ifs.rs
Normal file
337
base/src/functions/statistical/if_ifs.rs
Normal file
@@ -0,0 +1,337 @@
|
|||||||
|
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::functions::util::build_criteria;
|
||||||
|
use crate::{
|
||||||
|
calc_result::{CalcResult, Range},
|
||||||
|
expressions::parser::Node,
|
||||||
|
expressions::token::Error,
|
||||||
|
model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_countif(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() == 2 {
|
||||||
|
let arguments = vec![args[0].clone(), args[1].clone()];
|
||||||
|
self.fn_countifs(&arguments, cell)
|
||||||
|
} else {
|
||||||
|
CalcResult::new_args_number_error(cell)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// AVERAGEIF(criteria_range, criteria, [average_range])
|
||||||
|
/// if average_rage is missing then criteria_range will be used
|
||||||
|
pub(crate) fn fn_averageif(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() == 2 {
|
||||||
|
let arguments = vec![args[0].clone(), args[0].clone(), args[1].clone()];
|
||||||
|
self.fn_averageifs(&arguments, cell)
|
||||||
|
} else if args.len() == 3 {
|
||||||
|
let arguments = vec![args[2].clone(), args[0].clone(), args[1].clone()];
|
||||||
|
self.fn_averageifs(&arguments, cell)
|
||||||
|
} else {
|
||||||
|
CalcResult::new_args_number_error(cell)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// FIXME: This function shares a lot of code with apply_ifs. Can we merge them?
|
||||||
|
pub(crate) fn fn_countifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let args_count = args.len();
|
||||||
|
if args_count < 2 || !args_count.is_multiple_of(2) {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let case_count = args_count / 2;
|
||||||
|
// NB: this is a beautiful example of the borrow checker
|
||||||
|
// The order of these two definitions cannot be swapped.
|
||||||
|
let mut criteria = Vec::new();
|
||||||
|
let mut fn_criteria = Vec::new();
|
||||||
|
let ranges = &mut Vec::new();
|
||||||
|
for case_index in 0..case_count {
|
||||||
|
let criterion = self.evaluate_node_in_context(&args[case_index * 2 + 1], cell);
|
||||||
|
criteria.push(criterion);
|
||||||
|
// NB: We cannot do:
|
||||||
|
// fn_criteria.push(build_criteria(&criterion));
|
||||||
|
// because criterion doesn't live long enough
|
||||||
|
let result = self.evaluate_node_in_context(&args[case_index * 2], cell);
|
||||||
|
if result.is_error() {
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
if let CalcResult::Range { left, right } = result {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
// TODO test ranges are of the same size as sum_range
|
||||||
|
ranges.push(Range { left, right });
|
||||||
|
} else {
|
||||||
|
return CalcResult::new_error(Error::VALUE, cell, "Expected a range".to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for criterion in criteria.iter() {
|
||||||
|
fn_criteria.push(build_criteria(criterion));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut total = 0.0;
|
||||||
|
let first_range = &ranges[0];
|
||||||
|
let left_row = first_range.left.row;
|
||||||
|
let left_column = first_range.left.column;
|
||||||
|
let right_row = first_range.right.row;
|
||||||
|
let right_column = first_range.right.column;
|
||||||
|
|
||||||
|
let dimension = match self.workbook.worksheet(first_range.left.sheet) {
|
||||||
|
Ok(s) => s.dimension(),
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", first_range.left.sheet),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let max_row = dimension.max_row;
|
||||||
|
let max_column = dimension.max_column;
|
||||||
|
|
||||||
|
let open_row = left_row == 1 && right_row == LAST_ROW;
|
||||||
|
let open_column = left_column == 1 && right_column == LAST_COLUMN;
|
||||||
|
|
||||||
|
for row in left_row..right_row + 1 {
|
||||||
|
if open_row && row > max_row {
|
||||||
|
// If the row is larger than the max row in the sheet then all cells are empty.
|
||||||
|
// We compute it only once
|
||||||
|
let mut is_true = true;
|
||||||
|
for fn_criterion in fn_criteria.iter() {
|
||||||
|
if !fn_criterion(&CalcResult::EmptyCell) {
|
||||||
|
is_true = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if is_true {
|
||||||
|
total += ((LAST_ROW - max_row) * (right_column - left_column + 1)) as f64;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
for column in left_column..right_column + 1 {
|
||||||
|
if open_column && column > max_column {
|
||||||
|
// If the column is larger than the max column in the sheet then all cells are empty.
|
||||||
|
// We compute it only once
|
||||||
|
let mut is_true = true;
|
||||||
|
for fn_criterion in fn_criteria.iter() {
|
||||||
|
if !fn_criterion(&CalcResult::EmptyCell) {
|
||||||
|
is_true = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if is_true {
|
||||||
|
total += (LAST_COLUMN - max_column) as f64;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
let mut is_true = true;
|
||||||
|
for case_index in 0..case_count {
|
||||||
|
// We check if value in range n meets criterion n
|
||||||
|
let range = &ranges[case_index];
|
||||||
|
let fn_criterion = &fn_criteria[case_index];
|
||||||
|
let value = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: range.left.sheet,
|
||||||
|
row: range.left.row + row - first_range.left.row,
|
||||||
|
column: range.left.column + column - first_range.left.column,
|
||||||
|
});
|
||||||
|
if !fn_criterion(&value) {
|
||||||
|
is_true = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if is_true {
|
||||||
|
total += 1.0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Number(total)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn apply_ifs<F>(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
mut apply: F,
|
||||||
|
) -> Result<(), CalcResult>
|
||||||
|
where
|
||||||
|
F: FnMut(f64),
|
||||||
|
{
|
||||||
|
let args_count = args.len();
|
||||||
|
if args_count < 3 || args_count.is_multiple_of(2) {
|
||||||
|
return Err(CalcResult::new_args_number_error(cell));
|
||||||
|
}
|
||||||
|
let arg_0 = self.evaluate_node_in_context(&args[0], cell);
|
||||||
|
if arg_0.is_error() {
|
||||||
|
return Err(arg_0);
|
||||||
|
}
|
||||||
|
let sum_range = if let CalcResult::Range { left, right } = arg_0 {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
Range { left, right }
|
||||||
|
} else {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Expected a range".to_string(),
|
||||||
|
));
|
||||||
|
};
|
||||||
|
|
||||||
|
let case_count = (args_count - 1) / 2;
|
||||||
|
// NB: this is a beautiful example of the borrow checker
|
||||||
|
// The order of these two definitions cannot be swapped.
|
||||||
|
let mut criteria = Vec::new();
|
||||||
|
let mut fn_criteria = Vec::new();
|
||||||
|
let ranges = &mut Vec::new();
|
||||||
|
for case_index in 1..=case_count {
|
||||||
|
let criterion = self.evaluate_node_in_context(&args[case_index * 2], cell);
|
||||||
|
// NB: criterion might be an error. That's ok
|
||||||
|
criteria.push(criterion);
|
||||||
|
// NB: We cannot do:
|
||||||
|
// fn_criteria.push(build_criteria(&criterion));
|
||||||
|
// because criterion doesn't live long enough
|
||||||
|
let result = self.evaluate_node_in_context(&args[case_index * 2 - 1], cell);
|
||||||
|
if result.is_error() {
|
||||||
|
return Err(result);
|
||||||
|
}
|
||||||
|
if let CalcResult::Range { left, right } = result {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
// TODO test ranges are of the same size as sum_range
|
||||||
|
ranges.push(Range { left, right });
|
||||||
|
} else {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Expected a range".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for criterion in criteria.iter() {
|
||||||
|
fn_criteria.push(build_criteria(criterion));
|
||||||
|
}
|
||||||
|
|
||||||
|
let left_row = sum_range.left.row;
|
||||||
|
let left_column = sum_range.left.column;
|
||||||
|
let mut right_row = sum_range.right.row;
|
||||||
|
let mut right_column = sum_range.right.column;
|
||||||
|
|
||||||
|
if left_row == 1 && right_row == LAST_ROW {
|
||||||
|
right_row = match self.workbook.worksheet(sum_range.left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", sum_range.left.sheet),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if left_column == 1 && right_column == LAST_COLUMN {
|
||||||
|
right_column = match self.workbook.worksheet(sum_range.left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return Err(CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", sum_range.left.sheet),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in left_row..right_row + 1 {
|
||||||
|
for column in left_column..right_column + 1 {
|
||||||
|
let mut is_true = true;
|
||||||
|
for case_index in 0..case_count {
|
||||||
|
// We check if value in range n meets criterion n
|
||||||
|
let range = &ranges[case_index];
|
||||||
|
let fn_criterion = &fn_criteria[case_index];
|
||||||
|
let value = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: range.left.sheet,
|
||||||
|
row: range.left.row + row - sum_range.left.row,
|
||||||
|
column: range.left.column + column - sum_range.left.column,
|
||||||
|
});
|
||||||
|
if !fn_criterion(&value) {
|
||||||
|
is_true = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if is_true {
|
||||||
|
let v = self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: sum_range.left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
});
|
||||||
|
match v {
|
||||||
|
CalcResult::Number(n) => apply(n),
|
||||||
|
CalcResult::Error { .. } => return Err(v),
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_averageifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let mut total = 0.0;
|
||||||
|
let mut count = 0.0;
|
||||||
|
|
||||||
|
let average = |value: f64| {
|
||||||
|
total += value;
|
||||||
|
count += 1.0;
|
||||||
|
};
|
||||||
|
if let Err(e) = self.apply_ifs(args, cell, average) {
|
||||||
|
return e;
|
||||||
|
}
|
||||||
|
|
||||||
|
if count == 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "division by 0".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
CalcResult::Number(total / count)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_minifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let mut min = f64::INFINITY;
|
||||||
|
let apply_min = |value: f64| min = value.min(min);
|
||||||
|
if let Err(e) = self.apply_ifs(args, cell, apply_min) {
|
||||||
|
return e;
|
||||||
|
}
|
||||||
|
|
||||||
|
if min.is_infinite() {
|
||||||
|
min = 0.0;
|
||||||
|
}
|
||||||
|
CalcResult::Number(min)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_maxifs(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let mut max = -f64::INFINITY;
|
||||||
|
let apply_max = |value: f64| max = value.max(max);
|
||||||
|
if let Err(e) = self.apply_ifs(args, cell, apply_max) {
|
||||||
|
return e;
|
||||||
|
}
|
||||||
|
if max.is_infinite() {
|
||||||
|
max = 0.0;
|
||||||
|
}
|
||||||
|
CalcResult::Number(max)
|
||||||
|
}
|
||||||
|
}
|
||||||
124
base/src/functions/statistical/log_normal.rs
Normal file
124
base/src/functions/statistical/log_normal.rs
Normal file
@@ -0,0 +1,124 @@
|
|||||||
|
use statrs::distribution::{Continuous, ContinuousCDF, LogNormal};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_log_norm_dist(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let mean = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let std_dev = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let cumulative = match self.get_boolean(&args[3], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel domain checks
|
||||||
|
if x <= 0.0 || std_dev <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for LOGNORM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match LogNormal::new(mean, std_dev) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for LOGNORM.DIST".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(x) } else { dist.pdf(x) };
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for LOGNORM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_log_norm_inv(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
use statrs::distribution::{ContinuousCDF, LogNormal};
|
||||||
|
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let mean = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let std_dev = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel domain checks
|
||||||
|
if p <= 0.0 || p >= 1.0 || std_dev <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for LOGNORM.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match LogNormal::new(mean, std_dev) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for LOGNORM.INV".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = dist.inverse_cdf(p);
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameter for LOGNORM.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
26
base/src/functions/statistical/mod.rs
Normal file
26
base/src/functions/statistical/mod.rs
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
mod beta;
|
||||||
|
mod binom;
|
||||||
|
mod chisq;
|
||||||
|
mod correl;
|
||||||
|
mod count_and_average;
|
||||||
|
mod covariance;
|
||||||
|
mod devsq;
|
||||||
|
mod exponential;
|
||||||
|
mod fisher;
|
||||||
|
mod gamma;
|
||||||
|
mod gauss;
|
||||||
|
mod geomean;
|
||||||
|
mod hypegeom;
|
||||||
|
mod if_ifs;
|
||||||
|
mod log_normal;
|
||||||
|
mod normal;
|
||||||
|
mod pearson;
|
||||||
|
mod phi;
|
||||||
|
mod poisson;
|
||||||
|
mod rank_eq_avg;
|
||||||
|
mod standard_dev;
|
||||||
|
mod standardize;
|
||||||
|
mod t_dist;
|
||||||
|
mod variance;
|
||||||
|
mod weibull;
|
||||||
|
mod z_test;
|
||||||
325
base/src/functions/statistical/normal.rs
Normal file
325
base/src/functions/statistical/normal.rs
Normal file
@@ -0,0 +1,325 @@
|
|||||||
|
use statrs::distribution::{Continuous, ContinuousCDF, Normal, StudentsT};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// NORM.DIST(x, mean, standard_dev, cumulative)
|
||||||
|
pub(crate) fn fn_norm_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let mean = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let std_dev = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let cumulative = match self.get_boolean(&args[3], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Excel: standard_dev must be > 0
|
||||||
|
if std_dev <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "standard_dev must be > 0 in NORM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match Normal::new(mean, std_dev) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for NORM.DIST".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(x) } else { dist.pdf(x) };
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for NORM.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NORM.INV(probability, mean, standard_dev)
|
||||||
|
pub(crate) fn fn_norm_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let mean = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let std_dev = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if p <= 0.0 || p >= 1.0 || std_dev <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for NORM.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match Normal::new(mean, std_dev) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for NORM.INV".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let x = dist.inverse_cdf(p);
|
||||||
|
|
||||||
|
if !x.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for NORM.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NORM.S.DIST(z, cumulative)
|
||||||
|
pub(crate) fn fn_norm_s_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let z = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let cumulative = match self.get_boolean(&args[1], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let dist = match Normal::new(0.0, 1.0) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::ERROR,
|
||||||
|
origin: cell,
|
||||||
|
message: "Failed to construct standard normal distribution".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(z) } else { dist.pdf(z) };
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for NORM.S.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NORM.S.INV(probability)
|
||||||
|
pub(crate) fn fn_norm_s_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 1 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if p <= 0.0 || p >= 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "probability must be in (0,1) in NORM.S.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match Normal::new(0.0, 1.0) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::ERROR,
|
||||||
|
origin: cell,
|
||||||
|
message: "Failed to construct standard normal distribution".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let z = dist.inverse_cdf(p);
|
||||||
|
|
||||||
|
if !z.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for NORM.S.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(z)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_confidence_norm(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let alpha = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let std_dev = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let size = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.floor(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if alpha <= 0.0 || alpha >= 1.0 || std_dev <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for CONFIDENCE.NORM".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if size < 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Sample size must be at least 1".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let normal = match Normal::new(0.0, 1.0) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
"Failed to construct normal distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let quantile = normal.inverse_cdf(1.0 - alpha / 2.0);
|
||||||
|
if !quantile.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid quantile for CONFIDENCE.NORM".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let margin = quantile * std_dev / size.sqrt();
|
||||||
|
CalcResult::Number(margin)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_confidence_t(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let alpha = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let std_dev = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let size = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Domain checks
|
||||||
|
if alpha <= 0.0 || alpha >= 1.0 || std_dev <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for CONFIDENCE.T".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Need at least 2 observations so df = n - 1 > 0
|
||||||
|
if size < 2.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::DIV,
|
||||||
|
origin: cell,
|
||||||
|
message: "Sample size must be at least 2".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let df = size - 1.0;
|
||||||
|
|
||||||
|
let t_dist = match StudentsT::new(0.0, 1.0, df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
"Failed to construct Student's t distribution".to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Two-sided CI => use 1 - alpha/2
|
||||||
|
let t_crit = t_dist.inverse_cdf(1.0 - alpha / 2.0);
|
||||||
|
if !t_crit.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid quantile for CONFIDENCE.T".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let margin = t_crit * std_dev / size.sqrt();
|
||||||
|
CalcResult::Number(margin)
|
||||||
|
}
|
||||||
|
}
|
||||||
113
base/src/functions/statistical/pearson.rs
Normal file
113
base/src/functions/statistical/pearson.rs
Normal file
@@ -0,0 +1,113 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// PEARSON(array1, array2)
|
||||||
|
pub(crate) fn fn_pearson(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (_, _, values_left, values_right) = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(result) => result,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Flatten into (x, y) pairs, skipping non-numeric entries (None)
|
||||||
|
let mut n: f64 = 0.0;
|
||||||
|
let mut sum_x = 0.0;
|
||||||
|
let mut sum_y = 0.0;
|
||||||
|
let mut sum_x2 = 0.0;
|
||||||
|
let mut sum_y2 = 0.0;
|
||||||
|
let mut sum_xy = 0.0;
|
||||||
|
|
||||||
|
let len = values_left.len().min(values_right.len());
|
||||||
|
for i in 0..len {
|
||||||
|
match (values_left[i], values_right[i]) {
|
||||||
|
(Some(x), Some(y)) => {
|
||||||
|
n += 1.0;
|
||||||
|
sum_x += x;
|
||||||
|
sum_y += y;
|
||||||
|
sum_x2 += x * x;
|
||||||
|
sum_y2 += y * y;
|
||||||
|
sum_xy += x * y;
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// Ignore pairs where at least one side is non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if n < 2.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"PEARSON requires at least two numeric pairs".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pearson correlation:
|
||||||
|
// r = [ n*Σxy - (Σx)(Σy) ] / sqrt( [n*Σx² - (Σx)²] [n*Σy² - (Σy)²] )
|
||||||
|
let num = n * sum_xy - sum_x * sum_y;
|
||||||
|
let denom_x = n * sum_x2 - sum_x * sum_x;
|
||||||
|
let denom_y = n * sum_y2 - sum_y * sum_y;
|
||||||
|
|
||||||
|
if denom_x.abs() < 1e-15 || denom_y.abs() < 1e-15 {
|
||||||
|
// Zero variance in at least one series
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"PEARSON cannot be computed when one series has zero variance".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let denom = (denom_x * denom_y).sqrt();
|
||||||
|
|
||||||
|
CalcResult::Number(num / denom)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RSQ(array1, array2) = CORREL(array1, array2)^2
|
||||||
|
pub(crate) fn fn_rsq(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
let (_rows, _cols, values1, values2) = match self.fn_get_two_matrices(args, cell) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut n = 0.0_f64;
|
||||||
|
let mut sum_x = 0.0_f64;
|
||||||
|
let mut sum_y = 0.0_f64;
|
||||||
|
let mut sum_x2 = 0.0_f64;
|
||||||
|
let mut sum_y2 = 0.0_f64;
|
||||||
|
let mut sum_xy = 0.0_f64;
|
||||||
|
|
||||||
|
let len = values1.len().min(values2.len());
|
||||||
|
for i in 0..len {
|
||||||
|
if let (Some(x), Some(y)) = (values1[i], values2[i]) {
|
||||||
|
n += 1.0;
|
||||||
|
sum_x += x;
|
||||||
|
sum_y += y;
|
||||||
|
sum_x2 += x * x;
|
||||||
|
sum_y2 += y * y;
|
||||||
|
sum_xy += x * y;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if n < 2.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"RSQ requires at least two numeric data points in each range".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let num = n * sum_xy - sum_x * sum_y;
|
||||||
|
let denom_x = n * sum_x2 - sum_x * sum_x;
|
||||||
|
let denom_y = n * sum_y2 - sum_y * sum_y;
|
||||||
|
let denom = (denom_x * denom_y).sqrt();
|
||||||
|
|
||||||
|
if denom == 0.0 || !denom.is_finite() {
|
||||||
|
return CalcResult::new_error(Error::DIV, cell, "Division by zero in RSQ".to_string());
|
||||||
|
}
|
||||||
|
|
||||||
|
let r = num / denom;
|
||||||
|
CalcResult::Number(r * r)
|
||||||
|
}
|
||||||
|
}
|
||||||
21
base/src/functions/statistical/phi.rs
Normal file
21
base/src/functions/statistical/phi.rs
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{calc_result::CalcResult, expressions::parser::Node, model::Model};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// PHI(x) = standard normal PDF at x
|
||||||
|
pub(crate) fn fn_phi(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 1 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Standard normal PDF: (1 / sqrt(2π)) * exp(-x^2 / 2)
|
||||||
|
let result = (-(x * x) / 2.0).exp() / (2.0 * std::f64::consts::PI).sqrt();
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
94
base/src/functions/statistical/poisson.rs
Normal file
94
base/src/functions/statistical/poisson.rs
Normal file
@@ -0,0 +1,94 @@
|
|||||||
|
use statrs::distribution::{Discrete, DiscreteCDF, Poisson};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// =POISSON.DIST(x, mean, cumulative)
|
||||||
|
pub(crate) fn fn_poisson_dist(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
// x
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// mean (lambda)
|
||||||
|
let lambda = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[2], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x < 0.0 || lambda < 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for POISSON.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Guard against insane k for u64
|
||||||
|
if x < 0.0 || x > (u64::MAX as f64) {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for POISSON.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let k = x as u64;
|
||||||
|
|
||||||
|
// Special-case lambda = 0: degenerate distribution at 0
|
||||||
|
if lambda == 0.0 {
|
||||||
|
let result = if cumulative {
|
||||||
|
// For x >= 0, P(X <= x) = 1
|
||||||
|
1.0
|
||||||
|
} else {
|
||||||
|
// P(X = 0) = 1, P(X = k>0) = 0
|
||||||
|
if k == 0 {
|
||||||
|
1.0
|
||||||
|
} else {
|
||||||
|
0.0
|
||||||
|
}
|
||||||
|
};
|
||||||
|
return CalcResult::Number(result);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match Poisson::new(lambda) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for POISSON.DIST".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let prob = if cumulative { dist.cdf(k) } else { dist.pmf(k) };
|
||||||
|
|
||||||
|
if !prob.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for POISSON.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(prob)
|
||||||
|
}
|
||||||
|
}
|
||||||
202
base/src/functions/statistical/rank_eq_avg.rs
Normal file
202
base/src/functions/statistical/rank_eq_avg.rs
Normal file
@@ -0,0 +1,202 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// Helper to collect numeric values from the 2nd argument of RANK.*
|
||||||
|
fn collect_rank_values(
|
||||||
|
&mut self,
|
||||||
|
arg: &Node,
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> Result<Vec<f64>, CalcResult> {
|
||||||
|
let values = match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Array(array) => match self.values_from_array(array) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(e) => {
|
||||||
|
return Err(CalcResult::Error {
|
||||||
|
error: Error::VALUE,
|
||||||
|
origin: cell,
|
||||||
|
message: format!("Unsupported array argument: {}", e),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
},
|
||||||
|
CalcResult::Range { left, right } => self.values_from_range(left, right)?,
|
||||||
|
CalcResult::Boolean(value) => {
|
||||||
|
if !matches!(arg, Node::ReferenceKind { .. }) {
|
||||||
|
vec![Some(if value { 1.0 } else { 0.0 })]
|
||||||
|
} else {
|
||||||
|
return Err(CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Unsupported argument type".to_string(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
return Err(CalcResult::Error {
|
||||||
|
error: Error::NIMPL,
|
||||||
|
origin: cell,
|
||||||
|
message: "Unsupported argument type".to_string(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let numeric_values: Vec<f64> = values.into_iter().flatten().collect();
|
||||||
|
Ok(numeric_values)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RANK.EQ(number, ref, [order])
|
||||||
|
pub(crate) fn fn_rank_eq(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if !(2..=3).contains(&args.len()) {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
// number
|
||||||
|
let number = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// ref
|
||||||
|
let mut values = match self.collect_rank_values(&args[1], cell) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if values.is_empty() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "No numeric values for RANK.EQ".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// order: default 0 (descending)
|
||||||
|
let order = if args.len() == 2 {
|
||||||
|
0.0
|
||||||
|
} else {
|
||||||
|
match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
values.retain(|v| !v.is_nan());
|
||||||
|
|
||||||
|
// "better" = greater (descending) or smaller (ascending)
|
||||||
|
let mut better = 0;
|
||||||
|
let mut equal = 0;
|
||||||
|
|
||||||
|
if order == 0.0 {
|
||||||
|
// descending
|
||||||
|
for v in &values {
|
||||||
|
if *v > number {
|
||||||
|
better += 1;
|
||||||
|
} else if *v == number {
|
||||||
|
equal += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// ascending
|
||||||
|
for v in &values {
|
||||||
|
if *v < number {
|
||||||
|
better += 1;
|
||||||
|
} else if *v == number {
|
||||||
|
equal += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if equal == 0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NA,
|
||||||
|
origin: cell,
|
||||||
|
message: "Number not found in reference for RANK.EQ".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let rank = (better as f64) + 1.0;
|
||||||
|
CalcResult::Number(rank)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RANK.AVG(number, ref, [order])
|
||||||
|
pub(crate) fn fn_rank_avg(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if !(2..=3).contains(&args.len()) {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
// number
|
||||||
|
let number = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
// ref
|
||||||
|
let mut values = match self.collect_rank_values(&args[1], cell) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if values.is_empty() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "No numeric values for RANK.AVG".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// order: default 0 (descending)
|
||||||
|
let order = if args.len() == 2 {
|
||||||
|
0.0
|
||||||
|
} else {
|
||||||
|
match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
values.retain(|v| !v.is_nan());
|
||||||
|
|
||||||
|
// > or < depending on order
|
||||||
|
let mut better = 0;
|
||||||
|
let mut equal = 0;
|
||||||
|
|
||||||
|
if order == 0.0 {
|
||||||
|
// descending
|
||||||
|
for v in &values {
|
||||||
|
if *v > number {
|
||||||
|
better += 1;
|
||||||
|
} else if *v == number {
|
||||||
|
equal += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// ascending
|
||||||
|
for v in &values {
|
||||||
|
if *v < number {
|
||||||
|
better += 1;
|
||||||
|
} else if *v == number {
|
||||||
|
equal += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if equal == 0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NA,
|
||||||
|
origin: cell,
|
||||||
|
message: "Number not found in reference for RANK.AVG".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// For ties, average of the ranks. If the equal values occupy positions
|
||||||
|
// (better+1) ..= (better+equal), the average is:
|
||||||
|
// better + (equal + 1) / 2
|
||||||
|
let better_f = better as f64;
|
||||||
|
let equal_f = equal as f64;
|
||||||
|
let rank = better_f + (equal_f + 1.0) / 2.0;
|
||||||
|
|
||||||
|
CalcResult::Number(rank)
|
||||||
|
}
|
||||||
|
}
|
||||||
519
base/src/functions/statistical/standard_dev.rs
Normal file
519
base/src/functions/statistical/standard_dev.rs
Normal file
@@ -0,0 +1,519 @@
|
|||||||
|
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
||||||
|
use crate::expressions::parser::ArrayNode;
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_stdev_p(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"STDEV.P with no numeric data".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / n;
|
||||||
|
|
||||||
|
// clamp tiny negatives from FP noise
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var.sqrt())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_stdev_s(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count <= 1 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"STDEV.S requires at least two numeric values".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / (n - 1.0);
|
||||||
|
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var.sqrt())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_stdeva(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::String(_) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, 0.0);
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(value) => {
|
||||||
|
let val = if value { 1.0 } else { 0.0 };
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, val);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count <= 1 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"STDEVA requires at least two numeric values".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / (n - 1.0);
|
||||||
|
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var.sqrt())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_stdevpa(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::String(_) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, 0.0);
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(value) => {
|
||||||
|
let val = if value { 1.0 } else { 0.0 };
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, val);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"STDEVPA with no numeric data".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / n;
|
||||||
|
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var.sqrt())
|
||||||
|
}
|
||||||
|
}
|
||||||
38
base/src/functions/statistical/standardize.rs
Normal file
38
base/src/functions/statistical/standardize.rs
Normal file
@@ -0,0 +1,38 @@
|
|||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_standardize(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
// STANDARDIZE(x, mean, standard_dev)
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let mean = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let std_dev = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if std_dev <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "standard_dev must be > 0 in STANDARDIZE".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let z = (x - mean) / std_dev;
|
||||||
|
|
||||||
|
CalcResult::Number(z)
|
||||||
|
}
|
||||||
|
}
|
||||||
588
base/src/functions/statistical/t_dist.rs
Normal file
588
base/src/functions/statistical/t_dist.rs
Normal file
@@ -0,0 +1,588 @@
|
|||||||
|
use statrs::distribution::{Continuous, ContinuousCDF, StudentsT};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
fn mean(xs: &[f64]) -> f64 {
|
||||||
|
let n = xs.len();
|
||||||
|
if n == 0 {
|
||||||
|
return 0.0;
|
||||||
|
}
|
||||||
|
let mut s = 0.0;
|
||||||
|
for &x in xs {
|
||||||
|
s += x;
|
||||||
|
}
|
||||||
|
s / (n as f64)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn sample_var(xs: &[f64]) -> f64 {
|
||||||
|
let n = xs.len();
|
||||||
|
if n < 2 {
|
||||||
|
return 0.0;
|
||||||
|
}
|
||||||
|
let m = mean(xs);
|
||||||
|
let mut s = 0.0;
|
||||||
|
for &x in xs {
|
||||||
|
let d = x - m;
|
||||||
|
s += d * d;
|
||||||
|
}
|
||||||
|
s / ((n - 1) as f64)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum TTestType {
|
||||||
|
Paired,
|
||||||
|
TwoSampleEqualVar,
|
||||||
|
TwoSampleUnequalVar,
|
||||||
|
}
|
||||||
|
|
||||||
|
enum TTestTails {
|
||||||
|
OneTailed,
|
||||||
|
TwoTailed,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// T.DIST(x, deg_freedom, cumulative)
|
||||||
|
pub(crate) fn fn_t_dist(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[2], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if df < 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "deg_freedom must be >= 1 in T.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match StudentsT::new(0.0, 1.0, df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for T.DIST".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(x) } else { dist.pdf(x) };
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for T.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// T.DIST.2T(x, deg_freedom)
|
||||||
|
pub(crate) fn fn_t_dist_2t(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x < 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "x must be >= 0 in T.DIST.2T".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
if df < 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "deg_freedom must be >= 1 in T.DIST.2T".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match StudentsT::new(0.0, 1.0, df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for T.DIST.2T".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let upper_tail = 1.0 - dist.cdf(x);
|
||||||
|
let mut result = 2.0 * upper_tail;
|
||||||
|
|
||||||
|
result = result.clamp(0.0, 1.0);
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for T.DIST.2T".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// T.DIST.RT(x, deg_freedom)
|
||||||
|
pub(crate) fn fn_t_dist_rt(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if df < 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "deg_freedom must be >= 1 in T.DIST.RT".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match StudentsT::new(0.0, 1.0, df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for T.DIST.RT".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = 1.0 - dist.cdf(x);
|
||||||
|
|
||||||
|
if !result.is_finite() || result < 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for T.DIST.RT".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
// T.INV(probability, deg_freedom)
|
||||||
|
pub(crate) fn fn_t_inv(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if p <= 0.0 || p >= 1.0 || df < 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for T.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match StudentsT::new(0.0, 1.0, df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for T.INV".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let x = dist.inverse_cdf(p);
|
||||||
|
|
||||||
|
if !x.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for T.INV".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// T.INV.2T(probability, deg_freedom)
|
||||||
|
pub(crate) fn fn_t_inv_2t(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 2 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let p = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let df = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f.trunc(),
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if p <= 0.0 || p > 1.0 || df < 1.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for T.INV.2T".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match StudentsT::new(0.0, 1.0, df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for T.INV.2T".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Two-sided: F(x) = 1 - p/2
|
||||||
|
let target_cdf = 1.0 - p / 2.0;
|
||||||
|
let x = dist.inverse_cdf(target_cdf);
|
||||||
|
|
||||||
|
if !x.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for T.INV.2T".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(x.abs())
|
||||||
|
}
|
||||||
|
|
||||||
|
// T.TEST(array1, array2, tails, type)
|
||||||
|
pub(crate) fn fn_t_test(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let values1_opts = match self.evaluate_node_in_context(&args[0], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in first array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"First argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let values2_opts = match self.evaluate_node_in_context(&args[1], cell) {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(a) => match self.values_from_array(a) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in second array: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Second argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let tails = match self.get_number(&args[2], cell) {
|
||||||
|
Ok(f) => {
|
||||||
|
let tf = f.trunc();
|
||||||
|
if tf == 1.0 {
|
||||||
|
TTestTails::OneTailed
|
||||||
|
} else if tf == 2.0 {
|
||||||
|
TTestTails::TwoTailed
|
||||||
|
} else {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"tails must be 1 or 2".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
let test_type = match self.get_number(&args[3], cell) {
|
||||||
|
Ok(f) => {
|
||||||
|
let tf = f.trunc();
|
||||||
|
match tf {
|
||||||
|
1.0 => TTestType::Paired,
|
||||||
|
2.0 => TTestType::TwoSampleEqualVar,
|
||||||
|
3.0 => TTestType::TwoSampleUnequalVar,
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"type must be 1, 2, or 3".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let (values1, values2): (Vec<f64>, Vec<f64>) = if matches!(test_type, TTestType::Paired) {
|
||||||
|
values1_opts
|
||||||
|
.into_iter()
|
||||||
|
.zip(values2_opts)
|
||||||
|
.filter_map(|(o1, o2)| match (o1, o2) {
|
||||||
|
(Some(v1), Some(v2)) => Some((v1, v2)),
|
||||||
|
_ => None, // skip if either is None
|
||||||
|
})
|
||||||
|
.unzip()
|
||||||
|
} else {
|
||||||
|
// keep only numeric entries, ignore non-numeric (Option::None)
|
||||||
|
let v1: Vec<f64> = values1_opts.into_iter().flatten().collect();
|
||||||
|
let v2: Vec<f64> = values2_opts.into_iter().flatten().collect();
|
||||||
|
(v1, v2)
|
||||||
|
};
|
||||||
|
|
||||||
|
let n1 = values1.len();
|
||||||
|
let n2 = values2.len();
|
||||||
|
|
||||||
|
if n1 == 0 || n2 == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"T.TEST requires non-empty samples".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (t_stat, df) = match test_type {
|
||||||
|
TTestType::Paired => {
|
||||||
|
if n1 != n2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NA,
|
||||||
|
cell,
|
||||||
|
"For paired T.TEST, both samples must have the same length".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if n1 < 2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Paired T.TEST requires at least two pairs".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut diffs = Vec::with_capacity(n1);
|
||||||
|
for i in 0..n1 {
|
||||||
|
diffs.push(values1[i] - values2[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
let nd = diffs.len();
|
||||||
|
let md = mean(&diffs);
|
||||||
|
let vd = sample_var(&diffs);
|
||||||
|
if vd <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Zero variance in paired T.TEST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let sd = vd.sqrt();
|
||||||
|
let t_stat = md / (sd / (nd as f64).sqrt());
|
||||||
|
let df = (nd - 1) as f64;
|
||||||
|
(t_stat, df)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2: two-sample, equal variance (homoscedastic)
|
||||||
|
TTestType::TwoSampleEqualVar => {
|
||||||
|
if n1 < 2 || n2 < 2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Two-sample T.TEST type 2 requires at least two values in each sample"
|
||||||
|
.to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let m1 = mean(&values1);
|
||||||
|
let m2 = mean(&values2);
|
||||||
|
let v1 = sample_var(&values1);
|
||||||
|
let v2 = sample_var(&values2);
|
||||||
|
|
||||||
|
let df_i = (n1 + n2 - 2) as i32;
|
||||||
|
if df_i <= 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Degrees of freedom must be positive in T.TEST type 2".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let df = df_i as f64;
|
||||||
|
|
||||||
|
let sp2 = (((n1 - 1) as f64) * v1 + ((n2 - 1) as f64) * v2) / df; // pooled variance
|
||||||
|
|
||||||
|
if sp2 <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Zero pooled variance in T.TEST type 2".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let denom = (sp2 * (1.0 / (n1 as f64) + 1.0 / (n2 as f64))).sqrt();
|
||||||
|
if denom == 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Zero denominator in T.TEST type 2".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let t_stat = (m1 - m2) / denom;
|
||||||
|
(t_stat, df)
|
||||||
|
}
|
||||||
|
|
||||||
|
// two-sample, unequal variance (Welch)
|
||||||
|
TTestType::TwoSampleUnequalVar => {
|
||||||
|
if n1 < 2 || n2 < 2 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Two-sample T.TEST type 3 requires at least two values in each sample"
|
||||||
|
.to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let m1 = mean(&values1);
|
||||||
|
let m2 = mean(&values2);
|
||||||
|
let v1 = sample_var(&values1);
|
||||||
|
let v2 = sample_var(&values2);
|
||||||
|
|
||||||
|
let s1n = v1 / (n1 as f64);
|
||||||
|
let s2n = v2 / (n2 as f64);
|
||||||
|
let denom = (s1n + s2n).sqrt();
|
||||||
|
if denom == 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Zero denominator in T.TEST type 3".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let t_stat = (m1 - m2) / denom;
|
||||||
|
|
||||||
|
let num_df = (s1n + s2n).powi(2);
|
||||||
|
let den_df = (s1n * s1n) / ((n1 - 1) as f64) + (s2n * s2n) / ((n2 - 1) as f64);
|
||||||
|
if den_df == 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Invalid degrees of freedom in T.TEST type 3".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let df = num_df / den_df;
|
||||||
|
(t_stat, df)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if df <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Degrees of freedom must be positive in T.TEST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dist = match StudentsT::new(0.0, 1.0, df) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Invalid parameters for Student's t distribution".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let t_abs = t_stat.abs();
|
||||||
|
let cdf = dist.cdf(t_abs);
|
||||||
|
|
||||||
|
let mut p = match tails {
|
||||||
|
TTestTails::OneTailed => 1.0 - cdf,
|
||||||
|
TTestTails::TwoTailed => 2.0 * (1.0 - cdf),
|
||||||
|
};
|
||||||
|
|
||||||
|
// clamp tiny fp noise
|
||||||
|
if p < 0.0 && p > -1e-15 {
|
||||||
|
p = 0.0;
|
||||||
|
}
|
||||||
|
if p > 1.0 && p < 1.0 + 1e-15 {
|
||||||
|
p = 1.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(p)
|
||||||
|
}
|
||||||
|
}
|
||||||
518
base/src/functions/statistical/variance.rs
Normal file
518
base/src/functions/statistical/variance.rs
Normal file
@@ -0,0 +1,518 @@
|
|||||||
|
use crate::constants::{LAST_COLUMN, LAST_ROW};
|
||||||
|
use crate::expressions::parser::ArrayNode;
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
pub(crate) fn fn_var_p(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"VAR.P with no numeric data".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / n;
|
||||||
|
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_var_s(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count <= 1 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"VAR.S requires at least two numeric values".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / (n - 1.0);
|
||||||
|
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_vara(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..=row2 {
|
||||||
|
for column in column1..=column2 {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::String(_) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, 0.0);
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(value) => {
|
||||||
|
let val = if value { 1.0 } else { 0.0 };
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, val);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now (A semantics to be added)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now (A semantics to be added)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now (A semantics to be added)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count <= 1 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"VARA requires at least two numeric values".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / (n - 1.0);
|
||||||
|
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn fn_varpa(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
if args.is_empty() {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut sumsq = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn accumulate(sum: &mut f64, sumsq: &mut f64, count: &mut u64, value: f64) {
|
||||||
|
*sum += value;
|
||||||
|
*sumsq += value * value;
|
||||||
|
*count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg in args {
|
||||||
|
match self.evaluate_node_in_context(arg, cell) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::Range { left, right } => {
|
||||||
|
if left.sheet != right.sheet {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Ranges are in different sheets".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let row1 = left.row;
|
||||||
|
let mut row2 = right.row;
|
||||||
|
let column1 = left.column;
|
||||||
|
let mut column2 = right.column;
|
||||||
|
|
||||||
|
if row1 == 1 && row2 == LAST_ROW {
|
||||||
|
row2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_row,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if column1 == 1 && column2 == LAST_COLUMN {
|
||||||
|
column2 = match self.workbook.worksheet(left.sheet) {
|
||||||
|
Ok(s) => s.dimension().max_column,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::ERROR,
|
||||||
|
cell,
|
||||||
|
format!("Invalid worksheet index: '{}'", left.sheet),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
for row in row1..row2 + 1 {
|
||||||
|
for column in column1..(column2 + 1) {
|
||||||
|
match self.evaluate_cell(CellReferenceIndex {
|
||||||
|
sheet: left.sheet,
|
||||||
|
row,
|
||||||
|
column,
|
||||||
|
}) {
|
||||||
|
CalcResult::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
CalcResult::String(_) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, 0.0);
|
||||||
|
}
|
||||||
|
CalcResult::Boolean(value) => {
|
||||||
|
let val = if value { 1.0 } else { 0.0 };
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, val);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
CalcResult::Array(array) => {
|
||||||
|
for row in array {
|
||||||
|
for value in row {
|
||||||
|
match value {
|
||||||
|
ArrayNode::Number(value) => {
|
||||||
|
accumulate(&mut sum, &mut sumsq, &mut count, value);
|
||||||
|
}
|
||||||
|
ArrayNode::Error(error) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error,
|
||||||
|
origin: cell,
|
||||||
|
message: "Error in array".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
// ignore non-numeric for now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"VARPA with no numeric data".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mut var = (sumsq - (sum * sum) / n) / n;
|
||||||
|
|
||||||
|
if var < 0.0 && var > -1e-12 {
|
||||||
|
var = 0.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(var)
|
||||||
|
}
|
||||||
|
}
|
||||||
71
base/src/functions/statistical/weibull.rs
Normal file
71
base/src/functions/statistical/weibull.rs
Normal file
@@ -0,0 +1,71 @@
|
|||||||
|
use statrs::distribution::{Continuous, ContinuousCDF, Weibull};
|
||||||
|
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::parser::Node, expressions::token::Error, model::Model,
|
||||||
|
};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// WEIBULL.DIST(x, alpha, beta, cumulative)
|
||||||
|
pub(crate) fn fn_weibull_dist(
|
||||||
|
&mut self,
|
||||||
|
args: &[Node],
|
||||||
|
cell: CellReferenceIndex,
|
||||||
|
) -> CalcResult {
|
||||||
|
if args.len() != 4 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let x = match self.get_number_no_bools(&args[0], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let alpha = match self.get_number_no_bools(&args[1], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let beta = match self.get_number_no_bools(&args[2], cell) {
|
||||||
|
Ok(f) => f,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
let cumulative = match self.get_boolean(&args[3], cell) {
|
||||||
|
Ok(b) => b,
|
||||||
|
Err(e) => return e,
|
||||||
|
};
|
||||||
|
|
||||||
|
if x < 0.0 || alpha <= 0.0 || beta <= 0.0 {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for WEIBULL.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// statrs::Weibull: shape = k (alpha), scale = lambda (beta)
|
||||||
|
let dist = match Weibull::new(alpha, beta) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid parameters for WEIBULL.DIST".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = if cumulative { dist.cdf(x) } else { dist.pdf(x) };
|
||||||
|
|
||||||
|
if !result.is_finite() {
|
||||||
|
return CalcResult::Error {
|
||||||
|
error: Error::NUM,
|
||||||
|
origin: cell,
|
||||||
|
message: "Invalid result for WEIBULL.DIST".to_string(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
171
base/src/functions/statistical/z_test.rs
Normal file
171
base/src/functions/statistical/z_test.rs
Normal file
@@ -0,0 +1,171 @@
|
|||||||
|
use statrs::distribution::{ContinuousCDF, Normal};
|
||||||
|
|
||||||
|
use crate::expressions::token::Error;
|
||||||
|
use crate::expressions::types::CellReferenceIndex;
|
||||||
|
use crate::{calc_result::CalcResult, expressions::parser::Node, model::Model};
|
||||||
|
|
||||||
|
impl<'a> Model<'a> {
|
||||||
|
// Z.TEST(array, x, [sigma])
|
||||||
|
pub(crate) fn fn_z_test(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
|
// 2 or 3 arguments
|
||||||
|
if args.len() < 2 || args.len() > 3 {
|
||||||
|
return CalcResult::new_args_number_error(cell);
|
||||||
|
}
|
||||||
|
|
||||||
|
let array_arg = self.evaluate_node_in_context(&args[0], cell);
|
||||||
|
|
||||||
|
// Flatten first argument into Vec<Option<f64>> (numeric / non-numeric)
|
||||||
|
let values = match array_arg {
|
||||||
|
CalcResult::Range { left, right } => match self.values_from_range(left, right) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => return error,
|
||||||
|
},
|
||||||
|
CalcResult::Array(array) => match self.values_from_array(array) {
|
||||||
|
Ok(v) => v,
|
||||||
|
Err(error) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
format!("Error in array argument: {:?}", error),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
CalcResult::Number(v) => vec![Some(v)],
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Z.TEST first argument must be a range or array".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Collect basic stats on numeric entries
|
||||||
|
let mut sum = 0.0;
|
||||||
|
let mut count: u64 = 0;
|
||||||
|
|
||||||
|
for x in values.iter().flatten() {
|
||||||
|
sum += x;
|
||||||
|
count += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Excel: if array has no numeric values -> #N/A
|
||||||
|
if count == 0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NA,
|
||||||
|
cell,
|
||||||
|
"Z.TEST array has no numeric data".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let n = count as f64;
|
||||||
|
let mean = sum / n;
|
||||||
|
|
||||||
|
// x argument (hypothesized population mean)
|
||||||
|
let x_value = match self.evaluate_node_in_context(&args[1], cell) {
|
||||||
|
CalcResult::Number(v) => v,
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Z.TEST second argument (x) must be numeric".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Optional sigma
|
||||||
|
let mut sigma: Option<f64> = None;
|
||||||
|
if args.len() == 3 {
|
||||||
|
match self.evaluate_node_in_context(&args[2], cell) {
|
||||||
|
CalcResult::Number(v) => {
|
||||||
|
if v == 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Z.TEST sigma cannot be zero".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
sigma = Some(v);
|
||||||
|
}
|
||||||
|
error @ CalcResult::Error { .. } => return error,
|
||||||
|
_ => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::VALUE,
|
||||||
|
cell,
|
||||||
|
"Z.TEST sigma (third argument) must be numeric".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If sigma omitted, use sample standard deviation STDEV(array)
|
||||||
|
let sigma_value = if let Some(s) = sigma {
|
||||||
|
s
|
||||||
|
} else {
|
||||||
|
// Excel: if only one numeric value and sigma omitted -> #DIV/0!
|
||||||
|
if count <= 1 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Z.TEST requires at least two values when sigma is omitted".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compute sum of squared deviations
|
||||||
|
let mut sumsq_dev = 0.0;
|
||||||
|
for x in values.iter().flatten() {
|
||||||
|
let d = x - mean;
|
||||||
|
sumsq_dev += d * d;
|
||||||
|
}
|
||||||
|
|
||||||
|
let var = sumsq_dev / (n - 1.0);
|
||||||
|
if var <= 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Z.TEST standard deviation is zero".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
var.sqrt()
|
||||||
|
};
|
||||||
|
|
||||||
|
// Compute z statistic: (mean - x) / (sigma / sqrt(n))
|
||||||
|
let denom = sigma_value / n.sqrt();
|
||||||
|
if denom == 0.0 {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::DIV,
|
||||||
|
cell,
|
||||||
|
"Z.TEST denominator is zero".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let z = (mean - x_value) / denom;
|
||||||
|
|
||||||
|
// Standard normal CDF
|
||||||
|
let dist = match Normal::new(0.0, 1.0) {
|
||||||
|
Ok(d) => d,
|
||||||
|
Err(_) => {
|
||||||
|
return CalcResult::new_error(
|
||||||
|
Error::NUM,
|
||||||
|
cell,
|
||||||
|
"Cannot create standard normal distribution in Z.TEST".to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut p = 1.0 - dist.cdf(z);
|
||||||
|
|
||||||
|
// clamp tiny FP noise
|
||||||
|
if p < 0.0 && p > -1e-15 {
|
||||||
|
p = 0.0;
|
||||||
|
}
|
||||||
|
if p > 1.0 && p < 1.0 + 1e-15 {
|
||||||
|
p = 1.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CalcResult::Number(p)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -32,7 +32,7 @@ pub enum CellTableStatus {
|
|||||||
Filtered,
|
Filtered,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
fn get_table_for_cell(&self, sheet_index: u32, row: i32, column: i32) -> bool {
|
fn get_table_for_cell(&self, sheet_index: u32, row: i32, column: i32) -> bool {
|
||||||
let worksheet = match self.workbook.worksheet(sheet_index) {
|
let worksheet = match self.workbook.worksheet(sheet_index) {
|
||||||
Ok(ws) => ws,
|
Ok(ws) => ws,
|
||||||
|
|||||||
@@ -50,7 +50,7 @@ fn search(search_for: &str, text: &str, start: usize) -> Option<i32> {
|
|||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
pub(crate) fn fn_concat(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
pub(crate) fn fn_concat(&mut self, args: &[Node], cell: CellReferenceIndex) -> CalcResult {
|
||||||
let mut result = "".to_string();
|
let mut result = "".to_string();
|
||||||
for arg in args {
|
for arg in args {
|
||||||
@@ -149,7 +149,7 @@ impl Model {
|
|||||||
Ok(s) => s,
|
Ok(s) => s,
|
||||||
Err(s) => return s,
|
Err(s) => return s,
|
||||||
};
|
};
|
||||||
let d = format_number(value, &format_code, &self.locale);
|
let d = format_number(value, &format_code, self.locale);
|
||||||
if let Some(_e) = d.error {
|
if let Some(_e) = d.error {
|
||||||
return CalcResult::Error {
|
return CalcResult::Error {
|
||||||
error: Error::VALUE,
|
error: Error::VALUE,
|
||||||
@@ -1210,7 +1210,15 @@ impl Model {
|
|||||||
match self.evaluate_node_in_context(&args[0], cell) {
|
match self.evaluate_node_in_context(&args[0], cell) {
|
||||||
CalcResult::String(text) => {
|
CalcResult::String(text) => {
|
||||||
let currencies = vec!["$", "€"];
|
let currencies = vec!["$", "€"];
|
||||||
if let Ok((value, _)) = parse_formatted_number(&text, ¤cies) {
|
let (decimal_separator, group_separator) =
|
||||||
|
if self.locale.numbers.symbols.decimal == "," {
|
||||||
|
(b',', b'.')
|
||||||
|
} else {
|
||||||
|
(b'.', b',')
|
||||||
|
};
|
||||||
|
if let Ok((value, _)) =
|
||||||
|
parse_formatted_number(&text, ¤cies, decimal_separator, group_separator)
|
||||||
|
{
|
||||||
return CalcResult::Number(value);
|
return CalcResult::Number(value);
|
||||||
};
|
};
|
||||||
CalcResult::Error {
|
CalcResult::Error {
|
||||||
|
|||||||
@@ -1,7 +1,10 @@
|
|||||||
#[cfg(feature = "use_regex_lite")]
|
#[cfg(feature = "use_regex_lite")]
|
||||||
use regex_lite as regex;
|
use regex_lite as regex;
|
||||||
|
|
||||||
use crate::{calc_result::CalcResult, expressions::token::is_english_error_string};
|
use crate::{
|
||||||
|
calc_result::CalcResult, expressions::token::is_english_error_string,
|
||||||
|
number_format::to_excel_precision,
|
||||||
|
};
|
||||||
|
|
||||||
/// This test for exact match (modulo case).
|
/// This test for exact match (modulo case).
|
||||||
/// * strings are not cast into bools or numbers
|
/// * strings are not cast into bools or numbers
|
||||||
@@ -34,6 +37,8 @@ pub(crate) fn values_are_equal(left: &CalcResult, right: &CalcResult) -> bool {
|
|||||||
pub(crate) fn compare_values(left: &CalcResult, right: &CalcResult) -> i32 {
|
pub(crate) fn compare_values(left: &CalcResult, right: &CalcResult) -> i32 {
|
||||||
match (left, right) {
|
match (left, right) {
|
||||||
(CalcResult::Number(value1), CalcResult::Number(value2)) => {
|
(CalcResult::Number(value1), CalcResult::Number(value2)) => {
|
||||||
|
let value1 = to_excel_precision(*value1, 15);
|
||||||
|
let value2 = to_excel_precision(*value2, 15);
|
||||||
if (value2 - value1).abs() < f64::EPSILON {
|
if (value2 - value1).abs() < f64::EPSILON {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -118,7 +118,7 @@ fn linear_search(
|
|||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Model {
|
impl<'a> Model<'a> {
|
||||||
/// The XLOOKUP function searches a range or an array, and then returns the item corresponding
|
/// The XLOOKUP function searches a range or an array, and then returns the item corresponding
|
||||||
/// to the first match it finds. If no match exists, then XLOOKUP can return the closest (approximate) match.
|
/// to the first match it finds. If no match exists, then XLOOKUP can return the closest (approximate) match.
|
||||||
/// =XLOOKUP(lookup_value, lookup_array, return_array, [if_not_found], [match_mode], [search_mode])
|
/// =XLOOKUP(lookup_value, lookup_array, return_array, [if_not_found], [match_mode], [search_mode])
|
||||||
|
|||||||
Binary file not shown.
File diff suppressed because it is too large
Load Diff
@@ -2,13 +2,13 @@ use std::{collections::HashMap, sync::OnceLock};
|
|||||||
|
|
||||||
use bitcode::{Decode, Encode};
|
use bitcode::{Decode, Encode};
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct Booleans {
|
pub struct Booleans {
|
||||||
pub r#true: String,
|
pub r#true: String,
|
||||||
pub r#false: String,
|
pub r#false: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct Errors {
|
pub struct Errors {
|
||||||
pub r#ref: String,
|
pub r#ref: String,
|
||||||
pub name: String,
|
pub name: String,
|
||||||
@@ -24,10 +24,367 @@ pub struct Errors {
|
|||||||
pub null: String,
|
pub null: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
|
pub struct Functions {
|
||||||
|
pub and: String,
|
||||||
|
pub r#false: String,
|
||||||
|
pub r#if: String,
|
||||||
|
pub iferror: String,
|
||||||
|
pub ifna: String,
|
||||||
|
pub ifs: String,
|
||||||
|
pub not: String,
|
||||||
|
pub or: String,
|
||||||
|
pub switch: String,
|
||||||
|
pub r#true: String,
|
||||||
|
pub xor: String,
|
||||||
|
pub log: String,
|
||||||
|
pub log10: String,
|
||||||
|
pub ln: String,
|
||||||
|
pub sin: String,
|
||||||
|
pub cos: String,
|
||||||
|
pub tan: String,
|
||||||
|
pub asin: String,
|
||||||
|
pub acos: String,
|
||||||
|
pub atan: String,
|
||||||
|
pub sinh: String,
|
||||||
|
pub cosh: String,
|
||||||
|
pub tanh: String,
|
||||||
|
pub asinh: String,
|
||||||
|
pub acosh: String,
|
||||||
|
pub atanh: String,
|
||||||
|
pub acot: String,
|
||||||
|
pub acoth: String,
|
||||||
|
pub cot: String,
|
||||||
|
pub coth: String,
|
||||||
|
pub csc: String,
|
||||||
|
pub csch: String,
|
||||||
|
pub sec: String,
|
||||||
|
pub sech: String,
|
||||||
|
pub abs: String,
|
||||||
|
pub pi: String,
|
||||||
|
pub sqrt: String,
|
||||||
|
pub sqrtpi: String,
|
||||||
|
pub atan2: String,
|
||||||
|
pub power: String,
|
||||||
|
pub max: String,
|
||||||
|
pub min: String,
|
||||||
|
pub product: String,
|
||||||
|
pub rand: String,
|
||||||
|
pub randbetween: String,
|
||||||
|
pub round: String,
|
||||||
|
pub rounddown: String,
|
||||||
|
pub roundup: String,
|
||||||
|
pub sum: String,
|
||||||
|
pub sumif: String,
|
||||||
|
pub sumifs: String,
|
||||||
|
pub choose: String,
|
||||||
|
pub column: String,
|
||||||
|
pub columns: String,
|
||||||
|
pub index: String,
|
||||||
|
pub indirect: String,
|
||||||
|
pub hlookup: String,
|
||||||
|
pub lookup: String,
|
||||||
|
pub r#match: String,
|
||||||
|
pub offset: String,
|
||||||
|
pub row: String,
|
||||||
|
pub rows: String,
|
||||||
|
pub vlookup: String,
|
||||||
|
pub xlookup: String,
|
||||||
|
pub concatenate: String,
|
||||||
|
pub exact: String,
|
||||||
|
pub value: String,
|
||||||
|
pub t: String,
|
||||||
|
pub valuetotext: String,
|
||||||
|
pub concat: String,
|
||||||
|
pub find: String,
|
||||||
|
pub left: String,
|
||||||
|
pub len: String,
|
||||||
|
pub lower: String,
|
||||||
|
pub mid: String,
|
||||||
|
pub right: String,
|
||||||
|
pub search: String,
|
||||||
|
pub text: String,
|
||||||
|
pub trim: String,
|
||||||
|
pub unicode: String,
|
||||||
|
pub upper: String,
|
||||||
|
pub isnumber: String,
|
||||||
|
pub isnontext: String,
|
||||||
|
pub istext: String,
|
||||||
|
pub islogical: String,
|
||||||
|
pub isblank: String,
|
||||||
|
pub iserr: String,
|
||||||
|
pub iserror: String,
|
||||||
|
pub isna: String,
|
||||||
|
pub na: String,
|
||||||
|
pub isref: String,
|
||||||
|
pub isodd: String,
|
||||||
|
pub iseven: String,
|
||||||
|
pub errortype: String,
|
||||||
|
pub formulatext: String,
|
||||||
|
pub isformula: String,
|
||||||
|
pub r#type: String,
|
||||||
|
pub sheet: String,
|
||||||
|
pub average: String,
|
||||||
|
pub averagea: String,
|
||||||
|
pub avedev: String,
|
||||||
|
pub averageif: String,
|
||||||
|
pub averageifs: String,
|
||||||
|
pub count: String,
|
||||||
|
pub counta: String,
|
||||||
|
pub countblank: String,
|
||||||
|
pub countif: String,
|
||||||
|
pub countifs: String,
|
||||||
|
pub maxifs: String,
|
||||||
|
pub minifs: String,
|
||||||
|
pub geomean: String,
|
||||||
|
pub year: String,
|
||||||
|
pub day: String,
|
||||||
|
pub month: String,
|
||||||
|
pub eomonth: String,
|
||||||
|
pub date: String,
|
||||||
|
pub datedif: String,
|
||||||
|
pub datevalue: String,
|
||||||
|
pub edate: String,
|
||||||
|
pub networkdays: String,
|
||||||
|
pub networkdaysintl: String,
|
||||||
|
pub time: String,
|
||||||
|
pub timevalue: String,
|
||||||
|
pub hour: String,
|
||||||
|
pub minute: String,
|
||||||
|
pub second: String,
|
||||||
|
pub today: String,
|
||||||
|
pub now: String,
|
||||||
|
pub days: String,
|
||||||
|
pub days360: String,
|
||||||
|
pub weekday: String,
|
||||||
|
pub weeknum: String,
|
||||||
|
pub workday: String,
|
||||||
|
pub workdayintl: String,
|
||||||
|
pub yearfrac: String,
|
||||||
|
pub isoweeknum: String,
|
||||||
|
pub pmt: String,
|
||||||
|
pub pv: String,
|
||||||
|
pub rate: String,
|
||||||
|
pub nper: String,
|
||||||
|
pub fv: String,
|
||||||
|
pub ppmt: String,
|
||||||
|
pub ipmt: String,
|
||||||
|
pub npv: String,
|
||||||
|
pub mirr: String,
|
||||||
|
pub irr: String,
|
||||||
|
pub xirr: String,
|
||||||
|
pub xnpv: String,
|
||||||
|
pub rept: String,
|
||||||
|
pub textafter: String,
|
||||||
|
pub textbefore: String,
|
||||||
|
pub textjoin: String,
|
||||||
|
pub substitute: String,
|
||||||
|
pub ispmt: String,
|
||||||
|
pub rri: String,
|
||||||
|
pub sln: String,
|
||||||
|
pub syd: String,
|
||||||
|
pub nominal: String,
|
||||||
|
pub effect: String,
|
||||||
|
pub pduration: String,
|
||||||
|
pub tbillyield: String,
|
||||||
|
pub tbillprice: String,
|
||||||
|
pub tbilleq: String,
|
||||||
|
pub dollarde: String,
|
||||||
|
pub dollarfr: String,
|
||||||
|
pub ddb: String,
|
||||||
|
pub db: String,
|
||||||
|
pub cumprinc: String,
|
||||||
|
pub cumipmt: String,
|
||||||
|
pub besseli: String,
|
||||||
|
pub besselj: String,
|
||||||
|
pub besselk: String,
|
||||||
|
pub bessely: String,
|
||||||
|
pub erf: String,
|
||||||
|
pub erfprecise: String,
|
||||||
|
pub erfc: String,
|
||||||
|
pub erfcprecise: String,
|
||||||
|
pub bin2dec: String,
|
||||||
|
pub bin2hex: String,
|
||||||
|
pub bin2oct: String,
|
||||||
|
pub dec2bin: String,
|
||||||
|
pub dec2hex: String,
|
||||||
|
pub dec2oct: String,
|
||||||
|
pub hex2bin: String,
|
||||||
|
pub hex2dec: String,
|
||||||
|
pub hex2oct: String,
|
||||||
|
pub oct2bin: String,
|
||||||
|
pub oct2dec: String,
|
||||||
|
pub oct2hex: String,
|
||||||
|
pub bitand: String,
|
||||||
|
pub bitlshift: String,
|
||||||
|
pub bitor: String,
|
||||||
|
pub bitrshift: String,
|
||||||
|
pub bitxor: String,
|
||||||
|
pub complex: String,
|
||||||
|
pub imabs: String,
|
||||||
|
pub imaginary: String,
|
||||||
|
pub imargument: String,
|
||||||
|
pub imconjugate: String,
|
||||||
|
pub imcos: String,
|
||||||
|
pub imcosh: String,
|
||||||
|
pub imcot: String,
|
||||||
|
pub imcsc: String,
|
||||||
|
pub imcsch: String,
|
||||||
|
pub imdiv: String,
|
||||||
|
pub imexp: String,
|
||||||
|
pub imln: String,
|
||||||
|
pub imlog10: String,
|
||||||
|
pub imlog2: String,
|
||||||
|
pub impower: String,
|
||||||
|
pub improduct: String,
|
||||||
|
pub imreal: String,
|
||||||
|
pub imsec: String,
|
||||||
|
pub imsech: String,
|
||||||
|
pub imsin: String,
|
||||||
|
pub imsinh: String,
|
||||||
|
pub imsqrt: String,
|
||||||
|
pub imsub: String,
|
||||||
|
pub imsum: String,
|
||||||
|
pub imtan: String,
|
||||||
|
pub convert: String,
|
||||||
|
pub delta: String,
|
||||||
|
pub gestep: String,
|
||||||
|
pub subtotal: String,
|
||||||
|
pub exp: String,
|
||||||
|
pub fact: String,
|
||||||
|
pub factdouble: String,
|
||||||
|
pub sign: String,
|
||||||
|
pub radians: String,
|
||||||
|
pub degrees: String,
|
||||||
|
pub int: String,
|
||||||
|
pub even: String,
|
||||||
|
pub odd: String,
|
||||||
|
pub ceiling: String,
|
||||||
|
pub ceilingmath: String,
|
||||||
|
pub ceilingprecise: String,
|
||||||
|
pub floor: String,
|
||||||
|
pub floormath: String,
|
||||||
|
pub floorprecise: String,
|
||||||
|
pub isoceiling: String,
|
||||||
|
pub r#mod: String,
|
||||||
|
pub quotient: String,
|
||||||
|
pub mround: String,
|
||||||
|
pub trunc: String,
|
||||||
|
pub gcd: String,
|
||||||
|
pub lcm: String,
|
||||||
|
pub base: String,
|
||||||
|
pub decimal: String,
|
||||||
|
pub roman: String,
|
||||||
|
pub arabic: String,
|
||||||
|
pub combin: String,
|
||||||
|
pub combina: String,
|
||||||
|
pub sumsq: String,
|
||||||
|
pub n: String,
|
||||||
|
pub cell: String,
|
||||||
|
pub info: String,
|
||||||
|
pub sheets: String,
|
||||||
|
pub daverage: String,
|
||||||
|
pub dcount: String,
|
||||||
|
pub dget: String,
|
||||||
|
pub dmax: String,
|
||||||
|
pub dmin: String,
|
||||||
|
pub dsum: String,
|
||||||
|
pub dcounta: String,
|
||||||
|
pub dproduct: String,
|
||||||
|
pub dstdev: String,
|
||||||
|
pub dvar: String,
|
||||||
|
pub dvarp: String,
|
||||||
|
pub dstdevp: String,
|
||||||
|
pub betadist: String,
|
||||||
|
pub betainv: String,
|
||||||
|
pub binomdist: String,
|
||||||
|
pub binomdistrange: String,
|
||||||
|
pub binominv: String,
|
||||||
|
pub chisqdist: String,
|
||||||
|
pub chisqdistrt: String,
|
||||||
|
pub chisqinv: String,
|
||||||
|
pub chisqinvrt: String,
|
||||||
|
pub chisqtest: String,
|
||||||
|
pub confidencenorm: String,
|
||||||
|
pub confidencet: String,
|
||||||
|
pub covariancep: String,
|
||||||
|
pub covariances: String,
|
||||||
|
pub devsq: String,
|
||||||
|
pub expondist: String,
|
||||||
|
pub fdist: String,
|
||||||
|
pub fdistrt: String,
|
||||||
|
pub finv: String,
|
||||||
|
pub finvrt: String,
|
||||||
|
pub fisher: String,
|
||||||
|
pub fisherinv: String,
|
||||||
|
pub ftest: String,
|
||||||
|
pub gamma: String,
|
||||||
|
pub gammadist: String,
|
||||||
|
pub gammainv: String,
|
||||||
|
pub gammaln: String,
|
||||||
|
pub gammalnprecise: String,
|
||||||
|
pub hypgeomdist: String,
|
||||||
|
pub lognormdist: String,
|
||||||
|
pub lognorminv: String,
|
||||||
|
pub negbinomdist: String,
|
||||||
|
pub normdist: String,
|
||||||
|
pub norminv: String,
|
||||||
|
pub normsdist: String,
|
||||||
|
pub normsinv: String,
|
||||||
|
pub pearson: String,
|
||||||
|
pub phi: String,
|
||||||
|
pub poissondist: String,
|
||||||
|
pub standardize: String,
|
||||||
|
pub stdevp: String,
|
||||||
|
pub stdevs: String,
|
||||||
|
pub stdeva: String,
|
||||||
|
pub stdevpa: String,
|
||||||
|
pub tdist: String,
|
||||||
|
pub tdist2t: String,
|
||||||
|
pub tdistrt: String,
|
||||||
|
pub tinv: String,
|
||||||
|
pub tinv2t: String,
|
||||||
|
pub ttest: String,
|
||||||
|
pub varp: String,
|
||||||
|
pub vars: String,
|
||||||
|
pub varpa: String,
|
||||||
|
pub vara: String,
|
||||||
|
pub weibulldist: String,
|
||||||
|
pub ztest: String,
|
||||||
|
pub sumx2my2: String,
|
||||||
|
pub sumx2py2: String,
|
||||||
|
pub sumxmy2: String,
|
||||||
|
pub correl: String,
|
||||||
|
pub rsq: String,
|
||||||
|
pub intercept: String,
|
||||||
|
pub slope: String,
|
||||||
|
pub steyx: String,
|
||||||
|
pub gauss: String,
|
||||||
|
pub harmean: String,
|
||||||
|
pub kurt: String,
|
||||||
|
pub large: String,
|
||||||
|
pub maxa: String,
|
||||||
|
pub median: String,
|
||||||
|
pub mina: String,
|
||||||
|
pub rankavg: String,
|
||||||
|
pub rankeq: String,
|
||||||
|
pub skew: String,
|
||||||
|
pub skewp: String,
|
||||||
|
pub small: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Encode, Decode)]
|
||||||
pub struct Language {
|
pub struct Language {
|
||||||
|
pub name: String,
|
||||||
|
pub code: String,
|
||||||
pub booleans: Booleans,
|
pub booleans: Booleans,
|
||||||
pub errors: Errors,
|
pub errors: Errors,
|
||||||
|
pub functions: Functions,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_default_language() -> &'static Language {
|
||||||
|
#[allow(clippy::unwrap_used)]
|
||||||
|
get_language("en").unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
static LANGUAGES: OnceLock<HashMap<String, Language>> = OnceLock::new();
|
static LANGUAGES: OnceLock<HashMap<String, Language>> = OnceLock::new();
|
||||||
@@ -39,7 +396,7 @@ fn get_languages() -> &'static HashMap<String, Language> {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_language(id: &str) -> Result<&Language, String> {
|
pub fn get_language(id: &str) -> Result<&'static Language, String> {
|
||||||
get_languages()
|
get_languages()
|
||||||
.get(id)
|
.get(id)
|
||||||
.ok_or_else(|| format!("Language is not supported: '{id}'"))
|
.ok_or_else(|| format!("Language is not supported: '{id}'"))
|
||||||
|
|||||||
@@ -57,8 +57,10 @@ mod test;
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
pub mod mock_time;
|
pub mod mock_time;
|
||||||
|
|
||||||
|
pub use locale::get_supported_locales;
|
||||||
pub use model::get_milliseconds_since_epoch;
|
pub use model::get_milliseconds_since_epoch;
|
||||||
pub use model::Model;
|
pub use model::Model;
|
||||||
pub use user_model::BorderArea;
|
pub use user_model::BorderArea;
|
||||||
pub use user_model::ClipboardData;
|
pub use user_model::ClipboardData;
|
||||||
pub use user_model::UserModel;
|
pub use user_model::UserModel;
|
||||||
|
pub use utils::get_all_timezones;
|
||||||
|
|||||||
Binary file not shown.
File diff suppressed because one or more lines are too long
@@ -2,36 +2,39 @@ use std::{collections::HashMap, sync::OnceLock};
|
|||||||
|
|
||||||
use bitcode::{Decode, Encode};
|
use bitcode::{Decode, Encode};
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct Locale {
|
pub struct Locale {
|
||||||
pub dates: Dates,
|
pub dates: Dates,
|
||||||
pub numbers: NumbersProperties,
|
pub numbers: NumbersProperties,
|
||||||
pub currency: Currency,
|
pub currency: Currency,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct Currency {
|
pub struct Currency {
|
||||||
pub iso: String,
|
pub iso: String,
|
||||||
pub symbol: String,
|
pub symbol: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct NumbersProperties {
|
pub struct NumbersProperties {
|
||||||
pub symbols: NumbersSymbols,
|
pub symbols: NumbersSymbols,
|
||||||
pub decimal_formats: DecimalFormats,
|
pub decimal_formats: DecimalFormats,
|
||||||
pub currency_formats: CurrencyFormats,
|
pub currency_formats: CurrencyFormats,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct Dates {
|
pub struct Dates {
|
||||||
pub day_names: Vec<String>,
|
pub day_names: Vec<String>,
|
||||||
pub day_names_short: Vec<String>,
|
pub day_names_short: Vec<String>,
|
||||||
pub months: Vec<String>,
|
pub months: Vec<String>,
|
||||||
pub months_short: Vec<String>,
|
pub months_short: Vec<String>,
|
||||||
pub months_letter: Vec<String>,
|
pub months_letter: Vec<String>,
|
||||||
|
pub date_formats: DateFormats,
|
||||||
|
pub time_formats: DateFormats,
|
||||||
|
pub date_time_formats: DateFormats,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct NumbersSymbols {
|
pub struct NumbersSymbols {
|
||||||
pub decimal: String,
|
pub decimal: String,
|
||||||
pub group: String,
|
pub group: String,
|
||||||
@@ -49,7 +52,7 @@ pub struct NumbersSymbols {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// See: https://cldr.unicode.org/translation/number-currency-formats/number-and-currency-patterns
|
// See: https://cldr.unicode.org/translation/number-currency-formats/number-and-currency-patterns
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct CurrencyFormats {
|
pub struct CurrencyFormats {
|
||||||
pub standard: String,
|
pub standard: String,
|
||||||
pub standard_alpha_next_to_number: Option<String>,
|
pub standard_alpha_next_to_number: Option<String>,
|
||||||
@@ -59,11 +62,32 @@ pub struct CurrencyFormats {
|
|||||||
pub accounting_no_currency: String,
|
pub accounting_no_currency: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Encode, Decode, Clone)]
|
#[derive(Encode, Decode)]
|
||||||
pub struct DecimalFormats {
|
pub struct DecimalFormats {
|
||||||
pub standard: String,
|
pub standard: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Encode, Decode)]
|
||||||
|
pub struct DateFormats {
|
||||||
|
pub full: String,
|
||||||
|
pub long: String,
|
||||||
|
pub medium: String,
|
||||||
|
pub short: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Encode, Decode)]
|
||||||
|
pub struct TimeFormats {
|
||||||
|
pub full: String,
|
||||||
|
pub long: String,
|
||||||
|
pub medium: String,
|
||||||
|
pub short: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_default_locale() -> &'static Locale {
|
||||||
|
#[allow(clippy::unwrap_used)]
|
||||||
|
get_locale("en").unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
static LOCALES: OnceLock<HashMap<String, Locale>> = OnceLock::new();
|
static LOCALES: OnceLock<HashMap<String, Locale>> = OnceLock::new();
|
||||||
|
|
||||||
#[allow(clippy::expect_used)]
|
#[allow(clippy::expect_used)]
|
||||||
@@ -73,7 +97,12 @@ fn get_locales() -> &'static HashMap<String, Locale> {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_locale(id: &str) -> Result<&Locale, String> {
|
/// Get all available locale IDs.
|
||||||
|
pub fn get_supported_locales() -> Vec<String> {
|
||||||
|
get_locales().keys().cloned().collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_locale(id: &str) -> Result<&'static Locale, String> {
|
||||||
get_locales()
|
get_locales()
|
||||||
.get(id)
|
.get(id)
|
||||||
.ok_or_else(|| format!("Invalid locale: '{id}'"))
|
.ok_or_else(|| format!("Invalid locale: '{id}'"))
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user