quantwave-core 0.8.0

A high-performance, Polars-native technical analysis library for Rust.
Documentation 
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

//! Native O(1) streaming CDL3LINESTRIKE (matches `talib_rs::pattern::cdl_3linestrike`).

use crate::indicators::patterns::candle_settings::{
    CandleWindow, NEAR, RollingCandleAvg, candle_color,
};
use crate::traits::Next;

const LOOKBACK: usize = 5 + 3; // NEAR.avg_period (5) + 3

#[derive(Debug, Clone)]
#[allow(non_camel_case_types)]
pub struct CDL3LINESTRIKE {
    bars_seen: usize,
    window: CandleWindow,
    near_avg: RollingCandleAvg,
}

impl CDL3LINESTRIKE {
    pub fn new() -> Self {
        Self {
            bars_seen: 0,
            window: CandleWindow::new(4), // Needs i, i-1, i-2, i-3 => capacity 4
            near_avg: RollingCandleAvg::new(NEAR),
        }
    }
}

impl Default for CDL3LINESTRIKE {
    fn default() -> Self {
        Self::new()
    }
}

impl Next<(f64, f64, f64, f64)> for CDL3LINESTRIKE {
    type Output = f64;

    fn next(&mut self, (open, high, low, close): (f64, f64, f64, f64)) -> Self::Output {
        self.bars_seen += 1;

        self.window.push(open, high, low, close);
        self.near_avg.push(open, high, low, close);

        if self.bars_seen <= LOOKBACK {
            return 0.0;
        }

        let c3 = candle_color(self.window.bar(3).open, self.window.bar(3).close);
        let c2 = candle_color(self.window.bar(2).open, self.window.bar(2).close);
        let c1 = candle_color(self.window.bar(1).open, self.window.bar(1).close);
        let c0 = candle_color(self.window.bar(0).open, self.window.bar(0).close);

        if c3 == c2 && c2 == c1 && c0 != c1 {
            let progressive = if c3 == 1 {
                self.window.bar(2).close > self.window.bar(3).close
                    && self.window.bar(1).close > self.window.bar(2).close
            } else {
                self.window.bar(2).close < self.window.bar(3).close
                    && self.window.bar(1).close < self.window.bar(2).close
            };

            let opens_near = if c3 == 1 {
                self.window.bar(2).open >= self.window.bar(3).open.min(self.window.bar(3).close)
                    && self.window.bar(2).open <= self.window.bar(3).close + self.near_avg.val(3)
                    && self.window.bar(1).open
                        >= self.window.bar(2).open.min(self.window.bar(2).close)
                    && self.window.bar(1).open <= self.window.bar(2).close + self.near_avg.val(2)
            } else {
                self.window.bar(2).open <= self.window.bar(3).open.max(self.window.bar(3).close)
                    && self.window.bar(2).open >= self.window.bar(3).close - self.near_avg.val(3)
                    && self.window.bar(1).open
                        <= self.window.bar(2).open.max(self.window.bar(2).close)
                    && self.window.bar(1).open >= self.window.bar(2).close - self.near_avg.val(2)
            };

            let strike = if c3 == 1 {
                self.window.bar(0).open >= self.window.bar(1).close
                    && self.window.bar(0).close <= self.window.bar(3).open
            } else {
                self.window.bar(0).open <= self.window.bar(1).close
                    && self.window.bar(0).close >= self.window.bar(3).open
            };

            if progressive && opens_near && strike {
                return (c3 * 100) as f64;
            }
        }

        0.0
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    crate::test_pattern_parity!(
        test_cdl3linestrike_parity,
        CDL3LINESTRIKE,
        talib_rs::pattern::cdl_3linestrike
    );
}