nautilus_indicators/python/average/
dema.rs1use nautilus_core::python::to_pyvalue_err;
17use nautilus_model::{
18 data::{Bar, QuoteTick, TradeTick},
19 enums::PriceType,
20};
21use pyo3::prelude::*;
22
23use crate::{
24 average::dema::DoubleExponentialMovingAverage,
25 indicator::{Indicator, MovingAverage},
26};
27
28#[pyo3_stub_gen::derive::gen_stub_pymethods]
29#[pymethods]
30impl DoubleExponentialMovingAverage {
31 #[new]
34 #[pyo3(signature = (period, price_type=None))]
35 fn py_new(period: usize, price_type: Option<PriceType>) -> Self {
36 Self::new(period, price_type)
37 }
38
39 fn __repr__(&self) -> String {
40 format!("DoubleExponentialMovingAverage({})", self.period)
41 }
42
43 #[getter]
44 #[pyo3(name = "name")]
45 fn py_name(&self) -> String {
46 self.name()
47 }
48
49 #[getter]
50 #[pyo3(name = "period")]
51 const fn py_period(&self) -> usize {
52 self.period
53 }
54
55 #[getter]
56 #[pyo3(name = "price_type")]
57 const fn py_price_type(&self) -> PriceType {
58 self.price_type
59 }
60
61 #[getter]
62 #[pyo3(name = "count")]
63 const fn py_count(&self) -> usize {
64 self.count
65 }
66
67 #[getter]
68 #[pyo3(name = "value")]
69 const fn py_value(&self) -> f64 {
70 self.value
71 }
72
73 #[getter]
74 #[pyo3(name = "has_inputs")]
75 fn py_has_inputs(&self) -> bool {
76 self.has_inputs()
77 }
78
79 #[getter]
80 #[pyo3(name = "initialized")]
81 const fn py_initialized(&self) -> bool {
82 self.initialized
83 }
84
85 #[pyo3(name = "handle_quote_tick")]
86 fn py_handle_quote_tick(&mut self, quote: &QuoteTick) -> PyResult<()> {
87 self.handle_quote(quote).map_err(to_pyvalue_err)
88 }
89
90 #[pyo3(name = "handle_trade_tick")]
91 fn py_handle_trade_tick(&mut self, trade: &TradeTick) {
92 self.handle_trade(trade);
93 }
94
95 #[pyo3(name = "handle_bar")]
96 fn py_handle_bar(&mut self, bar: &Bar) {
97 self.handle_bar(bar);
98 }
99
100 #[pyo3(name = "reset")]
101 fn py_reset(&mut self) {
102 self.reset();
103 }
104
105 #[pyo3(name = "update_raw")]
106 fn py_update_raw(&mut self, value: f64) {
107 self.update_raw(value);
108 }
109}