quantwave-core 0.6.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

//! TA-Lib compatible streaming moving averages (SMA / EMA).

use crate::indicators::incremental::talib_ema::TalibEma;
use crate::indicators::incremental::talib_sma::TalibSma;
use crate::traits::Next;
use talib_rs::MaType;

/// Streaming MA matching TA-Lib `compute_ma` for SMA and EMA.
#[derive(Debug, Clone)]
pub enum MaStream {
    Sma(TalibSma),
    Ema(TalibEma),
}

impl MaStream {
    pub fn new(period: usize, ma_type: MaType) -> Self {
        match ma_type {
            MaType::Ema => Self::Ema(TalibEma::new(period)),
            _ => Self::Sma(TalibSma::new(period)),
        }
    }

    pub fn next(&mut self, v: f64) -> f64 {
        match self {
            Self::Sma(s) => s.next(v),
            Self::Ema(e) => e.next(v),
        }
    }
}