ganit-core 0.4.1

Spreadsheet formula engine — parser and evaluator for Excel-compatible formulas
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use crate::eval::coercion::to_number;
use crate::eval::functions::check_arity;
use crate::types::{ErrorKind, Value};

/// `MUNIT(dimension)` — returns an n×n identity matrix as a nested Array.
pub fn munit_fn(args: &[Value]) -> Value {
    if let Some(err) = check_arity(args, 1, 1) {
        return err;
    }
    let n = match to_number(args[0].clone()) {
        Err(e) => return e,
        Ok(v) => v,
    };
    if n <= 0.0 {
        return Value::Error(ErrorKind::Num);
    }
    let n = n as usize;
    let rows: Vec<Value> = (0..n)
        .map(|r| {
            let row: Vec<Value> = (0..n)
                .map(|c| Value::Number(if r == c { 1.0 } else { 0.0 }))
                .collect();
            Value::Array(row)
        })
        .collect();
    Value::Array(rows)
}