point_cloud.h

Go to the documentation of this file.

/*
 * Software License Agreement (BSD License)
 *
 *  Point Cloud Library (PCL) - www.pointclouds.org
 *  Copyright (c) 2010-2011, Willow Garage, Inc.
 *
 *  All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above
 *     copyright notice, this list of conditions and the following
 *     disclaimer in the documentation and/or other materials provided
 *     with the distribution.
 *   * Neither the name of Willow Garage, Inc. nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 *  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 *  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 *  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 *  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 *  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 *  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *  POSSIBILITY OF SUCH DAMAGE.
 *
 * $Id: point_cloud.h 3746 2011-12-31 22:19:47Z rusu $
 *
 */

#ifndef PCL_POINT_CLOUD_H_
#define PCL_POINT_CLOUD_H_

#include <Eigen/StdVector>
#include <Eigen/Geometry>
#include <cstddef>
#include <std_msgs/Header.h>
#include <pcl/pcl_macros.h>
#include <pcl/exceptions.h>
#include <pcl/cloud_properties.h>
#include <pcl/channel_properties.h>
#include <pcl/point_traits.h>
#include <pcl/for_each_type.h>
#include <boost/shared_ptr.hpp>
#include <map>
#include <boost/mpl/size.hpp>

namespace pcl
{
  namespace detail
  {
    struct FieldMapping
    {
      size_t serialized_offset;
      size_t struct_offset;
      size_t size;
    };
  } // namespace detail


  // Forward declarations
  template <typename PointT> class PointCloud;
  typedef std::vector<detail::FieldMapping> MsgFieldMap;
  template <typename PointT> struct NdCopyEigenPointFunctor;
  template <typename PointT> struct NdCopyPointEigenFunctor;
 
  namespace detail
  {
    template <typename PointT> boost::shared_ptr<pcl::MsgFieldMap>&
    getMapping (pcl::PointCloud<PointT>& p);
  } // namespace detail

  template <typename PointT>
  class PointCloud
  {
    public:
      PointCloud () : width (0), height (0), is_dense (true),
                      sensor_origin_ (Eigen::Vector4f::Zero ()), sensor_orientation_ (Eigen::Quaternionf::Identity ())
      {}

      inline PointCloud (PointCloud<PointT> &pc)
      {
        *this = pc;
      }

      inline PointCloud (const PointCloud<PointT> &pc)
      {
        *this = pc;
      }

      inline PointCloud (const PointCloud<PointT> &pc, 
                         const std::vector<int> &indices)
      {
        // Copy the obvious
        width    = indices.size ();
        height   = 1;
        is_dense = pc.is_dense;
        header = pc.header;
        sensor_origin_ = pc.sensor_origin_;
        sensor_orientation_ = pc.sensor_orientation_;

        assert (indices.size () <= pc.size ());
        points.resize (indices.size ());
        for (size_t i = 0; i < indices.size (); i++)
          points[i] = pc.points[indices[i]];
      }

      inline PointCloud (uint32_t width_, uint32_t height_)
        : points (width_ * height_)
        , width (width_)
        , height (height_)
        , is_dense (true)
        , sensor_origin_ (Eigen::Vector4f::Zero ())
        , sensor_orientation_ (Eigen::Quaternionf::Identity ())
      {}

      virtual ~PointCloud () {}


      inline PointCloud&
      operator += (const PointCloud& rhs)
      {
        // Make the resultant point cloud take the newest stamp
        if (rhs.header.stamp > header.stamp)
          header.stamp = rhs.header.stamp;

        size_t nr_points = points.size ();
        points.resize (nr_points + rhs.points.size ());
        for (size_t i = nr_points; i < points.size (); ++i)
          points[i] = rhs.points[i - nr_points];

        width    = (uint32_t) points.size ();
        height   = 1;
        if (rhs.is_dense && is_dense)
          is_dense = true;
        else
          is_dense = false;
        return (*this);
      }


      inline const PointCloud
      operator + (const PointCloud& rhs)
      {
        return (PointCloud (*this) += rhs);
      }
      

      inline const PointT&
      at (int u, int v) const
      {
        if (this->height > 1)
          return (points.at (v * this->width + u));
        else
          throw IsNotDenseException ("Can't use 2D indexing with a unorganized point cloud");
      }

      inline PointT&
      at (int u, int v)
      {
        if (this->height > 1)
          return (points.at (v * this->width + u));
        else
          throw IsNotDenseException ("Can't use 2D indexing with a unorganized point cloud");
      }


      inline const PointT&
      operator () (int u, int v) const
      {
        return (points[v * this->width + u]);
      }

      inline PointT&
      operator () (int u, int v)
      {
        return (points[v * this->width + u]);
      }


      inline bool
      isOrganized () const
      {
        return (height != 1);
      }
      

      inline Eigen::Map<Eigen::MatrixXf, Eigen::Aligned, Eigen::OuterStride<> > 
      getMatrixXfMap (int dim, int stride, int offset)
      {
        //return Eigen::Map<Eigen::MatrixXf, Eigen::Aligned, Eigen::OuterStride<> >((float*)(&points[0])+offset, dim, points.size(), Eigen::OuterStride<>(stride));
        return (Eigen::Map<Eigen::MatrixXf, Eigen::Aligned, Eigen::OuterStride<> >((float*)(&points[0])+offset, points.size (), dim, Eigen::OuterStride<> (stride)));
      }

      inline const Eigen::Map<const Eigen::MatrixXf, Eigen::Aligned, Eigen::OuterStride<> >
      getMatrixXfMap (int dim, int stride, int offset) const
      {
        return (Eigen::Map<Eigen::MatrixXf, Eigen::Aligned, Eigen::OuterStride<> >((float*)(&points[0])+offset, points.size (), dim, Eigen::OuterStride<> (stride)));
      }


      inline Eigen::Map<Eigen::MatrixXf, Eigen::Aligned, Eigen::OuterStride<> >
      getMatrixXfMap () 
      {
        return (getMatrixXfMap (sizeof (PointT) / sizeof (float),  sizeof (PointT) / sizeof (float), 0));
      }

      inline const Eigen::Map<const Eigen::MatrixXf, Eigen::Aligned, Eigen::OuterStride<> >
      getMatrixXfMap () const
      {
        return (getMatrixXfMap (sizeof (PointT) / sizeof (float),  sizeof (PointT) / sizeof (float), 0));
      }

      std_msgs::Header header;

      std::vector<PointT, Eigen::aligned_allocator<PointT> > points;

      uint32_t width;
      uint32_t height;

      bool is_dense;

      Eigen::Vector4f    sensor_origin_;
      Eigen::Quaternionf sensor_orientation_;

      typedef PointT PointType;  // Make the template class available from the outside
      typedef std::vector<PointT, Eigen::aligned_allocator<PointT> > VectorType;
      typedef boost::shared_ptr<PointCloud<PointT> > Ptr;
      typedef boost::shared_ptr<const PointCloud<PointT> > ConstPtr;

      // iterators
      typedef typename VectorType::iterator iterator;
      typedef typename VectorType::const_iterator const_iterator;
      inline iterator begin () { return (points.begin ()); }
      inline iterator end ()   { return (points.end ()); }
      inline const_iterator begin () const { return (points.begin ()); }
      inline const_iterator end () const  { return (points.end ()); }

      //capacity
      inline size_t size () const { return (points.size ()); }
      inline void reserve (size_t n) { points.reserve (n); }
      inline void resize (size_t n) { points.resize (n); }
      inline bool empty () const { return points.empty (); }

      //element access
      inline const PointT& operator[] (size_t n) const { return points[n]; }
      inline PointT& operator[] (size_t n) { return points[n]; }
      inline const PointT& at (size_t n) const { return points.at (n); }
      inline PointT& at (size_t n) { return points.at (n); }
      inline const PointT& front () const { return points.front (); }
      inline PointT& front () { return points.front (); }
      inline const PointT& back () const { return points.back (); }
      inline PointT& back () { return points.back (); }

      inline void 
      push_back (const PointT& pt)
      {
        points.push_back (pt);
        width = points.size ();
        height = 1;
      }

      inline iterator 
      insert (iterator position, const PointT& pt)
      {
        iterator it = points.insert (position, pt);
        width = points.size ();
        height = 1;
        return (it);
      }

      inline void 
      insert (iterator position, size_t n, const PointT& pt)
      {
        points.insert (position, n, pt);
        width = points.size ();
        height = 1;
      }

      template <class InputIterator> inline void 
      insert (iterator position, InputIterator first, InputIterator last)
      {
        points.insert (position, first, last);
        width = points.size ();
        height = 1;
      }

      inline iterator 
      erase (iterator position)
      {
        iterator it = points.erase (position); 
        width = points.size ();
        height = 1;
        return (it);
      }

      inline iterator 
      erase (iterator first, iterator last)
      {
        iterator it = points.erase (first, last);
        width = points.size ();
        height = 1;
        return (it);
      }

      inline void 
      swap (PointCloud<PointT> &rhs)
      {
        this->points.swap (rhs.points);
        std::swap (width, rhs.width);
        std::swap (height, rhs.height);
        std::swap (is_dense, rhs.is_dense);
        std::swap (sensor_origin_, rhs.sensor_origin_);
        std::swap (sensor_orientation_, rhs.sensor_orientation_);
      }

      inline void 
      clear ()
      {
        points.clear ();
        width = 0;
        height = 0;
      }

      inline Ptr 
      makeShared () { return Ptr (new PointCloud<PointT> (*this)); }

      inline ConstPtr 
      makeShared () const { return ConstPtr (new PointCloud<PointT> (*this)); }

    protected:
      // This is motivated by ROS integration. Users should not need to access mapping_.
      boost::shared_ptr<MsgFieldMap> mapping_;

      friend boost::shared_ptr<MsgFieldMap>& detail::getMapping<PointT>(pcl::PointCloud<PointT> &p);

    public:
      EIGEN_MAKE_ALIGNED_OPERATOR_NEW;
  };

  template <>
  class PointCloud<Eigen::MatrixXf> 
  {
    public:
      PointCloud () : 
        points (Eigen::MatrixXf (0, 0)), width (0), height (0), is_dense (true)
      {}

      inline PointCloud (PointCloud<Eigen::MatrixXf> &pc)
      {
        *this = pc;
      }
 
      template <typename PointT>
      inline PointCloud (PointCloud<PointT> &pc)
      {
        // Copy the obvious
        width    = pc.width;
        height   = pc.height;
        is_dense = pc.is_dense;
        properties.acquisition_time   = pc.header.stamp;
        properties.sensor_origin      = pc.sensor_origin_;//.head<3> ();
        properties.sensor_orientation = pc.sensor_orientation_;

        typedef typename pcl::traits::fieldList<PointT>::type FieldList;
        // Copy the fields
        // ...

        // Resize the array
        points.resize (pc.points.size (), boost::mpl::size<FieldList>::value);

        for (size_t cp = 0; cp < pc.points.size (); ++cp)
          pcl::for_each_type <FieldList> (NdCopyPointEigenFunctor<PointT> (pc.points[cp], points.row (cp)));
      }
 
      inline PointCloud (const PointCloud<Eigen::MatrixXf> &pc)
      {
        *this = pc;
      }

      template <typename PointT>
      inline PointCloud (const PointCloud<PointT> &pc)
      {
        // Copy the obvious
        width    = pc.width;
        height   = pc.height;
        is_dense = pc.is_dense;
        properties.acquisition_time   = pc.header.stamp;
        properties.sensor_origin      = pc.sensor_origin_;//.head<3> ();
        properties.sensor_orientation = pc.sensor_orientation_;

        typedef typename pcl::traits::fieldList<PointT>::type FieldList;

        // Copy the fields
        // ...

        // Resize the array
        points.resize (pc.points.size (), boost::mpl::size<FieldList>::value);

        for (size_t cp = 0; cp < pc.points.size (); ++cp)
          pcl::for_each_type <FieldList> (NdCopyPointEigenFunctor<PointT> (pc.points[cp], points.row (cp)));
      }

      inline PointCloud (const PointCloud &pc, 
                         const std::vector<int> &indices)
      {
        // Copy the obvious
        width    = indices.size ();
        height   = 1;
        is_dense = pc.is_dense;
        properties = pc.properties;
        channels = pc.channels;

        assert ((int)indices.size () <= pc.points.rows ());
        points.resize (indices.size (), pc.points.cols ());
        for (size_t i = 0; i < indices.size (); i++)
          points.row (i) = pc.points.row (indices[i]);
      }

      inline PointCloud (uint32_t _width, uint32_t _height, uint32_t _dim)
        : points (Eigen::MatrixXf (_width * _height, _dim))
        , width (_width)
        , height (_height)
        , is_dense (true)
      {}

      inline PointCloud (uint32_t _num_points, uint32_t _dim)
        : points (Eigen::MatrixXf (_num_points, _dim))
        , width (_num_points)
        , height (1)
        , is_dense (true)
      {}

      virtual ~PointCloud () {}


      inline PointCloud&
      operator += (const PointCloud& rhs)
      {
        if (rhs.properties.acquisition_time > properties.acquisition_time)
          properties.acquisition_time = rhs.properties.acquisition_time;

        properties.sensor_origin = Eigen::Vector4f::Zero ();
        properties.sensor_orientation = Eigen::Quaternionf::Identity ();

        int nr_points = points.rows ();
        points.resize (nr_points + rhs.points.rows (), points.cols ());
        for (int i = nr_points; i < points.rows (); ++i)
          points.row (i) = rhs.points.row (i - nr_points);

        channels = rhs.channels;
        width    = (uint32_t) points.rows ();
        height   = 1;
        if (rhs.is_dense && is_dense)
          is_dense = true;
        else
          is_dense = false;
        return (*this);
      }
      

      inline const PointCloud
      operator + (const PointCloud& rhs)
      {
        return (PointCloud (*this) += rhs);
      }


      inline Eigen::Map<Eigen::VectorXf>
      at (int u, int v)
      {
        if (height > 1)
          return (Eigen::VectorXf::Map (&points (v * width + u, 0), points.cols ()));
        else
          throw IsNotDenseException ("Can't use 2D indexing with a unorganized point cloud");
      }


      inline Eigen::Map<Eigen::VectorXf>
      operator () (int u, int v) 
      {
        return (Eigen::VectorXf::Map (&points (v * width + u, 0), points.cols ()));
      }


      inline bool
      isOrganized () const
      {
        return (height != 1);
      }
      
      pcl::CloudProperties properties;

      Eigen::MatrixXf points;

      std::map<std::string, pcl::ChannelProperties> channels;

      uint32_t width;
      uint32_t height;

      bool is_dense;

      typedef boost::shared_ptr<PointCloud<Eigen::MatrixXf> > Ptr;
      typedef boost::shared_ptr<const PointCloud<Eigen::MatrixXf> > ConstPtr;

      inline void 
      swap (PointCloud<Eigen::MatrixXf> &rhs)
      {
        std::swap (points, rhs.points);
        std::swap (width, rhs.width);
        std::swap (height, rhs.height);
        std::swap (is_dense, rhs.is_dense);
        std::swap (properties, rhs.properties);
        std::swap (channels, rhs.channels);
      }

      inline void 
      clear ()
      {
        points.resize (0, 0);
        width = 0;
        height = 0;
      }

      inline Ptr 
      makeShared () { return Ptr (new PointCloud<Eigen::MatrixXf> (*this)); }

      inline ConstPtr 
      makeShared () const { return ConstPtr (new PointCloud<Eigen::MatrixXf> (*this)); }

    public:
      EIGEN_MAKE_ALIGNED_OPERATOR_NEW;
  };


  template <typename PointT>
  struct NdCopyEigenPointFunctor
  {
    typedef typename traits::POD<PointT>::type Pod;
    
    NdCopyEigenPointFunctor (const Eigen::VectorXf &p1, PointT &p2)
      : p1_ (p1),
        p2_ (reinterpret_cast<Pod&>(p2)),
        f_idx_ (0) { }

    template<typename Key> inline void 
    operator() ()
    {
      //boost::fusion::at_key<Key> (p2_) = p1_[f_idx_++];
      typedef typename pcl::traits::datatype<PointT, Key>::type T;
      uint8_t* data_ptr = reinterpret_cast<uint8_t*>(&p2_) + pcl::traits::offset<PointT, Key>::value;
      *reinterpret_cast<T*>(data_ptr) = p1_[f_idx_++];
    }

    private:
      const Eigen::VectorXf &p1_;
      Pod &p2_;
      int f_idx_;
    public:
      EIGEN_MAKE_ALIGNED_OPERATOR_NEW;
   };


  template <typename PointT>
  struct NdCopyPointEigenFunctor
  {
    typedef typename traits::POD<PointT>::type Pod;
    
     NdCopyPointEigenFunctor (const PointT &p1, Eigen::VectorXf &p2)
      : p1_ (reinterpret_cast<const Pod&>(p1)), p2_ (p2), f_idx_ (0) { }

    template<typename Key> inline void 
    operator() ()
    {
      //p2_[f_idx_++] = boost::fusion::at_key<Key> (p1_);
      typedef typename pcl::traits::datatype<PointT, Key>::type T;
      const uint8_t* data_ptr = reinterpret_cast<const uint8_t*>(&p1_) + pcl::traits::offset<PointT, Key>::value;
      p2_[f_idx_++] = *reinterpret_cast<const T*>(data_ptr);
    }

    private:
      const Pod &p1_;
      Eigen::VectorXf &p2_;
      int f_idx_;
    public:
      EIGEN_MAKE_ALIGNED_OPERATOR_NEW;
   };

  namespace detail
  {
    template <typename PointT> boost::shared_ptr<pcl::MsgFieldMap>&
    getMapping (pcl::PointCloud<PointT>& p)
    {
      return (p.mapping_);
    }
  } // namespace detail

  template <typename PointT> std::ostream&
  operator << (std::ostream& s, const pcl::PointCloud<PointT> &p)
  {
    s << "points[]: " << p.points.size () << std::endl;
    s << "width: " << p.width << std::endl;
    s << "height: " << p.height << std::endl;
    s << "is_dense: " << p.is_dense << std::endl;
    return (s);
  }
}

#endif  //#ifndef PCL_POINT_CLOUD_H_