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
pub fn five_point_derivitive(data: &[f64]) -> Vec<f64>
{
let mut d = vec![f64::NAN; data.len()];
if data.len() >= 5 {
for i in 2..data.len()-2 {
let mut tmp = data[i-1].mul_add(-8.0, data[i-2]);
tmp = data[i+1].mul_add(8.0, tmp) - data[i+2];
d[i] = tmp / 12.0;
}
}
d
}
fn derivative(data: &[f64]) -> Vec<f64>
{
let mut d = vec![f64::NAN; data.len()];
if data.len() >= 3 {
for i in 1..data.len()-1 {
d[i] = (data[i+1] - data[i-1]) / 2.0;
}
}
if data.len() >= 2 {
d[0] = (data[1] - data[0]) / 2.0;
d[data.len() - 1] = (data[data.len() - 1] - data[data.len() - 2]) / 2.0;
}
d
}
pub fn derivative_merged(data: &[f64]) -> Vec<f64>
{
if data.len() < 5 {
return derivative(data);
}
let mut d = five_point_derivitive(data);
d[1] = (data[2] - data[0]) / 2.0;
d[data.len() - 2] = (data[data.len() - 1] - data[data.len() - 3]) / 2.0;
d[0] = (data[1] - data[0]) / 2.0;
d[data.len() - 1] = (data[data.len() - 1] - data[data.len() - 2]) / 2.0;
d
}