mercs2_probe 2.2.0

Mercenaries 2 modernization — headless WAD diagnostic/export tool. Runs the carved-out diagnostics from mercs2_engine::diag as subcommands (no window).
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

//! Dev bin: decode a model's diffuse textures (resident low-res AND hi-res streamed) to BMP so we can
//! eyeball whether the hi-res streaming assembly is correct (e.g. the PMC floor "shifted/transposed"
//! report). BMP is uncompressed → no extra deps. Separate binary so it isn't locked by a running game.
//!
//!   cargo run -p mercs2_probe --bin tex_dump -- 0x39AF17DC <out_dir>

use mercs2_engine::wad;
use mercs2_formats::texture::{TexFormat, TextureData};

fn rgb565(c: u16) -> [u8; 3] {
    let r = ((c >> 11) & 0x1f) as u32;
    let g = ((c >> 5) & 0x3f) as u32;
    let b = (c & 0x1f) as u32;
    [((r * 255 + 15) / 31) as u8, ((g * 255 + 31) / 63) as u8, ((b * 255 + 15) / 31) as u8]
}

/// Decode BC1/BC3 mip0 → RGB (ignores alpha; colour endpoints are what we care about here).
fn decode_mip0(td: &TextureData) -> Option<(u32, u32, Vec<[u8; 3]>)> {
    let (w, h) = (td.width as usize, td.height as usize);
    let (block_bytes, color_off) = match td.format {
        TexFormat::Bc1 => (8usize, 0usize),
        TexFormat::Bc3 => (16usize, 8usize),
    };
    let bw = (w + 3) / 4;
    let bh = (h + 3) / 4;
    if td.mip0.len() < bw * bh * block_bytes {
        return None;
    }
    let mut px = vec![[0u8; 3]; w * h];
    for by in 0..bh {
        for bx in 0..bw {
            let blk = &td.mip0[(by * bw + bx) * block_bytes + color_off..];
            let c0 = u16::from_le_bytes([blk[0], blk[1]]);
            let c1 = u16::from_le_bytes([blk[2], blk[3]]);
            let (e0, e1) = (rgb565(c0), rgb565(c1));
            let lerp = |a: [u8; 3], b: [u8; 3], t: u32| {
                [((a[0] as u32 * (3 - t) + b[0] as u32 * t) / 3) as u8,
                 ((a[1] as u32 * (3 - t) + b[1] as u32 * t) / 3) as u8,
                 ((a[2] as u32 * (3 - t) + b[2] as u32 * t) / 3) as u8]
            };
            // BC3 colour is always 4-colour; BC1 uses the c0>c1 rule.
            let pal = if td.format == TexFormat::Bc1 && c0 <= c1 {
                [e0, e1, [((e0[0] as u16 + e1[0] as u16) / 2) as u8, ((e0[1] as u16 + e1[1] as u16) / 2) as u8, ((e0[2] as u16 + e1[2] as u16) / 2) as u8], [0, 0, 0]]
            } else {
                [e0, e1, lerp(e0, e1, 1), lerp(e0, e1, 2)]
            };
            let bits = u32::from_le_bytes([blk[4], blk[5], blk[6], blk[7]]);
            for py in 0..4 {
                for pxi in 0..4 {
                    let (x, y) = (bx * 4 + pxi, by * 4 + py);
                    if x < w && y < h {
                        let idx = ((bits >> (2 * (py * 4 + pxi))) & 3) as usize;
                        px[y * w + x] = pal[idx];
                    }
                }
            }
        }
    }
    Some((td.width, td.height, px))
}

fn write_bmp(path: &str, w: u32, h: u32, px: &[[u8; 3]]) {
    let row_pad = (4 - (w as usize * 3) % 4) % 4;
    let img_size = (w as usize * 3 + row_pad) * h as usize;
    let mut f = Vec::with_capacity(54 + img_size);
    f.extend_from_slice(b"BM");
    f.extend_from_slice(&(54 + img_size as u32).to_le_bytes());
    f.extend_from_slice(&0u32.to_le_bytes());
    f.extend_from_slice(&54u32.to_le_bytes());
    f.extend_from_slice(&40u32.to_le_bytes());
    f.extend_from_slice(&(w as i32).to_le_bytes());
    f.extend_from_slice(&(h as i32).to_le_bytes()); // positive = bottom-up
    f.extend_from_slice(&1u16.to_le_bytes());
    f.extend_from_slice(&24u16.to_le_bytes());
    f.extend_from_slice(&0u32.to_le_bytes());
    f.extend_from_slice(&(img_size as u32).to_le_bytes());
    f.extend_from_slice(&[0u8; 16]);
    for y in (0..h as usize).rev() {
        for x in 0..w as usize {
            let p = px[y * w as usize + x];
            f.extend_from_slice(&[p[2], p[1], p[0]]); // BGR
        }
        f.extend(std::iter::repeat(0u8).take(row_pad));
    }
    std::fs::write(path, f).unwrap();
}

fn main() {
    let args: Vec<String> = std::env::args().collect();
    let mhash = args.get(1).and_then(|a| a.strip_prefix("0x")).and_then(|h| u32::from_str_radix(h, 16).ok()).unwrap_or(0x39AF17DC);
    let out = args.get(2).cloned().unwrap_or_else(|| ".".into());
    let mut w = wad::resolve_vz_wad(None).and_then(|p| wad::open(&p).ok()).expect("open vz.wad");

    // Work at the MTRL level: every material carries a FULL texture list (all slots), not just the
    // diffuse(0)/normal(2) the mesh builder keeps. Dump EVERY texture in EVERY slot so nothing is hidden.
    let container = wad::extract_container(&mut w, mhash).expect("extract container");
    let mats = mercs2_formats::texture::parse_mtrl(&container);
    println!("model 0x{mhash:08X}: {} materials (slot roles: 0=diffuse, 1=?middle, 2=normal)", mats.len());
    let mut texs: std::collections::BTreeSet<u32> = std::collections::BTreeSet::new();
    for (i, mat) in mats.iter().enumerate() {
        let slots: Vec<String> = mat.textures.iter().map(|h| format!("0x{h:08X}")).collect();
        println!("  material {i:2}: {} slots [{}]", mat.textures.len(), slots.join(", "));
        for &h in &mat.textures {
            if h != 0 && h != 0xFFFFFFFF {
                texs.insert(h);
            }
        }
    }
    // Metadata names: the texture's own NAME leaf (asset name), else a rainbow-table reverse of the hash.
    let rainbow = mercs2_engine::worldutil::rainbow_names(&texs);
    println!("== {} distinct textures; dumping (mip0), labelled by NAME leaf / rainbow reverse ==", texs.len());
    let (mut ok, mut named, mut fail) = (0u32, 0u32, 0u32);
    for th in &texs {
        let th = *th;
        // Container NAME leaf (borrow the wad's file+archive briefly).
        let leaf_name = {
            let (archive, file) = wad::archive_and_file(&mut w);
            mercs2_formats::texture::extract_texture_name(file, archive, th)
                .map(|s| s.trim().trim_end_matches('\0').to_string())
                .filter(|s| !s.is_empty())
        };
        let name = leaf_name.clone().or_else(|| rainbow.get(&th).cloned());
        if name.is_some() {
            named += 1;
        }
        match wad::extract_texture_hires(&mut w, th) {
            Ok(hi) => match decode_mip0(&hi) {
                Some((tw, tht, px)) => {
                    let tag = name
                        .as_deref()
                        .map(|n| {
                            let s: String = n.chars().map(|c| if c.is_ascii_alphanumeric() { c } else { '_' }).collect();
                            format!("_{}", &s[..s.len().min(48)])
                        })
                        .unwrap_or_default();
                    let p = format!("{out}/tex_{th:08X}{tag}_{tw}x{tht}.bmp");
                    write_bmp(&p, tw, tht, &px);
                    println!("  0x{th:08X}  {tw}x{tht} {:?}  name={}", hi.format, name.as_deref().unwrap_or("<none>"));
                    ok += 1;
                }
                None => { println!("  0x{th:08X}: decode failed"); fail += 1; }
            },
            Err(e) => { println!("  0x{th:08X}: unresolved ({e})"); fail += 1; }
        }
    }
    println!("== dumped {ok}, {named} named (of {}), {fail} unresolved ==", texs.len());
}