AttributeMeshIOBase.cpp

Go to the documentation of this file.

 
#include "lvr2/io/AttributeMeshIOBase.hpp"
namespace lvr2{
 
const std::string attribute_type<EdgeHandle>::attr_group            = "edge_attributes";
const std::string attribute_type<OptionalEdgeHandle>::attr_group    = "edge_attributes";
const std::string attribute_type<VertexHandle>::attr_group          = "vertex_attributes";
const std::string attribute_type<OptionalVertexHandle>::attr_group  = "vertex_attributes";
const std::string attribute_type<FaceHandle>::attr_group            = "face_attributes";
const std::string attribute_type<OptionalFaceHandle>::attr_group    = "face_attributes";
const std::string attribute_type<ClusterHandle>::attr_group         = "cluster_attributes";
const std::string attribute_type<OptionalClusterHandle>::attr_group = "cluster_attributes";
 
bool AttributeMeshIOBase::addMesh(const HalfEdgeMesh<BaseVec>& hem)
{
  FloatChannel vertices(hem.numVertices(), 3);
  IndexChannel indices(hem.numFaces(), 3);
 
  DenseVertexMap<size_t > new_indices;
  new_indices.reserve(hem.numVertices());
 
  size_t k = 0;
  for(size_t i=0; i<hem.nextVertexIndex(); i++)
  {
    VertexHandle vH(i);
    if(!hem.containsVertex(vH)) continue;
    new_indices.insert(vH, k);
    vertices[k++] = hem.getVertexPosition(vH);
  }
 
  Index i = 0;
  for(auto fH : hem.faces())
  {
    auto vHs = hem.getVerticesOfFace(fH);
    indices[i][0] = new_indices[vHs[0]];
    indices[i][1] = new_indices[vHs[1]];
    indices[i][2] = new_indices[vHs[2]];
    i++;
  }
  return addVertices(vertices) && addIndices(indices);
}
 
boost::optional<HalfEdgeMesh<BaseVec>> AttributeMeshIOBase::getMesh()
{
  auto vertices_opt = getVertices();
  auto indices_opt = getIndices();
 
  if(vertices_opt && indices_opt)
  {
    auto& vertices = vertices_opt.get();
    auto& indices = indices_opt.get();
 
    HalfEdgeMesh<BaseVec> hem;
    for (size_t i = 0; i < vertices.numElements(); i++)
    {
      hem.addVertex(vertices[i]);
    }
 
    for (size_t i = 0; i < indices.numElements(); i++)
    {
      const std::array<VertexHandle, 3>& face = indices[i];
      hem.addFace(face[0], face[1], face[2]);
    }
    return hem;
  }
  return boost::none;
}
 
}