#[cfg(any(feature = "rand", feature = "num-std", feature = "num-libm"))]
use super::Signal;
#[cfg(feature = "rand")]
use rand::{distributions::Distribution, Rng};
#[cfg(any(feature = "num-std", feature = "num-libm"))]
use crate::num::Phase;
#[cfg(feature = "rand")]
#[derive(Clone)]
pub struct Random<R, D, T> {
pub(super) rng: R,
pub(super) dist: D,
pub(super) _marker: core::marker::PhantomData<T>,
}
#[cfg(feature = "rand")]
impl<R, D, T> Signal for Random<R, D, T>
where
R: Rng,
D: Distribution<T>,
{
type Type = T;
#[inline]
fn next(&mut self) -> Self::Type {
self.dist.sample(&mut self.rng)
}
}
#[cfg(any(feature = "num-std", feature = "num-libm"))]
#[derive(Clone)]
pub struct Simplex<S: Signal<Type = f64>> {
perm: [u8; 256],
phase: Phase<S>,
}
#[cfg(any(feature = "num-std", feature = "num-libm"))]
impl<S: Signal<Type = f64>> Simplex<S> {
#[inline]
pub fn default(steps: S) -> Self {
Self {
perm: [
151, 160, 137, 91, 90, 15, 131, 13, 201, 95, 96, 53, 194, 233, 7, 225, 140, 36,
103, 30, 69, 142, 8, 99, 37, 240, 21, 10, 23, 190, 6, 148, 247, 120, 234, 75, 0,
26, 197, 62, 94, 252, 219, 203, 117, 35, 11, 32, 57, 177, 33, 88, 237, 149, 56, 87,
174, 20, 125, 136, 171, 168, 68, 175, 74, 165, 71, 134, 139, 48, 27, 166, 77, 146,
158, 231, 83, 111, 229, 122, 60, 211, 133, 230, 220, 105, 92, 41, 55, 46, 245, 40,
244, 102, 143, 54, 65, 25, 63, 161, 1, 216, 80, 73, 209, 76, 132, 187, 208, 89, 18,
169, 200, 196, 135, 130, 116, 188, 159, 86, 164, 100, 109, 198, 173, 186, 3, 64,
52, 217, 226, 250, 124, 123, 5, 202, 38, 147, 118, 126, 255, 82, 85, 212, 207, 206,
59, 227, 47, 16, 58, 17, 182, 189, 28, 42, 223, 183, 170, 213, 119, 248, 152, 2,
44, 154, 163, 70, 221, 153, 101, 155, 167, 43, 172, 9, 129, 22, 39, 253, 19, 98,
108, 110, 79, 113, 224, 232, 178, 185, 112, 104, 218, 246, 97, 228, 251, 34, 242,
193, 238, 210, 144, 12, 191, 179, 162, 241, 81, 51, 145, 235, 249, 14, 239, 107,
49, 192, 214, 31, 181, 199, 106, 157, 184, 84, 204, 176, 115, 121, 50, 45, 127, 4,
150, 254, 138, 236, 205, 93, 222, 114, 67, 29, 24, 72, 243, 141, 128, 195, 78, 66,
215, 61, 156, 180,
],
phase: Phase::new(steps, 256f64),
}
}
#[cfg(feature = "rand")]
pub fn with_rng<R: rand::Rng + ?Sized>(rng: &mut R, steps: S) -> Self {
use core::mem::MaybeUninit;
use rand::seq::SliceRandom;
let mut perm = [MaybeUninit::uninit(); 256];
for i in 0..=255u8 {
perm[i as usize] = MaybeUninit::new(i);
}
perm.shuffle(rng);
Self {
perm: unsafe { core::mem::transmute(perm) },
phase: Phase::new(steps, 256f64),
}
}
}
#[cfg(any(feature = "num-std", feature = "num-libm"))]
impl<S> Signal for Simplex<S>
where
S: Signal<Type = f64>,
{
type Type = f64;
fn next(&mut self) -> Self::Type {
fn hash(perm: &[u8; 256], i: i64) -> u8 {
unsafe { *perm.get_unchecked(i as u8 as usize) }
}
fn grad(hash: i64, x: f64) -> f64 {
let h = hash & 0x0F; let mut grad = (1 + (h & 7)) as f64; if (h & 8) != 0 {
grad = -grad;
}
grad * x
}
let phase = self.phase.next_phase();
let phase_floor = num_traits::Float::floor(phase);
let i0 = phase_floor as i64;
let i1 = i0 + 1;
let x0 = phase - phase_floor;
let x1 = x0 - 1.0;
let mut t0 = 1.0 - x0 * x0;
t0 *= t0;
let n0 = t0 * t0 * grad(hash(&self.perm, i0) as i64, x0);
let mut t1 = 1.0 - x1 * x1;
t1 *= t1;
let n1 = t1 * t1 * grad(hash(&self.perm, i1) as i64, x1);
0.395 * (n0 + n1)
}
}