[WIP] Audio Extractor

1 files changed, 9 insertions, 5 deletions

diff --git a/src/Companion.cpp b/src/Companion.cpp
index 16d6c11..218f018 100644
--- a/src/Companion.cpp
+++ b/src/Companion.cpp
@@ -4,6 +4,7 @@
 #include "utils/MIODecoder.h"
 #include "factories/RawFactory.h"
 #include "factories/BlobFactory.h"
+#include "factories/AudioFactory.h"
 #include "factories/TextureFactory.h"
 #include "factories/AnimFactory.h"
 
@@ -18,9 +19,10 @@ namespace fs = std::filesystem;
 static const std::unordered_map<std::string, RawFactory*> gFactories = {
     { ".png", new TextureFactory() },
     { ".anim", new AnimFactory() },
+    { ".aiff", new AudioFactory() },
     { ".bin", new BlobFactory(LUS::ResourceType::Blob) },
+    { ".m64", new BlobFactory(LUS::ResourceType::Blob) },
     { ".sbox", new BlobFactory(LUS::ResourceType::Blob, true) },
-    { ".aud", new BlobFactory(LUS::ResourceType::Blob, false) },
 };
 
 void Companion::Start() {
@@ -45,7 +47,7 @@ void Companion::ProcessAssets() {
     for( auto& [asset, data] : assets.items() ) {
         std::string extension = fs::path(asset).extension().string();
         if( gFactories.find(extension) == gFactories.end() ) {
-            std::cout << "No factory found for " << asset << '\n';
+            // std::cout << "No factory found for " << asset << '\n';
             continue;
         }
         LUS::BinaryWriter write = LUS::BinaryWriter();
@@ -57,7 +59,7 @@ void Companion::ProcessAssets() {
         RawFactory* factory = gFactories.at(extension);
 
         if(!factory->process(&write, entry, this->gRomData)){
-            std::cout << "Failed to process " << asset << '\n';
+            // std::cout << "Failed to process " << asset << '\n';
             continue;
         }
 
@@ -70,13 +72,15 @@ void Companion::ProcessAssets() {
         if(!isTexture) {
             // Write to file too
             std::string dpath = "debug/" + path;
-            fs::create_directories(fs::path(dpath).parent_path());
+            if(!fs::exists(fs::path(dpath).parent_path())){
+                fs::create_directories(fs::path(dpath).parent_path());
+            }
             std::ofstream output(dpath, std::ios::binary);
             output.write((char*)buffer.data(), buffer.size());
             output.close();
         }
 
-        std::cout << "Processed " << path << std::endl;
+        // std::cout << "Processed " << path << std::endl;
 
         write.Close();
     }