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}