355 lines
8.0 KiB
C++
355 lines
8.0 KiB
C++
|
|
// ScoreboardSerializer.cpp
|
|
|
|
|
|
#include "Globals.h"
|
|
#include "ScoreboardSerializer.h"
|
|
#include "OSSupport/GZipFile.h"
|
|
#include "FastNBT.h"
|
|
|
|
#include "../Scoreboard.h"
|
|
|
|
|
|
|
|
|
|
|
|
cScoreboardSerializer::cScoreboardSerializer(const AString & a_WorldName, cScoreboard * a_ScoreBoard):
|
|
m_ScoreBoard(a_ScoreBoard)
|
|
{
|
|
AString DataPath;
|
|
Printf(DataPath, "%s%cdata", a_WorldName.c_str(), cFile::PathSeparator());
|
|
|
|
m_Path = DataPath + cFile::PathSeparator() + "scoreboard.dat";
|
|
|
|
cFile::CreateFolder(DataPath);
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool cScoreboardSerializer::Load(void)
|
|
{
|
|
try
|
|
{
|
|
const auto Data = GZipFile::ReadRestOfFile(m_Path);
|
|
const cParsedNBT NBT(Data.GetView());
|
|
|
|
if (!NBT.IsValid())
|
|
{
|
|
// NBT Parsing failed
|
|
return false;
|
|
}
|
|
|
|
return LoadScoreboardFromNBT(NBT);
|
|
}
|
|
catch (const std::exception & Oops)
|
|
{
|
|
LOGWARNING("Failed to load scoreboard from \"%s\": %s", m_Path.c_str(), Oops.what());
|
|
return false;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool cScoreboardSerializer::Save(void)
|
|
{
|
|
cFastNBTWriter Writer;
|
|
SaveScoreboardToNBT(Writer);
|
|
Writer.Finish();
|
|
|
|
#ifndef NDEBUG
|
|
cParsedNBT TestParse(Writer.GetResult());
|
|
ASSERT(TestParse.IsValid());
|
|
#endif // !NDEBUG
|
|
|
|
GZipFile::Write(m_Path, Writer.GetResult());
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void cScoreboardSerializer::SaveScoreboardToNBT(cFastNBTWriter & a_Writer)
|
|
{
|
|
a_Writer.BeginCompound("data");
|
|
|
|
a_Writer.BeginList("Objectives", TAG_Compound);
|
|
|
|
for (cScoreboard::cObjectiveMap::const_iterator it = m_ScoreBoard->m_Objectives.begin(); it != m_ScoreBoard->m_Objectives.end(); ++it)
|
|
{
|
|
const cObjective & Objective = it->second;
|
|
|
|
a_Writer.BeginCompound("");
|
|
|
|
a_Writer.AddString("CriteriaName", cObjective::TypeToString(Objective.GetType()));
|
|
|
|
a_Writer.AddString("DisplayName", Objective.GetDisplayName());
|
|
a_Writer.AddString("Name", it->first);
|
|
|
|
a_Writer.EndCompound();
|
|
}
|
|
|
|
a_Writer.EndList(); // Objectives
|
|
|
|
a_Writer.BeginList("PlayerScores", TAG_Compound);
|
|
|
|
for (cScoreboard::cObjectiveMap::const_iterator it = m_ScoreBoard->m_Objectives.begin(); it != m_ScoreBoard->m_Objectives.end(); ++it)
|
|
{
|
|
const cObjective & Objective = it->second;
|
|
|
|
for (cObjective::cScoreMap::const_iterator it2 = Objective.m_Scores.begin(); it2 != Objective.m_Scores.end(); ++it2)
|
|
{
|
|
a_Writer.BeginCompound("");
|
|
|
|
a_Writer.AddInt("Score", it2->second);
|
|
|
|
a_Writer.AddString("Name", it2->first);
|
|
a_Writer.AddString("Objective", it->first);
|
|
|
|
a_Writer.EndCompound();
|
|
}
|
|
}
|
|
|
|
a_Writer.EndList(); // PlayerScores
|
|
|
|
a_Writer.BeginList("Teams", TAG_Compound);
|
|
|
|
for (cScoreboard::cTeamMap::const_iterator it = m_ScoreBoard->m_Teams.begin(); it != m_ScoreBoard->m_Teams.end(); ++it)
|
|
{
|
|
const cTeam & Team = it->second;
|
|
|
|
a_Writer.BeginCompound("");
|
|
|
|
a_Writer.AddByte("AllowFriendlyFire", Team.AllowsFriendlyFire() ? 1 : 0);
|
|
a_Writer.AddByte("SeeFriendlyInvisibles", Team.CanSeeFriendlyInvisible() ? 1 : 0);
|
|
|
|
a_Writer.AddString("DisplayName", Team.GetDisplayName());
|
|
a_Writer.AddString("Name", it->first);
|
|
|
|
a_Writer.AddString("Prefix", Team.GetPrefix());
|
|
a_Writer.AddString("Suffix", Team.GetSuffix());
|
|
|
|
a_Writer.BeginList("Players", TAG_String);
|
|
|
|
for (cTeam::cPlayerNameSet::const_iterator it2 = Team.m_Players.begin(); it2 != Team.m_Players.end(); ++it2)
|
|
{
|
|
a_Writer.AddString("", *it2);
|
|
}
|
|
|
|
a_Writer.EndList();
|
|
|
|
a_Writer.EndCompound();
|
|
}
|
|
|
|
a_Writer.EndList(); // Teams
|
|
|
|
a_Writer.BeginCompound("DisplaySlots");
|
|
|
|
cObjective * Objective = m_ScoreBoard->GetObjectiveIn(cScoreboard::dsList);
|
|
a_Writer.AddString("slot_0", (Objective == nullptr) ? "" : Objective->GetName());
|
|
|
|
Objective = m_ScoreBoard->GetObjectiveIn(cScoreboard::dsSidebar);
|
|
a_Writer.AddString("slot_1", (Objective == nullptr) ? "" : Objective->GetName());
|
|
|
|
Objective = m_ScoreBoard->GetObjectiveIn(cScoreboard::dsName);
|
|
a_Writer.AddString("slot_2", (Objective == nullptr) ? "" : Objective->GetName());
|
|
|
|
a_Writer.EndCompound(); // DisplaySlots
|
|
|
|
a_Writer.EndCompound(); // Data
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool cScoreboardSerializer::LoadScoreboardFromNBT(const cParsedNBT & a_NBT)
|
|
{
|
|
int Data = a_NBT.FindChildByName(0, "data");
|
|
if (Data < 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
int Objectives = a_NBT.FindChildByName(Data, "Objectives");
|
|
if (Objectives < 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
for (int Child = a_NBT.GetFirstChild(Objectives); Child >= 0; Child = a_NBT.GetNextSibling(Child))
|
|
{
|
|
AString CriteriaName, DisplayName, Name;
|
|
|
|
int CurrLine = a_NBT.FindChildByName(Child, "CriteriaName");
|
|
if (CurrLine >= 0)
|
|
{
|
|
CriteriaName = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "DisplayName");
|
|
if (CurrLine >= 0)
|
|
{
|
|
DisplayName = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "Name");
|
|
if (CurrLine >= 0)
|
|
{
|
|
Name = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
cObjective::eType Type = cObjective::StringToType(CriteriaName);
|
|
|
|
m_ScoreBoard->RegisterObjective(Name, DisplayName, Type);
|
|
}
|
|
|
|
int PlayerScores = a_NBT.FindChildByName(Data, "PlayerScores");
|
|
if (PlayerScores < 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
for (int Child = a_NBT.GetFirstChild(PlayerScores); Child >= 0; Child = a_NBT.GetNextSibling(Child))
|
|
{
|
|
AString Name, ObjectiveName;
|
|
|
|
cObjective::Score Score = 0;
|
|
|
|
int CurrLine = a_NBT.FindChildByName(Child, "Score");
|
|
if (CurrLine >= 0)
|
|
{
|
|
Score = a_NBT.GetInt(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "Name");
|
|
if (CurrLine >= 0)
|
|
{
|
|
Name = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "Objective");
|
|
if (CurrLine >= 0)
|
|
{
|
|
ObjectiveName = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
cObjective * Objective = m_ScoreBoard->GetObjective(ObjectiveName);
|
|
|
|
if (Objective)
|
|
{
|
|
Objective->SetScore(Name, Score);
|
|
}
|
|
}
|
|
|
|
int Teams = a_NBT.FindChildByName(Data, "Teams");
|
|
if (Teams < 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
for (int Child = a_NBT.GetFirstChild(Teams); Child >= 0; Child = a_NBT.GetNextSibling(Child))
|
|
{
|
|
AString Name, DisplayName, Prefix, Suffix;
|
|
|
|
bool AllowsFriendlyFire = true, CanSeeFriendlyInvisible = false;
|
|
|
|
int CurrLine = a_NBT.FindChildByName(Child, "Name");
|
|
if ((CurrLine >= 0) && (a_NBT.GetType(CurrLine) == TAG_String))
|
|
{
|
|
Name = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "DisplayName");
|
|
if ((CurrLine >= 0) && (a_NBT.GetType(CurrLine) == TAG_String))
|
|
{
|
|
DisplayName = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "Prefix");
|
|
if ((CurrLine >= 0) && (a_NBT.GetType(CurrLine) == TAG_String))
|
|
{
|
|
Prefix = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "Suffix");
|
|
if ((CurrLine >= 0) && (a_NBT.GetType(CurrLine) == TAG_String))
|
|
{
|
|
Suffix = a_NBT.GetString(CurrLine);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "AllowFriendlyFire");
|
|
if ((CurrLine >= 0) && (a_NBT.GetType(CurrLine) == TAG_Int))
|
|
{
|
|
AllowsFriendlyFire = (a_NBT.GetInt(CurrLine) != 0);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(Child, "SeeFriendlyInvisibles");
|
|
if ((CurrLine >= 0) && (a_NBT.GetType(CurrLine) == TAG_Int))
|
|
{
|
|
CanSeeFriendlyInvisible = (a_NBT.GetInt(CurrLine) != 0);
|
|
}
|
|
|
|
cTeam * Team = m_ScoreBoard->RegisterTeam(Name, DisplayName, Prefix, Suffix);
|
|
|
|
Team->SetFriendlyFire(AllowsFriendlyFire);
|
|
Team->SetCanSeeFriendlyInvisible(CanSeeFriendlyInvisible);
|
|
|
|
int Players = a_NBT.FindChildByName(Child, "Players");
|
|
if (Players < 0)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
for (int ChildB = a_NBT.GetFirstChild(Players); ChildB >= 0; ChildB = a_NBT.GetNextSibling(ChildB))
|
|
{
|
|
Team->AddPlayer(a_NBT.GetString(ChildB));
|
|
}
|
|
}
|
|
|
|
int DisplaySlots = a_NBT.FindChildByName(Data, "DisplaySlots");
|
|
if (DisplaySlots < 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
int CurrLine = a_NBT.FindChildByName(DisplaySlots, "slot_0");
|
|
if (CurrLine >= 0)
|
|
{
|
|
AString Name = a_NBT.GetString(CurrLine);
|
|
|
|
m_ScoreBoard->SetDisplay(Name, cScoreboard::dsList);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(DisplaySlots, "slot_1");
|
|
if (CurrLine >= 0)
|
|
{
|
|
AString Name = a_NBT.GetString(CurrLine);
|
|
|
|
m_ScoreBoard->SetDisplay(Name, cScoreboard::dsSidebar);
|
|
}
|
|
|
|
CurrLine = a_NBT.FindChildByName(DisplaySlots, "slot_2");
|
|
if (CurrLine >= 0)
|
|
{
|
|
AString Name = a_NBT.GetString(CurrLine);
|
|
|
|
m_ScoreBoard->SetDisplay(Name, cScoreboard::dsName);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|