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

#![feature(external_doc)]
use tsf_rs::TSF;
use ta_common::traits::Indicator;


#[doc(include = "../README.md")]
pub struct FOSC {
    period: u32,
    prev_tsf: Option<f64>,
    tsf: TSF,
}


impl FOSC {
    pub fn new(period: u32) -> FOSC {
        Self {
            period,
            prev_tsf: None,
            tsf: TSF::new(period),
        }
    }
}

impl Indicator<f64, Option<f64>> for FOSC {
    fn next(&mut self, input: f64) -> Option<f64> {
        let res = match self.prev_tsf {
            None => None,
            Some(prev) => {
                let fosc = 100.0 * ((input - prev) / input);
                Some(fosc)
            }
        };
        self.prev_tsf = self.tsf.next(input);
        res
    }

    fn reset(&mut self) {
        self.prev_tsf = None;
        self.tsf.reset();
    }
}


#[cfg(test)]
mod tests {
    use crate::FOSC;
    use ta_common::traits::Indicator;

    #[test]
    fn it_works() {
        let mut fosc = FOSC::new(5);
        assert_eq!(fosc.next(81.59), None);
        assert_eq!(fosc.next(81.06), None);
        assert_eq!(fosc.next(82.87), None);
        assert_eq!(fosc.next(83.00), None);
        assert_eq!(fosc.next(83.61), None);
        assert_eq!(fosc.next(83.15), Some(-1.2868310282621687));
        assert_eq!(fosc.next(82.84), Some(-1.6586190246257615));
        assert_eq!(fosc.next(83.99), Some(1.0346469817835624));
        assert_eq!(fosc.next(84.55), Some(1.0277942046126718));
        assert_eq!(fosc.next(84.36), Some(-0.0995732574679644));
        assert_eq!(fosc.next(85.53), Some(0.5997895475271895));
        assert_eq!(fosc.next(86.54), Some(0.6482551421308311));
        assert_eq!(fosc.next(86.89), Some(0.08286339049376355));
        assert_eq!(fosc.next(87.77), Some(0.157229121567756));
        assert_eq!(fosc.next(87.29), Some(-1.5832283193950765));
    }
}