104 lines
2.0 KiB
C++
104 lines
2.0 KiB
C++
#include "PackManager.h"
|
|
#include "EterLib/BufferPool.h"
|
|
#include <fstream>
|
|
#include <filesystem>
|
|
|
|
CPackManager::CPackManager()
|
|
: m_load_from_pack(true)
|
|
, m_pBufferPool(nullptr)
|
|
{
|
|
m_pBufferPool = new CBufferPool();
|
|
}
|
|
|
|
CPackManager::~CPackManager()
|
|
{
|
|
if (m_pBufferPool)
|
|
{
|
|
delete m_pBufferPool;
|
|
m_pBufferPool = nullptr;
|
|
}
|
|
}
|
|
|
|
bool CPackManager::AddPack(const std::string& path)
|
|
{
|
|
std::shared_ptr<CPack> pack = std::make_shared<CPack>();
|
|
|
|
if (!pack->Load(path))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
const auto& index = pack->GetIndex();
|
|
for (const auto& entry : index)
|
|
{
|
|
m_entries[entry.file_name] = std::make_pair(pack, entry);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CPackManager::GetFile(std::string_view path, TPackFile& result)
|
|
{
|
|
return GetFileWithPool(path, result, m_pBufferPool);
|
|
}
|
|
|
|
bool CPackManager::GetFileWithPool(std::string_view path, TPackFile& result, CBufferPool* pPool)
|
|
{
|
|
thread_local std::string buf;
|
|
NormalizePath(path, buf);
|
|
|
|
if (m_load_from_pack) {
|
|
auto it = m_entries.find(buf);
|
|
if (it != m_entries.end()) {
|
|
return it->second.first->GetFileWithPool(it->second.second, result, pPool);
|
|
}
|
|
}
|
|
else {
|
|
std::ifstream ifs(buf, std::ios::binary);
|
|
if (ifs.is_open()) {
|
|
ifs.seekg(0, std::ios::end);
|
|
size_t size = ifs.tellg();
|
|
ifs.seekg(0, std::ios::beg);
|
|
|
|
if (pPool) {
|
|
result = pPool->Acquire(size);
|
|
result.resize(size);
|
|
} else {
|
|
result.resize(size);
|
|
}
|
|
|
|
if (ifs.read((char*)result.data(), size)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool CPackManager::IsExist(std::string_view path) const
|
|
{
|
|
thread_local std::string buf;
|
|
NormalizePath(path, buf);
|
|
|
|
if (m_load_from_pack) {
|
|
auto it = m_entries.find(buf);
|
|
return it != m_entries.end();
|
|
}
|
|
else {
|
|
return std::filesystem::exists(buf);
|
|
}
|
|
}
|
|
|
|
void CPackManager::NormalizePath(std::string_view in, std::string& out) const
|
|
{
|
|
out.resize(in.size());
|
|
for (std::size_t i = 0; i < out.size(); ++i) {
|
|
if (in[i] == '\\')
|
|
out[i] = '/';
|
|
else
|
|
out[i] = static_cast<char>(std::tolower(in[i]));
|
|
}
|
|
}
|