2015-07-01 23:50:39 +00:00
|
|
|
#include "SpecBase.hpp"
|
2015-08-04 21:35:41 +00:00
|
|
|
#include "Blender/BlenderSupport.hpp"
|
2015-10-01 00:40:21 +00:00
|
|
|
#include "BlenderConnection.hpp"
|
2015-07-01 23:50:39 +00:00
|
|
|
|
|
|
|
namespace Retro
|
|
|
|
{
|
|
|
|
|
2015-08-04 21:35:41 +00:00
|
|
|
static LogVisor::LogModule Log("Retro::SpecBase");
|
|
|
|
|
2015-09-30 06:23:40 +00:00
|
|
|
bool SpecBase::canExtract(const ExtractPassInfo& info, std::list<ExtractReport>& reps)
|
2015-07-01 23:50:39 +00:00
|
|
|
{
|
2015-07-18 04:33:38 +00:00
|
|
|
m_disc = NOD::OpenDiscFromImage(info.srcpath.c_str(), m_isWii);
|
2015-07-16 01:57:34 +00:00
|
|
|
if (!m_disc)
|
2015-07-02 20:41:40 +00:00
|
|
|
return false;
|
2015-07-16 01:57:34 +00:00
|
|
|
const char* gameID = m_disc->getHeader().gameID;
|
2015-07-02 20:41:40 +00:00
|
|
|
|
2015-07-18 04:33:38 +00:00
|
|
|
m_standalone = true;
|
|
|
|
if (m_isWii && !memcmp(gameID, "R3M", 3))
|
|
|
|
m_standalone = false;
|
2015-07-01 23:50:39 +00:00
|
|
|
|
2015-07-18 04:33:38 +00:00
|
|
|
if (m_standalone && !checkStandaloneID(gameID))
|
2015-07-10 05:28:08 +00:00
|
|
|
return false;
|
|
|
|
|
2015-07-16 01:57:34 +00:00
|
|
|
char region = m_disc->getHeader().gameID[3];
|
2015-07-13 06:29:12 +00:00
|
|
|
static const HECL::SystemString regNONE = _S("");
|
|
|
|
static const HECL::SystemString regE = _S("NTSC");
|
|
|
|
static const HECL::SystemString regJ = _S("NTSC-J");
|
|
|
|
static const HECL::SystemString regP = _S("PAL");
|
|
|
|
const HECL::SystemString* regstr = ®NONE;
|
2015-07-12 18:07:58 +00:00
|
|
|
switch (region)
|
|
|
|
{
|
|
|
|
case 'E':
|
|
|
|
regstr = ®E;
|
|
|
|
break;
|
|
|
|
case 'J':
|
|
|
|
regstr = ®J;
|
|
|
|
break;
|
|
|
|
case 'P':
|
|
|
|
regstr = ®P;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2015-07-18 04:33:38 +00:00
|
|
|
if (m_standalone)
|
2015-08-05 21:46:07 +00:00
|
|
|
return checkFromStandaloneDisc(*m_disc, *regstr, info.extractArgs, reps);
|
2015-07-10 05:28:08 +00:00
|
|
|
else
|
2015-08-05 21:46:07 +00:00
|
|
|
return checkFromTrilogyDisc(*m_disc, *regstr, info.extractArgs, reps);
|
2015-07-01 23:50:39 +00:00
|
|
|
}
|
|
|
|
|
2015-09-30 06:23:40 +00:00
|
|
|
void SpecBase::doExtract(const ExtractPassInfo& info, FProgress progress)
|
2015-07-01 23:50:39 +00:00
|
|
|
{
|
2015-08-05 21:46:07 +00:00
|
|
|
if (!Blender::BuildMasterShader(m_masterShader))
|
2015-08-04 21:35:41 +00:00
|
|
|
Log.report(LogVisor::FatalError, "Unable to build master shader blend");
|
2015-07-18 04:33:38 +00:00
|
|
|
if (m_isWii)
|
|
|
|
{
|
|
|
|
/* Extract update partition for repacking later */
|
2015-10-01 00:40:21 +00:00
|
|
|
const HECL::SystemString& target = m_project.getProjectWorkingPath().getAbsolutePath();
|
2015-07-18 04:33:38 +00:00
|
|
|
NOD::DiscBase::IPartition* update = m_disc->getUpdatePartition();
|
2015-09-29 05:17:54 +00:00
|
|
|
NOD::ExtractionContext ctx = {true, info.force, nullptr};
|
2015-09-29 03:49:31 +00:00
|
|
|
|
2015-07-18 04:33:38 +00:00
|
|
|
if (update)
|
2015-07-20 23:25:16 +00:00
|
|
|
{
|
2015-09-29 05:17:54 +00:00
|
|
|
atUint64 idx = 0;
|
2015-09-02 22:00:40 +00:00
|
|
|
progress(_S("Update Partition"), _S(""), 0, 0.0);
|
2015-09-29 05:17:54 +00:00
|
|
|
const atUint64 nodeCount = update->getFSTRoot().rawEnd() - update->getFSTRoot().rawBegin();
|
|
|
|
ctx.progressCB = [&](const std::string& name) {
|
|
|
|
HECL::SystemStringView nameView(name);
|
|
|
|
progress(_S("Update Partition"), nameView.sys_str().c_str(), 0, idx / (float)nodeCount);
|
|
|
|
idx++;
|
|
|
|
};
|
|
|
|
|
2015-09-29 03:49:31 +00:00
|
|
|
update->getFSTRoot().extractToDirectory(target, ctx);
|
2015-09-02 22:00:40 +00:00
|
|
|
progress(_S("Update Partition"), _S(""), 0, 1.0);
|
2015-07-20 23:25:16 +00:00
|
|
|
}
|
2015-07-18 04:33:38 +00:00
|
|
|
|
|
|
|
if (!m_standalone)
|
|
|
|
{
|
2015-09-02 22:00:40 +00:00
|
|
|
progress(_S("Trilogy Files"), _S(""), 1, 0.0);
|
2015-07-18 04:33:38 +00:00
|
|
|
NOD::DiscBase::IPartition* data = m_disc->getDataPartition();
|
|
|
|
const NOD::DiscBase::IPartition::Node& root = data->getFSTRoot();
|
|
|
|
for (const NOD::DiscBase::IPartition::Node& child : root)
|
|
|
|
if (child.getKind() == NOD::DiscBase::IPartition::Node::NODE_FILE)
|
2015-09-29 03:49:31 +00:00
|
|
|
child.extractToDirectory(target, ctx);
|
2015-09-02 22:00:40 +00:00
|
|
|
progress(_S("Trilogy Files"), _S(""), 1, 1.0);
|
2015-07-18 04:33:38 +00:00
|
|
|
}
|
|
|
|
}
|
2015-08-05 21:46:07 +00:00
|
|
|
extractFromDisc(*m_disc, info.force, progress);
|
2015-07-01 23:50:39 +00:00
|
|
|
}
|
|
|
|
|
2015-09-30 06:23:40 +00:00
|
|
|
bool SpecBase::canCook(const HECL::ProjectPath& path)
|
2015-07-01 23:50:39 +00:00
|
|
|
{
|
2015-10-01 00:40:21 +00:00
|
|
|
if (HECL::IsPathBlend(path))
|
|
|
|
{
|
|
|
|
HECL::BlenderConnection& conn = HECL::BlenderConnection::SharedConnection();
|
|
|
|
if (!conn.openBlend(path.getAbsolutePath()))
|
|
|
|
return false;
|
|
|
|
if (conn.getBlendType() != HECL::BlenderConnection::TypeNone)
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else if (HECL::IsPathPNG(path))
|
|
|
|
{
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else if (HECL::IsPathYAML(path))
|
|
|
|
{
|
|
|
|
FILE* fp = HECL::Fopen(path.getAbsolutePath().c_str(), _S("r"));
|
|
|
|
bool retval = validateYAMLDNAType(fp);
|
|
|
|
fclose(fp);
|
|
|
|
return retval;
|
|
|
|
}
|
2015-07-22 19:05:18 +00:00
|
|
|
return false;
|
2015-07-01 23:50:39 +00:00
|
|
|
}
|
|
|
|
|
2015-09-30 06:23:40 +00:00
|
|
|
void SpecBase::doCook(const HECL::ProjectPath& path, const HECL::ProjectPath& cookedPath)
|
2015-07-01 23:50:39 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2015-08-05 21:46:07 +00:00
|
|
|
bool SpecBase::canPackage(const PackagePassInfo& info)
|
2015-07-01 23:50:39 +00:00
|
|
|
{
|
2015-07-22 19:05:18 +00:00
|
|
|
return false;
|
2015-07-01 23:50:39 +00:00
|
|
|
}
|
|
|
|
|
2015-08-05 21:46:07 +00:00
|
|
|
void SpecBase::gatherDependencies(const PackagePassInfo& info,
|
2015-07-01 23:50:39 +00:00
|
|
|
std::unordered_set<HECL::ProjectPath>& implicitsOut)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2015-08-05 21:46:07 +00:00
|
|
|
void SpecBase::doPackage(const PackagePassInfo& info)
|
2015-07-01 23:50:39 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|