/* Copyright (c) 2010-2016, Mathieu Labbe - IntRoLab - Universite de Sherbrooke 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. */ #include "rtabmap_conversions/MsgConversion.h" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace rtabmap_conversions { void transformToTF(const rtabmap::Transform & transform, tf::Transform & tfTransform) { if(!transform.isNull()) { tf::transformEigenToTF(transform.toEigen3d(), tfTransform); } else { tfTransform = tf::Transform(); } } rtabmap::Transform transformFromTF(const tf::Transform & transform) { Eigen::Affine3d eigenTf; tf::transformTFToEigen(transform, eigenTf); return rtabmap::Transform::fromEigen3d(eigenTf); } void transformToGeometryMsg(const rtabmap::Transform & transform, geometry_msgs::Transform & msg) { if(!transform.isNull()) { tf::transformEigenToMsg(transform.toEigen3d(), msg); // make sure the quaternion is normalized long double recipNorm = 1.0 / sqrt(msg.rotation.x * msg.rotation.x + msg.rotation.y * msg.rotation.y + msg.rotation.z * msg.rotation.z + msg.rotation.w * msg.rotation.w); msg.rotation.x *= recipNorm; msg.rotation.y *= recipNorm; msg.rotation.z *= recipNorm; msg.rotation.w *= recipNorm; } else { msg = geometry_msgs::Transform(); } } rtabmap::Transform transformFromGeometryMsg(const geometry_msgs::Transform & msg) { if(msg.rotation.w == 0 && msg.rotation.x == 0 && msg.rotation.y == 0 && msg.rotation.z ==0) { return rtabmap::Transform(); } Eigen::Affine3d tfTransform; tf::transformMsgToEigen(msg, tfTransform); return rtabmap::Transform::fromEigen3d(tfTransform); } void transformToPoseMsg(const rtabmap::Transform & transform, geometry_msgs::Pose & msg) { if(!transform.isNull()) { tf::poseEigenToMsg(transform.toEigen3d(), msg); } else { msg = geometry_msgs::Pose(); } } rtabmap::Transform transformFromPoseMsg(const geometry_msgs::Pose & msg, bool ignoreRotationIfNotSet) { if(msg.orientation.w == 0 && msg.orientation.x == 0 && msg.orientation.y == 0 && msg.orientation.z == 0) { if(ignoreRotationIfNotSet) { return rtabmap::Transform(msg.position.x, msg.position.y, msg.position.z, 0, 0, 0); } return rtabmap::Transform(); } Eigen::Affine3d tfPose; tf::poseMsgToEigen(msg, tfPose); return rtabmap::Transform::fromEigen3d(tfPose); } void toCvCopy(const rtabmap_msgs::RGBDImage & image, cv_bridge::CvImagePtr & rgb, cv_bridge::CvImagePtr & depth) { if(!image.rgb.data.empty()) { rgb = cv_bridge::toCvCopy(image.rgb); } else if(!image.rgb_compressed.data.empty()) { #ifdef CV_BRIDGE_HYDRO ROS_ERROR("Unsupported compressed image copy, please upgrade at least to ROS Indigo to use this."); #else rgb = cv_bridge::toCvCopy(image.rgb_compressed); #endif } if(!image.depth.data.empty()) { depth = cv_bridge::toCvCopy(image.depth); } else if(!image.depth_compressed.data.empty()) { cv_bridge::CvImagePtr ptr = boost::make_shared(); ptr->header = image.depth_compressed.header; ptr->image = rtabmap::uncompressImage(image.depth_compressed.data); ROS_ASSERT(ptr->image.empty() || ptr->image.type() == CV_32FC1 || ptr->image.type() == CV_16UC1); ptr->encoding = ptr->image.empty()?"":ptr->image.type() == CV_32FC1?sensor_msgs::image_encodings::TYPE_32FC1:sensor_msgs::image_encodings::TYPE_16UC1; depth = ptr; } } void toCvShare(const rtabmap_msgs::RGBDImageConstPtr & image, cv_bridge::CvImageConstPtr & rgb, cv_bridge::CvImageConstPtr & depth) { toCvShare(*image, image, rgb, depth); } void toCvShare(const rtabmap_msgs::RGBDImage & image, const boost::shared_ptr& trackedObject, cv_bridge::CvImageConstPtr & rgb, cv_bridge::CvImageConstPtr & depth) { if(!image.rgb.data.empty()) { rgb = cv_bridge::toCvShare(image.rgb, trackedObject); } else if(!image.rgb_compressed.data.empty()) { #ifdef CV_BRIDGE_HYDRO ROS_ERROR("Unsupported compressed image copy, please upgrade at least to ROS Indigo to use this."); #else rgb = cv_bridge::toCvCopy(image.rgb_compressed); #endif } if(!image.depth.data.empty()) { depth = cv_bridge::toCvShare(image.depth, trackedObject); } else if(!image.depth_compressed.data.empty()) { if(image.depth_compressed.format.compare("jpg")==0) { #ifdef CV_BRIDGE_HYDRO ROS_ERROR("Unsupported compressed image copy, please upgrade at least to ROS Indigo to use this."); #else depth = cv_bridge::toCvCopy(image.depth_compressed); #endif } else { cv_bridge::CvImagePtr ptr = boost::make_shared(); ptr->header = image.depth_compressed.header; ptr->image = rtabmap::uncompressImage(image.depth_compressed.data); ROS_ASSERT(ptr->image.empty() || ptr->image.type() == CV_32FC1 || ptr->image.type() == CV_16UC1); ptr->encoding = ptr->image.empty()?"":ptr->image.type() == CV_32FC1?sensor_msgs::image_encodings::TYPE_32FC1:sensor_msgs::image_encodings::TYPE_16UC1; depth = ptr; } } } void rgbdImageToROS(const rtabmap::SensorData & data, rtabmap_msgs::RGBDImage & msg, const std::string & sensorFrameId) { std_msgs::Header header; header.frame_id = sensorFrameId; header.stamp = ros::Time(data.stamp()); rtabmap::Transform localTransform; if(data.cameraModels().size()>1) { UERROR("Cannot convert multi-camera data to rgbd image"); return; } if(data.cameraModels().size() == 1) { //rgb+depth rtabmap_conversions::cameraModelToROS(data.cameraModels().front(), msg.rgb_camera_info); msg.rgb_camera_info.header = header; localTransform = data.cameraModels().front().localTransform(); } else if(data.stereoCameraModels().size() == 1) { //stereo rtabmap_conversions::cameraModelToROS(data.stereoCameraModels()[0].left(), msg.rgb_camera_info); rtabmap_conversions::cameraModelToROS(data.stereoCameraModels()[0].right(), msg.depth_camera_info); msg.rgb_camera_info.header = header; msg.depth_camera_info.header = header; localTransform = data.stereoCameraModels()[0].localTransform(); } if(!data.imageRaw().empty()) { cv_bridge::CvImage cvImg; cvImg.header = header; cvImg.image = data.imageRaw(); UASSERT(data.imageRaw().type()==CV_8UC1 || data.imageRaw().type()==CV_8UC3); cvImg.encoding = data.imageRaw().type()==CV_8UC1?sensor_msgs::image_encodings::MONO8:sensor_msgs::image_encodings::BGR8; cvImg.toImageMsg(msg.rgb); } else if(!data.imageCompressed().empty()) { ROS_ERROR("Conversion of compressed SensorData to RGBDImage is not implemented..."); } if(!data.depthOrRightRaw().empty()) { cv_bridge::CvImage cvDepth; cvDepth.header = header; cvDepth.image = data.depthOrRightRaw(); UASSERT(data.depthOrRightRaw().type()==CV_8UC1 || data.depthOrRightRaw().type()==CV_16UC1 || data.depthOrRightRaw().type()==CV_32FC1); cvDepth.encoding = data.depthOrRightRaw().type()==CV_8UC1?sensor_msgs::image_encodings::MONO8:data.depthOrRightRaw().type()==CV_16UC1?sensor_msgs::image_encodings::TYPE_16UC1:sensor_msgs::image_encodings::TYPE_32FC1; cvDepth.toImageMsg(msg.depth); } else if(!data.depthOrRightCompressed().empty()) { ROS_ERROR("Conversion of compressed SensorData to RGBDImage is not implemented..."); } //convert features if(!data.keypoints().empty()) { rtabmap_conversions::keypointsToROS(data.keypoints(), msg.key_points); } if(!data.keypoints3D().empty()) { rtabmap_conversions::points3fToROS(data.keypoints3D(), msg.points, localTransform.inverse()); } if(!data.descriptors().empty()) { msg.descriptors = rtabmap::compressData(data.descriptors()); } if(!data.globalDescriptors().empty()) { rtabmap_conversions::globalDescriptorToROS(data.globalDescriptors().front(), msg.global_descriptor); msg.global_descriptor.header = header; } } rtabmap::SensorData rgbdImageFromROS(const rtabmap_msgs::RGBDImageConstPtr & image) { rtabmap::SensorData data; cv_bridge::CvImageConstPtr imageMsg; cv_bridge::CvImageConstPtr depthMsg; toCvShare(image, imageMsg, depthMsg); rtabmap::StereoCameraModel stereoModel = stereoCameraModelFromROS(image->rgb_camera_info, image->depth_camera_info, rtabmap::Transform::getIdentity()); if(stereoModel.isValidForProjection()) { cv_bridge::CvImageConstPtr imageRectLeft = imageMsg; cv_bridge::CvImageConstPtr imageRectRight = depthMsg; if(!(imageRectLeft->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) ==0 || imageRectLeft->encoding.compare(sensor_msgs::image_encodings::MONO8) ==0 || imageRectLeft->encoding.compare(sensor_msgs::image_encodings::MONO16) ==0 || imageRectLeft->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || imageRectLeft->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0) || !(imageRectRight->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) ==0 || imageRectRight->encoding.compare(sensor_msgs::image_encodings::MONO8) ==0 || imageRectRight->encoding.compare(sensor_msgs::image_encodings::MONO16) ==0 || imageRectRight->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || imageRectRight->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0)) { ROS_ERROR("Input type must be image=mono8,mono16,rgb8,bgr8 (mono8 recommended), received types are %s (left) and %s (right)", imageRectLeft->encoding.c_str(), imageRectRight->encoding.c_str()); return data; } if(!imageRectLeft->image.empty() && !imageRectRight->image.empty()) { if(stereoModel.baseline() > 10.0) { static bool shown = false; if(!shown) { ROS_WARN("Detected baseline (%f m) is quite large! Is your " "right camera_info P(0,3) correctly set? Note that " "baseline=-P(0,3)/P(0,0). This warning is printed only once.", stereoModel.baseline()); shown = true; } } cv::Mat left, right; if(imageRectLeft->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || imageRectLeft->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0) { left = imageRectLeft->image; } else if(imageRectLeft->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0) { left = cv_bridge::cvtColor(imageRectLeft, "mono8")->image; } else { left = cv_bridge::cvtColor(imageRectLeft, "bgr8")->image; } if(imageRectRight->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || imageRectRight->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0) { right = imageRectRight->image; } else { right = cv_bridge::cvtColor(imageRectRight, "mono8")->image; } // data = rtabmap::SensorData( left, right, stereoModel, 0, rtabmap_conversions::timestampFromROS(image->header.stamp)); } else { ROS_WARN("Odom: input images empty?!?"); } } else //depth { ros::Time higherStamp; int imageWidth = imageMsg->image.cols; int imageHeight = imageMsg->image.rows; int depthWidth = depthMsg->image.cols; int depthHeight = depthMsg->image.rows; UASSERT_MSG( imageWidth/depthWidth == imageHeight/depthHeight, uFormat("rgb=%dx%d depth=%dx%d", imageWidth, imageHeight, depthWidth, depthHeight).c_str()); cv::Mat rgb; cv::Mat depth; rtabmap::CameraModel cameraModels; if(!(imageMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) ==0 || 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 || imageMsg->encoding.compare(sensor_msgs::image_encodings::BGRA8) == 0 || imageMsg->encoding.compare(sensor_msgs::image_encodings::RGBA8) == 0 || imageMsg->encoding.compare(sensor_msgs::image_encodings::BAYER_GRBG8) == 0) || !(depthMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_16UC1) == 0 || depthMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_32FC1) == 0 || depthMsg->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0)) { ROS_ERROR("Input type must be image=mono8,mono16,rgb8,bgr8,bgra8,rgba8 and " "image_depth=32FC1,16UC1,mono16. Current rgb=%s and depth=%s", imageMsg->encoding.c_str(), depthMsg->encoding.c_str()); return data; } cv_bridge::CvImageConstPtr ptrImage = imageMsg; if(imageMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1)==0 || imageMsg->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0 || imageMsg->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0) { // do nothing } else if(imageMsg->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0) { ptrImage = cv_bridge::cvtColor(imageMsg, "mono8"); } else { ptrImage = cv_bridge::cvtColor(imageMsg, "bgr8"); } cv_bridge::CvImageConstPtr ptrDepth = depthMsg; data = rtabmap::SensorData( ptrImage->image, ptrDepth->image, rtabmap_conversions::cameraModelFromROS(image->rgb_camera_info), 0, rtabmap_conversions::timestampFromROS(image->header.stamp)); } return data; } void compressedMatToBytes(const cv::Mat & compressed, std::vector & bytes) { UASSERT(compressed.empty() || compressed.type() == CV_8UC1); bytes.clear(); if(!compressed.empty()) { bytes.resize(compressed.cols * compressed.rows); memcpy(bytes.data(), compressed.data, bytes.size()); } } cv::Mat compressedMatFromBytes(const std::vector & bytes, bool copy) { cv::Mat out; if(bytes.size()) { out = cv::Mat(1, bytes.size(), CV_8UC1, (void*)bytes.data()); if(copy) { out = out.clone(); } } return out; } void infoFromROS(const rtabmap_msgs::Info & info, rtabmap::Statistics & stat) { stat.setExtended(true); // Extended // rtabmap_msgs::Info stat.setRefImageId(info.refId); stat.setLoopClosureId(info.loopClosureId); stat.setProximityDetectionId(info.proximityDetectionId); stat.setStamp(info.header.stamp.toSec()); stat.setLoopClosureTransform(rtabmap_conversions::transformFromGeometryMsg(info.loopClosureTransform)); //wmState stat.setWmState(info.wmState); //Posterior, likelihood, childCount std::map mapIntFloat; for(unsigned int i=0; i(info.posteriorKeys.at(i), info.posteriorValues.at(i))); } stat.setPosterior(mapIntFloat); mapIntFloat.clear(); for(unsigned int i=0; i(info.likelihoodKeys.at(i), info.likelihoodValues.at(i))); } stat.setLikelihood(mapIntFloat); mapIntFloat.clear(); for(unsigned int i=0; i(info.rawLikelihoodKeys.at(i), info.rawLikelihoodValues.at(i))); } stat.setRawLikelihood(mapIntFloat); std::map mapIntInt; for(unsigned int i=0; i(info.weightsKeys.at(i), info.weightsValues.at(i))); } stat.setWeights(mapIntInt); std::map mapIntStr; for(unsigned int i=0; i(info.labelsKeys.at(i), info.labelsValues.at(i))); } stat.setLabels(mapIntStr); stat.setLocalPath(info.localPath); stat.setCurrentGoalId(info.currentGoalId); std::map poses; std::multimap constraints; rtabmap::Transform t; mapGraphFromROS(info.odom_cache, poses, constraints, t); stat.setOdomCachePoses(poses); stat.setOdomCacheConstraints(constraints); // Statistics data for(unsigned int i=0; i(uValue(stats.data(), rtabmap::Statistics::kLoopLandmark_detected(), 0.0f)); rtabmap_conversions::transformToGeometryMsg(stats.loopClosureTransform(), info.loopClosureTransform); // Detailed info if(stats.extended()) { //wmState info.wmState = stats.wmState(); //Posterior, likelihood, childCount info.posteriorKeys = uKeys(stats.posterior()); info.posteriorValues = uValues(stats.posterior()); info.likelihoodKeys = uKeys(stats.likelihood()); info.likelihoodValues = uValues(stats.likelihood()); info.rawLikelihoodKeys = uKeys(stats.rawLikelihood()); info.rawLikelihoodValues = uValues(stats.rawLikelihood()); info.weightsKeys = uKeys(stats.weights()); info.weightsValues = uValues(stats.weights()); info.labelsKeys = uKeys(stats.labels()); info.labelsValues = uValues(stats.labels()); info.localPath = stats.localPath(); info.currentGoalId = stats.currentGoalId(); mapGraphToROS(stats.odomCachePoses(), stats.odomCacheConstraints(), stats.mapCorrection(), info.odom_cache); // Statistics data info.statsKeys = uKeys(stats.data()); info.statsValues = uValues(stats.data()); } } rtabmap::Link linkFromROS(const rtabmap_msgs::Link & msg) { cv::Mat information = cv::Mat(6,6,CV_64FC1, (void*)msg.information.data()).clone(); return rtabmap::Link(msg.fromId, msg.toId, (rtabmap::Link::Type)msg.type, transformFromGeometryMsg(msg.transform), information); } void linkToROS(const rtabmap::Link & link, rtabmap_msgs::Link & msg) { msg.fromId = link.from(); msg.toId = link.to(); msg.type = link.type(); if(link.infMatrix().type() == CV_64FC1 && link.infMatrix().cols == 6 && link.infMatrix().rows == 6) { memcpy(msg.information.data(), link.infMatrix().data, 36*sizeof(double)); } transformToGeometryMsg(link.transform(), msg.transform); } cv::KeyPoint keypointFromROS(const rtabmap_msgs::KeyPoint & msg) { return cv::KeyPoint(msg.pt.x, msg.pt.y, msg.size, msg.angle, msg.response, msg.octave, msg.class_id); } void keypointToROS(const cv::KeyPoint & kpt, rtabmap_msgs::KeyPoint & msg) { msg.angle = kpt.angle; msg.class_id = kpt.class_id; msg.octave = kpt.octave; msg.pt.x = kpt.pt.x; msg.pt.y = kpt.pt.y; msg.response = kpt.response; msg.size = kpt.size; } std::vector keypointsFromROS(const std::vector & msg) { std::vector v(msg.size()); for(unsigned int i=0; i & msg, std::vector & kpts, int xShift) { size_t outCurrentIndex = kpts.size(); kpts.resize(kpts.size()+msg.size()); for(unsigned int i=0; i & kpts, std::vector & msg) { msg.resize(kpts.size()); for(unsigned int i=0; i globalDescriptorsFromROS(const std::vector & msg) { if(!msg.empty()) { std::vector v(msg.size()); for(unsigned int i=0; i(); } void globalDescriptorsToROS(const std::vector & desc, std::vector & msg) { msg.clear(); if(!desc.empty()) { msg.resize(desc.size()); for(unsigned int i=0; i & msg) { rtabmap::EnvSensors v; if(!msg.empty()) { for(unsigned int i=0; i & msg) { msg.clear(); if(!sensors.empty()) { msg.resize(sensors.size()); int i=0; for(rtabmap::EnvSensors::const_iterator iter=sensors.begin(); iter!=sensors.end(); ++iter) { envSensorToROS(iter->second, msg[i++]); } } } cv::Point2f point2fFromROS(const rtabmap_msgs::Point2f & msg) { return cv::Point2f(msg.x, msg.y); } void point2fToROS(const cv::Point2f & kpt, rtabmap_msgs::Point2f & msg) { msg.x = kpt.x; msg.y = kpt.y; } std::vector points2fFromROS(const std::vector & msg) { std::vector v(msg.size()); for(unsigned int i=0; i & kpts, std::vector & msg) { msg.resize(kpts.size()); for(unsigned int i=0; i points3fFromROS(const std::vector & msg, const rtabmap::Transform & transform) { bool transformPoints = !transform.isNull() && !transform.isIdentity(); std::vector v(msg.size()); for(unsigned int i=0; i & msg, std::vector & points3, const rtabmap::Transform & transform) { size_t currentIndex = points3.size(); points3.resize(points3.size()+msg.size()); bool transformPoint = !transform.isNull() && !transform.isIdentity(); for(unsigned int i=0; i & pts, std::vector & msg, const rtabmap::Transform & transform) { msg.resize(pts.size()); bool transformPoints = !transform.isNull() && !transform.isIdentity(); for(unsigned int i=0; i=4 && (uStrContains(camInfo.distortion_model, "fisheye") || uStrContains(camInfo.distortion_model, "equidistant") || uStrContains(camInfo.distortion_model, "Kannala Brandt4"))) { D = cv::Mat::zeros(1, 6, CV_64FC1); D.at(0,0) = camInfo.D[0]; D.at(0,1) = camInfo.D[1]; D.at(0,4) = camInfo.D[2]; D.at(0,5) = camInfo.D[3]; } else if(camInfo.D.size()>8) { bool zerosAfter8 = true; for(size_t i=8; i(4); camInfo.D[0] = model.D_raw().at(0,0); camInfo.D[1] = model.D_raw().at(0,1); camInfo.D[2] = model.D_raw().at(0,4); camInfo.D[3] = model.D_raw().at(0,5); camInfo.distortion_model = "equidistant"; // fisheye } else { camInfo.D = std::vector(model.D_raw().cols); memcpy(camInfo.D.data(), model.D_raw().data, model.D_raw().cols*sizeof(double)); if(camInfo.D.size() > 5) { camInfo.distortion_model = "rational_polynomial"; } else { camInfo.distortion_model = "plumb_bob"; } } UASSERT(model.R().empty() || model.R().total() == 9); if(model.R().empty()) { memset(camInfo.R.elems, 0.0, 9*sizeof(double)); } else { memcpy(camInfo.R.elems, model.R().data, 9*sizeof(double)); } UASSERT(model.P().empty() || model.P().total() == 12); if(model.P().empty()) { memset(camInfo.P.elems, 0.0, 12*sizeof(double)); } else { memcpy(camInfo.P.elems, model.P().data, 12*sizeof(double)); } camInfo.binning_x = 1; camInfo.binning_y = 1; camInfo.roi.width = model.imageWidth(); camInfo.roi.height = model.imageHeight(); camInfo.width = model.imageWidth(); camInfo.height = model.imageHeight(); } rtabmap::StereoCameraModel stereoCameraModelFromROS( const sensor_msgs::CameraInfo & leftCamInfo, const sensor_msgs::CameraInfo & rightCamInfo, const rtabmap::Transform & localTransform, const rtabmap::Transform & stereoTransform) { return rtabmap::StereoCameraModel( "ros", cameraModelFromROS(leftCamInfo, localTransform), cameraModelFromROS(rightCamInfo, localTransform), stereoTransform); } rtabmap::StereoCameraModel stereoCameraModelFromROS( const sensor_msgs::CameraInfo & leftCamInfo, const sensor_msgs::CameraInfo & rightCamInfo, const std::string & frameId, tf::TransformListener & listener, double waitForTransform) { rtabmap::Transform localTransform = getTransform( frameId, leftCamInfo.header.frame_id, leftCamInfo.header.stamp, listener, waitForTransform); if(localTransform.isNull()) { return rtabmap::StereoCameraModel(); } rtabmap::Transform stereoTransform = getTransform( leftCamInfo.header.frame_id, rightCamInfo.header.frame_id, leftCamInfo.header.stamp, listener, waitForTransform); if(stereoTransform.isNull()) { return rtabmap::StereoCameraModel(); } return stereoCameraModelFromROS(leftCamInfo, rightCamInfo, localTransform, stereoTransform); } void mapDataFromROS( const rtabmap_msgs::MapData & msg, std::map & poses, std::multimap & links, std::map & signatures, rtabmap::Transform & mapToOdom) { //optimized graph mapGraphFromROS(msg.graph, poses, links, mapToOdom); //Data for(unsigned int i=0; i & poses, const std::multimap & links, const std::map & signatures, const rtabmap::Transform & mapToOdom, rtabmap_msgs::MapData & msg) { //Optimized graph mapGraphToROS(poses, links, mapToOdom, msg.graph); //Data msg.nodes.resize(signatures.size()); int index=0; for(std::multimap::const_iterator iter = signatures.begin(); iter!=signatures.end(); ++iter) { nodeToROS(iter->second, msg.nodes[index++]); } } void mapGraphFromROS( const rtabmap_msgs::MapGraph & msg, std::map & poses, std::multimap & links, rtabmap::Transform & mapToOdom) { //optimized graph UASSERT(msg.posesId.size() == msg.poses.size()); for(unsigned int i=0; i & poses, const std::multimap & links, const rtabmap::Transform & mapToOdom, rtabmap_msgs::MapGraph & msg) { //Optimized graph msg.posesId.resize(poses.size()); msg.poses.resize(poses.size()); int index = 0; for(std::map::const_iterator iter = poses.begin(); iter != poses.end(); ++iter) { msg.posesId[index] = iter->first; transformToPoseMsg(iter->second, msg.poses[index]); ++index; } msg.links.resize(links.size()); index=0; for(std::multimap::const_iterator iter = links.begin(); iter!=links.end(); ++iter) { linkToROS(iter->second, msg.links[index++]); } transformToGeometryMsg(mapToOdom, msg.mapToOdom); } rtabmap::SensorData sensorDataFromROS(const rtabmap_msgs::SensorData & msg) { rtabmap::SensorData s( cv::Mat(), msg.header.seq, msg.header.stamp.toSec(), compressedMatFromBytes(msg.user_data)); std::vector stereoModels; std::vector models; bool isStereo = !msg.right_camera_info.empty(); if(isStereo) { // stereo model if(msg.left_camera_info.size() == msg.right_camera_info.size() && msg.local_transform.size() == msg.right_camera_info.size()) { for(unsigned int i=0; i trackedObject; cv_bridge::CvImageConstPtr leftRawPtr = cv_bridge::toCvShare(msg.left, trackedObject); if(!(leftRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) ==0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO8) ==0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO16) ==0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::BGRA8) == 0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::RGBA8) == 0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::BAYER_GRBG8))) { ROS_ERROR("Input type must be image=mono8,mono16,rgb8,bgr8 (mono8 recommended), received type is %s. Will return data without left/rgb raw image.", leftRawPtr->encoding.c_str()); } else { if(leftRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || leftRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0) { left = leftRawPtr->image.clone(); } else if(leftRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0) { left = cv_bridge::cvtColor(leftRawPtr, "mono8")->image; } else { left = cv_bridge::cvtColor(leftRawPtr, "bgr8")->image; } } } if(!msg.right.data.empty()) { boost::shared_ptr trackedObject; cv_bridge::CvImageConstPtr rightRawPtr = cv_bridge::toCvShare(msg.right, trackedObject); if(!(rightRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) ==0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO8) ==0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO16) ==0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_16UC1) == 0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_32FC1) == 0)) { ROS_ERROR("Input type must be image=mono8,mono16,rgb8,bgr8,32FC1,16UC1, received type is %s. Will return data without right/depth raw image.", rightRawPtr->encoding.c_str()); } else { if(rightRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0 || (!isStereo && (rightRawPtr->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0|| rightRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_16UC1) == 0 || rightRawPtr->encoding.compare(sensor_msgs::image_encodings::TYPE_32FC1) == 0))) { right = rightRawPtr->image.clone(); } else { right = cv_bridge::cvtColor(rightRawPtr, "mono8")->image; } } } if(isStereo) { s.setStereoImage( compressedMatFromBytes(msg.left_compressed), compressedMatFromBytes(msg.right_compressed), stereoModels); if(!left.empty() && !right.empty()) { s.setStereoImage(left, right, stereoModels, false); } } else { s.setRGBDImage( compressedMatFromBytes(msg.left_compressed), compressedMatFromBytes(msg.right_compressed), models); if(!left.empty() && !right.empty()) { s.setRGBDImage(left, right, models, false); } } // Laser scan data if(!msg.laser_scan_compressed.empty()) { s.setLaserScan(rtabmap::LaserScan( compressedMatFromBytes(msg.laser_scan_compressed), msg.laser_scan_max_pts, msg.laser_scan_max_range, (rtabmap::LaserScan::Format)msg.laser_scan_format, transformFromGeometryMsg(msg.laser_scan_local_transform))); } if(!msg.laser_scan.data.empty()) { pcl::PCLPointCloud2 cloud; pcl_conversions::toPCL(msg.laser_scan, cloud); s.setLaserScan(rtabmap::LaserScan( rtabmap::util3d::laserScanFromPointCloud(cloud), msg.laser_scan_max_pts, msg.laser_scan_max_range, transformFromGeometryMsg(msg.laser_scan_local_transform)), false); UASSERT((rtabmap::LaserScan::Format)msg.laser_scan_format == s.laserScanRaw().format()); } //convert features std::vector keypoints; std::vector keypoints3D; cv::Mat descriptors; if(!msg.key_points.empty()) { keypoints = rtabmap_conversions::keypointsFromROS(msg.key_points); } if(!msg.points.empty()) { keypoints3D = rtabmap_conversions::points3fFromROS(msg.points); } if(!msg.descriptors.empty()) { descriptors = rtabmap::uncompressData(msg.descriptors); } s.setFeatures(keypoints, keypoints3D, descriptors); s.setGlobalDescriptors(rtabmap_conversions::globalDescriptorsFromROS(msg.global_descriptors)); s.setEnvSensors(rtabmap_conversions::envSensorsFromROS(msg.env_sensors)); s.setOccupancyGrid( compressedMatFromBytes(msg.grid_ground), compressedMatFromBytes(msg.grid_obstacles), compressedMatFromBytes(msg.grid_empty_cells), msg.grid_cell_size, point3fFromROS(msg.grid_view_point)); s.setGPS(rtabmap::GPS(msg.gps.stamp, msg.gps.longitude, msg.gps.latitude, msg.gps.altitude, msg.gps.error, msg.gps.bearing)); s.setIMU(rtabmap_conversions::imuFromROS(msg.imu, transformFromGeometryMsg(msg.imu_local_transform))); return s; } void sensorDataToROS(const rtabmap::SensorData & data, rtabmap_msgs::SensorData & msg, const std::string & frameId, bool copyRawData) { // add data msg.header.seq = data.id(); msg.header.stamp = ros::Time(data.stamp()); msg.header.frame_id = frameId; transformToPoseMsg(data.groundTruth(), msg.ground_truth_pose); msg.gps.stamp = data.gps().stamp(); msg.gps.longitude = data.gps().longitude(); msg.gps.latitude = data.gps().latitude(); msg.gps.altitude = data.gps().altitude(); msg.gps.error = data.gps().error(); msg.gps.bearing = data.gps().bearing(); //Calibration if(data.cameraModels().size()) { msg.left_camera_info.resize(data.cameraModels().size()); msg.local_transform.resize(data.cameraModels().size()); for(unsigned int i=0; i words; std::vector wordsKpts; std::vector words3D; cv::Mat wordsDescriptors = rtabmap::uncompressData(msg.word_descriptors); if(msg.word_id_keys.size() != msg.word_id_values.size()) { ROS_ERROR("Word ID keys and values should be the same size (%d, %d)!", (int)msg.word_id_keys.size(), (int)msg.word_id_values.size()); } if(!msg.word_kpts.empty() && msg.word_kpts.size() != msg.word_id_keys.size()) { ROS_ERROR("Word IDs and 2D keypoints should be the same size (%d, %d)!", (int)msg.word_id_keys.size(), (int)msg.word_kpts.size()); } if(!msg.word_pts.empty() && msg.word_pts.size() != msg.word_id_keys.size()) { ROS_ERROR("Word IDs and 3D points should be the same size (%d, %d)!", (int)msg.word_id_keys.size(), (int)msg.word_pts.size()); } if(!wordsDescriptors.empty() && wordsDescriptors.rows != (int)msg.word_id_keys.size()) { ROS_ERROR("Word IDs and descriptors should be the same size (%d, %d)!", (int)msg.word_id_keys.size(), wordsDescriptors.rows); wordsDescriptors = cv::Mat(); } rtabmap::Signature s( msg.id, msg.map_id, msg.weight, msg.stamp, msg.label, transformFromPoseMsg(msg.pose), transformFromPoseMsg(msg.data.ground_truth_pose)); if(msg.word_id_keys.size() == msg.word_id_values.size()) { for(unsigned int i=0; i::const_iterator iter=signature.getWords().begin(); iter!=signature.getWords().end(); ++iter) { msg.word_id_keys.at(i) = iter->first; msg.word_id_values.at(i) = iter->second; if(signature.getWordsKpts().size() == signature.getWords().size()) { if(msg.word_kpts.empty()) { msg.word_kpts.resize(signature.getWords().size()); } keypointToROS(signature.getWordsKpts().at(i), msg.word_kpts.at(i)); } if(signature.getWords3().size() == signature.getWords().size()) { if(msg.word_pts.empty()) { msg.word_pts.resize(signature.getWords().size()); } point3fToROS(signature.getWords3().at(i), msg.word_pts.at(i)); } ++i; } if(!signature.getWordsDescriptors().empty()) { if(signature.getWordsDescriptors().rows == (int)signature.getWords().size()) { msg.word_descriptors = rtabmap::compressData(signature.getWordsDescriptors()); } else { ROS_ERROR("Word IDs and descriptors must have the same size (%d vs %d)!", (int)signature.getWords().size(), signature.getWordsDescriptors().rows); } } sensorDataToROS(signature.sensorData(), msg.data); transformToPoseMsg(signature.getGroundTruthPose(), msg.data.ground_truth_pose); } rtabmap::Signature nodeDataFromROS(const rtabmap_msgs::Node & msg) { return nodeFromROS(msg); } void nodeDataToROS(const rtabmap::Signature & signature, rtabmap_msgs::Node & msg) { nodeToROS(signature, msg); } rtabmap::Signature nodeInfoFromROS(const rtabmap_msgs::Node & msg) { rtabmap::Signature s( msg.id, msg.map_id, msg.weight, msg.stamp, msg.label, transformFromPoseMsg(msg.pose), transformFromPoseMsg(msg.data.ground_truth_pose)); return s; } void nodeInfoToROS(const rtabmap::Signature & signature, rtabmap_msgs::Node & msg) { // add data msg.id = signature.id(); msg.map_id = signature.mapId(); msg.weight = signature.getWeight(); msg.stamp = signature.getStamp(); msg.label = signature.getLabel(); transformToPoseMsg(signature.getPose(), msg.pose); transformToPoseMsg(signature.getGroundTruthPose(), msg.data.ground_truth_pose); } std::map odomInfoToStatistics(const rtabmap::OdometryInfo & info) { std::map stats; stats.insert(std::make_pair("Odometry/TimeRegistration/ms", info.reg.totalTime*1000.0f)); stats.insert(std::make_pair("Odometry/RAM_usage/MB", info.memoryUsage)); // Based on rtabmap/MainWindow.cpp stats.insert(std::make_pair("Odometry/Features/", info.features)); stats.insert(std::make_pair("Odometry/Matches/", info.reg.matches)); stats.insert(std::make_pair("Odometry/MatchesRatio/", info.features<=0?0.0f:float(info.reg.inliers)/float(info.features))); stats.insert(std::make_pair("Odometry/Inliers/", info.reg.inliers)); stats.insert(std::make_pair("Odometry/InliersMeanDistance/m", info.reg.inliersMeanDistance)); stats.insert(std::make_pair("Odometry/InliersDistribution/", info.reg.inliersDistribution)); stats.insert(std::make_pair("Odometry/InliersRatio/", info.reg.inliers)); stats.insert(std::make_pair("Odometry/ICPInliersRatio/", info.reg.icpInliersRatio)); stats.insert(std::make_pair("Odometry/ICPRotation/rad", info.reg.icpRotation)); stats.insert(std::make_pair("Odometry/ICPTranslation/m", info.reg.icpTranslation)); stats.insert(std::make_pair("Odometry/ICPStructuralComplexity/", info.reg.icpStructuralComplexity)); stats.insert(std::make_pair("Odometry/ICPStructuralDistribution/", info.reg.icpStructuralDistribution)); stats.insert(std::make_pair("Odometry/ICPCorrespondences/", info.reg.icpCorrespondences)); stats.insert(std::make_pair("Odometry/StdDevLin/", sqrt((float)info.reg.covariance.at(0,0)))); stats.insert(std::make_pair("Odometry/StdDevAng/", sqrt((float)info.reg.covariance.at(5,5)))); stats.insert(std::make_pair("Odometry/VarianceLin/", (float)info.reg.covariance.at(0,0))); stats.insert(std::make_pair("Odometry/VarianceAng/", (float)info.reg.covariance.at(5,5))); stats.insert(std::make_pair("Odometry/TimeEstimation/ms", info.timeEstimation*1000.0f)); stats.insert(std::make_pair("Odometry/TimeFiltering/ms", info.timeParticleFiltering*1000.0f)); stats.insert(std::make_pair("Odometry/LocalMapSize/", info.localMapSize)); stats.insert(std::make_pair("Odometry/LocalScanMapSize/", info.localScanMapSize)); stats.insert(std::make_pair("Odometry/LocalKeyFrames/", info.localKeyFrames)); stats.insert(std::make_pair("Odometry/LocalBundleOutliers/", info.localBundleOutliers)); stats.insert(std::make_pair("Odometry/LocalBundleConstraints/", info.localBundleConstraints)); stats.insert(std::make_pair("Odometry/LocalBundleTime/ms", info.localBundleTime*1000.0f)); stats.insert(std::make_pair("Odometry/localBundleAvgInlierDistance/pix", info.localBundleAvgInlierDistance)); stats.insert(std::make_pair("Odometry/localBundleMaxKeyFramesForInlier/", info.localBundleMaxKeyFramesForInlier)); float32 localBundleAvgInlierDistance int32 localBundleMaxKeyFramesForInlier stats.insert(std::make_pair("Odometry/KeyFrameAdded/", info.keyFrameAdded?1.0f:0.0f)); stats.insert(std::make_pair("Odometry/Interval/ms", (float)info.interval)); stats.insert(std::make_pair("Odometry/Distance/m", info.distanceTravelled)); float x,y,z,roll,pitch,yaw; float dist = 0.0f, speed=0.0f; if(!info.transform.isNull()) { info.transform.getTranslationAndEulerAngles(x,y,z,roll,pitch,yaw); dist = info.transform.getNorm(); stats.insert(std::make_pair("Odometry/T/m", dist)); stats.insert(std::make_pair("Odometry/Tx/m", x)); stats.insert(std::make_pair("Odometry/Ty/m", y)); stats.insert(std::make_pair("Odometry/Tz/m", z)); stats.insert(std::make_pair("Odometry/Troll/deg", roll*180.0/CV_PI)); stats.insert(std::make_pair("Odometry/Tpitch/deg", pitch*180.0/CV_PI)); stats.insert(std::make_pair("Odometry/Tyaw/deg", yaw*180.0/CV_PI)); if(info.interval>0.0) { speed = dist/info.interval; stats.insert(std::make_pair("Odometry/Speed/kph", speed*3.6)); stats.insert(std::make_pair("Odometry/Speed/mph", speed*2.237)); stats.insert(std::make_pair("Odometry/Speed/mps", speed)); } } if(!info.transformGroundTruth.isNull()) { if(!info.transform.isNull()) { rtabmap::Transform diff = info.transformGroundTruth.inverse()*info.transform; stats.insert(std::make_pair("Odometry/TG_error_lin/m", diff.getNorm())); stats.insert(std::make_pair("Odometry/TG_error_ang/deg", diff.getAngle()*180.0/CV_PI)); } info.transformGroundTruth.getTranslationAndEulerAngles(x,y,z,roll,pitch,yaw); dist = info.transformGroundTruth.getNorm(); stats.insert(std::make_pair("Odometry/TG/m", dist)); stats.insert(std::make_pair("Odometry/TGx/m", x)); stats.insert(std::make_pair("Odometry/TGy/m", y)); stats.insert(std::make_pair("Odometry/TGz/m", z)); stats.insert(std::make_pair("Odometry/TGroll/deg", roll*180.0/CV_PI)); stats.insert(std::make_pair("Odometry/TGpitch/deg", pitch*180.0/CV_PI)); stats.insert(std::make_pair("Odometry/TGyaw/deg", yaw*180.0/CV_PI)); if(info.interval>0.0) { speed = dist/info.interval; stats.insert(std::make_pair("Odometry/SpeedG/kph", speed*3.6)); stats.insert(std::make_pair("Odometry/SpeedG/mph", speed*2.237)); stats.insert(std::make_pair("Odometry/SpeedG/mps", speed)); } } return stats; } rtabmap::OdometryInfo odomInfoFromROS(const rtabmap_msgs::OdomInfo & msg, bool ignoreData) { rtabmap::OdometryInfo info; info.lost = msg.lost; info.reg.matches = msg.matches; info.reg.inliers = msg.inliers; info.reg.icpInliersRatio = msg.icpInliersRatio; info.reg.icpRotation = msg.icpRotation; info.reg.icpTranslation = msg.icpTranslation; info.reg.icpStructuralComplexity = msg.icpStructuralComplexity; info.reg.icpStructuralDistribution = msg.icpStructuralDistribution; info.reg.icpCorrespondences = msg.icpCorrespondences; info.reg.covariance = cv::Mat(6,6,CV_64FC1, (void*)msg.covariance.data()).clone(); info.features = msg.features; info.localMapSize = msg.localMapSize; info.localScanMapSize = msg.localScanMapSize; info.localKeyFrames = msg.localKeyFrames; info.localBundleOutliers = msg.localBundleOutliers; info.localBundleConstraints = msg.localBundleConstraints; info.localBundleTime = msg.localBundleTime; info.localBundleAvgInlierDistance = msg.localBundleAvgInlierDistance; info.localBundleMaxKeyFramesForInlier = msg.localBundleMaxKeyFramesForInlier; UASSERT(msg.localBundleModels.size() == msg.localBundleIds.size()); UASSERT(msg.localBundleModels.size() == msg.localBundlePoses.size()); for(size_t i=0; i models; for(size_t j=0; j >::const_iterator iter=info.localBundleModels.begin(); iter!=info.localBundleModels.end(); ++iter) { msg.localBundleIds.push_back(iter->first); UASSERT(info.localBundlePoses.find(iter->first)!=info.localBundlePoses.end()); geometry_msgs::Pose pose; transformToPoseMsg(info.localBundlePoses.at(iter->first), pose); msg.localBundlePoses.push_back(pose); rtabmap_msgs::CameraModels models; for(size_t i=0; isecond.size(); ++i) { rtabmap_msgs::CameraModel modelMsg; cameraModelToROS(iter->second[i], modelMsg.camera_info); transformToGeometryMsg(iter->second[i].localTransform(), modelMsg.local_transform); models.models.push_back(modelMsg); } msg.localBundleModels.push_back(models); } msg.keyFrameAdded = info.keyFrameAdded; msg.timeEstimation = info.timeEstimation; msg.timeParticleFiltering = info.timeParticleFiltering; msg.stamp = info.stamp; msg.interval = info.interval; msg.distanceTravelled = info.distanceTravelled; msg.memoryUsage = info.memoryUsage; msg.gravityRollError = info.gravityRollError; msg.gravityPitchError = info.gravityPitchError; msg.type = info.type; transformToGeometryMsg(info.transform, msg.transform); transformToGeometryMsg(info.transformFiltered, msg.transformFiltered); transformToGeometryMsg(info.transformGroundTruth, msg.transformGroundTruth); transformToGeometryMsg(info.guess, msg.guess); if(!ignoreData) { msg.wordsKeys = uKeys(info.words); keypointsToROS(uValues(info.words), msg.wordsValues); msg.wordMatches = info.reg.matchesIDs; msg.wordInliers = info.reg.inliersIDs; points2fToROS(info.refCorners, msg.refCorners); points2fToROS(info.newCorners, msg.newCorners); msg.cornerInliers = info.cornerInliers; msg.localMapKeys = uKeys(info.localMap); points3fToROS(uValues(info.localMap), msg.localMapValues); pcl_conversions::moveFromPCL(*rtabmap::util3d::laserScanToPointCloud2(info.localScanMap, info.localScanMap.localTransform()), msg.localScanMap); } } cv::Mat userDataFromROS(const rtabmap_msgs::UserData & dataMsg) { cv::Mat data; if(!dataMsg.data.empty()) { if(dataMsg.cols > 0 && dataMsg.rows > 0 && dataMsg.type >= 0) { data = cv::Mat(dataMsg.rows, dataMsg.cols, dataMsg.type, (void*)dataMsg.data.data()).clone(); } else { if(dataMsg.cols != (int)dataMsg.data.size() || dataMsg.rows != 1 || dataMsg.type != CV_8UC1) { ROS_ERROR("cols, rows and type fields of the UserData msg " "are not correctly set (cols=%d, rows=%d, type=%d)! We assume that the data " "is compressed (cols=%d, rows=1, type=%d(CV_8UC1)).", dataMsg.cols, dataMsg.rows, dataMsg.type, (int)dataMsg.data.size(), CV_8UC1); } data = cv::Mat(1, dataMsg.data.size(), CV_8UC1, (void*)dataMsg.data.data()).clone(); } } return data; } void userDataToROS(const cv::Mat & data, rtabmap_msgs::UserData & dataMsg, bool compress) { if(!data.empty()) { if(compress) { dataMsg.data = rtabmap::compressData(data); dataMsg.rows = 1; dataMsg.cols = dataMsg.data.size(); dataMsg.type = CV_8UC1; } else { dataMsg.data.resize(data.step[0] * data.rows); // use step for non-contiguous matrices memcpy(dataMsg.data.data(), data.data, dataMsg.data.size()); dataMsg.rows = data.rows; dataMsg.cols = data.cols; dataMsg.type = data.type(); } } } rtabmap::IMU imuFromROS(const sensor_msgs::Imu & msg, const rtabmap::Transform & localTransform) { return rtabmap::IMU( cv::Vec4d(msg.orientation.x, msg.orientation.y, msg.orientation.z, msg.orientation.w), cv::Mat(3,3,CV_64FC1,(void*)msg.orientation_covariance.data()).clone(), cv::Vec3d(msg.angular_velocity.x, msg.angular_velocity.y, msg.angular_velocity.z), cv::Mat(3,3,CV_64FC1,(void*)msg.angular_velocity_covariance.data()).clone(), cv::Vec3d(msg.linear_acceleration.x, msg.linear_acceleration.y, msg.linear_acceleration.z), cv::Mat(3,3,CV_64FC1,(void*)msg.linear_acceleration_covariance.data()).clone(), localTransform); } void imuToROS(const rtabmap::IMU & imu, sensor_msgs::Imu & msg) { msg.orientation.x = imu.orientation()[0]; msg.orientation.y = imu.orientation()[1]; msg.orientation.z = imu.orientation()[2]; msg.orientation.w = imu.orientation()[3]; if(!imu.orientationCovariance().empty()) { memcpy((void*)msg.orientation_covariance.data(), imu.orientationCovariance().data, 9*sizeof(double)); } msg.angular_velocity.x = imu.angularVelocity()[0]; msg.angular_velocity.y = imu.angularVelocity()[1]; msg.angular_velocity.z = imu.angularVelocity()[2]; if(!imu.angularVelocityCovariance().empty()) { memcpy((void*)msg.angular_velocity_covariance.data(), imu.angularVelocityCovariance().data, 9*sizeof(double)); } msg.linear_acceleration.x = imu.linearAcceleration()[0]; msg.linear_acceleration.y = imu.linearAcceleration()[1]; msg.linear_acceleration.z = imu.linearAcceleration()[2]; if(!imu.linearAccelerationCovariance().empty()) { memcpy((void*)msg.linear_acceleration_covariance.data(), imu.linearAccelerationCovariance().data, 9*sizeof(double)); } } rtabmap::Landmarks landmarksFromROS( const std::map > & tags, const std::string & frameId, const std::string & odomFrameId, const ros::Time & odomStamp, tf::TransformListener & listener, double waitForTransform, double defaultLinVariance, double defaultAngVariance) { //tag detections rtabmap::Landmarks landmarks; for(std::map >::const_iterator iter=tags.begin(); iter!=tags.end(); ++iter) { if(iter->first <=0) { ROS_ERROR("Invalid landmark received! IDs should be > 0 (it is %d). Ignoring this landmark.", iter->first); continue; } rtabmap::Transform baseToCamera = rtabmap_conversions::getTransform( frameId, iter->second.first.header.frame_id, iter->second.first.header.stamp, listener, waitForTransform); if(baseToCamera.isNull()) { ROS_ERROR("Cannot transform tag pose from \"%s\" frame to \"%s\" frame!", iter->second.first.header.frame_id.c_str(), frameId.c_str()); continue; } rtabmap::Transform baseToTag = baseToCamera * transformFromPoseMsg(iter->second.first.pose.pose); if(!baseToTag.isNull()) { // Correction of the global pose accounting the odometry movement since we received it rtabmap::Transform correction = rtabmap_conversions::getMovingTransform( frameId, odomFrameId, odomStamp, iter->second.first.header.stamp, listener, waitForTransform); if(!correction.isNull()) { baseToTag = correction * baseToTag; } else { ROS_WARN("Could not adjust tag pose accordingly to latest odometry pose. " "If odometry is small since it received the tag pose and " "covariance is large, this should not be a problem."); } cv::Mat covariance = cv::Mat(6,6, CV_64FC1, (void*)iter->second.first.pose.covariance.data()).clone(); if(covariance.empty() || !uIsFinite(covariance.at(0,0)) || covariance.at(0,0)<=0.0f) { covariance = cv::Mat::eye(6,6,CV_64FC1); covariance(cv::Range(0,3), cv::Range(0,3)) *= defaultLinVariance; covariance(cv::Range(3,6), cv::Range(3,6)) *= defaultAngVariance; } landmarks.insert(std::make_pair(iter->first, rtabmap::Landmark(iter->first, iter->second.second, baseToTag, covariance))); } } return landmarks; } rtabmap::Transform getTransform( const std::string & fromFrameId, const std::string & toFrameId, const ros::Time & stamp, tf::TransformListener & listener, double waitForTransform) { // TF ready? rtabmap::Transform transform; try { if(waitForTransform > 0.0 && !stamp.isZero()) { //if(!tfBuffer_.canTransform(fromFrameId, toFrameId, stamp, ros::Duration(1))) std::string errorMsg; if(!listener.waitForTransform(fromFrameId, toFrameId, stamp, ros::Duration(waitForTransform), ros::Duration(0.01), &errorMsg)) { ROS_WARN("Could not get transform from %s to %s after %f seconds (for stamp=%f)! Error=\"%s\".", fromFrameId.c_str(), toFrameId.c_str(), waitForTransform, stamp.toSec(), errorMsg.c_str()); return transform; } } tf::StampedTransform tmp; listener.lookupTransform(fromFrameId, toFrameId, stamp, tmp); transform = rtabmap_conversions::transformFromTF(tmp); } catch(tf::TransformException & ex) { ROS_WARN("(getting transform %s -> %s) %s", fromFrameId.c_str(), toFrameId.c_str(), ex.what()); } return transform; } // get moving transform accordingly to a fixed frame. For example get // transform between moving /base_link between two stamps accordingly to /odom frame. rtabmap::Transform getMovingTransform( const std::string & movingFrame, const std::string & fixedFrame, const ros::Time & stampFrom, const ros::Time & stampTo, tf::TransformListener & listener, double waitForTransform) { // TF ready? rtabmap::Transform transform; try { ros::Time stamp = stampTo>stampFrom?stampTo:stampFrom; if(waitForTransform > 0.0 && !stamp.isZero()) { std::string errorMsg; if(!listener.waitForTransform(movingFrame, fixedFrame, stamp, ros::Duration(waitForTransform), ros::Duration(0.01), &errorMsg)) { ROS_WARN("Could not get transform from %s to %s accordingly to %s after %f seconds (for stamps=%f -> %f)! Error=\"%s\".", movingFrame.c_str(), movingFrame.c_str(), fixedFrame.c_str(), waitForTransform, stampTo.toSec(), stampFrom.toSec(), errorMsg.c_str()); return transform; } } tf::StampedTransform tmp; listener.lookupTransform(movingFrame, stampFrom, movingFrame, stampTo, fixedFrame, tmp); transform = rtabmap_conversions::transformFromTF(tmp); } catch(tf::TransformException & ex) { ROS_WARN("(getting transform movement of %s according to fixed %s) %s", movingFrame.c_str(), fixedFrame.c_str(), ex.what()); } return transform; } bool convertRGBDMsgs( const std::vector & imageMsgs, const std::vector & depthMsgs, const std::vector & cameraInfoMsgs, const std::vector & depthCameraInfoMsgs, const std::string & frameId, const std::string & odomFrameId, const ros::Time & odomStamp, cv::Mat & rgb, cv::Mat & depth, std::vector & cameraModels, std::vector & stereoCameraModels, tf::TransformListener & listener, double waitForTransform, bool alreadRectifiedImages, const std::vector > & localKeyPointsMsgs, const std::vector > & localPoints3dMsgs, const std::vector & localDescriptorsMsgs, std::vector * localKeyPoints, std::vector * localPoints3d, cv::Mat * localDescriptors) { UASSERT(!cameraInfoMsgs.empty() && (cameraInfoMsgs.size() == imageMsgs.size() || imageMsgs.empty()) && (cameraInfoMsgs.size() == depthMsgs.size() || depthMsgs.empty()) && (cameraInfoMsgs.size() == depthCameraInfoMsgs.size() || depthCameraInfoMsgs.empty())); int imageWidth = imageMsgs.size()?imageMsgs[0]->image.cols:cameraInfoMsgs[0].width; int imageHeight = imageMsgs.size()?imageMsgs[0]->image.rows:cameraInfoMsgs[0].height; int depthWidth = depthMsgs.size()?depthMsgs[0]->image.cols:0; int depthHeight = depthMsgs.size()?depthMsgs[0]->image.rows:0; bool isDepth = depthMsgs.empty() || (depthMsgs[0].get() != 0 && ( depthMsgs[0]->encoding.compare(sensor_msgs::image_encodings::TYPE_16UC1) == 0 || depthMsgs[0]->encoding.compare(sensor_msgs::image_encodings::TYPE_32FC1) == 0 || depthMsgs[0]->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0)); // Note that right image can be also MONO16, check the camera info if Tx is set, if so assume it is stereo instead if(isDepth && !depthMsgs.empty() && depthMsgs[0]->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0 && cameraInfoMsgs.size() == depthCameraInfoMsgs.size()) { isDepth = cameraInfoMsgs[0].P.elems[3] == 0.0 && depthCameraInfoMsgs[0].P.elems[3] == 0.0; static bool warned = false; if(!warned && isDepth) { ROS_WARN("Input depth/left image has encoding \"mono16\" and " "camera info P[3] is null for both cameras, thus image is " "considered a depth image. If the depth image is in " "fact the right image, please convert the right image to " "\"mono8\". This warning is shown only once."); warned = true; } } if(isDepth && !depthMsgs.empty()) { UASSERT_MSG( imageWidth/depthWidth == imageHeight/depthHeight, uFormat("rgb=%dx%d depth=%dx%d", imageWidth, imageHeight, depthWidth, depthHeight).c_str()); } int cameraCount = cameraInfoMsgs.size(); for(unsigned int i=0; iencoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO8) ==0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO16) ==0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::BGRA8) == 0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::RGBA8) == 0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::BAYER_GRBG8) == 0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::BAYER_RGGB8) == 0)) { ROS_ERROR("Input rgb/left type must be image=mono8,mono16,rgb8,bgr8,bgra8,rgba8. Current rgb/left=%s", imageMsgs[i]->encoding.c_str()); return false; } UASSERT_MSG(imageMsgs[i]->image.cols == imageWidth && imageMsgs[i]->image.rows == imageHeight, uFormat("imageWidth=%d vs %d imageHeight=%d vs %d", imageWidth, imageMsgs[i]->image.cols, imageHeight, imageMsgs[i]->image.rows).c_str()); } if(!depthMsgs.empty()) { if(isDepth && !(depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::TYPE_16UC1) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::TYPE_32FC1) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0)) { ROS_ERROR("Input depth type must be image_depth=32FC1,16UC1,mono16. Current depth=%s", depthMsgs[i]->encoding.c_str()); return false; } else if(!isDepth && !(depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::BGRA8) == 0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::RGBA8) == 0)) { ROS_ERROR("Input right type must be image=mono8,mono16,rgb8,bgr8,bgra8,rgba8. Current right=%s", depthMsgs[i]->encoding.c_str()); return false; } } ros::Time stamp; if(isDepth && !depthMsgs.empty()) { UASSERT_MSG(depthMsgs[i]->image.cols == depthWidth && depthMsgs[i]->image.rows == depthHeight, uFormat("depthWidth=%d vs %d imageHeight=%d vs %d", depthWidth, depthMsgs[i]->image.cols, depthHeight, depthMsgs[i]->image.rows).c_str()); stamp = depthMsgs[i]->header.stamp; } else if(!imageMsgs.empty()) { stamp = imageMsgs[i]->header.stamp; } else { stamp = cameraInfoMsgs[i].header.stamp; } // use depth's stamp so that geometry is sync to odom, use rgb frame as we assume depth is registered (normally depth msg should have same frame than rgb) rtabmap::Transform localTransform = rtabmap_conversions::getTransform(frameId, !imageMsgs.empty()?imageMsgs[i]->header.frame_id:cameraInfoMsgs[i].header.frame_id, stamp, listener, waitForTransform); if(localTransform.isNull()) { ROS_ERROR("TF of received image %d at time %fs is not set!", i, stamp.toSec()); return false; } // sync with odometry stamp if(!odomFrameId.empty() && odomStamp != stamp) { rtabmap::Transform sensorT = getMovingTransform( frameId, odomFrameId, odomStamp, stamp, listener, waitForTransform); if(sensorT.isNull()) { ROS_WARN("Could not get odometry value for image stamp (%fs). Latest odometry " "stamp is %fs. The image pose will not be synchronized with odometry.", stamp.toSec(), odomStamp.toSec()); } else { //ROS_WARN("RGBD correction = %s (time diff=%fs)", sensorT.prettyPrint().c_str(), fabs(stamp.toSec()-odomStamp.toSec())); localTransform = sensorT * localTransform; } } if(!imageMsgs.empty()) { cv_bridge::CvImageConstPtr ptrImage = imageMsgs[i]; if(imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1)==0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0 || imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0) { // do nothing } else if(imageMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0) { ptrImage = cv_bridge::cvtColor(imageMsgs[i], "mono8"); } else { ptrImage = cv_bridge::cvtColor(imageMsgs[i], "bgr8"); } // initialize if(rgb.empty()) { rgb = cv::Mat(imageHeight, imageWidth*cameraCount, ptrImage->image.type()); } if(ptrImage->image.type() == rgb.type()) { ptrImage->image.copyTo(cv::Mat(rgb, cv::Rect(i*imageWidth, 0, imageWidth, imageHeight))); } else { ROS_ERROR("Some RGB/left images are not the same type!"); return false; } } if(!depthMsgs.empty()) { if(isDepth) { cv_bridge::CvImageConstPtr ptrDepth = depthMsgs[i]; cv::Mat subDepth = ptrDepth->image; if(depth.empty()) { depth = cv::Mat(depthHeight, depthWidth*cameraCount, subDepth.type()); } if(subDepth.type() == depth.type()) { subDepth.copyTo(cv::Mat(depth, cv::Rect(i*depthWidth, 0, depthWidth, depthHeight))); } else { ROS_ERROR("Some Depth images are not the same type!"); return false; } } else { cv_bridge::CvImageConstPtr ptrImage = depthMsgs[i]; if( depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1)==0 || depthMsgs[i]->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0) { // do nothing } else { ptrImage = cv_bridge::cvtColor(depthMsgs[i], "mono8"); } // initialize if(depth.empty()) { depth = cv::Mat(depthHeight, depthWidth*cameraCount, ptrImage->image.type()); } if(ptrImage->image.type() == depth.type()) { ptrImage->image.copyTo(cv::Mat(depth, cv::Rect(i*depthWidth, 0, depthWidth, depthHeight))); } else { ROS_ERROR("Some right images are not the same type!"); return false; } } } if(isDepth) { cameraModels.push_back(rtabmap_conversions::cameraModelFromROS(cameraInfoMsgs[i], localTransform)); } else //stereo { UASSERT(cameraInfoMsgs.size() == depthCameraInfoMsgs.size()); rtabmap::Transform stereoTransform; if(!alreadRectifiedImages) { if(depthCameraInfoMsgs[i].header.frame_id.empty() || cameraInfoMsgs[i].header.frame_id.empty()) { if(depthCameraInfoMsgs[i].P[3] == 0.0 && cameraInfoMsgs[i].P[3] == 0) { ROS_ERROR("Parameter %s is false but the frame_id in one of the camera_info " "topic is empty, so TF between the cameras cannot be computed!", rtabmap::Parameters::kRtabmapImagesAlreadyRectified().c_str()); return false; } else { static bool warned = false; if(!warned) { ROS_WARN("Parameter %s is false but the frame_id in one of the " "camera_info topic is empty, so TF between the cameras cannot be " "computed! However, the baseline can be computed from the calibration, " "we will use this one instead of TF. This message is only printed once...", rtabmap::Parameters::kRtabmapImagesAlreadyRectified().c_str()); warned = true; } } } else { stereoTransform = getTransform( depthCameraInfoMsgs[i].header.frame_id, cameraInfoMsgs[i].header.frame_id, cameraInfoMsgs[i].header.stamp, listener, waitForTransform); if(stereoTransform.isNull()) { ROS_ERROR("Parameter %s is false but we cannot get TF between the two cameras!", rtabmap::Parameters::kRtabmapImagesAlreadyRectified().c_str()); return false; } else if(stereoTransform.isIdentity()) { ROS_ERROR("Parameter %s is false but we cannot get a valid TF between the two cameras! " "Identity transform returned between left and right cameras. Verify that if TF between " "the cameras is valid: \"rosrun tf tf_echo %s %s\".", rtabmap::Parameters::kRtabmapImagesAlreadyRectified().c_str(), depthCameraInfoMsgs[i].header.frame_id.c_str(), cameraInfoMsgs[i].header.frame_id.c_str()); return false; } } } rtabmap::StereoCameraModel stereoModel = rtabmap_conversions::stereoCameraModelFromROS(cameraInfoMsgs[i], depthCameraInfoMsgs[i], localTransform, stereoTransform); if(stereoModel.baseline() > 10.0) { static bool shown = false; if(!shown) { ROS_WARN("Detected baseline (%f m) is quite large! Is your " "right camera_info P(0,3) correctly set? Note that " "baseline=-P(0,3)/P(0,0). You may need to calibrate your camera. " "This warning is printed only once.", stereoModel.baseline()); shown = true; } } else if(stereoModel.baseline() == 0 && alreadRectifiedImages) { rtabmap::Transform stereoTransform; if( !cameraInfoMsgs[i].header.frame_id.empty() && !depthCameraInfoMsgs[i].header.frame_id.empty()) { stereoTransform = getTransform( cameraInfoMsgs[i].header.frame_id, depthCameraInfoMsgs[i].header.frame_id, cameraInfoMsgs[i].header.stamp, listener, waitForTransform); } if(stereoTransform.isNull() || stereoTransform.x()<=0) { if(cameraInfoMsgs[i].header.frame_id.empty() || depthCameraInfoMsgs[i].header.frame_id.empty()) { ROS_WARN("We cannot estimated the baseline of the rectified images with tf! (camera_info topics have empty frame_id)"); } else { ROS_WARN("We cannot estimated the baseline of the rectified images with tf! (%s->%s = %s)", depthCameraInfoMsgs[i].header.frame_id.c_str(), cameraInfoMsgs[i].header.frame_id.c_str(), stereoTransform.prettyPrint().c_str()); } } else { static bool warned = false; if(!warned) { ROS_WARN("Right camera info doesn't have Tx set but we are assuming that stereo images are already rectified (see %s parameter). While not " "recommended, we used TF to get the baseline (%s->%s = %fm) for convenience (e.g., D400 ir stereo issue). It is preferred to feed " "a valid right camera info if stereo images are already rectified. This message is only printed once...", rtabmap::Parameters::kRtabmapImagesAlreadyRectified().c_str(), depthCameraInfoMsgs[i].header.frame_id.c_str(), cameraInfoMsgs[i].header.frame_id.c_str(), stereoTransform.x()); warned = true; } stereoModel = rtabmap::StereoCameraModel( stereoModel.left().fx(), stereoModel.left().fy(), stereoModel.left().cx(), stereoModel.left().cy(), stereoTransform.x(), stereoModel.localTransform(), stereoModel.left().imageSize()); } } stereoCameraModels.push_back(stereoModel); } if(localKeyPoints && localKeyPointsMsgs.size() == cameraInfoMsgs.size()) { rtabmap_conversions::keypointsFromROS(localKeyPointsMsgs[i], *localKeyPoints, imageWidth*i); } if(localPoints3d && localPoints3dMsgs.size() == cameraInfoMsgs.size()) { // Points should be in base frame rtabmap_conversions::points3fFromROS(localPoints3dMsgs[i], *localPoints3d, localTransform); } if(localDescriptors && localDescriptorsMsgs.size() == cameraInfoMsgs.size()) { localDescriptors->push_back(localDescriptorsMsgs[i]); } } return true; } bool convertStereoMsg( const cv_bridge::CvImageConstPtr& leftImageMsg, const cv_bridge::CvImageConstPtr& rightImageMsg, const sensor_msgs::CameraInfo& leftCamInfoMsg, const sensor_msgs::CameraInfo& rightCamInfoMsg, const std::string & frameId, const std::string & odomFrameId, const ros::Time & odomStamp, cv::Mat & left, cv::Mat & right, rtabmap::StereoCameraModel & stereoModel, tf::TransformListener & listener, double waitForTransform, bool alreadyRectified) { UASSERT(leftImageMsg.get() && rightImageMsg.get()); if(!(leftImageMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || leftImageMsg->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0 || leftImageMsg->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0 || leftImageMsg->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || leftImageMsg->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0 || leftImageMsg->encoding.compare(sensor_msgs::image_encodings::BGRA8) == 0 || leftImageMsg->encoding.compare(sensor_msgs::image_encodings::RGBA8) == 0) || !(rightImageMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || rightImageMsg->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0 || rightImageMsg->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0 || rightImageMsg->encoding.compare(sensor_msgs::image_encodings::BGR8) == 0 || rightImageMsg->encoding.compare(sensor_msgs::image_encodings::RGB8) == 0 || rightImageMsg->encoding.compare(sensor_msgs::image_encodings::BGRA8) == 0 || rightImageMsg->encoding.compare(sensor_msgs::image_encodings::RGBA8) == 0)) { ROS_ERROR("Input type must be image=mono8,mono16,rgb8,bgr8,bgra8,rgba8"); ROS_ERROR("Input type must be image=mono8,mono16,rgb8,bgr8,bgra8,rgba8 Current left=%s and right=%s", leftImageMsg->encoding.c_str(), rightImageMsg->encoding.c_str()); return false; } if(leftImageMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || leftImageMsg->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0) { left = leftImageMsg->image.clone(); } else if(leftImageMsg->encoding.compare(sensor_msgs::image_encodings::MONO16) == 0) { left = cv_bridge::cvtColor(leftImageMsg, "mono8")->image; } else { left = cv_bridge::cvtColor(leftImageMsg, "bgr8")->image; } if(rightImageMsg->encoding.compare(sensor_msgs::image_encodings::TYPE_8UC1) == 0 || rightImageMsg->encoding.compare(sensor_msgs::image_encodings::MONO8) == 0) { right = rightImageMsg->image.clone(); } else { right = cv_bridge::cvtColor(rightImageMsg, "mono8")->image; } rtabmap::Transform localTransform = getTransform(frameId, leftImageMsg->header.frame_id, leftImageMsg->header.stamp, listener, waitForTransform); if(localTransform.isNull()) { return false; } // sync with odometry stamp if(!odomFrameId.empty() && odomStamp != leftImageMsg->header.stamp) { rtabmap::Transform sensorT = getMovingTransform( frameId, odomFrameId, odomStamp, leftImageMsg->header.stamp, listener, waitForTransform); if(sensorT.isNull()) { ROS_WARN("Could not get odometry value for stereo msg stamp (%fs). Latest odometry " "stamp is %fs. The stereo image pose will not be synchronized with odometry.", leftImageMsg->header.stamp.toSec(), odomStamp.toSec()); } else { localTransform = sensorT * localTransform; } } rtabmap::Transform stereoTransform; if(!alreadyRectified) { stereoTransform = getTransform( rightCamInfoMsg.header.frame_id, leftCamInfoMsg.header.frame_id, leftCamInfoMsg.header.stamp, listener, waitForTransform); if(stereoTransform.isNull()) { ROS_ERROR("Parameter %s is false but we cannot get TF between the two cameras!", rtabmap::Parameters::kRtabmapImagesAlreadyRectified().c_str()); return false; } } stereoModel = rtabmap_conversions::stereoCameraModelFromROS(leftCamInfoMsg, rightCamInfoMsg, localTransform, stereoTransform); if(stereoModel.baseline() > 10.0) { static bool shown = false; if(!shown) { ROS_WARN("Detected baseline (%f m) is quite large! Is your " "right camera_info P(0,3) correctly set? Note that " "baseline=-P(0,3)/P(0,0). You may need to calibrate your camera. " "This warning is printed only once.", stereoModel.baseline()); shown = true; } } else if(stereoModel.baseline() == 0 && alreadyRectified) { rtabmap::Transform stereoTransform = getTransform( leftCamInfoMsg.header.frame_id, rightCamInfoMsg.header.frame_id, leftCamInfoMsg.header.stamp, listener, waitForTransform); if(stereoTransform.isNull() || stereoTransform.x()<=0) { ROS_WARN("We cannot estimated the baseline of the rectified images with tf! (%s->%s = %s)", rightCamInfoMsg.header.frame_id.c_str(), leftCamInfoMsg.header.frame_id.c_str(), stereoTransform.prettyPrint().c_str()); } else { static bool warned = false; if(!warned) { ROS_WARN("Right camera info doesn't have Tx set but we are assuming that stereo images are already rectified (see %s parameter). While not " "recommended, we used TF to get the baseline (%s->%s = %fm) for convenience (e.g., D400 ir stereo issue). It is preferred to feed " "a valid right camera info if stereo images are already rectified. This message is only printed once...", rtabmap::Parameters::kRtabmapImagesAlreadyRectified().c_str(), rightCamInfoMsg.header.frame_id.c_str(), leftCamInfoMsg.header.frame_id.c_str(), stereoTransform.x()); warned = true; } stereoModel = rtabmap::StereoCameraModel( stereoModel.left().fx(), stereoModel.left().fy(), stereoModel.left().cx(), stereoModel.left().cy(), stereoTransform.x(), stereoModel.localTransform(), stereoModel.left().imageSize()); } } return true; } bool convertScanMsg( const sensor_msgs::LaserScan & scan2dMsg, const std::string & frameId, const std::string & odomFrameId, const ros::Time & odomStamp, rtabmap::LaserScan & scan, tf::TransformListener & listener, double waitForTransform, bool outputInFrameId) { // scan message validation check if(scan2dMsg.angle_increment == 0.0f) { ROS_ERROR("convertScanMsg: angle_increment should not be 0!"); return false; } if(scan2dMsg.range_min > scan2dMsg.range_max) { ROS_ERROR("convertScanMsg: range_min (%f) should be smaller than range_max (%f)!", scan2dMsg.range_min, scan2dMsg.range_max); return false; } if(scan2dMsg.angle_increment > 0 && scan2dMsg.angle_max < scan2dMsg.angle_min) { ROS_ERROR("convertScanMsg: Angle increment (%f) should be negative if angle_min(%f) > angle_max(%f)!", scan2dMsg.angle_increment, scan2dMsg.angle_min, scan2dMsg.angle_max); return false; } else if (scan2dMsg.angle_increment < 0 && scan2dMsg.angle_max > scan2dMsg.angle_min) { ROS_ERROR("convertScanMsg: Angle increment (%f) should positive if angle_min(%f) < angle_max(%f)!", scan2dMsg.angle_increment, scan2dMsg.angle_min, scan2dMsg.angle_max); return false; } // make sure the frame of the laser is updated during the whole scan time rtabmap::Transform tmpT = getMovingTransform( scan2dMsg.header.frame_id, odomFrameId.empty()?frameId:odomFrameId, scan2dMsg.header.stamp, scan2dMsg.header.stamp + ros::Duration().fromSec(scan2dMsg.ranges.size()*scan2dMsg.time_increment), listener, waitForTransform); if(tmpT.isNull()) { return false; } rtabmap::Transform scanLocalTransform = getTransform( frameId, scan2dMsg.header.frame_id, scan2dMsg.header.stamp, listener, waitForTransform); if(scanLocalTransform.isNull()) { return false; } //transform in frameId_ frame sensor_msgs::PointCloud2 scanOut; laser_geometry::LaserProjection projection; projection.transformLaserScanToPointCloud(odomFrameId.empty()?frameId:odomFrameId, scan2dMsg, scanOut, listener); //transform back in laser frame rtabmap::Transform laserToOdom = getTransform( scan2dMsg.header.frame_id, odomFrameId.empty()?frameId:odomFrameId, scan2dMsg.header.stamp, listener, waitForTransform); if(laserToOdom.isNull()) { return false; } // sync with odometry stamp if(!odomFrameId.empty() && odomStamp != scan2dMsg.header.stamp) { rtabmap::Transform sensorT = getMovingTransform( frameId, odomFrameId, odomStamp, scan2dMsg.header.stamp, listener, waitForTransform); if(sensorT.isNull()) { ROS_WARN("Could not get odometry value for laser scan stamp (%fs). Latest odometry " "stamp is %fs. The laser scan pose will not be synchronized with odometry.", scan2dMsg.header.stamp.toSec(), odomStamp.toSec()); } else { //ROS_WARN("scan correction = %s (time diff=%fs)", sensorT.prettyPrint().c_str(), fabs(scan2dMsg->header.stamp.toSec()-odomStamp.toSec())); scanLocalTransform = sensorT * scanLocalTransform; } } if(outputInFrameId) { laserToOdom *= scanLocalTransform; } bool hasIntensity = false; for(unsigned int i=0; i::Ptr pclScan(new pcl::PointCloud); pcl::fromROSMsg(scanOut, *pclScan); pclScan->is_dense = true; data = rtabmap::util3d::laserScan2dFromPointCloud(*pclScan, laserToOdom).data(); // put back in laser frame format = rtabmap::LaserScan::kXYI; } else { pcl::PointCloud::Ptr pclScan(new pcl::PointCloud); pcl::fromROSMsg(scanOut, *pclScan); pclScan->is_dense = true; data = rtabmap::util3d::laserScan2dFromPointCloud(*pclScan, laserToOdom).data(); // put back in laser frame format = rtabmap::LaserScan::kXY; } rtabmap::Transform zAxis(0,0,1,0,0,0); if((scanLocalTransform.rotation()*zAxis).z() < 0) { cv::Mat flipScan; cv::flip(data, flipScan, 1); data = flipScan; } scan = rtabmap::LaserScan( data, format, scan2dMsg.range_min, scan2dMsg.range_max, scan2dMsg.angle_min, scan2dMsg.angle_max, scan2dMsg.angle_increment, outputInFrameId?rtabmap::Transform::getIdentity():scanLocalTransform); return true; } bool convertScan3dMsg( const sensor_msgs::PointCloud2 & scan3dMsg, const std::string & frameId, const std::string & odomFrameId, const ros::Time & odomStamp, rtabmap::LaserScan & scan, tf::TransformListener & listener, double waitForTransform, int maxPoints, float maxRange, bool is2D) { UASSERT_MSG(scan3dMsg.data.size() == scan3dMsg.row_step*scan3dMsg.height, uFormat("data=%d row_step=%d height=%d", scan3dMsg.data.size(), scan3dMsg.row_step, scan3dMsg.height).c_str()); rtabmap::Transform scanLocalTransform = getTransform(frameId, scan3dMsg.header.frame_id, scan3dMsg.header.stamp, listener, waitForTransform); if(scanLocalTransform.isNull()) { ROS_ERROR("TF of received scan cloud at time %fs is not set, aborting rtabmap update.", scan3dMsg.header.stamp.toSec()); return false; } // sync with odometry stamp if(!odomFrameId.empty() && odomStamp != scan3dMsg.header.stamp) { rtabmap::Transform sensorT = getMovingTransform( frameId, odomFrameId, odomStamp, scan3dMsg.header.stamp, listener, waitForTransform); if(sensorT.isNull()) { ROS_WARN("Could not get odometry value for laser scan stamp (%fs). Latest odometry " "stamp is %fs. The 3d laser scan pose will not be synchronized with odometry.", scan3dMsg.header.stamp.toSec(), odomStamp.toSec()); } else { scanLocalTransform = sensorT * scanLocalTransform; } } scan = rtabmap::util3d::laserScanFromPointCloud(scan3dMsg, true, is2D); scan = rtabmap::LaserScan(scan, maxPoints, maxRange, scanLocalTransform); return true; } bool deskew_impl( const sensor_msgs::PointCloud2 & input, sensor_msgs::PointCloud2 & output, const std::string & fixedFrameId, tf::TransformListener * listener, double waitForTransform, bool slerp, const rtabmap::Transform & velocity, double previousStamp) { if(listener != 0) { if(input.header.frame_id.empty()) { ROS_ERROR("Input cloud has empty frame_id!"); return false; } if(fixedFrameId.empty()) { ROS_ERROR("fixedFrameId parameter should be set!"); return false; } } else { if(!slerp) { ROS_ERROR("slerp should be true when constant velocity model is used!"); return false; } if(previousStamp <= 0.0) { ROS_ERROR("previousStamp should be >0 when constant velocity model is used!"); return false; } if(velocity.isNull()) { ROS_ERROR("velocity should be valid when constant velocity model is used!"); return false; } } int offsetTime = -1; int offsetX = -1; int offsetY = -1; int offsetZ = -1; int timeDatatype = 6; for(size_t i=0; i input.height; // Get latest timestamp ros::Time firstStamp; ros::Time lastStamp; if(timeDatatype == 6) // UINT32 { unsigned int nsecFirst = *((const unsigned int*)(&input.data[0]+offsetTime)); unsigned int nsecLast = *((const unsigned int*)(&input.data[(input.width-1)*input.point_step + input.row_step*(input.height-1)]+offsetTime)); if(nsecFirst > nsecLast) { // scans are not ordered, we need to search min/max static bool warned = false; if(!warned) { ROS_WARN("Timestamp channel is not ordered, we will have to parse every scans to " "determinate first and last time offsets. This will add slightly computation " "time. This warning is only shown once."); warned = true; } if(timeOnColumns) { for(size_t i=0; i nsecLast) { nsecLast = nsec; } } } else { for(size_t i=0; i nsecLast) { nsecLast = nsec; } } } } firstStamp = input.header.stamp+ros::Duration(0, nsecFirst); lastStamp = input.header.stamp+ros::Duration(0, nsecLast); } else if(timeDatatype == 7) // FLOAT32 { float secFirst = *((const float*)(&input.data[0]+offsetTime)); float secLast = *((const float*)(&input.data[(input.width-1)*input.point_step + input.row_step*(input.height-1)]+offsetTime)); if(secFirst > secLast) { // scans are not ordered, we need to search min/max static bool warned = false; if(!warned) { ROS_WARN("Timestamp channel is not ordered, we will have to parse every scans to " "determinate first and last time offsets. This will add slightly computation " "time. This warning is only shown once."); warned = true; } if(timeOnColumns) { for(size_t i=0; i secLast) { secLast = sec; } } } else { for(size_t i=0; i secLast) { secLast = sec; } } } } firstStamp = input.header.stamp+ros::Duration().fromSec(secFirst); lastStamp = input.header.stamp+ros::Duration().fromSec(secLast); } else if(timeDatatype == 8) // FLOAT64 { double secFirst = *((const double*)(&input.data[0]+offsetTime)); double secLast = *((const double*)(&input.data[(input.width-1)*input.point_step + input.row_step*(input.height-1)]+offsetTime)); if(secFirst > secLast) { // scans are not ordered, we need to search min/max static bool warned = false; if(!warned) { ROS_WARN("Timestamp channel is not ordered, we will have to parse every scans to " "determinate first and last time offsets. This will add slightly computation " "time. This warning is only shown once."); warned = true; } if(timeOnColumns) { for(size_t i=0; i secLast) { secLast = sec; } } } else { for(size_t i=0; i secLast) { secLast = sec; } } } } if(secFirst > 1.e18) { // convert nanoseconds to seconds secFirst /= 1.e9; secLast /= 1.e9; } else if(secFirst > 1.e15) { // convert microseconds to seconds secFirst /= 1.e6; secLast /= 1.e6; } else if(secFirst > 1.e12) { // convert milliseconds to seconds secFirst /= 1.e3; secLast /= 1.e3; } firstStamp = ros::Time(secFirst); lastStamp = ros::Time(secLast); } else { ROS_ERROR("Not supported time datatype %d!", timeDatatype); return false; } if(!(timeDatatype >=6 && timeDatatype<=8)) { ROS_ERROR("Only lidar timestamp channel data type 6, 7 or 8 is supported! (received %d)", timeDatatype); return false; } if(lastStamp < firstStamp) { ROS_ERROR("Last stamp (%f) is smaller than first stamp (%f) (header=%f)!", lastStamp.toSec(), firstStamp.toSec(), input.header.stamp.toSec()); return false; } else if(lastStamp == firstStamp) { ROS_ERROR("First and last stamps in the scan are the same (%f) (header=%f)!", lastStamp.toSec(), input.header.stamp.toSec()); return false; } std::string errorMsg; if(listener != 0 && waitForTransform>0.0 && !listener->waitForTransform( input.header.frame_id, firstStamp, input.header.frame_id, lastStamp, fixedFrameId, ros::Duration(waitForTransform), ros::Duration(0.01), &errorMsg)) { ROS_ERROR("Could not estimate motion of %s accordingly to fixed frame %s between stamps %f and %f! (%s)", input.header.frame_id.c_str(), fixedFrameId.c_str(), firstStamp.toSec(), lastStamp.toSec(), errorMsg.c_str()); return false; } rtabmap::Transform firstPose; rtabmap::Transform lastPose; double scanTime = 0; if(slerp) { if(listener != 0) { firstPose = rtabmap_conversions::getMovingTransform( input.header.frame_id, fixedFrameId, input.header.stamp, firstStamp, *listener, 0); lastPose = rtabmap_conversions::getMovingTransform( input.header.frame_id, fixedFrameId, input.header.stamp, lastStamp, *listener, 0); } else { float vx,vy,vz, vroll,vpitch,vyaw; velocity.getTranslationAndEulerAngles(vx,vy,vz, vroll,vpitch,vyaw); // We need three poses: // 1- The pose of base frame in odom frame at first stamp // 2- The pose of base frame in odom frame at msg stamp // 3- The pose of base frame in odom frame at last stamp UASSERT(firstStamp.toSec() >= previousStamp); UASSERT(lastStamp.toSec() > previousStamp); double dt1 = firstStamp.toSec() - previousStamp; double dt2 = input.header.stamp.toSec() - previousStamp; double dt3 = lastStamp.toSec() - previousStamp; rtabmap::Transform p1(vx*dt1, vy*dt1, vz*dt1, vroll*dt1, vpitch*dt1, vyaw*dt1); rtabmap::Transform p2(vx*dt2, vy*dt2, vz*dt2, vroll*dt2, vpitch*dt2, vyaw*dt2); rtabmap::Transform p3(vx*dt3, vy*dt3, vz*dt3, vroll*dt3, vpitch*dt3, vyaw*dt3); // First and last poses are relative to stamp of the msg firstPose = p2.inverse() * p1; lastPose = p2.inverse() * p3; } if(firstPose.isNull()) { ROS_ERROR("Could not get transform of %s accordingly to %s between stamps %f and %f!", input.header.frame_id.c_str(), fixedFrameId.empty()?"velocity":fixedFrameId.c_str(), firstStamp.toSec(), input.header.stamp.toSec()); return false; } if(lastPose.isNull()) { ROS_ERROR("Could not get transform of %s accordingly to %s between stamps %f and %f!", input.header.frame_id.c_str(), fixedFrameId.empty()?"velocity":fixedFrameId.c_str(), lastStamp.toSec(), input.header.stamp.toSec()); return false; } scanTime = lastStamp.toSec() - firstStamp.toSec(); } //else tf will be used to get more accurate transforms output = input; ros::Time stamp; UTimer processingTime; if(timeOnColumns) { // ouster point cloud: // t1 t2 ... // ring1 ring1 ... // ring2 ring2 ... // ring3 ring4 ... // ring4 ring3 ... for(size_t u=0; u 1.e18) { // convert nanoseconds to seconds sec /= 1.e9; } else if(sec > 1.e15) { // convert microseconds to seconds sec /= 1.e6; } else if(sec > 1.e12) { // sec milliseconds to seconds sec /= 1.e3; } stamp = ros::Time(sec); } rtabmap::Transform transform; if(slerp) { transform = firstPose.interpolate((stamp-firstStamp).toSec() / scanTime, lastPose); } else { transform = rtabmap_conversions::getMovingTransform( output.header.frame_id, fixedFrameId, output.header.stamp, stamp, *listener, 0); if(transform.isNull()) { ROS_ERROR("Could not get transform of %s accordingly to %s between stamps %f and %f!", output.header.frame_id.c_str(), fixedFrameId.c_str(), stamp.toSec(), output.header.stamp.toSec()); return false; } } for(size_t v=0; v 1.e18) { // convert nanoseconds to seconds sec /= 1.e9; } else if(sec > 1.e15) { // convert microseconds to seconds sec /= 1.e6; } else if(sec > 1.e12) { // sec milliseconds to seconds sec /= 1.e3; } stamp = ros::Time(sec); } rtabmap::Transform transform; if(slerp) { transform = firstPose.interpolate((stamp-firstStamp).toSec() / scanTime, lastPose); } else { transform = rtabmap_conversions::getMovingTransform( output.header.frame_id, fixedFrameId, output.header.stamp, stamp, *listener, 0); if(transform.isNull()) { ROS_ERROR("Could not get transform of %s accordingly to %s between stamps %f and %f!", output.header.frame_id.c_str(), fixedFrameId.c_str(), stamp.toSec(), output.header.stamp.toSec()); return false; } } for(size_t u=0; u