#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)
{
	for(size_t i = 0; i < m_assets.size(); ++i)
	{
		if(strcmp(m_assets[i].name, assetname.c_str()) == 0)
		{
			return m_assets[i];
		}
	}
}

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)
	{
		std::cerr << "Asset '" << assetname << "' not found!" << std::endl;
		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;
}