CFD求解器中C++配置文件变量读取的代码优化咨询

针对你当前用大量else if处理输入变量的问题，最直接的优化方式是用哈希表（std::unordered_map）关联变量名与对应的赋值操作，这样新增变量时只需要往映射表中添加一行配置，无需修改判断逻辑，大幅降低维护成本。

具体实现步骤 #

1. 保留原有的求解器参数变量定义（比如double CFL, xbegin, xend, dt, gamma; int nx;）
2. 用std::unordered_map存储每个变量名对应的赋值lambda表达式，建立变量名到操作的映射
3. 简化读取循环：用while(input >> var >> val)替代原有的while(true)+eof判断，输入失败（包括文件结束）时自动退出循环
4. 增加未知变量的处理逻辑，输出警告方便调试

优化后的代码示例 #

// 预先定义求解器参数变量
double CFL, xbegin, xend, dt, gamma;
int nx;

// If input file is properly provided, try to read values from it
else
{
    std::ifstream input(input_file); // Read input file

    // Check if file is opened, return error otherwise
    if(input.is_open())
    {
        // 构建变量名到赋值操作的映射表
        std::unordered_map<std::string, std::function<void(const std::string&)>> var_map = {
            {"CFL:", [&](const std::string& val) { CFL = std::stod(val); }},
            {"nx:", [&](const std::string& val) { nx = std::stoi(val); }},
            {"xbegin:", [&](const std::string& val) { xbegin = std::stod(val); }},
            {"xend:", [&](const std::string& val) { xend = std::stod(val); }},
            {"dt:", [&](const std::string& val) { dt = std::stod(val); }},
            {"gamma:", [&](const std::string& val) { gamma = std::stod(val); }} // 修正原代码笔误：之前误将gamma的值赋给了dt
        };

        std::string var;
        std::string val;
        // 简化读取循环：输入成功时继续，失败（含文件尾）时退出
        while(input >> var >> val)
        {
            // 查找映射表中的变量名
            auto it = var_map.find(var);
            if(it != var_map.end())
            {
                // 执行对应的赋值操作
                it->second(val);
            }
            else
            {
                // 处理未知变量，输出警告
                std::cerr << "Warning: Unknown variable '" << var << "' in input file" << std::endl;
            }
        }

        input.close();
    }
    else
    {
        std::cerr << "Error: Failed to open input file '" << input_file << "'" << std::endl;
        // 可添加错误处理逻辑，比如终止程序
    }
}

额外优化建议 #

• 可以把映射表的构建逻辑抽成单独函数（比如init_var_map()），让主读取逻辑更清晰
• 如果需要支持数组、枚举等复杂参数类型，可扩展lambda表达式的处理逻辑
• 原代码中gamma:的赋值存在笔误，上述示例已修正

内容的提问来源于stack exchange，提问作者ccruz1000