我想在C++中使用ifstream和ofstream来模仿C#的BinaryReader / BinaryWriter功能

I'd like to use ifstream and ofstream in C++ to mimic C#'s BinaryReader / BinaryWriter functionality

本文关键字:BinaryReader 功能 BinaryWriter ofstream C++ ifstream      更新时间:2023-10-16

我正在寻找一种方法来写浮点数/int/字符串到一个文件,并读取它们作为浮点数/int/字符串。(基本读写方式为ios::binary)

最后是我自己写的。只是想和大家分享一下。

它可能没有被优化,但我有一些困难找到c++代码模仿c#的BinaryReader & BinaryWriter 类。所以我创建了一个类来处理读和写。

快速注意事项:

1) "BM"只是我的类的前缀。

2) BMLogging是一个助手类,它简单地做:

cout << "bla bla bla" << endl;

所以你可以忽略对BMLogging的调用,我保留它们是为了突出显示我们可以警告用户的情况。

代码如下:

#include <iostream>
#include <fstream>
using namespace std;
// Create the macro so we don't repeat the code over and over again.
#define BMBINARY_READ(reader,value) reader.read((char *)&value, sizeof(value))
enum BMBinaryIOMode
{
    None = 0,
    Read,
    Write
};
class BMBinaryIO
{
    // the output file stream to write onto a file
    ofstream writer;
    // the input file stream to read from a file
    ifstream reader;
    // the filepath of the file we're working with
    string filePath;
    // the current active mode.
    BMBinaryIOMode currentMode;
public:
    BMBinaryIO()
    {
        currentMode = BMBinaryIOMode::None;
    }
    // the destructor will be responsible for checking if we forgot to close
    // the file
    ~BMBinaryIO()
    {
        if(writer.is_open())
        {
            BMLogging::error(BMLoggingClass::BinaryIO, "You forgot to call close() after finishing with the file! Closing it...");
            writer.close();
        }
        if(reader.is_open())
        {
            BMLogging::error(BMLoggingClass::BinaryIO, "You forgot to call close() after finishing with the file! Closing it...");
            reader.close();
        }   
    }
    // opens a file with either read or write mode. Returns whether
    // the open operation was successful
    bool open(string fileFullPath, BMBinaryIOMode mode)
    {
        filePath = fileFullPath;
        BMLogging::info(BMLoggingClass::BinaryIO, "Opening file: " + filePath);
        // Write mode
        if(mode == BMBinaryIOMode::Write)
        {
            currentMode = mode;
            // check if we had a previously opened file to close it
            if(writer.is_open())
                writer.close();
            writer.open(filePath, ios::binary);
            if(!writer.is_open())
            {
                BMLogging::error(BMLoggingClass::BinaryIO, "Could not open file for write: " + filePath);
                currentMode = BMBinaryIOMode::None;
            }
        }
        // Read mode
        else if(mode == BMBinaryIOMode::Read)
        {
            currentMode = mode;
            // check if we had a previously opened file to close it
            if(reader.is_open())
                reader.close();
            reader.open(filePath, ios::binary);
            if(!reader.is_open())
            {
                BMLogging::error(BMLoggingClass::BinaryIO, "Could not open file for read: " + filePath);
                currentMode = BMBinaryIOMode::None;
            }
        }
        // if the mode is still the NONE/initial one -> we failed
        return currentMode == BMBinaryIOMode::None ? false : true;
    }
    // closes the file
    void close()
    {
        if(currentMode == BMBinaryIOMode::Write)
        {
            writer.close();
        }
        else if(currentMode == BMBinaryIOMode::Read)
        {
            reader.close();
        }
    }
    bool checkWritabilityStatus()
    {
        if(currentMode != BMBinaryIOMode::Write)
        {
            BMLogging::error(BMLoggingClass::BinaryIO, "Trying to write with a non Writable mode!");
            return false;
        }
        return true;
    }
    // Generic write method that will write any value to a file (except a string,
    // for strings use writeString instead).
    void write(void *value, size_t size)
    {
        if(!checkWritabilityStatus())
            return;
        // write the value to the file.
        writer.write((const char *)value, size);
    }
    // Writes a string to the file
    void writeString(string str)
    {
        if(!checkWritabilityStatus())
            return;
        // first add a  at the end of the string so we can detect
        // the end of string when reading it
        str += '';
        // create char pointer from string.
        char* text = (char *)(str.c_str());
        // find the length of the string.
        unsigned long size = str.size();
        // write the whole string including the null.
        writer.write((const char *)text, size);
    }
    // helper to check if we're allowed to read
    bool checkReadabilityStatus()
    {
        if(currentMode != BMBinaryIOMode::Read)
        {
            BMLogging::error(BMLoggingClass::BinaryIO, "Trying to read with a non Readable mode!");
            return false;
        }
        // check if we hit the end of the file.
        if(reader.eof())
        {
            BMLogging::error(BMLoggingClass::BinaryIO, "Trying to read but reached the end of file!");
            reader.close();
            currentMode = BMBinaryIOMode::None;
            return false;
        }
        return true;
    }
    // reads a boolean value
    bool readBoolean()
    {
        if(checkReadabilityStatus())
        {
            bool value = false;
            BMBINARY_READ(reader, value);
            return value;
        }
        return false;
    }
    // reads a character value
    char readChar()
    {
        if(checkReadabilityStatus())
        {
            char value = 0;
            BMBINARY_READ(reader, value);
            return value;
        }
        return 0;
    }
    // read an integer value
    int readInt()
    {
        if(checkReadabilityStatus())
        {
            int value = 0;
            BMBINARY_READ(reader, value);
            return value;
        }
        return 0;
    }
    // read a float value
    float readFloat()
    {
        if(checkReadabilityStatus())
        {
            float value = 0;
            BMBINARY_READ(reader, value);
            return value;
        }
        return 0;
    }   
    // read a double value
    double readDouble()
    {
        if(checkReadabilityStatus())
        {
            double value = 0;
            BMBINARY_READ(reader, value);
            return value;
        }
        return 0;
    }
    // read a string value
    string readString()
    {
        if(checkReadabilityStatus())
        {
            char c;
            string result = "";
            while((c = readChar()) != '')
            {
                result += c;
            }
            return result;
        }
        return "";
    }
};

编辑:我用这些替换了上面所有的读/写方法:(也更新了使用代码)

// Generic write method that will write any value to a file (except a string,
// for strings use writeString instead)
template<typename T>
void write(T &value)
{
    if(!checkWritabilityStatus())
        return;
    // write the value to the file.
    writer.write((const char *)&value, sizeof(value));
}
// Writes a string to the file
void writeString(string str)
{
    if(!checkWritabilityStatus())
        return;
    // first add a  at the end of the string so we can detect
    // the end of string when reading it
    str += '';
    // create char pointer from string.
    char* text = (char *)(str.c_str());
    // find the length of the string.
    unsigned long size = str.size();
    // write the whole string including the null.
    writer.write((const char *)text, size);
}
// reads any type of value except strings.
template<typename T>
T read()
{
    checkReadabilityStatus();
    T value;
    reader.read((char *)&value, sizeof(value));
    return value;
}
// reads any type of value except strings.
template<typename T>
void read(T &value)
{
    if(checkReadabilityStatus())
    {
        reader.read((char *)&value, sizeof(value));
    }
}
// read a string value
string readString()
{
    if(checkReadabilityStatus())
    {
        char c;
        string result = "";
        while((c = read<char>()) != '')
        {
            result += c;
        }
        return result;
    }
    return "";
}
// read a string value
void readString(string &result)
{
    if(checkReadabilityStatus())
    {
        char c;
        result = "";
        while((c = read<char>()) != '')
        {
            result += c;
        }
    }
}

你可以这样写:

string myPath = "somepath to the file";
BMBinaryIO binaryIO;
if(binaryIO.open(myPath, BMBinaryIOMode::Write))
{
    float value = 165;
    binaryIO.write(value);
    char valueC = 'K';
    binaryIO.write(valueC);
    double valueD = 1231.99;
    binaryIO.write(valueD);
    string valueStr = "spawnAt(100,200)";
    binaryIO.writeString(valueStr);
    valueStr = "helpAt(32,3)";
    binaryIO.writeString(valueStr);
    binaryIO.close();
}

下面是你如何使用它来读取:

string myPath = "some path to the same file";
if(binaryIO.open(myPath, BMBinaryIOMode::Read))
{
    cout << binaryIO.read<float>() << endl;
    cout << binaryIO.read<char>() << endl;
    double valueD = 0;
    binaryIO.read(valueD); // or you could use read<double()
    cout << valueD << endl;
    cout << binaryIO.readString() << endl;
    cout << binaryIO.readString() << endl;
    binaryIO.close();
}

编辑2:您甚至可以在一行中写入/读取整个结构:

struct Vertex {
    float x, y;
};
Vertex vtx; vtx.x = 2.5f; vtx.y = 10.0f;
// to write it
binaryIO.write(vtx);
// to read it
Vertex vtxRead;
binaryIO.read(vtxRead); // option 1
vtxRead = binaryIO.read<Vertex>(); // option 2

希望我的代码足够清晰

我将ifstreamofstream分为ibfstreamobfstream。我做了一个小的辅助类来检测我正在编译/运行的机器的端序。然后,我为ibfstreamobfstream添加了一个标志,指示原始类型中的字节是否应该翻转。这些类还具有读/写基本类型和这些类型的数组的方法,可以根据需要翻转字节顺序。最后,我为这些类默认设置了ios::binary

我经常在小端机器上工作,想要写大端文件,反之亦然。这是在一个程序中使用的,该程序对各种格式的3D图形文件做了大量的I/O。

我将ifstreamofstream子类化为ibfstreamobfstream。我做了一个类来检测我正在编译/运行的机器的端序。然后,我为ibfstreamobfstream添加了一个标志,指示原始类型中的字节是否应该翻转。这些类还具有读写基本类型和数组的方法,可以根据需要翻转字节顺序。

我经常在小端机器上工作,想要写大端文件,反之亦然。这是在一个程序中使用的,该程序对各种格式的3D图形文件做了大量的I/O。