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
|
/* 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 <list>
#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(const std::string& mapName, const std::string& modName, bool serverDemo);
~CDemoRecorder();
void WriteSetupText(const std::string& text);
void SaveToDemo(const unsigned char* buf, const unsigned length, const float modGameTime);
/**
@brief assign a map name for the demo file
*/
void SetName(const std::string& mapName, const std::string& modName, bool serverDemo);
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::list< 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;
std::stringstream demoStream;
std::vector<PlayerStatistics> playerStats;
std::vector< std::vector<TeamStatistics> > teamStats;
std::vector<unsigned char> winningAllyTeams;
};
#endif
|
