ScoreboardSerializer.cpp

Go to the documentation of this file.

 
// 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)
{
    auto DataPath = fmt::format(FMT_STRING("{}{}data"), a_WorldName, cFile::PathSeparator());
    m_Path = DataPath + cFile::PathSeparator() + "scoreboard.dat";
    cFile::CreateFolder(DataPath);
}
 
 
 
 
 
bool cScoreboardSerializer::Load()
{
    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;
}