jbonsai 0.4.1

Voice synthesis library for Text-to-Speech applications (Currently HTS Engine rewrite in Rust language).
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

use super::{
    coefficients::{Coefficients, GeneralizedCoefficients},
    mglsa::MelGeneralizedLogSpectrumApproximation,
    mlsa::MelLogSpectrumApproximation,
};

/// Stage's variant is mostly Zero, therefore I decided to ignore large_enum_variant.
#[derive(Debug, Clone)]
#[allow(clippy::large_enum_variant)]
pub enum Stage {
    NonZero {
        stage: usize,
        gamma: f64,
        coefficients: GeneralizedCoefficients,
        filter: MelGeneralizedLogSpectrumApproximation,
    },
    Zero {
        coefficients: Coefficients,
        filter: MelLogSpectrumApproximation<6>,
    },
}

impl Stage {
    pub fn new(stage: usize, nmcp: usize) -> Self {
        if stage == 0 {
            Self::Zero {
                coefficients: Coefficients::new(&[]),
                filter: MelLogSpectrumApproximation::new(nmcp),
            }
        } else {
            let gamma = -1.0 / stage as f64;
            Self::NonZero {
                stage,
                gamma,
                coefficients: GeneralizedCoefficients::new(&[], gamma),
                filter: MelGeneralizedLogSpectrumApproximation::new(stage, nmcp),
            }
        }
    }
}