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]
22impl AroonOscillator {
23 #[new]
24 #[must_use]
25 pub fn py_new(period: usize) -> Self {
26 Self::new(period)
27 }
28
29 fn __repr__(&self) -> String {
30 format!("AroonOscillator({})", self.period)
31 }
32
33 #[getter]
34 #[pyo3(name = "name")]
35 fn py_name(&self) -> String {
36 self.name()
37 }
38
39 #[getter]
40 #[pyo3(name = "period")]
41 const fn py_period(&self) -> usize {
42 self.period
43 }
44
45 #[getter]
46 #[pyo3(name = "has_inputs")]
47 fn py_has_inputs(&self) -> bool {
48 self.has_inputs()
49 }
50
51 #[getter]
52 #[pyo3(name = "count")]
53 const fn py_count(&self) -> usize {
54 self.count
55 }
56
57 #[getter]
58 #[pyo3(name = "aroon_up")]
59 const fn py_aroon_up(&self) -> f64 {
60 self.aroon_up
61 }
62
63 #[getter]
64 #[pyo3(name = "aroon_down")]
65 const fn py_aroon_down(&self) -> f64 {
66 self.aroon_down
67 }
68
69 #[getter]
70 #[pyo3(name = "value")]
71 const fn py_value(&self) -> f64 {
72 self.value
73 }
74
75 #[getter]
76 #[pyo3(name = "initialized")]
77 const fn py_initialized(&self) -> bool {
78 self.initialized
79 }
80
81 #[pyo3(name = "update_raw")]
82 fn py_update_raw(&mut self, high: f64, low: f64) {
83 self.update_raw(high, low);
84 }
85
86 #[pyo3(name = "handle_quote_tick")]
87 const fn py_handle_quote_tick(&mut self, _quote: &QuoteTick) {
88 }
90
91 #[pyo3(name = "handle_trade_tick")]
92 const fn py_handle_trade_tick(&mut self, _trade: &TradeTick) {
93 }
95
96 #[pyo3(name = "handle_bar")]
97 fn py_handle_bar(&mut self, bar: &Bar) {
98 self.update_raw((&bar.close).into(), (&bar.close).into());
99 }
100
101 #[pyo3(name = "reset")]
102 fn py_reset(&mut self) {
103 self.reset();
104 }
105}