/
WeaponDefHandler.cpp
62 lines (42 loc) · 1.41 KB
/
WeaponDefHandler.cpp
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
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#include <algorithm>
#include <cctype>
#include <iostream>
#include <stdexcept>
#include "WeaponDefHandler.h"
#include "Lua/LuaParser.h"
#include "Sim/Misc/DamageArrayHandler.h"
#include "System/Exceptions.h"
#include "System/StringUtil.h"
static CWeaponDefHandler gWeaponDefHandler;
CWeaponDefHandler* weaponDefHandler = &gWeaponDefHandler;
void CWeaponDefHandler::Init(LuaParser* defsParser)
{
const LuaTable& rootTable = defsParser->GetRoot().SubTable("WeaponDefs");
if (!rootTable.IsValid())
throw content_error("Error loading WeaponDefs");
std::vector<std::string> weaponNames;
rootTable.GetKeys(weaponNames);
weaponDefsVector.reserve(weaponNames.size());
weaponDefIDs.reserve(weaponNames.size());
for (int wid = 0; wid < weaponNames.size(); wid++) {
const std::string& name = weaponNames[wid];
const LuaTable wdTable = rootTable.SubTable(name);
weaponDefsVector.emplace_back(wdTable, name, wid);
weaponDefIDs[name] = wid;
}
}
const WeaponDef* CWeaponDefHandler::GetWeaponDef(std::string wdName) const
{
StringToLowerInPlace(wdName);
const auto it = weaponDefIDs.find(wdName);
if (it == weaponDefIDs.end())
return nullptr;
return &weaponDefsVector[it->second];
}
const WeaponDef* CWeaponDefHandler::GetWeaponDefByID(int id) const
{
if (!IsValidWeaponDefID(id))
return nullptr;
return &weaponDefsVector[id];
}