calibrate.cpp

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

#include <fstream>
#include <ctime>

#include <ros/ros.h>
#include <rosbag/bag.h>
#include <rosbag/view.h>
#include <rosbag/query.h>

#include <std_msgs/String.h>
#include <robot_calibration_msgs/CalibrationData.h>
#include <robot_calibration_msgs/CaptureConfig.h>

#include <robot_calibration/capture/chain_manager.h>
#include <robot_calibration/capture/feature_finder.h>

#include <camera_calibration_parsers/parse.h>
#include <robot_calibration/ceres/optimizer.h>
#include <robot_calibration/camera_info.h>

#include <boost/foreach.hpp>  // for rosbag iterator

/*
 * usage:
 *  calibrate --manual
 *  calibrate calibration_poses.bag
 *  calibrate --from-bag calibration_data.bag
 */
int main(int argc, char** argv)
{
  ros::init(argc, argv,"robot_calibration");
  ros::NodeHandle nh("~");

  // Should we be stupidly verbose?
  bool verbose;
  nh.param<bool>("verbose", verbose, false);

  // The calibration data
  std_msgs::String description_msg;
  std::vector<robot_calibration_msgs::CalibrationData> data;

  // What bag to use to load calibration poses out of (for capture)
  std::string pose_bag_name("calibration_poses.bag");
  if (argc > 1)
    pose_bag_name = argv[1];

  if (pose_bag_name.compare("--from-bag") != 0)
  {
    // No name provided for a calibration bag file, must do capture
    robot_calibration::ChainManager chain_manager_(nh);
    robot_calibration::FeatureFinderMap finders_;
    if (!robot_calibration::loadFeatureFinders(nh, finders_))
    {
      ROS_FATAL("Unable to load feature finders!");
      return -1;
    }

    ros::Publisher pub = nh.advertise<robot_calibration_msgs::CalibrationData>("/calibration_data", 10);
    ros::Publisher urdf_pub = nh.advertise<std_msgs::String>("/robot_description", 1, true);  // latched

    // Get the robot_description and republish it
    if (!nh.getParam("/robot_description", description_msg.data))
    {
      ROS_FATAL("robot_description not set!");
      return -1;
    }
    urdf_pub.publish(description_msg);

    // Load a set of calibration poses
    std::vector<robot_calibration_msgs::CaptureConfig> poses;
    if (pose_bag_name.compare("--manual") != 0)
    {
      ROS_INFO_STREAM("Opening " << pose_bag_name);
      rosbag::Bag bag;
      try
      {
        bag.open(pose_bag_name, rosbag::bagmode::Read);
      }
      catch (rosbag::BagException)
      {
        ROS_FATAL_STREAM("Cannot open " << pose_bag_name);
        return -1;
      }
      rosbag::View data_view(bag, rosbag::TopicQuery("calibration_joint_states"));

      BOOST_FOREACH (rosbag::MessageInstance const m, data_view)
      {
        robot_calibration_msgs::CaptureConfig::ConstPtr msg = m.instantiate<robot_calibration_msgs::CaptureConfig>();
        if (msg == NULL)
        {
          // Try to load older style bags
          sensor_msgs::JointState::ConstPtr js_msg = m.instantiate<sensor_msgs::JointState>();
          if (js_msg != NULL)
          {
            robot_calibration_msgs::CaptureConfig config;
            config.joint_states = *js_msg;
            // Assume all finders should find this pose (old style config):
            for (robot_calibration::FeatureFinderMap::iterator it = finders_.begin();
                 it != finders_.end();
                 it++)
            {
              config.features.push_back(it->first);
            }
            poses.push_back(config);
          }
        }
        else
        {
          poses.push_back(*msg);
        }
      }
    }
    else
    {
      ROS_INFO("Using manual calibration mode...");
    }

    // For each pose in the capture sequence.
    for (unsigned pose_idx = 0;
         (pose_idx < poses.size() || poses.size() == 0) && ros::ok();
         ++pose_idx)
    {
      robot_calibration_msgs::CalibrationData msg;

      if (poses.size() == 0)
      {
        // Manual calibration, wait for keypress
        ROS_INFO("Press key when arm is ready... (type 'done' to finish capture)");
        std::string throwaway;
        std::getline(std::cin, throwaway);
        if (throwaway.compare("done") == 0)
          break;
        if (!ros::ok())
          break;
      }
      else
      {
        // Move head/arm to pose
        if (!chain_manager_.moveToState(poses[pose_idx].joint_states))
        {
          ROS_WARN("Unable to move to desired state for sample %u.", pose_idx);
          continue;
        }
      }

      // Regardless of manual vs. automatic, wait for joints to settle
      chain_manager_.waitToSettle();

      // Make sure sensor data is up to date after settling
      ros::Duration(0.1).sleep();

      // Get pose of the features
      bool found_all_features = true;
      if (poses.size() == 0)
      {
        // In manual mode, we need to capture all features
        for (robot_calibration::FeatureFinderMap::iterator it = finders_.begin();
             it != finders_.end();
             it++)
        {
          if (!it->second->find(&msg))
          {
            ROS_WARN("%s failed to capture features.", it->first.c_str());
            found_all_features = false;
            break;
          }
        }
      }
      else
      {
        // Capture only the intended features for this sample
        for (size_t i = 0; i < poses[pose_idx].features.size(); i++)
        {
          std::string feature = poses[pose_idx].features[i];
          if (!finders_[feature]->find(&msg))
          {
            ROS_WARN("%s failed to capture features.", feature.c_str());
            found_all_features = false;
            break;
          }
        }
      }

      // Make sure we succeeded
      if (found_all_features)
      {
        ROS_INFO("Captured pose %u", pose_idx);
      }
      else
      {
        ROS_WARN("Failed to capture sample %u.", pose_idx);
        continue;
      }

      // Fill in joint values
      chain_manager_.getState(&msg.joint_states);

      // Publish calibration data message.
      pub.publish(msg);

      // Add to samples
      data.push_back(msg);
    }

    ROS_INFO("Done capturing samples");
  }
  else
  {
    // Load calibration data from bagfile
    std::string data_bag_name("/tmp/calibration_data.bag");
    if (argc > 2)
      data_bag_name = argv[2];
    ROS_INFO_STREAM("Loading calibration data from " << data_bag_name);

    rosbag::Bag bag_;
    try
    {
      bag_.open(data_bag_name, rosbag::bagmode::Read);
    }
    catch (rosbag::BagException)
    {
      ROS_FATAL_STREAM("Cannot open " << data_bag_name);
      return -1;
    }

    // Get robot_description from bag file
    rosbag::View model_view_(bag_, rosbag::TopicQuery("robot_description"));
    if (model_view_.size() < 1)
    {
      std::cerr << "robot_description topic not found in bag file." << std::endl;
      return -1;
    }
    std_msgs::String::ConstPtr description_ = model_view_.begin()->instantiate<std_msgs::String>();
    description_msg = *description_;

    // Parse calibration_data topic
    rosbag::View data_view_(bag_, rosbag::TopicQuery("calibration_data"));
    BOOST_FOREACH (rosbag::MessageInstance const m, data_view_)
    {
      robot_calibration_msgs::CalibrationData::ConstPtr msg = m.instantiate<robot_calibration_msgs::CalibrationData>();
      data.push_back(*msg);
    }
  }

  // Create instance of optimizer
  robot_calibration::OptimizationParams params;
  params.LoadFromROS(nh);
  robot_calibration::Optimizer opt(description_msg.data);
  opt.optimize(params, data, verbose);
  if (verbose)
  {
    std::cout << "Parameter Offsets:" << std::endl;
    std::cout << opt.getOffsets()->getOffsetYAML() << std::endl;
  }

  // Update the URDF
  std::string s = opt.getOffsets()->updateURDF(description_msg.data);

  // Generate datecode
  char datecode[80];
  {
    std::time_t t = std::time(NULL);
    std::strftime(datecode, 80, "%Y_%m_%d_%H_%M_%S", std::localtime(&t));
  }

  // Save updated URDF
  {
    std::stringstream urdf_name;
    urdf_name << "/tmp/calibrated_" << datecode << ".urdf";
    std::ofstream file;
    file.open(urdf_name.str().c_str());
    file << s;
    file.close();
  }

  // Output camera calibration
  {
    std::stringstream depth_name;
    depth_name << "/tmp/depth_" << datecode << ".yaml";
    camera_calibration_parsers::writeCalibration(depth_name.str(), "",
        robot_calibration::updateCameraInfo(
                         opt.getOffsets()->get("camera_fx"),
                         opt.getOffsets()->get("camera_fy"),
                         opt.getOffsets()->get("camera_cx"),
                         opt.getOffsets()->get("camera_cy"),
                         data[0].observations[0].ext_camera_info.camera_info));  // TODO avoid hardcoding index

    std::stringstream rgb_name;
    rgb_name << "/tmp/rgb_" << datecode << ".yaml";
    camera_calibration_parsers::writeCalibration(rgb_name.str(), "",
        robot_calibration::updateCameraInfo(
                         opt.getOffsets()->get("camera_fx"),
                         opt.getOffsets()->get("camera_fy"),
                         opt.getOffsets()->get("camera_cx"),
                         opt.getOffsets()->get("camera_cy"),
                         data[0].observations[0].ext_camera_info.camera_info));  // TODO avoid hardcoding index
  }

  // Output the calibration yaml
  {
    std::stringstream yaml_name;
    yaml_name << "/tmp/calibration_" << datecode << ".yaml";
    std::ofstream file;
    file.open(yaml_name.str().c_str());
    file << opt.getOffsets()->getOffsetYAML();
    file << "depth_info: depth_" << datecode << ".yaml" << std::endl;
    file << "rgb_info: rgb_" << datecode << ".yaml" << std::endl;
    file << "urdf: calibrated_" << datecode << ".urdf" << std::endl;
    file.close();
  }

  ROS_INFO("Done calibrating");

  return 0;
}