nautilus_indicators/python/momentum/
stochastics.rs1use nautilus_model::data::Bar;
17use pyo3::prelude::*;
18
19use crate::{indicator::Indicator, momentum::stochastics::Stochastics};
20
21#[pymethods]
22impl Stochastics {
23 #[new]
24 #[must_use]
25 pub fn py_new(period_k: usize, period_d: usize) -> Self {
26 Self::new(period_k, period_d)
27 }
28
29 fn __repr__(&self) -> String {
30 format!("Stochastics({},{})", self.period_k, self.period_d)
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_k")]
41 const fn py_period_k(&self) -> usize {
42 self.period_k
43 }
44
45 #[getter]
46 #[pyo3(name = "period_d")]
47 const fn py_period_d(&self) -> usize {
48 self.period_d
49 }
50
51 #[getter]
52 #[pyo3(name = "has_inputs")]
53 fn py_has_inputs(&self) -> bool {
54 self.has_inputs()
55 }
56
57 #[getter]
58 #[pyo3(name = "value_k")]
59 const fn py_value_k(&self) -> f64 {
60 self.value_k
61 }
62
63 #[getter]
64 #[pyo3(name = "value_d")]
65 const fn py_value_d(&self) -> f64 {
66 self.value_d
67 }
68
69 #[getter]
70 #[pyo3(name = "initialized")]
71 const fn py_initialized(&self) -> bool {
72 self.initialized
73 }
74
75 #[pyo3(name = "update_raw")]
76 fn py_update_raw(&mut self, high: f64, low: f64, close: f64) {
77 self.update_raw(high, low, close);
78 }
79
80 #[pyo3(name = "handle_bar")]
81 fn py_handle_bar(&mut self, bar: &Bar) {
82 self.handle_bar(bar);
83 }
84
85 #[pyo3(name = "reset")]
86 fn py_reset(&mut self) {
87 self.reset();
88 }
89}