Files
rtabmap_ros/src/CoreWrapper.cpp
T

3704 lines
115 KiB
C++
Raw Normal View History

/*
2016-07-17 22:06:11 -04:00
Copyright (c) 2010-2016, Mathieu Labbe - IntRoLab - Universite de Sherbrooke
2014-08-11 17:03:39 +00:00
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 the Universite de Sherbrooke 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 HOLDER 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.
*/
2016-07-27 16:44:48 -07:00
#include "rtabmap_ros/CoreWrapper.h"
2015-05-14 00:42:21 -04:00
#include <stdio.h>
#include <ros/ros.h>
#include "pluginlib/class_list_macros.h"
2015-01-30 15:31:44 -05:00
#include <nav_msgs/Path.h>
#include <std_msgs/Int32MultiArray.h>
2015-02-06 16:28:28 -05:00
#include <std_msgs/Bool.h>
#include <geometry_msgs/PoseArray.h>
#include <sensor_msgs/image_encodings.h>
#include <cv_bridge/cv_bridge.h>
#include <pcl_conversions/pcl_conversions.h>
2017-08-25 18:06:01 -04:00
#include <pcl/io/io.h>
2015-05-14 00:42:21 -04:00
#include <visualization_msgs/MarkerArray.h>
2015-05-14 00:42:21 -04:00
#include <rtabmap/utilite/UTimer.h>
#include <rtabmap/utilite/UDirectory.h>
2015-05-14 00:42:21 -04:00
#include <rtabmap/utilite/UFile.h>
#include <rtabmap/utilite/UConversion.h>
#include <rtabmap/utilite/UStl.h>
2015-05-30 20:08:20 -04:00
#include <rtabmap/utilite/UMath.h>
#include <rtabmap/core/util2d.h>
#include <rtabmap/core/util3d.h>
#include <rtabmap/core/util3d_transforms.h>
2017-08-25 18:06:01 -04:00
#include <rtabmap/core/util3d_surface.h>
2014-08-18 23:16:16 +00:00
#include <rtabmap/core/Memory.h>
2015-05-30 20:08:20 -04:00
#include <rtabmap/core/OdometryEvent.h>
2016-06-28 19:02:01 -04:00
#include <rtabmap/core/Version.h>
#include <rtabmap/core/OccupancyGrid.h>
#include <rtabmap/core/DBDriver.h>
#include <rtabmap/core/Registration.h>
2018-02-01 22:16:13 -05:00
#include <rtabmap/core/Graph.h>
2013-12-11 00:12:44 +00:00
2018-09-29 13:51:48 -04:00
#ifdef WITH_OCTOMAP_MSGS
2016-06-28 19:02:01 -04:00
#ifdef RTABMAP_OCTOMAP
2015-02-20 19:00:31 -05:00
#include <octomap_msgs/conversions.h>
2016-06-28 19:02:01 -04:00
#include <rtabmap/core/OctoMap.h>
#endif
2015-05-01 13:41:08 -04:00
#endif
#define BAD_COVARIANCE 9999
2015-02-20 19:00:31 -05:00
//msgs
2014-11-25 17:13:15 -05:00
#include "rtabmap_ros/Info.h"
#include "rtabmap_ros/MapData.h"
#include "rtabmap_ros/MapGraph.h"
2014-11-25 17:13:15 -05:00
#include "rtabmap_ros/GetMap.h"
#include "rtabmap_ros/PublishMap.h"
#include "rtabmap_ros/Path.h"
2013-12-11 00:12:44 +00:00
2014-11-25 17:13:15 -05:00
#include "rtabmap_ros/MsgConversion.h"
using namespace rtabmap;
namespace rtabmap_ros {
CoreWrapper::CoreWrapper() :
CommonDataSubscriber(false),
2013-12-11 00:12:44 +00:00
paused_(false),
lastPose_(Transform::getIdentity()),
lastPoseIntermediate_(false),
latestNodeWasReached_(false),
2013-12-11 00:12:44 +00:00
frameId_("base_link"),
odomFrameId_(""),
mapFrameId_("map"),
groundTruthFrameId_(""), // e.g., "world"
groundTruthBaseFrameId_(""), // e.g., "base_link_gt"
2013-12-11 00:12:44 +00:00
configPath_(""),
odomDefaultAngVariance_(0.001),
odomDefaultLinVariance_(0.001),
landmarkDefaultAngVariance_(0.001),
landmarkDefaultLinVariance_(0.001),
waitForTransform_(true),
2016-03-17 17:14:43 -04:00
waitForTransformDuration_(0.2), // 200 ms
useActionForGoal_(false),
useSavedMap_(true),
genScan_(false),
genScanMaxDepth_(4.0),
2016-04-12 15:18:29 -04:00
genScanMinDepth_(0.0),
scanCloudMaxPoints_(0),
2015-05-14 00:42:21 -04:00
mapToOdom_(rtabmap::Transform::getIdentity()),
2014-10-18 00:00:42 +00:00
transformThread_(0),
tfThreadRunning_(false),
stereoToDepth_(false),
odomSensorSync_(false),
2013-12-11 00:12:44 +00:00
rate_(Parameters::defaultRtabmapDetectionRate()),
createIntermediateNodes_(Parameters::defaultRtabmapCreateIntermediateNodes()),
2018-02-01 22:16:13 -05:00
maxMappingNodes_(Parameters::defaultGridGlobalMaxNodes()),
previousStamp_(0),
mbClient_(0)
{
2018-10-17 11:43:38 -04:00
char * rosHomePath = getenv("ROS_HOME");
std::string workingDir = rosHomePath?rosHomePath:UDirectory::homeDir()+"/.ros";
databasePath_ = workingDir+"/"+rtabmap::Parameters::getDefaultDatabaseName();
2017-05-24 16:03:47 -04:00
globalPose_.header.stamp = ros::Time(0);
}
void CoreWrapper::onInit()
{
ros::NodeHandle & nh = getNodeHandle();
ros::NodeHandle & pnh = getPrivateNodeHandle();
mapsManager_.init(nh, pnh, getName(), true);
bool publishTf = true;
2013-12-11 00:12:44 +00:00
double tfDelay = 0.05; // 20 Hz
double tfTolerance = 0.1; // 100 ms
2013-12-11 00:12:44 +00:00
pnh.param("config_path", configPath_, configPath_);
pnh.param("database_path", databasePath_, databasePath_);
2013-12-11 00:12:44 +00:00
pnh.param("frame_id", frameId_, frameId_);
pnh.param("odom_frame_id", odomFrameId_, odomFrameId_); // set to use odom from TF
pnh.param("map_frame_id", mapFrameId_, mapFrameId_);
pnh.param("ground_truth_frame_id", groundTruthFrameId_, groundTruthFrameId_);
pnh.param("ground_truth_base_frame_id", groundTruthBaseFrameId_, frameId_);
if(pnh.hasParam("depth_cameras") && !pnh.hasParam("depth_cameras"))
{
NODELET_ERROR("\"depth_cameras\" parameter doesn't exist "
"anymore! It is replaced by \"rgbd_cameras\" parameter "
"used when \"subscribe_rgbd\" is true");
}
2013-12-11 00:12:44 +00:00
pnh.param("publish_tf", publishTf, publishTf);
pnh.param("tf_delay", tfDelay, tfDelay);
if(pnh.hasParam("tf_prefix"))
{
ROS_ERROR("tf_prefix parameter has been removed, use directly map_frame_id, odom_frame_id and frame_id parameters.");
}
pnh.param("tf_tolerance", tfTolerance, tfTolerance);
pnh.param("odom_tf_angular_variance", odomDefaultAngVariance_, odomDefaultAngVariance_);
pnh.param("odom_tf_linear_variance", odomDefaultLinVariance_, odomDefaultLinVariance_);
pnh.param("landmark_angular_variance", landmarkDefaultAngVariance_, landmarkDefaultAngVariance_);
pnh.param("landmark_linear_variance", landmarkDefaultLinVariance_, landmarkDefaultLinVariance_);
pnh.param("wait_for_transform", waitForTransform_, waitForTransform_);
pnh.param("wait_for_transform_duration", waitForTransformDuration_, waitForTransformDuration_);
pnh.param("use_action_for_goal", useActionForGoal_, useActionForGoal_);
pnh.param("use_saved_map", useSavedMap_, useSavedMap_);
pnh.param("gen_scan", genScan_, genScan_);
pnh.param("gen_scan_max_depth", genScanMaxDepth_, genScanMaxDepth_);
2016-04-12 15:18:29 -04:00
pnh.param("gen_scan_min_depth", genScanMinDepth_, genScanMinDepth_);
pnh.param("scan_cloud_max_points", scanCloudMaxPoints_, scanCloudMaxPoints_);
if(pnh.hasParam("scan_cloud_normal_k"))
{
ROS_WARN("rtabmap: Parameter \"scan_cloud_normal_k\" has been removed. RTAB-Map's parameter \"%s\" should be used instead. "
"The value is copied. Use \"%s\" to avoid this warning.",
Parameters::kMemLaserScanNormalK().c_str(),
Parameters::kMemLaserScanNormalK().c_str());
double value;
pnh.getParam("scan_cloud_normal_k", value);
uInsert(parameters_, ParametersPair(Parameters::kMemLaserScanNormalK(), uNumber2Str(value)));
}
pnh.param("stereo_to_depth", stereoToDepth_, stereoToDepth_);
pnh.param("odom_sensor_sync", odomSensorSync_, odomSensorSync_);
if(pnh.hasParam("flip_scan"))
{
NODELET_WARN("Parameter \"flip_scan\" doesn't exist anymore. Rtabmap now "
"detects automatically if the laser is upside down with /tf, then if so, it "
"switches scan values.");
}
NODELET_INFO("rtabmap: frame_id = %s", frameId_.c_str());
if(!odomFrameId_.empty())
{
NODELET_INFO("rtabmap: odom_frame_id = %s", odomFrameId_.c_str());
}
if(!groundTruthFrameId_.empty())
{
NODELET_INFO("rtabmap: ground_truth_frame_id = %s -> ground_truth_base_frame_id = %s",
groundTruthFrameId_.c_str(),
groundTruthBaseFrameId_.c_str());
}
NODELET_INFO("rtabmap: map_frame_id = %s", mapFrameId_.c_str());
NODELET_INFO("rtabmap: use_action_for_goal = %s", useActionForGoal_?"true":"false");
NODELET_INFO("rtabmap: tf_delay = %f", tfDelay);
NODELET_INFO("rtabmap: tf_tolerance = %f", tfTolerance);
NODELET_INFO("rtabmap: odom_sensor_sync = %s", odomSensorSync_?"true":"false");
bool subscribeStereo = false;
pnh.param("subscribe_stereo", subscribeStereo, subscribeStereo);
if(subscribeStereo)
{
NODELET_INFO("rtabmap: stereo_to_depth = %s", stereoToDepth_?"true":"false");
}
2013-12-11 00:12:44 +00:00
2014-11-25 17:13:15 -05:00
infoPub_ = nh.advertise<rtabmap_ros::Info>("info", 1);
mapDataPub_ = nh.advertise<rtabmap_ros::MapData>("mapData", 1);
mapGraphPub_ = nh.advertise<rtabmap_ros::MapGraph>("mapGraph", 1);
landmarksPub_ = nh.advertise<geometry_msgs::PoseArray>("landmarks", 1);
labelsPub_ = nh.advertise<visualization_msgs::MarkerArray>("labels", 1);
2017-08-18 15:07:56 -04:00
mapPathPub_ = nh.advertise<nav_msgs::Path>("mapPath", 1);
localGridObstacle_ = nh.advertise<sensor_msgs::PointCloud2>("local_grid_obstacle", 1);
localGridEmpty_ = nh.advertise<sensor_msgs::PointCloud2>("local_grid_empty", 1);
localGridGround_ = nh.advertise<sensor_msgs::PointCloud2>("local_grid_ground", 1);
localizationPosePub_ = nh.advertise<geometry_msgs::PoseWithCovarianceStamped>("localization_pose", 1);
initialPoseSub_ = nh.subscribe("initialpose", 1, &CoreWrapper::initialPoseCallback, this);
2015-01-30 15:31:44 -05:00
// planning topics
goalSub_ = nh.subscribe("goal", 1, &CoreWrapper::goalCallback, this);
goalNodeSub_ = nh.subscribe("goal_node", 1, &CoreWrapper::goalNodeCallback, this);
nextMetricGoalPub_ = nh.advertise<geometry_msgs::PoseStamped>("goal_out", 1);
2015-02-06 16:28:28 -05:00
goalReachedPub_ = nh.advertise<std_msgs::Bool>("goal_reached", 1);
globalPathPub_ = nh.advertise<nav_msgs::Path>("global_path", 1);
localPathPub_ = nh.advertise<nav_msgs::Path>("local_path", 1);
globalPathNodesPub_ = nh.advertise<rtabmap_ros::Path>("global_path_nodes", 1);
localPathNodesPub_ = nh.advertise<rtabmap_ros::Path>("local_path_nodes", 1);
2015-01-30 15:31:44 -05:00
ros::Publisher nextMetricGoal_;
ros::Publisher goalReached_;
ros::Publisher path_;
configPath_ = uReplaceChar(configPath_, '~', UDirectory::homeDir());
databasePath_ = uReplaceChar(databasePath_, '~', UDirectory::homeDir());
if(configPath_.size() && configPath_.at(0) != '/')
{
configPath_ = UDirectory::currentDir(true) + configPath_;
}
if(databasePath_.size() && databasePath_.at(0) != '/')
{
databasePath_ = UDirectory::currentDir(true) + databasePath_;
}
ParametersMap allParameters = Parameters::getDefaultParameters();
// remove Odom parameters
for(ParametersMap::iterator iter=allParameters.begin(); iter!=allParameters.end();)
{
if(iter->first.find("Odom") == 0)
{
allParameters.erase(iter++);
}
else
{
++iter;
}
}
uInsert(allParameters, ParametersPair(Parameters::kRGBDCreateOccupancyGrid(), "true")); // default true in ROS
2018-10-17 11:43:38 -04:00
char * rosHomePath = getenv("ROS_HOME");
std::string workingDir = rosHomePath?rosHomePath:UDirectory::homeDir()+"/.ros";
uInsert(allParameters, ParametersPair(Parameters::kRtabmapWorkingDirectory(), workingDir)); // change default to ~/.ros
2013-12-11 00:12:44 +00:00
// load parameters
loadParameters(configPath_, parameters_);
for(ParametersMap::iterator iter=parameters_.begin(); iter!=parameters_.end();)
{
if(iter->first.find("Odom") == 0)
{
parameters_.erase(iter++);
}
else
{
++iter;
}
}
2013-12-11 00:12:44 +00:00
// update parameters with user input parameters (private)
for(ParametersMap::iterator iter=allParameters.begin(); iter!=allParameters.end(); ++iter)
2013-12-11 00:12:44 +00:00
{
std::string vStr;
bool vBool;
int vInt;
double vDouble;
if(pnh.getParam(iter->first, vStr))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), vStr.c_str());
2013-12-11 00:12:44 +00:00
if(iter->first.compare(Parameters::kRtabmapWorkingDirectory()) == 0)
{
vStr = uReplaceChar(vStr, '~', UDirectory::homeDir());
2013-12-11 00:12:44 +00:00
}
else if(iter->first.compare(Parameters::kKpDictionaryPath()) == 0)
{
vStr = uReplaceChar(vStr, '~', UDirectory::homeDir());
2013-12-11 00:12:44 +00:00
}
uInsert(parameters_, ParametersPair(iter->first, vStr));
2013-12-11 00:12:44 +00:00
}
else if(pnh.getParam(iter->first, vBool))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), uBool2Str(vBool).c_str());
uInsert(parameters_, ParametersPair(iter->first, uBool2Str(vBool)));
2013-12-11 00:12:44 +00:00
}
else if(pnh.getParam(iter->first, vDouble))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), uNumber2Str(vDouble).c_str());
uInsert(parameters_, ParametersPair(iter->first, uNumber2Str(vDouble)));
2013-12-11 00:12:44 +00:00
}
2015-01-21 13:57:57 -05:00
else if(pnh.getParam(iter->first, vInt))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), uNumber2Str(vInt).c_str());
uInsert(parameters_, ParametersPair(iter->first, uNumber2Str(vInt)));
2015-01-21 13:57:57 -05:00
}
2013-12-11 00:12:44 +00:00
}
2012-03-03 01:46:30 +00:00
//parse input arguments
std::vector<std::string> argList = getMyArgv();
char * argv[argList.size()];
bool deleteDbOnStart = false;
for(unsigned int i=0; i<argList.size(); ++i)
{
argv[i] = &argList[i].at(0);
if(strcmp(argv[i], "--delete_db_on_start") == 0 || strcmp(argv[i], "-d") == 0)
{
deleteDbOnStart = true;
}
}
rtabmap::ParametersMap parameters = rtabmap::Parameters::parseArguments(argList.size(), argv);
for(ParametersMap::const_iterator iter=parameters.begin(); iter!=parameters.end(); ++iter)
{
uInsert(parameters_, ParametersPair(iter->first, iter->second));
NODELET_INFO("Update RTAB-Map parameter \"%s\"=\"%s\" from arguments", iter->first.c_str(), iter->second.c_str());
}
// Backward compatibility
for(std::map<std::string, std::pair<bool, std::string> >::const_iterator iter=Parameters::getRemovedParameters().begin();
iter!=Parameters::getRemovedParameters().end();
++iter)
{
std::string vStr;
bool vBool;
int vInt;
double vDouble;
std::string paramValue;
if(pnh.getParam(iter->first, vStr))
{
paramValue = vStr;
}
else if(pnh.getParam(iter->first, vBool))
{
paramValue = uBool2Str(vBool);
}
else if(pnh.getParam(iter->first, vDouble))
{
paramValue = uNumber2Str(vDouble);
}
else if(pnh.getParam(iter->first, vInt))
{
paramValue = uNumber2Str(vInt);
}
if(!paramValue.empty())
{
if(iter->second.first)
{
// can be migrated
uInsert(parameters_, ParametersPair(iter->second.second, paramValue));
NODELET_WARN("Rtabmap: Parameter name changed: \"%s\" -> \"%s\". Please update your launch file accordingly. Value \"%s\" is still set to the new parameter name.",
iter->first.c_str(), iter->second.second.c_str(), paramValue.c_str());
}
else
{
if(iter->second.second.empty())
{
NODELET_ERROR("Rtabmap: Parameter \"%s\" doesn't exist anymore!",
iter->first.c_str());
}
else
{
NODELET_ERROR("Rtabmap: Parameter \"%s\" doesn't exist anymore! You may look at this similar parameter: \"%s\"",
iter->first.c_str(), iter->second.second.c_str());
}
2015-06-28 20:40:57 -04:00
}
}
}
// Backward compatibility (MapsManager)
mapsManager_.backwardCompatibilityParameters(pnh, parameters_);
bool subscribeScan2d = false;
bool subscribeScan3d = false;
pnh.param("subscribe_scan", subscribeScan2d, subscribeScan2d);
pnh.param("subscribe_scan_cloud", subscribeScan3d, subscribeScan3d);
if((subscribeScan2d || subscribeScan3d) && parameters_.find(Parameters::kGridFromDepth()) == parameters_.end())
{
NODELET_WARN("Setting \"%s\" parameter to false (default true) as \"subscribe_scan\" or \"subscribe_scan_cloud\" is "
"true. The occupancy grid map will be constructed from "
"laser scans. To get occupancy grid map from cloud projection, set \"%s\" "
"to true. To suppress this warning, "
"add <param name=\"%s\" type=\"string\" value=\"false\"/>",
Parameters::kGridFromDepth().c_str(),
Parameters::kGridFromDepth().c_str(),
Parameters::kGridFromDepth().c_str());
parameters_.insert(ParametersPair(Parameters::kGridFromDepth(), "false"));
}
if((subscribeScan2d || subscribeScan3d) && parameters_.find(Parameters::kGridRangeMax()) == parameters_.end())
{
NODELET_INFO("Setting \"%s\" parameter to 0 (default %f) as \"subscribe_scan\" or \"subscribe_scan_cloud\" is true.",
Parameters::kGridRangeMax().c_str(),
Parameters::defaultGridRangeMax());
parameters_.insert(ParametersPair(Parameters::kGridRangeMax(), "0"));
}
if(subscribeScan3d && parameters_.find(Parameters::kIcpPointToPlaneRadius()) == parameters_.end())
{
NODELET_INFO("Setting \"%s\" parameter to 0 (default %f) as \"subscribe_scan_cloud\" is true.",
Parameters::kIcpPointToPlaneRadius().c_str(),
Parameters::defaultIcpPointToPlaneRadius());
parameters_.insert(ParametersPair(Parameters::kIcpPointToPlaneRadius(), "0"));
}
int regStrategy = Parameters::defaultRegStrategy();
Parameters::parse(parameters_, Parameters::kRegStrategy(), regStrategy);
if(parameters_.find(Parameters::kRGBDProximityPathMaxNeighbors()) == parameters_.end() &&
(regStrategy == Registration::kTypeIcp || regStrategy == Registration::kTypeVisIcp))
{
if(subscribeScan2d)
{
NODELET_WARN("Setting \"%s\" parameter to 10 (default 0) as \"subscribe_scan\" is "
"true and \"%s\" uses ICP. Proximity detection by space will be also done by merging close "
"scans. To disable, set \"%s\" to 0. To suppress this warning, "
"add <param name=\"%s\" type=\"string\" value=\"10\"/>",
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRegStrategy().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str());
parameters_.insert(ParametersPair(Parameters::kRGBDProximityPathMaxNeighbors(), "10"));
}
else if(subscribeScan3d)
{
NODELET_WARN("Setting \"%s\" parameter to 1 (default 0) as \"subscribe_scan_cloud\" is "
"true and \"%s\" uses ICP. To disable, set \"%s\" to 0. To suppress this warning, "
"add <param name=\"%s\" type=\"string\" value=\"1\"/>",
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRegStrategy().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str());
parameters_.insert(ParametersPair(Parameters::kRGBDProximityPathMaxNeighbors(), "1"));
}
}
int estimationType = Parameters::defaultVisEstimationType();
Parameters::parse(parameters_, Parameters::kVisEstimationType(), estimationType);
int cameras = 0;
bool subscribeRGBD = false;
pnh.param("rgbd_cameras", cameras, cameras);
pnh.param("subscribe_rgbd", subscribeRGBD, subscribeRGBD);
if(subscribeRGBD && cameras> 1 && estimationType>0)
{
NODELET_WARN("Setting \"%s\" parameter to 0 (%d is not supported "
"for multi-cameras) as \"subscribe_rgbd\" is "
"true and \"rgbd_cameras\">1. Set \"%s\" to 0 to suppress this warning.",
Parameters::kVisEstimationType().c_str(),
estimationType,
Parameters::kVisEstimationType().c_str());
uInsert(parameters_, ParametersPair(Parameters::kVisEstimationType(), "0"));
}
// modify default parameters with those in the database
if(!deleteDbOnStart)
{
ParametersMap dbParameters;
rtabmap::DBDriver * driver = rtabmap::DBDriver::create();
if(driver->openConnection(databasePath_))
{
dbParameters = driver->getLastParameters(); // parameter migration is already done
}
delete driver;
for(ParametersMap::iterator iter=dbParameters.begin(); iter!=dbParameters.end(); ++iter)
{
if(iter->first.compare(Parameters::kRtabmapWorkingDirectory()) == 0)
{
// ignore working directory
continue;
}
if(parameters_.find(iter->first) == parameters_.end() &&
allParameters.find(iter->first) != allParameters.end() &&
allParameters.find(iter->first)->second.compare(iter->second) !=0)
{
NODELET_INFO("Update RTAB-Map parameter \"%s\"=\"%s\" from database", iter->first.c_str(), iter->second.c_str());
parameters_.insert(*iter);
}
}
}
ParametersMap modifiedParameters = parameters_;
// Add all other parameters (not copied if already exists)
parameters_.insert(allParameters.begin(), allParameters.end());
2013-12-11 00:12:44 +00:00
// set public parameters
2014-06-22 03:35:17 +00:00
nh.setParam("is_rtabmap_paused", paused_);
2015-03-20 15:17:01 -04:00
for(ParametersMap::iterator iter=parameters_.begin(); iter!=parameters_.end(); ++iter)
2013-12-11 00:12:44 +00:00
{
nh.setParam(iter->first, iter->second);
}
2015-03-20 15:17:01 -04:00
if(parameters_.find(Parameters::kRtabmapDetectionRate()) != parameters_.end())
2013-12-11 00:12:44 +00:00
{
Parameters::parse(parameters_, Parameters::kRtabmapDetectionRate(), rate_);
NODELET_INFO("RTAB-Map detection rate = %f Hz", rate_);
}
if(parameters_.find(Parameters::kRtabmapCreateIntermediateNodes()) != parameters_.end())
{
Parameters::parse(parameters_, Parameters::kRtabmapCreateIntermediateNodes(), createIntermediateNodes_);
if(createIntermediateNodes_)
{
NODELET_INFO("Create intermediate nodes");
if(rate_ == 0.0f)
{
NODELET_INFO("Subscribe to inter odom messges");
interOdomSub_ = nh.subscribe("inter_odom", 1, &CoreWrapper::interOdomCallback, this);
}
}
2013-12-11 00:12:44 +00:00
}
2018-02-01 22:16:13 -05:00
if(parameters_.find(Parameters::kGridGlobalMaxNodes()) != parameters_.end())
{
Parameters::parse(parameters_, Parameters::kGridGlobalMaxNodes(), maxMappingNodes_);
if(maxMappingNodes_>0)
{
NODELET_INFO("Max mapping nodes = %d", maxMappingNodes_);
}
}
2013-12-11 00:12:44 +00:00
if(paused_)
{
NODELET_WARN("Node paused... don't forget to call service \"resume\" to start rtabmap.");
}
if(deleteDbOnStart)
{
if(UFile::erase(databasePath_) == 0)
{
NODELET_INFO("rtabmap: Deleted database \"%s\" (--delete_db_on_start or -d are set).", databasePath_.c_str());
}
2013-12-11 00:12:44 +00:00
}
if(databasePath_.size())
{
NODELET_INFO("rtabmap: Using database from \"%s\" (%ld MB).", databasePath_.c_str(), UFile::length(databasePath_)/(1024*1024));
}
else
{
NODELET_INFO("rtabmap: database_path parameter not set, the map will not be saved.");
}
mapsManager_.setParameters(parameters_);
2013-12-11 00:12:44 +00:00
// Init RTAB-Map
2015-03-20 15:17:01 -04:00
rtabmap_.init(parameters_, databasePath_);
2013-12-11 00:12:44 +00:00
if(rtabmap_.getMemory() && useSavedMap_)
{
float xMin, yMin, gridCellSize;
cv::Mat map = rtabmap_.getMemory()->load2DMap(xMin, yMin, gridCellSize);
if(!map.empty())
{
NODELET_INFO("rtabmap: 2D occupancy grid map loaded (%dx%d).", map.cols, map.rows);
mapsManager_.set2DMap(map, xMin, yMin, gridCellSize, rtabmap_.getLocalOptimizedPoses(), rtabmap_.getMemory());
}
}
2015-07-26 17:46:34 -04:00
if(databasePath_.size() && rtabmap_.getMemory())
{
NODELET_INFO("rtabmap: Database version = \"%s\".", rtabmap_.getMemory()->getDatabaseVersion().c_str());
2015-07-26 17:46:34 -04:00
}
2013-12-11 00:12:44 +00:00
// setup services
updateSrv_ = nh.advertiseService("update_parameters", &CoreWrapper::updateRtabmapCallback, this);
resetSrv_ = nh.advertiseService("reset", &CoreWrapper::resetRtabmapCallback, this);
pauseSrv_ = nh.advertiseService("pause", &CoreWrapper::pauseRtabmapCallback, this);
resumeSrv_ = nh.advertiseService("resume", &CoreWrapper::resumeRtabmapCallback, this);
triggerNewMapSrv_ = nh.advertiseService("trigger_new_map", &CoreWrapper::triggerNewMapCallback, this);
2015-03-20 15:17:01 -04:00
backupDatabase_ = nh.advertiseService("backup", &CoreWrapper::backupDatabaseCallback, this);
2014-07-09 21:12:24 +00:00
setModeLocalizationSrv_ = nh.advertiseService("set_mode_localization", &CoreWrapper::setModeLocalizationCallback, this);
setModeMappingSrv_ = nh.advertiseService("set_mode_mapping", &CoreWrapper::setModeMappingCallback, this);
getMapDataSrv_ = nh.advertiseService("get_map_data", &CoreWrapper::getMapDataCallback, this);
getMapSrv_ = nh.advertiseService("get_map", &CoreWrapper::getMapCallback, this);
getProbMapSrv_ = nh.advertiseService("get_prob_map", &CoreWrapper::getProbMapCallback, this);
getGridMapSrv_ = nh.advertiseService("get_grid_map", &CoreWrapper::getGridMapCallback, this);
getProjMapSrv_ = nh.advertiseService("get_proj_map", &CoreWrapper::getProjMapCallback, this);
publishMapDataSrv_ = nh.advertiseService("publish_map", &CoreWrapper::publishMapCallback, this);
2018-07-09 14:53:21 -04:00
getPlanSrv_ = nh.advertiseService("get_plan", &CoreWrapper::getPlanCallback, this);
getPlanNodesSrv_ = nh.advertiseService("get_plan_nodes", &CoreWrapper::getPlanNodesCallback, this);
setGoalSrv_ = nh.advertiseService("set_goal", &CoreWrapper::setGoalCallback, this);
2015-06-21 20:17:31 -04:00
cancelGoalSrv_ = nh.advertiseService("cancel_goal", &CoreWrapper::cancelGoalCallback, this);
setLabelSrv_ = nh.advertiseService("set_label", &CoreWrapper::setLabelCallback, this);
listLabelsSrv_ = nh.advertiseService("list_labels", &CoreWrapper::listLabelsCallback, this);
2018-09-29 13:51:48 -04:00
#ifdef WITH_OCTOMAP_MSGS
2016-06-28 19:02:01 -04:00
#ifdef RTABMAP_OCTOMAP
2015-02-20 19:00:31 -05:00
octomapBinarySrv_ = nh.advertiseService("octomap_binary", &CoreWrapper::octomapBinaryCallback, this);
octomapFullSrv_ = nh.advertiseService("octomap_full", &CoreWrapper::octomapFullCallback, this);
2016-06-28 19:02:01 -04:00
#endif
2015-05-01 13:41:08 -04:00
#endif
//private services
setLogDebugSrv_ = pnh.advertiseService("log_debug", &CoreWrapper::setLogDebug, this);
setLogInfoSrv_ = pnh.advertiseService("log_info", &CoreWrapper::setLogInfo, this);
setLogWarnSrv_ = pnh.advertiseService("log_warning", &CoreWrapper::setLogWarn, this);
setLogErrorSrv_ = pnh.advertiseService("log_error", &CoreWrapper::setLogError, this);
2013-12-11 00:12:44 +00:00
2015-03-12 17:01:29 -04:00
int optimizeIterations = 0;
Parameters::parse(parameters_, Parameters::kOptimizerIterations(), optimizeIterations);
2015-03-12 17:01:29 -04:00
if(publishTf && optimizeIterations != 0)
2014-10-18 00:00:42 +00:00
{
tfThreadRunning_ = true;
transformThread_ = new boost::thread(boost::bind(&CoreWrapper::publishLoop, this, tfDelay, tfTolerance));
2014-10-18 00:00:42 +00:00
}
else if(publishTf)
2014-10-18 00:00:42 +00:00
{
NODELET_WARN("Graph optimization is disabled (%s=0), the tf between frame \"%s\" and odometry frame will not be published. You can safely ignore this warning if you are using map_optimizer node.",
Parameters::kOptimizerIterations().c_str(), mapFrameId_.c_str());
2014-10-18 00:00:42 +00:00
}
setupCallbacks(nh, pnh, getName()); // do it at the end
if(!this->isDataSubscribed())
{
bool isRGBD = uStr2Bool(parameters_.at(Parameters::kRGBDEnabled()).c_str());
if(isRGBD)
{
NODELET_WARN("ROS param subscribe_depth, subscribe_stereo and subscribe_rgbd are false, but RTAB-Map "
"parameter \"%s\" is true! Please set subscribe_depth, subscribe_stereo or subscribe_rgbd "
"to true to use rtabmap node for RGB-D SLAM, set \"%s\" to false for loop closure "
"detection on images-only or set subscribe_rgb to true to localize a single RGB camera against pre-built 3D map.",
Parameters::kRGBDEnabled().c_str(),
Parameters::kRGBDEnabled().c_str());
}
ros::NodeHandle rgb_nh(nh, "rgb");
ros::NodeHandle rgb_pnh(pnh, "rgb");
image_transport::ImageTransport rgb_it(rgb_nh);
image_transport::TransportHints hintsRgb("raw", ros::TransportHints(), rgb_pnh);
defaultSub_ = rgb_it.subscribe("image", 1, &CoreWrapper::defaultCallback, this);
NODELET_INFO("\n%s subscribed to:\n %s", getName().c_str(), defaultSub_.getTopic().c_str());
}
else if(!this->isSubscribedToDepth() &&
!this->isSubscribedToStereo() &&
!this->isSubscribedToRGBD() &&
!this->isSubscribedToRGB() &&
(this->isSubscribedToScan2d() || this->isSubscribedToScan3d() || this->isSubscribedToOdom()))
{
NODELET_WARN("There is no image subscription, bag-of-words loop closure detection will be disabled...");
int kpMaxFeatures = Parameters::defaultKpMaxFeatures();
int registrationStrategy = Parameters::defaultRegStrategy();
Parameters::parse(parameters_, Parameters::kKpMaxFeatures(), kpMaxFeatures);
Parameters::parse(parameters_, Parameters::kRegStrategy(), registrationStrategy);
bool updateParams = false;
if(kpMaxFeatures!= -1)
{
uInsert(parameters_, ParametersPair(Parameters::kKpMaxFeatures(), "-1"));
NODELET_WARN("Setting %s=-1 (bag-of-words disabled)", Parameters::kKpMaxFeatures().c_str());
updateParams = true;
}
if((this->isSubscribedToScan2d() || this->isSubscribedToScan3d()) && registrationStrategy != 1)
{
uInsert(parameters_, ParametersPair(Parameters::kRegStrategy(), "1"));
NODELET_WARN("Setting %s=1 (ICP)", Parameters::kRegStrategy().c_str());
updateParams = true;
if(modifiedParameters.find(Parameters::kRGBDProximityPathMaxNeighbors()) == modifiedParameters.end())
{
if(this->isSubscribedToScan2d())
{
NODELET_WARN("Setting \"%s\" parameter to 10 (default 0) as \"subscribe_scan\" is "
"true and \"%s\" uses ICP. Proximity detection by space will be also done by merging close "
"scans. To disable, set \"%s\" to 0. To suppress this warning, "
"add <param name=\"%s\" type=\"string\" value=\"10\"/>",
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRegStrategy().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str());
uInsert(parameters_, ParametersPair(Parameters::kRGBDProximityPathMaxNeighbors(), "10"));
}
else if(this->isSubscribedToScan3d())
{
NODELET_WARN("Setting \"%s\" parameter to 1 (default 0) as \"subscribe_scan_cloud\" is "
"true and \"%s\" uses ICP. To disable, set \"%s\" to 0. To suppress this warning, "
"add <param name=\"%s\" type=\"string\" value=\"1\"/>",
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRegStrategy().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str(),
Parameters::kRGBDProximityPathMaxNeighbors().c_str());
uInsert(parameters_, ParametersPair(Parameters::kRGBDProximityPathMaxNeighbors(), "1"));
}
}
}
if(updateParams)
{
rtabmap_.parseParameters(parameters_);
}
}
userDataAsyncSub_ = nh.subscribe("user_data_async", 1, &CoreWrapper::userDataAsyncCallback, this);
globalPoseAsyncSub_ = nh.subscribe("global_pose", 1, &CoreWrapper::globalPoseAsyncCallback, this);
gpsFixAsyncSub_ = nh.subscribe("gps/fix", 1, &CoreWrapper::gpsFixAsyncCallback, this);
#ifdef WITH_APRILTAG_ROS
tagDetectionsSub_ = nh.subscribe("tag_detections", 1, &CoreWrapper::tagDetectionsAsyncCallback, this);
#endif
imuSub_ = nh.subscribe("imu", 100, &CoreWrapper::imuAsyncCallback, this);
}
CoreWrapper::~CoreWrapper()
{
2013-12-11 00:12:44 +00:00
if(transformThread_)
{
tfThreadRunning_ = false;
2013-12-11 00:12:44 +00:00
transformThread_->join();
delete transformThread_;
}
this->saveParameters(configPath_);
2014-06-22 03:35:17 +00:00
ros::NodeHandle nh;
for(ParametersMap::iterator iter=parameters_.begin(); iter!=parameters_.end(); ++iter)
2014-06-22 03:35:17 +00:00
{
nh.deleteParam(iter->first);
}
nh.deleteParam("is_rtabmap_paused");
printf("rtabmap: Saving database/long-term memory... (located at %s)\n", databasePath_.c_str());
if(rtabmap_.getMemory())
{
// save the grid map
float xMin=0.0f, yMin=0.0f, gridCellSize = 0.05f;
cv::Mat pixels = mapsManager_.getGridMap(xMin, yMin, gridCellSize);
if(!pixels.empty())
{
printf("rtabmap: 2D occupancy grid map saved.\n");
rtabmap_.getMemory()->save2DMap(pixels, xMin, yMin, gridCellSize);
}
}
rtabmap_.close();
printf("rtabmap: Saving database/long-term memory...done! (located at %s, %ld MB)\n", databasePath_.c_str(), UFile::length(databasePath_)/(1024*1024));
delete mbClient_;
}
void CoreWrapper::loadParameters(const std::string & configFile, ParametersMap & parameters)
{
2013-12-11 00:12:44 +00:00
if(!configFile.empty())
{
NODELET_INFO("Loading parameters from %s", configFile.c_str());
2013-12-11 00:12:44 +00:00
if(!UFile::exists(configFile.c_str()))
{
NODELET_WARN("Config file doesn't exist! It will be generated...");
2013-12-11 00:12:44 +00:00
}
Parameters::readINI(configFile.c_str(), parameters);
}
}
2013-12-11 00:12:44 +00:00
void CoreWrapper::saveParameters(const std::string & configFile)
{
2013-12-11 00:12:44 +00:00
if(!configFile.empty())
{
2013-12-11 00:12:44 +00:00
printf("Saving parameters to %s\n", configFile.c_str());
2013-12-11 00:12:44 +00:00
if(!UFile::exists(configFile.c_str()))
{
2013-12-11 00:12:44 +00:00
printf("Config file doesn't exist, a new one will be created.\n");
}
Parameters::writeINI(configFile.c_str(), parameters_);
2013-12-11 00:12:44 +00:00
}
else
{
NODELET_INFO("Parameters are not saved! (No configuration file provided...)");
2013-12-11 00:12:44 +00:00
}
}
void CoreWrapper::publishLoop(double tfDelay, double tfTolerance)
2013-12-11 00:12:44 +00:00
{
if(tfDelay == 0)
return;
ros::Rate r(1.0 / tfDelay);
while(tfThreadRunning_)
2013-12-11 00:12:44 +00:00
{
if(!odomFrameId_.empty())
{
mapToOdomMutex_.lock();
ros::Time tfExpiration = ros::Time::now() + ros::Duration(tfTolerance);
2015-05-14 00:42:21 -04:00
geometry_msgs::TransformStamped msg;
msg.child_frame_id = odomFrameId_;
msg.header.frame_id = mapFrameId_;
msg.header.stamp = tfExpiration;
rtabmap_ros::transformToGeometryMsg(mapToOdom_, msg.transform);
tfBroadcaster_.sendTransform(msg);
2013-12-11 00:12:44 +00:00
mapToOdomMutex_.unlock();
}
r.sleep();
}
}
void CoreWrapper::defaultCallback(const sensor_msgs::ImageConstPtr & imageMsg)
{
if(!paused_)
{
ros::Time stamp = imageMsg->header.stamp;
if(stamp.toSec() == 0.0)
{
ROS_WARN("A null stamp has been detected in the input topic. Make sure the stamp is set.");
return;
}
2013-12-11 00:12:44 +00:00
if(rate_>0.0f)
{
if(previousStamp_.toSec() > 0.0 && stamp.toSec() > previousStamp_.toSec() && stamp - previousStamp_ < ros::Duration(1.0f/rate_))
2013-12-11 00:12:44 +00:00
{
return;
}
}
previousStamp_ = stamp;
2013-12-11 00:12:44 +00:00
2014-07-22 19:00:06 +00:00
if(!(imageMsg->encoding.compare(sensor_msgs::image_encodings::MONO8) ==0 ||
imageMsg->encoding.compare(sensor_msgs::image_encodings::MONO16) ==0 ||
imageMsg->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 ||
imageMsg->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0))
{
NODELET_ERROR("Input type must be image=mono8,mono16,rgb8,bgr8");
2014-07-22 19:00:06 +00:00
return;
}
cv_bridge::CvImageConstPtr ptrImage;
if(imageMsg->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0 ||
imageMsg->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0)
{
ptrImage = cv_bridge::toCvShare(imageMsg, "mono8");
}
else
{
ptrImage = cv_bridge::toCvShare(imageMsg, "bgr8");
}
2014-12-01 14:00:19 -05:00
// process data
UTimer timer;
if(rtabmap_.isIDsGenerated() || ptrImage->header.seq > 0)
{
if(!rtabmap_.process(ptrImage->image.clone(), ptrImage->header.seq))
{
NODELET_WARN("RTAB-Map could not process the data received! (ROS id = %d)", ptrImage->header.seq);
2014-12-01 14:00:19 -05:00
}
else
{
this->publishStats(ros::Time::now());
2014-12-01 14:00:19 -05:00
}
}
else if(!rtabmap_.isIDsGenerated())
{
NODELET_WARN("Ignoring received image because its sequence ID=0. Please "
2014-12-01 14:00:19 -05:00
"set \"Mem/GenerateIds\"=\"true\" to ignore ros generated sequence id. "
"Use only \"Mem/GenerateIds\"=\"false\" for once-time run of RTAB-Map and "
"when you need to have IDs output of RTAB-map synchronised with the source "
"image sequence ID.");
}
NODELET_INFO("rtabmap: Update rate=%fs, Limit=%fs, Processing time = %fs (%d local nodes)",
2014-12-01 14:00:19 -05:00
1.0f/rate_,
rtabmap_.getTimeThreshold()/1000.0f,
timer.ticks(),
rtabmap_.getWMSize()+rtabmap_.getSTMSize());
2013-12-11 00:12:44 +00:00
}
}
bool CoreWrapper::odomUpdate(const nav_msgs::OdometryConstPtr & odomMsg, ros::Time stamp)
{
if(!paused_)
{
Transform odom = rtabmap_ros::transformFromPoseMsg(odomMsg->pose.pose);
if(!odom.isNull())
{
Transform odomTF = rtabmap_ros::getTransform(odomMsg->header.frame_id, frameId_, stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(odomTF.isNull())
{
static bool shown = false;
if(!shown)
{
NODELET_WARN("We received odometry message, but we cannot get the "
"corresponding TF %s->%s at data stamp %fs (odom msg stamp is %fs). Make sure TF of odometry is "
"also published to get more accurate pose estimation. This "
"warning is only printed once.", odomMsg->header.frame_id.c_str(), frameId_.c_str(), stamp.toSec(), odomMsg->header.stamp.toSec());
shown = true;
}
stamp = odomMsg->header.stamp;
}
else
{
odom = odomTF;
}
}
2016-06-01 14:41:06 -04:00
if(!lastPose_.isIdentity() && !odom.isNull() && (odom.isIdentity() || (odomMsg->pose.covariance[0] >= BAD_COVARIANCE && odomMsg->twist.covariance[0] >= BAD_COVARIANCE)))
{
2016-06-01 14:41:06 -04:00
UWARN("Odometry is reset (identity pose or high variance (%f) detected). Increment map id!", MAX(odomMsg->pose.covariance[0], odomMsg->twist.covariance[0]));
rtabmap_.triggerNewMap();
2017-05-23 16:31:06 -04:00
covariance_ = cv::Mat();
}
lastPoseIntermediate_ = false;
lastPose_ = odom;
lastPoseStamp_ = stamp;
2016-06-01 14:41:06 -04:00
// Only update variance if odom is not null
if(!odom.isNull())
{
2017-05-23 16:31:06 -04:00
cv::Mat covariance;
2018-02-01 22:16:13 -05:00
double variance = odomMsg->twist.covariance[0];
if(variance == BAD_COVARIANCE || variance <= 0.0f)
{
//use the one of the pose
2017-05-23 16:31:06 -04:00
covariance = cv::Mat(6,6,CV_64FC1, (void*)odomMsg->pose.covariance.data()).clone();
covariance /= 2.0;
}
2017-05-23 16:31:06 -04:00
else
{
2017-05-23 16:31:06 -04:00
covariance = cv::Mat(6,6,CV_64FC1, (void*)odomMsg->twist.covariance.data()).clone();
}
if(uIsFinite(covariance.at<double>(0,0)) &&
covariance.at<double>(0,0) != 1.0 &&
covariance.at<double>(0,0)>0.0)
2016-06-01 14:41:06 -04:00
{
// Use largest covariance error (to be independent of the odometry frame rate)
if(covariance_.empty() || covariance.at<double>(0,0) > covariance_.at<double>(0,0))
2017-05-23 16:31:06 -04:00
{
covariance_ = covariance;
}
2016-06-01 14:41:06 -04:00
}
}
// Throttle
bool ignoreFrame = false;
if(stamp.toSec() == 0.0)
{
ROS_WARN("A null stamp has been detected in the input topics. Make sure the stamp in all input topics is set.");
ignoreFrame = true;
}
if(rate_>0.0f)
{
if(previousStamp_.toSec() > 0.0 && stamp.toSec() > previousStamp_.toSec() && stamp - previousStamp_ < ros::Duration(1.0f/rate_))
{
ignoreFrame = true;
}
}
if(ignoreFrame)
{
if(createIntermediateNodes_)
{
lastPoseIntermediate_ = true;
}
else
{
return false;
}
}
else if(!ignoreFrame)
{
previousStamp_ = stamp;
}
return true;
}
return false;
}
bool CoreWrapper::odomTFUpdate(const ros::Time & stamp)
{
if(!paused_)
{
// Odom TF ready?
Transform odom = rtabmap_ros::getTransform(odomFrameId_, frameId_, stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(odom.isNull())
{
return false;
}
if(!lastPose_.isIdentity() && odom.isIdentity())
{
UWARN("Odometry is reset (identity pose detected). Increment map id!");
rtabmap_.triggerNewMap();
2017-05-23 16:31:06 -04:00
covariance_ = cv::Mat();
}
lastPoseIntermediate_ = false;
lastPose_ = odom;
lastPoseStamp_ = stamp;
bool ignoreFrame = false;
if(stamp.toSec() == 0.0)
{
ROS_WARN("A null stamp has been detected in the input topics. Make sure the stamp in all input topics is set.");
ignoreFrame = true;
}
if(rate_>0.0f)
{
if(previousStamp_.toSec() > 0.0 && stamp.toSec() > previousStamp_.toSec() && stamp - previousStamp_ < ros::Duration(1.0f/rate_))
{
ignoreFrame = true;
}
}
if(ignoreFrame)
{
if(createIntermediateNodes_)
{
lastPoseIntermediate_ = true;
}
else
{
return false;
}
}
else if(!ignoreFrame)
{
previousStamp_ = stamp;
}
return true;
}
return false;
}
void CoreWrapper::commonDepthCallback(
const nav_msgs::OdometryConstPtr & odomMsg,
const rtabmap_ros::UserDataConstPtr & userDataMsg,
const std::vector<cv_bridge::CvImageConstPtr> & imageMsgs,
const std::vector<cv_bridge::CvImageConstPtr> & depthMsgs,
const std::vector<sensor_msgs::CameraInfo> & cameraInfoMsgs,
const sensor_msgs::LaserScanConstPtr& scan2dMsg,
const sensor_msgs::PointCloud2ConstPtr& scan3dMsg,
const rtabmap_ros::OdomInfoConstPtr& odomInfoMsg)
{
std::string odomFrameId = odomFrameId_;
if(odomMsg.get())
{
odomFrameId = odomMsg->header.frame_id;
if(scan2dMsg.get())
{
if(!odomUpdate(odomMsg, scan2dMsg->header.stamp))
{
return;
}
}
else if(scan3dMsg.get())
{
if(!odomUpdate(odomMsg, scan3dMsg->header.stamp))
{
return;
}
}
else if(imageMsgs.size() == 0 || imageMsgs[0].get() == 0 || !odomUpdate(odomMsg, imageMsgs[0]->header.stamp))
{
return;
}
}
else if(scan2dMsg.get())
{
if(!odomTFUpdate(scan2dMsg->header.stamp))
{
return;
}
}
else if(scan3dMsg.get())
{
if(!odomTFUpdate(scan3dMsg->header.stamp))
{
return;
}
}
else if(imageMsgs.size() == 0 || imageMsgs[0].get() == 0 || !odomTFUpdate(imageMsgs[0]->header.stamp))
{
return;
}
commonDepthCallbackImpl(odomFrameId,
userDataMsg,
imageMsgs,
depthMsgs,
cameraInfoMsgs,
scan2dMsg,
scan3dMsg,
odomInfoMsg);
}
void CoreWrapper::commonDepthCallbackImpl(
const std::string & odomFrameId,
const rtabmap_ros::UserDataConstPtr & userDataMsg,
const std::vector<cv_bridge::CvImageConstPtr> & imageMsgs,
const std::vector<cv_bridge::CvImageConstPtr> & depthMsgs,
const std::vector<sensor_msgs::CameraInfo> & cameraInfoMsgs,
const sensor_msgs::LaserScanConstPtr& scan2dMsg,
const sensor_msgs::PointCloud2ConstPtr& scan3dMsg,
const rtabmap_ros::OdomInfoConstPtr& odomInfoMsg)
{
cv::Mat rgb;
cv::Mat depth;
std::vector<rtabmap::CameraModel> cameraModels;
if(!rtabmap_ros::convertRGBDMsgs(
imageMsgs,
depthMsgs,
cameraInfoMsgs,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
rgb,
depth,
cameraModels,
tfListener_,
waitForTransform_?waitForTransformDuration_:0.0))
{
NODELET_ERROR("Could not convert rgb/depth msgs! Aborting rtabmap update...");
return;
}
2016-06-15 16:47:46 -04:00
UASSERT(uContains(parameters_, rtabmap::Parameters::kMemSaveDepth16Format()));
if(!depth.empty() && depth.type() == CV_32FC1 && uStr2Bool(parameters_.at(Parameters::kMemSaveDepth16Format())))
{
depth = rtabmap::util2d::cvtDepthFromFloat(depth);
static bool shown = false;
if(!shown)
{
NODELET_WARN("Save depth data to 16 bits format: depth type detected is "
"32FC1, use 16UC1 depth format to avoid this conversion "
"(or set parameter \"Mem/SaveDepth16Format=false\" to use "
"32bits format). This message is only printed once...");
shown = true;
}
}
LaserScan scan;
bool genMaxScanPts = 0;
if(scan2dMsg.get() == 0 && scan3dMsg.get() == 0 && !depth.empty() && genScan_)
{
2017-08-25 18:06:01 -04:00
pcl::PointCloud<pcl::PointXYZ>::Ptr scanCloud2d(new pcl::PointCloud<pcl::PointXYZ>);
*scanCloud2d = util3d::laserScanFromDepthImages(
depth,
cameraModels,
genScanMaxDepth_,
genScanMinDepth_);
genMaxScanPts += depth.cols;
scan = LaserScan(rtabmap::util3d::laserScan2dFromPointCloud(*scanCloud2d), 0, genScanMaxDepth_, LaserScan::kXY);
}
else if(scan2dMsg.get() != 0)
{
if(!rtabmap_ros::convertScanMsg(
scan2dMsg,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
scan,
tfListener_,
waitForTransform_?waitForTransformDuration_:0,
// backward compatibility, project 2D scan in /base_link frame
rtabmap_.getMemory() && uStrNumCmp(rtabmap_.getMemory()->getDatabaseVersion(), "0.11.10") < 0))
{
NODELET_ERROR("Could not convert laser scan msg! Aborting rtabmap update...");
return;
}
}
else if(scan3dMsg.get() != 0)
{
if(!rtabmap_ros::convertScan3dMsg(
scan3dMsg,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
scan,
tfListener_,
waitForTransform_?waitForTransformDuration_:0,
scanCloudMaxPoints_))
{
NODELET_ERROR("Could not convert 3d laser scan msg! Aborting rtabmap update...");
return;
}
}
cv::Mat userData;
if(userDataMsg.get())
{
userData = rtabmap_ros::userDataFromROS(*userDataMsg);
UScopeMutex lock(userDataMutex_);
if(!userData_.empty())
{
NODELET_WARN("Synchronized and asynchronized user data topics cannot be used at the same time. Async user data dropped!");
userData_ = cv::Mat();
}
}
else
{
UScopeMutex lock(userDataMutex_);
userData = userData_;
userData_ = cv::Mat();
}
SensorData data(
scan,
rgb,
depth,
cameraModels,
lastPoseIntermediate_?-1:imageMsgs[0]->header.seq,
rtabmap_ros::timestampFromROS(lastPoseStamp_),
userData);
OdometryInfo odomInfo;
if(odomInfoMsg.get())
{
odomInfo = odomInfoFromROS(*odomInfoMsg);
}
process(lastPoseStamp_,
data,
lastPose_,
odomFrameId,
covariance_,
odomInfo);
2017-05-23 16:31:06 -04:00
covariance_ = cv::Mat();
}
void CoreWrapper::commonStereoCallback(
const nav_msgs::OdometryConstPtr & odomMsg,
const rtabmap_ros::UserDataConstPtr & userDataMsg,
const cv_bridge::CvImageConstPtr& leftImageMsg,
const cv_bridge::CvImageConstPtr& rightImageMsg,
const sensor_msgs::CameraInfo& leftCamInfoMsg,
const sensor_msgs::CameraInfo& rightCamInfoMsg,
const sensor_msgs::LaserScanConstPtr& scan2dMsg,
const sensor_msgs::PointCloud2ConstPtr& scan3dMsg,
const rtabmap_ros::OdomInfoConstPtr& odomInfoMsg)
{
std::string odomFrameId = odomFrameId_;
if(odomMsg.get())
{
odomFrameId = odomMsg->header.frame_id;
if(scan2dMsg.get())
{
if(!odomUpdate(odomMsg, scan2dMsg->header.stamp))
{
return;
}
}
else if(scan3dMsg.get())
{
if(!odomUpdate(odomMsg, scan3dMsg->header.stamp))
{
return;
}
}
else if(leftImageMsg.get() == 0 || !odomUpdate(odomMsg, leftImageMsg->header.stamp))
{
return;
}
}
else if(scan2dMsg.get())
{
if(!odomTFUpdate(scan2dMsg->header.stamp))
{
return;
}
}
else if(scan3dMsg.get())
{
if(!odomTFUpdate(scan3dMsg->header.stamp))
{
return;
}
}
else if(leftImageMsg.get() == 0 || !odomTFUpdate(leftImageMsg->header.stamp))
{
return;
}
cv::Mat left;
cv::Mat right;
StereoCameraModel stereoModel;
if(!rtabmap_ros::convertStereoMsg(
leftImageMsg,
rightImageMsg,
leftCamInfoMsg,
rightCamInfoMsg,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
left,
right,
stereoModel,
tfListener_,
waitForTransform_?waitForTransformDuration_:0.0))
{
NODELET_ERROR("Could not convert stereo msgs! Aborting rtabmap update...");
return;
}
if(stereoToDepth_)
{
// cv::stereoBM() see "$ rosrun rtabmap_ros rtabmap --params | grep StereoBM" for parameters
cv::Mat disparity = rtabmap::util2d::disparityFromStereoImages(
left,
right,
parameters_);
if(disparity.empty())
{
NODELET_ERROR("Could not compute disparity image (\"stereo_to_depth\" is true)!");
return;
}
cv::Mat depth = rtabmap::util2d::depthFromDisparity(
disparity,
stereoModel.left().fx(),
stereoModel.baseline());
if(depth.empty())
{
NODELET_ERROR("Could not compute depth image (\"stereo_to_depth\" is true)!");
return;
}
UASSERT(depth.type() == CV_16UC1 || depth.type() == CV_32FC1);
// move to common depth callback
cv_bridge::CvImagePtr imgDepth(new cv_bridge::CvImage);
if(depth.type() == CV_16UC1)
{
imgDepth->encoding = sensor_msgs::image_encodings::TYPE_16UC1;
}
else // CV_32FC1
{
imgDepth->encoding = sensor_msgs::image_encodings::TYPE_32FC1;
}
imgDepth->image = depth;
imgDepth->header = leftImageMsg->header;
std::vector<cv_bridge::CvImageConstPtr> rgbImages(1);
std::vector<cv_bridge::CvImageConstPtr> depthImages(1);
std::vector<sensor_msgs::CameraInfo> cameraInfos(1);
rgbImages[0] = leftImageMsg;
depthImages[0] = imgDepth;
cameraInfos[0] = leftCamInfoMsg;
commonDepthCallbackImpl(odomFrameId, rtabmap_ros::UserDataConstPtr(), rgbImages, depthImages, cameraInfos, scan2dMsg, scan3dMsg, odomInfoMsg);
return;
}
LaserScan scan;
if(scan2dMsg.get() != 0)
{
if(!rtabmap_ros::convertScanMsg(
scan2dMsg,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
scan,
tfListener_,
waitForTransform_?waitForTransformDuration_:0,
// backward compatibility, project 2D scan in /base_link frame
rtabmap_.getMemory() && uStrNumCmp(rtabmap_.getMemory()->getDatabaseVersion(), "0.11.10") < 0))
{
NODELET_ERROR("Could not convert laser scan msg! Aborting rtabmap update...");
return;
}
}
else if(scan3dMsg.get() != 0)
{
if(!rtabmap_ros::convertScan3dMsg(
scan3dMsg,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
scan,
tfListener_,
waitForTransform_?waitForTransformDuration_:0,
scanCloudMaxPoints_))
{
NODELET_ERROR("Could not convert 3d laser scan msg! Aborting rtabmap update...");
return;
}
}
cv::Mat userData;
if(userDataMsg.get())
{
userData = rtabmap_ros::userDataFromROS(*userDataMsg);
UScopeMutex lock(userDataMutex_);
if(!userData_.empty())
{
NODELET_WARN("Synchronized and asynchronized user data topics cannot be used at the same time. Async user data dropped!");
userData_ = cv::Mat();
}
}
else
{
UScopeMutex lock(userDataMutex_);
userData = userData_;
userData_ = cv::Mat();
}
SensorData data(
scan,
left,
right,
stereoModel,
lastPoseIntermediate_?-1:leftImageMsg->header.seq,
rtabmap_ros::timestampFromROS(lastPoseStamp_),
userData);
OdometryInfo odomInfo;
if(odomInfoMsg.get())
{
odomInfo = odomInfoFromROS(*odomInfoMsg);
}
process(lastPoseStamp_,
data,
lastPose_,
odomFrameId,
covariance_,
odomInfo);
covariance_ = cv::Mat();
}
void CoreWrapper::commonLaserScanCallback(
const nav_msgs::OdometryConstPtr & odomMsg,
const rtabmap_ros::UserDataConstPtr & userDataMsg,
const sensor_msgs::LaserScanConstPtr& scan2dMsg,
const sensor_msgs::PointCloud2ConstPtr& scan3dMsg,
const rtabmap_ros::OdomInfoConstPtr& odomInfoMsg)
{
UASSERT(scan2dMsg.get() || scan3dMsg.get());
std::string odomFrameId = odomFrameId_;
if(odomMsg.get())
{
odomFrameId = odomMsg->header.frame_id;
if(scan2dMsg.get())
{
if(!odomUpdate(odomMsg, scan2dMsg->header.stamp))
{
return;
}
}
else if(scan3dMsg.get())
{
if(!odomUpdate(odomMsg, scan3dMsg->header.stamp))
{
return;
}
}
else
{
return;
}
}
else if(scan2dMsg.get())
{
if(!odomTFUpdate(scan2dMsg->header.stamp))
{
return;
}
}
else if(scan3dMsg.get())
{
if(!odomTFUpdate(scan3dMsg->header.stamp))
{
return;
}
}
else
{
return;
}
LaserScan scan;
if(scan2dMsg.get() != 0)
{
if(!rtabmap_ros::convertScanMsg(
scan2dMsg,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
scan,
tfListener_,
waitForTransform_?waitForTransformDuration_:0,
// backward compatibility, project 2D scan in /base_link frame
rtabmap_.getMemory() && uStrNumCmp(rtabmap_.getMemory()->getDatabaseVersion(), "0.11.10") < 0))
{
NODELET_ERROR("Could not convert laser scan msg! Aborting rtabmap update...");
return;
}
}
else if(scan3dMsg.get() != 0)
{
if(!rtabmap_ros::convertScan3dMsg(
scan3dMsg,
frameId_,
odomSensorSync_?odomFrameId:"",
lastPoseStamp_,
scan,
tfListener_,
waitForTransform_?waitForTransformDuration_:0,
scanCloudMaxPoints_))
{
NODELET_ERROR("Could not convert 3d laser scan msg! Aborting rtabmap update...");
return;
}
}
cv::Mat userData;
if(userDataMsg.get())
{
userData = rtabmap_ros::userDataFromROS(*userDataMsg);
UScopeMutex lock(userDataMutex_);
if(!userData_.empty())
{
NODELET_WARN("Synchronized and asynchronized user data topics cannot be used at the same time. Async user data dropped!");
userData_ = cv::Mat();
}
}
else
{
UScopeMutex lock(userDataMutex_);
userData = userData_;
userData_ = cv::Mat();
}
cv::Mat rgb = cv::Mat::zeros(2,1,CV_8UC1);
cv::Mat depth = cv::Mat::zeros(2,1,CV_16UC1);
CameraModel model(
1,
1,
0.5,
1,
scan.localTransform()*Transform(0,0,1,0, -1,0,0,0, 0,-1,0,0),
0,
cv::Size(1,2));
SensorData data(
scan,
rgb,
depth,
model,
lastPoseIntermediate_?-1:scan2dMsg.get() != 0?scan2dMsg->header.seq:scan3dMsg->header.seq,
rtabmap_ros::timestampFromROS(lastPoseStamp_),
userData);
OdometryInfo odomInfo;
if(odomInfoMsg.get())
{
odomInfo = odomInfoFromROS(*odomInfoMsg);
}
process(lastPoseStamp_,
data,
lastPose_,
odomFrameId,
covariance_,
odomInfo);
covariance_ = cv::Mat();
}
void CoreWrapper::commonOdomCallback(
const nav_msgs::OdometryConstPtr & odomMsg,
const rtabmap_ros::UserDataConstPtr & userDataMsg,
const rtabmap_ros::OdomInfoConstPtr& odomInfoMsg)
{
UASSERT(odomMsg.get());
std::string odomFrameId = odomFrameId_;
odomFrameId = odomMsg->header.frame_id;
if(!odomUpdate(odomMsg, odomMsg->header.stamp))
{
return;
}
cv::Mat userData;
if(userDataMsg.get())
{
userData = rtabmap_ros::userDataFromROS(*userDataMsg);
UScopeMutex lock(userDataMutex_);
if(!userData_.empty())
{
NODELET_WARN("Synchronized and asynchronized user data topics cannot be used at the same time. Async user data dropped!");
userData_ = cv::Mat();
}
}
else
{
UScopeMutex lock(userDataMutex_);
userData = userData_;
userData_ = cv::Mat();
}
cv::Mat rgb = cv::Mat::zeros(2,1,CV_8UC1);
cv::Mat depth = cv::Mat::zeros(2,1,CV_16UC1);
CameraModel model(
1,
1,
0.5,
1,
Transform(0,0,1,0, -1,0,0,0, 0,-1,0,0),
0,
cv::Size(1,2));
SensorData data(
rgb,
depth,
model,
lastPoseIntermediate_?-1:odomMsg->header.seq,
rtabmap_ros::timestampFromROS(lastPoseStamp_),
userData);
OdometryInfo odomInfo;
if(odomInfoMsg.get())
{
odomInfo = odomInfoFromROS(*odomInfoMsg);
}
process(lastPoseStamp_,
data,
lastPose_,
odomFrameId,
covariance_,
odomInfo);
2017-05-23 16:31:06 -04:00
covariance_ = cv::Mat();
}
2013-12-11 00:12:44 +00:00
void CoreWrapper::process(
const ros::Time & stamp,
SensorData & data,
2013-12-11 00:12:44 +00:00
const Transform & odom,
const std::string & odomFrameId,
const cv::Mat & odomCovariance,
const OdometryInfo & odomInfo)
2013-12-11 00:12:44 +00:00
{
UTimer timer;
if(rtabmap_.isIDsGenerated() || data.id() > 0)
2013-12-11 00:12:44 +00:00
{
// Add intermediate nodes?
for(std::list<nav_msgs::Odometry>::iterator iter=interOdoms_.begin(); iter!=interOdoms_.end();)
{
if(iter->header.stamp < lastPoseStamp_)
{
Transform interOdom = rtabmap_ros::transformFromPoseMsg(iter->pose.pose);
if(!interOdom.isNull())
{
cv::Mat covariance;
double variance = iter->twist.covariance[0];
if(variance == BAD_COVARIANCE || variance <= 0.0f)
{
//use the one of the pose
covariance = cv::Mat(6,6,CV_64FC1, (void*)iter->pose.covariance.data()).clone();
covariance /= 2.0;
}
else
{
covariance = cv::Mat(6,6,CV_64FC1, (void*)iter->twist.covariance.data()).clone();
}
if(!uIsFinite(covariance.at<double>(0,0)) || covariance.at<double>(0,0)<=0.0f)
{
covariance = cv::Mat::eye(6,6,CV_64FC1);
if(odomDefaultLinVariance_ > 0.0f)
{
covariance.at<double>(0,0) = odomDefaultLinVariance_;
covariance.at<double>(1,1) = odomDefaultLinVariance_;
covariance.at<double>(2,2) = odomDefaultLinVariance_;
}
if(odomDefaultAngVariance_ > 0.0f)
{
covariance.at<double>(3,3) = odomDefaultAngVariance_;
covariance.at<double>(4,4) = odomDefaultAngVariance_;
covariance.at<double>(5,5) = odomDefaultAngVariance_;
}
}
cv::Mat rgb = cv::Mat::zeros(2,1,CV_8UC1);
cv::Mat depth = cv::Mat::zeros(2,1,CV_16UC1);
CameraModel model(
1,
1,
0.5,
1,
Transform(0,0,1,0, -1,0,0,0, 0,-1,0,0),
0,
cv::Size(1,2));
SensorData interData(rgb, depth, model, -1, rtabmap_ros::timestampFromROS(iter->header.stamp));
Transform gt;
if(!groundTruthFrameId_.empty())
{
gt = rtabmap_ros::getTransform(groundTruthFrameId_, groundTruthBaseFrameId_, iter->header.stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
}
interData.setGroundTruth(gt);
rtabmap_.process(interData, interOdom, covariance);
}
interOdoms_.erase(iter++);
}
else if(iter->header.stamp == lastPoseStamp_)
{
interOdoms_.erase(iter++);
break;
}
else
{
break;
}
}
//Add async stuff
Transform groundTruthPose;
if(!groundTruthFrameId_.empty())
{
groundTruthPose = rtabmap_ros::getTransform(groundTruthFrameId_, groundTruthBaseFrameId_, lastPoseStamp_, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
}
data.setGroundTruth(groundTruthPose);
//global pose
if(!globalPose_.header.stamp.isZero())
{
// assume sensor is fixed
Transform sensorToBase = rtabmap_ros::getTransform(
globalPose_.header.frame_id,
frameId_,
lastPoseStamp_,
tfListener_,
waitForTransform_?waitForTransformDuration_:0.0);
if(!sensorToBase.isNull())
{
Transform globalPose = rtabmap_ros::transformFromPoseMsg(globalPose_.pose.pose);
globalPose *= sensorToBase; // transform global pose from sensor frame to robot base frame
// Correction of the global pose accounting the odometry movement since we received it
Transform correction = rtabmap_ros::getTransform(
frameId_,
odomFrameId,
globalPose_.header.stamp,
lastPoseStamp_,
tfListener_,
waitForTransform_?waitForTransformDuration_:0.0);
if(!correction.isNull())
{
globalPose *= correction;
}
else
{
NODELET_WARN("Could not adjust global pose accordingly to latest odometry pose. "
"If odometry is small since it received the global pose and "
"covariance is large, this should not be a problem.");
}
cv::Mat globalPoseCovariance = cv::Mat(6,6, CV_64FC1, (void*)globalPose_.pose.covariance.data()).clone();
data.setGlobalPose(globalPose, globalPoseCovariance);
}
}
globalPose_.header.stamp = ros::Time(0);
if(gps_.stamp() > 0.0)
{
data.setGPS(gps_);
}
gps_ = rtabmap::GPS();
//tag detections
Landmarks landmarks = rtabmap_ros::landmarksFromROS(
tags_,
frameId_,
odomFrameId,
lastPoseStamp_,
tfListener_,
waitForTransform_?waitForTransformDuration_:0,
landmarkDefaultLinVariance_,
landmarkDefaultAngVariance_);
tags_.clear();
if(!landmarks.empty())
{
data.setLandmarks(landmarks);
}
// IMU
if(!imus_.empty())
{
double stampDiff = 0.0;
Transform t = Transform::getClosestTransform(imus_, data.stamp(), &stampDiff);
if(!t.isNull() && stampDiff == 0.0)
{
Eigen::Quaterniond q = t.getQuaterniond();
data.setIMU(IMU(cv::Vec4d(q.x(), q.y(), q.z(), q.w()), cv::Mat::eye(3,3,CV_64FC1),
cv::Vec3d(), cv::Mat(),
cv::Vec3d(), cv::Mat(),
Transform::getIdentity()));
}
else
{
ROS_WARN("We are receiving imu data (buffer=%d), but cannot interpolate "
"imu transform at time %f (closest is at %f). IMU won't be added to graph.",
(int)imus_.size(), data.stamp(), stampDiff);
}
}
double timeRtabmap = 0.0;
2016-06-28 19:02:01 -04:00
double timeUpdateMaps = 0.0;
double timePublishMaps = 0.0;
2017-05-23 16:31:06 -04:00
cv::Mat covariance = odomCovariance;
if(covariance.empty() || !uIsFinite(covariance.at<double>(0,0)) || covariance.at<double>(0,0)<=0.0f)
{
covariance = cv::Mat::eye(6,6,CV_64FC1);
2017-05-23 16:31:06 -04:00
if(odomDefaultLinVariance_ > 0.0f)
{
covariance.at<double>(0,0) = odomDefaultLinVariance_;
covariance.at<double>(1,1) = odomDefaultLinVariance_;
covariance.at<double>(2,2) = odomDefaultLinVariance_;
}
if(odomDefaultAngVariance_ > 0.0f)
{
covariance.at<double>(3,3) = odomDefaultAngVariance_;
covariance.at<double>(4,4) = odomDefaultAngVariance_;
covariance.at<double>(5,5) = odomDefaultAngVariance_;
}
}
std::map<std::string, float> externalStats;
std::vector<float> odomVelocity;
if(odomInfo.timeEstimation != 0.0f)
{
externalStats.insert(std::make_pair("Odometry/LocalBundle/ms", odomInfo.localBundleTime*1000.0f));
externalStats.insert(std::make_pair("Odometry/LocalBundleConstraints/", odomInfo.localBundleConstraints));
externalStats.insert(std::make_pair("Odometry/LocalBundleOutliers/", odomInfo.localBundleOutliers));
externalStats.insert(std::make_pair("Odometry/TotalTime/ms", odomInfo.timeEstimation*1000.0f));
externalStats.insert(std::make_pair("Odometry/Registration/ms", odomInfo.reg.totalTime*1000.0f));
float speed = 0.0f;
if(odomInfo.interval>0.0)
speed = odomInfo.transform.x()/odomInfo.interval*3.6;
externalStats.insert(std::make_pair("Odometry/Speed/kph", speed));
externalStats.insert(std::make_pair("Odometry/Inliers/", odomInfo.reg.inliers));
externalStats.insert(std::make_pair("Odometry/Features/", odomInfo.features));
externalStats.insert(std::make_pair("Odometry/DistanceTravelled/m", odomInfo.distanceTravelled));
externalStats.insert(std::make_pair("Odometry/KeyFrameAdded/", odomInfo.keyFrameAdded));
externalStats.insert(std::make_pair("Odometry/LocalKeyFrames/", odomInfo.localKeyFrames));
externalStats.insert(std::make_pair("Odometry/LocalMapSize/", odomInfo.localMapSize));
externalStats.insert(std::make_pair("Odometry/LocalScanMapSize/", odomInfo.localScanMapSize));
2018-02-01 22:16:13 -05:00
externalStats.insert(std::make_pair("Odometry/RAM_usage/MB", odomInfo.memoryUsage));
if(odomInfo.interval>0.0)
{
odomVelocity.resize(6);
float x,y,z,roll,pitch,yaw;
odomInfo.transform.getTranslationAndEulerAngles(x,y,z,roll,pitch,yaw);
odomVelocity[0] = x/odomInfo.interval;
odomVelocity[1] = y/odomInfo.interval;
odomVelocity[2] = z/odomInfo.interval;
odomVelocity[3] = roll/odomInfo.interval;
odomVelocity[4] = pitch/odomInfo.interval;
odomVelocity[5] = yaw/odomInfo.interval;
}
}
2018-02-01 22:16:13 -05:00
if(rtabmapROSStats_.size())
{
externalStats.insert(rtabmapROSStats_.begin(), rtabmapROSStats_.end());
rtabmapROSStats_.clear();
}
if(rtabmap_.process(data, odom, covariance, odomVelocity, externalStats))
2013-12-11 00:12:44 +00:00
{
timeRtabmap = timer.ticks();
2013-12-11 00:12:44 +00:00
mapToOdomMutex_.lock();
2015-05-14 00:42:21 -04:00
mapToOdom_ = rtabmap_.getMapCorrection();
2013-12-11 00:12:44 +00:00
odomFrameId_ = odomFrameId;
mapToOdomMutex_.unlock();
if(data.id() < 0)
{
NODELET_INFO("Intermediate node added");
}
else
{
// Publish local graph, info
this->publishStats(stamp);
if(localizationPosePub_.getNumSubscribers() &&
!rtabmap_.getStatistics().localizationCovariance().empty())
{
geometry_msgs::PoseWithCovarianceStamped poseMsg;
poseMsg.header.frame_id = mapFrameId_;
poseMsg.header.stamp = stamp;
rtabmap_ros::transformToPoseMsg(mapToOdom_*odom, poseMsg.pose.pose);
poseMsg.pose.covariance;
const cv::Mat & cov = rtabmap_.getStatistics().localizationCovariance();
memcpy(poseMsg.pose.covariance.data(), cov.data, cov.total()*sizeof(double));
localizationPosePub_.publish(poseMsg);
}
std::map<int, rtabmap::Transform> filteredPoses(rtabmap_.getLocalOptimizedPoses().lower_bound(1), rtabmap_.getLocalOptimizedPoses().end());
// create a tmp signature with latest sensory data if latest signature was ignored
std::map<int, rtabmap::Signature> tmpSignature;
if(rtabmap_.getMemory() == 0 ||
filteredPoses.size() == 0 ||
rtabmap_.getMemory()->getLastSignatureId() != filteredPoses.rbegin()->first ||
rtabmap_.getMemory()->getLastWorkingSignature() == 0 ||
rtabmap_.getMemory()->getLastWorkingSignature()->sensorData().gridCellSize() == 0 ||
2018-02-16 20:00:32 -05:00
(!mapsManager_.getOccupancyGrid()->isGridFromDepth() && data.laserScanRaw().is2d())) // 2d laser scan would fill empty space for latest data
{
SensorData tmpData = data;
tmpData.setId(0);
tmpSignature.insert(std::make_pair(0, Signature(0, -1, 0, data.stamp(), "", odom, Transform(), tmpData)));
filteredPoses.insert(std::make_pair(0, mapToOdom_*odom));
2018-02-01 22:16:13 -05:00
}
if(maxMappingNodes_ > 0 && filteredPoses.size()>1)
{
std::map<int, Transform> nearestPoses;
std::vector<int> nodes = graph::findNearestNodes(filteredPoses, mapToOdom_*odom, maxMappingNodes_);
for(std::vector<int>::iterator iter=nodes.begin(); iter!=nodes.end(); ++iter)
{
std::map<int, Transform>::iterator pter = filteredPoses.find(*iter);
if(pter != filteredPoses.end())
{
nearestPoses.insert(*pter);
}
}
//add latest/zero and make sure those on a planned path are not filtered
2018-02-01 22:16:13 -05:00
std::set<int> onPath;
if(rtabmap_.getPath().size())
{
std::vector<int> nextNodes = rtabmap_.getPathNextNodes();
onPath.insert(nextNodes.begin(), nextNodes.end());
}
for(std::map<int, Transform>::iterator iter=filteredPoses.begin(); iter!=filteredPoses.end(); ++iter)
{
if(iter->first == 0 || onPath.find(iter->first) != onPath.end())
2018-02-01 22:16:13 -05:00
{
nearestPoses.insert(*iter);
}
else if(onPath.empty())
{
break;
}
}
filteredPoses = nearestPoses;
}
// Update maps
filteredPoses = mapsManager_.updateMapCaches(
filteredPoses,
rtabmap_.getMemory(),
false,
false,
tmpSignature);
2016-06-28 19:02:01 -04:00
timeUpdateMaps = timer.ticks();
mapsManager_.publishMaps(filteredPoses, stamp, mapFrameId_);
// update goal if planning is enabled
if(!currentMetricGoal_.isNull())
{
if(rtabmap_.getPath().size() == 0)
{
// Don't send status yet if move_base actionlib is used unless it failed,
// let move_base finish reaching the goal
if(mbClient_ == 0 || rtabmap_.getPathStatus() <= 0)
{
if(rtabmap_.getPathStatus() > 0)
{
// Goal reached
NODELET_INFO("Planning: Publishing goal reached!");
}
else if(rtabmap_.getPathStatus() <= 0)
{
NODELET_WARN("Planning: Plan failed!");
if(mbClient_ && mbClient_->isServerConnected())
{
mbClient_->cancelGoal();
}
}
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = rtabmap_.getPathStatus() > 0;
goalReachedPub_.publish(result);
}
currentMetricGoal_.setNull();
lastPublishedMetricGoal_.setNull();
goalFrameId_.clear();
latestNodeWasReached_ = false;
}
}
else
{
currentMetricGoal_ = rtabmap_.getPose(rtabmap_.getPathCurrentGoalId());
if(!currentMetricGoal_.isNull())
{
// Adjust the target pose relative to last node
if(rtabmap_.getPathCurrentGoalId() == rtabmap_.getPath().back().first && rtabmap_.getLocalOptimizedPoses().size())
{
if(latestNodeWasReached_ ||
rtabmap_.getLastLocalizationPose().getDistance(currentMetricGoal_) < rtabmap_.getLocalRadius())
{
latestNodeWasReached_ = true;
Transform goalLocalTransform = Transform::getIdentity();
if(!goalFrameId_.empty() && goalFrameId_.compare(frameId_) != 0)
{
Transform localT = rtabmap_ros::getTransform(frameId_, goalFrameId_, ros::Time::now(), tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(!localT.isNull())
{
goalLocalTransform = localT.inverse().to3DoF();
}
}
currentMetricGoal_ *= rtabmap_.getPathTransformToGoal()*goalLocalTransform;
}
}
// publish next goal with updated currentMetricGoal_
publishCurrentGoal(stamp);
// publish local path
publishLocalPath(stamp);
// publish global path
publishGlobalPath(stamp);
}
else
{
NODELET_ERROR("Planning: Local map broken, current goal id=%d (the robot may have moved to far from planned nodes)",
rtabmap_.getPathCurrentGoalId());
rtabmap_.clearPath(-1);
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = false;
goalReachedPub_.publish(result);
}
currentMetricGoal_.setNull();
2017-06-01 13:08:28 -04:00
lastPublishedMetricGoal_.setNull();
goalFrameId_.clear();
latestNodeWasReached_ = false;
}
}
}
2016-06-28 19:02:01 -04:00
timePublishMaps = timer.ticks();
}
}
else
{
timeRtabmap = timer.ticks();
}
NODELET_INFO("rtabmap (%d): Rate=%.2fs, Limit=%.3fs, RTAB-Map=%.4fs, Maps update=%.4fs pub=%.4fs (local map=%d, WM=%d)",
rtabmap_.getLastLocationId(),
rate_>0?1.0f/rate_:0,
rtabmap_.getTimeThreshold()/1000.0f,
timeRtabmap,
2016-06-28 19:02:01 -04:00
timeUpdateMaps,
timePublishMaps,
(int)rtabmap_.getLocalOptimizedPoses().size(),
rtabmap_.getWMSize()+rtabmap_.getSTMSize());
2018-02-01 22:16:13 -05:00
rtabmapROSStats_.insert(std::make_pair(std::string("RtabmapROS/HasSubscribers/"), mapsManager_.hasSubscribers()?1:0));
rtabmapROSStats_.insert(std::make_pair(std::string("RtabmapROS/TimeRtabmap/ms"), timeRtabmap*1000.0f));
rtabmapROSStats_.insert(std::make_pair(std::string("RtabmapROS/TimeUpdatingMaps/ms"), timeUpdateMaps*1000.0f));
rtabmapROSStats_.insert(std::make_pair(std::string("RtabmapROS/TimePublishing/ms"), timePublishMaps*1000.0f));
rtabmapROSStats_.insert(std::make_pair(std::string("RtabmapROS/TimeTotal/ms"), (timeRtabmap+timeUpdateMaps+timePublishMaps)*1000.0f));
}
2013-12-11 00:12:44 +00:00
else if(!rtabmap_.isIDsGenerated())
2012-03-03 01:46:30 +00:00
{
NODELET_WARN("Ignoring received image because its sequence ID=0. Please "
2013-12-11 00:12:44 +00:00
"set \"Mem/GenerateIds\"=\"true\" to ignore ros generated sequence id. "
"Use only \"Mem/GenerateIds\"=\"false\" for once-time run of RTAB-Map and "
"when you need to have IDs output of RTAB-map synchronized with the source "
2013-12-11 00:12:44 +00:00
"image sequence ID.");
2012-03-03 01:46:30 +00:00
}
}
void CoreWrapper::userDataAsyncCallback(const rtabmap_ros::UserDataConstPtr & dataMsg)
{
if(!paused_)
{
UScopeMutex lock(userDataMutex_);
static bool warningShow = false;
if(!userData_.empty() && !warningShow)
{
ROS_WARN("Overwriting previous user data set. When asynchronous user "
"data input topic rate is higher than "
"map update rate (current %s=%f), only latest data is saved "
"in the next node created. This message will is shown only once.",
Parameters::kRtabmapDetectionRate().c_str(), rate_);
warningShow = true;
}
userData_ = rtabmap_ros::userDataFromROS(*dataMsg);
}
}
void CoreWrapper::globalPoseAsyncCallback(const geometry_msgs::PoseWithCovarianceStampedConstPtr & globalPoseMsg)
{
if(!paused_)
{
globalPose_ = *globalPoseMsg;
}
}
void CoreWrapper::gpsFixAsyncCallback(const sensor_msgs::NavSatFixConstPtr & gpsFixMsg)
{
if(!paused_)
{
double error = 10.0;
if(gpsFixMsg->position_covariance_type != sensor_msgs::NavSatFix::COVARIANCE_TYPE_UNKNOWN)
{
double variance = uMax3(gpsFixMsg->position_covariance.at(0), gpsFixMsg->position_covariance.at(4), gpsFixMsg->position_covariance.at(8));
if(variance>0.0)
{
error = sqrt(variance);
}
}
gps_ = rtabmap::GPS(
gpsFixMsg->header.stamp.toSec(),
gpsFixMsg->longitude,
gpsFixMsg->latitude,
gpsFixMsg->altitude,
error,
0);
}
}
#ifdef WITH_APRILTAG_ROS
void CoreWrapper::tagDetectionsAsyncCallback(const apriltag_ros::AprilTagDetectionArray & tagDetections)
{
if(!paused_)
{
for(unsigned int i=0; i<tagDetections.detections.size(); ++i)
{
2018-12-10 13:17:37 -05:00
if(tagDetections.detections[i].id.size() >= 1)
{
2018-12-10 20:22:02 -05:00
geometry_msgs::PoseWithCovarianceStamped p = tagDetections.detections[i].pose;
p.header = tagDetections.header;
if(!tagDetections.detections[i].pose.header.frame_id.empty())
{
p.header.frame_id = tagDetections.detections[i].pose.header.frame_id;
}
if(!tagDetections.detections[i].pose.header.stamp.isZero())
{
p.header.stamp = tagDetections.detections[i].pose.header.stamp;
}
2018-12-10 20:22:02 -05:00
uInsert(tags_, std::make_pair(tagDetections.detections[i].id[0], p));
}
}
}
}
#endif
void CoreWrapper::imuAsyncCallback(const sensor_msgs::ImuConstPtr & msg)
{
if(!paused_)
{
if(msg->orientation.x == 0 && msg->orientation.y == 0 && msg->orientation.z == 0 && msg->orientation.w == 0)
{
UERROR("IMU received doesn't have orientation set, it is ignored.");
}
else
{
double stamp = msg->header.stamp.toSec();
rtabmap::Transform localTransform = rtabmap::Transform::getIdentity();
if(frameId_.compare(msg->header.frame_id) != 0)
{
localTransform = getTransform(frameId_, msg->header.frame_id, msg->header.stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(localTransform.isNull())
{
return;
}
}
Transform orientation(0,0,0, msg->orientation.x, msg->orientation.y, msg->orientation.z, msg->orientation.w);
imus_.insert(std::make_pair(msg->header.stamp.toSec(), orientation*localTransform.inverse()));
if(imus_.size() > 1000)
{
imus_.erase(imus_.begin());
}
}
}
}
void CoreWrapper::interOdomCallback(const nav_msgs::OdometryConstPtr & msg)
{
if(!paused_)
{
interOdoms_.push_back(*msg);
}
}
void CoreWrapper::initialPoseCallback(const geometry_msgs::PoseWithCovarianceStampedConstPtr & msg)
{
Transform intialPose = rtabmap_ros::transformFromPoseMsg(msg->pose.pose);
if(intialPose.isNull())
{
NODELET_ERROR("Pose received is null!");
return;
}
rtabmap_.setInitialPose(intialPose);
}
void CoreWrapper::goalCommonCallback(
int id,
const std::string & label,
const std::string & frameId,
const Transform & pose,
const ros::Time & stamp,
double * planningTime)
2015-01-30 15:31:44 -05:00
{
UTimer timer;
if(id == 0 && !label.empty() && rtabmap_.getMemory())
2015-01-30 15:31:44 -05:00
{
id = rtabmap_.getMemory()->getSignatureIdByLabel(label);
}
if(id > 0)
{
NODELET_INFO("Planning: set goal to node %d", id);
}
else if(id < 0)
{
NODELET_INFO("Planning: set goal to landmark %d", id);
}
else if(!pose.isNull())
{
NODELET_INFO("Planning: set goal %s", pose.prettyPrint().c_str());
}
if(planningTime)
{
*planningTime = 0.0;
}
bool success = false;
if((id != 0 && rtabmap_.computePath(id, true)) ||
(!pose.isNull() && rtabmap_.computePath(pose)))
{
if(planningTime)
{
*planningTime = timer.elapsed();
}
NODELET_INFO("Planning: Time computing path = %f s", timer.ticks());
const std::vector<std::pair<int, Transform> > & poses = rtabmap_.getPath();
currentMetricGoal_.setNull();
2017-06-01 13:08:28 -04:00
lastPublishedMetricGoal_.setNull();
goalFrameId_.clear();
latestNodeWasReached_ = false;
if(poses.size() == 0)
2015-01-30 15:31:44 -05:00
{
NODELET_WARN("Planning: Goal already reached (RGBD/GoalReachedRadius=%fm).",
rtabmap_.getGoalReachedRadius());
rtabmap_.clearPath(1);
2015-02-06 16:28:28 -05:00
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = true;
2015-02-06 16:28:28 -05:00
goalReachedPub_.publish(result);
}
success = true;
2015-01-30 15:31:44 -05:00
}
else
{
currentMetricGoal_ = rtabmap_.getPose(rtabmap_.getPathCurrentGoalId());
if(!currentMetricGoal_.isNull())
{
NODELET_INFO("Planning: Path successfully created (size=%d)", (int)poses.size());
goalFrameId_ = frameId;
// Adjust the target pose relative to last node
if(rtabmap_.getPathCurrentGoalId() == rtabmap_.getPath().back().first && rtabmap_.getLocalOptimizedPoses().size())
{
if(rtabmap_.getLastLocalizationPose().getDistance(currentMetricGoal_) < rtabmap_.getLocalRadius())
{
latestNodeWasReached_ = true;
Transform goalLocalTransform = Transform::getIdentity();
if(!goalFrameId_.empty() && goalFrameId_.compare(frameId_) != 0)
{
Transform localT = rtabmap_ros::getTransform(frameId_, goalFrameId_, stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(!localT.isNull())
{
goalLocalTransform = localT.inverse().to3DoF();
}
}
currentMetricGoal_ *= rtabmap_.getPathTransformToGoal() * goalLocalTransform;
}
}
publishCurrentGoal(stamp);
publishLocalPath(stamp);
publishGlobalPath(stamp);
// Just output the path on screen
std::stringstream stream;
for(std::vector<std::pair<int, Transform> >::const_iterator iter=poses.begin(); iter!=poses.end(); ++iter)
{
if(iter != poses.begin())
{
stream << " ";
}
stream << iter->first;
}
NODELET_INFO("Global path: [%s]", stream.str().c_str());
success=true;
}
else
{
NODELET_ERROR("Pose of node %d not found!? Cannot send a metric goal...", rtabmap_.getPathCurrentGoalId());
}
2015-01-30 15:31:44 -05:00
}
}
else if(!label.empty())
{
NODELET_ERROR("Planning: Node with label \"%s\" not found!", label.c_str());
}
else if(pose.isNull())
{
if(id > 0)
{
NODELET_ERROR("Planning: Could not plan to node %d! The node is not in map's graph (look for warnings before this message for more details).", id);
}
else if(id < 0)
{
NODELET_ERROR("Planning: Could not plan to landmark %d! The landmark is not in map's graph (look for warnings before this message for more details).", id);
}
else
{
NODELET_ERROR("Planning: Node id should be > 0 !");
}
}
2015-01-30 15:31:44 -05:00
else
{
NODELET_ERROR("Planning: A node near the goal's pose not found! The pose may be to far from the graph (RGBD/LocalRadius=%f m)", rtabmap_.getLocalRadius());
}
if(!success)
{
rtabmap_.clearPath(-1);
2015-02-06 16:28:28 -05:00
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = false;
2015-02-06 16:28:28 -05:00
goalReachedPub_.publish(result);
}
2015-01-30 15:31:44 -05:00
}
}
void CoreWrapper::goalCallback(const geometry_msgs::PoseStampedConstPtr & msg)
{
Transform targetPose = rtabmap_ros::transformFromPoseMsg(msg->pose);
if(targetPose.isNull())
{
NODELET_ERROR("Pose received is null!");
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = false;
goalReachedPub_.publish(result);
}
return;
}
// transform goal in /map frame
if(mapFrameId_.compare(msg->header.frame_id) != 0)
{
Transform t = rtabmap_ros::getTransform(mapFrameId_, msg->header.frame_id, msg->header.stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(t.isNull())
{
NODELET_ERROR("Cannot transform goal pose from \"%s\" frame to \"%s\" frame!",
msg->header.frame_id.c_str(), mapFrameId_.c_str());
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = false;
goalReachedPub_.publish(result);
}
return;
}
targetPose = t * targetPose;
}
goalCommonCallback(0, "", "", targetPose, msg->header.stamp);
}
void CoreWrapper::goalNodeCallback(const rtabmap_ros::GoalConstPtr & msg)
{
if(msg->node_id == 0 && msg->node_label.empty())
{
NODELET_ERROR("Node id or label should be set!");
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = false;
goalReachedPub_.publish(result);
}
return;
}
goalCommonCallback(msg->node_id, msg->node_label, msg->frame_id, Transform(), msg->header.stamp);
}
2013-12-11 00:12:44 +00:00
bool CoreWrapper::updateRtabmapCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
2013-12-11 00:12:44 +00:00
ros::NodeHandle nh;
for(rtabmap::ParametersMap::iterator iter=parameters_.begin(); iter!=parameters_.end(); ++iter)
{
2013-12-11 00:12:44 +00:00
std::string vStr;
bool vBool;
int vInt;
double vDouble;
if(nh.getParam(iter->first, vStr))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), vStr.c_str());
2013-12-11 00:12:44 +00:00
iter->second = vStr;
}
else if(nh.getParam(iter->first, vBool))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), uBool2Str(vBool).c_str());
2013-12-11 00:12:44 +00:00
iter->second = uBool2Str(vBool);
}
else if(nh.getParam(iter->first, vInt))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), uNumber2Str(vInt).c_str());
2013-12-11 00:12:44 +00:00
iter->second = uNumber2Str(vInt).c_str();
}
else if(nh.getParam(iter->first, vDouble))
{
NODELET_INFO("Setting RTAB-Map parameter \"%s\"=\"%s\"", iter->first.c_str(), uNumber2Str(vDouble).c_str());
2013-12-11 00:12:44 +00:00
iter->second = uNumber2Str(vDouble).c_str();
}
}
NODELET_INFO("rtabmap: Updating parameters");
if(parameters_.find(Parameters::kRtabmapDetectionRate()) != parameters_.end())
2013-12-11 00:12:44 +00:00
{
rate_ = uStr2Float(parameters_.at(Parameters::kRtabmapDetectionRate()));
NODELET_INFO("RTAB-Map rate detection = %f Hz", rate_);
2013-12-11 00:12:44 +00:00
}
rtabmap_.parseParameters(parameters_);
mapsManager_.setParameters(parameters_);
2013-12-11 00:12:44 +00:00
return true;
}
bool CoreWrapper::resetRtabmapCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Reset");
rtabmap_.resetMemory();
2017-05-23 16:31:06 -04:00
covariance_ = cv::Mat();
lastPose_.setIdentity();
lastPoseIntermediate_ = false;
2015-01-30 15:31:44 -05:00
currentMetricGoal_.setNull();
2017-06-01 13:08:28 -04:00
lastPublishedMetricGoal_.setNull();
goalFrameId_.clear();
latestNodeWasReached_ = false;
mapsManager_.clear();
previousStamp_ = ros::Time(0);
globalPose_.header.stamp = ros::Time(0);
gps_ = rtabmap::GPS();
tags_.clear();
userDataMutex_.lock();
userData_ = cv::Mat();
userDataMutex_.unlock();
imus_.clear();
2019-11-13 17:00:02 -05:00
interOdoms_.clear();
2013-12-11 00:12:44 +00:00
return true;
}
bool CoreWrapper::pauseRtabmapCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
if(paused_)
{
NODELET_WARN("rtabmap: Already paused!");
2013-12-11 00:12:44 +00:00
}
else
{
paused_ = true;
NODELET_INFO("rtabmap: paused!");
2013-12-11 00:12:44 +00:00
ros::NodeHandle nh;
nh.setParam("is_rtabmap_paused", true);
}
return true;
}
bool CoreWrapper::resumeRtabmapCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
if(!paused_)
{
NODELET_WARN("rtabmap: Already running!");
2013-12-11 00:12:44 +00:00
}
else
{
paused_ = false;
NODELET_INFO("rtabmap: resumed!");
2013-12-11 00:12:44 +00:00
ros::NodeHandle nh;
nh.setParam("is_rtabmap_paused", false);
}
return true;
}
bool CoreWrapper::triggerNewMapCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Trigger new map");
2013-12-11 00:12:44 +00:00
rtabmap_.triggerNewMap();
return true;
}
2015-03-20 15:17:01 -04:00
bool CoreWrapper::backupDatabaseCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("Backup: Saving memory...");
2015-03-20 15:17:01 -04:00
rtabmap_.close();
NODELET_INFO("Backup: Saving memory... done!");
2015-03-20 15:17:01 -04:00
2017-05-23 16:31:06 -04:00
covariance_ = cv::Mat();
2015-03-20 15:17:01 -04:00
lastPose_.setIdentity();
currentMetricGoal_.setNull();
2017-06-01 13:08:28 -04:00
lastPublishedMetricGoal_.setNull();
goalFrameId_.clear();
2015-03-20 15:17:01 -04:00
latestNodeWasReached_ = false;
userDataMutex_.lock();
userData_ = cv::Mat();
userDataMutex_.unlock();
globalPose_.header.stamp = ros::Time(0);
gps_ = rtabmap::GPS();
tags_.clear();
2015-03-20 15:17:01 -04:00
NODELET_INFO("Backup: Saving \"%s\" to \"%s\"...", databasePath_.c_str(), (databasePath_+".back").c_str());
2015-03-20 15:17:01 -04:00
UFile::copy(databasePath_, databasePath_+".back");
NODELET_INFO("Backup: Saving \"%s\" to \"%s\"... done!", databasePath_.c_str(), (databasePath_+".back").c_str());
2015-03-20 15:17:01 -04:00
NODELET_INFO("Backup: Reloading memory...");
2015-03-20 15:17:01 -04:00
rtabmap_.init(parameters_, databasePath_);
NODELET_INFO("Backup: Reloading memory... done!");
2015-03-20 15:17:01 -04:00
return true;
}
2014-07-09 21:12:24 +00:00
bool CoreWrapper::setModeLocalizationCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Set localization mode");
2014-07-09 21:12:24 +00:00
rtabmap::ParametersMap parameters;
parameters.insert(rtabmap::ParametersPair(rtabmap::Parameters::kMemIncrementalMemory(), "false"));
ros::NodeHandle & nh = getNodeHandle();
nh.setParam(rtabmap::Parameters::kMemIncrementalMemory(), "false");
2014-07-09 21:12:24 +00:00
rtabmap_.parseParameters(parameters);
return true;
}
bool CoreWrapper::setModeMappingCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Set mapping mode");
2014-07-09 21:12:24 +00:00
rtabmap::ParametersMap parameters;
parameters.insert(rtabmap::ParametersPair(rtabmap::Parameters::kMemIncrementalMemory(), "true"));
ros::NodeHandle & nh = getNodeHandle();
nh.setParam(rtabmap::Parameters::kMemIncrementalMemory(), "true");
2014-07-09 21:12:24 +00:00
rtabmap_.parseParameters(parameters);
return true;
}
bool CoreWrapper::setLogDebug(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Set log level to Debug");
ULogger::setLevel(ULogger::kDebug);
return true;
}
bool CoreWrapper::setLogInfo(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Set log level to Info");
ULogger::setLevel(ULogger::kInfo);
return true;
}
bool CoreWrapper::setLogWarn(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Set log level to Warning");
ULogger::setLevel(ULogger::kWarning);
return true;
}
bool CoreWrapper::setLogError(std_srvs::Empty::Request&, std_srvs::Empty::Response&)
{
NODELET_INFO("rtabmap: Set log level to Error");
ULogger::setLevel(ULogger::kError);
return true;
}
bool CoreWrapper::getMapDataCallback(rtabmap_ros::GetMap::Request& req, rtabmap_ros::GetMap::Response& res)
2013-12-11 00:12:44 +00:00
{
NODELET_INFO("rtabmap: Getting map (global=%s optimized=%s graphOnly=%s)...",
req.global?"true":"false",
req.optimized?"true":"false",
req.graphOnly?"true":"false");
std::map<int, Signature> signatures;
2013-12-11 00:12:44 +00:00
std::map<int, Transform> poses;
std::multimap<int, rtabmap::Link> constraints;
2013-12-11 00:12:44 +00:00
if(req.graphOnly)
2013-12-11 00:12:44 +00:00
{
rtabmap_.getGraph(
poses,
constraints,
req.optimized,
2015-05-30 20:08:20 -04:00
req.global,
&signatures);
}
else
{
rtabmap_.get3DMap(
signatures,
2013-12-11 00:12:44 +00:00
poses,
constraints,
req.optimized,
req.global);
}
2014-12-14 16:44:13 -05:00
//RGB-D SLAM data
2015-05-30 20:08:20 -04:00
rtabmap_ros::mapDataToROS(poses,
2014-12-14 16:44:13 -05:00
constraints,
2015-05-30 20:08:20 -04:00
signatures,
mapToOdom_,
2015-05-30 20:08:20 -04:00
res.data);
res.data.header.stamp = ros::Time::now();
res.data.header.frame_id = mapFrameId_;
return true;
}
bool CoreWrapper::getProjMapCallback(nav_msgs::GetMap::Request &req, nav_msgs::GetMap::Response &res)
{
if(parameters_.find(Parameters::kGridFromDepth()) != parameters_.end() &&
!uStr2Bool(parameters_.at(Parameters::kGridFromDepth())))
{
NODELET_WARN("/get_proj_map service is deprecated! Call /get_grid_map service "
"instead with <param name=\"%s\" type=\"string\" value=\"true\"/>. "
"Do \"$ rosrun rtabmap_ros rtabmap --params | grep Grid\" to see "
"all occupancy grid parameters.",
Parameters::kGridFromDepth().c_str());
}
else
{
NODELET_WARN("/get_proj_map service is deprecated! Call /get_grid_map service instead.");
}
return getGridMapCallback(req, res);
}
bool CoreWrapper::getGridMapCallback(nav_msgs::GetMap::Request &req, nav_msgs::GetMap::Response &res)
{
NODELET_WARN("/get_grid_map service is deprecated! Call /get_map service instead.");
return getMapCallback(req, res);
}
bool CoreWrapper::getMapCallback(nav_msgs::GetMap::Request &req, nav_msgs::GetMap::Response &res)
{
// create the grid map
float xMin=0.0f, yMin=0.0f, gridCellSize = 0.05f;
cv::Mat pixels = mapsManager_.getGridMap(xMin, yMin, gridCellSize);
if(!pixels.empty())
2018-02-01 22:16:13 -05:00
{
//init
res.map.info.resolution = gridCellSize;
res.map.info.origin.position.x = 0.0;
res.map.info.origin.position.y = 0.0;
res.map.info.origin.position.z = 0.0;
res.map.info.origin.orientation.x = 0.0;
res.map.info.origin.orientation.y = 0.0;
res.map.info.origin.orientation.z = 0.0;
res.map.info.origin.orientation.w = 1.0;
2018-02-01 22:16:13 -05:00
res.map.info.width = pixels.cols;
res.map.info.height = pixels.rows;
res.map.info.origin.position.x = xMin;
res.map.info.origin.position.y = yMin;
res.map.data.resize(res.map.info.width * res.map.info.height);
2015-02-20 19:00:31 -05:00
memcpy(res.map.data.data(), pixels.data, res.map.info.width * res.map.info.height);
res.map.header.frame_id = mapFrameId_;
res.map.header.stamp = ros::Time::now();
return true;
}
return false;
}
bool CoreWrapper::getProbMapCallback(nav_msgs::GetMap::Request &req, nav_msgs::GetMap::Response &res)
{
// create the grid map
float xMin=0.0f, yMin=0.0f, gridCellSize = 0.05f;
cv::Mat pixels = mapsManager_.getGridProbMap(xMin, yMin, gridCellSize);
if(!pixels.empty())
{
//init
res.map.info.resolution = gridCellSize;
res.map.info.origin.position.x = 0.0;
res.map.info.origin.position.y = 0.0;
res.map.info.origin.position.z = 0.0;
res.map.info.origin.orientation.x = 0.0;
res.map.info.origin.orientation.y = 0.0;
res.map.info.origin.orientation.z = 0.0;
res.map.info.origin.orientation.w = 1.0;
res.map.info.width = pixels.cols;
res.map.info.height = pixels.rows;
res.map.info.origin.position.x = xMin;
res.map.info.origin.position.y = yMin;
res.map.data.resize(res.map.info.width * res.map.info.height);
memcpy(res.map.data.data(), pixels.data, res.map.info.width * res.map.info.height);
res.map.header.frame_id = mapFrameId_;
res.map.header.stamp = ros::Time::now();
return true;
}
return false;
}
2014-11-25 17:13:15 -05:00
bool CoreWrapper::publishMapCallback(rtabmap_ros::PublishMap::Request& req, rtabmap_ros::PublishMap::Response& res)
{
NODELET_INFO("rtabmap: Publishing map...");
if(mapDataPub_.getNumSubscribers() ||
(!req.graphOnly && mapsManager_.hasSubscribers()) ||
2017-08-18 15:07:56 -04:00
(req.graphOnly && (labelsPub_.getNumSubscribers() || mapGraphPub_.getNumSubscribers() || mapPathPub_.getNumSubscribers())))
2015-02-20 19:00:31 -05:00
{
std::map<int, Transform> poses;
std::multimap<int, rtabmap::Link> constraints;
2015-05-30 20:08:20 -04:00
std::map<int, Signature > signatures;
2015-02-20 19:00:31 -05:00
if(req.graphOnly)
{
rtabmap_.getGraph(
poses,
constraints,
req.optimized,
2015-05-30 20:08:20 -04:00
req.global,
&signatures);
}
else
{
rtabmap_.get3DMap(
signatures,
poses,
constraints,
req.optimized,
req.global);
}
ros::Time now = ros::Time::now();
2015-05-30 20:08:20 -04:00
if(mapDataPub_.getNumSubscribers())
{
2015-05-30 20:08:20 -04:00
rtabmap_ros::MapDataPtr msg(new rtabmap_ros::MapData);
msg->header.stamp = now;
msg->header.frame_id = mapFrameId_;
2014-12-14 16:44:13 -05:00
2015-05-30 20:08:20 -04:00
rtabmap_ros::mapDataToROS(poses,
2015-02-20 19:00:31 -05:00
constraints,
2015-05-30 20:08:20 -04:00
signatures,
mapToOdom_,
2015-05-30 20:08:20 -04:00
*msg);
2015-02-20 19:00:31 -05:00
2015-05-30 20:08:20 -04:00
mapDataPub_.publish(msg);
}
if(mapGraphPub_.getNumSubscribers())
{
rtabmap_ros::MapGraphPtr msg(new rtabmap_ros::MapGraph);
msg->header.stamp = now;
msg->header.frame_id = mapFrameId_;
rtabmap_ros::mapGraphToROS(poses,
constraints,
mapToOdom_,
*msg);
mapGraphPub_.publish(msg);
}
bool pubLabels = labelsPub_.getNumSubscribers();
visualization_msgs::MarkerArray markers;
if((landmarksPub_.getNumSubscribers() || pubLabels) && !poses.empty() && poses.begin()->first < 0)
{
geometry_msgs::PoseArrayPtr msg(new geometry_msgs::PoseArray);
msg->header.stamp = now;
msg->header.frame_id = mapFrameId_;
for(std::map<int, Transform>::const_iterator iter=poses.begin(); iter!=poses.end() && iter->first<0; ++iter)
{
geometry_msgs::Pose p;
rtabmap_ros::transformToPoseMsg(iter->second, p);
msg->poses.push_back(p);
if(pubLabels)
{
// Add landmark ids
visualization_msgs::Marker marker;
marker.header.frame_id = mapFrameId_;
marker.header.stamp = now;
marker.ns = "landmarks";
marker.id = iter->first;
marker.action = visualization_msgs::Marker::ADD;
marker.pose.position.x = iter->second.x();
marker.pose.position.y = iter->second.y();
marker.pose.position.z = iter->second.z();
marker.pose.orientation.x = 0.0;
marker.pose.orientation.y = 0.0;
marker.pose.orientation.z = 0.0;
marker.pose.orientation.w = 1.0;
marker.scale.x = 1;
marker.scale.y = 1;
marker.scale.z = 0.35;
marker.color.a = 0.5;
marker.color.r = 1.0;
marker.color.g = 1.0;
marker.color.b = 0.0;
marker.lifetime = ros::Duration(2.0f/rate_);
marker.type = visualization_msgs::Marker::TEXT_VIEW_FACING;
marker.text = uNumber2Str(iter->first);
markers.markers.push_back(marker);
}
}
landmarksPub_.publish(msg);
}
if(!req.graphOnly && mapsManager_.hasSubscribers())
{
std::map<int, Transform> filteredPoses(poses.lower_bound(1), poses.end());
if(maxMappingNodes_ > 0 && filteredPoses.size()>1)
2018-02-01 22:16:13 -05:00
{
std::map<int, Transform> nearestPoses;
std::vector<int> nodes = graph::findNearestNodes(filteredPoses, filteredPoses.rbegin()->second, maxMappingNodes_);
for(std::vector<int>::iterator iter=nodes.begin(); iter!=nodes.end(); ++iter)
{
std::map<int, Transform>::iterator pter = filteredPoses.find(*iter);
if(pter != filteredPoses.end())
{
nearestPoses.insert(*pter);
}
}
}
2015-02-20 19:00:31 -05:00
if(signatures.size())
{
filteredPoses = mapsManager_.updateMapCaches(
2018-02-01 22:16:13 -05:00
filteredPoses,
rtabmap_.getMemory(),
false,
false,
signatures);
2015-02-20 19:00:31 -05:00
}
else
{
2018-02-01 22:16:13 -05:00
filteredPoses = mapsManager_.getFilteredPoses(filteredPoses);
2015-02-20 19:00:31 -05:00
}
mapsManager_.publishMaps(filteredPoses, now, mapFrameId_);
2015-02-20 19:00:31 -05:00
}
2017-08-18 15:07:56 -04:00
bool pubPath = mapPathPub_.getNumSubscribers();
if(pubLabels || pubPath)
{
2015-05-30 20:08:20 -04:00
if(poses.size() && signatures.size())
{
2017-08-18 15:07:56 -04:00
nav_msgs::Path path;
if(pubPath)
{
path.poses.resize(poses.size());
}
int oi=0;
2015-05-30 20:08:20 -04:00
for(std::map<int, Signature>::const_iterator iter=signatures.begin();
iter!=signatures.end();
++iter)
{
std::map<int, Transform>::const_iterator poseIter= poses.find(iter->first);
if(poseIter!=poses.end())
{
2017-08-18 15:07:56 -04:00
if(pubLabels)
{
2017-08-18 15:07:56 -04:00
// Add labels
if(!iter->second.getLabel().empty())
{
visualization_msgs::Marker marker;
marker.header.frame_id = mapFrameId_;
marker.header.stamp = now;
marker.ns = "labels";
marker.id = iter->first;
2017-08-18 15:07:56 -04:00
marker.action = visualization_msgs::Marker::ADD;
marker.pose.position.x = poseIter->second.x();
marker.pose.position.y = poseIter->second.y();
marker.pose.position.z = poseIter->second.z();
marker.pose.orientation.x = 0.0;
marker.pose.orientation.y = 0.0;
marker.pose.orientation.z = 0.0;
marker.pose.orientation.w = 1.0;
marker.scale.x = 1;
marker.scale.y = 1;
marker.scale.z = 0.5;
marker.color.a = 0.7;
marker.color.r = 1.0;
marker.color.g = 0.0;
marker.color.b = 0.0;
marker.type = visualization_msgs::Marker::TEXT_VIEW_FACING;
marker.text = iter->second.getLabel();
markers.markers.push_back(marker);
}
// Add node ids
visualization_msgs::Marker marker;
marker.header.frame_id = mapFrameId_;
marker.header.stamp = now;
2017-08-18 15:07:56 -04:00
marker.ns = "ids";
marker.id = iter->first;
marker.action = visualization_msgs::Marker::ADD;
marker.pose.position.x = poseIter->second.x();
marker.pose.position.y = poseIter->second.y();
marker.pose.position.z = poseIter->second.z();
marker.pose.orientation.x = 0.0;
marker.pose.orientation.y = 0.0;
marker.pose.orientation.z = 0.0;
marker.pose.orientation.w = 1.0;
marker.scale.x = 1;
marker.scale.y = 1;
2017-08-18 15:07:56 -04:00
marker.scale.z = 0.2;
marker.color.a = 0.5;
marker.color.r = 1.0;
2017-08-18 15:07:56 -04:00
marker.color.g = 1.0;
marker.color.b = 1.0;
marker.lifetime = ros::Duration(2.0f/rate_);
marker.type = visualization_msgs::Marker::TEXT_VIEW_FACING;
2017-08-18 15:07:56 -04:00
marker.text = uNumber2Str(iter->first);
markers.markers.push_back(marker);
}
2017-08-18 15:07:56 -04:00
if(pubPath)
{
rtabmap_ros::transformToPoseMsg(poseIter->second, path.poses.at(oi).pose);
path.poses.at(oi).header.frame_id = mapFrameId_;
path.poses.at(oi).header.stamp = ros::Time(iter->second.getStamp());
++oi;
}
}
}
2017-08-18 15:07:56 -04:00
if(pubLabels && markers.markers.size())
{
labelsPub_.publish(markers);
}
2017-08-18 15:07:56 -04:00
if(pubPath && oi)
{
path.header.frame_id = mapFrameId_;
path.header.stamp = now;
path.poses.resize(oi);
mapPathPub_.publish(path);
}
}
}
2014-12-14 16:44:13 -05:00
}
else
{
2015-02-20 19:00:31 -05:00
UWARN("No subscribers, don't need to publish!");
}
2015-02-20 19:00:31 -05:00
return true;
}
2018-07-09 15:26:39 -04:00
bool CoreWrapper::getPlanCallback(nav_msgs::GetPlan::Request &req, nav_msgs::GetPlan::Response &res)
2018-07-09 14:53:21 -04:00
{
Transform pose = rtabmap_ros::transformFromPoseMsg(req.goal.pose);
UTimer timer;
if(!pose.isNull())
{
// transform goal in /map frame
Transform coordinateTransform = Transform::getIdentity();
2018-07-09 14:53:21 -04:00
if(mapFrameId_.compare(req.goal.header.frame_id) != 0)
{
coordinateTransform = rtabmap_ros::getTransform(mapFrameId_, req.goal.header.frame_id, req.goal.header.stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(coordinateTransform.isNull())
2018-07-09 14:53:21 -04:00
{
NODELET_ERROR("Cannot transform goal pose from \"%s\" frame to \"%s\" frame!",
req.goal.header.frame_id.c_str(), mapFrameId_.c_str());
return false;
2018-07-09 14:53:21 -04:00
}
pose = coordinateTransform * pose;
2018-07-09 14:53:21 -04:00
}
// To convert back the poses in goal frame
coordinateTransform = coordinateTransform.inverse();
2018-07-09 15:26:39 -04:00
if(rtabmap_.computePath(pose, req.tolerance))
2018-07-09 14:53:21 -04:00
{
NODELET_INFO("Planning: Time computing path = %f s", timer.ticks());
const std::vector<std::pair<int, Transform> > & poses = rtabmap_.getPath();
res.plan.header.frame_id = req.goal.header.frame_id;
res.plan.header.stamp = req.goal.header.stamp;
2018-07-09 14:53:21 -04:00
if(poses.size() == 0)
{
NODELET_WARN("Planning: Goal already reached (RGBD/GoalReachedRadius=%fm).",
rtabmap_.getGoalReachedRadius());
// just set the goal directly
res.plan.poses.resize(1);
rtabmap_ros::transformToPoseMsg(coordinateTransform*pose, res.plan.poses[0].pose);
2018-07-09 14:53:21 -04:00
}
else
{
res.plan.poses.resize(poses.size());
int oi = 0;
for(std::vector<std::pair<int, Transform> >::const_iterator iter=poses.begin(); iter!=poses.end(); ++iter)
{
res.plan.poses[oi].header = res.plan.header;
rtabmap_ros::transformToPoseMsg(coordinateTransform*iter->second, res.plan.poses[oi].pose);
2018-07-09 14:53:21 -04:00
++oi;
}
if(!rtabmap_.getPathTransformToGoal().isIdentity())
{
res.plan.poses.resize(res.plan.poses.size()+1);
res.plan.poses[res.plan.poses.size()-1].header = res.plan.header;
2018-07-09 14:53:21 -04:00
Transform p = rtabmap_.getPath().back().second*rtabmap_.getPathTransformToGoal();
rtabmap_ros::transformToPoseMsg(coordinateTransform*p, res.plan.poses[res.plan.poses.size()-1].pose);
2018-07-09 14:53:21 -04:00
}
// Just output the path on screen
std::stringstream stream;
for(std::vector<std::pair<int, Transform> >::const_iterator iter=poses.begin(); iter!=poses.end(); ++iter)
{
if(iter != poses.begin())
{
stream << " ";
}
stream << iter->first;
}
NODELET_INFO("Planned path: [%s]", stream.str().c_str());
}
}
rtabmap_.clearPath(0);
}
return true;
}
bool CoreWrapper::getPlanNodesCallback(rtabmap_ros::GetPlan::Request &req, rtabmap_ros::GetPlan::Response &res)
{
Transform pose = rtabmap_ros::transformFromPoseMsg(req.goal.pose);
UTimer timer;
if(req.goal_node > 0 || !pose.isNull())
{
Transform coordinateTransform = Transform::getIdentity();
// transform goal in /map frame
if(mapFrameId_.compare(req.goal.header.frame_id) != 0)
{
coordinateTransform = rtabmap_ros::getTransform(mapFrameId_, req.goal.header.frame_id, req.goal.header.stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(coordinateTransform.isNull())
{
NODELET_ERROR("Cannot transform goal pose from \"%s\" frame to \"%s\" frame!",
req.goal.header.frame_id.c_str(), mapFrameId_.c_str());
return false;
}
if(!pose.isNull())
{
pose = coordinateTransform * pose;
}
}
// To convert back the poses in goal frame
coordinateTransform = coordinateTransform.inverse();
if((req.goal_node > 0 && rtabmap_.computePath(req.goal_node, req.tolerance)) ||
(req.goal_node <= 0 && rtabmap_.computePath(pose, req.tolerance)))
{
NODELET_INFO("Planning: Time computing path = %f s", timer.ticks());
const std::vector<std::pair<int, Transform> > & poses = rtabmap_.getPath();
res.plan.header.frame_id = mapFrameId_;
res.plan.header.stamp = req.goal_node > 0?ros::Time::now():req.goal.header.stamp;
if(poses.size() == 0)
{
NODELET_WARN("Planning: Goal already reached (RGBD/GoalReachedRadius=%fm).",
rtabmap_.getGoalReachedRadius());
if(!pose.isNull())
{
// just set the goal directly
res.plan.poses.resize(1);
res.plan.nodeIds.resize(1);
rtabmap_ros::transformToPoseMsg(coordinateTransform*pose, res.plan.poses[0]);
res.plan.nodeIds[0] = 0;
}
}
else
{
res.plan.poses.resize(poses.size());
res.plan.nodeIds.resize(poses.size());
int oi = 0;
for(std::vector<std::pair<int, Transform> >::const_iterator iter=poses.begin(); iter!=poses.end(); ++iter)
{
rtabmap_ros::transformToPoseMsg(coordinateTransform*iter->second, res.plan.poses[oi]);
res.plan.nodeIds[oi] = iter->first;
++oi;
}
if(!rtabmap_.getPathTransformToGoal().isIdentity())
{
res.plan.poses.resize(res.plan.poses.size()+1);
res.plan.nodeIds.resize(res.plan.nodeIds.size()+1);
Transform p = rtabmap_.getPath().back().second*rtabmap_.getPathTransformToGoal();
rtabmap_ros::transformToPoseMsg(coordinateTransform*p, res.plan.poses[res.plan.poses.size()-1]);
res.plan.nodeIds[res.plan.nodeIds.size()-1] = 0;
}
// Just output the path on screen
std::stringstream stream;
for(std::vector<std::pair<int, Transform> >::const_iterator iter=poses.begin(); iter!=poses.end(); ++iter)
{
if(iter != poses.begin())
{
stream << " ";
}
stream << iter->first;
}
NODELET_INFO("Planned path: [%s]", stream.str().c_str());
}
}
rtabmap_.clearPath(0);
}
return true;
}
bool CoreWrapper::setGoalCallback(rtabmap_ros::SetGoal::Request& req, rtabmap_ros::SetGoal::Response& res)
{
double planningTime = 0.0;
goalCommonCallback(req.node_id, req.node_label, req.frame_id, Transform(), ros::Time::now(), &planningTime);
const std::vector<std::pair<int, Transform> > & path = rtabmap_.getPath();
res.path_ids.resize(path.size());
res.path_poses.resize(path.size());
res.planning_time = planningTime;
for(unsigned int i=0; i<path.size(); ++i)
{
res.path_ids[i] = path[i].first;
rtabmap_ros::transformToPoseMsg(path[i].second, res.path_poses[i]);
}
return true;
}
2015-06-21 20:17:31 -04:00
bool CoreWrapper::cancelGoalCallback(std_srvs::Empty::Request& req, std_srvs::Empty::Response& res)
{
if(rtabmap_.getPath().size())
{
NODELET_WARN("Goal cancelled!");
rtabmap_.clearPath(0);
2015-06-21 20:17:31 -04:00
currentMetricGoal_.setNull();
2017-06-01 13:08:28 -04:00
lastPublishedMetricGoal_.setNull();
goalFrameId_.clear();
2015-06-21 20:17:31 -04:00
latestNodeWasReached_ = false;
if(goalReachedPub_.getNumSubscribers())
{
std_msgs::Bool result;
result.data = false;
goalReachedPub_.publish(result);
}
2015-06-21 20:17:31 -04:00
}
if(mbClient_ && mbClient_->isServerConnected())
2017-06-01 13:08:28 -04:00
{
mbClient_->cancelGoal();
2017-06-01 13:08:28 -04:00
}
2015-06-21 20:17:31 -04:00
return true;
}
bool CoreWrapper::setLabelCallback(rtabmap_ros::SetLabel::Request& req, rtabmap_ros::SetLabel::Response& res)
{
if(rtabmap_.labelLocation(req.node_id, req.node_label))
{
if(req.node_id > 0)
{
NODELET_INFO("Set label \"%s\" to node %d", req.node_label.c_str(), req.node_id);
}
else
{
NODELET_INFO("Set label \"%s\" to last node", req.node_label.c_str());
}
}
else
{
if(req.node_id > 0)
{
NODELET_ERROR("Could not set label \"%s\" to node %d", req.node_label.c_str(), req.node_id);
}
else
{
NODELET_ERROR("Could not set label \"%s\" to last node", req.node_label.c_str());
}
}
return true;
}
bool CoreWrapper::listLabelsCallback(rtabmap_ros::ListLabels::Request& req, rtabmap_ros::ListLabels::Response& res)
{
if(rtabmap_.getMemory())
{
std::map<int, std::string> labels = rtabmap_.getMemory()->getAllLabels();
res.ids = uKeys(labels);
res.labels = uValues(labels);
NODELET_INFO("List labels service: %d labels found.", (int)res.labels.size());
}
return true;
}
void CoreWrapper::publishStats(const ros::Time & stamp)
{
2015-03-25 14:52:05 -04:00
UDEBUG("Publishing stats...");
const rtabmap::Statistics & stats = rtabmap_.getStatistics();
if(infoPub_.getNumSubscribers())
{
//NODELET_INFO("Sending RtabmapInfo msg (last_id=%d)...", stat.refImageId());
2014-11-25 17:13:15 -05:00
rtabmap_ros::InfoPtr msg(new rtabmap_ros::Info);
msg->header.stamp = stamp;
msg->header.frame_id = mapFrameId_;
rtabmap_ros::infoToROS(stats, *msg);
2014-12-14 16:44:13 -05:00
infoPub_.publish(msg);
}
2015-05-30 20:08:20 -04:00
if(mapDataPub_.getNumSubscribers())
{
2015-05-30 20:08:20 -04:00
rtabmap_ros::MapDataPtr msg(new rtabmap_ros::MapData);
msg->header.stamp = stamp;
msg->header.frame_id = mapFrameId_;
2012-06-24 17:19:34 +00:00
2018-11-09 11:38:08 -05:00
std::map<int, Signature> signatures;
if(stats.getLastSignatureData().id() > 0)
{
signatures.insert(std::make_pair(stats.getLastSignatureData().id(), stats.getLastSignatureData()));
}
2015-05-30 20:08:20 -04:00
rtabmap_ros::mapDataToROS(
stats.poses(),
stats.constraints(),
2018-11-09 11:38:08 -05:00
signatures,
2015-05-30 20:08:20 -04:00
stats.mapCorrection(),
*msg);
2015-05-30 20:08:20 -04:00
mapDataPub_.publish(msg);
}
if(mapGraphPub_.getNumSubscribers())
{
rtabmap_ros::MapGraphPtr msg(new rtabmap_ros::MapGraph);
msg->header.stamp = stamp;
msg->header.frame_id = mapFrameId_;
rtabmap_ros::mapGraphToROS(
stats.poses(),
stats.constraints(),
stats.mapCorrection(),
*msg);
mapGraphPub_.publish(msg);
}
if(localGridObstacle_.getNumSubscribers() && !stats.getLastSignatureData().sensorData().gridObstacleCellsRaw().empty())
{
pcl::PCLPointCloud2::Ptr cloud = rtabmap::util3d::laserScanToPointCloud2(LaserScan::backwardCompatibility(stats.getLastSignatureData().sensorData().gridObstacleCellsRaw()));
sensor_msgs::PointCloud2 msg;
pcl_conversions::moveFromPCL(*cloud, msg);
msg.header.stamp = stamp;
msg.header.frame_id = frameId_;
localGridObstacle_.publish(msg);
}
if(localGridEmpty_.getNumSubscribers() && !stats.getLastSignatureData().sensorData().gridEmptyCellsRaw().empty())
{
pcl::PCLPointCloud2::Ptr cloud = rtabmap::util3d::laserScanToPointCloud2(LaserScan::backwardCompatibility(stats.getLastSignatureData().sensorData().gridEmptyCellsRaw()));
sensor_msgs::PointCloud2 msg;
pcl_conversions::moveFromPCL(*cloud, msg);
msg.header.stamp = stamp;
msg.header.frame_id = frameId_;
localGridEmpty_.publish(msg);
}
if(localGridGround_.getNumSubscribers() && !stats.getLastSignatureData().sensorData().gridGroundCellsRaw().empty())
{
pcl::PCLPointCloud2::Ptr cloud = rtabmap::util3d::laserScanToPointCloud2(LaserScan::backwardCompatibility(stats.getLastSignatureData().sensorData().gridGroundCellsRaw()));
sensor_msgs::PointCloud2 msg;
pcl_conversions::moveFromPCL(*cloud, msg);
msg.header.stamp = stamp;
msg.header.frame_id = frameId_;
localGridGround_.publish(msg);
}
2017-08-18 15:07:56 -04:00
bool pubLabels = labelsPub_.getNumSubscribers();
visualization_msgs::MarkerArray markers;
if((landmarksPub_.getNumSubscribers() || pubLabels) && !stats.poses().empty() && stats.poses().begin()->first < 0)
{
geometry_msgs::PoseArrayPtr msg(new geometry_msgs::PoseArray);
msg->header.stamp = stamp;
msg->header.frame_id = mapFrameId_;
for(std::map<int, Transform>::const_iterator iter=stats.poses().begin(); iter!=stats.poses().end() && iter->first<0; ++iter)
{
geometry_msgs::Pose p;
rtabmap_ros::transformToPoseMsg(iter->second, p);
msg->poses.push_back(p);
if(pubLabels)
{
// Add landmark ids
visualization_msgs::Marker marker;
marker.header.frame_id = mapFrameId_;
marker.header.stamp = stamp;
marker.ns = "landmarks";
marker.id = iter->first;
marker.action = visualization_msgs::Marker::ADD;
marker.pose.position.x = iter->second.x();
marker.pose.position.y = iter->second.y();
marker.pose.position.z = iter->second.z();
marker.pose.orientation.x = 0.0;
marker.pose.orientation.y = 0.0;
marker.pose.orientation.z = 0.0;
marker.pose.orientation.w = 1.0;
marker.scale.x = 1;
marker.scale.y = 1;
marker.scale.z = 0.35;
marker.color.a = 0.7;
marker.color.r = 0.0;
marker.color.g = 1.0;
marker.color.b = 0.0;
marker.lifetime = ros::Duration(2.0f/rate_);
marker.type = visualization_msgs::Marker::TEXT_VIEW_FACING;
marker.text = uNumber2Str(iter->first);
markers.markers.push_back(marker);
}
}
landmarksPub_.publish(msg);
}
2017-08-18 15:07:56 -04:00
bool pubPath = mapPathPub_.getNumSubscribers();
if(pubLabels || pubPath)
{
2018-11-09 11:38:08 -05:00
if(stats.poses().size())
{
2017-08-18 15:07:56 -04:00
nav_msgs::Path path;
if(pubPath)
{
path.poses.resize(stats.poses().size());
}
int oi = 0;
2018-11-09 11:38:08 -05:00
for(std::map<int, Transform>::const_iterator poseIter=stats.poses().begin();
poseIter!=stats.poses().end();
++poseIter)
{
2018-11-19 18:00:39 -05:00
if(pubLabels && rtabmap_.getMemory())
{
2018-11-09 11:38:08 -05:00
// Add labels
2018-11-19 18:00:39 -05:00
std::map<int, std::string>::const_iterator lter = rtabmap_.getMemory()->getAllLabels().find(poseIter->first);
if(lter != rtabmap_.getMemory()->getAllLabels().end() && !lter->second.empty())
{
visualization_msgs::Marker marker;
marker.header.frame_id = mapFrameId_;
marker.header.stamp = stamp;
2018-11-09 11:38:08 -05:00
marker.ns = "labels";
marker.id = -poseIter->first;
marker.action = visualization_msgs::Marker::ADD;
marker.pose.position.x = poseIter->second.x();
marker.pose.position.y = poseIter->second.y();
marker.pose.position.z = poseIter->second.z();
marker.pose.orientation.x = 0.0;
marker.pose.orientation.y = 0.0;
marker.pose.orientation.z = 0.0;
marker.pose.orientation.w = 1.0;
marker.scale.x = 1;
marker.scale.y = 1;
2018-11-09 11:38:08 -05:00
marker.scale.z = 0.5;
marker.color.a = 0.7;
marker.color.r = 1.0;
2018-11-09 11:38:08 -05:00
marker.color.g = 0.0;
marker.color.b = 0.0;
marker.type = visualization_msgs::Marker::TEXT_VIEW_FACING;
2018-11-09 11:38:08 -05:00
marker.text = lter->second;
markers.markers.push_back(marker);
}
2018-11-09 11:38:08 -05:00
// Add node ids
visualization_msgs::Marker marker;
marker.header.frame_id = mapFrameId_;
marker.header.stamp = stamp;
marker.ns = "ids";
marker.id = poseIter->first;
marker.action = visualization_msgs::Marker::ADD;
marker.pose.position.x = poseIter->second.x();
marker.pose.position.y = poseIter->second.y();
marker.pose.position.z = poseIter->second.z();
marker.pose.orientation.x = 0.0;
marker.pose.orientation.y = 0.0;
marker.pose.orientation.z = 0.0;
marker.pose.orientation.w = 1.0;
marker.scale.x = 1;
marker.scale.y = 1;
marker.scale.z = 0.2;
marker.color.a = 0.5;
marker.color.r = 1.0;
marker.color.g = 1.0;
marker.color.b = 1.0;
marker.lifetime = ros::Duration(2.0f/rate_);
marker.type = visualization_msgs::Marker::TEXT_VIEW_FACING;
marker.text = uNumber2Str(poseIter->first);
markers.markers.push_back(marker);
}
if(pubPath)
{
rtabmap_ros::transformToPoseMsg(poseIter->second, path.poses.at(oi).pose);
path.poses.at(oi).header.frame_id = mapFrameId_;
path.poses.at(oi).header.stamp = stamp;
++oi;
}
}
2017-08-18 15:07:56 -04:00
if(pubLabels && markers.markers.size())
{
labelsPub_.publish(markers);
}
2017-08-18 15:07:56 -04:00
if(pubPath && oi)
{
path.header.frame_id = mapFrameId_;
path.header.stamp = stamp;
path.poses.resize(oi);
mapPathPub_.publish(path);
}
2014-12-14 16:44:13 -05:00
}
}
}
2013-12-11 00:12:44 +00:00
void CoreWrapper::publishCurrentGoal(const ros::Time & stamp)
{
2017-06-01 13:08:28 -04:00
if(!currentMetricGoal_.isNull() && currentMetricGoal_ != lastPublishedMetricGoal_)
{
NODELET_INFO("Publishing next goal: %d -> %s",
rtabmap_.getPathCurrentGoalId(), currentMetricGoal_.prettyPrint().c_str());
geometry_msgs::PoseStamped poseMsg;
poseMsg.header.frame_id = mapFrameId_;
poseMsg.header.stamp = stamp;
rtabmap_ros::transformToPoseMsg(currentMetricGoal_, poseMsg.pose);
if(useActionForGoal_)
{
if(mbClient_ == 0 || !mbClient_->isServerConnected())
{
NODELET_INFO("Connecting to move_base action server...");
if(mbClient_ == 0)
{
mbClient_ = new MoveBaseClient("move_base", true);
}
mbClient_->waitForServer(ros::Duration(5.0));
}
if(mbClient_ && mbClient_->isServerConnected())
{
move_base_msgs::MoveBaseGoal goal;
goal.target_pose = poseMsg;
mbClient_->sendGoal(goal,
boost::bind(&CoreWrapper::goalDoneCb, this, _1, _2),
boost::bind(&CoreWrapper::goalActiveCb, this),
boost::bind(&CoreWrapper::goalFeedbackCb, this, _1));
2017-06-01 13:08:28 -04:00
lastPublishedMetricGoal_ = currentMetricGoal_;
}
else
{
NODELET_ERROR("Cannot connect to move_base action server (called \"%s\")!", this->getNodeHandle().resolveName("move_base").c_str());
}
}
if(nextMetricGoalPub_.getNumSubscribers())
{
nextMetricGoalPub_.publish(poseMsg);
2017-06-01 13:08:28 -04:00
if(!useActionForGoal_)
{
lastPublishedMetricGoal_ = currentMetricGoal_;
}
}
}
}
// Called once when the goal completes
void CoreWrapper::goalDoneCb(const actionlib::SimpleClientGoalState& state,
const move_base_msgs::MoveBaseResultConstPtr& result)
{
bool ignore = false;
2015-02-06 16:59:10 -05:00
if(!currentMetricGoal_.isNull())
{
2015-02-06 16:59:10 -05:00
if(state == actionlib::SimpleClientGoalState::SUCCEEDED)
{
if(rtabmap_.getPath().size() &&
rtabmap_.getPathCurrentGoalId() != rtabmap_.getPath().back().first &&
(!uContains(rtabmap_.getLocalOptimizedPoses(), rtabmap_.getPath().back().first) || !latestNodeWasReached_))
{
NODELET_WARN("Planning: move_base reached current goal but it is not "
"the last one planned by rtabmap. A new goal should be sent when "
"rtabmap will be able to retrieve next locations on the path.");
ignore = true;
}
else
{
NODELET_INFO("Planning: move_base success!");
}
2015-02-06 16:59:10 -05:00
}
else
{
NODELET_ERROR("Planning: move_base failed for some reason. Aborting the plan...");
2015-02-06 16:59:10 -05:00
}
if(!ignore && goalReachedPub_.getNumSubscribers())
2015-02-06 16:59:10 -05:00
{
std_msgs::Bool result;
result.data = state == actionlib::SimpleClientGoalState::SUCCEEDED;
goalReachedPub_.publish(result);
}
2015-02-06 16:28:28 -05:00
}
if(!ignore)
{
rtabmap_.clearPath(1);
currentMetricGoal_.setNull();
2017-06-01 13:08:28 -04:00
lastPublishedMetricGoal_.setNull();
goalFrameId_.clear();
latestNodeWasReached_ = false;
}
}
// Called once when the goal becomes active
void CoreWrapper::goalActiveCb()
{
//NODELET_INFO("Planning: Goal just went active");
}
// Called every time feedback is received for the goal
void CoreWrapper::goalFeedbackCb(const move_base_msgs::MoveBaseFeedbackConstPtr& feedback)
{
2015-02-06 16:28:28 -05:00
//Transform basePosition = rtabmap_ros::transformFromPoseMsg(feedback->base_position.pose);
//NODELET_INFO("Planning: feedback base_position = %s", basePosition.prettyPrint().c_str());
}
void CoreWrapper::publishLocalPath(const ros::Time & stamp)
{
if(rtabmap_.getPath().size())
{
std::vector<std::pair<int, Transform> > poses = rtabmap_.getPathNextPoses();
if(poses.size())
{
if(localPathPub_.getNumSubscribers() || localPathNodesPub_.getNumSubscribers())
{
nav_msgs::Path path;
rtabmap_ros::Path pathNodes;
path.header.frame_id = pathNodes.header.frame_id = mapFrameId_;
path.header.stamp = pathNodes.header.stamp = stamp;
path.poses.resize(poses.size());
pathNodes.nodeIds.resize(poses.size());
pathNodes.poses.resize(poses.size());
int oi = 0;
for(std::vector<std::pair<int, Transform> >::iterator iter=poses.begin(); iter!=poses.end(); ++iter)
{
path.poses[oi].header = path.header;
rtabmap_ros::transformToPoseMsg(iter->second, path.poses[oi].pose);
pathNodes.poses[oi] = path.poses[oi].pose;
pathNodes.nodeIds[oi] = iter->first;
++oi;
}
if(localPathPub_.getNumSubscribers())
{
localPathPub_.publish(path);
}
if(localPathNodesPub_.getNumSubscribers())
{
localPathNodesPub_.publish(pathNodes);
}
}
}
}
}
void CoreWrapper::publishGlobalPath(const ros::Time & stamp)
{
if((globalPathPub_.getNumSubscribers() || globalPathNodesPub_.getNumSubscribers()) && rtabmap_.getPath().size())
{
Transform pose = uValue(rtabmap_.getLocalOptimizedPoses(), rtabmap_.getPathCurrentGoalId(), Transform());
if(!pose.isNull() && rtabmap_.getPathCurrentGoalIndex() < rtabmap_.getPath().size())
{
// transform the global path in the goal referential
Transform t = pose * rtabmap_.getPath().at(rtabmap_.getPathCurrentGoalIndex()).second.inverse();
nav_msgs::Path path;
rtabmap_ros::Path pathNodes;
path.header.frame_id = pathNodes.header.frame_id = mapFrameId_;
path.header.stamp = pathNodes.header.stamp = stamp;
path.poses.resize(rtabmap_.getPath().size());
pathNodes.nodeIds.resize(rtabmap_.getPath().size());
pathNodes.poses.resize(rtabmap_.getPath().size());
int oi = 0;
for(std::vector<std::pair<int, Transform> >::const_iterator iter=rtabmap_.getPath().begin(); iter!=rtabmap_.getPath().end(); ++iter)
{
path.poses[oi].header = path.header;
rtabmap_ros::transformToPoseMsg(t*iter->second, path.poses[oi].pose);
pathNodes.poses[oi] = path.poses[oi].pose;
pathNodes.nodeIds[oi] = iter->first;
++oi;
}
Transform goalLocalTransform = Transform::getIdentity();
if(!goalFrameId_.empty() && goalFrameId_.compare(frameId_) != 0)
{
Transform localT = rtabmap_ros::getTransform(frameId_, goalFrameId_, stamp, tfListener_, waitForTransform_?waitForTransformDuration_:0.0);
if(!localT.isNull())
{
goalLocalTransform = localT.inverse().to3DoF();
}
}
if(!rtabmap_.getPathTransformToGoal().isIdentity() || !goalLocalTransform.isIdentity())
{
path.poses.resize(path.poses.size()+1);
path.poses[path.poses.size()-1].header = path.header;
pathNodes.nodeIds.resize(pathNodes.nodeIds.size()+1);
pathNodes.poses.resize(pathNodes.poses.size()+1);
Transform p = t * rtabmap_.getPath().back().second*rtabmap_.getPathTransformToGoal() * goalLocalTransform;
rtabmap_ros::transformToPoseMsg(p, path.poses[path.poses.size()-1].pose);
pathNodes.poses[pathNodes.poses.size()-1] = path.poses[path.poses.size()-1].pose;
pathNodes.nodeIds[pathNodes.nodeIds.size()-1] = 0;
}
if(globalPathPub_.getNumSubscribers())
{
globalPathPub_.publish(path);
}
if(globalPathNodesPub_.getNumSubscribers())
{
globalPathNodesPub_.publish(pathNodes);
}
}
}
}
2018-09-29 13:51:48 -04:00
#ifdef WITH_OCTOMAP_MSGS
2016-06-28 19:02:01 -04:00
#ifdef RTABMAP_OCTOMAP
2015-02-20 19:00:31 -05:00
bool CoreWrapper::octomapBinaryCallback(
octomap_msgs::GetOctomap::Request &req,
octomap_msgs::GetOctomap::Response &res)
{
NODELET_INFO("Sending binary map data on service request");
2015-02-20 19:00:31 -05:00
res.map.header.frame_id = mapFrameId_;
res.map.header.stamp = ros::Time::now();
std::map<int, Transform> poses = rtabmap_.getLocalOptimizedPoses();
2018-02-01 22:16:13 -05:00
if(maxMappingNodes_ > 0 && poses.size()>1)
{
std::map<int, Transform> nearestPoses;
std::vector<int> nodes = graph::findNearestNodes(poses, poses.rbegin()->second, maxMappingNodes_);
for(std::vector<int>::iterator iter=nodes.begin(); iter!=nodes.end(); ++iter)
{
std::map<int, Transform>::iterator pter = poses.find(*iter);
if(pter != poses.end())
{
nearestPoses.insert(*pter);
}
}
poses = nearestPoses;
}
poses = mapsManager_.updateMapCaches(poses, rtabmap_.getMemory(), false, true);
2016-06-28 19:02:01 -04:00
const rtabmap::OctoMap * octomap = mapsManager_.getOctomap();
bool success = octomap->octree()->size() && octomap_msgs::binaryMapToMsg(*octomap->octree(), res.map);
2015-02-20 19:00:31 -05:00
return success;
}
bool CoreWrapper::octomapFullCallback(
octomap_msgs::GetOctomap::Request &req,
octomap_msgs::GetOctomap::Response &res)
{
NODELET_INFO("Sending full map data on service request");
2015-02-20 19:00:31 -05:00
res.map.header.frame_id = mapFrameId_;
res.map.header.stamp = ros::Time::now();
std::map<int, Transform> poses = rtabmap_.getLocalOptimizedPoses();
2018-02-01 22:16:13 -05:00
if(maxMappingNodes_ > 0 && poses.size()>1)
{
std::map<int, Transform> nearestPoses;
std::vector<int> nodes = graph::findNearestNodes(poses, poses.rbegin()->second, maxMappingNodes_);
for(std::vector<int>::iterator iter=nodes.begin(); iter!=nodes.end(); ++iter)
{
std::map<int, Transform>::iterator pter = poses.find(*iter);
if(pter != poses.end())
{
nearestPoses.insert(*pter);
}
}
poses = nearestPoses;
}
poses = mapsManager_.updateMapCaches(poses, rtabmap_.getMemory(), false, true);
2016-06-28 19:02:01 -04:00
const rtabmap::OctoMap * octomap = mapsManager_.getOctomap();
bool success = octomap->octree()->size() && octomap_msgs::fullMapToMsg(*octomap->octree(), res.map);
2015-02-20 19:00:31 -05:00
return success;
}
2015-05-01 13:41:08 -04:00
#endif
2016-06-28 19:02:01 -04:00
#endif
2015-02-20 19:00:31 -05:00
PLUGINLIB_EXPORT_CLASS(rtabmap_ros::CoreWrapper, nodelet::Nodelet);
2013-12-11 00:12:44 +00:00
}