maitakedayo_rpn_calc/
lib.rs

1// ドキュメントを記述 --- (*1)
2//! # RPN Calc
3//! Reverse Polish notation (RPN) Calc.
4//! # Example
5//! ```
6//! let src = String::from("1 2 + 3 * ");
7//! let a = rpn_calc::eval(src).unwrap();
8//! println!("{}", a); // →9
9//! ```
10
11/// RPN式を計算する関数
12pub fn eval(src: String) -> Result<f64, String> {
13    // 空白でトークンを区切る
14    let tokens = src.split_whitespace();
15    let mut stack:Vec<f64> = vec![];
16    // 繰り返し要素を計算 --- (*2)
17    for tok in tokens {
18        let t = tok.trim();
19        if t == "" { continue; }
20        // 数値ならスタックにPUSH
21        match t.parse::<f64>() {
22            Ok(v) => { stack.push(v); continue; },
23            Err(_) => 0.0,
24        };
25        // 演算子なら2回POPして計算結果をPUSH
26        let b = stack.pop().unwrap_or(0.0);
27        let a = stack.pop().unwrap_or(0.0);
28        match t {
29            "+" => stack.push(a + b),
30            "-" => stack.push(a - b),
31            "*" => stack.push(a * b),
32            "/" => stack.push(a / b),
33            "%" => stack.push(a % b),
34            _ => return Err(format!("invalid operator: {}", t)),
35        }
36    }
37    // 結果を返す --- (*3)
38    if stack.len() == 0 { return Err(format!("no result")); }
39    if stack.len() > 1 { 
40        return Err(format!("too many value in stack")); 
41    } 
42    Ok(stack.pop().unwrap_or(0.0))
43}
44
45/// より手軽に使う
46pub fn eval_str(src: &str) -> String {
47    match eval(String::from(src)) {
48        Ok(v) => format!("{}", v),
49        Err(e) => format!("[ERROR] {}", e),
50    }
51}
52
53// ライブラリを利用するテストを記述 --- (*4)
54#[cfg(test)]
55mod tests {
56    use super::*;
57    #[test]
58    fn it_works() {
59        // eval を使う
60        assert_eq!(eval("1 3 +".to_string()), Ok(4.0));
61        assert_eq!(eval("2 3 *".to_string()), Ok(6.0));
62        assert_eq!(eval("6 3 /".to_string()), Ok(2.0));
63        assert_eq!(eval("6 3 - 1 -".to_string()), Ok(2.0));
64        // eval_str を使う
65        assert_eq!(eval_str("1 2 3 + +"), "6".to_string());
66        assert_eq!(eval_str("1 2 3 * +"), "7".to_string());
67    }
68}
69
maitakedayo_rpn_calc/lib.rs

maitakedayo_rpn_calc/
lib.rs
