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

use crate::aroon::AROON;
use ta_common::traits::Indicator;
#[doc(include = "../docs/aroon.md")]
pub struct AROONOSC {
    period: u32,
    aroon: AROON,

}

impl AROONOSC {
    pub fn new(period: u32) -> AROONOSC {
        Self {
            period,
            aroon: AROON::new(period),
        }
    }
}

impl Indicator<[f64; 2], Option<f64>> for AROONOSC {
    fn next(&mut self, input: [f64; 2]) -> Option<f64> {
        let aroon = self.aroon.next(input);
        return aroon.map(|v| v[1] - v[0]);
    }

    fn reset(&mut self) {
        self.aroon.reset();
    }
}


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

    use crate::aroonosc::AROONOSC;

    #[test]
    fn aaron_works() {
        let mut aroonosc = AROONOSC::new(5);
        assert_eq!(aroonosc.next([82.15, 81.29, ]), None);
        assert_eq!(aroonosc.next([81.89, 80.64, ]), None);
        assert_eq!(aroonosc.next([83.03, 81.31, ]), None);
        assert_eq!(aroonosc.next([83.30, 82.65, ]), None);
        assert_eq!(aroonosc.next([83.85, 83.07, ]), None);
        assert_eq!(aroonosc.next([83.90, 83.11, ]), Some(80.00));
        assert_eq!(aroonosc.next([83.33, 82.49, ]), Some(60.00));
        assert_eq!(aroonosc.next([84.30, 82.30, ]), Some(0.00));
        assert_eq!(aroonosc.next([84.84, 84.15, ]), Some(20.00));
        assert_eq!(aroonosc.next([85.00, 84.11, ]), Some(40.00));
        assert_eq!(aroonosc.next([85.90, 84.03, ]), Some(60.00));
        assert_eq!(aroonosc.next([86.58, 85.39, ]), Some(80.00));
        assert_eq!(aroonosc.next([86.98, 85.76, ]), Some(40.00));
        assert_eq!(aroonosc.next([88.00, 87.17, ]), Some(60.00));
        assert_eq!(aroonosc.next([87.87, 87.01, ]), Some(60.00));
    }
}