Refactor Mesh serialization and model system.
This commit is contained in:
antopilo
225
Nuake/src/Resource/ModelLoader.cpp
Normal file
225
Nuake/src/Resource/ModelLoader.cpp
Normal file
@@ -0,0 +1,225 @@
|
||||
#include "src/Resource/ModelLoader.h"
|
||||
|
||||
#include "src/Core/FileSystem.h"
|
||||
#include "src/Core/Logger.h"
|
||||
|
||||
#include "src/Core/String.h"
|
||||
|
||||
#include "src/Resource/Model.h"
|
||||
namespace Nuake
|
||||
{
|
||||
ModelLoader::ModelLoader() {}
|
||||
ModelLoader::~ModelLoader() {}
|
||||
|
||||
Ref<Model> ModelLoader::LoadModel(const std::string& path)
|
||||
{
|
||||
Ref<Model> model = CreateRef<Model>(path);
|
||||
|
||||
Assimp::Importer import;
|
||||
import.SetPropertyFloat("PP_GSN_MAX_SMOOTHING_ANGLE", 90);
|
||||
|
||||
auto importFlags =
|
||||
aiProcess_Triangulate |
|
||||
aiProcess_GenSmoothNormals |
|
||||
aiProcess_FixInfacingNormals |
|
||||
aiProcess_CalcTangentSpace;
|
||||
|
||||
modelDir = path + "/../";
|
||||
const aiScene* scene = import.ReadFile(FileSystem::Root + path, importFlags);
|
||||
if (!scene || scene->mFlags & AI_SCENE_FLAGS_INCOMPLETE || !scene->mRootNode)
|
||||
{
|
||||
std::string assimpErrorMsg = std::string(import.GetErrorString());
|
||||
std::string logMsg = "[Failed to load model] - " + assimpErrorMsg;
|
||||
Logger::Log(logMsg, WARNING);
|
||||
|
||||
return model;
|
||||
}
|
||||
|
||||
ProcessNode(scene->mRootNode, scene);
|
||||
|
||||
for (const auto& mesh : m_Meshes)
|
||||
{
|
||||
model->AddMesh(mesh);
|
||||
}
|
||||
|
||||
return model;
|
||||
}
|
||||
|
||||
void ModelLoader::ProcessNode(aiNode* node, const aiScene* scene)
|
||||
{
|
||||
for (uint32_t i = 0; i < node->mNumMeshes; i++)
|
||||
{
|
||||
aiMesh* mesh = scene->mMeshes[node->mMeshes[i]];
|
||||
m_Meshes.push_back(ProcessMesh(mesh, scene));
|
||||
}
|
||||
|
||||
for (uint32_t i = 0; i < node->mNumChildren; i++)
|
||||
{
|
||||
ProcessNode(node->mChildren[i], scene);
|
||||
}
|
||||
}
|
||||
|
||||
Ref<Mesh> ModelLoader::ProcessMesh(aiMesh* node, const aiScene* scene)
|
||||
{
|
||||
auto vertices = ProcessVertices(node);
|
||||
auto indices = ProcessIndices(node);
|
||||
auto material = ProcessMaterials(scene, node);
|
||||
|
||||
Ref<Mesh> mesh = CreateRef<Mesh>();
|
||||
mesh->AddSurface(vertices, indices);
|
||||
mesh->SetMaterial(material);
|
||||
|
||||
return mesh;
|
||||
}
|
||||
|
||||
std::vector<Vertex> ModelLoader::ProcessVertices(aiMesh* mesh)
|
||||
{
|
||||
auto vertices = std::vector<Vertex>();
|
||||
for (uint32_t i = 0; i < mesh->mNumVertices; i++)
|
||||
{
|
||||
Vertex vertex;
|
||||
vertex.texture = 1.0f;
|
||||
|
||||
Vector3 current;
|
||||
|
||||
// Position
|
||||
current.x = mesh->mVertices[i].x;
|
||||
current.y = mesh->mVertices[i].y;
|
||||
current.z = mesh->mVertices[i].z;
|
||||
vertex.position = current;
|
||||
|
||||
// Normals
|
||||
current.x = mesh->mNormals[i].x;
|
||||
current.y = mesh->mNormals[i].y;
|
||||
current.z = mesh->mNormals[i].z;
|
||||
vertex.normal = current;
|
||||
|
||||
// Tangents
|
||||
if (mesh->mTangents)
|
||||
{
|
||||
current.x = mesh->mTangents[i].x;
|
||||
current.y = mesh->mTangents[i].y;
|
||||
current.z = mesh->mTangents[i].z;
|
||||
vertex.tangent = current;
|
||||
}
|
||||
|
||||
if (mesh->mBitangents)
|
||||
{
|
||||
current.x = mesh->mBitangents[i].x;
|
||||
current.y = mesh->mBitangents[i].y;
|
||||
current.z = mesh->mBitangents[i].z;
|
||||
vertex.bitangent = current;
|
||||
}
|
||||
|
||||
vertex.uv = glm::vec2(0.0f, 0.0f);
|
||||
|
||||
// Does it contain UVs?
|
||||
if (mesh->mTextureCoords[0])
|
||||
{
|
||||
float u = mesh->mTextureCoords[0][i].x;
|
||||
float v = mesh->mTextureCoords[0][i].y;
|
||||
vertex.uv = Vector2(u, v);
|
||||
}
|
||||
|
||||
vertices.push_back(vertex);
|
||||
}
|
||||
|
||||
return vertices;
|
||||
}
|
||||
|
||||
std::vector<uint32_t> ModelLoader::ProcessIndices(aiMesh* mesh)
|
||||
{
|
||||
auto indices = std::vector<uint32_t>();
|
||||
for (uint32_t i = 0; i < mesh->mNumFaces; i++)
|
||||
{
|
||||
aiFace face = mesh->mFaces[i];
|
||||
for (uint32_t j = 0; j < face.mNumIndices; j++)
|
||||
{
|
||||
indices.push_back(face.mIndices[j]);
|
||||
}
|
||||
}
|
||||
|
||||
return indices;
|
||||
}
|
||||
|
||||
Ref<Material> ModelLoader::ProcessMaterials(const aiScene* scene, aiMesh* mesh)
|
||||
{
|
||||
if (mesh->mMaterialIndex < 0)
|
||||
return nullptr;
|
||||
|
||||
aiMaterial* materialNode = scene->mMaterials[mesh->mMaterialIndex];
|
||||
|
||||
Ref<Material> material = CreateRef<Material>();
|
||||
|
||||
aiString str;
|
||||
if (materialNode->GetTextureCount(aiTextureType_DIFFUSE) > 0)
|
||||
{
|
||||
materialNode->GetTexture(aiTextureType_DIFFUSE, 0, &str);
|
||||
Ref<Texture> albedoTexture = ProcessTextures(scene, str.C_Str());
|
||||
material->SetAlbedo(albedoTexture);
|
||||
}
|
||||
|
||||
if (materialNode->GetTextureCount(aiTextureType_NORMALS) > 0)
|
||||
{
|
||||
materialNode->GetTexture(aiTextureType_NORMALS, 0, &str);
|
||||
Ref<Texture> albedoTexture = ProcessTextures(scene, str.C_Str());
|
||||
material->SetNormal(albedoTexture);
|
||||
}
|
||||
|
||||
if (materialNode->GetTextureCount(aiTextureType_METALNESS) > 0)
|
||||
{
|
||||
materialNode->GetTexture(aiTextureType_METALNESS, 0, &str);
|
||||
Ref<Texture> albedoTexture = ProcessTextures(scene, str.C_Str());
|
||||
material->SetMetalness(albedoTexture);
|
||||
}
|
||||
|
||||
if (materialNode->GetTextureCount(aiTextureType_AMBIENT_OCCLUSION) > 0)
|
||||
{
|
||||
materialNode->GetTexture(aiTextureType_AMBIENT_OCCLUSION, 0, &str);
|
||||
Ref<Texture> albedoTexture = ProcessTextures(scene, str.C_Str());
|
||||
material->SetAO(albedoTexture);
|
||||
}
|
||||
|
||||
if (materialNode->GetTextureCount(aiTextureType_DIFFUSE_ROUGHNESS) > 0)
|
||||
{
|
||||
materialNode->GetTexture(aiTextureType_DIFFUSE_ROUGHNESS, 0, &str);
|
||||
Ref<Texture> albedoTexture = ProcessTextures(scene, str.C_Str());
|
||||
material->SetRoughness(albedoTexture);
|
||||
}
|
||||
|
||||
if (materialNode->GetTextureCount(aiTextureType_DISPLACEMENT) > 0)
|
||||
{
|
||||
materialNode->GetTexture(aiTextureType_DISPLACEMENT, 0, &str);
|
||||
Ref<Texture> albedoTexture = ProcessTextures(scene, str.C_Str());
|
||||
material->SetDisplacement(albedoTexture);
|
||||
}
|
||||
|
||||
return material;
|
||||
}
|
||||
|
||||
Ref<Texture> ModelLoader::ProcessTextures(const aiScene* scene, const std::string& path)
|
||||
{
|
||||
// Load embedded textures, a texture is embedded is the path
|
||||
// starts with a '*'.
|
||||
if (String::BeginsWith(path, "*"))
|
||||
{
|
||||
uint32_t textureIndex = std::atoi(String::Split(path, '*')[1].c_str());
|
||||
const aiTexture* aitexture = scene->GetEmbeddedTexture(path.c_str());
|
||||
|
||||
Vector2 textureSize = Vector2(aitexture->mWidth, aitexture->mHeight);
|
||||
auto texture = CreateRef<Texture>(textureSize, (unsigned char*)aitexture->pcData, textureSize.x);
|
||||
return texture;
|
||||
}
|
||||
|
||||
std::string texturePath = FileSystem::Root + modelDir + path;
|
||||
if (!FileSystem::FileExists(texturePath))
|
||||
{
|
||||
std::string textureNotFoundmsg = "Texture file couldn't be found: " + texturePath;
|
||||
Logger::Log(textureNotFoundmsg, Nuake::LOG_TYPE::WARNING);
|
||||
|
||||
texturePath = "resources/Textures/default/Default.png";
|
||||
}
|
||||
|
||||
return TextureManager::Get()->GetTexture(texturePath);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user