attract 0.0.0

A small library for visualsing dynamical systems, particularly strange attractors.
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
34
35
36
37
38
39

//! Implementation of a Circle generator.
//!
//! This generator samples points from a circular region in the complex plane using
//! a uniform distribution.

use nalgebra::Complex;
use num_traits::{Float, FloatConst};
use rand::{Rng, distr::uniform::SampleUniform};

use crate::Generator;

/// Generator for sampling points from a circular region in the complex plane.
#[derive(Debug, Clone, Copy)]
pub struct Circle<T> {
    /// Center point of the circle in the complex plane.
    centre: Complex<T>,
    /// Radius of the circle.
    radius: T,
}

impl<T: Float> Circle<T> {
    /// Creates a new `Circle` generator with the specified center and radius.
    #[inline]
    pub fn new(centre: Complex<T>, radius: T) -> Self {
        debug_assert!(radius > T::zero(), "Radius must be positive");
        Self { centre, radius }
    }
}

impl<T: Float + FloatConst + SampleUniform> Generator<T> for Circle<T> {
    #[inline]
    fn sample<R: Rng>(&self, rng: &mut R) -> Complex<T> {
        let theta = rng.random_range(T::zero()..T::TAU());
        let rho = rng.random_range(T::zero()..self.radius).sqrt();
        let re = self.centre.re + rho * theta.cos();
        let im = self.centre.im + rho * theta.sin();
        self.centre + Complex::new(re, im)
    }
}