Merge pull request #10 from EctoplasmNWN/erf-extractor

Add Tool_ErfExtractor

Author: Liareth

README

@@ -18,4 +18,5 @@ There are some tools in the Tools subdirectory:
 
 - 2da_merge allows merging a 2da into a base 2da, overwriting rows that are already present or inserting ones that are not.
 - generate_placeable_blueprints allows the user to generate a series of blueprints from placeables defined in 2da using a base blueprint
-- key_bif_extractor allows extracting all resources in a KEY from their BIFs.
+- key_bif_extractor allows extracting all resources in a KEY from their BIFs.
+- erf_extractor allows extracting all resources from an ERF.

Tools/CMakeLists.txt

@@ -2,6 +2,10 @@ add_executable(2da_merge Tool_2daMerge.cpp)
 target_link_libraries(2da_merge FileFormats)
 set_target_properties(2da_merge PROPERTIES FOLDER "Tools")
 
+add_executable(erf_extractor Tool_ErfExtractor.cpp)
+target_link_libraries(erf_extractor FileFormats)
+set_target_properties(erf_extractor PROPERTIES FOLDER "Tools")
+
 add_executable(key_bif_extractor Tool_KeyBifExtractor.cpp)
 target_link_libraries(key_bif_extractor FileFormats)
 set_target_properties(key_bif_extractor PROPERTIES FOLDER "Tools")

Tools/Tool_ErfExtractor.cpp

@@ -0,0 +1,54 @@
+#include "FileFormats/Erf.hpp"
+#include "Utility/Assert.hpp"
+
+namespace
+{
+
+int extract_erf(const char* out_path, const char* in_path)
+{
+    using namespace FileFormats::Erf;
+
+    Raw::Erf erf_raw;
+    bool loaded = Raw::Erf::ReadFromFile(in_path, &erf_raw);
+
+    if (!loaded)
+    {
+        std::printf("Failed to open %s.", in_path);
+        return 1;
+    }
+
+    Friendly::Erf erf(std::move(erf_raw));
+
+    for (const Friendly::ErfResource& resource : erf.GetResources())
+    {
+        char path[512];
+        sprintf(path, "%s/%s.%s", out_path, resource.m_ResRef.c_str(), FileFormats::Resource::StringFromResourceType(resource.m_ResType));
+
+        FILE* file = std::fopen(path, "wb");
+        if (file)
+        {
+            std::printf("Writing %s.\n", path);
+            std::fwrite(resource.m_DataBlock->GetData(), resource.m_DataBlock->GetDataLength(), 1, file);
+            std::fclose(file);
+        }
+        else
+        {
+            std::printf("Failed to open %s\n", path);
+        }
+    }
+
+    return 0;
+}
+
+}
+
+int main(int argc, char** argv)
+{
+    if (argc != 3)
+    {
+        std::printf("erf_extractor [out_dir] [erf_path]\n");
+        return 1;
+    }
+
+    return extract_erf(argv[1], argv[2]);
+}