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
|
use crate::ctx::Ctx;
/// Represents the data present in a quad-tree node.
/// May be the base-level repr, or a node with 4 children.
pub enum Quad<A: Default, B: Copy> {
/// Base cell in grid.
/// May actually be a chunk of cells for performance.
Base(A),
/// Node with 4 children.
Node(Box<[Node<A, B>;4]>),
/// Children may be generated from Node.
Cached,
}
/// Represends a node in a quadtree.
/// Has a depth denoting the number of nodes below it.
/// Nodes should only be siblings of nodes with the same depth.
/// Data stored inside a quadtree node, including children, are in `data`.
pub struct Node<A: Default, B: Copy> {
pub depth: usize,
pub compr: B,
pub data: Quad<A, B>,
}
impl<A: Default, B: Copy> Node<A, B> {
/// Creates a new tree from a single base node
pub fn new_base(base: A, ctx: &mut Ctx) -> Self {
Node {
depth: 0,
compr: ctx.compress_base(&base),
data: Quad::Base(base),
}
}
/// Creates a new tree with a single empty base node
pub fn new_empty(ctx: &mut Ctx) -> Self {
Self::new_base(Default::default(), ctx)
}
pub fn new_cached(compr: B, depth: usize) -> Self {
Node {
depth,
compr,
data: Quad::Cached,
}
}
/// Creates a new node double the size by centering the current node
/// on a node double the size.
pub fn pad_empty(self, ctx: &mut Ctx) -> Self {
todo!()
}
}
|