SpaceCadetPinballPPC/SpaceCadetPinball/partman.cpp
Muzychenko Andrey 69ecce88df Score: inject 3DPB msg font into dat struct.
Fixed double free in sound.
2021-10-10 12:22:21 +03:00

135 lines
3.5 KiB
C++

#include "pch.h"
#include "partman.h"
#include "gdrv.h"
#include "GroupData.h"
#include "zdrv.h"
short partman::_field_size[] =
{
2, -1, 2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 0
};
DatFile* partman::load_records(LPCSTR lpFileName, bool fullTiltMode)
{
datFileHeader header{};
dat8BitBmpHeader bmpHeader{};
dat16BitBmpHeader zMapHeader{};
auto fileHandle = fopen(lpFileName, "rb");
if (fileHandle == nullptr)
return nullptr;
fread(&header, 1, sizeof header, fileHandle);
if (strcmp("PARTOUT(4.0)RESOURCE", header.FileSignature) != 0)
{
fclose(fileHandle);
return nullptr;
}
auto datFile = new DatFile();
if (!datFile)
{
fclose(fileHandle);
return nullptr;
}
datFile->AppName = header.AppName;
datFile->Description = header.Description;
if (header.Unknown)
{
auto unknownBuf = new char[header.Unknown];
if (!unknownBuf)
{
fclose(fileHandle);
delete datFile;
return nullptr;
}
fread(unknownBuf, 1, header.Unknown, fileHandle);
delete[] unknownBuf;
}
datFile->Groups.reserve(header.NumberOfGroups);
bool abort = false;
for (auto groupIndex = 0; !abort && groupIndex < header.NumberOfGroups; ++groupIndex)
{
auto entryCount = LRead<uint8_t>(fileHandle);
auto groupData = new GroupData(groupIndex);
groupData->ReserveEntries(entryCount);
for (auto entryIndex = 0; entryIndex < entryCount; ++entryIndex)
{
auto entryData = new EntryData();
auto entryType = static_cast<FieldTypes>(LRead<uint8_t>(fileHandle));
entryData->EntryType = entryType;
int fixedSize = _field_size[static_cast<int>(entryType)];
size_t fieldSize = fixedSize >= 0 ? fixedSize : LRead<uint32_t>(fileHandle);
entryData->FieldSize = static_cast<int>(fieldSize);
if (entryType == FieldTypes::Bitmap8bit)
{
fread(&bmpHeader, 1, sizeof(dat8BitBmpHeader), fileHandle);
assertm(bmpHeader.Size + sizeof(dat8BitBmpHeader) == fieldSize, "partman: Wrong bitmap field size");
assertm(bmpHeader.Resolution <= 2, "partman: bitmap resolution out of bounds");
auto bmp = new gdrv_bitmap8(bmpHeader);
entryData->Buffer = reinterpret_cast<char*>(bmp);
fread(bmp->IndexedBmpPtr, 1, bmpHeader.Size, fileHandle);
}
else if (entryType == FieldTypes::Bitmap16bit)
{
/*Full tilt has extra byte(@0:resolution) in zMap*/
auto zMapResolution = 0u;
if (fullTiltMode)
{
zMapResolution = LRead<uint8_t>(fileHandle);
fieldSize--;
assertm(zMapResolution <= 2, "partman: zMap resolution out of bounds");
}
fread(&zMapHeader, 1, sizeof(dat16BitBmpHeader), fileHandle);
auto length = fieldSize - sizeof(dat16BitBmpHeader);
auto zMap = new zmap_header_type(zMapHeader.Width, zMapHeader.Height, zMapHeader.Stride);
zMap->Resolution = zMapResolution;
if (zMapHeader.Stride * zMapHeader.Height * 2u == length)
{
fread(zMap->ZPtr1, 1, length, fileHandle);
}
else
{
// 3DPB .dat has zeroed zMap headers, in groups 497 and 498, skip them.
fseek(fileHandle, static_cast<int>(length), SEEK_CUR);
}
entryData->Buffer = reinterpret_cast<char*>(zMap);
}
else
{
auto entryBuffer = new char[fieldSize];
entryData->Buffer = entryBuffer;
if (!entryBuffer)
{
abort = true;
break;
}
fread(entryBuffer, 1, fieldSize, fileHandle);
}
groupData->AddEntry(entryData);
}
datFile->Groups.push_back(groupData);
}
fclose(fileHandle);
if (datFile->Groups.size() == header.NumberOfGroups)
{
datFile->Finalize();
return datFile;
}
delete datFile;
return nullptr;
}