research/argus
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(pyargus): implement new for PySignal

Browse source
This commit is contained in:
Anand Balakrishnan 2023-09-07 15:44:46 -07:00
parent e2cfe3da56
commit 8027f86213
No known key found for this signature in database
2 changed files with 18 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

28
pyargus/src/signals.rs
View file

@ -40,6 +40,18 @@ pub struct PySignal {
pub(crate) signal: SignalKind,
}
impl PySignal {
pub fn new<T>(signal: T, interpolation: PyInterp) -> Self
where
T: Into<SignalKind>,
{
Self {
interpolation,
signal: signal.into(),
}
}
}
#[pymethods]
impl PySignal {
#[getter]
@ -106,23 +118,13 @@ macro_rules! impl_signals {
#[derive(Debug, Copy, Clone)]
pub struct [<$ty_name Signal>];
impl [<$ty_name Signal>] {
#[inline]
pub fn super_type(signal: SignalKind) -> PySignal {
PySignal {
interpolation: PyInterp::Linear,
signal,
}
}
}
#[pymethods]
impl [<$ty_name Signal>] {
/// Create a new empty signal
#[new]
#[pyo3(signature = ())]
fn new() -> (Self, PySignal) {
(Self, Self::super_type(Signal::<$ty>::new().into()))
(Self, PySignal::new(Signal::<$ty>::new(), PyInterp::Linear))
}
/// Create a new signal with constant value
@ -130,7 +132,7 @@ macro_rules! impl_signals {
fn constant(_: &PyType, py: Python<'_>, value: $ty) -> PyResult<Py<Self>> {
Py::new(
py,
(Self, Self::super_type(Signal::constant(value).into()))
(Self, PySignal::new(Signal::constant(value), PyInterp::Linear))
)
}
@ -144,7 +146,7 @@ macro_rules! impl_signals {
Python::with_gil(|py| {
Py::new(
py,
(Self, Self::super_type(ret.into()))
(Self, PySignal::new(ret, PyInterp::Linear))
)
})
}