hand_control/src/filtre.cpp
Louis-Guillaume DUBOIS 74386e1a9f try : better filtre ?
2015-04-30 17:23:17 +02:00

63 lines
1.5 KiB
C++

#include <ros/ros.h>
#include <pcl_ros/point_cloud.h>
#include <pcl/point_types.h>
#include <pcl/filters/passthrough.h>
typedef pcl::PointCloud<pcl::PointXYZRGB> PointCloud;
class Callback {
public:
void
operator()(const PointCloud::ConstPtr& msg)
{
ROS_INFO("PointCloud received");
// pour publier un shared_ptr (mieux)
PointCloud::Ptr pcl(new PointCloud());
// filtrage
double zmax(0.);
ros::param::getCached("/filtre/zmax", zmax);
ROS_INFO("zmax : %f", zmax);
for (int i = 0; i < msg->points.size(); ++i)
{
if (msg->points[i].z > zmax)
pcl->points.push_back(msg->points[i]);
}
// publication
publisher.publish(pcl);
ROS_INFO("PointCloud published");
}
Callback(ros::Publisher& pub) : publisher(pub) {}
private:
ros::Publisher publisher;
};
int
main(int argc, char** argv)
{
ros::init(argc, argv, "filtre");
ros::NodeHandle node("filtre");
// récupération des paramètres
double zmax(0);
if (node.getParam("zmax", zmax))
{
ROS_INFO("zmax : %f" , zmax);
} else {
node.setParam("zmax", 50.0);
node.getParam("zmax", zmax);
ROS_INFO("zmax : %f (default value)", zmax);
}
// initialisation
ros::Publisher publisher = node.advertise<PointCloud>("output", 1);
Callback callback(publisher);
ros::Subscriber subscriber = node.subscribe<PointCloud>("input", 1, callback);
// démarrage
ROS_INFO("node started");
ros::spin();
ROS_INFO("exit");
return 0;
}