15 lines
426 B
Rust
15 lines
426 B
Rust
|
use super::common::compute_ht_core;
|
||
|
|
use numpy::{IntoPyArray, PyArray1, PyReadonlyArray1};
|
||
|
|
use pyo3::prelude::*;
|
||
|
|
|
||
|
|
/// Hilbert Transform Dominant Cycle Period in bars.
|
||
|
|
#[pyfunction]
|
||
|
|
pub fn ht_dcperiod<'py>(
|
||
|
|
py: Python<'py>,
|
||
|
|
close: PyReadonlyArray1<'py, f64>,
|
||
|
|
) -> PyResult<Bound<'py, PyArray1<f64>>> {
|
||
|
|
let prices = close.as_slice()?;
|
||
|
|
let core = compute_ht_core(prices);
|
||
|
|
Ok(core.dc_period.into_pyarray(py))
|
||
|
|
}
|