nautilus_indicators/python/volatility/
vr.rs1use nautilus_core::python::to_pyvalue_err;
17use nautilus_model::data::{Bar, QuoteTick, TradeTick};
18use pyo3::prelude::*;
19
20use crate::{indicator::Indicator, python::float_precision, volatility::vr::VolatilityRatio};
21
22#[pymethods]
23#[pyo3_stub_gen::derive::gen_stub_pymethods]
24impl VolatilityRatio {
25 #[new]
27 fn py_new(period: usize) -> PyResult<Self> {
28 Self::new_checked(period).map_err(to_pyvalue_err)
29 }
30
31 fn __repr__(&self) -> String {
32 format!("VolatilityRatio({})", self.period)
33 }
34
35 #[getter]
36 #[pyo3(name = "name")]
37 fn py_name(&self) -> String {
38 self.name()
39 }
40
41 #[getter]
42 #[pyo3(name = "period")]
43 const fn py_period(&self) -> usize {
44 self.period
45 }
46
47 #[getter]
48 #[pyo3(name = "has_inputs")]
49 fn py_has_inputs(&self) -> bool {
50 self.has_inputs()
51 }
52
53 #[getter]
54 #[pyo3(name = "value")]
55 const fn py_value(&self) -> f64 {
56 self.value
57 }
58
59 #[getter]
60 #[pyo3(name = "count")]
61 const fn py_count(&self) -> usize {
62 self.count
63 }
64
65 #[getter]
66 #[pyo3(name = "initialized")]
67 const fn py_initialized(&self) -> bool {
68 self.initialized
69 }
70
71 #[pyo3(name = "update_raw")]
73 fn py_update_raw(&mut self, high: f64, low: f64, close: f64) {
74 self.update_raw(high, low, close);
75 }
76
77 #[pyo3(name = "handle_quote_tick")]
78 fn py_handle_quote_tick(&mut self, quote: &QuoteTick) -> PyResult<()> {
79 self.handle_quote(quote).map_err(to_pyvalue_err)
80 }
81
82 #[pyo3(name = "handle_trade_tick")]
83 fn py_handle_trade_tick(&mut self, trade: &TradeTick) {
84 self.handle_trade(trade);
85 }
86
87 #[pyo3(name = "handle_bar")]
88 fn py_handle_bar(&mut self, bar: &Bar) -> PyResult<()> {
89 float_precision::check_bar(bar)?;
90 self.handle_bar(bar);
91 Ok(())
92 }
93
94 #[pyo3(name = "reset")]
95 fn py_reset(&mut self) {
96 self.reset();
97 }
98}