nyahh
parent
9147a85dc3
commit
60ec2051d8
|
|
@ -0,0 +1,146 @@
|
||||||
|
/*
|
||||||
|
* <Short Description>
|
||||||
|
* Copyright (C) 2024 Brett Terpstra
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef BLT_WITH_GRAPHICS_OBJ_LOADER_H
|
||||||
|
#define BLT_WITH_GRAPHICS_OBJ_LOADER_H
|
||||||
|
|
||||||
|
#include "blt/math/vectors.h"
|
||||||
|
#include "blt/std/hashmap.h"
|
||||||
|
#include <utility>
|
||||||
|
#include <vector>
|
||||||
|
#include <string_view>
|
||||||
|
|
||||||
|
namespace blt::gfx
|
||||||
|
{
|
||||||
|
|
||||||
|
typedef blt::vec3f vertex_t;
|
||||||
|
typedef blt::vec2f uv_t;
|
||||||
|
typedef blt::vec3f normal_t;
|
||||||
|
|
||||||
|
class model_data
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
private:
|
||||||
|
std::vector<vertex_t> vertices;
|
||||||
|
std::vector<uv_t> uvs;
|
||||||
|
std::vector<normal_t> normals;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct face_t
|
||||||
|
{
|
||||||
|
std::int32_t vertex, uv, normal;
|
||||||
|
};
|
||||||
|
|
||||||
|
static inline bool operator==(const face_t& f1, const face_t& f2)
|
||||||
|
{
|
||||||
|
return f1.vertex == f2.vertex && f1.uv == f2.uv && f1.normal == f2.normal;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct face_hash
|
||||||
|
{
|
||||||
|
size_t operator()(const face_t& face) const
|
||||||
|
{
|
||||||
|
std::hash<std::int32_t> hasher;
|
||||||
|
return hasher(face.vertex) ^ hasher(face.uv) ^ hasher(face.normal);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct face_eq
|
||||||
|
{
|
||||||
|
bool operator()(const face_t& f1, const face_t& f2) const
|
||||||
|
{
|
||||||
|
return f1 == f2;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct constructed_vertex_t
|
||||||
|
{
|
||||||
|
vertex_t vertex;
|
||||||
|
uv_t uv;
|
||||||
|
normal_t normal;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct triangle_t
|
||||||
|
{
|
||||||
|
std::int32_t v[3];
|
||||||
|
};
|
||||||
|
|
||||||
|
struct quad_t
|
||||||
|
{
|
||||||
|
std::int32_t v[4];
|
||||||
|
};
|
||||||
|
|
||||||
|
struct object_data
|
||||||
|
{
|
||||||
|
std::string object_name;
|
||||||
|
std::vector<triangle_t> indices;
|
||||||
|
};
|
||||||
|
|
||||||
|
class obj_objects_t
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
std::vector<constructed_vertex_t> vertex_data_;
|
||||||
|
std::vector<object_data> objects_;
|
||||||
|
public:
|
||||||
|
obj_objects_t(std::vector<constructed_vertex_t>&& vertex_data, std::vector<object_data>&& objects):
|
||||||
|
vertex_data_(vertex_data), objects_(objects)
|
||||||
|
{}
|
||||||
|
|
||||||
|
inline const std::vector<constructed_vertex_t>& vertex_data()
|
||||||
|
{
|
||||||
|
return vertex_data_;
|
||||||
|
};
|
||||||
|
|
||||||
|
inline const std::vector<object_data>& objects()
|
||||||
|
{
|
||||||
|
return objects_;
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
class char_tokenizer;
|
||||||
|
|
||||||
|
class obj_loader
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
std::vector<vertex_t> vertices;
|
||||||
|
std::vector<uv_t> uvs;
|
||||||
|
std::vector<normal_t> normals;
|
||||||
|
|
||||||
|
// maps between face (constructed vertex) -> vertex indices
|
||||||
|
HASHMAP<face_t, std::int32_t, face_hash, face_eq> vertex_map;
|
||||||
|
std::vector<constructed_vertex_t> vertex_data;
|
||||||
|
object_data current_object;
|
||||||
|
std::vector<object_data> data;
|
||||||
|
private:
|
||||||
|
bool handle_vertex_and_normals(float x, float y, float z, char type);
|
||||||
|
|
||||||
|
void parse_vertex_line(char_tokenizer& tokenizer);
|
||||||
|
|
||||||
|
void parse_face(char_tokenizer& tokenizer);
|
||||||
|
|
||||||
|
void handle_face_vertex(const std::vector<std::string>& face_list, std::int32_t* arr);
|
||||||
|
|
||||||
|
public:
|
||||||
|
obj_objects_t parseFile(std::string_view file);
|
||||||
|
};
|
||||||
|
|
||||||
|
obj_objects_t quick_load(std::string_view file);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif //BLT_WITH_GRAPHICS_OBJ_LOADER_H
|
||||||
|
|
@ -0,0 +1,219 @@
|
||||||
|
/*
|
||||||
|
* <Short Description>
|
||||||
|
* Copyright (C) 2024 Brett Terpstra
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#include <blt/parse/obj_loader.h>
|
||||||
|
#include <blt/std/loader.h>
|
||||||
|
#include <blt/std/string.h>
|
||||||
|
#include <blt/std/logging.h>
|
||||||
|
#include <cctype>
|
||||||
|
#include <charconv>
|
||||||
|
#include "blt/std/assert.h"
|
||||||
|
#include "blt/std/utility.h"
|
||||||
|
|
||||||
|
namespace blt::gfx
|
||||||
|
{
|
||||||
|
class char_tokenizer
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
std::string_view string;
|
||||||
|
std::size_t current_pos = 0;
|
||||||
|
public:
|
||||||
|
explicit char_tokenizer(std::string_view view): string(view)
|
||||||
|
{}
|
||||||
|
|
||||||
|
inline char advance()
|
||||||
|
{
|
||||||
|
return string[current_pos++];
|
||||||
|
}
|
||||||
|
|
||||||
|
inline bool has_next(size_t offset = 0)
|
||||||
|
{
|
||||||
|
return current_pos + offset < string.size();
|
||||||
|
}
|
||||||
|
|
||||||
|
inline std::string_view read_fully()
|
||||||
|
{
|
||||||
|
return blt::string::trim(string.substr(current_pos));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T = float>
|
||||||
|
T get(std::string_view str)
|
||||||
|
{
|
||||||
|
T x;
|
||||||
|
const auto [ptr, ec] = std::from_chars(str.data(), str.data() + str.size(), x);
|
||||||
|
// probably not needed.
|
||||||
|
if (ec != std::errc())
|
||||||
|
{
|
||||||
|
// int i;
|
||||||
|
// const auto [ptr2, ec2] = std::from_chars(str.data(), str.data() + str.size(), i);
|
||||||
|
// if (ec2 == std::errc())
|
||||||
|
// {
|
||||||
|
// x = static_cast<float>(i);
|
||||||
|
// } else
|
||||||
|
// {
|
||||||
|
BLT_WARN("Unable to parse string '%s' into number!", std::string(str).c_str());
|
||||||
|
x = 0;
|
||||||
|
// }
|
||||||
|
}
|
||||||
|
return x;
|
||||||
|
}
|
||||||
|
|
||||||
|
void obj_loader::parse_vertex_line(char_tokenizer& tokenizer)
|
||||||
|
{
|
||||||
|
char type = tokenizer.advance();
|
||||||
|
|
||||||
|
if (type == 'p')
|
||||||
|
{
|
||||||
|
BLT_WARN("Unexpected type '%c' (not supported)", type);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto elements = blt::string::split(std::string(tokenizer.read_fully()), " ");
|
||||||
|
BLT_ASSERT(elements.size() >= 2 && "Current line doesn't have enough arguments to process!");
|
||||||
|
float x = get(elements[0]), y = get(elements[1]);
|
||||||
|
if (elements.size() < 3)
|
||||||
|
{
|
||||||
|
if (type == 't')
|
||||||
|
uvs.push_back(uv_t{x, y});
|
||||||
|
else
|
||||||
|
BLT_ERROR("Unable to parse line '%s' type '%c' not recognized for arg count", std::string(tokenizer.read_fully()).c_str(), type);
|
||||||
|
} else
|
||||||
|
{
|
||||||
|
float z = get(elements[2]);
|
||||||
|
if (!handle_vertex_and_normals(x, y, z, type))
|
||||||
|
BLT_ERROR("Unable to parse line '%s' type '%c' not recognized", std::string(tokenizer.read_fully()).c_str(), type);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool obj_loader::handle_vertex_and_normals(float x, float y, float z, char type)
|
||||||
|
{
|
||||||
|
if (std::isspace(type))
|
||||||
|
{
|
||||||
|
vertices.push_back(vertex_t{x, y, z});
|
||||||
|
} else if (type == 'n')
|
||||||
|
{
|
||||||
|
normals.push_back(normal_t{x, y, z});
|
||||||
|
} else
|
||||||
|
return false;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
obj_objects_t quick_load(std::string_view file)
|
||||||
|
{
|
||||||
|
return obj_loader().parseFile(file);
|
||||||
|
}
|
||||||
|
|
||||||
|
obj_objects_t obj_loader::parseFile(std::string_view file)
|
||||||
|
{
|
||||||
|
auto lines = blt::fs::getLinesFromFile(std::string(file));
|
||||||
|
for (const auto& line : lines)
|
||||||
|
{
|
||||||
|
char_tokenizer token(line);
|
||||||
|
if (!token.has_next() || token.read_fully().empty())
|
||||||
|
continue;
|
||||||
|
switch (token.advance())
|
||||||
|
{
|
||||||
|
case '#':
|
||||||
|
continue;
|
||||||
|
case 'f':
|
||||||
|
parse_face(token);
|
||||||
|
break;
|
||||||
|
case 'v':
|
||||||
|
parse_vertex_line(token);
|
||||||
|
break;
|
||||||
|
case 'o':
|
||||||
|
{
|
||||||
|
if (!current_object.indices.empty())
|
||||||
|
data.push_back(current_object);
|
||||||
|
current_object = {};
|
||||||
|
current_object.object_name = token.read_fully();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case 'm':
|
||||||
|
{
|
||||||
|
BLT_TRACE("Material '%s' needs to be loaded!", std::string(token.read_fully()).c_str());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case 'u':
|
||||||
|
{
|
||||||
|
BLT_TRACE("Using material '%s'", std::string(token.read_fully()).c_str());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case 's':
|
||||||
|
BLT_TRACE("Using shading: %s", std::string(token.read_fully()).c_str());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data.push_back(current_object);
|
||||||
|
return {std::move(vertex_data), std::move(data)};
|
||||||
|
}
|
||||||
|
|
||||||
|
void obj_loader::parse_face(char_tokenizer& tokenizer)
|
||||||
|
{
|
||||||
|
auto faces = blt::string::split(std::string(tokenizer.read_fully()), ' ');
|
||||||
|
if (faces.size() == 3)
|
||||||
|
{
|
||||||
|
triangle_t triangle{};
|
||||||
|
handle_face_vertex(faces, triangle.v);
|
||||||
|
current_object.indices.push_back(triangle);
|
||||||
|
} else if (faces.size() == 4)
|
||||||
|
{
|
||||||
|
quad_t quad{};
|
||||||
|
handle_face_vertex(faces, quad.v);
|
||||||
|
triangle_t t1{};
|
||||||
|
triangle_t t2{};
|
||||||
|
|
||||||
|
for (int i = 0; i < 3; i++)
|
||||||
|
t1.v[i] = quad.v[i];
|
||||||
|
t2.v[0] = quad.v[0];
|
||||||
|
t2.v[1] = quad.v[2];
|
||||||
|
t2.v[2] = quad.v[3];
|
||||||
|
|
||||||
|
current_object.indices.push_back(t1);
|
||||||
|
current_object.indices.push_back(t2);
|
||||||
|
} else
|
||||||
|
BLT_WARN("Unsupported vertex count! %d", faces.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
void obj_loader::handle_face_vertex(const std::vector<std::string>& face_list, int32_t* arr)
|
||||||
|
{
|
||||||
|
for (const auto& pair : blt::enumerate(face_list))
|
||||||
|
{
|
||||||
|
auto indices = blt::string::split(pair.second, '/');
|
||||||
|
BLT_ASSERT(indices.size() == 3 && "Must have vertex, uv, and normal indices!!");
|
||||||
|
|
||||||
|
auto vi = get<std::int32_t>(indices[0]);
|
||||||
|
auto ui = get<std::int32_t>(indices[1]);
|
||||||
|
auto ni = get<std::int32_t>(indices[2]);
|
||||||
|
|
||||||
|
face_t face{vi, ui, ni};
|
||||||
|
|
||||||
|
auto loc = vertex_map.find(face);
|
||||||
|
if (loc == vertex_map.end())
|
||||||
|
{
|
||||||
|
auto index = static_cast<std::int32_t>(vertex_data.size());
|
||||||
|
vertex_data.push_back({vertices[vi], uvs[ui], normals[ni]});
|
||||||
|
vertex_map.insert({face, index});
|
||||||
|
arr[pair.first] = index;
|
||||||
|
} else
|
||||||
|
{
|
||||||
|
arr[pair.first] = loc->second;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue