Fix render system problem when reloading current level (#205)

BoomingTech · May 24, 2022 · 204567e · 204567e
1 parent 4a38fb3
commit 204567e
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/engine/source/editor/source/editor_ui.cpp b/engine/source/editor/source/editor_ui.cpp
@@ -320,6 +320,7 @@ namespace Pilot
                 if (ImGui::MenuItem("Reload Current Level"))
                 {
                     g_runtime_global_context.m_world_manager->reloadCurrentLevel();
+                    g_runtime_global_context.m_render_system->clearForLevelReloading();
                     g_editor_global_context.m_scene_manager->onGObjectSelected(k_invalid_gobject_id);
                 }
                 if (ImGui::MenuItem("Save Current Level"))

diff --git a/engine/source/runtime/function/render/render_system.cpp b/engine/source/runtime/function/render/render_system.cpp
@@ -170,7 +170,14 @@ namespace Pilot
 
     GuidAllocator<GameObjectPartId>& RenderSystem::getGOInstanceIdAllocator() { return m_instance_id_allocator; }
 
-    GuidAllocator<MeshSourceDesc>& RenderSystem::getMeshAssetIdAllocator() { return m_mesh_asset_id_allocator; }
+    GuidAllocator<MeshSourceDesc>& RenderSystem::getMeshAssetIdAllocator() { return m_mesh_asset_id_allocator; }
+
+    void RenderSystem::clearForLevelReloading()
+    {
+        m_instance_id_allocator.clear();
+        m_mesh_object_id_map.clear();
+        m_render_scene->m_render_entities.clear();
+    }
 
     void RenderSystem::setRenderPipelineType(RENDER_PIPELINE_TYPE pipeline_type)
     {

diff --git a/engine/source/runtime/function/render/render_system.h b/engine/source/runtime/function/render/render_system.h
@@ -53,6 +53,8 @@ namespace Pilot
         GuidAllocator<GameObjectPartId>& getGOInstanceIdAllocator();
         GuidAllocator<MeshSourceDesc>&   getMeshAssetIdAllocator();
 
+        void clearForLevelReloading();
+
     private:
         RENDER_PIPELINE_TYPE m_render_pipeline_type {RENDER_PIPELINE_TYPE::DEFERRED_PIPELINE};