有条件地逐行读取文件

Question

现在我有那个代码：

if (minimizator_weighted) {
    while (dataFile >> t >> e >> sigma) { // read data file value by value
        /* some code */
    }
}
else {
    while (dataFile >> t >> e) { // all the same but will not read standard deviation
        /* almost the same code */
    }
}

您可以看到if 和else 流之间的唯一区别是while 循环的条件。我想知道是否有可能优化该 sn-p 并重用代码？如果我能写一些就太好了：

while ((minimizator_weighted) ? (dataFile >> t >> e >> sigma) : (dataFile >> t >> e)) { ... }

但我不确定这个技巧是否正确......你能告诉我一些优化吗？谢谢！

编辑 这是完整的代码sn-p

if (minimizator_weighted) {
    while (dataFile >> t >> e >> sigma) { // read data file value by value
        data_set::pt point;
        point.t = t;
        point.e = e;
        point.c_vis = 0.0;
        point.c_invis = 0.0;

        if (std::abs(sigma) <= GSL_SQRT_DBL_MIN) // check for division-by-zero error
            sigma = 1.0;

        point.sigma = sigma;

        set.curve.push_back(point); // store point

        data_numPoints++; // collect some stats
        set.curveAvg += e;
    }   
}   
else {
    while (dataFile >> t >> e) { // all the same but will not read standard deviation
        data_set::pt point;
        point.t = t;
        point.e = e;
        point.c_vis = 0.0;
        point.c_invis = 0.0;

        set.curve.push_back(point);

        data_numPoints++;
        set.curveAvg += e;
    }   
}

Answer 1

您的架构向我们展示了 3 个错误：

1. sigma 是某些对象的未使用变量
2. 对象行为由外部变量决定 (minimizator_weighted)
3. 每次访问这些对象之一时都必须使用minimizator_weighted

这些是根本性的设计缺陷，不仅会导致您在这方面付出代价，还会让您在整个程序中付出代价。不过，修复将需要重新架构。让我向您展示如何做到这一点，您可以自行决定是继续解决问题还是修复架构。

首先您需要 2 个基本类型，第 1^st 将与您已经定义但没有 sigma 的 data_set::pt 匹配，然后我们将用data_set::pt_weighted 扩展它以添加sigma：

struct pt {
    double t;
    double e;
    double c_vis;
    double c_invis;
};

struct pt_weighted : pt {
    double sigma;
};

现在我们将编写提取运算符，再次从 data_set::pt 的提取运算符开始，并将其扩展为 data_set::pt_weighted：

istream& operator>> (istream& lhs, data_set::pt& rhs) {
    rhs.c_vis = 0.0;
    rhs.c_invis = 0.0;

    return lhs >> rhs.t >> rhs.e;
}

istream& operator>> (istream& lhs, data_set::pt_weighted& rhs) {
    lhs >> static_cast<data_set::pt&>(rhs) >> rhs.sigma;

    // check for division-by-zero error
    if(std::abs(rhs.sigma) <= GSL_SQRT_DBL_MIN) rhs.sigma = 1.0;

    return lhs;
}

您需要从这里开始使用模板。首先，您需要将set.curve 模板化为data_set::pt 或data_set::pt_weighted 的容器，然后您的功能需要更改为：

template <typename T>
void foo() {
    for(T point; dataFile >> point;) {
        set.curve.push_back(point); // store point

        data_numPoints++; // collect some stats
        set.curveAvg += point.e;            
    }
}

如果您无法在运行时建立 minimizator_weighted，则需要调用 foo，例如：

minimizator_weighted ? foo<data_set::pt_weighted>() : foo<data_set::pt>();

Answer 2

我想我会这样表达：

data_set::pt collect_point(std::istream& is, bool minimizator_weighted)
{
    data_set::pt point;
    is >> point.t >> point.e;
    point.c_vis = 0.0;
    point.c_invis = 0.0;
    if (minimizator_weighted) {
        is >> point.sigma;
        point.sigma = std::max(point.sigma, GSL_SQRT_DBL_MIN);
    }
    return point;
}

void test()
{
  int data_numPoints = 0;
  data_set set;

  while (dataFile)
  {
    auto point = collect_point(dataFile, minimizator_weighted);
    auto e = point.e;
    set.curve.push_back(std::move(point)); // store point
    data_numPoints++; // collect some stats
    set.curveAvg += e;
  }
}

Answer 3

添加一个间接级别

bool read_data_line1(istream& dataFile, T& t, E& e, Sig& sigma)
{ return dataFile >> t >> e >> sigma; }

bool read_data_line2(istream& dataFile, T& t, E& e, Sig&)
{ return dataFile >> t >> e; }

auto read_data_line_func = minimizator_weighted ? read_data_line1 : read_data_line2;
while(read_data_line_func(dataFile, t, e, sigma))
{
    data_set::pt point;
    point.t = t;
    point.e = e;
    point.c_vis = 0.0;
    point.c_invis = 0.0;

    if (minimizator_weighted)
    {
      if (std::abs(sigma) <= GSL_SQRT_DBL_MIN) // check for division-by-zero error
        sigma = 1.0;
      point.sigma = sigma;
    }

    set.curve.push_back(point); // store point

    data_numPoints++; // collect some stats
    set.curveAvg += e;
}