haqumei 0.2.2

Haqumei is a Japanese Grapheme-to-Phoneme (G2P) library implemented in Rust.
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

macro_rules! impl_batch_method_haqumei {
    (
        $(#[$meta:meta])*
        $batch_method:ident => $inner_method:ident -> $ret_type:ty
    ) => {
        $(#[$meta])*
        ///
        /// `HaqumeiOptions` の `modify_kanji_yomi` が無効な場合、マルチスレッドで処理を行います。
        /// 有効な場合は、シングルスレッドでの逐次処理にフォールバックします。
        pub fn $batch_method<S>(&mut self, texts: &[S]) -> Result<Vec<$ret_type>, HaqumeiError>
        where
            S: AsRef<str> + Sync,
        {
            if !self.options.modify_kanji_yomi {
                let dict = GLOBAL_MECAB_DICTIONARY.load_full();
                if !dict.model.is_initialized() {
                    return Err(HaqumeiError::GlobalDictionaryNotInitialized);
                }
                let options = self.options;

                return texts
                    .par_iter()
                    .map_init(
                        || {
                            let ojt = OpenJTalk::from_shared_dictionary(dict.clone())
                                .expect("Failed to initialize OpenJTalk worker");
                            Haqumei::from_open_jtalk(ojt, options).unwrap()
                        },
                        |haqumei, text| haqumei.$inner_method(text.as_ref()),
                    )
                    .collect();
            }

            texts.iter().map(|text| self.$inner_method(text.as_ref())).collect()
        }
    };
}

macro_rules! impl_batch_method_openjtalk {
    (
        $(#[$meta:meta])*
        $batch_method:ident => $inner_method:ident -> $ret_type:ty
    ) => {
        $(#[$meta])*
        ///
        #[doc = concat!(
            "複数のテキストに対して並行して `",
            stringify!($batch_method),
            "` を実行します。"
        )]
        pub fn $batch_method<S>(&mut self, texts: &[S]) -> Result<Vec<$ret_type>, HaqumeiError>
        where
            S: AsRef<str> + Sync,
        {
            let dict = GLOBAL_MECAB_DICTIONARY.load_full();
            if !dict.model.is_initialized() {
                return Err(HaqumeiError::GlobalDictionaryNotInitialized);
            }

            texts
            .par_iter()
            .map_init(
                || {
                    OpenJTalk::from_shared_dictionary(dict.clone())
                        .expect("Failed to initialize OpenJTalk worker")
                },
                |ojt, text| ojt.$inner_method(text.as_ref()),
            )
            .collect()
        }
    };
}