mindus 3.0.1

A library for working with mindustry data formats (eg schematics and maps) (fork of plandustry)
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

#![feature(let_chains)]
use image::codecs::png::PngDecoder;
use image::DynamicImage;
use std::fs::File;
use std::io::{BufReader, Write as _};
use std::iter::Iterator;
use std::path::Path;
use walkdir::WalkDir;

macro_rules! wr {
    ($dst:expr => $($arg:tt)*) => { writeln!($dst, $($arg)*).unwrap() };
}

fn kebab2bigsnek(kebab: &str) -> String {
    let mut n = String::new();
    n.reserve(kebab.len());
    for c in kebab.chars() {
        if c == '-' {
            n.push('_');
        } else {
            n.push(c.to_ascii_uppercase());
        }
    }
    n
}

fn main() {
    let _ = std::fs::remove_dir_all("target/out");
    let walkdir = WalkDir::new("assets");
    println!("cargo:rerun-if-changed=assets/");
    println!("cargo:rerun-if-changed=build.rs");
    let o = std::env::var("OUT_DIR").unwrap();
    let o = Path::new(&o);
    let mut full = File::create(o.join("full.rs")).unwrap();
    // let mut half = File::create(o.join("half.rs")).unwrap();
    let mut quar = File::create(o.join("quar.rs")).unwrap();
    let mut eigh = File::create(o.join("eigh.rs")).unwrap();
    let mut n = 5usize;

    wr!(full => "pub mod full {{");
    wr!(full => "pub static EMPTY: LazyLock<RgbaImage> = LazyLock::new(|| RgbaImage::new(32, 32));");

    wr!(quar => "pub mod quar {{");
    wr!(quar => "pub static EMPTY: LazyLock<RgbaImage> = LazyLock::new(|| RgbaImage::new(8, 8));");

    wr!(eigh => "pub mod eigh {{");
    wr!(eigh => "pub static EMPTY: LazyLock<RgbaImage> = LazyLock::new(|| RgbaImage::new(4, 4));");

    for mut file in [&full, &quar, &eigh] {
        file.write_all(b"macro_rules!img{($v:expr)=>{{static TMP:LazyLock<RgbaImage>=LazyLock::new(||$v);&TMP}};}\n").unwrap();
        wr!(file => "use image::RgbaImage;");
        wr!(file => "use crate::utils::Lock as LazyLock;");
    }
    for i in 1..=16 {
        n += 1;
        wr!(full => "pub static BUILD{}: &LazyLock<RgbaImage> = &EMPTY;", i);
        wr!(quar => "pub static BUILD{}: &LazyLock<RgbaImage> = &EMPTY;", i);
        wr!(eigh => "pub static BUILD{}: &LazyLock<RgbaImage> = &EMPTY;", i);
    }

    let mut warmup = File::create(o.join("warmup.rs")).unwrap();
    wr!(warmup => "/// SAFETY: this function must only be called once.");
    wr!(warmup => "pub unsafe fn warmup() {{");
    wr!(warmup => "LazyLock::load(&EMPTY);");
    for e in walkdir.into_iter().filter_map(|e| e.ok()) {
        let path = e.path();
        if path.is_file() && let Some(e) = path.extension() && e == "png" {
            let p = DynamicImage::from_decoder(PngDecoder::new(BufReader::new(File::open(path).unwrap())).unwrap()).unwrap().into_rgba8();
            let f = path.to_str().unwrap();
            if f.contains("-bottom.png") || f.contains("-liquid.png") || f.contains("-team") {
                continue;
            }
            let path = path.with_extension("");
            let path = kebab2bigsnek(path.file_name().unwrap().to_str().unwrap());
            macro_rules! writ {
                ($ext:ident / $scale:literal) => {
                    let mut buf = File::create(o.join(n.to_string() + "-" + stringify!($ext))).unwrap();
                    let new = if $scale == 1 {
                        p.clone()
                    } else {
                        // boulders
                        let (mx, my) = if p.width() + p.height() == 48+48 {
                            (32, 32)
                        // vents (dont match VENT_CONDENSER, do match (RHYOLITE_VENT)
                        } else if path.contains("_VENT") {
                            (32, 32)
                        } else {
                            (p.height(), p.width())
                        };
                        image::imageops::resize(
                            &p,
                            mx / $scale,
                            my / $scale,
                            image::imageops::Nearest,
                        )
                    };
                    let x = new.width();
                    let y = new.height();
                    buf.write_all(&new.into_raw()).unwrap();
                    wr!($ext =>
                        r#"pub static {path}: &LazyLock<RgbaImage> = img!(unsafe {{ RgbaImage::from_vec({x}, {y}, include_bytes!(concat!(env!("OUT_DIR"), "/{n}-{}")).to_vec()).unwrap_unchecked() }});"#,
                        stringify!($ext)
                    );
                };
            }
            writ!(full / 1);
            // writ!(half + 0.5);
            writ!(quar / 4);
            writ!(eigh / 8);
            wr!(warmup => "LazyLock::load({path});");
            n += 1;
        }
    }
    warmup.write_all(b"}").unwrap();
    for mut f in [full, eigh, quar] {
        // brazillian literal
        f.write_all(br#"include!(concat!(env!("OUT_DIR"), "/warmup.rs"));"#)
            .unwrap();
        f.write_all(b"}").unwrap();
    }
}