wickra_core/indicators/
high_wave.rs1use crate::ohlcv::Candle;
4use crate::traits::Indicator;
5
6#[derive(Debug, Clone, Default)]
40pub struct HighWave {
41 has_emitted: bool,
42}
43
44impl HighWave {
45 pub const fn new() -> Self {
47 Self { has_emitted: false }
48 }
49}
50
51impl Indicator for HighWave {
52 type Input = Candle;
53 type Output = f64;
54
55 #[inline]
56 fn update(&mut self, candle: Candle) -> Option<f64> {
57 self.has_emitted = true;
58 let range = candle.high - candle.low;
59 if range <= 0.0 {
60 return Some(0.0);
61 }
62 let upper = candle.high - candle.open.max(candle.close);
63 let lower = candle.open.min(candle.close) - candle.low;
64 if upper >= 0.4 * range && lower >= 0.4 * range {
65 return Some(1.0);
66 }
67 Some(0.0)
68 }
69
70 fn reset(&mut self) {
71 self.has_emitted = false;
72 }
73
74 #[inline]
75 fn warmup_period(&self) -> usize {
76 1
77 }
78
79 #[inline]
80 fn is_ready(&self) -> bool {
81 self.has_emitted
82 }
83
84 #[inline]
85 fn name(&self) -> &'static str {
86 "HighWave"
87 }
88}
89
90#[cfg(test)]
91mod tests {
92 use super::*;
93 use crate::traits::BatchExt;
94
95 fn c(open: f64, high: f64, low: f64, close: f64, ts: i64) -> Candle {
96 Candle::new(open, high, low, close, 1.0, ts).unwrap()
97 }
98
99 #[test]
100 fn accessors_and_metadata() {
101 let t = HighWave::new();
102 assert_eq!(t.name(), "HighWave");
103 assert_eq!(t.warmup_period(), 1);
104 assert!(!t.is_ready());
105 }
106
107 #[test]
108 fn high_wave_is_plus_one() {
109 let mut t = HighWave::new();
110 assert_eq!(t.update(c(10.0, 12.0, 8.0, 10.3, 0)), Some(1.0));
111 }
112
113 #[test]
114 fn short_upper_shadow_yields_zero() {
115 let mut t = HighWave::new();
116 assert_eq!(t.update(c(11.5, 12.0, 8.0, 11.7, 0)), Some(0.0));
118 }
119
120 #[test]
121 fn short_lower_shadow_yields_zero() {
122 let mut t = HighWave::new();
123 assert_eq!(t.update(c(8.3, 12.0, 8.0, 8.5, 0)), Some(0.0));
125 }
126
127 #[test]
128 fn big_body_yields_zero() {
129 let mut t = HighWave::new();
130 assert_eq!(t.update(c(8.5, 12.0, 8.0, 11.5, 0)), Some(0.0));
132 }
133
134 #[test]
135 fn zero_range_yields_zero() {
136 let mut t = HighWave::new();
137 assert_eq!(t.update(c(10.0, 10.0, 10.0, 10.0, 0)), Some(0.0));
138 }
139
140 #[test]
141 fn batch_equals_streaming() {
142 let candles: Vec<Candle> = (0..40)
143 .map(|i| {
144 let base = 100.0 + i as f64;
145 c(base, base + 3.0, base - 3.0, base + 0.2, i)
146 })
147 .collect();
148 let mut a = HighWave::new();
149 let mut b = HighWave::new();
150 assert_eq!(
151 a.batch(&candles),
152 candles.iter().map(|x| b.update(*x)).collect::<Vec<_>>()
153 );
154 }
155
156 #[test]
157 fn reset_clears_state() {
158 let mut t = HighWave::new();
159 t.update(c(10.0, 12.0, 8.0, 10.3, 0));
160 assert!(t.is_ready());
161 t.reset();
162 assert!(!t.is_ready());
163 }
164}