shapemaker / src / random / region.rs

use crate::{Containable, Point, Region};
use rand::{Rng, distr::uniform::SampleRange, seq::IteratorRandom};

impl Region {
    pub fn random_end(&self, rng: &mut impl Rng, start: Point) -> Point {
        // End anchors are always a square diagonal from the start anchor (for now)
        // that means taking steps of the form n * (one of (1, 1), (1, -1), (-1, 1), (-1, -1))
        // Except that the end anchor needs to stay in the bounds of the shape.

        // Determine all possible end anchors that are in a square diagonal from the start anchor
        let mut possible_end_anchors = vec![];

        // shapes can end on the next cell, since that's where they end
        let actual_region = self.enlarged(1, 1);

        for x in actual_region.mirrored_width_range() {
            for y in actual_region.mirrored_height_range() {
                let end_anchor = start.translated(x, y);

                if end_anchor == start {
                    continue;
                }

                // Check that the end anchor is in a square diagonal from the start anchor and that the end anchor is in bounds
                if x.abs() == y.abs() && actual_region.contains(&end_anchor) {
                    possible_end_anchors.push(end_anchor);
                }
            }
        }

        // Pick a random end anchor from the possible end anchors
        possible_end_anchors[rng.random_range(0..possible_end_anchors.len())]
    }

    pub fn random(rng: &mut impl Rng, within: &Region) -> Self {
        let start = Point::random(rng, within);
        let end = within.random_end(rng, start);
        Region::from(if start.x() > end.x() {
            (end, start)
        } else {
            (start, end)
        })
    }

    pub fn random_point(&self, rng: &mut impl Rng) -> Point {
        Point::random(rng, self)
    }

    pub fn random_point_except(
        &self,
        rng: &mut impl Rng,
        except: &Region,
    ) -> Point {
        self.iter()
            .filter(|p| !except.contains(p))
            .choose(rng)
            .unwrap()
    }
}

impl SampleRange<Point> for Region {
    fn is_empty(&self) -> bool {
        self.is_empty()
    }

    fn sample_single<R: rand::RngCore + ?Sized>(
        self,
        rng: &mut R,
    ) -> Result<Point, rand::distr::uniform::Error> {
        if self.is_empty() {
            return Err(rand::distr::uniform::Error::EmptyRange);
        }

        Ok(Point::Corner(
            rng.random_range(self.x_range()),
            rng.random_range(self.y_range()),
        ))
    }
}