parse actor label

Author: moritz-h

libsave/include/SatisfactorySave/GameTypes/UE/Engine/GameFramework/Actor.h

@@ -13,11 +13,21 @@ namespace SatisfactorySave {
     public:
         FObjectReferenceDisc Owner;
         std::vector<FObjectReferenceDisc> Components;
+        std::string ActorLabel;
 
         void serialize(Archive& ar) override {
-            ar << Owner;
-            ar << Components;
+            if (!ar.IsPersistent()) {
+                ar << Owner;
+                ar << Components;
+            }
             UObject::serialize(ar);
+            if (ar.IsPersistent()) {
+                bool bIsCooked = true;
+                ar << bIsCooked;
+                if (bIsCooked) {
+                    ar << ActorLabel;
+                }
+            }
         }
     };
 

libsave/include/SatisfactorySave/IO/Archive/Archive.h

@@ -150,6 +150,14 @@ namespace SatisfactorySave {
                                                        : std::numeric_limits<int32_t>::max();
         }
 
+        inline bool IsPersistent() const {
+            return ArIsPersistent;
+        }
+
+        inline void SetIsPersistent(bool bInIsPersistent) {
+            ArIsPersistent = bInIsPersistent;
+        }
+
     protected:
         Archive() = default;
         virtual ~Archive() = default;
@@ -162,5 +170,6 @@ namespace SatisfactorySave {
 
         std::stack<int32_t> save_version_stack_;
         std::stack<int32_t> lightweight_version_stack_;
+        uint8_t ArIsPersistent : 1 = 0;
     };
 } // namespace SatisfactorySave

libsave/include/SatisfactorySave/Pak/AssetObjectVisitor.h

@@ -5,6 +5,7 @@
 #include "../GameTypes/UE/Engine/Engine/StaticMesh.h"
 #include "../GameTypes/UE/Engine/Engine/Texture2D.h"
 #include "SatisfactorySave/GameTypes/UE/CoreUObject/UObject/Object.h"
+#include "SatisfactorySave/GameTypes/UE/Engine/GameFramework/Actor.h"
 #include "satisfactorysave_export.h"
 
 namespace SatisfactorySave {
@@ -13,12 +14,15 @@ namespace SatisfactorySave {
         virtual ~AssetObjectVisitor() = default;
 
         virtual void visit(UObject& o) = 0;
+        virtual void visit(AActor& o) = 0;
         virtual void visit(UStaticMesh& o) = 0;
         virtual void visit(UTexture2D& o) = 0;
 
         void dispatch(UObject& obj) {
             if (typeid(obj) == typeid(UObject)) {
                 visit(static_cast<UObject&>(obj));
+            } else if (typeid(obj) == typeid(AActor)) {
+                visit(static_cast<AActor&>(obj));
             } else if (typeid(obj) == typeid(UStaticMesh)) {
                 visit(static_cast<UStaticMesh&>(obj));
             } else if (typeid(obj) == typeid(UTexture2D)) {

libsave/src/Pak/AssetExport.cpp

@@ -2,12 +2,15 @@
 
 #include "GameTypes/UE/Engine/Engine/StaticMesh.h"
 #include "GameTypes/UE/Engine/Engine/Texture2D.h"
+#include "GameTypes/UE/Engine/GameFramework/Actor.h"
 
 std::shared_ptr<SatisfactorySave::UObject> SatisfactorySave::AssetUObjectFactory(const std::string& class_name) {
     if (class_name == "/Script/Engine/StaticMesh") {
         return std::make_shared<UStaticMesh>();
     } else if (class_name == "/Script/Engine/Texture2D") {
         return std::make_shared<UTexture2D>();
+    } else if (class_name == "/Script/Engine/WorldPartitionHLOD") {
+        return std::make_shared<AActor>();
     } else {
         return std::make_shared<UObject>();
     }

libsave/src/Pak/AssetFile.cpp

@@ -11,8 +11,10 @@ SatisfactorySave::AssetFile::AssetFile(std::shared_ptr<PakManager> pakManager, s
     std::unique_ptr<IStream> ubulkStream)
     : IStreamArchive(std::move(uassetStream)),
       pakManager_(std::move(pakManager)) {
+    SetIsPersistent(true);
     if (ubulkStream != nullptr) {
         ubulk_ar_ = std::make_unique<IStreamArchive>(std::move(ubulkStream));
+        ubulk_ar_->SetIsPersistent(true);
     }
 
     // Check old asset format

libsave/src/Pak/IoStoreFile.cpp

@@ -52,6 +52,7 @@ SatisfactorySave::IoStoreFile::IoStoreFile(Private p, std::shared_ptr<PakManager
     }
 
     IStreamArchive utocAr(path);
+    utocAr.SetIsPersistent(true);
     utocAr << utoc_;
 
     // Debug validation
@@ -61,6 +62,7 @@ SatisfactorySave::IoStoreFile::IoStoreFile(Private p, std::shared_ptr<PakManager
 
     if (!utoc_.DirectoryIndexBuffer.empty()) {
         IStreamArchive dirIndexAr(vector_to_char_span(utoc_.DirectoryIndexBuffer));
+        dirIndexAr.SetIsPersistent(true);
 
         FIoDirectoryIndexResource dirIndexRes;
         dirIndexAr << dirIndexRes;
@@ -69,6 +71,7 @@ SatisfactorySave::IoStoreFile::IoStoreFile(Private p, std::shared_ptr<PakManager
     }
 
     ucasAr_ = std::make_unique<IStreamArchive>(ucas_path);
+    ucasAr_->SetIsPersistent(true);
 }
 
 std::vector<std::string> SatisfactorySave::IoStoreFile::getAllAssetFilenames() const {

libsave/src/Pak/PakFile.cpp

@@ -16,6 +16,7 @@ SatisfactorySave::PakFile::PakFile(Private p, std::shared_ptr<PakManager> pakMan
     }
 
     pakAr_ = std::make_unique<IStreamArchive>(pakPath);
+    pakAr_->SetIsPersistent(true);
     auto& ar = *pakAr_;
 
     if (ar.size() < 221) {

libsave/src/Pak/PakManager.cpp

@@ -21,6 +21,7 @@ void SatisfactorySave::PakManager::init(const std::filesystem::path& gameDir) {
     auto globalUtoc = IoStoreFile::create(shared_from_this(), globalUtocPath);
     const auto globalBuf = globalUtoc->readChunkContent(0);
     IStreamArchive globalAr(globalBuf);
+    globalAr.SetIsPersistent(true);
     globalAr << GlobalNameMap;
     globalAr << ScriptObjectEntries;
     if (globalAr.tell() != globalBuf.size()) {

map/src/MapWindow/UI/AssetObjectEditor.cpp

@@ -27,6 +27,13 @@ void Satisfactory3DMap::UI::AssetObjectEditor::AssetUObjectEditor::visit(s::UObj
     });
 }
 
+void Satisfactory3DMap::UI::AssetObjectEditor::AssetUObjectEditor::visit(s::AActor& o) {
+    visit(static_cast<s::UObject&>(o));
+    EditorSectionWrap(0, "AActor", [&]() {
+        EditorString("ActorLabel", o.ActorLabel);
+    });
+}
+
 void Satisfactory3DMap::UI::AssetObjectEditor::AssetUObjectEditor::visit(s::UStaticMesh& o) {
     visit(static_cast<s::UObject&>(o));
     if (ImGui::CollapsingHeader("UStaticMesh", ImGuiTreeNodeFlags_DefaultOpen)) {

map/src/MapWindow/UI/AssetObjectEditor.h

@@ -41,6 +41,7 @@ namespace Satisfactory3DMap::UI {
             explicit AssetUObjectEditor(AssetObjectEditor& parent) : parent_(parent) {}
 
             void visit(s::UObject& o) override;
+            void visit(s::AActor& o) override;
             void visit(s::UStaticMesh& o) override;
             void visit(s::UTexture2D& o) override;
         };