nautilus_indicators/python/momentum/
psl.rs1use nautilus_model::data::Bar;
17use pyo3::prelude::*;
18
19use crate::{average::MovingAverageType, indicator::Indicator, momentum::psl::PsychologicalLine};
20
21#[pymethods]
22impl PsychologicalLine {
23 #[new]
24 #[pyo3(signature = (period, ma_type=None))]
25 #[must_use]
26 pub fn py_new(period: usize, ma_type: Option<MovingAverageType>) -> Self {
27 Self::new(period, ma_type)
28 }
29
30 fn __repr__(&self) -> String {
31 format!("PsychologicalLine({},{})", self.period, self.ma_type)
32 }
33
34 #[getter]
35 #[pyo3(name = "name")]
36 fn py_name(&self) -> String {
37 self.name()
38 }
39
40 #[getter]
41 #[pyo3(name = "period")]
42 const fn py_period(&self) -> usize {
43 self.period
44 }
45
46 #[getter]
47 #[pyo3(name = "has_inputs")]
48 fn py_has_inputs(&self) -> bool {
49 self.has_inputs()
50 }
51
52 #[getter]
53 #[pyo3(name = "value")]
54 const fn py_value(&self) -> f64 {
55 self.value
56 }
57
58 #[getter]
59 #[pyo3(name = "initialized")]
60 const fn py_initialized(&self) -> bool {
61 self.initialized
62 }
63
64 #[pyo3(name = "update_raw")]
65 fn py_update_raw(&mut self, close: f64) {
66 self.update_raw(close);
67 }
68
69 #[pyo3(name = "handle_bar")]
70 fn py_handle_bar(&mut self, bar: &Bar) {
71 self.handle_bar(bar);
72 }
73
74 #[pyo3(name = "reset")]
75 fn py_reset(&mut self) {
76 self.reset();
77 }
78}