misaki-rs 0.4.0

A self-contained, POS-aware Grapheme-to-Phoneme (G2P) engine for Rust, optimized for TTS models like Kokoro.
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

use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MToken {
    pub text: String,
    pub tag: String,
    pub whitespace: String,
    pub phonemes: Option<String>,
    pub start_ts: Option<f64>,
    pub end_ts: Option<f64>,
    #[serde(rename = "_")]
    pub underscore: Option<Underscore>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct Underscore {
    pub is_head: bool,
    pub alias: Option<String>,
    pub stress: Option<f64>,
    pub currency: Option<String>,
    pub num_flags: String,
    pub prespace: bool,
    pub rating: Option<i32>,
}

impl MToken {
    pub fn new(text: String, tag: String, whitespace: String) -> Self {
        Self {
            text,
            tag,
            whitespace,
            phonemes: None,
            start_ts: None,
            end_ts: None,
            underscore: Some(Underscore {
                is_head: true,
                num_flags: String::new(),
                prespace: false,
                ..Default::default()
            }),
        }
    }

    pub fn underscore_mut(&mut self) -> &mut Underscore {
        if self.underscore.is_none() {
            self.underscore = Some(Underscore::default());
        }
        self.underscore.as_mut().unwrap()
    }

    pub fn underscore(&self) -> &Underscore {
        static DEFAULT_UNDERSCORE: Underscore = Underscore {
            is_head: false,
            alias: None,
            stress: None,
            currency: None,
            num_flags: String::new(),
            prespace: false,
            rating: None,
        };
        self.underscore.as_ref().unwrap_or(&DEFAULT_UNDERSCORE)
    }
}