Function source_stft_bct

src/framework/modules/hift_vocoder.cpp:483–513

}

std::vector<float> source_stft_bct(
    const HiftVocoderWeights & weights,
    const std::vector<float> & source,
    int64_t & stft_frames,
    std::vector<float> & window) {
    const audio::STFTConfig window_config{
        weights.config.istft_n_fft,
        weights.config.istft_hop,
        weights.config.istft_n_fft,
        true,
        audio::STFTPadMode::Reflect,
        audio::STFTFamily::Kokoro,
    };
    window = audio::get_cached_stft_window(window_config);
    const auto complex = audio::STFT().compute_complex(
        source,
        window,
        1,
        static_cast<int64_t>(source.size()),
        {weights.config.istft_n_fft, weights.config.istft_hop, weights.config.istft_n_fft});
    stft_frames = complex.shape[2];
    std::vector<float> stft(static_cast<size_t>(kHiftStftChannels * stft_frames), 0.0F);
    for (int64_t f = 0; f < kHiftStftBins; ++f) {
        for (int64_t t = 0; t < stft_frames; ++t) {
            const size_t complex_index = static_cast<size_t>((f * stft_frames + t) * 2);
            stft[static_cast<size_t>(f * stft_frames + t)] = complex.values[complex_index];
            stft[static_cast<size_t>((kHiftStftBins + f) * stft_frames + t)] = complex.values[complex_index + 1];
        }
    }
    return stft;
}

std::vector<float> waveform_from_post(
    const HiftVocoderWeights & weights,