Function main

tests/unittests/test_audio_utility_api.cpp:63–119

}  // namespace

int main() {
    try {
        const auto root = std::filesystem::temp_directory_path() / "audio_cpp_audio_utility_api_test";
        std::filesystem::remove_all(root);
        std::filesystem::create_directories(root);
        const engine::audio::AudioUtilityPaths paths{repo_path("assets/framework/audio_utilities"), cuda_backend()};

        const auto input48 = root / "input48.wav";
        engine::audio::write_pcm16_wav(input48, 48000, 1, tone(48000, 1920, 220.0f));
        const auto rnnoise_out = root / "rnnoise.wav";
        engine::audio::denoise_file(input48, rnnoise_out, "rnnoise", paths);
        require_wav_shape(rnnoise_out, 48000);

        const auto dfn_out = root / "deepfilternet2.wav";
        engine::audio::denoise_file(input48, dfn_out, "deepfilternet2", paths);
        require_wav_shape(dfn_out, 48000);

        const auto input16 = root / "input16.wav";
        engine::audio::write_pcm16_wav(input16, 16000, 1, tone(16000, 800, 330.0f));
        const auto zipenhancer_out = root / "zipenhancer.wav";
        engine::audio::denoise_file(input16, zipenhancer_out, "zipenhancer", paths);
        require_wav_shape(zipenhancer_out, 16000);

        const auto flashsr_out = root / "flashsr.wav";
        engine::audio::super_resolve_file(input16, flashsr_out, "flashsr", paths);
        require_wav_shape(flashsr_out, 48000);

        const auto input_dir = root / "batch_in";
        const auto output_dir = root / "batch_out";
        std::filesystem::create_directories(input_dir);
        engine::audio::write_pcm16_wav(input_dir / "one.wav", 48000, 1, tone(48000, 1440, 200.0f));
        engine::audio::write_pcm16_wav(input_dir / "two.wav", 48000, 1, tone(48000, 1440, 300.0f));
        const auto rnnoise_batch = engine::audio::denoise_directory(input_dir, output_dir / "rnnoise", "rnnoise", paths);
        require(rnnoise_batch.outputs.size() == 2, "RNNoise directory output count mismatch");
        for (const auto & output : rnnoise_batch.outputs) {
            require_wav_shape(output, 48000);
        }

        const auto zipenhancer_batch = engine::audio::denoise_directory(input_dir, output_dir / "zipenhancer", "zipenhancer", paths);
        require(zipenhancer_batch.outputs.size() == 2, "ZipEnhancer directory output count mismatch");
        for (const auto & output : zipenhancer_batch.outputs) {
            require_wav_shape(output, 16000);
        }

        const auto flashsr_batch = engine::audio::super_resolve_directory(input_dir, output_dir / "flashsr", "flashsr", paths);
        require(flashsr_batch.outputs.size() == 2, "FlashSR directory output count mismatch");
        for (const auto & output : flashsr_batch.outputs) {
            require_wav_shape(output, 48000);
        }

        std::cout << "audio_utility_api_test passed\n";
    } catch (const std::exception & ex) {
        std::cerr << "audio_utility_api_test failed: " << ex.what() << "\n";
        return 1;
    }
    return 0;
}