153 lines
3.1 KiB
C++
153 lines
3.1 KiB
C++
|
|
#include "demofile.h"
|
|
#include "demotypes.h"
|
|
#include <cassert>
|
|
#include <utility>
|
|
|
|
// DemoFileReader
|
|
|
|
DemoFileReader::DemoFileReader(FILE* fp):
|
|
m_demoFp(fp)
|
|
{
|
|
}
|
|
|
|
bool DemoFileReader::IsOk() const
|
|
{
|
|
return ferror(m_demoFp) == 0;
|
|
}
|
|
|
|
void DemoFileReader::ReadDemoHeader(demoheader_t& header)
|
|
{
|
|
fread(&header, sizeof(demoheader_t), 1, m_demoFp);
|
|
}
|
|
|
|
int32_t DemoFileReader::ReadRawData(uint8_t* buffer, int32_t maxLength)
|
|
{
|
|
FILE* fp = m_demoFp;
|
|
|
|
int32_t size;
|
|
fread(&size, sizeof(int32_t), 1, fp);
|
|
|
|
if (buffer && (maxLength < size))
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
if (buffer)
|
|
{
|
|
fread(buffer, 1, size, fp);
|
|
}
|
|
else
|
|
{
|
|
fseek(fp, size, SEEK_CUR);
|
|
}
|
|
return size;
|
|
}
|
|
|
|
Array<uint8_t> DemoFileReader::ReadRawData(int32_t maxLength)
|
|
{
|
|
FILE* fp = m_demoFp;
|
|
|
|
int32_t size;
|
|
fread(&size, sizeof(int32_t), 1, fp);
|
|
|
|
Array<uint8_t> data;
|
|
if (maxLength < size)
|
|
{
|
|
return std::move(data);
|
|
}
|
|
|
|
data.reset(size);
|
|
fread(data.begin(), 1, size, fp);
|
|
return std::move(data);
|
|
}
|
|
|
|
void DemoFileReader::ReadSequenceInfo(int32_t& seqNum1, int32_t& seqNum2)
|
|
{
|
|
FILE* fp = m_demoFp;
|
|
fread(&seqNum1, sizeof(int32_t), 1, fp);
|
|
fread(&seqNum2, sizeof(int32_t), 1, fp);
|
|
}
|
|
|
|
void DemoFileReader::ReadCmdInfo(democmdinfo_t& info)
|
|
{
|
|
fread(&info, sizeof(democmdinfo_t), 1, m_demoFp);
|
|
}
|
|
|
|
void DemoFileReader::ReadCmdHeader(unsigned char& cmd, int32_t& tick)
|
|
{
|
|
FILE* fp = m_demoFp;
|
|
fread(&cmd, 1, sizeof(unsigned char), fp);
|
|
fread(&tick, 1, sizeof(int32_t), fp);
|
|
if (cmd > dem_lastcmd)
|
|
{
|
|
cmd = dem_stop;
|
|
}
|
|
}
|
|
|
|
int32_t DemoFileReader::ReadUserCmd(int32_t& cmdNum, uint8_t* buffer, int32_t maxLength)
|
|
{
|
|
fread(&cmdNum, sizeof(int32_t), 1, m_demoFp);
|
|
return ReadRawData(buffer, maxLength);
|
|
}
|
|
|
|
Array<uint8_t> DemoFileReader::ReadUserCmd(int32_t& cmdNum, int32_t maxLength)
|
|
{
|
|
fread(&cmdNum, sizeof(int32_t), 1, m_demoFp);
|
|
return ReadRawData(maxLength);
|
|
}
|
|
|
|
// DemoFileWriter
|
|
|
|
DemoFileWriter::DemoFileWriter(FILE* fp) :
|
|
m_demoFp(fp)
|
|
{
|
|
}
|
|
|
|
FILE* DemoFileWriter::GetFp() const
|
|
{
|
|
return m_demoFp;
|
|
}
|
|
|
|
bool DemoFileWriter::IsOk() const
|
|
{
|
|
return ferror(m_demoFp) == 0;
|
|
}
|
|
|
|
void DemoFileWriter::WriteDemoHeader(const demoheader_t& header)
|
|
{
|
|
fwrite(&header, sizeof(demoheader_t), 1, m_demoFp);
|
|
}
|
|
|
|
void DemoFileWriter::WriteRawData(const uint8_t* buffer, int32_t length)
|
|
{
|
|
FILE* fp = m_demoFp;
|
|
fwrite(&length, sizeof(int32_t), 1, fp);
|
|
fwrite(buffer, length, 1, fp);
|
|
}
|
|
|
|
void DemoFileWriter::WriteSequenceInfo(int32_t seqNum1, int32_t seqNum2)
|
|
{
|
|
FILE* fp = m_demoFp;
|
|
fwrite(&seqNum1, sizeof(int32_t), 1, fp);
|
|
fwrite(&seqNum2, sizeof(int32_t), 1, fp);
|
|
}
|
|
|
|
void DemoFileWriter::WriteCmdInfo(const democmdinfo_t& info)
|
|
{
|
|
fwrite(&info, sizeof(democmdinfo_t), 1, m_demoFp);
|
|
}
|
|
|
|
void DemoFileWriter::WriteCmdHeader(unsigned char cmd, int32_t tick)
|
|
{
|
|
FILE* fp = m_demoFp;
|
|
fwrite(&cmd, sizeof(unsigned char), 1, fp);
|
|
fwrite(&tick, sizeof(int32_t), 1, fp);
|
|
}
|
|
|
|
void DemoFileWriter::WriteUserCmd(int32_t cmdNum, const uint8_t* buffer, int32_t length)
|
|
{
|
|
fwrite(&cmdNum, sizeof(int32_t), 1, m_demoFp);
|
|
WriteRawData(buffer, length);
|
|
}
|