obj loader refactor
This commit is contained in:
h4570
@@ -0,0 +1,207 @@
|
||||
/*
|
||||
# ______ ____ ___
|
||||
# | \/ ____| |___|
|
||||
# | | | \ | |
|
||||
#-----------------------------------------------------------------------
|
||||
# Copyright 2022, tyra - https://github.com/h4570/tyra
|
||||
# Licenced under Apache License 2.0
|
||||
# Sandro Sobczyński <sandro.sobczynski@gmail.com>
|
||||
*/
|
||||
|
||||
#include <cstring>
|
||||
#include <stdio.h>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "math/vec2.hpp"
|
||||
#include "math/vec4.hpp"
|
||||
#include "renderer/models/color.hpp"
|
||||
#include "debug/debug.hpp"
|
||||
#include "loaders/3d/obj_loader/obj_loader.hpp"
|
||||
|
||||
#define TINYOBJLOADER_USE_MAPBOX_EARCUT
|
||||
#define TINYOBJLOADER_IMPLEMENTATION
|
||||
#include "loaders/3d/obj_loader/tiny_obj_loader.hpp"
|
||||
|
||||
namespace Tyra {
|
||||
|
||||
ObjLoader::ObjLoader() {}
|
||||
|
||||
ObjLoader::~ObjLoader() {}
|
||||
|
||||
MeshBuilder2Data* ObjLoader::load(const char* fullpath, const u16& count,
|
||||
const float& scale, const bool& invertY) {
|
||||
std::string path = fullpath;
|
||||
std::string basePath = getPathFromFilename(path);
|
||||
|
||||
TYRA_ASSERT(!path.empty(), "Provided path is empty!");
|
||||
|
||||
auto rawFilename = getFilenameWithoutExtension(path);
|
||||
auto extension = getExtensionOfFilename(path);
|
||||
|
||||
auto* result = new MeshBuilder2Data();
|
||||
|
||||
tinyobj::ObjReaderConfig readerConfig;
|
||||
readerConfig.triangulate = count == 1;
|
||||
readerConfig.triangulation_method = count == 1 ? "earcut" : "simple";
|
||||
readerConfig.mtl_search_path = basePath;
|
||||
|
||||
for (u16 i = 1; i <= count; i++) {
|
||||
std::string filePath = rawFilename + std::to_string(i) + "." + extension;
|
||||
if (count == 1) filePath = path;
|
||||
|
||||
tinyobj::ObjReader reader;
|
||||
|
||||
if (!reader.ParseFromFile(filePath, readerConfig)) {
|
||||
if (!reader.Error().empty()) {
|
||||
TYRA_TRAP("TinyObjLoader: ", reader.Error());
|
||||
}
|
||||
TYRA_TRAP("Unknown TinyObjLoader error!");
|
||||
}
|
||||
|
||||
if (!reader.Warning().empty()) {
|
||||
TYRA_WARN("TinyObjReader: ", reader.Warning());
|
||||
}
|
||||
|
||||
auto& attrib = reader.GetAttrib();
|
||||
auto& shapes = reader.GetShapes();
|
||||
auto& materials = reader.GetMaterials();
|
||||
|
||||
TYRA_ASSERT(
|
||||
materials.size() > 0,
|
||||
"No material data found! Please add .mtl file(s) and assign them via "
|
||||
"mtlib in obj file");
|
||||
|
||||
if (i == 1) {
|
||||
setInitialData(result, attrib, shapes, materials, count);
|
||||
}
|
||||
|
||||
importFrame(result, attrib, shapes, materials, i - 1, scale, invertY,
|
||||
count);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
void ObjLoader::setInitialData(
|
||||
MeshBuilder2Data* output, const tinyobj::attrib_t& attrib,
|
||||
const std::vector<tinyobj::shape_t>& shapes,
|
||||
const std::vector<tinyobj::material_t>& materials, const u16& framesCount) {
|
||||
if (attrib.texcoords.size()) output->textureCoordsEnabled = true;
|
||||
|
||||
if (attrib.normals.size()) output->normalsEnabled = true;
|
||||
|
||||
TYRA_ASSERT(materials.size() > 0,
|
||||
"No material data found! Please add .mtl "
|
||||
"file(s) and assign them via mtlib in obj "
|
||||
"file");
|
||||
|
||||
for (size_t i = 0; i < materials.size(); i++) {
|
||||
auto* material = new MeshBuilder2MaterialData();
|
||||
|
||||
material->name = materials[i].name;
|
||||
|
||||
material->ambient.set(materials[i].ambient[0] * 255.0F,
|
||||
materials[i].ambient[1] * 255.0F,
|
||||
materials[i].ambient[2] * 255.0F, 128.0F);
|
||||
|
||||
for (size_t j = 0; j < framesCount; j++) {
|
||||
auto* frame = new MeshBuilder2MaterialFrameData();
|
||||
material->frames.push_back(frame);
|
||||
}
|
||||
|
||||
output->materials.push_back(material);
|
||||
}
|
||||
}
|
||||
|
||||
void ObjLoader::importFrame(MeshBuilder2Data* output,
|
||||
const tinyobj::attrib_t& attrib,
|
||||
const std::vector<tinyobj::shape_t>& shapes,
|
||||
const std::vector<tinyobj::material_t>& materials,
|
||||
const u16& frameIndex, const float& scale,
|
||||
const bool& invertY, const u16& count) {
|
||||
for (size_t i = 0; i < shapes.size(); i++) {
|
||||
const auto& mesh = shapes[i].mesh;
|
||||
|
||||
// Loop over shapes
|
||||
for (size_t s = 0; s < shapes.size(); s++) {
|
||||
size_t index_offset = 0;
|
||||
// Loop over faces
|
||||
for (size_t f = 0; f < mesh.num_face_vertices.size(); f++) {
|
||||
auto materialId = mesh.material_ids[f];
|
||||
auto* outFrame = output->materials[materialId]->frames[frameIndex];
|
||||
|
||||
auto isAllocated = outFrame->vertices != nullptr;
|
||||
|
||||
if (!isAllocated) {
|
||||
auto vertCount = mesh.num_face_vertices.size() * 3;
|
||||
|
||||
outFrame->count = vertCount;
|
||||
outFrame->vertices = new Vec4[vertCount];
|
||||
|
||||
if (output->textureCoordsEnabled)
|
||||
outFrame->textureCoords = new Vec4[vertCount];
|
||||
|
||||
if (output->normalsEnabled) outFrame->normals = new Vec4[vertCount];
|
||||
} else {
|
||||
TYRA_ASSERT(outFrame->count == mesh.num_face_vertices.size() * 3,
|
||||
"Multiple usage of the same \"usemtl\" is not supported! "
|
||||
"Please merge them!");
|
||||
}
|
||||
|
||||
size_t fv = size_t(mesh.num_face_vertices[f]);
|
||||
|
||||
TYRA_ASSERT(count == 1 || fv == 3,
|
||||
"Please triangulate obj files if you are animating!",
|
||||
"Recommended Blender options: ",
|
||||
"- Obj exporting: triangulate off, keep vertex order",
|
||||
"- Object modifiers: triangulate (as first!), then other "
|
||||
"modifiers");
|
||||
|
||||
// Loop over vertices in the face.
|
||||
for (size_t v = 0; v < fv; v++) {
|
||||
// access to vertex
|
||||
tinyobj::index_t idx = mesh.indices[index_offset + v];
|
||||
tinyobj::real_t vx =
|
||||
attrib.vertices[3 * size_t(idx.vertex_index) + 0];
|
||||
tinyobj::real_t vy =
|
||||
attrib.vertices[3 * size_t(idx.vertex_index) + 1];
|
||||
tinyobj::real_t vz =
|
||||
attrib.vertices[3 * size_t(idx.vertex_index) + 2];
|
||||
|
||||
outFrame->vertices[index_offset + v].set(vx, vy, vz, 1.0F);
|
||||
outFrame->vertices[index_offset + v] *= scale;
|
||||
|
||||
// Check if `normal_index` is zero or positive. negative = no normal
|
||||
// data
|
||||
if (idx.normal_index >= 0) {
|
||||
tinyobj::real_t nx =
|
||||
attrib.normals[3 * size_t(idx.normal_index) + 0];
|
||||
tinyobj::real_t ny =
|
||||
attrib.normals[3 * size_t(idx.normal_index) + 1];
|
||||
tinyobj::real_t nz =
|
||||
attrib.normals[3 * size_t(idx.normal_index) + 2];
|
||||
|
||||
outFrame->normals[index_offset + v].set(nx, ny, nz, 1.0F);
|
||||
}
|
||||
|
||||
// Check if `texcoord_index` is zero or positive. negative =
|
||||
// notexcoord data
|
||||
if (idx.texcoord_index >= 0) {
|
||||
tinyobj::real_t tx =
|
||||
attrib.texcoords[2 * size_t(idx.texcoord_index) + 0];
|
||||
tinyobj::real_t ty =
|
||||
attrib.texcoords[2 * size_t(idx.texcoord_index) + 1];
|
||||
|
||||
auto finalY = invertY ? 1.0F - ty : ty;
|
||||
outFrame->textureCoords[index_offset + v].set(tx, finalY, 1.0F,
|
||||
0.0F);
|
||||
}
|
||||
}
|
||||
|
||||
index_offset += fv;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace Tyra
|
||||
Reference in New Issue
Block a user