#![allow(clippy::doc_markdown)]
use crate::ohlcv::Candle;
use crate::traits::Indicator;
#[derive(Debug, Clone, Default)]
pub struct TdCamouflage {
prev: Option<Candle>,
last_value: Option<f64>,
}
impl TdCamouflage {
#[must_use]
pub fn new() -> Self {
Self::default()
}
pub const fn value(&self) -> Option<f64> {
self.last_value
}
}
impl Indicator for TdCamouflage {
type Input = Candle;
type Output = f64;
fn update(&mut self, candle: Candle) -> Option<f64> {
let Some(prev) = self.prev else {
self.prev = Some(candle);
self.last_value = Some(0.0);
return Some(0.0);
};
let v = if candle.close < prev.close && candle.close > candle.open && candle.low < prev.low
{
1.0
} else if candle.close > prev.close && candle.close < candle.open && candle.high > prev.high
{
-1.0
} else {
0.0
};
self.prev = Some(candle);
self.last_value = Some(v);
Some(v)
}
fn reset(&mut self) {
self.prev = None;
self.last_value = None;
}
fn warmup_period(&self) -> usize {
2
}
fn is_ready(&self) -> bool {
self.last_value.is_some()
}
fn name(&self) -> &'static str {
"TDCamouflage"
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::traits::BatchExt;
fn c(open: f64, high: f64, low: f64, close: f64) -> Candle {
Candle::new_unchecked(open, high, low, close, 0.0, 0)
}
#[test]
fn accessors_and_metadata() {
let td = TdCamouflage::new();
assert_eq!(td.warmup_period(), 2);
assert_eq!(td.name(), "TDCamouflage");
assert!(!td.is_ready());
assert_eq!(td.value(), None);
}
#[test]
fn first_bar_seeds_without_signal() {
let mut td = TdCamouflage::new();
assert_eq!(td.update(c(10.0, 11.0, 9.0, 10.0)), Some(0.0));
assert!(td.update(c(10.0, 11.0, 8.0, 9.5)).is_some());
}
#[test]
fn bullish_camouflage_buy() {
let mut td = TdCamouflage::new();
td.update(c(10.0, 11.0, 8.0, 10.0));
assert_eq!(td.update(c(9.0, 10.0, 7.0, 9.5)), Some(1.0));
}
#[test]
fn bearish_camouflage_sell() {
let mut td = TdCamouflage::new();
td.update(c(10.0, 11.0, 8.0, 10.0));
assert_eq!(td.update(c(11.0, 12.0, 10.0, 10.5)), Some(-1.0));
}
#[test]
fn no_pattern_is_zero() {
let mut td = TdCamouflage::new();
td.update(c(10.0, 11.0, 9.0, 10.0));
assert_eq!(td.update(c(10.0, 11.5, 9.5, 11.0)), Some(0.0));
}
#[test]
fn reset_clears_state() {
let mut td = TdCamouflage::new();
td.update(c(10.0, 11.0, 9.0, 10.0));
td.update(c(9.0, 10.0, 7.0, 9.5));
assert!(td.is_ready());
td.reset();
assert!(!td.is_ready());
assert_eq!(td.update(c(10.0, 11.0, 9.0, 10.0)), Some(0.0));
}
#[test]
fn batch_equals_streaming() {
let candles: Vec<Candle> = (0..40)
.map(|i| {
let b = 100.0 + (f64::from(i) * 0.4).sin() * 5.0;
c(b, b + 1.0, b - 1.0, b + 0.2)
})
.collect();
let batch = TdCamouflage::new().batch(&candles);
let mut b = TdCamouflage::new();
let streamed: Vec<_> = candles.iter().map(|x| b.update(*x)).collect();
assert_eq!(batch, streamed);
}
}