CharacterTemplateDataStore.cpp

Go to the documentation of this file.

/*
 * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or (at your
 * option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */
 
#include "CharacterTemplateDataStore.h"
#include "DatabaseEnv.h"
#include "DB2Stores.h"
#include "Log.h"
#include "Timer.h"
 
namespace
{
 CharacterTemplateContainer _characterTemplateStore;
}
 
void CharacterTemplateDataStore::LoadCharacterTemplates()
{
 uint32 oldMSTime = getMSTime();
 _characterTemplateStore.clear();
 
 std::unordered_map<uint32, std::vector<CharacterTemplateClass>> characterTemplateClasses;
 
 if (QueryResult classesResult = WorldDatabase.Query("SELECT TemplateId, FactionGroup, Class FROM character_template_class"))
 {
 do
 {
 Field* fields = classesResult->Fetch();
 
 uint32 templateId = fields[0].GetUInt32();
 uint8 factionGroup = fields[1].GetUInt8();
 uint8 classID = fields[2].GetUInt8();
 
 if (!((factionGroup & (FACTION_MASK_PLAYER | FACTION_MASK_ALLIANCE)) == (FACTION_MASK_PLAYER | FACTION_MASK_ALLIANCE)) &&
 !((factionGroup & (FACTION_MASK_PLAYER | FACTION_MASK_HORDE)) == (FACTION_MASK_PLAYER | FACTION_MASK_HORDE)))
 {
 TC_LOG_ERROR("sql.sql", "Faction group {} defined for character template {} in `character_template_class` is invalid. Skipped.", factionGroup, templateId);
 continue;
 }
 
 ChrClassesEntry const* classEntry = sChrClassesStore.LookupEntry(classID);
 if (!classEntry)
 {
 TC_LOG_ERROR("sql.sql", "Class {} defined for character template {} in `character_template_class` does not exists, skipped.", classID, templateId);
 continue;
 }
 
 characterTemplateClasses[templateId].emplace_back(factionGroup, classID);
 }
 while (classesResult->NextRow());
 }
 else
 {
 TC_LOG_INFO("server.loading", ">> Loaded 0 character template classes. DB table `character_template_class` is empty.");
 }
 
 QueryResult templates = WorldDatabase.Query("SELECT Id, Name, Description, Level FROM character_template");
 if (!templates)
 {
 TC_LOG_INFO("server.loading", ">> Loaded 0 character templates. DB table `character_template` is empty.");
 return;
 }
 
 do
 {
 Field* fields = templates->Fetch();
 
 CharacterTemplate templ;
 templ.TemplateSetId = fields[0].GetUInt32();
 templ.Name = fields[1].GetString();
 templ.Description = fields[2].GetString();
 templ.Level = fields[3].GetUInt8();
 templ.Classes = std::move(characterTemplateClasses[templ.TemplateSetId]);
 
 if (templ.Classes.empty())
 {
 TC_LOG_ERROR("sql.sql", "Character template {} does not have any classes defined in `character_template_class`. Skipped.", templ.TemplateSetId);
 continue;
 }
 
 _characterTemplateStore[templ.TemplateSetId] = templ;
 }
 while (templates->NextRow());
 
 TC_LOG_INFO("server.loading", ">> Loaded {} character templates in {} ms.", _characterTemplateStore.size(), GetMSTimeDiffToNow(oldMSTime));
}
 
CharacterTemplateContainer const& CharacterTemplateDataStore::GetCharacterTemplates() const
{
 return _characterTemplateStore;
}
 
CharacterTemplate const* CharacterTemplateDataStore::GetCharacterTemplate(uint32 templateId) const
{
 auto itr = _characterTemplateStore.find(templateId);
 if (itr != _characterTemplateStore.end())
 return &itr->second;
 
 return nullptr;
}
 
CharacterTemplateDataStore* CharacterTemplateDataStore::Instance()
{
 static CharacterTemplateDataStore instance;
 return &instance;
}