PCL的fromROSMsg()和toROSMsg()不能正确处理xyz之外其他field的数据长度

      使用PCL的PCL的fromROSMsg()函数将ROS的sensor_msgs::PointCloud2类型数据转换为PCL的pcl::PointCloud<T>类型数据时,假如T只是PointXYZ没问题,假如是PointXYZI,intensity这个field的数据类型是float,但是数据长度就是不对的,数据占8个字节而不是float的4个字节!toROSMsg()将PCL的pcl::PointCloud<T>类型数据转换为ROS的sensor_msgs::PointCloud2类型数据时则对数据类型和长度不作检查,完全照搬PCL里的PointXYZI的数据类型和长度,intensity占8个字节这样的有问题数据,rviz可以正常解释和播放展示,自己开发的软件在读取pcl库调用toROSMsg()转换生成的bag时通常根据intensity是float类型而认为数据长度是4个字节而读取到不正常的intensity数据而不能正常播放展示,这时需要自己实现类似fromROSMsg()和toROSMsg()这样的功能函数。

翻看了一下PCL相关源码,里面fromROSMsg()和toROSMsg()函数相关的代码如下:

/opt/ros/melodic/include/pcl_conversions/pcl_conversions.h

namespace pcl {
...
template<typename T>
  void toROSMsg(const pcl::PointCloud<T> &pcl_cloud, sensor_msgs::PointCloud2 &cloud)
  {
    pcl::PCLPointCloud2 pcl_pc2;
    pcl::toPCLPointCloud2(pcl_cloud, pcl_pc2);
    pcl_conversions::moveFromPCL(pcl_pc2, cloud);
  }

template<typename T>
  void fromROSMsg(const sensor_msgs::PointCloud2 &cloud, pcl::PointCloud<T> &pcl_cloud)
  {
    pcl::PCLPointCloud2 pcl_pc2;
    pcl_conversions::toPCL(cloud, pcl_pc2);
    pcl::fromPCLPointCloud2(pcl_pc2, pcl_cloud);
  }
  ...
}




namespace pcl_conversions {
  ...
  inline
  void fromPCL(const pcl::PCLPointField &pcl_pf, sensor_msgs::PointField &pf)
  {
    pf.name = pcl_pf.name;
    pf.offset = pcl_pf.offset;
    pf.datatype = pcl_pf.datatype;
    pf.count = pcl_pf.count;
  }

  inline
  void fromPCL(const std::vector<pcl::PCLPointField> &pcl_pfs, std::vector<sensor_msgs::PointField> &pfs)
  {
    pfs.resize(pcl_pfs.size());
    std::vector<pcl::PCLPointField>::const_iterator it = pcl_pfs.begin();
    int i = 0;
    for(; it != pcl_pfs.end(); ++it, ++i) {
      fromPCL(*(it), pfs[i]);
    }
  }

  inline
  void copyPCLPointCloud2MetaData(const pcl::PCLPointCloud2 &pcl_pc2, sensor_msgs::PointCloud2 &pc2)
  {
    fromPCL(pcl_pc2.header, pc2.header);
    pc2.height = pcl_pc2.height;
    pc2.width = pcl_pc2.width;
    fromPCL(pcl_pc2.fields, pc2.fields);
    //printf("pc2.fields.size() %d, ", pc2.fields.size());
    //printf("pcl_pc2.fields/pc2.fields: offset %d/%d, datatype %d/%d\n", pcl_pc2.fields[3].offset, pc2.fields[3].offset, pcl_pc2.fields[3].datatype, pc2.fields[3].datatype);
    pc2.is_bigendian = pcl_pc2.is_bigendian;
    pc2.point_step = pcl_pc2.point_step;
    pc2.row_step = pcl_pc2.row_step;
    pc2.is_dense = pcl_pc2.is_dense;
  }


  inline
  void moveFromPCL(pcl::PCLPointCloud2 &pcl_pc2, sensor_msgs::PointCloud2 &pc2)
  {
    copyPCLPointCloud2MetaData(pcl_pc2, pc2);
    pc2.data.swap(pcl_pc2.data);
  }

  inline
  void toPCL(const sensor_msgs::PointCloud2 &pc2, pcl::PCLPointCloud2 &pcl_pc2)
  {
    copyPointCloud2MetaData(pc2, pcl_pc2);
    pcl_pc2.data = pc2.data;
  }

  inline
  void copyPointCloud2MetaData(const sensor_msgs::PointCloud2 &pc2, pcl::PCLPointCloud2 &pcl_pc2)
  {
    toPCL(pc2.header, pcl_pc2.header);
    pcl_pc2.height = pc2.height;
    pcl_pc2.width = pc2.width;
    toPCL(pc2.fields, pcl_pc2.fields);
    pcl_pc2.is_bigendian = pc2.is_bigendian;
    pcl_pc2.point_step = pc2.point_step;
    pcl_pc2.row_step = pc2.row_step;
    pcl_pc2.is_dense = pc2.is_dense;
  }
  
  inline
  void toPCL(const sensor_msgs::PointField &pf, pcl::PCLPointField &pcl_pf)
  {
    pcl_pf.name = pf.name;
    pcl_pf.offset = pf.offset;
    pcl_pf.datatype = pf.datatype;
    pcl_pf.count = pf.count;
  }

  inline
  void toPCL(const std::vector<sensor_msgs::PointField> &pfs, std::vector<pcl::PCLPointField> &pcl_pfs)
  {
    pcl_pfs.resize(pfs.size());
    std::vector<sensor_msgs::PointField>::const_iterator it = pfs.begin();
    int i = 0;
    for(; it != pfs.end(); ++it, ++i) {
      toPCL(*(it), pcl_pfs[i]);
    }
  }

   inline
  void toPCL(const std_msgs::Header &header, pcl::PCLHeader &pcl_header)
  {
    toPCL(header.stamp, pcl_header.stamp);
    pcl_header.seq = header.seq;
    pcl_header.frame_id = header.frame_id;
  }
  inline
  void toPCL(const ros::Time &stamp, std::uint64_t &pcl_stamp)
  {
    pcl_stamp = stamp.toNSec() / 1000ull;  // Convert from ns to us
  }
  ...
}

/usr/include/pcl-1.8/pcl/conversions.h

template<typename PointT> void
  toPCLPointCloud2 (const pcl::PointCloud<PointT>& cloud, pcl::PCLPointCloud2& msg)
  {
    // Ease the user's burden on specifying width/height for unorganized datasets
    if (cloud.width == 0 && cloud.height == 0)
    {
      msg.width  = static_cast<uint32_t>(cloud.points.size ());
      msg.height = 1;
    }
    else
    {
      assert (cloud.points.size () == cloud.width * cloud.height);
      msg.height = cloud.height;
      msg.width  = cloud.width;
    }

    // Fill point cloud binary data (padding and all)
    size_t data_size = sizeof (PointT) * cloud.points.size ();
    msg.data.resize (data_size);
    if (data_size)
    {
      memcpy(&msg.data[0], &cloud.points[0], data_size);
    }

    // Fill fields metadata
    msg.fields.clear ();
    for_each_type<typename traits::fieldList<PointT>::type> (detail::FieldAdder<PointT>(msg.fields));
    for(auto f: msg.fields) {
      printf("f.name %s, offset %d, datatype %d\n", f.name.c_str(), f.offset, f.datatype);
    }
    msg.header     = cloud.header;
    msg.point_step = sizeof (PointT);
    msg.row_step   = static_cast<uint32_t> (sizeof (PointT) * msg.width);
    msg.is_dense   = cloud.is_dense;
    /// @todo msg.is_bigendian = ?;
  }

 struct FieldAdder
    {
      FieldAdder (std::vector<pcl::PCLPointField>& fields) : fields_ (fields) {};

      template<typename U> void operator() ()
      {
        pcl::PCLPointField f;
        f.name = traits::name<PointT, U>::value;
        f.offset = traits::offset<PointT, U>::value;
        f.datatype = traits::datatype<PointT, U>::value;
        f.count = traits::datatype<PointT, U>::size;
        fields_.push_back (f);
      }

      std::vector<pcl::PCLPointField>& fields_;
    };

/usr/include/pcl-1.8/pcl/point_traits.h
// offset
    template<class PointT, typename Tag>
    struct offset : offset<typename POD<PointT>::type, Tag>
    {
      // Contents of specialization:
      // static const size_t value;

      // Avoid infinite compile-time recursion
      BOOST_MPL_ASSERT_MSG((!boost::is_same<PointT, typename POD<PointT>::type>::value),
                           POINT_TYPE_NOT_PROPERLY_REGISTERED, (PointT&));
    };

/usr/include/pcl-1.8/pcl/register_point_struct.h

#define POINT_CLOUD_REGISTER_FIELD_OFFSET(r, name, elem)                \
  template<> struct offset<name, pcl::fields::BOOST_PP_TUPLE_ELEM(3, 2, elem)> \
  {                                                                     \
    static const size_t value = offsetof(name, BOOST_PP_TUPLE_ELEM(3, 1, elem)); \
  };

#define POINT_CLOUD_REGISTER_POINT_STRUCT(name, fseq)               \
  POINT_CLOUD_REGISTER_POINT_STRUCT_I(name,                         \
    BOOST_PP_CAT(POINT_CLOUD_REGISTER_POINT_STRUCT_X fseq, 0))      \

/usr/include/pcl-1.8/pcl/point_types.h
POINT_CLOUD_REGISTER_POINT_STRUCT (pcl::_PointXYZI,
    (float, x, x)
    (float, y, y)
    (float, z, z)
    (float, intensity, intensity)
)
POINT_CLOUD_REGISTER_POINT_WRAPPER(pcl::PointXYZI, pcl::_PointXYZI)




/usr/include/pcl-1.8/pcl/PCLPointField.h
namespace pcl
{
  struct PCLPointField
  {
    PCLPointField () : name (), offset (0), datatype (0), count (0)
    {}

    std::string name;

    pcl::uint32_t offset;
    pcl::uint8_t datatype;
    pcl::uint32_t count;

    enum PointFieldTypes { INT8 = 1,
                           UINT8 = 2,
                           INT16 = 3,
                           UINT16 = 4,
                           INT32 = 5,
                           UINT32 = 6,
                           FLOAT32 = 7,
                           FLOAT64 = 8 };

/usr/include/pcl-1.8/pcl/register_point_struct.h




自己实现类似fromROSMsg()和toROSMsg()函数功能的代码如下:

void ROSMsg2PCL(sensor_msgs::PointCloud2& ros_msgs, cloud::Ptr cloud_ptr) {
  cloud_ptr->header.seq = ros_msgs.header.seq;
  cloud_ptr->header.frame_id = ros_msgs.header.frame_id;
  cloud_ptr->header.stamp.fromNSec(ros_msgs->header.stamp / 1000ull); // ns => us
  cloud_ptr->height = ros_msgs.height;
  cloud_ptr->width = ros_msgs.width;
  cloud_ptr->is_dense = ros_msgs.is_dense;
  
  int points_num = ros_msgs.height * ros_msgs.width;
  cloud_ptr->points.resize(points_num * 4 * 4);  // XYZI
  float * pdata = reinterpret_cast<float*>(&ros_msgs.data[0]);
  for (auto i = 0; i < points_num; ++i) {
      cloud_ptr->points[i*4 + 0] = pdata[i*4 + 0];
      cloud_ptr->points[i*4 + 1] = pdata[i*4 + 1];
      cloud_ptr->points[i*4 + 2] = pdata[i*4 + 2];
      if (ros_msgs.point_step == 16) // 4+4+4+4
        cloud_ptr->points[i*4 + 3] = pdata[i*4 + 3];
      /*有的雷达输出的rosmsg在fields的排放顺序有特殊安排,需要使用POINT_CLOUD_REGISTER_POINT_STRUCT注册特定的struct来读取XYZI,例如:
         POINT_CLOUD_REGISTER_POINT_STRUCT(OusterPointXYZIRT,
           (float, x, x)
           (float, y, y)
           (float, z, z)
           (float, intensity, intensity))
      即便是保证了XYZI的排放顺序,intensity可能占了8个字节,前4个字节为0后4个字节才是intensity值,不同雷达处理可能不一样,所以这里的处理只是示例,不同雷达是情况而定
      */
      else if (ros_msgs.point_step == 20) //4+4+4+8
        cloud_ptr->points[i*4 + 3] = pdata[i*4 + 4];
  }
}

void initRosMsgFields(std::vector<sensor_msgs::PointField>& fields) {
  fields.resize(4);
  fields[0].name = "x";
  fields[0].offset = 0;
  fields[0].datatype = 7;
  fields[0].count = 1;
  fields[1].name = "y";
  fields[1].offset = 4;
  fields[1].datatype = 7;
  fields[1].count = 1;
  fields[2].name = "z";
  fields[2].offset = 8;
  fields[2].datatype = 7;
  fields[2].count = 1;
  fields[3].name = "intensity";
  fields[3].offset = 12;
  fields[3].datatype = 7;
  fields[3].count = 1;
}

void PCL2ROSMSg(cloud::Ptr cloud_ptr, sensor_msgs::PointCloud2& ros_msgs) {
    //set header with PCL Cloud header for common use. 
    ros_msgs.header.stamp.fromNSec(cloud_ptr->header.stamp * 1000ull); // us => ns
    ros_msgs.header.seq = cloud_ptr->header.seq;
    ros_msgs.header.frame_id = cloud_ptr->header.frame_id;

    initRosMsgFields(ros_msgs.fields);

    ros_msgs.width = cloud_ptr->width;   // 1
    ros_msgs.height = cloud_ptr->height; // point_num
    ros_msgs.point_step = ros_msgs.fields.size() * sizeof(float); // 4 * 4
    ros_msgs.row_step   = static_cast<uint32_t> (ros_msgs.point_step * ros_msgs.width);
    ros_msgs.is_dense   = cloud_ptr->is_dense;

    int data_size = cloud_ptr->size() * ros_msgs.point_step;
    if (data_size > 0) {
       ros_msgs.data.resize(data_size);
       //memcpy(&ros_msgs.data[0], &cloud_ptr->points[0], data_size);
       float * pdata = reinterpret_cast<float*>(&ros_msgs.data[0]);
       for (int i = 0; i < cloud_ptr->size(); i++ ) {
          pcl::PointXYZI pt = cloud_ptr->points[i];
	  pdata[i*4 + 0] = pt.x;
	  pdata[i*4 + 1] = pt.y;
	  pdata[i*4 + 2] = pt.z;
	  pdata[i*4+ 3] = pt.intensity;
       }
    }
}

  • 10
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Arnold-FY-Chen

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值