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
40
41
42
43
44
45
46
47
48
49

use flo_curves::{bezier, BezierCurveFactory, Coord2, Geo, Line};
use rand::Rng;

use crate::parser::{Pattern, Cell};

/// Ported method for extracting colour information from height from [the official editor](https://gitlab.com/PITR_DEV/ultrakill.pattern-editor/-/blob/master/src/helpers/ColorHelper.js#L5)
pub fn evaluate_height(height: i32) -> f64 {
    let emulated_x = ((height + 10) as f64 / 30.0).clamp(0.0, f64::MAX);
    let col_curve = bezier::Curve::from_points(
        Coord2(0.0, 0.0),
        (Coord2(0.0, 255.0), Coord2(0.6, 0.0)),
        Coord2(1.0, 255.0),
    );
    let line = ColLine::from_points(Coord2(emulated_x, 0.0), Coord2(emulated_x, 255.0));

    bezier::curve_intersects_line(&col_curve, &line)[0].0 * 255.0
}

/// Generate a pattern with random heights in the range -10..=20, and random prefabs for each cell
pub fn random_pattern() -> Pattern {
    let mut rng = rand::thread_rng();
    let mut pattern = Pattern(vec![vec![Cell::new(); 16]; 16]);

    for row in pattern.0.iter_mut() {
        for cell in row {
            cell.set_height(rng.gen_range(-10..=20));
            cell.set_prefab(rand::random())
        }
    }

    pattern
}

#[doc(hidden)]
pub struct ColLine(Coord2, Coord2);

impl Geo for ColLine {
    type Point = Coord2;
}

impl Line for ColLine {
    fn from_points(p1: Self::Point, p2: Self::Point) -> Self {
        ColLine(p1, p2)
    }

    fn points(&self) -> (Self::Point, Self::Point) {
        (self.0, self.1)
    }
}