use num::complex::ComplexFloat;
use option_trait::Maybe;
use crate::{
systems::{Ar, Rpk, Rtf, Sos, Ss, SsAMatrix, SsBMatrix, SsCMatrix, SsDMatrix, Tf, Zpk},
quantities::{ListOrSingle, MaybeList, MaybeLists, MaybeOwnedList},
System
};
pub trait MaybeRtfOrSystem<D>
where
D: ComplexFloat
{
}
impl<D> MaybeRtfOrSystem<D> for ()
where
D: ComplexFloat
{
}
impl<W, S> MaybeRtfOrSystem<W> for Rtf<W, S>
where
W: ComplexFloat<Real = <S::Set as ComplexFloat>::Real>,
S::Set: Into<W>,
S: System
{
}
impl<T, B, A> MaybeRtfOrSystem<T> for Tf<T, B, A>
where
T: ComplexFloat,
B: MaybeLists<T>,
A: MaybeList<T>
{
}
impl<T, Z, P, K, R> MaybeRtfOrSystem<K> for Zpk<T, Z, P, K>
where
T: ComplexFloat<Real = R>,
K: ComplexFloat<Real = R>,
Z: MaybeList<T>,
P: MaybeList<T>
{
}
impl<T, A, B, C, D> MaybeRtfOrSystem<T> for Ss<T, A, B, C, D>
where
T: ComplexFloat,
A: SsAMatrix<T, B, C, D>,
B: SsBMatrix<T, A, C, D>,
C: SsCMatrix<T, A, B, D>,
D: SsDMatrix<T, A, B, C>
{
}
impl<T, B, A, S> MaybeRtfOrSystem<T> for Sos<T, B, A, S>
where
T: ComplexFloat,
B: Maybe<[T; 3]> + MaybeOwnedList<T>,
A: Maybe<[T; 3]> + MaybeOwnedList<T>,
S: MaybeList<Tf<T, B, A>>
{
}
impl<T, R, P, RP, K> MaybeRtfOrSystem<T> for Rpk<T, R, P, RP, K>
where
T: ComplexFloat,
R: ComplexFloat<Real = T::Real>,
P: ComplexFloat<Real = T::Real>,
RP: MaybeList<(R, P)>,
K: MaybeList<T>
{
}
impl<T, A, AV> MaybeRtfOrSystem<T> for Ar<T, A, AV>
where
T: ComplexFloat,
A: MaybeList<T>,
AV: ListOrSingle<(A, T::Real)>
{
}