mirror of
https://github.com/rwengine/openrw.git
synced 2024-11-07 11:22:45 +01:00
124 lines
2.2 KiB
C++
124 lines
2.2 KiB
C++
#include "LoaderIMG.h"
|
|
#include <cstring>
|
|
|
|
LoaderIMG::LoaderIMG()
|
|
: m_version(GTAIIIVC)
|
|
, m_assetCount(0)
|
|
{
|
|
|
|
}
|
|
|
|
bool LoaderIMG::load(const std::string& filename)
|
|
{
|
|
std::string dirName = filename;
|
|
dirName.append(".dir");
|
|
|
|
FILE* fp = fopen(dirName.c_str(), "rb");
|
|
if(fp)
|
|
{
|
|
fseek(fp, 0, SEEK_END);
|
|
unsigned long fileSize = ftell(fp);
|
|
fseek(fp, 0, SEEK_SET);
|
|
|
|
m_assets.resize(fileSize / 32);
|
|
|
|
m_assetCount = fileSize / 32;
|
|
|
|
fread(&m_assets[0], sizeof(LoaderIMGFile), fileSize / 32, fp);
|
|
|
|
fclose(fp);
|
|
m_archive = filename;
|
|
return true;
|
|
}
|
|
else
|
|
return false;
|
|
}
|
|
|
|
/// Get the information of a asset in the examining archive
|
|
LoaderIMGFile LoaderIMG::getAssetInfo(const std::string& assetname)
|
|
{
|
|
LoaderIMGFile file;
|
|
for(size_t i = 0; i < m_assets.size(); ++i)
|
|
{
|
|
if(strcmp(m_assets[i].name, assetname.c_str()) == 0)
|
|
{
|
|
file = m_assets[i];
|
|
}
|
|
}
|
|
return file;
|
|
}
|
|
|
|
char* LoaderIMG::loadToMemory(const std::string& assetname)
|
|
{
|
|
LoaderIMGFile assetInfo;
|
|
bool found = false;
|
|
for(size_t i = 0; i < m_assets.size(); ++i)
|
|
{
|
|
if(strcmp(m_assets[i].name, assetname.c_str()) == 0)
|
|
{
|
|
assetInfo = m_assets[i];
|
|
found = true;
|
|
}
|
|
}
|
|
|
|
if(!found)
|
|
{
|
|
printf("No such asset found.\n");
|
|
return 0;
|
|
}
|
|
|
|
std::string dirName = m_archive;
|
|
dirName.append(".img");
|
|
|
|
FILE* fp = fopen(dirName.c_str(), "rb");
|
|
if(fp)
|
|
{
|
|
char* raw_data = new char[assetInfo.size * 2048];
|
|
|
|
fseek(fp, assetInfo.offset * 2048, SEEK_SET);
|
|
fread(raw_data, 2048, assetInfo.size, fp);
|
|
|
|
fclose(fp);
|
|
return raw_data;
|
|
}
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
/// Writes the contents of assetname to filename
|
|
bool LoaderIMG::saveAsset(const std::string& assetname, const std::string& filename)
|
|
{
|
|
char* raw_data = loadToMemory(assetname);
|
|
if(!raw_data)
|
|
return false;
|
|
|
|
FILE* dumpFile = fopen(filename.c_str(), "wb");
|
|
if(dumpFile)
|
|
{
|
|
fwrite(raw_data, getAssetInfo(assetname).size * 2048, 1, dumpFile);
|
|
fclose(dumpFile);
|
|
|
|
printf("=> IMG: Saved %s to disk with filename %s\n", assetname.c_str(), filename.c_str());
|
|
|
|
delete[] raw_data;
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
delete[] raw_data;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/// Get the information of an asset by its index
|
|
LoaderIMGFile LoaderIMG::getAssetInfoByIndex(size_t index)
|
|
{
|
|
return m_assets[index];
|
|
}
|
|
|
|
|
|
uint32_t LoaderIMG::getAssetCount()
|
|
{
|
|
return m_assetCount;
|
|
}
|