Skip to content

Commit

Permalink
IMPL: FPVector & Vector for Vec<Complex<f64>> (#35)
Browse files Browse the repository at this point in the history
  • Loading branch information
Axect committed Jan 16, 2021
1 parent 42e804b commit 97fcd46
Showing 1 changed file with 83 additions and 1 deletion.
84 changes: 83 additions & 1 deletion src/structure/complex.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
use num_complex::Complex;
use crate::traits::fp::FPVector;
use crate::traits::math::{Vector, Normed, Norm, InnerProduct};
use crate::traits::sugar::VecOps;

impl Vector for Complex<f64> {
type Scalar = Self;
Expand All @@ -23,7 +25,7 @@ impl Normed for Complex<f64> {
match kind {
Norm::L1 => self.l1_norm(),
Norm::L2 => Complex::<f64>::norm(*self),
_ => panic!("No more norms form complex"),
_ => unimplemented!(),
}
}

Expand All @@ -40,3 +42,83 @@ impl InnerProduct for Complex<f64> {
self.conj() * rhs
}
}

impl FPVector for Vec<Complex<f64>> {
type Scalar = Complex<f64>;

fn fmap<F>(&self, f: F) -> Self
where
F: Fn(Self::Scalar) -> Self::Scalar {
self.iter().map(|&x| f(x)).collect()
}

fn zip_with<F>(&self, f: F, other: &Self) -> Self
where
F: Fn(Self::Scalar, Self::Scalar) -> Self::Scalar {
self.iter().zip(other.iter()).map(|(&x, &y)| f(x,y)).collect()
}

fn reduce<F, T>(&self, init: T, f: F) -> Self::Scalar
where
F: Fn(Self::Scalar, Self::Scalar) -> Self::Scalar,
T: Into<Self::Scalar> {
self.iter().fold(init.into(), |x, &y| f(x,y))
}

fn filter<F>(&self, f: F) -> Self
where
F: Fn(Self::Scalar) -> bool {
self.into_iter().filter(|&x| f(*x)).map(|&t| t).collect()
}

fn take(&self, n: usize) -> Self {
self.iter().take(n).map(|&x| x).collect()
}

fn skip(&self, n: usize) -> Self {
self.iter().skip(n).map(|&x| x).collect()
}

fn sum(&self) -> Self::Scalar {
self.iter().sum()
}

fn prod(&self) -> Self::Scalar {
self.iter().product()
}
}

impl Vector for Vec<Complex<f64>> {
type Scalar = Complex<f64>;

fn add_vec<'a, 'b>(&'a self, rhs: &'b Self) -> Self {
self.zip_with(|x, y| x + y, rhs)
}

fn sub_vec<'a, 'b>(&'a self, rhs: &'b Self) -> Self {
self.zip_with(|x, y| x - y, rhs)
}

fn mul_scalar(&self, rhs: Self::Scalar) -> Self {
self.fmap(|x| x * rhs)
}
}

impl Normed for Vec<Complex<f64>> {
type UnsignedScalar = f64;

fn norm(&self, kind: Norm) -> Self::UnsignedScalar {
match kind {
Norm::L1 => self.iter().map(|x| Complex::<f64>::norm(*x).abs()).sum(),
_ => unimplemented!()
}
}

fn normalize(&self, kind: Norm) -> Self
where
Self: Sized {
unimplemented!()
}
}

impl VecOps for Vec<Complex<f64>> {}

0 comments on commit 97fcd46

Please sign in to comment.