Initial support for AssetNameMap

DataSpec/AssetNameMap.cpp

@@ -0,0 +1,80 @@
+#include "AssetNameMap.hpp"
+#include "athena/MemoryReader.hpp"
+
+extern "C" uint8_t ASSET_NAME_MP32[];
+extern "C" uint32_t ASSET_NAME_MP32_SZ;
+extern "C" uint8_t ASSET_NAME_MP64[];
+extern "C" uint32_t ASSET_NAME_MP64_SZ;
+
+namespace DataSpec
+{
+namespace AssetNameMap
+{
+logvisor::Module Log("AssetNameMap");
+
+struct SAsset
+{
+    std::string name;
+    std::string directory;
+    hecl::FourCC type;
+    SAsset() = default;
+    SAsset(athena::io::IStreamReader& in)
+    {
+        uint32_t nameLen = in.readUint32Big();
+        name = in.readString(nameLen);
+        uint32_t dirLen = in.readUint32Big();
+        directory = in.readString(dirLen);
+        type = in.readUint32Big();
+    }
+};
+
+static std::unordered_map<uint64_t, SAsset> g_AssetNameMap;
+static bool g_AssetNameMapInit = false;
+
+void LoadAssetMap(athena::io::MemoryReader ar)
+{
+    if (!ar.hasError())
+    {
+        hecl::FourCC magic = ar.readUint32Big();
+        if (magic != FOURCC('AIDM'))
+            Log.report(logvisor::Error, _S("Unable to load asset map; Assets will not have proper filenames for most files."));
+        else
+        {
+            uint32_t assetCount = ar.readUint32Big();
+            for (uint32_t i = 0 ; i<assetCount ; ++i)
+            {
+                uint64_t id = ar.readUint64Big();
+                g_AssetNameMap[id] = SAsset(ar);
+            }
+        }
+    }
+}
+
+void InitAssetNameMap()
+{
+    if (g_AssetNameMapInit)
+        return;
+
+    /* First load the 32bit map for MP1/2 */
+    {
+        athena::io::MemoryReader ar(ASSET_NAME_MP32, ASSET_NAME_MP32_SZ);
+        LoadAssetMap(ar);
+    }
+    /* Now load the 64bit map for MP3 */
+    {
+        athena::io::MemoryReader ar(ASSET_NAME_MP64, ASSET_NAME_MP64_SZ);
+        LoadAssetMap(ar);
+    }
+    g_AssetNameMapInit = true;
+}
+
+const std::string* TranslateIdToName(const UniqueID32& id)
+{
+    if (g_AssetNameMap.find(id.toUint64()) == g_AssetNameMap.end())
+        return nullptr;
+
+    return &g_AssetNameMap[id.toUint64()].name;
+}
+
+}
+}