bspa 0.1.1

Beam search packing algorithm
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

use std::collections::{HashMap, HashSet};

use bspa::{Area, Placement, Rect, RectGroup};
use image::{ImageBuffer, Pixel, Rgb};
use rand::prelude::*;

fn gen_color<const MIN: u8, const MAX: u8>(rng: &mut ThreadRng) -> Rgb<u8> {
    Rgb([
        rng.random_range(MIN..MAX),
        rng.random_range(MIN..MAX),
        rng.random_range(MIN..MAX),
    ])
}

fn gen_color_map(list: &HashSet<Rect>) -> HashMap<Rect, Rgb<u8>> {
    let mut rng = rand::rng();
    list.iter()
        .map(|&x| (x, gen_color::<64, 255>(&mut rng)))
        .collect::<HashMap<_, _>>()
}

pub fn save_rg(list: &HashSet<Rect>, data: &RectGroup, name: &str) {
    let color_map = gen_color_map(list);

    let w = data.w();
    let h = data.h();
    let mut atlas = ImageBuffer::from_pixel(w, h, Rgb([0u8, 0u8, 0u8]));

    for bp in data.list.iter() {
        let pixel = *color_map.get(&bp.item).unwrap();

        for x in 1..bp.item.w() - 1 {
            for y in 1..bp.item.h() - 1 {
                atlas.put_pixel(bp.x + x, bp.y + y, pixel);
            }
        }
    }
    atlas.save(name).unwrap();
}

pub fn save_pg(list: &HashSet<Rect>, data: &[Placement<RectGroup>], name: &str) {
    let mut rng = rand::rng();
    let color_map = gen_color_map(list);

    let w = data.iter().map(|p| p.x + p.w()).max().unwrap() * 2;
    let h = data.iter().map(|p| p.y + p.h()).max().unwrap() * 2;
    let mut atlas = ImageBuffer::from_pixel(w, h, Rgb([0u8, 0u8, 0u8]));

    for p in data {
        let tint = gen_color::<64, 192>(&mut rng);
        for r in p.item.list.iter() {
            let mut pixel = *color_map
                .get(&r.item)
                .unwrap_or_else(|| panic!("{:?}", r.item));
            pixel.blend(&tint);
            pixel.apply(|p| p.saturating_add(16));

            for x in 1..(r.item.w() * 2 - 1) {
                for y in 1..(r.item.h() * 2 - 1) {
                    atlas.put_pixel((p.x + r.x) * 2 + x, (p.y + r.y) * 2 + y, pixel);
                }
            }
        }
    }

    atlas.save(name).unwrap();
}