wehub-resource-sync
75 lines
2.5 KiB
C++
75 lines
2.5 KiB
C++
#include "mnn_tts_config.hpp"
|
|
|
|
MNNTTSConfig::MNNTTSConfig(const std::string &config_json_path)
|
|
{
|
|
// 检查文件是否存在且是常规文件
|
|
if (!fs::exists(config_json_path) || !fs::is_regular_file(config_json_path))
|
|
{
|
|
throw std::runtime_error("Config file not found or is not a regular file: " + config_json_path);
|
|
}
|
|
|
|
std::ifstream file(config_json_path);
|
|
if (!file.is_open())
|
|
{
|
|
throw std::runtime_error("Failed to open config file: " + config_json_path);
|
|
}
|
|
|
|
try
|
|
{
|
|
file >> raw_config_data_; // 直接从文件流解析JSON
|
|
}
|
|
catch (const nlohmann::json::parse_error &e)
|
|
{
|
|
throw std::runtime_error("Error parsing config.json (" + config_json_path + "): " + e.what());
|
|
}
|
|
catch (const std::exception &e)
|
|
{
|
|
throw std::runtime_error("Error reading config.json (" + config_json_path + "): " + e.what());
|
|
}
|
|
|
|
try
|
|
{
|
|
model_type_ = get_value_from_json<std::string>(raw_config_data_, "model_type");
|
|
model_path_ = get_value_from_json<std::string>(raw_config_data_, "model_path");
|
|
asset_folder_ = get_value_from_json<std::string>(raw_config_data_, "asset_folder");
|
|
cache_folder_ = get_value_from_json<std::string>(raw_config_data_, "cache_folder");
|
|
sample_rate_ = get_value_from_json<int>(raw_config_data_, "sample_rate");
|
|
}
|
|
catch (const std::runtime_error &e)
|
|
{
|
|
// 捕获 get_value_from_json 抛出的异常，并添加更多上下文信息
|
|
throw std::runtime_error("Error in config file " + config_json_path + ": " + e.what());
|
|
}
|
|
}
|
|
|
|
// 新增：支持参数覆盖的构造函数
|
|
MNNTTSConfig::MNNTTSConfig(const std::string &config_file_path,
|
|
const std::map<std::string, std::string> &overrides)
|
|
: MNNTTSConfig(config_file_path) // 委托给原构造函数
|
|
{
|
|
// 应用参数覆盖
|
|
applyOverrides(overrides);
|
|
}
|
|
|
|
// 应用参数覆盖
|
|
void MNNTTSConfig::applyOverrides(const std::map<std::string, std::string> &overrides) {
|
|
if (overrides.empty()) {
|
|
return;
|
|
}
|
|
|
|
for (const auto& [key, value] : overrides) {
|
|
try {
|
|
if (key == "model_type") {
|
|
model_type_ = value;
|
|
} else if (key == "sample_rate") {
|
|
sample_rate_ = std::stoi(value);
|
|
}
|
|
// 可以继续添加其他参数的覆盖逻辑
|
|
} catch (const std::exception &e) {
|
|
// 忽略无法转换的参数，使用配置文件中的默认值
|
|
std::cerr << "Warning: Failed to override parameter '" << key
|
|
<< "' with value '" << value << "': " << e.what() << std::endl;
|
|
}
|
|
}
|
|
}
|