mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-11-14 04:08:20 +00:00
87 lines
3.0 KiB
C++
87 lines
3.0 KiB
C++
#include "CDLootTableTable.h"
|
|
#include "CDClientManager.h"
|
|
#include "CDComponentsRegistryTable.h"
|
|
#include "CDItemComponentTable.h"
|
|
#include "eReplicaComponentType.h"
|
|
|
|
// Sort the tables by their rarity so the highest rarity items are first.
|
|
void SortTable(LootTableEntries& table) {
|
|
auto* componentsRegistryTable = CDClientManager::Instance().GetTable<CDComponentsRegistryTable>();
|
|
auto* itemComponentTable = CDClientManager::Instance().GetTable<CDItemComponentTable>();
|
|
// We modify the table in place so the outer loop keeps track of what is sorted
|
|
// and the inner loop finds the highest rarity item and swaps it with the current position
|
|
// of the outer loop.
|
|
for (auto oldItrOuter = table.begin(); oldItrOuter != table.end(); oldItrOuter++) {
|
|
auto lootToInsert = oldItrOuter;
|
|
// Its fine if this starts at 0, even if this doesnt match lootToInsert as the actual highest will
|
|
// either be found and overwrite these values, or the original is somehow zero and is still the highest rarity.
|
|
uint32_t highestLootRarity = 0;
|
|
for (auto oldItrInner = oldItrOuter; oldItrInner != table.end(); oldItrInner++) {
|
|
uint32_t itemComponentId = componentsRegistryTable->GetByIDAndType(oldItrInner->itemid, eReplicaComponentType::ITEM);
|
|
uint32_t rarity = itemComponentTable->GetItemComponentByID(itemComponentId).rarity;
|
|
if (rarity > highestLootRarity) {
|
|
highestLootRarity = rarity;
|
|
lootToInsert = oldItrInner;
|
|
}
|
|
}
|
|
std::swap(*oldItrOuter, *lootToInsert);
|
|
}
|
|
}
|
|
|
|
CDLootTable CDLootTableTable::ReadRow(CppSQLite3Query& tableData) const {
|
|
CDLootTable entry{};
|
|
if (tableData.eof()) return entry;
|
|
entry.itemid = tableData.getIntField("itemid", -1);
|
|
entry.MissionDrop = tableData.getIntField("MissionDrop", -1) == 1 ? true : false;
|
|
entry.sortPriority = tableData.getIntField("sortPriority", -1);
|
|
return entry;
|
|
}
|
|
|
|
void CDLootTableTable::LoadValuesFromDatabase() {
|
|
|
|
// First, get the size of the table
|
|
uint32_t size = 0;
|
|
auto tableSize = CDClientDatabase::ExecuteQuery("SELECT COUNT(*) FROM LootTable");
|
|
while (!tableSize.eof()) {
|
|
size = tableSize.getIntField(0, 0);
|
|
|
|
tableSize.nextRow();
|
|
}
|
|
|
|
// Reserve the size
|
|
this->entries.reserve(size);
|
|
|
|
// Now get the data
|
|
auto tableData = CDClientDatabase::ExecuteQuery("SELECT * FROM LootTable");
|
|
while (!tableData.eof()) {
|
|
CDLootTable entry;
|
|
uint32_t lootTableIndex = tableData.getIntField("LootTableIndex", -1);
|
|
|
|
this->entries[lootTableIndex].push_back(ReadRow(tableData));
|
|
tableData.nextRow();
|
|
}
|
|
for (auto& [id, table] : this->entries) {
|
|
SortTable(table);
|
|
}
|
|
}
|
|
|
|
const LootTableEntries& CDLootTableTable::GetTable(uint32_t tableId) {
|
|
auto itr = this->entries.find(tableId);
|
|
if (itr != this->entries.end()) {
|
|
return itr->second;
|
|
}
|
|
|
|
auto query = CDClientDatabase::CreatePreppedStmt("SELECT * FROM LootTable WHERE LootTableIndex = ?;");
|
|
query.bind(1, static_cast<int32_t>(tableId));
|
|
auto tableData = query.execQuery();
|
|
|
|
while (!tableData.eof()) {
|
|
CDLootTable entry;
|
|
this->entries[tableId].push_back(ReadRow(tableData));
|
|
tableData.nextRow();
|
|
}
|
|
SortTable(this->entries[tableId]);
|
|
|
|
return this->entries[tableId];
|
|
}
|