v01
This commit is contained in:
Ivan
254
thirdparty/Pangolin/src/geometry/geometry_obj.cpp
vendored
Normal file
254
thirdparty/Pangolin/src/geometry/geometry_obj.cpp
vendored
Normal file
@@ -0,0 +1,254 @@
|
||||
/* This file is part of the Pangolin Project.
|
||||
* http://github.com/stevenlovegrove/Pangolin
|
||||
*
|
||||
* Copyright (c) 2011 Steven Lovegrove
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person
|
||||
* obtaining a copy of this software and associated documentation
|
||||
* files (the "Software"), to deal in the Software without
|
||||
* restriction, including without limitation the rights to use,
|
||||
* copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the
|
||||
* Software is furnished to do so, subject to the following
|
||||
* conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be
|
||||
* included in all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||||
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
||||
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||||
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
||||
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
||||
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
||||
* OTHER DEALINGS IN THE SOFTWARE.
|
||||
*/
|
||||
|
||||
#include <unordered_set>
|
||||
|
||||
#include <pangolin/geometry/geometry_obj.h>
|
||||
#include <tinyobj/tiny_obj_loader.h>
|
||||
#include <pangolin/utils/variadic_all.h>
|
||||
#include <pangolin/utils/simple_math.h>
|
||||
|
||||
// Needed for enum type. TODO: Make this independent of GL
|
||||
#include <pangolin/gl/glplatform.h>
|
||||
#include <pangolin/image/image_io.h>
|
||||
#include <pangolin/utils/file_utils.h>
|
||||
|
||||
namespace std {
|
||||
|
||||
template<>
|
||||
struct hash<tinyobj::index_t> {
|
||||
|
||||
std::size_t operator()(const tinyobj::index_t & t) const noexcept {
|
||||
static std::hash<int> h;
|
||||
return h(t.vertex_index) ^ h(t.normal_index) ^ h(t.texcoord_index);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
} // namespace std
|
||||
|
||||
namespace tinyobj {
|
||||
|
||||
bool operator==(const index_t a, const index_t b) {
|
||||
return a.vertex_index == b.vertex_index && a.normal_index == b.normal_index && a.texcoord_index == b.texcoord_index;
|
||||
}
|
||||
|
||||
} // namespace tinyobj
|
||||
|
||||
namespace pangolin {
|
||||
|
||||
template<typename T>
|
||||
Geometry::Element ConvertVectorToGeometryElement(const std::vector<T>& v, size_t count_per_element, const std::string& attribute_name )
|
||||
{
|
||||
PANGO_ASSERT(v.size() % count_per_element == 0);
|
||||
const size_t num_elements = v.size() / count_per_element;
|
||||
|
||||
// Allocate buffer and copy into it
|
||||
Geometry::Element el(sizeof(T) * count_per_element, num_elements);
|
||||
el.CopyFrom(Image<T>(v.data(), count_per_element, num_elements));
|
||||
el.attributes[attribute_name] = el.template UnsafeReinterpret<T>();
|
||||
return el;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
Image<T> GetImageWrapper(std::vector<T>& vec, size_t count_per_element)
|
||||
{
|
||||
PANGO_ASSERT(vec.size() % count_per_element == 0);
|
||||
if(vec.size()) {
|
||||
return Image<T>(vec.data(), count_per_element, vec.size() / count_per_element, count_per_element * sizeof(T));
|
||||
}else{
|
||||
return Image<T>();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
pangolin::Geometry LoadGeometryObj(const std::string& filename)
|
||||
{
|
||||
pangolin::Geometry geom;
|
||||
|
||||
tinyobj::attrib_t attrib;
|
||||
std::vector<tinyobj::shape_t> shapes;
|
||||
std::vector<tinyobj::material_t> materials;
|
||||
|
||||
std::string warn;
|
||||
std::string err;
|
||||
if(tinyobj::LoadObj(&attrib, &shapes, &materials, &warn, &err, filename.c_str(), PathParent(filename).c_str())) {
|
||||
PANGO_ASSERT(attrib.vertices.size() % 3 == 0);
|
||||
PANGO_ASSERT(attrib.normals.size() % 3 == 0);
|
||||
PANGO_ASSERT(attrib.colors.size() % 3 == 0);
|
||||
PANGO_ASSERT(attrib.texcoords.size() % 2 == 0);
|
||||
|
||||
// Load textures - a bit of a hack for now.
|
||||
for(size_t i=0; i < materials.size(); ++i) {
|
||||
if(!materials[i].diffuse_texname.empty()) {
|
||||
const std::string tex_name = FormatString("texture_%",i);
|
||||
try {
|
||||
TypedImage& tex_image = geom.textures[tex_name];
|
||||
tex_image = LoadImage(PathParent(filename) + "/" + materials[i].diffuse_texname);
|
||||
const int row_bytes = tex_image.w * tex_image.fmt.bpp / 8;
|
||||
std::vector<unsigned char> tmp_row(row_bytes);
|
||||
for (std::size_t y=0; y < (tex_image.h >> 1); ++y) {
|
||||
std::memcpy(tmp_row.data(), tex_image.RowPtr(y), row_bytes);
|
||||
std::memcpy(tex_image.RowPtr(y), tex_image.RowPtr(tex_image.h - 1 - y), row_bytes);
|
||||
std::memcpy(tex_image.RowPtr(tex_image.h - 1 - y), tmp_row.data(), row_bytes);
|
||||
}
|
||||
} catch(const std::exception& e) {
|
||||
pango_print_warn("Unable to read texture '%s'\n", tex_name.c_str());
|
||||
geom.textures.erase(tex_name);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// PANGO_ASSERT(all_of(
|
||||
// [&](const size_t& v){return (v == 0) || (v == num_verts);},
|
||||
// attrib.normals.size() / 3,
|
||||
// attrib.colors.size() / 3));
|
||||
|
||||
// Get rid of color buffer if all elements are equal.
|
||||
if ( std::adjacent_find( attrib.colors.begin(), attrib.colors.end(), std::not_equal_to<float>() ) == attrib.colors.end() )
|
||||
{
|
||||
attrib.colors.clear();
|
||||
}
|
||||
|
||||
Image<float> tiny_vs = GetImageWrapper(attrib.vertices, 3);
|
||||
Image<float> tiny_ns = GetImageWrapper(attrib.normals, 3);
|
||||
Image<float> tiny_cs = GetImageWrapper(attrib.colors, 3);
|
||||
Image<float> tiny_ts = GetImageWrapper(attrib.texcoords, 2);
|
||||
|
||||
// Some vertices are used with multiple texture coordinates or multiple normals, and will need to be split.
|
||||
// First, we'll find the number of unique combinations of vertex, texture, and normal indices used, as each
|
||||
// will result in a separate vertex in the buffers
|
||||
std::unordered_map<tinyobj::index_t, std::size_t> reindex_map;
|
||||
|
||||
for(auto& shape : shapes) {
|
||||
|
||||
if(shape.mesh.indices.size() == 0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const size_t num_indices = shape.mesh.indices.size() ;
|
||||
|
||||
for(size_t i=0; i < num_indices; ++i) {
|
||||
|
||||
const tinyobj::index_t index = shape.mesh.indices[i];
|
||||
|
||||
if (reindex_map.find(index) == reindex_map.end()) {
|
||||
|
||||
reindex_map.insert(std::pair<tinyobj::index_t, std::size_t>(index, reindex_map.size()));
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
const int num_unique_verts = reindex_map.size();
|
||||
|
||||
// Create unified verts attribute
|
||||
auto& verts = geom.buffers["geometry"];
|
||||
Image<float> new_vs, new_ns, new_cs, new_ts;
|
||||
{
|
||||
verts.Reinitialise(sizeof(float)*(tiny_vs.w + tiny_ns.w + tiny_cs.w + tiny_ts.w),num_unique_verts);
|
||||
size_t float_offset = 0;
|
||||
if(tiny_vs.IsValid()) {
|
||||
new_vs = verts.UnsafeReinterpret<float>().SubImage(float_offset,0,3,num_unique_verts);
|
||||
verts.attributes["vertex"] = new_vs;
|
||||
float_offset += 3;
|
||||
for (auto& el : reindex_map) {
|
||||
new_vs.Row(el.second).CopyFrom(tiny_vs.Row(el.first.vertex_index));
|
||||
}
|
||||
}
|
||||
if(tiny_ns.IsValid()) {
|
||||
new_ns = verts.UnsafeReinterpret<float>().SubImage(float_offset,0,3,num_unique_verts);
|
||||
verts.attributes["normal"] = new_ns;
|
||||
float_offset += 3;
|
||||
for (auto& el : reindex_map) {
|
||||
new_ns.Row(el.second).CopyFrom(tiny_ns.Row(el.first.normal_index));
|
||||
}
|
||||
}
|
||||
if(tiny_cs.IsValid()) {
|
||||
new_cs = verts.UnsafeReinterpret<float>().SubImage(float_offset,0,3,num_unique_verts);
|
||||
verts.attributes["color"] = new_cs;
|
||||
float_offset += 3;
|
||||
for (auto& el : reindex_map) {
|
||||
new_cs.Row(el.second).CopyFrom(tiny_cs.Row(el.first.vertex_index));
|
||||
}
|
||||
}
|
||||
if(tiny_ts.IsValid()) {
|
||||
new_ts = verts.UnsafeReinterpret<float>().SubImage(float_offset,0,2,num_unique_verts);
|
||||
verts.attributes["uv"] = new_ts;
|
||||
float_offset += 2;
|
||||
for (auto& el : reindex_map) {
|
||||
new_ts.Row(el.second).CopyFrom(tiny_ts.Row(el.first.texcoord_index));
|
||||
}
|
||||
}
|
||||
PANGO_ASSERT(float_offset * sizeof(float) == verts.w);
|
||||
}
|
||||
|
||||
for(auto& shape : shapes) {
|
||||
if(shape.mesh.indices.size() == 0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
auto faces = geom.objects.emplace(shape.name, Geometry::Element());
|
||||
|
||||
if(std::all_of( shape.mesh.num_face_vertices.begin(), shape.mesh.num_face_vertices.end(),
|
||||
[](unsigned char num){return num==3;}
|
||||
)) {
|
||||
// tri mesh
|
||||
const size_t num_indices = shape.mesh.indices.size() ;
|
||||
const size_t num_faces = shape.mesh.indices.size() / 3;
|
||||
PANGO_ASSERT(num_indices % 3 == 0);
|
||||
|
||||
// Use vert_ibo as our new IBO
|
||||
faces->second.Reinitialise(3*sizeof(uint32_t), num_faces);
|
||||
Image<uint32_t> new_ibo = faces->second.UnsafeReinterpret<uint32_t>().SubImage(0,0,3,num_faces);
|
||||
for(size_t f=0; f < num_faces; ++f) {
|
||||
for(size_t v=0; v < 3; ++v) {
|
||||
new_ibo(v,f) = reindex_map[shape.mesh.indices[3 * f + v]];
|
||||
}
|
||||
}
|
||||
faces->second.attributes["vertex_indices"] = new_ibo;
|
||||
|
||||
}else if(std::all_of( shape.mesh.num_face_vertices.begin(), shape.mesh.num_face_vertices.end(),
|
||||
[](unsigned char num){return num==4;}
|
||||
)) {
|
||||
// Quad mesh
|
||||
throw std::runtime_error("Do not support quad meshes yet.");
|
||||
}else{
|
||||
// ???
|
||||
throw std::runtime_error("Do not support meshes with mixed triangles and quads.");
|
||||
}
|
||||
}
|
||||
}else{
|
||||
throw std::runtime_error(FormatString("Unable to load OBJ file '%'. Error: '%'", filename, err));
|
||||
}
|
||||
|
||||
return geom;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user