xiaoqi
/
ModelCompiler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compile mesh bones and weights into binary 

Rig structure still WIP
This commit is contained in:
Xiao Qi 2022-11-12 15:55:41 +08:00
parent 310765f4f6
commit 66baaf374d
3 changed files with 65 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
src/Libraries/MeshCompiler.cpp
View File

@ -74,10 +74,27 @@ namespace SH_COMP
meshData.vertexTangent.reserve(mesh.mNumVertices);
meshData.texCoords.reserve(mesh.mNumVertices);
meshData.bonesInfo.resize(mesh.mNumBones);
meshData.bones.resize(mesh.mNumBones);
for (auto i {0}; i < mesh.mNumBones; ++i)
{
auto& bone = *mesh.mBones[i];
std::cout << "bone" << std::endl;
auto const& bone = *mesh.mBones[i];
auto& newBone = meshData.bones[i];
auto& newBoneInfo = meshData.bonesInfo[i];
newBone.name = bone.mName.C_Str();
newBoneInfo.charCount = newBone.name.length();
std::memcpy(&newBone.offset, &bone.mOffsetMatrix, sizeof(SHMat4));
newBone.weights.resize(bone.mNumWeights);
for (auto j {0}; j < bone.mNumWeights; ++j)
{
newBone.weights[j].index = bone.mWeights[j].mVertexId;
newBone.weights[j].weight = bone.mWeights[j].mWeight;
}
newBoneInfo.weightCount = bone.mNumWeights;
}
for (size_t i{ 0 }; i < mesh.mNumVertices; ++i)
@ -129,7 +146,7 @@ namespace SH_COMP
}
}
void MeshCompiler::BuildHeaders(MeshAsset& asset) noexcept
void MeshCompiler::BuildHeaders(ModelAsset& asset) noexcept
{
for (auto const& mesh : asset.meshes)
{
@ -139,6 +156,7 @@ namespace SH_COMP
head.charCount = mesh.name.size();
head.indexCount = mesh.indices.size();
head.vertexCount = mesh.vertexPosition.size();
head.boneCount = mesh.bonesInfo.size();
asset.header.meshCount++;
}
@ -188,18 +206,18 @@ namespace SH_COMP
);
}
void MeshCompiler::LoadFromFile(AssetPath path, MeshAsset& asset) noexcept
void MeshCompiler::LoadFromFile(AssetPath path, ModelAsset& asset) noexcept
{
const aiScene* scene = aiImporter.ReadFile(path.string().c_str(),0
//aiProcess_Triangulate // Make sure we get triangles rather than nvert polygons
//| aiProcess_GenUVCoords // Convert any type of mapping to uv mapping
//| aiProcess_TransformUVCoords // preprocess UV transformations (scaling, translation ...)
//| aiProcess_FindInstances // search for instanced meshes and remove them by references to one master
//| aiProcess_CalcTangentSpace // calculate tangents and bitangents if possible
//| aiProcess_JoinIdenticalVertices // join identical vertices/ optimize indexing
//| aiProcess_FindInvalidData // detect invalid model data, such as invalid normal vectors
//| aiProcess_FlipUVs // flip the V to match the Vulkans way of doing UVs
//| aiProcess_ValidateDataStructure
const aiScene* scene = aiImporter.ReadFile(path.string().c_str(),
aiProcess_Triangulate // Make sure we get triangles rather than nvert polygons
| aiProcess_GenUVCoords // Convert any type of mapping to uv mapping
| aiProcess_TransformUVCoords // preprocess UV transformations (scaling, translation ...)
| aiProcess_FindInstances // search for instanced meshes and remove them by references to one master
| aiProcess_CalcTangentSpace // calculate tangents and bitangents if possible
| aiProcess_JoinIdenticalVertices // join identical vertices/ optimize indexing
| aiProcess_FindInvalidData // detect invalid model data, such as invalid normal vectors
| aiProcess_FlipUVs // flip the V to match the Vulkans way of doing UVs
| aiProcess_ValidateDataStructure
);
if (!scene || !scene->HasMeshes())
@ -215,7 +233,7 @@ namespace SH_COMP
aiImporter.FreeScene();
}
void MeshCompiler::CompileMeshBinary(AssetPath path, MeshAsset const& asset) noexcept
void MeshCompiler::CompileMeshBinary(AssetPath path, ModelAsset const& asset) noexcept
{
std::string newPath{ path.string().substr(0, path.string().find_last_of('.')) };
newPath += MODEL_EXTENSION;
@ -254,7 +272,7 @@ namespace SH_COMP
{
RigNode* current = new RigNode();
current->name = source.mName.C_Str();
std::memcpy(&current->transform, &source.mTransformation, sizeof(float) * 16);
std::memcpy(&current->transform, &source.mTransformation, sizeof(SHMat4));
for (auto i {0}; i < source.mNumChildren; ++i)
{
@ -269,7 +287,7 @@ namespace SH_COMP
void MeshCompiler::LoadAndCompile(AssetPath path) noexcept
{
auto const asset = new MeshAsset();
auto const asset = new ModelAsset();
asset->rig.root = nullptr;

8
src/Libraries/MeshCompiler.h
View File

@ -33,13 +33,13 @@ namespace SH_COMP
static void ProcessNode(aiNode const& node, aiScene const& scene, MeshVectorRef meshes, RigNode*& root) noexcept;
static void ExtractAnimations(aiScene const& scene, AnimVectorRef anims) noexcept;
static void GetMesh(aiMesh const& mesh, MeshData& meshData) noexcept;
static void BuildHeaders(MeshAsset& asset) noexcept;
static void BuildHeaders(ModelAsset& asset) noexcept;
static void WriteMeshHeader(std::ofstream& file, MeshDataHeader const& header);
static void WriteMeshData(std::ofstream& file, MeshDataHeader const& header, MeshData const& data);
static void WriteMeshData(std::ofstream& file, MeshDataHeader const& header, MeshData const& asset);
static void LoadFromFile(AssetPath path, MeshAsset& asset) noexcept;
static void CompileMeshBinary(AssetPath path, MeshAsset const& asset) noexcept;
static void LoadFromFile(AssetPath path, ModelAsset& asset) noexcept;
static void CompileMeshBinary(AssetPath path, ModelAsset const& asset) noexcept;
static void BuildArmature(aiNode const& node, RigNode*& root) noexcept;
static void CopyNode(aiNode const& source, RigNode* parent) noexcept;

30
src/Types/MeshAsset.h
View File

@ -38,6 +38,26 @@ namespace SH_COMP
uint32_t vertexCount;
uint32_t indexCount;
uint32_t charCount;
uint32_t boneCount;
};
struct MeshBoneInfo
{
uint32_t charCount;
uint32_t weightCount; // Size should be same as boneCount
};
struct BoneWeight
{
uint32_t index;
float weight;
};
struct MeshBone
{
std::string name;
SHMat4 offset;
std::vector<BoneWeight> weights;
};
struct MeshData
@ -48,11 +68,14 @@ namespace SH_COMP
std::vector<SHVec3> vertexNormal;
std::vector<SHVec2> texCoords;
std::vector<uint32_t> indices;
std::vector<MeshBoneInfo> bonesInfo;
std::vector<MeshBone> bones;
};
struct RigNode
{
std::string name;
uint32_t id;
SHMat4 transform;
std::vector<RigNode*> children;
};
@ -62,15 +85,14 @@ namespace SH_COMP
RigNode* root;
};
struct MeshAssetHeader
struct ModelAssetHeader
{
size_t meshCount;
};
struct MeshAsset
struct ModelAsset
{
MeshAssetHeader header;
RigData rig;
ModelAssetHeader header;
std::vector<MeshDataHeader> headers;
std::vector<MeshData> meshes;
};