mesh_property_types.h

Go to the documentation of this file.

#ifndef COB_MESH_PROPERTY_TYPES_H
#define COB_MESH_PROPERTY_TYPES_H

#include <OpenMesh/Core/IO/MeshIO.hh>
#include <OpenMesh/Core/Mesh/TriMesh_ArrayKernelT.hh>

namespace cob_3d_meshing
{
  class DefaultMeshTraits
  {
  public:
    typedef OpenMesh::TriMesh_ArrayKernelT<> MeshT;
    typedef MeshT::VertexHandle VertexHandle;
    typedef MeshT::FaceHandle FaceHandle;
  };


  namespace MeshProperties
  {
    //______________________________ Normals __________________________________
    template<typename PointNormalT,
             typename MeshTraits = DefaultMeshTraits>
    class Normals
    {
    public:
      typedef PointNormalT T1;
      typedef typename MeshTraits::MeshT MeshT;
      typedef typename MeshTraits::VertexHandle VertexHandle;
      typedef typename pcl::PointCloud<T1>::ConstPtr DataPtrT;
      typedef OpenMesh::Vec3f PropT;
      typedef OpenMesh::VPropHandleT<PropT> PropHandle;

      Normals(const DataPtrT& data) { data_ = data; }

      inline static std::string name() { return "Normals"; }

      inline void init(MeshT* mesh_hdl)
      {
        mesh_ = mesh_hdl;
        mesh_->add_property(prop_hdl_, name());
      }

      inline void setVertex(const VertexHandle& vertex_hdl, int idx)
      {
        Eigen::Vector3f n = (*data_)[idx].getNormalVector3fMap();
        mesh_->property(prop_hdl_, vertex_hdl) = PropT(n(0), n(1), n(2));
      }

    private:
      MeshT* mesh_;
      DataPtrT data_;
      PropHandle prop_hdl_;
    };

    //______________________________ Labels __________________________________
    template<typename PointLabelT,
             typename MeshTraits = DefaultMeshTraits>
    class Labels
    {
    public:
      typedef PointLabelT T1;
      typedef typename MeshTraits::MeshT MeshT;
      typedef typename MeshTraits::VertexHandle VertexHandle;
      typedef typename pcl::PointCloud<T1>::ConstPtr DataPtrT;
      typedef int PropT;
      typedef OpenMesh::VPropHandleT<PropT> PropHandle;

      Labels(const DataPtrT& data) { data_ = data; }

      static std::string name() { return "Labels"; }

      inline void init(MeshT* mesh_hdl)
      {
        mesh_ = mesh_hdl;
        mesh_->add_property( prop_hdl_, name() );
      }

      inline void setVertex(const VertexHandle& vertex_hdl, int idx)
      {
        mesh_->property( prop_hdl_, vertex_hdl ) = (*data_)[idx].label;
      }

    private:
      MeshT* mesh_;
      DataPtrT data_;
      PropHandle prop_hdl_;
    };

    //______________________________ Color ____________________________________

    //______________________________ PlaneCoef ________________________________
  }
}

#endif