mirror of https://github.com/libAthena/athena.git
60 lines
1.7 KiB
C++
60 lines
1.7 KiB
C++
|
#include "ZQuestFileWriter.hpp"
|
||
|
|
#include "InvalidOperationException.hpp"
|
||
|
|
#include "ZQuest.hpp"
|
||
|
|
#include "Compression.hpp"
|
||
|
|
|
||
|
|
namespace zelda
|
||
|
|
{
|
||
|
|
|
||
|
|
ZQuestFileWriter::ZQuestFileWriter(Uint8* data, Uint64 length)
|
||
|
|
: base(data, length)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
ZQuestFileWriter::ZQuestFileWriter(const std::string& filename)
|
||
|
|
: base(filename)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
void ZQuestFileWriter::write(ZQuest* quest, bool compress)
|
||
|
|
{
|
||
|
|
if (!quest)
|
||
|
|
throw InvalidOperationException("ZQuestFileWriter::writer -> quest cannot be NULL");
|
||
|
|
|
||
|
|
base::writeUInt32(ZQuest::Magic);
|
||
|
|
base::writeUInt32(ZQuest::Version);
|
||
|
|
Uint8* questData = quest->data();
|
||
|
|
Uint32 compLen = quest->length();
|
||
|
|
if (compress)
|
||
|
|
{
|
||
|
|
Uint8* compData = new Uint8[quest->length() + 0x20]; // add 20 bytes because sometimes the file grows with compression
|
||
|
|
io::Compression::compressZlib(questData, quest->length(), compData, &compLen);
|
||
|
|
|
||
|
|
// if the compressed data is the same length or larger than the original data, just store the original
|
||
|
|
if (compLen >= quest->length())
|
||
|
|
{
|
||
|
|
compLen = quest->length();
|
||
|
|
// Delete the compressed data since we won't be using it
|
||
|
|
delete[] compData;
|
||
|
|
compData = NULL;
|
||
|
|
base::writeUInt32(quest->length());
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Don't do delete on data
|
||
|
|
questData = compData;
|
||
|
|
base::writeUInt32(compLen);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
base::writeUInt32(quest->length());
|
||
|
|
|
||
|
|
base::writeUInt32(quest->length());
|
||
|
|
base::writeUInt32(quest->game());
|
||
|
|
base::writeUInt16(quest->endian() == BigEndian ? 0xFEFF : 0xFFFE);
|
||
|
|
base::seek(0x0A);
|
||
|
|
base::writeUBytes(questData, compLen);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // zelda
|