tulip_rs 0.1.8

High-performance technical analysis library — 100+ indicators and 60+ candlestick patterns with SIMD acceleration
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

use crate::indicators::adosc::State;
use crate::indicators::simd_indicators::{
    ad_simd::calc_simd as calc_ad_simd, ema_simd::calc_simd as calc_ema_simd,
};
use std::simd::Simd;

#[cfg(feature = "simd_assets")]
pub use crate::indicators::simd_indicators::by_asset::adosc::indicator_by_assets;

#[cfg(feature = "simd_options")]
pub use crate::indicators::simd_indicators::by_option::adosc::indicator_by_options;

/// SIMD-parallel state for computing the Chaikin AD Oscillator (ADOSC) across `N` assets
/// simultaneously. Each field is a SIMD vector where lane `i` corresponds to asset `i`.
pub struct SimdState<const N: usize> {
    /// Running Accumulation/Distribution line value for each asset.
    pub ad: Simd<f64, N>,
    /// Short-period EMA of the AD line for each asset.
    pub short_ema: Simd<f64, N>,
    /// Long-period EMA of the AD line for each asset.
    pub long_ema: Simd<f64, N>,
}
impl<const N: usize> SimdState<N> {
    /// Gathers `N` scalar [`State`] references into a single `SimdState`,
    /// packing each field into a SIMD lane.
    pub fn new(states: &[&mut State]) -> Self {
        let mut ad = [0.0; N];
        let mut short_ema = [0.0; N];
        let mut long_ema = [0.0; N];

        for i in 0..N {
            ad[i] = states[i].ad;
            short_ema[i] = states[i].short_ema;
            long_ema[i] = states[i].long_ema;
        }
        Self {
            ad: Simd::from_array(ad),
            short_ema: Simd::from_array(short_ema),
            long_ema: Simd::from_array(long_ema),
        }
    }
    /// Scatters the SIMD state back into an array of `N` scalar [`State`] values.
    pub fn to_states(&self) -> [State; N] {
        let ad = self.ad.to_array();
        let short_ema = self.short_ema.to_array();
        let long_ema = self.long_ema.to_array();

        let states: [State; N] =
            std::array::from_fn(|i| State::new(ad[i], short_ema[i], long_ema[i]));

        states
    }
    /// Writes the SIMD state back into `N` existing mutable scalar [`State`] references in place,
    /// avoiding allocation compared to [`to_states`].
    pub fn write_states(&self, states: &mut [&mut State]) {
        let ad = self.ad.to_array();
        let short_ema = self.short_ema.to_array();
        let long_ema = self.long_ema.to_array();

        for i in 0..N {
            states[i].ad = ad[i];
            states[i].short_ema = short_ema[i];
            states[i].long_ema = long_ema[i];
        }
    }
}

/// Advances the Chaikin AD Oscillator (ADOSC) by one bar for `N` assets simultaneously.
///
/// Updates the AD line, then applies short- and long-period EMA smoothing. The oscillator value
/// is the difference between the two EMAs (`short_ema - long_ema`).
///
/// # Arguments
///
/// * `state` - Mutable SIMD state holding per-asset AD, short EMA, and long EMA.
/// * `inputs` - Tuple of `(high, low, close, volume)` SIMD vectors for the current bar.
/// * `multipliers` - Tuple of `(short_multiplier, long_multiplier)` EMA smoothing factors,
///   each itself a `(per, inv_per)` pair.
///
/// # Returns
///
/// ADOSC values (`short_ema - long_ema`) for all `N` lanes.
#[inline(always)]
pub fn calc_simd<const N: usize>(
    state: &mut SimdState<N>,
    inputs: (Simd<f64, N>, Simd<f64, N>, Simd<f64, N>, Simd<f64, N>),
    multipliers: ((Simd<f64, N>, Simd<f64, N>), (Simd<f64, N>, Simd<f64, N>)),
) -> Simd<f64, N> {
    let (high, low, close, volume) = inputs;
    let (short_multiplier, long_multiplier) = multipliers;

    state.ad = calc_ad_simd(state.ad, high, low, close, volume);
    state.short_ema = calc_ema_simd(state.ad, state.short_ema, short_multiplier);
    state.long_ema = calc_ema_simd(state.ad, state.long_ema, long_multiplier);

    state.short_ema - state.long_ema
}