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