123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195 |
- #include <iostream>
- #include <vector>
- #include <random>
- #include <nlohmann/json.hpp>
- #include "signalstats_wrapper.h"
- #include "ThreadPool.h"
- void test1();
- void test2();
- int main() {
- // 初始化Python解释器
- // signalstats_wrapper::initialize();
-
- // try {
- // // 创建测试音频信号 (C++版本)
- // const int sample_rate = 44100;
- // const int duration_seconds = 1;
- // const int num_samples = sample_rate * duration_seconds;
-
- // std::vector<double> audio_signal(num_samples);
- // std::random_device rd;
- // std::mt19937 gen(rd());
- // std::normal_distribution<> dist(0.0, 1.0);
-
- // for (auto& sample : audio_signal) {
- // sample = dist(gen);
- // }
-
- // std::cout << "生成测试音频信号完成" << std::endl;
-
- // // 调用C++接口函数
- // nlohmann::json output;
- // std::vector<std::string> window_params = {"tukey", "0.25"};
-
- // nlohmann::json& result = signalstats_wrapper::detect_signal_wrapper(
- // output,
- // audio_signal,
- // sample_rate,
- // 3e-3, // silence_threshold
- // -70.0, // db_threshold
- // -70.0, // cv_threshold
- // window_params,
- // 256, // nperseg
- // 32, // noverlap
- // 256, // nfft
- // true // debug
- // );
-
- // // 处理输出结果
- // if (!output.empty()) {
- // std::cout << "信号检测完成,输出结果: " << output.dump(4) << std::endl;
- // // 这里可以添加具体的结果解析逻辑
- // } else {
- // std::cout << "信号检测完成,但无输出结果" << std::endl;
- // }
-
- // std::cout << "程序执行完成" << std::endl;
-
- // } catch (const std::exception& e) {
- // std::cerr << "错误: " << e.what() << std::endl;
- // return 1;
- // }
- // test1();
- // 初始化Python解释器
- // signalstats_wrapper::initialize();
- CPPTP.add_task(test1);
- // std::this_thread::sleep_for(std::chrono::seconds(1));
- CPPTP.add_task(test2);
- // std::this_thread::sleep_for(std::chrono::seconds(5));
-
-
- std::this_thread::sleep_for(std::chrono::seconds(10)); // 等待任务执行
- // signalstats_wrapper::finalize();
-
- return 0;
- }
- void test1()
- {
- signalstats_wrapper::initialize();
- try {
- // 创建测试音频信号 (C++版本)
- const int sample_rate = 44100;
- const int duration_seconds = 1;
- const int num_samples = sample_rate * duration_seconds;
-
- std::vector<double> audio_signal(num_samples);
- std::random_device rd;
- std::mt19937 gen(rd());
- std::normal_distribution<> dist(0.0, 1.0);
-
- for (auto& sample : audio_signal) {
- sample = dist(gen);
- }
-
- std::cout << "生成测试音频信号完成" << std::endl;
-
- // 调用C++接口函数
- nlohmann::json output;
- std::vector<std::string> window_params = {"tukey", "0.25"};
-
- nlohmann::json& result = signalstats_wrapper::detect_signal_wrapper(
- output,
- audio_signal,
- sample_rate,
- 3e-3, // silence_threshold
- -70.0, // db_threshold
- -70.0, // cv_threshold
- window_params,
- 256, // nperseg
- 32, // noverlap
- 256, // nfft
- true // debug
- );
-
- // 处理输出结果
- if (!output.empty()) {
- std::cout << "信号检测完成,输出结果: " << output.dump(4) << std::endl;
- // 这里可以添加具体的结果解析逻辑
- } else {
- std::cout << "信号检测完成,但无输出结果" << std::endl;
- }
-
- std::cout << "程序执行完成" << std::endl;
-
- } catch (const std::exception& e) {
- std::cerr << "错误: " << e.what() << std::endl;
- return;
- }
- signalstats_wrapper::finalize();
- }
- void test2()
- {
- signalstats_wrapper::initialize();
- try {
- // 创建测试音频信号 (C++版本)
- const int sample_rate = 44100;
- const int duration_seconds = 1;
- const int num_samples = sample_rate * duration_seconds;
-
- std::vector<double> audio_signal(num_samples);
- std::random_device rd;
- std::mt19937 gen(rd());
- std::normal_distribution<> dist(0.0, 1.0);
-
- for (auto& sample : audio_signal) {
- sample = dist(gen);
- }
-
- std::cout << "生成测试音频信号完成" << std::endl;
-
- // 调用C++接口函数
- nlohmann::json output;
- std::vector<std::string> window_params = {"tukey", "0.25"};
-
- nlohmann::json& result = signalstats_wrapper::detect_signal_wrapper(
- output,
- audio_signal,
- sample_rate,
- 3e-3, // silence_threshold
- -70.0, // db_threshold
- -70.0, // cv_threshold
- window_params,
- 256, // nperseg
- 32, // noverlap
- 256, // nfft
- true // debug
- );
-
- // 处理输出结果
- if (!output.empty()) {
- std::cout << "信号检测完成,输出结果: " << output.dump(4) << std::endl;
- // 这里可以添加具体的结果解析逻辑
- } else {
- std::cout << "信号检测完成,但无输出结果" << std::endl;
- }
-
- std::cout << "程序执行完成" << std::endl;
-
- } catch (const std::exception& e) {
- std::cerr << "错误: " << e.what() << std::endl;
- return;
- }
- signalstats_wrapper::finalize();
- }
|