Question

我有当前代码将文本文件读入内存：

std::streampos fsize = 0;
std::ifstream file(fileName, std::ios::binary); // open file

if(!file.good()) {
    std::cout << "Error opening file";
    return 0;
}

// get length of file
file.seekg(0, ios::end);
fsize = file.tellg();

// allocate memory
char *memory = new char[fsize];

// read data as a block
file.seekg (0, ios::beg);
file.read (memory, fsize);

file.close(); // close file

return fsize;

现在我有代码迭代它。如果该行以'v'开头，它将读取前面3个浮点值，如果它以'n'开头，则相同，但是读入不同的数组。

char* p = memory;       // pointer to start of memory
char* e = memory + fsize;   // pointer to end of memory

while (p != e) {
    if (memcmp(p, "v", 1) == 0) { 
        sscanf(p, "v %f %f %f", &a[vI], &b[vI], &c[vI]);
        vI++;
    } else if (memcmp(p, "n",  1) == 0) {
        sscanf(p, "v %f %f %f", &d[nI], &e[nI], &f[nI]);
        nI++;           
    while (*p++ != (char) 0x0A);
}

我知道必须有更好/更安全的方法来做到这一点。

Answer 1

我假设你有一个文本文件。这可以做得更简单。首先，不要以二进制模式打开文件，只需按行读取。以下是可能的实现：

template<class output_iterator>
void read_file(std::istream &input, output_iterator v1, output_iterator v2,
               output_iterator v1) {
    std::string line_buffer;

    while(std::getline(input, line_buffer)) { // read each line of text
        if(line_buffer[0] == 'v') {
            std::stringstream line_stream(line_buffer.substr(1)); // drop the 'v'
            // read three consecutive floats
            line_stream >> *v1++ >> *v2++ >> *v3++; 
        }
    }
}

此代码假定以'v'开头的行格式正确。您可以像这样使用它：

std::vector<float> values1, values2, values3;
std::fstream input_file(fileName);

read_file(input_file, std::back_inserter(values1), std::back_inserter(values2),
          std::back_inserter(values3));

Answer 2

如果您的系统支持，请使用mmap（即任何类似* nix的系统）。这将（很快）为您提供文件内容的char*。如果你的文件很大，这很好用，因为它使用虚拟内存系统为你缓存一切 - 即你不必等待所有数据被复制。 mmap函数立即返回，虚拟内存的相关部分已映射到您的文件。

int fd = open(binaryRAWFileName, O_RDONLY);
... should do some error check to ensure fd != -1

// get the size of the file
struct stat sb;
if (fstat(fd, &sb) == -1) {
        ... there was an error with fstat
}

char * memory = mmap(NULL  // we don't care where the memory is
        , sb.st_size      // length of the file
        , PROT_READ   
        , MAP_PRIVATE
        , fd            // the file descriptor of course
        , 0);

最后，memcmp似乎有点毫无意义，长度为1.为什么不只是if(*p=='v')而不是if(memcmp(p, "v", 1) == 0)

通过内存迭代读取值

2 个答案: