CSSZombieEscape/demboyz
7
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Partial implementation of demo file writer

Browse Source
This commit is contained in:
Jordan Cristiano 2015-05-27 22:58:19 -04:00
parent 05d1518653
commit 2ed3320b84
1 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
demboyz/demowriter.cpp
View File

@ -1,6 +1,12 @@
#include "idemowriter.h" #include "idemowriter.h"
#include "demotypes.h"
#include "demofile.h"
#include "netmessages/nethandlers.h"
#include "netmessages/netcontants.h"
#include "bitbuf.h"
#include <cstdio> #include <cstdio>
#include <memory>
class DemoWriter: public IDemoWriter class DemoWriter: public IDemoWriter
{ {
@ -14,6 +20,11 @@ public:
virtual void EndCommandPacket() override final; virtual void EndCommandPacket() override final;
virtual void WriteNetPacket(NetPacket& packet, SourceGameContext& context) override final; virtual void WriteNetPacket(NetPacket& packet, SourceGameContext& context) override final;
private:
DemoFileWriter m_writer;
bf_write m_cmdPacketBuf;
std::unique_ptr<uint8_t[]> m_packetBuffer;
}; };
IDemoWriter* IDemoWriter::CreateDemoWriter(void* outputFp) IDemoWriter* IDemoWriter::CreateDemoWriter(void* outputFp)
@ -21,12 +32,16 @@ IDemoWriter* IDemoWriter::CreateDemoWriter(void* outputFp)
return new DemoWriter(reinterpret_cast<FILE*>(outputFp)); return new DemoWriter(reinterpret_cast<FILE*>(outputFp));
} }
DemoWriter::DemoWriter(FILE* outputFp) DemoWriter::DemoWriter(FILE* outputFp):
m_writer(outputFp),
m_packetBuffer(new uint8_t[NET_MAX_PAYLOAD])
{ {
m_cmdPacketBuf.StartWriting(m_packetBuffer.get(), NET_MAX_PAYLOAD);
} }
void DemoWriter::StartWriting(demoheader_t& header) void DemoWriter::StartWriting(demoheader_t& header)
{ {
m_writer.WriteDemoHeader(header);
} }
void DemoWriter::EndWriting() void DemoWriter::EndWriting()
@ -35,12 +50,21 @@ void DemoWriter::EndWriting()
void DemoWriter::StartCommandPacket(CommandPacket& packet) void DemoWriter::StartCommandPacket(CommandPacket& packet)
{ {
m_writer.WriteCmdHeader(packet.cmd, packet.tick);
if (packet.cmd == dem_packet || packet.cmd == dem_signon)
{
m_writer.WriteCmdInfo(*packet.cmdInfo);
m_writer.WriteSequenceInfo(packet.sequenceInfo1, packet.sequenceInfo2);
}
m_cmdPacketBuf.Reset();
} }
void DemoWriter::EndCommandPacket() void DemoWriter::EndCommandPacket()
{ {
m_writer.WriteRawData(m_cmdPacketBuf.GetBasePointer(), m_cmdPacketBuf.GetNumBytesWritten());
} }
void DemoWriter::WriteNetPacket(NetPacket& packet, SourceGameContext& context) void DemoWriter::WriteNetPacket(NetPacket& packet, SourceGameContext& context)
{ {
NetHandlers::NetMsg_BitWrite(packet.type, m_cmdPacketBuf, context, packet.data);
} }