ruut_functions/
param.rs

1use crate::{Func, F1D, F2D, F3D};
2
3impl Func {
4    fn set_par(&mut self, name: &str, val: f64) {
5        match self {
6            Func::Param(c, v) if c == name => {
7                *v = val;
8            }
9            Func::Add(arr) | Func::Mul(arr) => {
10                for el in arr {
11                    el.set_par(name, val)
12                }
13            }
14            Func::Pow(base, exp) => {
15                base.set_par(name, val);
16                exp.set_par(name, val);
17            }
18            Func::S(_, arg) => arg.set_par(name, val),
19            _ => (),
20        }
21    }
22}
23
24impl F1D {
25    /// Set param value by name
26    /// ```
27    /// use ruut_functions::{f1d, F1D};
28    ///
29    /// let mut f = f1d!("x+[a]^2");
30    /// f.set_par("a", 6.9);
31    /// assert!(f.eval(4.20) - 51.81 < 0.0001);
32    /// f.set_par("a", 1.2);
33    /// assert!(f.eval(4.20) - 5.64 < 0.0001);
34    /// ```
35    pub fn set_par(&mut self, name: &str, val: f64) {
36        self.0.set_par(name, val)
37    }
38}
39impl F2D {
40    /// Set param value by name
41    /// ```
42    /// use ruut_functions::{f2d, F2D};
43    ///
44    /// let mut f = f2d!("x+y[a]^2"); // Default value of param in 0
45    /// f.set_par("a", 6.9);
46    /// assert!(f.eval(4.20, 1.0) - 51.81 < 0.0001);
47    /// f.set_par("a", 1.2);
48    /// assert!(f.eval(4.20, 1.0) - 5.64 < 0.0001);
49    /// ```
50    pub fn set_par(&mut self, name: &str, val: f64) {
51        self.0.set_par(name, val)
52    }
53}
54impl F3D {
55    /// Set param value by name
56    /// ```
57    /// use ruut_functions::{f3d, F3D};
58    ///
59    /// let mut f = f3d!("x+y[a]^2+z"); // Default value of param in 0
60    /// f.set_par("a", 6.9);
61    /// assert!(f.eval(4.20, 1.0,7.0) - 58.81 < 0.0001);
62    /// f.set_par("a", 1.2);
63    /// assert!(f.eval(4.20, 1.0, 7.0) - 12.64 < 0.0001);
64    /// ```
65    pub fn set_par(&mut self, name: &str, val: f64) {
66        self.0.set_par(name, val)
67    }
68}
69
70#[test]
71fn test_param() {
72    use crate::{f1d, f2d, f3d};
73    let mut f = f1d!("[a]^2+x[a]+ln([a])");
74    f.set_par("a", 6.);
75    assert!(f.eval(0.) - 37.791 < 0.001);
76
77    let mut f = f2d!("e^(-[eta]xy)");
78    f.set_par("a", 12.);
79    assert_eq!(f.eval(1., 1.), 1.);
80    f.set_par("eta", 0.5);
81    assert!(f.eval(1., 1.) - 0.60653 < 0.00001);
82
83    let mut f = f3d!("sin(xyz[b])");
84    f.set_par("b", 1.);
85}