nautilus_indicators/average/
mod.rs1pub mod ama;
19pub mod dema;
20pub mod ema;
21pub mod hma;
22pub mod lr;
23pub mod rma;
24pub mod sma;
25pub mod vidya;
26pub mod vwap;
27pub mod wma;
28
29use nautilus_model::enums::PriceType;
30use strum::{AsRefStr, Display, EnumIter, EnumString, FromRepr};
31
32use crate::{
33 average::{
34 dema::DoubleExponentialMovingAverage, ema::ExponentialMovingAverage,
35 hma::HullMovingAverage, rma::WilderMovingAverage, sma::SimpleMovingAverage,
36 },
37 indicator::MovingAverage,
38};
39
40#[repr(C)]
41#[derive(
42 Copy,
43 Clone,
44 Debug,
45 Display,
46 Hash,
47 PartialEq,
48 Eq,
49 PartialOrd,
50 Ord,
51 AsRefStr,
52 FromRepr,
53 EnumIter,
54 EnumString,
55)]
56#[strum(ascii_case_insensitive)]
57#[strum(serialize_all = "SCREAMING_SNAKE_CASE")]
58#[cfg_attr(
59 feature = "python",
60 pyo3::pyclass(
61 frozen,
62 eq,
63 eq_int,
64 hash,
65 module = "nautilus_trader.core.nautilus_pyo3.indicators",
66 from_py_object,
67 )
68)]
69pub enum MovingAverageType {
70 Simple,
71 Exponential,
72 DoubleExponential,
73 Wilder,
74 Hull,
75}
76
77#[derive(Debug)]
78pub struct MovingAverageFactory;
79
80impl MovingAverageFactory {
81 #[must_use]
82 pub fn create(
83 moving_average_type: MovingAverageType,
84 period: usize,
85 ) -> Box<dyn MovingAverage + Send + Sync> {
86 let price_type = Some(PriceType::Last);
87
88 match moving_average_type {
89 MovingAverageType::Simple => Box::new(SimpleMovingAverage::new(period, price_type)),
90 MovingAverageType::Exponential => {
91 Box::new(ExponentialMovingAverage::new(period, price_type))
92 }
93 MovingAverageType::DoubleExponential => {
94 Box::new(DoubleExponentialMovingAverage::new(period, price_type))
95 }
96 MovingAverageType::Wilder => Box::new(WilderMovingAverage::new(period, price_type)),
97 MovingAverageType::Hull => Box::new(HullMovingAverage::new(period, price_type)),
98 }
99 }
100}