wallrnd 1.2.2

use crate::prelude::*;
use rand::{rngs::ThreadRng, Rng};

#[derive(Clone, Copy, Debug)]
pub struct SaltItem {
    pub color: Color,
    pub likeliness: f64,
    pub variability: usize,
}

#[derive(Clone, Debug, Default)]
pub struct Salt(pub Vec<SaltItem>);

impl SaltItem {
    fn sample(&self, rng: &mut ThreadRng) -> Option<Color> {
        if rng.gen::<f64>() < self.likeliness {
            Some(self.color.variate(rng, self.variability))
        } else {
            None
        }
    }
}

impl Salt {
    pub fn sample(&self, rng: &mut ThreadRng) -> Option<Color> {
        for item in self.0.iter() {
            if let Some(c) = item.sample(rng) {
                return Some(c);
            }
        }
        None
    }

    pub fn none() -> Self {
        Self(Vec::new())
    }
}