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