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]
22#[pyo3_stub_gen::derive::gen_stub_pymethods]
23impl PsychologicalLine {
24 #[new]
26 #[pyo3(signature = (period, ma_type=None))]
27 #[must_use]
28 pub fn py_new(period: usize, ma_type: Option<MovingAverageType>) -> Self {
29 Self::new(period, ma_type)
30 }
31
32 fn __repr__(&self) -> String {
33 format!("PsychologicalLine({},{})", self.period, self.ma_type)
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 = "value")]
56 const fn py_value(&self) -> f64 {
57 self.value
58 }
59
60 #[getter]
61 #[pyo3(name = "initialized")]
62 const fn py_initialized(&self) -> bool {
63 self.initialized
64 }
65
66 #[pyo3(name = "update_raw")]
67 fn py_update_raw(&mut self, close: f64) {
68 self.update_raw(close);
69 }
70
71 #[pyo3(name = "handle_bar")]
72 fn py_handle_bar(&mut self, bar: &Bar) {
73 self.handle_bar(bar);
74 }
75
76 #[pyo3(name = "reset")]
77 fn py_reset(&mut self) {
78 self.reset();
79 }
80}