nautilus_indicators/python/momentum/
aroon.rs1use nautilus_model::data::{Bar, QuoteTick, TradeTick};
17use pyo3::prelude::*;
18
19use crate::{indicator::Indicator, momentum::aroon::AroonOscillator};
20
21#[pymethods]
22#[pyo3_stub_gen::derive::gen_stub_pymethods]
23impl AroonOscillator {
24 #[new]
27 #[must_use]
28 pub fn py_new(period: usize) -> Self {
29 Self::new(period)
30 }
31
32 fn __repr__(&self) -> String {
33 format!("AroonOscillator({})", self.period)
34 }
35
36 #[getter]
37 #[pyo3(name = "name")]
38 fn py_name(&self) -> String {
39 self.name()
40 }
41
42 #[getter]
43 #[pyo3(name = "period")]
44 const fn py_period(&self) -> usize {
45 self.period
46 }
47
48 #[getter]
49 #[pyo3(name = "has_inputs")]
50 fn py_has_inputs(&self) -> bool {
51 self.has_inputs()
52 }
53
54 #[getter]
55 #[pyo3(name = "count")]
56 const fn py_count(&self) -> usize {
57 self.count
58 }
59
60 #[getter]
61 #[pyo3(name = "aroon_up")]
62 const fn py_aroon_up(&self) -> f64 {
63 self.aroon_up
64 }
65
66 #[getter]
67 #[pyo3(name = "aroon_down")]
68 const fn py_aroon_down(&self) -> f64 {
69 self.aroon_down
70 }
71
72 #[getter]
73 #[pyo3(name = "value")]
74 const fn py_value(&self) -> f64 {
75 self.value
76 }
77
78 #[getter]
79 #[pyo3(name = "initialized")]
80 const fn py_initialized(&self) -> bool {
81 self.initialized
82 }
83
84 #[pyo3(name = "update_raw")]
85 fn py_update_raw(&mut self, high: f64, low: f64) {
86 self.update_raw(high, low);
87 }
88
89 #[pyo3(name = "handle_quote_tick")]
90 const fn py_handle_quote_tick(&mut self, _quote: &QuoteTick) {
91 }
93
94 #[pyo3(name = "handle_trade_tick")]
95 const fn py_handle_trade_tick(&mut self, _trade: &TradeTick) {
96 }
98
99 #[pyo3(name = "handle_bar")]
100 fn py_handle_bar(&mut self, bar: &Bar) {
101 self.update_raw((&bar.close).into(), (&bar.close).into());
102 }
103
104 #[pyo3(name = "reset")]
105 fn py_reset(&mut self) {
106 self.reset();
107 }
108}