mirror of
https://github.com/HarbourMasters/Shipwright.git
synced 2024-11-30 05:12:18 -05:00
c80f9fbd57
* WIP Multiversion support * GC PAL Non-MQ support complete * Updated OtrGui to handle different game versions * Added version file * Added new extract mode to ZAPD and optimized OTR gen time * Fixed bug causing crash * Further optimized OTRExporter, saving around ~20 seconds. * ZAPD is now multi-threaded. * Fixed merge issue * Fixed memory leak and fog issue on pause screen. * Additional fog fixes. Co-authored-by: Jack Walker <7463599+Jack-Walker@users.noreply.github.com>
613 lines
14 KiB
C++
613 lines
14 KiB
C++
#include "SetMesh.h"
|
|
|
|
#include "Globals.h"
|
|
#include "Utils/BitConverter.h"
|
|
#include "Utils/Path.h"
|
|
#include "Utils/StringHelper.h"
|
|
#include "WarningHandler.h"
|
|
#include "ZBackground.h"
|
|
#include "ZFile.h"
|
|
#include "ZRoom/ZRoom.h"
|
|
|
|
void GenDListDeclarations(ZRoom* zRoom, ZFile* parent, ZDisplayList* dList);
|
|
|
|
SetMesh::SetMesh(ZFile* nParent) : ZRoomCommand(nParent)
|
|
{
|
|
}
|
|
|
|
void SetMesh::ParseRawData()
|
|
{
|
|
ZRoomCommand::ParseRawData();
|
|
auto& parentRawData = parent->GetRawData();
|
|
meshHeaderType = parentRawData.at(segmentOffset);
|
|
|
|
switch (meshHeaderType)
|
|
{
|
|
case 0:
|
|
polyType = std::make_shared<PolygonType2>(parent, segmentOffset, zRoom);
|
|
break;
|
|
|
|
case 1:
|
|
polyType = std::make_shared<PolygonType1>(parent, segmentOffset, zRoom);
|
|
break;
|
|
|
|
case 2:
|
|
polyType = std::make_shared<PolygonType2>(parent, segmentOffset, zRoom);
|
|
break;
|
|
|
|
default:
|
|
HANDLE_ERROR(WarningType::InvalidExtractedData,
|
|
StringHelper::Sprintf("unknown meshHeaderType: %i", meshHeaderType), "");
|
|
}
|
|
|
|
polyType->ParseRawData();
|
|
}
|
|
|
|
void SetMesh::DeclareReferences(const std::string& prefix)
|
|
{
|
|
polyType->SetName(polyType->GetDefaultName(prefix));
|
|
polyType->DeclareReferences(prefix);
|
|
polyType->DeclareAndGenerateOutputCode(prefix);
|
|
}
|
|
|
|
// TODO: is this really needed?
|
|
void GenDListDeclarations(ZRoom* zRoom, ZFile* parent, ZDisplayList* dList)
|
|
{
|
|
if (dList == nullptr)
|
|
return;
|
|
|
|
dList->DeclareReferences(zRoom->GetName());
|
|
|
|
for (ZDisplayList* otherDList : dList->otherDLists)
|
|
GenDListDeclarations(zRoom, parent, otherDList);
|
|
}
|
|
|
|
std::string SetMesh::GenDListExterns(ZDisplayList* dList)
|
|
{
|
|
std::string sourceOutput;
|
|
|
|
sourceOutput += StringHelper::Sprintf("extern Gfx %sDL_%06X[];\n", zRoom->GetName().c_str(),
|
|
dList->GetRawDataIndex());
|
|
|
|
for (ZDisplayList* otherDList : dList->otherDLists)
|
|
sourceOutput += GenDListExterns(otherDList);
|
|
|
|
return sourceOutput;
|
|
}
|
|
|
|
std::string SetMesh::GetBodySourceCode() const
|
|
{
|
|
std::string list;
|
|
Globals::Instance->GetSegmentedPtrName(cmdArg2, parent, "", list, parent->workerID);
|
|
return StringHelper::Sprintf("SCENE_CMD_MESH(%s)", list.c_str());
|
|
}
|
|
|
|
std::string SetMesh::GetCommandCName() const
|
|
{
|
|
return "SCmdMesh";
|
|
}
|
|
|
|
RoomCommand SetMesh::GetRoomCommand() const
|
|
{
|
|
return RoomCommand::SetMesh;
|
|
}
|
|
|
|
PolygonDlist::PolygonDlist(ZFile* nParent) : ZResource(nParent)
|
|
{
|
|
}
|
|
|
|
void PolygonDlist::ParseRawData()
|
|
{
|
|
const auto& rawData = parent->GetRawData();
|
|
switch (polyType)
|
|
{
|
|
case 2:
|
|
x = BitConverter::ToInt16BE(rawData, rawDataIndex + 0);
|
|
y = BitConverter::ToInt16BE(rawData, rawDataIndex + 2);
|
|
z = BitConverter::ToInt16BE(rawData, rawDataIndex + 4);
|
|
unk_06 = BitConverter::ToInt16BE(rawData, rawDataIndex + 6);
|
|
|
|
opa = BitConverter::ToUInt32BE(rawData, rawDataIndex + 8);
|
|
xlu = BitConverter::ToUInt32BE(rawData, rawDataIndex + 12);
|
|
break;
|
|
|
|
default:
|
|
opa = BitConverter::ToUInt32BE(rawData, rawDataIndex);
|
|
xlu = BitConverter::ToUInt32BE(rawData, rawDataIndex + 4);
|
|
break;
|
|
}
|
|
}
|
|
|
|
void PolygonDlist::DeclareReferences(const std::string& prefix)
|
|
{
|
|
opaDList = MakeDlist(opa, prefix);
|
|
xluDList = MakeDlist(xlu, prefix);
|
|
}
|
|
|
|
std::string PolygonDlist::GetBodySourceCode() const
|
|
{
|
|
std::string bodyStr;
|
|
std::string opaStr;
|
|
std::string xluStr;
|
|
Globals::Instance->GetSegmentedPtrName(opa, parent, "Gfx", opaStr, parent->workerID);
|
|
Globals::Instance->GetSegmentedPtrName(xlu, parent, "Gfx", xluStr, parent->workerID);
|
|
|
|
if (polyType == 2)
|
|
{
|
|
bodyStr += StringHelper::Sprintf("{ %6i, %6i, %6i }, %6i, ", x, y, z, unk_06);
|
|
}
|
|
|
|
bodyStr += StringHelper::Sprintf("%s, %s", opaStr.c_str(), xluStr.c_str());
|
|
|
|
return bodyStr;
|
|
}
|
|
|
|
void PolygonDlist::GetSourceOutputCode(const std::string& prefix)
|
|
{
|
|
std::string bodyStr = StringHelper::Sprintf("\n\t%s\n", GetBodySourceCode().c_str());
|
|
|
|
Declaration* decl = parent->GetDeclaration(rawDataIndex);
|
|
|
|
if (decl == nullptr)
|
|
DeclareVar(prefix, bodyStr);
|
|
else
|
|
decl->text = bodyStr;
|
|
}
|
|
|
|
std::string PolygonDlist::GetSourceTypeName() const
|
|
{
|
|
switch (polyType)
|
|
{
|
|
case 2:
|
|
return "PolygonDlist2";
|
|
|
|
default:
|
|
return "PolygonDlist";
|
|
}
|
|
}
|
|
|
|
ZResourceType PolygonDlist::GetResourceType() const
|
|
{
|
|
// TODO
|
|
return ZResourceType::Error;
|
|
}
|
|
|
|
size_t PolygonDlist::GetRawDataSize() const
|
|
{
|
|
switch (polyType)
|
|
{
|
|
case 2:
|
|
return 0x10;
|
|
|
|
default:
|
|
return 0x08;
|
|
}
|
|
}
|
|
|
|
void PolygonDlist::SetPolyType(uint8_t nPolyType)
|
|
{
|
|
polyType = nPolyType;
|
|
}
|
|
|
|
ZDisplayList* PolygonDlist::MakeDlist(segptr_t ptr, [[maybe_unused]] const std::string& prefix)
|
|
{
|
|
if (ptr == 0)
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
uint32_t dlistAddress = Seg2Filespace(ptr, parent->baseAddress);
|
|
|
|
int32_t dlistLength = ZDisplayList::GetDListLength(
|
|
parent->GetRawData(), dlistAddress,
|
|
Globals::Instance->game == ZGame::OOT_SW97 ? DListType::F3DEX : DListType::F3DZEX);
|
|
ZDisplayList* dlist = new ZDisplayList(parent);
|
|
parent->AddResource(dlist);
|
|
dlist->ExtractFromBinary(dlistAddress, dlistLength);
|
|
dlist->SetName(dlist->GetDefaultName(prefix));
|
|
GenDListDeclarations(zRoom, parent, dlist);
|
|
|
|
return dlist;
|
|
}
|
|
|
|
/* BgImage */
|
|
|
|
BgImage::BgImage(ZFile* nParent) : ZResource(nParent)
|
|
{
|
|
}
|
|
|
|
BgImage::BgImage(bool nIsSubStruct, const std::string& prefix, uint32_t nRawDataIndex,
|
|
ZFile* nParent)
|
|
: BgImage(nParent)
|
|
{
|
|
rawDataIndex = nRawDataIndex;
|
|
parent = nParent;
|
|
isSubStruct = nIsSubStruct;
|
|
|
|
name = GetDefaultName(prefix);
|
|
|
|
ParseRawData();
|
|
sourceBackground = MakeBackground(source, prefix);
|
|
}
|
|
|
|
void BgImage::ParseRawData()
|
|
{
|
|
size_t pad = 0x00;
|
|
const auto& rawData = parent->GetRawData();
|
|
if (!isSubStruct)
|
|
{
|
|
pad = 0x04;
|
|
|
|
unk_00 = BitConverter::ToUInt16BE(rawData, rawDataIndex + 0x00);
|
|
id = BitConverter::ToUInt8BE(rawData, rawDataIndex + 0x02);
|
|
}
|
|
source = BitConverter::ToUInt32BE(rawData, rawDataIndex + pad + 0x00);
|
|
unk_0C = BitConverter::ToUInt32BE(rawData, rawDataIndex + pad + 0x04);
|
|
tlut = BitConverter::ToUInt32BE(rawData, rawDataIndex + pad + 0x08);
|
|
width = BitConverter::ToUInt16BE(rawData, rawDataIndex + pad + 0x0C);
|
|
height = BitConverter::ToUInt16BE(rawData, rawDataIndex + pad + 0x0E);
|
|
fmt = BitConverter::ToUInt8BE(rawData, rawDataIndex + pad + 0x10);
|
|
siz = BitConverter::ToUInt8BE(rawData, rawDataIndex + pad + 0x11);
|
|
mode0 = BitConverter::ToUInt16BE(rawData, rawDataIndex + pad + 0x12);
|
|
tlutCount = BitConverter::ToUInt16BE(rawData, rawDataIndex + pad + 0x14);
|
|
}
|
|
|
|
ZBackground* BgImage::MakeBackground(segptr_t ptr, const std::string& prefix)
|
|
{
|
|
if (ptr == 0)
|
|
return nullptr;
|
|
|
|
uint32_t backAddress = Seg2Filespace(ptr, parent->baseAddress);
|
|
|
|
ZBackground* background = new ZBackground(parent);
|
|
background->ExtractFromFile(backAddress);
|
|
|
|
std::string defaultName = background->GetDefaultName(prefix);
|
|
background->SetName(defaultName);
|
|
background->SetOutName(defaultName);
|
|
|
|
background->DeclareVar(prefix, "");
|
|
parent->resources.push_back(background);
|
|
|
|
return background;
|
|
}
|
|
|
|
size_t BgImage::GetRawDataSize() const
|
|
{
|
|
return 0x1C;
|
|
}
|
|
|
|
std::string BgImage::GetBodySourceCode() const
|
|
{
|
|
std::string bodyStr = " ";
|
|
if (!isSubStruct)
|
|
{
|
|
bodyStr += "{ \n ";
|
|
}
|
|
|
|
if (!isSubStruct)
|
|
{
|
|
bodyStr += StringHelper::Sprintf("0x%04X, ", unk_00);
|
|
bodyStr += StringHelper::Sprintf("%i, ", id);
|
|
bodyStr += "\n ";
|
|
bodyStr += " ";
|
|
}
|
|
|
|
std::string backgroundName;
|
|
Globals::Instance->GetSegmentedPtrName(source, parent, "", backgroundName, parent->workerID);
|
|
bodyStr += StringHelper::Sprintf("%s, ", backgroundName.c_str());
|
|
bodyStr += "\n ";
|
|
if (!isSubStruct)
|
|
{
|
|
bodyStr += " ";
|
|
}
|
|
|
|
bodyStr += StringHelper::Sprintf("0x%08X, ", unk_0C);
|
|
bodyStr += StringHelper::Sprintf("0x%08X, ", tlut);
|
|
bodyStr += "\n ";
|
|
if (!isSubStruct)
|
|
{
|
|
bodyStr += " ";
|
|
}
|
|
|
|
bodyStr += StringHelper::Sprintf("%i, ", width);
|
|
bodyStr += StringHelper::Sprintf("%i, ", height);
|
|
bodyStr += "\n ";
|
|
if (!isSubStruct)
|
|
{
|
|
bodyStr += " ";
|
|
}
|
|
|
|
bodyStr += StringHelper::Sprintf("%i, ", fmt);
|
|
bodyStr += StringHelper::Sprintf("%i, ", siz);
|
|
bodyStr += "\n ";
|
|
if (!isSubStruct)
|
|
{
|
|
bodyStr += " ";
|
|
}
|
|
|
|
bodyStr += StringHelper::Sprintf("0x%04X, ", mode0);
|
|
bodyStr += StringHelper::Sprintf("0x%04X, ", tlutCount);
|
|
if (!isSubStruct)
|
|
{
|
|
bodyStr += " \n }, ";
|
|
}
|
|
else
|
|
{
|
|
bodyStr += "\n";
|
|
}
|
|
|
|
return bodyStr;
|
|
}
|
|
|
|
std::string BgImage::GetSourceTypeName() const
|
|
{
|
|
return "BgImage";
|
|
}
|
|
|
|
ZResourceType BgImage::GetResourceType() const
|
|
{
|
|
// TODO
|
|
return ZResourceType::Error;
|
|
}
|
|
|
|
/* PolygonType section */
|
|
|
|
PolygonTypeBase::PolygonTypeBase(ZFile* nParent, uint32_t nRawDataIndex, ZRoom* nRoom)
|
|
: ZResource(nParent), zRoom{nRoom}
|
|
{
|
|
rawDataIndex = nRawDataIndex;
|
|
type = BitConverter::ToUInt8BE(parent->GetRawData(), rawDataIndex);
|
|
}
|
|
|
|
void PolygonTypeBase::DeclareAndGenerateOutputCode(const std::string& prefix)
|
|
{
|
|
std::string bodyStr = GetBodySourceCode();
|
|
|
|
Declaration* decl = parent->GetDeclaration(rawDataIndex);
|
|
if (decl == nullptr)
|
|
{
|
|
DeclareVar(prefix, bodyStr);
|
|
}
|
|
else
|
|
{
|
|
decl->text = bodyStr;
|
|
}
|
|
}
|
|
|
|
std::string PolygonTypeBase::GetSourceTypeName() const
|
|
{
|
|
switch (type)
|
|
{
|
|
case 2:
|
|
return "PolygonType2";
|
|
|
|
case 1:
|
|
return "PolygonType1";
|
|
|
|
default:
|
|
return "PolygonType0";
|
|
}
|
|
}
|
|
|
|
ZResourceType PolygonTypeBase::GetResourceType() const
|
|
{
|
|
// TODO
|
|
return ZResourceType::Error;
|
|
}
|
|
|
|
PolygonType1::PolygonType1(ZFile* nParent, uint32_t nRawDataIndex, ZRoom* nRoom)
|
|
: PolygonTypeBase(nParent, nRawDataIndex, nRoom), single(nParent)
|
|
{
|
|
}
|
|
|
|
void PolygonType1::ParseRawData()
|
|
{
|
|
const auto& rawData = parent->GetRawData();
|
|
|
|
format = BitConverter::ToUInt8BE(rawData, rawDataIndex + 0x01);
|
|
dlist = BitConverter::ToUInt32BE(rawData, rawDataIndex + 0x04);
|
|
|
|
if (format == 2)
|
|
{
|
|
count = BitConverter::ToUInt8BE(rawData, rawDataIndex + 0x08);
|
|
list = BitConverter::ToUInt32BE(rawData, rawDataIndex + 0x0C);
|
|
}
|
|
|
|
if (dlist != 0)
|
|
{
|
|
PolygonDlist polyGfxList(parent);
|
|
polyGfxList.zRoom = zRoom;
|
|
polyGfxList.SetPolyType(type);
|
|
polyGfxList.ExtractFromFile(Seg2Filespace(dlist, parent->baseAddress));
|
|
polyGfxList.DeclareReferences(zRoom->GetName());
|
|
polyDLists.push_back(polyGfxList);
|
|
}
|
|
}
|
|
|
|
void PolygonType1::DeclareReferences(const std::string& prefix)
|
|
{
|
|
polyDLists.at(0).GetSourceOutputCode(prefix);
|
|
|
|
uint32_t listAddress;
|
|
std::string bgImageArrayBody;
|
|
switch (format)
|
|
{
|
|
case 1:
|
|
single = BgImage(true, prefix, rawDataIndex + 0x08, parent);
|
|
break;
|
|
|
|
case 2:
|
|
if (list != 0)
|
|
{
|
|
listAddress = Seg2Filespace(list, parent->baseAddress);
|
|
uint32_t auxPtr = listAddress;
|
|
for (size_t i = 0; i < count; ++i)
|
|
{
|
|
BgImage bg(false, prefix, auxPtr, parent);
|
|
multiList.push_back(bg);
|
|
auxPtr += bg.GetRawDataSize();
|
|
bgImageArrayBody += bg.GetBodySourceCode();
|
|
if (i + 1 < count)
|
|
{
|
|
bgImageArrayBody += "\n";
|
|
}
|
|
}
|
|
|
|
Declaration* decl = parent->GetDeclaration(listAddress);
|
|
if (decl == nullptr)
|
|
{
|
|
parent->AddDeclarationArray(
|
|
listAddress, DeclarationAlignment::Align4,
|
|
count * multiList.at(0).GetRawDataSize(), multiList.at(0).GetSourceTypeName(),
|
|
multiList.at(0).GetName().c_str(), count, bgImageArrayBody);
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
HANDLE_ERROR(WarningType::InvalidExtractedData,
|
|
StringHelper::Sprintf("unknown format: %i", format), "");
|
|
break;
|
|
}
|
|
}
|
|
|
|
size_t PolygonType1::GetRawDataSize() const
|
|
{
|
|
switch (format)
|
|
{
|
|
case 1:
|
|
return 0x20;
|
|
|
|
case 2:
|
|
return 0x10;
|
|
}
|
|
return 0x20;
|
|
}
|
|
|
|
std::string PolygonType1::GetBodySourceCode() const
|
|
{
|
|
std::string bodyStr = "\n ";
|
|
|
|
bodyStr += "{ ";
|
|
bodyStr += StringHelper::Sprintf("%i, %i, ", type, format);
|
|
|
|
std::string dlistStr;
|
|
Globals::Instance->GetSegmentedPtrName(dlist, parent, "", dlistStr, parent->workerID);
|
|
|
|
bodyStr += StringHelper::Sprintf("%s, ", dlistStr.c_str());
|
|
bodyStr += "}, \n";
|
|
|
|
std::string listStr = "NULL";
|
|
switch (format)
|
|
{
|
|
case 1:
|
|
bodyStr += single.GetBodySourceCode();
|
|
break;
|
|
case 2:
|
|
Globals::Instance->GetSegmentedPtrName(list, parent, "BgImage", listStr, parent->workerID);
|
|
bodyStr += StringHelper::Sprintf(" %i, %s, \n", count, listStr.c_str());
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return bodyStr;
|
|
}
|
|
|
|
std::string PolygonType1::GetSourceTypeName() const
|
|
{
|
|
switch (format)
|
|
{
|
|
case 1:
|
|
return "MeshHeader1Single";
|
|
|
|
case 2:
|
|
return "MeshHeader1Multi";
|
|
}
|
|
return "ERROR";
|
|
// return "PolygonType1";
|
|
}
|
|
|
|
PolygonType2::PolygonType2(ZFile* nParent, uint32_t nRawDataIndex, ZRoom* nRoom)
|
|
: PolygonTypeBase(nParent, nRawDataIndex, nRoom)
|
|
{
|
|
}
|
|
|
|
void PolygonType2::ParseRawData()
|
|
{
|
|
const auto& rawData = parent->GetRawData();
|
|
|
|
num = BitConverter::ToUInt8BE(rawData, rawDataIndex + 0x01);
|
|
|
|
start = BitConverter::ToUInt32BE(rawData, rawDataIndex + 0x04);
|
|
end = BitConverter::ToUInt32BE(rawData, rawDataIndex + 0x08);
|
|
|
|
uint32_t currentPtr = GETSEGOFFSET(start);
|
|
for (size_t i = 0; i < num; i++)
|
|
{
|
|
PolygonDlist entry(parent);
|
|
entry.zRoom = zRoom;
|
|
entry.SetPolyType(type);
|
|
entry.ExtractFromFile(currentPtr);
|
|
entry.DeclareReferences(zRoom->GetName());
|
|
polyDLists.push_back(entry);
|
|
currentPtr += entry.GetRawDataSize();
|
|
}
|
|
}
|
|
|
|
void PolygonType2::DeclareReferences(const std::string& prefix)
|
|
{
|
|
if (num > 0)
|
|
{
|
|
std::string declaration;
|
|
|
|
for (size_t i = 0; i < polyDLists.size(); i++)
|
|
{
|
|
declaration +=
|
|
StringHelper::Sprintf("\t{ %s },", polyDLists.at(i).GetBodySourceCode().c_str());
|
|
if (i + 1 < polyDLists.size())
|
|
declaration += "\n";
|
|
}
|
|
|
|
std::string polyDlistType = polyDLists.at(0).GetSourceTypeName();
|
|
std::string polyDListName;
|
|
polyDListName = StringHelper::Sprintf("%s%s_%06X", prefix.c_str(), polyDlistType.c_str(),
|
|
GETSEGOFFSET(start));
|
|
|
|
Declaration* decl = parent->AddDeclarationArray(
|
|
GETSEGOFFSET(start), DeclarationAlignment::Align4,
|
|
polyDLists.size() * polyDLists.at(0).GetRawDataSize(), polyDlistType, polyDListName,
|
|
polyDLists.size(), declaration);
|
|
decl->forceArrayCnt = true;
|
|
}
|
|
|
|
parent->AddDeclaration(GETSEGOFFSET(end), DeclarationAlignment::Align4, 4, "s32",
|
|
StringHelper::Sprintf("%s_terminatorMaybe_%06X",
|
|
parent->GetName().c_str(), GETSEGOFFSET(end)),
|
|
"0x01000000");
|
|
}
|
|
|
|
std::string PolygonType2::GetBodySourceCode() const
|
|
{
|
|
std::string listName;
|
|
Globals::Instance->GetSegmentedPtrName(start, parent, "", listName, parent->workerID);
|
|
|
|
std::string body = StringHelper::Sprintf("\n %i, %i,\n", type, polyDLists.size());
|
|
body += StringHelper::Sprintf(" %s,\n", listName.c_str());
|
|
body +=
|
|
StringHelper::Sprintf(" %s + ARRAY_COUNTU(%s)\n", listName.c_str(), listName.c_str());
|
|
return body;
|
|
}
|
|
|
|
size_t PolygonType2::GetRawDataSize() const
|
|
{
|
|
return 0x0C;
|
|
}
|
|
|
|
DeclarationAlignment PolygonType2::GetDeclarationAlignment() const
|
|
{
|
|
return DeclarationAlignment::Align4;
|
|
}
|