二进制读取器在读取确切的字节数时不会触发eof位

时间:2014-09-16 14:19:59

标签: c++ ifstream eof ofstream

我正在使用以下代码将图像写入二进制文件:

std::ofstream edgefile("C:\\****\\edge.bin", std::ofstream::binary | std::ofstream::app | std::ofstream::out);
Mat edges;
    Canny(bilat, edges, cthr1, cthr2, 3); //cany sliders

    if (writeedge){
        int rows = edges.rows;
        int cols = edges.cols;
        edgefile.write(reinterpret_cast<const char *>(&rows), sizeof(int));
        edgefile.write(reinterpret_cast<const char *>(&cols), sizeof(int));
        edgefile.write(reinterpret_cast<char*>(edges.data), edges.rows*edges.cols*sizeof(uchar));
        cout << "writen r:" << rows << "C: " << cols << "Bytes: " << edges.rows*edges.cols*sizeof(uchar) << endl;
    }

然后用这个读取相同的图像:

std::ifstream infile;
int main(int argc, char* argv[])
{
    int * ptr;
    ptr = new int;
    int rows;
    int cols;
    infile.open("C:\\****\\edge.bin", std::ofstream::binary | std::ofstream::app | std::ofstream::in);
    while (!infile.eof())
    {
            infile.read(reinterpret_cast<char*>(ptr), sizeof(int));
            rows = *ptr;
            infile.read(reinterpret_cast<char*>(ptr), sizeof(int));
            cols = *ptr;
            Mat ed(rows, cols, CV_8UC1, Scalar::all(0));
            infile.read(reinterpret_cast<char*>(ed.data), rows * cols * (sizeof uchar));
            cout << "writen r: " << rows << " C: " << cols << " Bytes: " << rows * cols * (sizeof uchar) << endl;
            imshow("God Knows", ed);
            cvWaitKey();        
    }
    infile.close();
    return 0;
}

图像被准确读取,但是最后没有触发eof位,因此将最后的ptr值相乘并在末尾读取另一个空白图像。在此之后,循环结束。如何在不重置当前读取位置的情况下检查下一位是否为EOF位?

(我知道如果再读取1个字节,它将触发EOF位)

1 个答案:

答案 0 :(得分:1)

在尝试读取文件末尾后,位设置为,这就是流的工作方式。

您可以轻松地重构主循环以在第一次读取后检查状态。这是有效的,因为read的返回值是对流的引用,并且将引用转换为bool来检查流是否仍然处于良好状态(即没有EOF)。

while (infile.read(reinterpret_cast<char*>(ptr), sizeof(int)))
{
    // ...