led_finder.h

Go to the documentation of this file.

/*
 * Copyright (C) 2015 Fetch Robotics Inc.
 * Copyright (C) 2013-2014 Unbounded Robotics Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
// Author: Michael Ferguson
 
#ifndef ROBOT_CALIBRATION_CAPTURE_LED_FINDER_H
#define ROBOT_CALIBRATION_CAPTURE_LED_FINDER_H
 
#include <ros/ros.h>
#include <robot_calibration/capture/depth_camera.h>
#include <robot_calibration/plugins/feature_finder.h>
 
#include <tf/transform_listener.h>
#include <sensor_msgs/Image.h>
#include <sensor_msgs/PointCloud2.h>
#include <geometry_msgs/PointStamped.h>
#include <robot_calibration_msgs/CalibrationData.h>
#include <robot_calibration_msgs/GripperLedCommandAction.h>
#include <actionlib/client/simple_action_client.h>
 
namespace robot_calibration
{
 
class LedFinder : public FeatureFinder
{
  struct CloudDifferenceTracker
  {
 
    CloudDifferenceTracker(std::string frame, double x, double y, double z);
 
    bool process(sensor_msgs::PointCloud2& cloud,
                 sensor_msgs::PointCloud2& prev,
                 geometry_msgs::Point& led_point,
                 double max_distance,
                 double weight);
 
    // Have we found the LED?
    bool isFound(const sensor_msgs::PointCloud2& cloud,
                 double threshold);
 
    // Gives a refined centroid using multiple points
    bool getRefinedCentroid(const sensor_msgs::PointCloud2& cloud,
                            geometry_msgs::PointStamped& centroid);
 
    // Reset the tracker
    void reset(size_t height, size_t width);
 
    // Get an image of tracker status
    sensor_msgs::Image getImage();
 
    std::vector<double> diff_;
    double max_;
    int max_idx_;
    int count_;
    size_t height_, width_;
    std::string frame_;  // frame of led coordinates
    geometry_msgs::Point point_;  //coordinates of led this is tracking
  };
 
  typedef actionlib::SimpleActionClient<robot_calibration_msgs::GripperLedCommandAction> LedClient;
 
public:
  LedFinder();
  bool init(const std::string& name, ros::NodeHandle & n);
  bool find(robot_calibration_msgs::CalibrationData * msg);
 
private:
  void cameraCallback(const sensor_msgs::PointCloud2::ConstPtr& cloud);
  bool waitForCloud();
 
  ros::Subscriber subscriber_;  
  ros::Publisher publisher_;  
  boost::scoped_ptr<LedClient> client_;
 
  bool waiting_;
  sensor_msgs::PointCloud2 cloud_;
 
  std::vector<boost::shared_ptr<ros::Publisher> > tracker_publishers_;
  std::vector<CloudDifferenceTracker> trackers_;
  std::vector<uint8_t> codes_;
 
  tf::TransformListener listener_;
  DepthCameraInfoManager depth_camera_manager_;
 
  /*
   * ROS Parameters
   */
  double max_error_;    
  double max_inconsistency_;
 
  double threshold_;    
  int max_iterations_;  
 
  bool output_debug_;   
 
  std::string camera_sensor_name_;
  std::string chain_sensor_name_;
};
 
}  // namespace robot_calibration
 
#endif  // ROBOT_CALIBRATION_CAPTURE_LED_FINDER_H