stochastic-rs-distributions 2.0.0

Probability distributions with SIMD bulk sampling.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

//! # Complex
//!
//! $$
//! \mathbb P(X\in A)=\int_A f_X(x)dx\ \text{or}\ \sum_{x\in A}p_X(x)
//! $$
//!
use num_complex::Complex;
use num_traits::Num;
use rand::Rng;
use rand::distr::Distribution;

#[derive(Clone, Copy, Debug)]
pub struct ComplexDistribution<Re, Im = Re> {
  re: Re,
  im: Im,
}

impl<Re, Im> ComplexDistribution<Re, Im> {
  pub fn new(re: Re, im: Im) -> Self {
    ComplexDistribution { re, im }
  }
}

impl<T, Re, Im> Distribution<Complex<T>> for ComplexDistribution<Re, Im>
where
  T: Num + Clone,
  Re: Distribution<T>,
  Im: Distribution<T>,
{
  fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> Complex<T> {
    Complex::new(self.re.sample(rng), self.im.sample(rng))
  }
}