nucleation 0.3.16

A high-performance Minecraft schematic parser and utility library
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use super::Shape;

#[derive(Clone)]
pub struct Disk {
    pub center: (f64, f64, f64),
    pub radius: f64,
    pub normal: (f64, f64, f64),
    pub thickness: f64,
    normal_len: f64,
}

impl Disk {
    pub fn new(
        center: (f64, f64, f64),
        radius: f64,
        normal: (f64, f64, f64),
        thickness: f64,
    ) -> Self {
        let normal_len = (normal.0 * normal.0 + normal.1 * normal.1 + normal.2 * normal.2).sqrt();
        Self {
            center,
            radius,
            normal,
            thickness,
            normal_len,
        }
    }

    fn normalized_normal(&self) -> (f64, f64, f64) {
        if self.normal_len == 0.0 {
            (0.0, 1.0, 0.0)
        } else {
            (
                self.normal.0 / self.normal_len,
                self.normal.1 / self.normal_len,
                self.normal.2 / self.normal_len,
            )
        }
    }
}

impl Shape for Disk {
    fn contains(&self, x: i32, y: i32, z: i32) -> bool {
        let n = self.normalized_normal();
        let dx = x as f64 - self.center.0;
        let dy = y as f64 - self.center.1;
        let dz = z as f64 - self.center.2;

        let plane_dist = (dx * n.0 + dy * n.1 + dz * n.2).abs();
        if plane_dist > self.thickness / 2.0 {
            return false;
        }

        let rx = dx - (dx * n.0 + dy * n.1 + dz * n.2) * n.0;
        let ry = dy - (dx * n.0 + dy * n.1 + dz * n.2) * n.1;
        let rz = dz - (dx * n.0 + dy * n.1 + dz * n.2) * n.2;
        let radial = (rx * rx + ry * ry + rz * rz).sqrt();
        radial <= self.radius
    }

    fn points(&self) -> Vec<(i32, i32, i32)> {
        let mut points = Vec::new();
        self.for_each_point(|x, y, z| points.push((x, y, z)));
        points
    }

    fn normal_at(&self, _x: i32, _y: i32, _z: i32) -> (f64, f64, f64) {
        self.normalized_normal()
    }

    fn bounds(&self) -> (i32, i32, i32, i32, i32, i32) {
        let r = self.radius.ceil() as i32 + 1;
        let t = (self.thickness / 2.0).ceil() as i32 + 1;
        let b = r + t;
        let cx = self.center.0.round() as i32;
        let cy = self.center.1.round() as i32;
        let cz = self.center.2.round() as i32;
        (cx - b, cy - b, cz - b, cx + b, cy + b, cz + b)
    }

    fn for_each_point<F>(&self, mut f: F)
    where
        F: FnMut(i32, i32, i32),
    {
        let (min_x, min_y, min_z, max_x, max_y, max_z) = self.bounds();
        for x in min_x..=max_x {
            for y in min_y..=max_y {
                for z in min_z..=max_z {
                    if self.contains(x, y, z) {
                        f(x, y, z);
                    }
                }
            }
        }
    }
}