55 lines
1.5 KiB
Rust
55 lines
1.5 KiB
Rust
// This module implement several methods to fill some
|
|
// missing features of num-complex when it comes to randomness.
|
|
|
|
use na::RealField;
|
|
use num_complex::Complex;
|
|
use quickcheck::{Arbitrary, Gen};
|
|
use rand::distributions::{Distribution, Standard};
|
|
use rand::Rng;
|
|
|
|
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
|
|
pub struct RandComplex<N>(pub Complex<N>);
|
|
|
|
impl<N: Arbitrary + RealField> Arbitrary for RandComplex<N> {
|
|
#[inline]
|
|
fn arbitrary(rng: &mut Gen) -> Self {
|
|
let im = Arbitrary::arbitrary(rng);
|
|
let re = Arbitrary::arbitrary(rng);
|
|
RandComplex(Complex::new(re, im))
|
|
}
|
|
}
|
|
|
|
impl<N: RealField> Distribution<RandComplex<N>> for Standard
|
|
where
|
|
Standard: Distribution<N>,
|
|
{
|
|
#[inline]
|
|
fn sample<'a, G: Rng + ?Sized>(&self, rng: &'a mut G) -> RandComplex<N> {
|
|
let re = rng.gen();
|
|
let im = rng.gen();
|
|
RandComplex(Complex::new(re, im))
|
|
}
|
|
}
|
|
|
|
// This is a wrapper similar to RandComplex, but for non-complex.
|
|
// This exists only to make generic tests easier to write.
|
|
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
|
|
pub struct RandScalar<N>(pub N);
|
|
|
|
impl<N: Arbitrary> Arbitrary for RandScalar<N> {
|
|
#[inline]
|
|
fn arbitrary(rng: &mut Gen) -> Self {
|
|
RandScalar(Arbitrary::arbitrary(rng))
|
|
}
|
|
}
|
|
|
|
impl<N: RealField> Distribution<RandScalar<N>> for Standard
|
|
where
|
|
Standard: Distribution<N>,
|
|
{
|
|
#[inline]
|
|
fn sample<'a, G: Rng + ?Sized>(&self, rng: &'a mut G) -> RandScalar<N> {
|
|
RandScalar(self.sample(rng))
|
|
}
|
|
}
|