使用istream_iterator时如何忽略最后一个空行

时间:2014-04-16 22:27:48

标签: c++ file-io istream-iterator

我们说我有一个像这样的2D点的简单文本文件:

10
0.000   0.010
0.000   0.260
0.000   0.510
0.000   0.760
0.000   1.010
0.000   1.260
0.000   1.510
0.000   1.760
0.000   2.010
0.000   2.260
// Blank line here

我为IO使用了一个简单的结构:

template <typename T>
struct point
{
    point (T x = 0, T y = 0) : x (x), y (y) {}

    T x ;
    T y ;

    friend std::istream& operator>> (std::istream &is, point &p) {
        is >> p.x >> p.y ;
        return is ;
    }
};

我原来的代码是:

int main (void)
{
    std::string strFile = "Points.txt" ;

    std::ifstream file ;
    file.exceptions (std::ios::failbit | std::ios::badbit) ;
    std::vector <point <double> > vec ;

    try {
        file.open (strFile) ;

        int nPoints = 0 ;
        file >> nPoints ;

        for (int n = 0; n < nPoints; ++n) {
            point <double> p ;
            file >> p ;
            vec.push_back (p) ;
        }
    }

    catch (std::ios_base::failure &e) {
        std::cerr << e.what () << "\n" ;
        return 1 ;
    }

    return 0 ;
}

这很好用,但本着no raw loops的精神,我想摆脱for循环。

这是我的新代码:

int main (void)
{
    std::string strFile = "Points.txt" ;

    std::ifstream file ;
    file.exceptions (std::ios::failbit | std::ios::badbit) ;
    std::vector <point <double> > vec ;

    try {
        file.open (strFile) ;

        int nPoints = 0 ;
        file >> nPoints ;

        std::copy (
            std::istream_iterator <point <double> > (file), 
            std::istream_iterator <point <double> > (), 
            std::back_inserter (vec)
        ) ;
    }

    catch (std::ios_base::failure &e) {
        std::cerr << e.what () << "\n" ;
        return 1 ;
    }

    return 0 ;
}

所有内容都被正确复制,但不幸的是,读取最后一个空行会导致设置失败位。

我能想到解决这个问题的唯一方法就是难看。我可以:

  • point::operator>>()函数
  • 中放置try-catch子句
  • 检查已存在的catch子句中的vec.size()

是否有一种忽略最后空白行的优雅方式?

1 个答案:

答案 0 :(得分:2)

更改

    std::copy (
        std::istream_iterator <point <double> > (file), 
        std::istream_iterator <point <double> > (), 
        std::back_inserter (vec)
    ) ;

    std::copy_n (
        std::istream_iterator <point <double> > (file), 
        nPoints, 
        std::back_inserter (vec)
    ) ;