Mesh Component: "Multi-threaded" texture loading

Author: PR3C14D0

src/private/Core/GameObject/Component/Mesh.cpp

@@ -392,6 +392,10 @@ void Mesh::LoadModel(std::string filename) {
 	Assimp::Importer importer;
 	const aiScene* scene = importer.ReadFile(filename, NULL);
 
+	ResourceManager* resMgr = ResourceManager::GetInstance();
+	std::atomic<int> taskPending = 0;
+	std::vector<std::thread> threads;
+
 	for (UINT i = 0; i < scene->mNumMeshes; i++) {
 		aiMesh* mesh = scene->mMeshes[i];
 		aiMaterial* material = scene->mMaterials[mesh->mMaterialIndex];
@@ -443,26 +447,41 @@ void Mesh::LoadModel(std::string filename) {
 		if (material->GetTextureCount(aiTextureType_BASE_COLOR) > 0 && material->GetTexture(aiTextureType_DIFFUSE, 0, &texPath) == AI_SUCCESS) {
 			const aiTexture* texture = scene->GetEmbeddedTexture(texPath.C_Str());
 			if (texture != nullptr) {
-				ResourceManager* resMgr = ResourceManager::GetInstance();
 				std::string name = std::string(texPath.C_Str());
-				ComPtr<ID3D12Resource> resource;
-				resMgr->LoadTexture((BYTE*)texture->pcData, texture->mWidth, name, resource);
-				resource->SetName(L"Mesh Base color");
-				this->m_textures[i] = resource;
+				taskPending++;
+				threads.emplace_back([=, &taskPending, this]() {
+					if (!texture) {
+						taskPending--;
+						return;
+					}
+					ComPtr<ID3D12Resource> resource;
+					resMgr->LoadTexture((BYTE*)texture->pcData, texture->mWidth, name, resource);
+					resource->SetName(L"Mesh Base color");
+					this->m_textures[i] = resource;
+					taskPending--;
+				});
 			}
 		}
 
 		if (material->GetTextureCount(aiTextureType_METALNESS) > 0 && material->GetTexture(aiTextureType_METALNESS, 0, &metalPath) == AI_SUCCESS) {
 			const aiTexture* texture = scene->GetEmbeddedTexture(metalPath.C_Str());
 
 			if (texture != nullptr) {
-				ResourceManager* resMgr = ResourceManager::GetInstance();
 				std::string name = std::string(metalPath.C_Str());
+				if (!texture) {
+					taskPending--;
+					return;
+				}
 				ComPtr<ID3D12Resource> resource;
 				resMgr->LoadTexture((BYTE*)texture->pcData, texture->mWidth, name, resource);
 				resource->SetName(L"Mesh Metallic Roughness");
 				this->m_ORMTextures[i] = resource;
+				taskPending--;
 			}
 		}
 	}
+
+	for (std::thread& t : threads) {
+		if (t.joinable()) t.join();
+	}
 }

src/public/Core/GameObject/Component/Mesh.h

@@ -12,6 +12,9 @@
 #include "Util.h"
 #include <string>
 
+#include <thread>
+#include <atomic>
+
 #include "Core/GameObject/Component/Component.h"
 #include "Core/Renderer/Shader.h"
 #include "Core/Renderer/Descriptor.h"