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;
}

}