1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#ifndef DEMO_RECORDER
#define DEMO_RECORDER
#include <vector>
#include <sstream>
#include <zlib.h>
#include "Demo.h"
#include "Game/Players/PlayerStatistics.h"
#include "Sim/Misc/TeamStatistics.h"
/**
* @brief Used to record demos
*/
class CDemoRecorder : public CDemo
{
public:
CDemoRecorder() { memset(&fileHeader, 0, sizeof(fileHeader)); }
CDemoRecorder(const std::string& mapName, const std::string& modName, bool serverDemo);
CDemoRecorder(const CDemoRecorder&) = delete;
CDemoRecorder(CDemoRecorder&& r) { *this = std::move(r); }
~CDemoRecorder();
CDemoRecorder& operator = (const CDemoRecorder&) = delete;
CDemoRecorder& operator = (CDemoRecorder&& r) {
memcpy(&fileHeader, &r.fileHeader, sizeof(fileHeader));
memset(&r.fileHeader, 0, sizeof(fileHeader));
std::swap(file, r.file);
std::swap(demoName, r.demoName);
std::swap(playerStats, r.playerStats);
std::swap(teamStats, r.teamStats);
std::swap(winningAllyTeams, r.winningAllyTeams);
std::swap(isServerDemo, r.isServerDemo);
return *this;
}
bool IsValid() const { return (file != nullptr); }
void WriteSetupText(const std::string& text);
void SaveToDemo(const unsigned char* buf, const unsigned length, const float modGameTime);
void SetStream();
void SetName(const std::string& mapName, const std::string& modName);
const std::string& GetName() const { return demoName; }
void SetGameID(const unsigned char* buf);
void SetTime(int gameTime, int wallclockTime);
void AddNewPlayer(const std::string& name, int playerNum);
void InitializeStats(int numPlayers, int numTeams);
void SetPlayerStats(int playerNum, const PlayerStatistics& stats);
void SetTeamStats(int teamNum, const std::vector<TeamStatistics>& stats);
void SetWinningAllyTeams(const std::vector<unsigned char>& winningAllyTeams);
private:
unsigned int WriteFileHeader(bool updateStreamLength);
void SetFileHeader();
void WritePlayerStats();
void WriteTeamStats();
void WriteWinnerList();
void WriteDemoFile();
private:
gzFile file = nullptr;
std::vector<PlayerStatistics> playerStats;
std::vector< std::vector<TeamStatistics> > teamStats;
std::vector<unsigned char> winningAllyTeams;
bool isServerDemo = false;
};
#endif
|
