From 23c6f5789970570317a5f813eaa227c9c20dadc3 Mon Sep 17 00:00:00 2001
From: SHAM-DP <s.srishamharan@digipen.edu>
Date: Tue, 28 Feb 2023 21:30:24 +0800
Subject: [PATCH] Prefab Data Resolution WIP

---
 .../Serialization/Prefab/SHPrefabManager.h    |  3 +-
 .../src/Serialization/SHSerialization.cpp     | 29 ++++++++++--
 .../Serialization/SHSerializationHelper.hpp   | 46 +++++++++++++++++--
 3 files changed, 69 insertions(+), 9 deletions(-)

diff --git a/SHADE_Engine/src/Serialization/Prefab/SHPrefabManager.h b/SHADE_Engine/src/Serialization/Prefab/SHPrefabManager.h
index 6222a736..0c56ce4d 100644
--- a/SHADE_Engine/src/Serialization/Prefab/SHPrefabManager.h
+++ b/SHADE_Engine/src/Serialization/Prefab/SHPrefabManager.h
@@ -5,7 +5,7 @@
 #include "ECS_Base/General/SHFamily.h"
 #include "ECS_Base/Components/SHComponent.h"
 #include <unordered_map>
-
+#include <>
 
 namespace SHADE
 {
@@ -21,6 +21,7 @@ namespace SHADE
     };
 
     using PrefabMap = std::unordered_map<AssetID, std::vector<EntityID>>;
+    //using PrefabDatabase = std::unordered_multimap<AssetID, 
     using PrefabEntitiesComponentStatusData = std::unordered_map<EntityID, std::unordered_map<SHFamilyID<SHComponent>,PrefabEntityComponentStatus>>;
 
     static AssetID GetPrefabAssetID(EntityID eid) noexcept;
diff --git a/SHADE_Engine/src/Serialization/SHSerialization.cpp b/SHADE_Engine/src/Serialization/SHSerialization.cpp
index aae82503..4e2036d3 100644
--- a/SHADE_Engine/src/Serialization/SHSerialization.cpp
+++ b/SHADE_Engine/src/Serialization/SHSerialization.cpp
@@ -16,6 +16,7 @@
 #include "Scripting/SHScriptEngine.h"
 #include "Tools/FileIO/SHFileIO.h"
 #include "Prefab/SHPrefabManager.h"
+#include "Assets/Asset Types/SHPrefabAsset.h"
 
 namespace SHADE
 {
@@ -418,15 +419,34 @@ namespace SHADE
 
   void SHSerialization::InitializeEntity(YAML::Node const& entityNode, EntityID const& eid)
   {
-    auto const componentsNode = entityNode[ComponentsNode];
-    if (!componentsNode)
+    auto componentsNode = entityNode[ComponentsNode];
+    if (!componentsNode.IsDefined())
       return;
+    
     SHSerializationHelper::InitializeComponentFromNode<SHTransformComponent>(componentsNode, eid);
+    SHSerializationHelper::InitializeComponentFromNode<SHColliderComponent>(componentsNode, eid);
+
+    bool isPrefab = entityNode[PrefabID].IsDefined();
+    if(isPrefab)
+    {
+      auto prefabAssetID = entityNode[PrefabID].as<AssetID>();
+      if(auto prefabAsset = SHAssetManager::GetData<SHPrefabAsset>(prefabAssetID))
+      {
+        auto prefabNode = YAML::Load(prefabAsset->data);
+        if(prefabNode[ComponentsNode].IsDefined())
+        {
+          componentsNode = prefabNode[ComponentsNode];
+        }
+
+        if (prefabNode[ScriptsNode].IsDefined())
+          SHSystemManager::GetSystem<SHScriptEngine>()->DeserialiseScripts(eid, prefabNode[ScriptsNode]);
+      }
+    }
+
     SHSerializationHelper::InitializeComponentFromNode<SHCameraComponent>(componentsNode, eid);
     SHSerializationHelper::InitializeComponentFromNode<SHCameraArmComponent>(componentsNode, eid);
     SHSerializationHelper::InitializeComponentFromNode<SHRigidBodyComponent>(componentsNode, eid);
     SHSerializationHelper::InitializeComponentFromNode<SHRenderable>(componentsNode, eid);
-    SHSerializationHelper::InitializeComponentFromNode<SHColliderComponent>(componentsNode, eid);
     
     SHSerializationHelper::InitializeComponentFromNode<SHCanvasComponent>(componentsNode, eid);
     SHSerializationHelper::InitializeComponentFromNode<SHButtonComponent>(componentsNode, eid);
@@ -437,7 +457,8 @@ namespace SHADE
     SHSerializationHelper::InitializeComponentFromNode<SHAnimatorComponent>(componentsNode, eid);
     SHSerializationHelper::InitializeComponentFromNode<SHUIComponent>(componentsNode, eid);
     SHSerializationHelper::InitializeComponentFromNode<SHAudioListenerComponent>(componentsNode, eid);
-    if (entityNode[ScriptsNode].IsDefined())
+
+    if (!isPrefab && entityNode[ScriptsNode].IsDefined())
       SHSystemManager::GetSystem<SHScriptEngine>()->DeserialiseScripts(eid, entityNode[ScriptsNode]);
   }
 
diff --git a/SHADE_Engine/src/Serialization/SHSerializationHelper.hpp b/SHADE_Engine/src/Serialization/SHSerializationHelper.hpp
index a4936528..645b3927 100644
--- a/SHADE_Engine/src/Serialization/SHSerializationHelper.hpp
+++ b/SHADE_Engine/src/Serialization/SHSerializationHelper.hpp
@@ -209,7 +209,7 @@ namespace SHADE
         auto component = SHComponentManager::GetComponent_s<ComponentType>(eid);
         if (componentsNode.IsNull() || !component)
           return false;
-        auto componentNode = GetComponentNode<ComponentType>(componentsNode, eid);
+        auto componentNode = GetComponentNode<ComponentType>(componentsNode, component);
         if (componentNode.IsNull() || !componentNode.IsDefined())
           return false;
         if (componentNode[IsActive.data()].IsDefined())
@@ -242,9 +242,46 @@ namespace SHADE
     }
 
     template <typename ComponentType, std::enable_if_t<std::is_base_of_v<SHComponent, ComponentType>, bool> = true>
-    static YAML::Node GetComponentNode(YAML::Node const& componentsNode, EntityID const& eid)
+    static bool InitializeComponentFromNode(YAML::Node const& componentsNode, ComponentType* component)
+    {
+      if constexpr (YAML::HasYAMLConv<ComponentType>())
+      {
+        if (componentsNode.IsNull() || !component)
+          return false;
+        auto componentNode = GetComponentNode<ComponentType>(componentsNode, component);
+        if (componentNode.IsNull() || !componentNode.IsDefined())
+          return false;
+        if (componentNode[IsActive.data()].IsDefined())
+          component->isActive = componentNode[IsActive.data()].as<bool>();
+        YAML::convert<ComponentType>::decode(componentNode, *component);
+        return true;
+      }
+      else
+      {
+        if (componentsNode.IsNull() && !component)
+          return false;
+        auto rttrType = rttr::type::get<ComponentType>();
+        auto componentNode = componentsNode[rttrType.get_name().data()];
+        if (!componentNode.IsDefined())
+          return false;
+        if (componentNode[IsActive.data()].IsDefined())
+          component->isActive = componentNode[IsActive.data()].as<bool>();
+
+        auto properties = rttrType.get_properties();
+        for (auto const& prop : properties)
+        {
+          if (componentNode[prop.get_name().data()].IsDefined())
+          {
+            InitializeProperty<ComponentType>(component, prop, componentNode[prop.get_name().data()]);
+          }
+        }
+      }
+      return true;
+    }
+
+    template <typename ComponentType, std::enable_if_t<std::is_base_of_v<SHComponent, ComponentType>, bool> = true>
+    static YAML::Node GetComponentNode(YAML::Node const& componentsNode, ComponentType* component)
     {
-      auto component = SHComponentManager::GetComponent_s<ComponentType>(eid);
       if (componentsNode.IsNull() && !component)
         return {};
       auto rttrType = rttr::type::get<ComponentType>();
@@ -307,7 +344,8 @@ namespace SHADE
     template<>
     static void FetchAssetsFromComponent<SHRenderable>(YAML::Node const& componentsNode, EntityID const& eid, AssetQueue& assetQueue)
     {
-      auto node = GetComponentNode<SHRenderable>(componentsNode, eid);
+      SHRenderable* renderableComponent = SHComponentManager::GetComponent_s<SHRenderable>(eid);
+      auto node = GetComponentNode<SHRenderable>(componentsNode, renderableComponent);
       if(!node.IsDefined())
         return;
       if (auto const& meshNode = node[YAML::convert<SHRenderable>::MESH_YAML_TAG.data()]; meshNode.IsDefined())