#include "amuse/AudioGroupProject.hpp" #include "amuse/AudioGroupData.hpp" #include "athena/MemoryReader.hpp" #include "athena/FileWriter.hpp" #include "athena/FileReader.hpp" namespace amuse { static bool AtEnd32(athena::io::IStreamReader& r) { uint32_t v = r.readUint32Big(); r.seek(-4, athena::Current); return v == 0xffffffff; } static bool AtEnd16(athena::io::IStreamReader& r) { uint16_t v = r.readUint16Big(); r.seek(-2, athena::Current); return v == 0xffff; } template static void ReadRangedObjectIds(NameDB* db, athena::io::IStreamReader& r, NameDB::Type tp) { uint16_t id; athena::io::Read::Do({}, id, r); if ((id & 0x8000) == 0x8000) { uint16_t endId; athena::io::Read::Do({}, endId, r); for (uint16_t i = uint16_t(id & 0x7fff); i <= uint16_t(endId & 0x7fff); ++i) { ObjectId useId = i; if (tp == NameDB::Type::Layer) useId.id |= 0x8000; db->registerPair(NameDB::generateName(useId, tp), useId); } } else { db->registerPair(NameDB::generateName(id, tp), id); } } AudioGroupProject::AudioGroupProject(athena::io::IStreamReader& r, GCNDataTag) { while (!AtEnd32(r)) { GroupHeader header; header.read(r); GroupId::CurNameDB->registerPair(NameDB::generateName(header.groupId, NameDB::Type::Group), header.groupId); /* Sound Macros */ r.seek(header.soundMacroIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SoundMacroId::CurNameDB, r, NameDB::Type::SoundMacro); /* Samples */ r.seek(header.samplIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SampleId::CurNameDB, r, NameDB::Type::Sample); /* Tables */ r.seek(header.tableIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(TableId::CurNameDB, r, NameDB::Type::Table); /* Keymaps */ r.seek(header.keymapIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(KeymapId::CurNameDB, r, NameDB::Type::Keymap); /* Layers */ r.seek(header.layerIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(LayersId::CurNameDB, r, NameDB::Type::Layer); if (header.type == GroupType::Song) { auto& idx = m_songGroups[header.groupId]; idx = MakeObj(); /* Normal pages */ r.seek(header.pageTableOff, athena::Begin); while (!AtEnd16(r)) { SongGroupIndex::PageEntryDNA entry; entry.read(r); idx->m_normPages[entry.programNo] = entry; } /* Drum pages */ r.seek(header.drumTableOff, athena::Begin); while (!AtEnd16(r)) { SongGroupIndex::PageEntryDNA entry; entry.read(r); idx->m_drumPages[entry.programNo] = entry; } /* MIDI setups */ r.seek(header.midiSetupsOff, athena::Begin); while (r.position() < header.groupEndOff) { uint16_t songId = r.readUint16Big(); r.seek(2, athena::Current); std::array& setup = idx->m_midiSetups[songId]; for (int i = 0; i < 16 ; ++i) setup[i].read(r); SongId::CurNameDB->registerPair(NameDB::generateName(songId, NameDB::Type::Song), songId); } } else if (header.type == GroupType::SFX) { auto& idx = m_sfxGroups[header.groupId]; idx = MakeObj(); /* SFX entries */ r.seek(header.pageTableOff, athena::Begin); uint16_t count = r.readUint16Big(); r.seek(2, athena::Current); idx->m_sfxEntries.reserve(count); for (int i = 0; i < count; ++i) { SFXGroupIndex::SFXEntryDNA entry; entry.read(r); idx->m_sfxEntries[entry.sfxId.id] = entry; SFXId::CurNameDB->registerPair( NameDB::generateName(entry.sfxId.id, NameDB::Type::SFX), entry.sfxId.id); } } r.seek(header.groupEndOff, athena::Begin); } } template AudioGroupProject AudioGroupProject::_AudioGroupProject(athena::io::IStreamReader& r, bool absOffs) { AudioGroupProject ret; while (!AtEnd32(r)) { atInt64 groupBegin = r.position(); atInt64 subDataOff = absOffs ? 0 : groupBegin + 8; GroupHeader header; header.read(r); GroupId::CurNameDB->registerPair(NameDB::generateName(header.groupId, NameDB::Type::Group), header.groupId); /* Sound Macros */ r.seek(subDataOff + header.soundMacroIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SoundMacroId::CurNameDB, r, NameDB::Type::SoundMacro); /* Samples */ r.seek(subDataOff + header.samplIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SampleId::CurNameDB, r, NameDB::Type::Sample); /* Tables */ r.seek(subDataOff + header.tableIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(TableId::CurNameDB, r, NameDB::Type::Table); /* Keymaps */ r.seek(subDataOff + header.keymapIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(KeymapId::CurNameDB, r, NameDB::Type::Keymap); /* Layers */ r.seek(subDataOff + header.layerIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(LayersId::CurNameDB, r, NameDB::Type::Layer); if (header.type == GroupType::Song) { auto& idx = ret.m_songGroups[header.groupId]; idx = MakeObj(); if (absOffs) { /* Normal pages */ r.seek(header.pageTableOff, athena::Begin); while (!AtEnd16(r)) { SongGroupIndex::PageEntryDNA entry; entry.read(r); idx->m_normPages[entry.programNo] = entry; } /* Drum pages */ r.seek(header.drumTableOff, athena::Begin); while (!AtEnd16(r)) { SongGroupIndex::PageEntryDNA entry; entry.read(r); idx->m_drumPages[entry.programNo] = entry; } /* MIDI setups */ r.seek(header.midiSetupsOff, athena::Begin); while (r.position() < header.groupEndOff) { uint16_t songId; athena::io::Read::Do({}, songId, r); r.seek(2, athena::Current); std::array& setup = idx->m_midiSetups[songId]; for (int i = 0; i < 16 ; ++i) setup[i].read(r); SongId::CurNameDB->registerPair(NameDB::generateName(songId, NameDB::Type::Song), songId); } } else { /* Normal pages */ r.seek(subDataOff + header.pageTableOff, athena::Begin); while (!AtEnd16(r)) { SongGroupIndex::MusyX1PageEntryDNA entry; entry.read(r); idx->m_normPages[entry.programNo] = entry; } /* Drum pages */ r.seek(subDataOff + header.drumTableOff, athena::Begin); while (!AtEnd16(r)) { SongGroupIndex::MusyX1PageEntryDNA entry; entry.read(r); idx->m_drumPages[entry.programNo] = entry; } /* MIDI setups */ r.seek(subDataOff + header.midiSetupsOff, athena::Begin); while (r.position() < groupBegin + header.groupEndOff) { uint16_t songId; athena::io::Read::Do({}, songId, r); r.seek(2, athena::Current); std::array& setup = idx->m_midiSetups[songId]; for (int i = 0; i < 16 ; ++i) { SongGroupIndex::MusyX1MIDISetup ent; ent.read(r); setup[i] = ent; } SongId::CurNameDB->registerPair(NameDB::generateName(songId, NameDB::Type::Song), songId); } } } else if (header.type == GroupType::SFX) { auto& idx = ret.m_sfxGroups[header.groupId]; idx = MakeObj(); /* SFX entries */ r.seek(subDataOff + header.pageTableOff, athena::Begin); uint16_t count; athena::io::Read::Do({}, count, r); r.seek(2, athena::Current); idx->m_sfxEntries.reserve(count); for (int i = 0; i < count; ++i) { SFXGroupIndex::SFXEntryDNA entry; entry.read(r); r.seek(2, athena::Current); idx->m_sfxEntries[entry.sfxId.id] = entry; SFXId::CurNameDB->registerPair( NameDB::generateName(entry.sfxId.id, NameDB::Type::SFX), entry.sfxId.id); } } if (absOffs) r.seek(header.groupEndOff, athena::Begin); else r.seek(groupBegin + header.groupEndOff, athena::Begin); } return ret; } AudioGroupProject AudioGroupProject::CreateAudioGroupProject(const AudioGroupData& data) { athena::io::MemoryReader r(data.getProj(), data.getProjSize()); switch (data.getDataFormat()) { case DataFormat::GCN: default: return AudioGroupProject(r, GCNDataTag{}); case DataFormat::N64: return _AudioGroupProject(r, data.getAbsoluteProjOffsets()); case DataFormat::PC: return _AudioGroupProject(r, data.getAbsoluteProjOffsets()); } } std::string ParseStringSlashId(const std::string& str, uint16_t& idOut) { size_t slashPos = str.find('/'); if (slashPos == std::string::npos) return {}; idOut = uint16_t(strtoul(str.data() + slashPos + 1, nullptr, 0)); return {str.begin(), str.begin() + slashPos}; } AudioGroupProject AudioGroupProject::CreateAudioGroupProject(SystemStringView groupPath) { AudioGroupProject ret; SystemString projPath(groupPath); projPath += _S("/!project.yaml"); athena::io::FileReader fi(projPath); if (!fi.hasError()) { athena::io::YAMLDocReader r; if (r.parse(&fi) && !r.readString("DNAType").compare("amuse::Project")) { if (auto __v = r.enterSubRecord("songGroups")) { ret.m_songGroups.reserve(r.getCurNode()->m_mapChildren.size()); for (const auto& grp : r.getCurNode()->m_mapChildren) { if (auto __r = r.enterSubRecord(grp.first.c_str())) { uint16_t groupId; std::string groupName = ParseStringSlashId(grp.first, groupId); if (groupName.empty() || groupId == 0xffff) continue; GroupId::CurNameDB->registerPair(groupName, groupId); auto& idx = ret.m_songGroups[groupId]; idx = MakeObj(); if (auto __v2 = r.enterSubRecord("normPages")) { idx->m_normPages.reserve(r.getCurNode()->m_mapChildren.size()); for (const auto& pg : r.getCurNode()->m_mapChildren) if (auto __r2 = r.enterSubRecord(pg.first.c_str())) idx->m_normPages[strtoul(pg.first.c_str(), nullptr, 0)].read(r); } if (auto __v2 = r.enterSubRecord("drumPages")) { idx->m_drumPages.reserve(r.getCurNode()->m_mapChildren.size()); for (const auto& pg : r.getCurNode()->m_mapChildren) if (auto __r2 = r.enterSubRecord(pg.first.c_str())) idx->m_drumPages[strtoul(pg.first.c_str(), nullptr, 0)].read(r); } if (auto __v2 = r.enterSubRecord("songs")) { idx->m_midiSetups.reserve(r.getCurNode()->m_mapChildren.size()); for (const auto& song : r.getCurNode()->m_mapChildren) { size_t chanCount; if (auto __v3 = r.enterSubVector(song.first.c_str(), chanCount)) { uint16_t songId; std::string songName = ParseStringSlashId(song.first, songId); if (songName.empty() || songId == 0xffff) continue; SongId::CurNameDB->registerPair(songName, songId); std::array& setup = idx->m_midiSetups[songId]; for (int i = 0; i < 16 && i < chanCount; ++i) if (auto __r2 = r.enterSubRecord(nullptr)) setup[i].read(r); } } } } } } if (auto __v = r.enterSubRecord("sfxGroups")) { ret.m_sfxGroups.reserve(r.getCurNode()->m_mapChildren.size()); for (const auto& grp : r.getCurNode()->m_mapChildren) { if (auto __r = r.enterSubRecord(grp.first.c_str())) { uint16_t groupId; std::string groupName = ParseStringSlashId(grp.first, groupId); if (groupName.empty() || groupId == 0xffff) continue; GroupId::CurNameDB->registerPair(groupName, groupId); auto& idx = ret.m_sfxGroups[groupId]; idx = MakeObj(); for (const auto& sfx : r.getCurNode()->m_mapChildren) if (auto __r2 = r.enterSubRecord(sfx.first.c_str())) { uint16_t sfxId; std::string sfxName = ParseStringSlashId(sfx.first, sfxId); if (sfxName.empty() || sfxId == 0xffff) continue; SFXId::CurNameDB->registerPair(sfxName, sfxId); idx->m_sfxEntries[sfxId].read(r); } } } } } } return ret; } void AudioGroupProject::BootstrapObjectIDs(athena::io::IStreamReader& r, GCNDataTag) { while (!AtEnd32(r)) { GroupHeader header; header.read(r); GroupId::CurNameDB->registerPair(NameDB::generateName(header.groupId, NameDB::Type::Group), header.groupId); /* Sound Macros */ r.seek(header.soundMacroIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SoundMacroId::CurNameDB, r, NameDB::Type::SoundMacro); /* Samples */ r.seek(header.samplIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SampleId::CurNameDB, r, NameDB::Type::Sample); /* Tables */ r.seek(header.tableIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(TableId::CurNameDB, r, NameDB::Type::Table); /* Keymaps */ r.seek(header.keymapIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(KeymapId::CurNameDB, r, NameDB::Type::Keymap); /* Layers */ r.seek(header.layerIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(LayersId::CurNameDB, r, NameDB::Type::Layer); if (header.type == GroupType::Song) { /* MIDI setups */ r.seek(header.midiSetupsOff, athena::Begin); while (r.position() < header.groupEndOff) { uint16_t id = r.readUint16Big(); SongId::CurNameDB->registerPair(NameDB::generateName(id, NameDB::Type::Song), id); r.seek(2 + 5 * 16, athena::Current); } } else if (header.type == GroupType::SFX) { /* SFX entries */ r.seek(header.pageTableOff, athena::Begin); uint16_t count = r.readUint16Big(); r.seek(2, athena::Current); for (int i = 0; i < count; ++i) { SFXGroupIndex::SFXEntryDNA entry; entry.read(r); SFXId::CurNameDB->registerPair( NameDB::generateName(entry.sfxId.id, NameDB::Type::SFX), entry.sfxId.id); } } r.seek(header.groupEndOff, athena::Begin); } } template void AudioGroupProject::BootstrapObjectIDs(athena::io::IStreamReader& r, bool absOffs) { while (!AtEnd32(r)) { atInt64 groupBegin = r.position(); atInt64 subDataOff = absOffs ? 0 : groupBegin + 8; GroupHeader header; header.read(r); GroupId::CurNameDB->registerPair(NameDB::generateName(header.groupId, NameDB::Type::Group), header.groupId); /* Sound Macros */ r.seek(subDataOff + header.soundMacroIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SoundMacroId::CurNameDB, r, NameDB::Type::SoundMacro); /* Samples */ r.seek(subDataOff + header.samplIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(SampleId::CurNameDB, r, NameDB::Type::Sample); /* Tables */ r.seek(subDataOff + header.tableIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(TableId::CurNameDB, r, NameDB::Type::Table); /* Keymaps */ r.seek(subDataOff + header.keymapIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(KeymapId::CurNameDB, r, NameDB::Type::Keymap); /* Layers */ r.seek(subDataOff + header.layerIdsOff, athena::Begin); while (!AtEnd16(r)) ReadRangedObjectIds(LayersId::CurNameDB, r, NameDB::Type::Layer); if (header.type == GroupType::Song) { /* MIDI setups */ if (absOffs) { r.seek(header.midiSetupsOff, athena::Begin); while (r.position() < header.groupEndOff) { uint16_t id; athena::io::Read::Do({}, id, r); SongId::CurNameDB->registerPair(NameDB::generateName(id, NameDB::Type::Song), id); r.seek(2 + 5 * 16, athena::Current); } } else { r.seek(subDataOff + header.midiSetupsOff, athena::Begin); while (r.position() < groupBegin + header.groupEndOff) { uint16_t id; athena::io::Read::Do({}, id, r); SongId::CurNameDB->registerPair(NameDB::generateName(id, NameDB::Type::Song), id); r.seek(2 + 8 * 16, athena::Current); } } } else if (header.type == GroupType::SFX) { /* SFX entries */ r.seek(subDataOff + header.pageTableOff, athena::Begin); uint16_t count; athena::io::Read::Do({}, count, r); r.seek(2, athena::Current); for (int i = 0; i < count; ++i) { SFXGroupIndex::SFXEntryDNA entry; entry.read(r); r.seek(2, athena::Current); SFXId::CurNameDB->registerPair( NameDB::generateName(entry.sfxId.id, NameDB::Type::SFX), entry.sfxId.id); } } if (absOffs) r.seek(header.groupEndOff, athena::Begin); else r.seek(groupBegin + header.groupEndOff, athena::Begin); } } void AudioGroupProject::BootstrapObjectIDs(const AudioGroupData& data) { athena::io::MemoryReader r(data.getProj(), data.getProjSize()); switch (data.getDataFormat()) { case DataFormat::GCN: default: BootstrapObjectIDs(r, GCNDataTag{}); break; case DataFormat::N64: BootstrapObjectIDs(r, data.getAbsoluteProjOffsets()); break; case DataFormat::PC: BootstrapObjectIDs(r, data.getAbsoluteProjOffsets()); break; } } const SongGroupIndex* AudioGroupProject::getSongGroupIndex(int groupId) const { auto search = m_songGroups.find(groupId); if (search != m_songGroups.cend()) return search->second.get(); return nullptr; } const SFXGroupIndex* AudioGroupProject::getSFXGroupIndex(int groupId) const { auto search = m_sfxGroups.find(groupId); if (search != m_sfxGroups.cend()) return search->second.get(); return nullptr; } bool AudioGroupProject::toYAML(SystemStringView groupPath) const { athena::io::YAMLDocWriter w("amuse::Project"); if (!m_songGroups.empty()) { if (auto __v = w.enterSubRecord("songGroups")) { for (const auto& p : SortUnorderedMap(m_songGroups)) { char groupString[64]; snprintf(groupString, 64, "%s/0x%04X", GroupId::CurNameDB->resolveNameFromId(p.first).data(), int(p.first.id)); if (auto __r = w.enterSubRecord(groupString)) { if (!p.second.get()->m_normPages.empty()) { if (auto __v2 = w.enterSubRecord("normPages")) { for (const auto& pg : SortUnorderedMap(p.second.get()->m_normPages)) { char name[16]; snprintf(name, 16, "%d", pg.first); if (auto __r2 = w.enterSubRecord(name)) { w.setStyle(athena::io::YAMLNodeStyle::Flow); pg.second.get().write(w); } } } } if (!p.second.get()->m_drumPages.empty()) { if (auto __v2 = w.enterSubRecord("drumPages")) { for (const auto& pg : SortUnorderedMap(p.second.get()->m_drumPages)) { char name[16]; snprintf(name, 16, "%d", pg.first); if (auto __r2 = w.enterSubRecord(name)) { w.setStyle(athena::io::YAMLNodeStyle::Flow); pg.second.get().write(w); } } } } if (!p.second.get()->m_midiSetups.empty()) { if (auto __v2 = w.enterSubRecord("songs")) { for (const auto& song : SortUnorderedMap(p.second.get()->m_midiSetups)) { char songString[64]; snprintf(songString, 64, "%s/0x%04X", SongId::CurNameDB->resolveNameFromId(song.first).data(), int(song.first.id)); if (auto __v3 = w.enterSubVector(songString)) for (int i = 0; i < 16; ++i) if (auto __r2 = w.enterSubRecord(nullptr)) { w.setStyle(athena::io::YAMLNodeStyle::Flow); song.second.get()[i].write(w); } } } } } } } } if (!m_sfxGroups.empty()) { if (auto __v = w.enterSubRecord("sfxGroups")) { for (const auto& p : SortUnorderedMap(m_sfxGroups)) { char groupString[64]; snprintf(groupString, 64, "%s/0x%04X", GroupId::CurNameDB->resolveNameFromId(p.first).data(), int(p.first.id)); if (auto __r = w.enterSubRecord(groupString)) { for (const auto& sfx : SortUnorderedMap(p.second.get()->m_sfxEntries)) { char sfxString[64]; snprintf(sfxString, 64, "%s/0x%04X", SFXId::CurNameDB->resolveNameFromId(sfx.first).data(), int(sfx.first.id)); if (auto __r2 = w.enterSubRecord(sfxString)) { w.setStyle(athena::io::YAMLNodeStyle::Flow); sfx.second.get().write(w); } } } } } } SystemString projPath(groupPath); projPath += _S("/!project.yaml"); athena::io::FileWriter fo(projPath); return w.finish(&fo); } }