mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
4c123e20d0
CIOStream: * Use flag based system instead (common flags are aliased; uses std::ios_base::openmode). * 'ReadString' now writes into a string buffer (reference) that needs to be passed in by programmer (avoids copy). * 'm_nSize' is now synced with any write operations. CSigCache: * Fixed bug causing crash when 'DecompressBlob' failed (results were never checked). * Light formatting improvements.
102 lines
2.6 KiB
C++
102 lines
2.6 KiB
C++
#pragma once
|
|
|
|
class CIOStream
|
|
{
|
|
public:
|
|
enum Mode_t
|
|
{
|
|
NONE = 0,
|
|
READ = std::ios::in,
|
|
WRITE = std::ios::out,
|
|
BINARY = std::ios::binary,
|
|
};
|
|
|
|
CIOStream();
|
|
CIOStream(const fs::path& fsFileFullPath, int nFlags);
|
|
~CIOStream();
|
|
|
|
bool Open(const fs::path& fsFileFullPath, int nFlags);
|
|
void Close();
|
|
void Flush();
|
|
|
|
void ComputeFileSize();
|
|
|
|
std::streampos GetPosition(Mode_t mode);
|
|
void SetPosition(std::streampos nOffset, Mode_t mode);
|
|
|
|
const std::filebuf* GetData() const;
|
|
const std::streampos GetSize() const;
|
|
|
|
bool IsReadable();
|
|
bool IsWritable() const;
|
|
|
|
bool IsEof() const;
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: reads any value from the file
|
|
//-----------------------------------------------------------------------------
|
|
template<typename T>
|
|
void Read(T& tValue)
|
|
{
|
|
if (IsReadable())
|
|
m_Stream.read(reinterpret_cast<char*>(&tValue), sizeof(tValue));
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: reads any value from the file with specified size
|
|
//-----------------------------------------------------------------------------
|
|
template<typename T>
|
|
void Read(T& tValue, size_t nSize)
|
|
{
|
|
if (IsReadable())
|
|
m_Stream.read(reinterpret_cast<char*>(&tValue), nSize);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: reads any value from the file and returns it
|
|
//-----------------------------------------------------------------------------
|
|
template<typename T>
|
|
T Read()
|
|
{
|
|
T value{};
|
|
if (!IsReadable())
|
|
return value;
|
|
|
|
m_Stream.read(reinterpret_cast<char*>(&value), sizeof(value));
|
|
return value;
|
|
}
|
|
bool ReadString(string& svOut);
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: writes any value to the file
|
|
//-----------------------------------------------------------------------------
|
|
template<typename T>
|
|
void Write(T tValue)
|
|
{
|
|
if (!IsWritable())
|
|
return;
|
|
|
|
m_Stream.write(reinterpret_cast<const char*>(&tValue), sizeof(tValue));
|
|
m_nSize += sizeof(tValue);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: writes any value to the file with specified size
|
|
//-----------------------------------------------------------------------------
|
|
template<typename T>
|
|
void Write(T tValue, size_t nSize)
|
|
{
|
|
if (!IsWritable())
|
|
return;
|
|
|
|
m_Stream.write(reinterpret_cast<const char*>(tValue), nSize);
|
|
m_nSize += nSize;
|
|
}
|
|
bool WriteString(const string& svInput);
|
|
|
|
private:
|
|
std::streampos m_nSize; // File size.
|
|
int m_nFlags; // Stream flags.
|
|
fstream m_Stream; // I/O stream.
|
|
};
|