romkit 0.1.0

Identify game ROMs and extract their data and assets as structured files.
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

//! A Gen 4 game as assets:
//!
//! ```text
//! species.json      every species and form, with evolutions and level-up moves
//! moves.json
//! abilities.json
//! text/<bank>.json  one file per text bank, its messages in markup form
//! sprites/pokemon/<variant>/<species>.png  80×80 battle sprites: front, back, with -shiny,
//!                                          and with -female where females look different
//! sprites/icons/<species>.png              32×32 box icons
//! sprites/trainers/<class>.png             80×80 trainer class sprites
//! sprites/items/<item>.png                 32×32 item icons
//! models/…, textures/…                    the overworld in 3D, see [`field`](super::field)
//! sound/...                                music as MIDI, samples and cries as WAV (`sound.rs`)
//! ```

use serde::Serialize;

use super::Gen4;
use crate::games::{Game, GameId, json};
use crate::io::ByteSource;
use crate::{Error, Result};

const TABLES: [&str; 3] = ["species.json", "moves.json", "abilities.json"];

#[derive(Serialize)]
struct TextBank {
    bank: usize,
    name: Option<&'static str>,
    /// See [`Message::markup`](super::text::Message::markup).
    messages: Vec<String>,
}

/// A bank's asset name: its descriptive name where known, else its number.
fn bank_stem(bank: usize, name: Option<&str>) -> String {
    name.map_or_else(|| format!("{bank:04}"), str::to_owned)
}

impl<S: ByteSource> Game for Gen4<S> {
    fn id(&self) -> GameId {
        self.game
    }

    fn assets(&self) -> Result<Vec<String>> {
        let text = self.text()?;
        let banks =
            (0..text.len()).map(|bank| format!("text/{}.json", bank_stem(bank, text.name(bank))));
        let tables = TABLES.into_iter().map(str::to_owned);
        let field = self.field_assets()?;
        Ok(tables
            .chain(banks)
            .chain(self.sprite_assets()?)
            .chain(field)
            .chain(self.sound_assets()?)
            .collect())
    }

    fn extract(&self, path: &str) -> Result<Vec<u8>> {
        if let Some(asset) = self.extract_field(path) {
            return asset;
        }
        match path {
            "species.json" => json(self.species()?),
            "moves.json" => json(self.moves()?),
            "abilities.json" => json(self.abilities()?),
            _ if path.starts_with("sprites/") => self.extract_sprite(path),
            _ if path.starts_with("sound/") => self.extract_sound(path),
            _ => {
                let text = self.text()?;
                let stem = path
                    .strip_prefix("text/")
                    .and_then(|file| file.strip_suffix(".json"));
                let bank = stem
                    .and_then(|stem| text.find(stem).or_else(|| stem.parse().ok()))
                    .filter(|&bank| bank < text.len())
                    .ok_or_else(|| Error::NotFound(path.to_owned()))?;
                json(&TextBank {
                    bank,
                    name: text.name(bank),
                    messages: text
                        .bank(bank)?
                        .iter()
                        .map(|message| message.markup())
                        .collect(),
                })
            }
        }
    }
}