mirror of
https://github.com/Laupetin/OpenAssetTools.git
synced 2025-06-08 22:08:29 -05:00
93 lines
2.9 KiB
C++
93 lines
2.9 KiB
C++
#include "AssetLoaderStringTable.h"
|
|
|
|
#include <cstring>
|
|
|
|
#include "Csv/CsvStream.h"
|
|
#include "Game/T6/CommonT6.h"
|
|
#include "Game/T6/T6.h"
|
|
#include "Pool/GlobalAssetPool.h"
|
|
|
|
using namespace T6;
|
|
|
|
void* AssetLoaderStringTable::CreateEmptyAsset(const std::string& assetName, MemoryManager* memory)
|
|
{
|
|
auto* stringTable = memory->Create<StringTable>();
|
|
memset(stringTable, 0, sizeof(StringTable));
|
|
stringTable->name = memory->Dup(assetName.c_str());
|
|
return stringTable;
|
|
}
|
|
|
|
bool AssetLoaderStringTable::CanLoadFromRaw() const
|
|
{
|
|
return true;
|
|
}
|
|
|
|
bool AssetLoaderStringTable::LoadFromRaw(const std::string& assetName, ISearchPath* searchPath, MemoryManager* memory, IAssetLoadingManager* manager, Zone* zone) const
|
|
{
|
|
const auto file = searchPath->Open(assetName);
|
|
if (!file.IsOpen())
|
|
return false;
|
|
|
|
auto* stringTable = memory->Create<StringTable>();
|
|
stringTable->name = memory->Dup(assetName.c_str());
|
|
|
|
std::vector<std::vector<std::string>> csvLines;
|
|
std::vector<std::string> currentLine;
|
|
auto maxCols = 0u;
|
|
const CsvInputStream csv(*file.m_stream);
|
|
|
|
while (csv.NextRow(currentLine))
|
|
{
|
|
if (currentLine.size() > maxCols)
|
|
maxCols = currentLine.size();
|
|
csvLines.emplace_back(std::move(currentLine));
|
|
}
|
|
|
|
stringTable->columnCount = maxCols;
|
|
stringTable->rowCount = csvLines.size();
|
|
const auto cellCount = static_cast<unsigned>(stringTable->rowCount) * static_cast<unsigned>(stringTable->columnCount);
|
|
|
|
if (cellCount)
|
|
{
|
|
stringTable->values = static_cast<StringTableCell*>(memory->Alloc(sizeof(StringTableCell) * cellCount));
|
|
stringTable->cellIndex = static_cast<int16_t*>(memory->Alloc(sizeof(int16_t) * cellCount));
|
|
|
|
for (auto c = 0u; c < cellCount; c++)
|
|
stringTable->cellIndex[c] = static_cast<int16_t>(c);
|
|
|
|
for (auto row = 0u; row < csvLines.size(); row++)
|
|
{
|
|
const auto& rowValues = csvLines[row];
|
|
for (auto col = 0u; col < maxCols; col++)
|
|
{
|
|
auto& cell = stringTable->values[row * maxCols + col];
|
|
if (col >= rowValues.size() || rowValues[col].empty())
|
|
cell.string = "";
|
|
else
|
|
cell.string = memory->Dup(rowValues[col].c_str());
|
|
|
|
cell.hash = Common::Com_HashString(cell.string);
|
|
}
|
|
}
|
|
|
|
|
|
std::sort(&stringTable->cellIndex[0], &stringTable->cellIndex[cellCount - 1], [stringTable, maxCols](const int16_t a, const int16_t b)
|
|
{
|
|
auto compareResult = stringTable->values[a].hash - stringTable->values[b].hash;
|
|
if (compareResult == 0)
|
|
compareResult = a % maxCols - b % maxCols;
|
|
return compareResult < 0;
|
|
});
|
|
}
|
|
|
|
else
|
|
{
|
|
stringTable->values = nullptr;
|
|
stringTable->cellIndex = nullptr;
|
|
}
|
|
|
|
manager->AddAsset(ASSET_TYPE_STRINGTABLE, assetName, stringTable);
|
|
|
|
return true;
|
|
}
|