mirror of https://github.com/AxioDL/metaforce.git
103 lines
3.1 KiB
C++
103 lines
3.1 KiB
C++
#include "CSNG.hpp"
|
|
#include "amuse/SongConverter.hpp"
|
|
|
|
namespace DataSpec::DNAMP1
|
|
{
|
|
|
|
bool CSNG::Extract(PAKEntryReadStream& rs, const hecl::ProjectPath& outPath)
|
|
{
|
|
hecl::ProjectPath midPath = outPath.getWithExtension(_S(".mid"), true);
|
|
hecl::ProjectPath yamlPath = outPath.getWithExtension(_S(".yaml"), true);
|
|
|
|
Header head;
|
|
head.read(rs);
|
|
|
|
{
|
|
athena::io::YAMLDocWriter dw("CSNG");
|
|
dw.writeUint32("midiSetupId", head.midiSetupId);
|
|
dw.writeUint32("songGroupId", head.songGroupId);
|
|
if (auto rec = dw.enterSubRecord("agscId"))
|
|
head.agscId.write(dw);
|
|
|
|
athena::io::FileWriter w(yamlPath.getAbsolutePath());
|
|
if (w.hasError())
|
|
return false;
|
|
dw.finish(&w);
|
|
}
|
|
|
|
{
|
|
auto sng = rs.readUBytes(head.sngLength);
|
|
int version;
|
|
bool isBig;
|
|
auto midi = amuse::SongConverter::SongToMIDI(sng.get(), version, isBig);
|
|
|
|
athena::io::FileWriter w(midPath.getAbsolutePath());
|
|
if (w.hasError())
|
|
return false;
|
|
w.writeUBytes(midi.data(), midi.size());
|
|
}
|
|
|
|
/* Update !songs.yaml for Amuse editor */
|
|
hecl::ProjectPath audGrp(outPath.getParentPath().getParentPath(), _S("AudioGrp"));
|
|
audGrp.makeDirChain(true);
|
|
hecl::ProjectPath songsPath(audGrp, _S("!songs.yaml"));
|
|
std::experimental::optional<athena::io::FileReader> r;
|
|
if (songsPath.isFile())
|
|
r.emplace(songsPath.getAbsolutePath());
|
|
athena::io::YAMLDocWriter ydw("amuse::Songs", r ? &*r : nullptr);
|
|
r = std::experimental::nullopt;
|
|
char id[16];
|
|
snprintf(id, 16, "%04X", head.midiSetupId);
|
|
ydw.writeString(id, hecl::Format("../MidiData/%s", midPath.getLastComponentUTF8().data()));
|
|
athena::io::FileWriter w(songsPath.getAbsolutePath());
|
|
ydw.finish(&w);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CSNG::Cook(const hecl::ProjectPath& inPath, const hecl::ProjectPath& outPath)
|
|
{
|
|
hecl::ProjectPath midPath = inPath.getWithExtension(_S(".mid"), true);
|
|
hecl::ProjectPath yamlPath = inPath.getWithExtension(_S(".yaml"), true);
|
|
|
|
std::vector<uint8_t> sngData;
|
|
{
|
|
athena::io::FileReader midR(midPath.getAbsolutePath());
|
|
if (midR.hasError())
|
|
return false;
|
|
|
|
uint32_t midLen = midR.length();
|
|
std::vector<uint8_t> midData;
|
|
midData.resize(midLen);
|
|
midR.readUBytesToBuf(midData.data(), midLen);
|
|
sngData = amuse::SongConverter::MIDIToSong(midData, 1, true);
|
|
}
|
|
|
|
athena::io::FileWriter w(outPath.getAbsolutePath());
|
|
if (w.hasError())
|
|
return false;
|
|
|
|
{
|
|
athena::io::FileReader yamlR(yamlPath.getAbsolutePath());
|
|
if (yamlR.hasError())
|
|
return false;
|
|
athena::io::YAMLDocReader dr;
|
|
if (!dr.parse(&yamlR))
|
|
return false;
|
|
|
|
Header head;
|
|
head.midiSetupId = dr.readUint32("midiSetupId");
|
|
head.songGroupId = dr.readUint32("songGroupId");
|
|
if (auto rec = dr.enterSubRecord("agscId"))
|
|
head.agscId.read(dr);
|
|
head.sngLength = sngData.size();
|
|
head.write(w);
|
|
}
|
|
|
|
w.writeUBytes(sngData.data(), sngData.size());
|
|
|
|
return true;
|
|
}
|
|
|
|
}
|