Wickra 0.1.0: streaming-first technical indicators
A multi-language technical analysis library: 25 indicators across trend,
momentum, volatility, and volume families, every one a state machine with
O(1) per-tick updates. Batch evaluation is provided by a blanket extension
trait over the streaming primitive, so live trading bots and historical
backtests run the same code path.
What ships in this initial drop:
crates/wickra-core - 25 indicators, Indicator/BatchExt/Chain traits,
OHLCV types with validation; 171 unit tests,
property tests, Wilder/Bollinger textbook tests.
crates/wickra - top-level facade + criterion benches for every
indicator at 1K/10K/100K series sizes.
crates/wickra-data - streaming CSV reader, tick-to-candle aggregator,
multi-timeframe resampler, Binance Spot kline
WebSocket adapter behind feature live-binance;
11 unit + 1 doctest.
bindings/python - PyO3 + maturin, NumPy I/O, type stubs (.pyi),
56 pytest tests including streaming==batch
equivalence, Wilder reference values, lifecycle.
bindings/node - napi-rs native module, TypeScript .d.ts
auto-generated, 7 node --test cases.
bindings/wasm - wasm-bindgen ES module for browser/bundler/Node;
interactive HTML demo at examples/index.html.
examples/ - Python and Rust scripts: backtest, live trading,
parallel multi-asset, multi-timeframe, Binance.
benchmarks/ - cross-library comparison against TA-Lib,
pandas-ta, finta, talipp; Wickra wins every
category by 11-1030x (batch) and 17x+ streaming.
.github/workflows/ - CI matrix (Rust + Python + Node + WASM on
Linux/macOS/Windows), release pipeline for
PyPI wheels and npm.
Indicators (25):
Trend SMA EMA WMA DEMA TEMA HMA KAMA
Momentum RSI MACD Stochastic CCI ROC WilliamsR ADX MFI TRIX
AwesomeOscillator Aroon
Volatility BollingerBands ATR Keltner Donchian PSAR
Volume OBV VWAP (cumulative + rolling)
cargo clippy --workspace --all-targets -D warnings is clean. License: Apache-2.0.
This commit is contained in:
@@ -0,0 +1,298 @@
|
||||
//! Average Directional Index (ADX) with +DI / -DI components.
|
||||
|
||||
use crate::error::{Error, Result};
|
||||
use crate::ohlcv::Candle;
|
||||
use crate::traits::Indicator;
|
||||
|
||||
/// ADX output: the three Wilder lines.
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub struct AdxOutput {
|
||||
/// Plus Directional Indicator.
|
||||
pub plus_di: f64,
|
||||
/// Minus Directional Indicator.
|
||||
pub minus_di: f64,
|
||||
/// Average Directional Index (smoothed |DX|).
|
||||
pub adx: f64,
|
||||
}
|
||||
|
||||
/// Wilder's Average Directional Index.
|
||||
///
|
||||
/// Uses Wilder smoothing throughout. First `period` candles seed the directional
|
||||
/// movement / true range sums; the next `period` candles produce DX values that
|
||||
/// seed the ADX. The first complete `AdxOutput` is emitted after `2 * period`
|
||||
/// candles.
|
||||
#[allow(clippy::struct_field_names)] // adx_value pairs with adx (the output line) — renaming hurts clarity
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Adx {
|
||||
period: usize,
|
||||
prev: Option<Candle>,
|
||||
|
||||
// Wilder-smoothed sums during seeding.
|
||||
tr_seed: f64,
|
||||
plus_dm_seed: f64,
|
||||
minus_dm_seed: f64,
|
||||
seed_count: usize,
|
||||
|
||||
// Smoothed running values after seeding.
|
||||
tr_smooth: Option<f64>,
|
||||
plus_dm_smooth: Option<f64>,
|
||||
minus_dm_smooth: Option<f64>,
|
||||
|
||||
// ADX seeding.
|
||||
dx_buf: Vec<f64>,
|
||||
adx_value: Option<f64>,
|
||||
last_plus_di: f64,
|
||||
last_minus_di: f64,
|
||||
}
|
||||
|
||||
impl Adx {
|
||||
/// # Errors
|
||||
/// Returns [`Error::PeriodZero`] if `period == 0`.
|
||||
pub fn new(period: usize) -> Result<Self> {
|
||||
if period == 0 {
|
||||
return Err(Error::PeriodZero);
|
||||
}
|
||||
Ok(Self {
|
||||
period,
|
||||
prev: None,
|
||||
tr_seed: 0.0,
|
||||
plus_dm_seed: 0.0,
|
||||
minus_dm_seed: 0.0,
|
||||
seed_count: 0,
|
||||
tr_smooth: None,
|
||||
plus_dm_smooth: None,
|
||||
minus_dm_smooth: None,
|
||||
dx_buf: Vec::with_capacity(period),
|
||||
adx_value: None,
|
||||
last_plus_di: 0.0,
|
||||
last_minus_di: 0.0,
|
||||
})
|
||||
}
|
||||
|
||||
/// Configured period.
|
||||
pub const fn period(&self) -> usize {
|
||||
self.period
|
||||
}
|
||||
}
|
||||
|
||||
fn directional_movement(prev: &Candle, current: &Candle) -> (f64, f64) {
|
||||
let up = current.high - prev.high;
|
||||
let down = prev.low - current.low;
|
||||
let plus_dm = if up > down && up > 0.0 { up } else { 0.0 };
|
||||
let minus_dm = if down > up && down > 0.0 { down } else { 0.0 };
|
||||
(plus_dm, minus_dm)
|
||||
}
|
||||
|
||||
impl Indicator for Adx {
|
||||
type Input = Candle;
|
||||
type Output = AdxOutput;
|
||||
|
||||
fn update(&mut self, candle: Candle) -> Option<AdxOutput> {
|
||||
let Some(prev) = self.prev else {
|
||||
self.prev = Some(candle);
|
||||
return None;
|
||||
};
|
||||
self.prev = Some(candle);
|
||||
|
||||
let tr = candle.true_range(Some(prev.close));
|
||||
let (plus_dm, minus_dm) = directional_movement(&prev, &candle);
|
||||
let n = self.period as f64;
|
||||
|
||||
let (tr_v, plus_v, minus_v) = if let (Some(t), Some(p), Some(m)) =
|
||||
(self.tr_smooth, self.plus_dm_smooth, self.minus_dm_smooth)
|
||||
{
|
||||
let t_new = t - t / n + tr;
|
||||
let p_new = p - p / n + plus_dm;
|
||||
let m_new = m - m / n + minus_dm;
|
||||
self.tr_smooth = Some(t_new);
|
||||
self.plus_dm_smooth = Some(p_new);
|
||||
self.minus_dm_smooth = Some(m_new);
|
||||
(t_new, p_new, m_new)
|
||||
} else {
|
||||
self.tr_seed += tr;
|
||||
self.plus_dm_seed += plus_dm;
|
||||
self.minus_dm_seed += minus_dm;
|
||||
self.seed_count += 1;
|
||||
if self.seed_count < self.period {
|
||||
return None;
|
||||
}
|
||||
self.tr_smooth = Some(self.tr_seed);
|
||||
self.plus_dm_smooth = Some(self.plus_dm_seed);
|
||||
self.minus_dm_smooth = Some(self.minus_dm_seed);
|
||||
(self.tr_seed, self.plus_dm_seed, self.minus_dm_seed)
|
||||
};
|
||||
|
||||
let plus_di = if tr_v == 0.0 {
|
||||
0.0
|
||||
} else {
|
||||
100.0 * plus_v / tr_v
|
||||
};
|
||||
let minus_di = if tr_v == 0.0 {
|
||||
0.0
|
||||
} else {
|
||||
100.0 * minus_v / tr_v
|
||||
};
|
||||
self.last_plus_di = plus_di;
|
||||
self.last_minus_di = minus_di;
|
||||
|
||||
let dx_den = plus_di + minus_di;
|
||||
let dx = if dx_den == 0.0 {
|
||||
0.0
|
||||
} else {
|
||||
100.0 * (plus_di - minus_di).abs() / dx_den
|
||||
};
|
||||
|
||||
if let Some(prev_adx) = self.adx_value {
|
||||
let new_adx = (prev_adx * (n - 1.0) + dx) / n;
|
||||
self.adx_value = Some(new_adx);
|
||||
return Some(AdxOutput {
|
||||
plus_di,
|
||||
minus_di,
|
||||
adx: new_adx,
|
||||
});
|
||||
}
|
||||
|
||||
self.dx_buf.push(dx);
|
||||
if self.dx_buf.len() == self.period {
|
||||
let seed = self.dx_buf.iter().sum::<f64>() / n;
|
||||
self.adx_value = Some(seed);
|
||||
return Some(AdxOutput {
|
||||
plus_di,
|
||||
minus_di,
|
||||
adx: seed,
|
||||
});
|
||||
}
|
||||
None
|
||||
}
|
||||
|
||||
fn reset(&mut self) {
|
||||
self.prev = None;
|
||||
self.tr_seed = 0.0;
|
||||
self.plus_dm_seed = 0.0;
|
||||
self.minus_dm_seed = 0.0;
|
||||
self.seed_count = 0;
|
||||
self.tr_smooth = None;
|
||||
self.plus_dm_smooth = None;
|
||||
self.minus_dm_smooth = None;
|
||||
self.dx_buf.clear();
|
||||
self.adx_value = None;
|
||||
self.last_plus_di = 0.0;
|
||||
self.last_minus_di = 0.0;
|
||||
}
|
||||
|
||||
fn warmup_period(&self) -> usize {
|
||||
2 * self.period
|
||||
}
|
||||
|
||||
fn is_ready(&self) -> bool {
|
||||
self.adx_value.is_some()
|
||||
}
|
||||
|
||||
fn name(&self) -> &'static str {
|
||||
"ADX"
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::traits::BatchExt;
|
||||
use approx::assert_relative_eq;
|
||||
|
||||
fn c(h: f64, l: f64, cl: f64) -> Candle {
|
||||
Candle::new(cl, h, l, cl, 1.0, 0).unwrap()
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn pure_uptrend_yields_plus_di_dominant() {
|
||||
// Strict uptrend: highs increase, lows increase, ADX should trend up,
|
||||
// +DI should dominate -DI.
|
||||
let candles: Vec<Candle> = (0..50)
|
||||
.map(|i| {
|
||||
let base = 100.0 + f64::from(i) * 2.0;
|
||||
c(base + 1.0, base - 0.5, base + 0.5)
|
||||
})
|
||||
.collect();
|
||||
let mut adx = Adx::new(14).unwrap();
|
||||
let last = adx
|
||||
.batch(&candles)
|
||||
.into_iter()
|
||||
.flatten()
|
||||
.last()
|
||||
.expect("emits");
|
||||
assert!(
|
||||
last.plus_di > last.minus_di,
|
||||
"+DI {} should exceed -DI {}",
|
||||
last.plus_di,
|
||||
last.minus_di
|
||||
);
|
||||
assert!(last.adx > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn pure_downtrend_yields_minus_di_dominant() {
|
||||
let candles: Vec<Candle> = (0..50)
|
||||
.rev()
|
||||
.map(|i| {
|
||||
let base = 100.0 + f64::from(i) * 2.0;
|
||||
c(base + 1.0, base - 0.5, base + 0.5)
|
||||
})
|
||||
.collect();
|
||||
let mut adx = Adx::new(14).unwrap();
|
||||
let last = adx
|
||||
.batch(&candles)
|
||||
.into_iter()
|
||||
.flatten()
|
||||
.last()
|
||||
.expect("emits");
|
||||
assert!(last.minus_di > last.plus_di);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn rejects_zero_period() {
|
||||
assert!(Adx::new(0).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn batch_equals_streaming() {
|
||||
let candles: Vec<Candle> = (0..60)
|
||||
.map(|i| {
|
||||
let base = 100.0 + (f64::from(i) * 0.3).sin() * 5.0;
|
||||
c(base + 1.0, base - 1.0, base)
|
||||
})
|
||||
.collect();
|
||||
let mut a = Adx::new(14).unwrap();
|
||||
let mut b = Adx::new(14).unwrap();
|
||||
assert_eq!(
|
||||
a.batch(&candles),
|
||||
candles.iter().map(|x| b.update(*x)).collect::<Vec<_>>()
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn reset_clears_state() {
|
||||
let candles: Vec<Candle> = (0..40).map(|_| c(11.0, 9.0, 10.0)).collect();
|
||||
let mut adx = Adx::new(14).unwrap();
|
||||
adx.batch(&candles);
|
||||
adx.reset();
|
||||
assert!(!adx.is_ready());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn outputs_remain_finite() {
|
||||
let candles: Vec<Candle> = (0..200)
|
||||
.map(|i| {
|
||||
let m = 100.0 + (f64::from(i) * 0.2).sin() * 5.0;
|
||||
c(m + 1.0, m - 1.0, m)
|
||||
})
|
||||
.collect();
|
||||
let mut adx = Adx::new(14).unwrap();
|
||||
for v in adx.batch(&candles).into_iter().flatten() {
|
||||
assert!(v.plus_di.is_finite() && v.minus_di.is_finite() && v.adx.is_finite());
|
||||
}
|
||||
// Sanity: ADX is bounded by 100.
|
||||
let last = adx.batch(&candles).into_iter().flatten().last().unwrap();
|
||||
assert!(last.adx <= 100.0 + 1e-6);
|
||||
assert_relative_eq!(0.0_f64.max(last.adx), last.adx, epsilon = 1e-9);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user