Files
rtabmap/corelib/src/camera/CameraStereoZed.cpp
T

1022 lines
35 KiB
C++
Raw Normal View History

2018-10-01 19:33:56 -04:00
/*
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/core/camera/CameraStereoZed.h>
#include <rtabmap/utilite/UTimer.h>
#include <rtabmap/utilite/UThread.h>
2018-10-01 19:33:56 -04:00
#include <rtabmap/utilite/UConversion.h>
#include <rtabmap/utilite/UMath.h>
2018-10-01 19:33:56 -04:00
#ifdef RTABMAP_ZED
#include <sl/Camera.hpp>
#endif
namespace rtabmap
{
#ifdef RTABMAP_ZED
static cv::Mat slMat2cvMat(sl::Mat& input) {
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
//convert MAT_TYPE to CV_TYPE
int cv_type = -1;
switch (input.getDataType()) {
case sl::MAT_TYPE_32F_C1: cv_type = CV_32FC1; break;
case sl::MAT_TYPE_32F_C2: cv_type = CV_32FC2; break;
case sl::MAT_TYPE_32F_C3: cv_type = CV_32FC3; break;
case sl::MAT_TYPE_32F_C4: cv_type = CV_32FC4; break;
case sl::MAT_TYPE_8U_C1: cv_type = CV_8UC1; break;
case sl::MAT_TYPE_8U_C2: cv_type = CV_8UC2; break;
case sl::MAT_TYPE_8U_C3: cv_type = CV_8UC3; break;
case sl::MAT_TYPE_8U_C4: cv_type = CV_8UC4; break;
default: break;
}
// cv::Mat data requires a uchar* pointer. Therefore, we get the uchar1 pointer from sl::Mat (getPtr<T>())
//cv::Mat and sl::Mat will share the same memory pointer
return cv::Mat(input.getHeight(), input.getWidth(), cv_type, input.getPtr<sl::uchar1>(sl::MEM_CPU));
2020-01-31 20:17:10 +01:00
#else
//convert MAT_TYPE to CV_TYPE
int cv_type = -1;
switch (input.getDataType()) {
case sl::MAT_TYPE::F32_C1: cv_type = CV_32FC1; break;
case sl::MAT_TYPE::F32_C2: cv_type = CV_32FC2; break;
case sl::MAT_TYPE::F32_C3: cv_type = CV_32FC3; break;
case sl::MAT_TYPE::F32_C4: cv_type = CV_32FC4; break;
case sl::MAT_TYPE::U8_C1: cv_type = CV_8UC1; break;
case sl::MAT_TYPE::U8_C2: cv_type = CV_8UC2; break;
case sl::MAT_TYPE::U8_C3: cv_type = CV_8UC3; break;
case sl::MAT_TYPE::U8_C4: cv_type = CV_8UC4; break;
default: break;
}
// cv::Mat data requires a uchar* pointer. Therefore, we get the uchar1 pointer from sl::Mat (getPtr<T>())
//cv::Mat and sl::Mat will share the same memory pointer
return cv::Mat(input.getHeight(), input.getWidth(), cv_type, input.getPtr<sl::uchar1>(sl::MEM::CPU));
#endif
}
static Transform zedPoseToTransform(const sl::Pose & pose)
{
return Transform(
pose.pose_data.m[0], pose.pose_data.m[1], pose.pose_data.m[2], pose.pose_data.m[3],
pose.pose_data.m[4], pose.pose_data.m[5], pose.pose_data.m[6], pose.pose_data.m[7],
pose.pose_data.m[8], pose.pose_data.m[9], pose.pose_data.m[10], pose.pose_data.m[11]);
}
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
static IMU zedIMUtoIMU(const sl::IMUData & imuData, const Transform & imuLocalTransform)
{
sl::Orientation orientation = imuData.pose_data.getOrientation();
//Convert zed imu orientation from camera frame to world frame ENU!
Transform opticalTransform(0, 0, 1, 0, -1, 0, 0, 0, 0, -1, 0, 0);
Transform orientationT(0,0,0, orientation.ox, orientation.oy, orientation.oz, orientation.ow);
orientationT = opticalTransform * orientationT;
static double deg2rad = 0.017453293;
Eigen::Vector4d accT = Eigen::Vector4d(imuData.linear_acceleration.v[0], imuData.linear_acceleration.v[1], imuData.linear_acceleration.v[2], 1);
Eigen::Vector4d gyrT = Eigen::Vector4d(imuData.angular_velocity.v[0]*deg2rad, imuData.angular_velocity.v[1]*deg2rad, imuData.angular_velocity.v[2]*deg2rad, 1);
cv::Mat orientationCov = (cv::Mat_<double>(3,3)<<
imuData.pose_covariance[21], imuData.pose_covariance[22], imuData.pose_covariance[23],
imuData.pose_covariance[27], imuData.pose_covariance[28], imuData.pose_covariance[29],
imuData.pose_covariance[33], imuData.pose_covariance[34], imuData.pose_covariance[35]);
cv::Mat angCov = (cv::Mat_<double>(3,3)<<
imuData.angular_velocity_convariance.r[0], imuData.angular_velocity_convariance.r[1], imuData.angular_velocity_convariance.r[2],
imuData.angular_velocity_convariance.r[3], imuData.angular_velocity_convariance.r[4], imuData.angular_velocity_convariance.r[5],
imuData.angular_velocity_convariance.r[6], imuData.angular_velocity_convariance.r[7], imuData.angular_velocity_convariance.r[8]);
cv::Mat accCov = (cv::Mat_<double>(3,3)<<
imuData.linear_acceleration_convariance.r[0], imuData.linear_acceleration_convariance.r[1], imuData.linear_acceleration_convariance.r[2],
imuData.linear_acceleration_convariance.r[3], imuData.linear_acceleration_convariance.r[4], imuData.linear_acceleration_convariance.r[5],
imuData.linear_acceleration_convariance.r[6], imuData.linear_acceleration_convariance.r[7], imuData.linear_acceleration_convariance.r[8]);
Eigen::Quaternionf quat = orientationT.getQuaternionf();
return IMU(
cv::Vec4d(quat.x(), quat.y(), quat.z(), quat.w()),
orientationCov,
cv::Vec3d(gyrT[0], gyrT[1], gyrT[2]),
angCov,
cv::Vec3d(accT[0], accT[1], accT[2]),
accCov,
imuLocalTransform);
}
2020-01-31 20:17:10 +01:00
#else
static IMU zedIMUtoIMU(const sl::SensorsData & sensorData, const Transform & imuLocalTransform)
2020-01-31 20:17:10 +01:00
{
sl::Orientation orientation = sensorData.imu.pose.getOrientation();
//Convert zed imu orientation from camera frame to world frame ENU!
Transform opticalTransform(0, 0, 1, 0, -1, 0, 0, 0, 0, -1, 0, 0);
Transform orientationT(0,0,0, orientation.ox, orientation.oy, orientation.oz, orientation.ow);
orientationT = opticalTransform * orientationT;
static double deg2rad = 0.017453293;
Eigen::Vector4d accT = Eigen::Vector4d(sensorData.imu.linear_acceleration.v[0], sensorData.imu.linear_acceleration.v[1], sensorData.imu.linear_acceleration.v[2], 1);
Eigen::Vector4d gyrT = Eigen::Vector4d(sensorData.imu.angular_velocity.v[0]*deg2rad, sensorData.imu.angular_velocity.v[1]*deg2rad, sensorData.imu.angular_velocity.v[2]*deg2rad, 1);
cv::Mat orientationCov = (cv::Mat_<double>(3,3)<<
sensorData.imu.pose_covariance.r[0], sensorData.imu.pose_covariance.r[1], sensorData.imu.pose_covariance.r[2],
sensorData.imu.pose_covariance.r[3], sensorData.imu.pose_covariance.r[4], sensorData.imu.pose_covariance.r[5],
sensorData.imu.pose_covariance.r[6], sensorData.imu.pose_covariance.r[7], sensorData.imu.pose_covariance.r[8]);
cv::Mat angCov = (cv::Mat_<double>(3,3)<<
sensorData.imu.angular_velocity_covariance.r[0], sensorData.imu.angular_velocity_covariance.r[1], sensorData.imu.angular_velocity_covariance.r[2],
sensorData.imu.angular_velocity_covariance.r[3], sensorData.imu.angular_velocity_covariance.r[4], sensorData.imu.angular_velocity_covariance.r[5],
sensorData.imu.angular_velocity_covariance.r[6], sensorData.imu.angular_velocity_covariance.r[7], sensorData.imu.angular_velocity_covariance.r[8]);
cv::Mat accCov = (cv::Mat_<double>(3,3)<<
sensorData.imu.linear_acceleration_covariance.r[0], sensorData.imu.linear_acceleration_covariance.r[1], sensorData.imu.linear_acceleration_covariance.r[2],
sensorData.imu.linear_acceleration_covariance.r[3], sensorData.imu.linear_acceleration_covariance.r[4], sensorData.imu.linear_acceleration_covariance.r[5],
sensorData.imu.linear_acceleration_covariance.r[6], sensorData.imu.linear_acceleration_covariance.r[7], sensorData.imu.linear_acceleration_covariance.r[8]);
Eigen::Quaternionf quat = orientationT.getQuaternionf();
return IMU(
cv::Vec4d(quat.x(), quat.y(), quat.z(), quat.w()),
orientationCov,
cv::Vec3d(gyrT[0], gyrT[1], gyrT[2]),
angCov,
cv::Vec3d(accT[0], accT[1], accT[2]),
accCov,
imuLocalTransform);
}
// sl::SensorsData::imu.is_available only means the camera has an IMU; a returned
// sample can still contain NaN pose/accel/gyro (e.g. before the IMU fusion has
// initialized, or in SVO/STREAM mode). Validate the actual measurements.
static bool isImuValid(const sl::SensorsData & sensorData)
{
if(!sensorData.imu.is_available)
{
return false;
}
const sl::float3 & acc = sensorData.imu.linear_acceleration;
const sl::float3 & gyr = sensorData.imu.angular_velocity;
const sl::Orientation ori = sensorData.imu.pose.getOrientation();
return uIsFinite(acc.v[0]) && uIsFinite(acc.v[1]) && uIsFinite(acc.v[2]) &&
uIsFinite(gyr.v[0]) && uIsFinite(gyr.v[1]) && uIsFinite(gyr.v[2]) &&
uIsFinite(ori.ox) && uIsFinite(ori.oy) && uIsFinite(ori.oz) && uIsFinite(ori.ow);
}
2020-01-31 20:17:10 +01:00
#endif
class ZedIMUThread: public UThread
{
public:
ZedIMUThread(float rate, sl::Camera * zed, CameraStereoZed * camera, const Transform & imuLocalTransform, bool accurate)
{
UASSERT(rate > 0.0f);
UASSERT(zed != 0 && camera != 0);
rate_ = rate;
zed_= zed;
camera_ = camera;
accurate_ = accurate;
imuLocalTransform_ = imuLocalTransform;
}
private:
virtual void mainLoopBegin()
{
frameRateTimer_.start();
}
virtual void mainLoop()
{
double delay = 1000.0/double(rate_);
int sleepTime = delay - 1000.0f*frameRateTimer_.getElapsedTime();
if(sleepTime > 0)
{
if(accurate_)
{
if(sleepTime > 1)
{
uSleep(sleepTime-1);
}
// Add precision at the cost of a small overhead
delay/=1000.0;
while(frameRateTimer_.getElapsedTime() < delay-0.000001)
{
//
}
}
else
{
uSleep(sleepTime);
}
}
frameRateTimer_.start();
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
sl::IMUData imudata;
bool res = zed_->getIMUData(imudata, sl::TIME_REFERENCE_IMAGE);
if(res == sl::SUCCESS && imudata.valid)
{
this->postInterIMU(zedIMUtoIMU(imudata, imuLocalTransform_), UTimer::now());
}
2020-01-31 20:17:10 +01:00
#else
sl::SensorsData sensordata;
sl::ERROR_CODE res = zed_->getSensorsData(sensordata, sl::TIME_REFERENCE::CURRENT);
if(res == sl::ERROR_CODE::SUCCESS && isImuValid(sensordata))
2020-01-31 20:17:10 +01:00
{
camera_->postInterIMUPublic(zedIMUtoIMU(sensordata, imuLocalTransform_), double(sensordata.imu.timestamp.getNanoseconds())/10e8);
2020-01-31 20:17:10 +01:00
}
#endif
}
float rate_;
sl::Camera * zed_;
CameraStereoZed * camera_;
bool accurate_;
Transform imuLocalTransform_;
UTimer frameRateTimer_;
};
#endif
bool CameraStereoZed::available()
{
#ifdef RTABMAP_ZED
return true;
#else
return false;
#endif
}
2023-04-16 18:47:15 -07:00
int CameraStereoZed::sdkVersion()
{
#ifdef RTABMAP_ZED
return ZED_SDK_MAJOR_VERSION;
#else
return -1;
#endif
}
#ifdef RTABMAP_ZED
static void backwardCompatibility(int & resolution, int & quality)
{
// -1 = AUTO, 0=HD4K 1=QHDPLUS 2=HD2K 3=HD1536 4=HD1080 5=HD1200 6=HD720 7=SVGA 8=VGA 9=XVGA 10=TXVGA
#if ZED_SDK_MAJOR_VERSION < 4
// Zed3 Supported: 2=HD2K 4=HD1080 6=HD720 8=VGA
if(resolution == 0 || resolution == 1) // 0=HD4K 1=QHDPLUS
{
UWARN("Zed SDK v3 doesn't support HD4K and QHDPLUS, setting HD2K.");
resolution = 2; // 2=HD2K
}
if(resolution == 3 || resolution == 5) // 3=HD1536 5=HD1200
{
UWARN("Zed SDK v3 doesn't support HD1536 and HD1200, setting HD1080.");
resolution = 4; // 4=HD1080
}
if(resolution == 7 || resolution >=9) // 7=SVGA 9=XVGA 10=TXVGA
{
UWARN("Zed SDK v3 doesn't support SVGA, XVGA and TXVGA, setting VGA.");
resolution = 8; // 8=VGA
}
if(quality == 4) { // 4=NEURAL_LIGHT 6=NEURAL_PLUS
UWARN("Zed SDK v3 doesn't support NEURAL_LIGHT and NEURAL PLUS, setting NEURAL.");
quality = 5; // NEURAL
}
#else
if(resolution == -1)
{
resolution = int(sl::RESOLUTION::AUTO); // AUTO
}
#if ZED_SDK_MAJOR_VERSION < 5
// Zed4 Supported: 0=HD4K 1=QHDPLUS 2=HD2K 4=HD1080 5=HD1200 6=HD720 7=SVGA 8=VGA
if(resolution == 3) // 3=HD1536
{
UWARN("Zed SDK v4 doesn't support HD1536, setting HD1200.");
resolution_ = 5; // 5=HD1200
}
if(resolution >=9) // 9=XVGA 10=TXVGA
{
UWARN("Zed SDK v4 doesn't support XVGA and TXVGA, setting VGA.");
resolution = 8; // 8=VGA
}
if(quality > 5) { // NEURAL_PLUS
UWARN("Zed SDK v4 doesn't support NEURAL_LIGHT and NEURAL PLUS, setting NEURAL.");
quality = 5; // NEURAL
}
#endif
#endif
}
#endif
2023-04-16 18:47:15 -07:00
CameraStereoZed::CameraStereoZed(
int deviceId,
int resolution,
int quality,
int sensingMode,
int confidenceThr,
bool computeOdometry,
float imageRate,
const Transform & localTransform,
bool selfCalibration,
bool odomForce3DoF,
int texturenessConfidenceThr) :
Camera(imageRate, localTransform)
#ifdef RTABMAP_ZED
,
zed_(0),
src_(CameraVideo::kUsbDevice),
usbDevice_(deviceId),
svoFilePath_(""),
resolution_(resolution),
quality_(quality),
selfCalibration_(selfCalibration),
sensingMode_(sensingMode),
confidenceThr_(confidenceThr),
texturenessConfidenceThr_(texturenessConfidenceThr),
computeOdometry_(computeOdometry),
lost_(true),
force3DoF_(odomForce3DoF),
imuPublishingThread_(0)
#endif
{
UDEBUG("");
#ifdef RTABMAP_ZED
backwardCompatibility(resolution_, quality_);
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
UASSERT(resolution_ >= sl::RESOLUTION_HD2K && resolution_ <sl::RESOLUTION_LAST);
UASSERT(quality_ >= sl::DEPTH_MODE_NONE && quality_ <sl::DEPTH_MODE_LAST);
UASSERT(sensingMode_ >= sl::SENSING_MODE_STANDARD && sensingMode_ <sl::SENSING_MODE_LAST);
UASSERT(confidenceThr_ >= 0 && confidenceThr_ <=100);
2020-01-31 20:17:10 +01:00
#else
sl::RESOLUTION res = static_cast<sl::RESOLUTION>(resolution_);
sl::DEPTH_MODE qual = static_cast<sl::DEPTH_MODE>(quality_);
UASSERT(qual >= sl::DEPTH_MODE::NONE && qual < sl::DEPTH_MODE::LAST);
2023-04-16 18:47:15 -07:00
#if ZED_SDK_MAJOR_VERSION < 4
UASSERT(res >= sl::RESOLUTION::HD2K && res < sl::RESOLUTION::LAST);
2023-04-16 18:47:15 -07:00
sl::SENSING_MODE sens = static_cast<sl::SENSING_MODE>(sensingMode_);
2020-01-31 20:17:10 +01:00
UASSERT(sens >= sl::SENSING_MODE::STANDARD && sens < sl::SENSING_MODE::LAST);
2023-04-16 18:47:15 -07:00
#else
UASSERT(res >= sl::RESOLUTION::HD4K && res < sl::RESOLUTION::LAST);
2023-04-16 18:47:15 -07:00
UASSERT(sensingMode_ >= 0 && sensingMode_ < 2);
#endif
2020-01-31 20:17:10 +01:00
UASSERT(confidenceThr_ >= 0 && confidenceThr_ <=100);
UASSERT(texturenessConfidenceThr_ >= 0 && texturenessConfidenceThr_ <=100);
2020-01-31 20:17:10 +01:00
#endif
#endif
}
CameraStereoZed::CameraStereoZed(
const std::string & filePath,
int quality,
int sensingMode,
int confidenceThr,
bool computeOdometry,
float imageRate,
const Transform & localTransform,
bool selfCalibration,
bool odomForce3DoF,
int texturenessConfidenceThr) :
Camera(imageRate, localTransform)
#ifdef RTABMAP_ZED
,
zed_(0),
src_(CameraVideo::kVideoFile),
usbDevice_(0),
svoFilePath_(filePath),
#if ZED_SDK_MAJOR_VERSION < 3
resolution_(sl::RESOLUTION_HD720),
#elif ZED_SDK_MAJOR_VERSION < 4
resolution_(sl::RESOLUTION::HD720),
#else
resolution_(int(sl::RESOLUTION::AUTO)),
#endif
quality_(quality),
selfCalibration_(selfCalibration),
sensingMode_(sensingMode),
confidenceThr_(confidenceThr),
texturenessConfidenceThr_(texturenessConfidenceThr),
computeOdometry_(computeOdometry),
lost_(true),
force3DoF_(odomForce3DoF),
imuPublishingThread_(0)
#endif
{
UDEBUG("");
#ifdef RTABMAP_ZED
backwardCompatibility(resolution_, quality_);
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
UASSERT(resolution_ >= sl::RESOLUTION_HD2K && resolution_ <sl::RESOLUTION_LAST);
UASSERT(quality_ >= sl::DEPTH_MODE_NONE && quality_ <sl::DEPTH_MODE_LAST);
UASSERT(sensingMode_ >= sl::SENSING_MODE_STANDARD && sensingMode_ <sl::SENSING_MODE_LAST);
UASSERT(confidenceThr_ >= 0 && confidenceThr_ <=100);
2020-01-31 20:17:10 +01:00
#else
sl::RESOLUTION res = static_cast<sl::RESOLUTION>(resolution_);
sl::DEPTH_MODE qual = static_cast<sl::DEPTH_MODE>(quality_);
UASSERT(res >= sl::RESOLUTION::HD2K && res < sl::RESOLUTION::LAST);
UASSERT(qual >= sl::DEPTH_MODE::NONE && qual < sl::DEPTH_MODE::LAST);
2023-04-16 18:47:15 -07:00
#if ZED_SDK_MAJOR_VERSION < 4
sl::SENSING_MODE sens = static_cast<sl::SENSING_MODE>(sensingMode_);
2020-01-31 20:17:10 +01:00
UASSERT(sens >= sl::SENSING_MODE::STANDARD && sens < sl::SENSING_MODE::LAST);
2023-04-16 18:47:15 -07:00
#else
UASSERT(sensingMode_ >= 0 && sensingMode_ < 2);
#endif
2020-01-31 20:17:10 +01:00
UASSERT(confidenceThr_ >= 0 && confidenceThr_ <=100);
UASSERT(texturenessConfidenceThr_ >= 0 && texturenessConfidenceThr_ <=100);
2020-01-31 20:17:10 +01:00
#endif
#endif
}
CameraStereoZed::~CameraStereoZed()
{
#ifdef RTABMAP_ZED
if(imuPublishingThread_)
{
imuPublishingThread_->join(true);
}
delete imuPublishingThread_;
delete zed_;
#endif
}
std::string CameraStereoZed::getNeuralModelWarning(int quality)
{
(void)quality;
#if defined(RTABMAP_ZED) && ZED_SDK_MAJOR_VERSION >= 5
sl::DEPTH_MODE depthMode = (sl::DEPTH_MODE)quality;
if(depthMode == sl::DEPTH_MODE::NEURAL_LIGHT ||
depthMode == sl::DEPTH_MODE::NEURAL ||
depthMode == sl::DEPTH_MODE::NEURAL_PLUS)
{
sl::AI_MODELS aiModel =
depthMode == sl::DEPTH_MODE::NEURAL_LIGHT ? sl::AI_MODELS::NEURAL_LIGHT_DEPTH :
depthMode == sl::DEPTH_MODE::NEURAL_PLUS ? sl::AI_MODELS::NEURAL_PLUS_DEPTH :
sl::AI_MODELS::NEURAL_DEPTH;
sl::AI_Model_status status = sl::checkAIModelStatus(aiModel);
if(!status.downloaded || !status.optimized)
{
return uFormat("The selected ZED NEURAL depth model is not ready yet (downloaded=%s, "
"optimized=%s): the first start may take significantly longer while the "
"ZED SDK downloads and/or optimizes the model for your GPU. Subsequent "
"starts will be faster.",
status.downloaded?"true":"false", status.optimized?"true":"false");
}
}
#endif
return std::string();
}
2018-10-01 19:33:56 -04:00
bool CameraStereoZed::init(const std::string & calibrationFolder, const std::string & cameraName)
{
UDEBUG("");
#ifdef RTABMAP_ZED
if(imuPublishingThread_)
{
imuPublishingThread_->join(true);
delete imuPublishingThread_;
imuPublishingThread_=0;
}
2018-10-01 19:33:56 -04:00
if(zed_)
{
delete zed_;
zed_ = 0;
}
lost_ = true;
sl::InitParameters param;
param.camera_resolution=static_cast<sl::RESOLUTION>(resolution_);
2020-01-31 20:17:10 +01:00
param.camera_fps=getImageRate();
2018-10-01 19:33:56 -04:00
param.depth_mode=(sl::DEPTH_MODE)quality_;
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
param.camera_linux_id=usbDevice_;
2018-10-01 19:33:56 -04:00
param.coordinate_units=sl::UNIT_METER;
param.coordinate_system=(sl::COORDINATE_SYSTEM)sl::COORDINATE_SYSTEM_IMAGE ;
2020-01-31 20:17:10 +01:00
#else
param.coordinate_units=sl::UNIT::METER;
param.coordinate_system=sl::COORDINATE_SYSTEM::IMAGE ;
#endif
2018-10-01 19:33:56 -04:00
param.sdk_verbose=true;
param.sdk_gpu_id=-1;
param.depth_minimum_distance=-1;
param.camera_disable_self_calib=!selfCalibration_;
sl::ERROR_CODE r = sl::ERROR_CODE::SUCCESS;
if(src_ == CameraVideo::kVideoFile)
{
UINFO("svo file = %s", svoFilePath_.c_str());
zed_ = new sl::Camera(); // Use in SVO playback mode
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
param.svo_input_filename=svoFilePath_.c_str();
2020-01-31 20:17:10 +01:00
#else
param.input.setFromSVOFile(svoFilePath_.c_str());
#endif
2018-10-01 19:33:56 -04:00
r = zed_->open(param);
}
else
{
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION >= 3
param.input.setFromCameraID(usbDevice_);
#endif
2018-10-01 19:33:56 -04:00
UINFO("Resolution=%d imagerate=%f device=%d", resolution_, getImageRate(), usbDevice_);
zed_ = new sl::Camera(); // Use in Live Mode
r = zed_->open(param);
}
#if ZED_SDK_MAJOR_VERSION >= 3
// CORRUPTED_SDK_INSTALLATION on open() is typically a NEURAL depth mode whose optional
// neural/TensorRT runtime files aren't installed. Give a clear, actionable error.
if(r == sl::ERROR_CODE::CORRUPTED_SDK_INSTALLATION &&
#if ZED_SDK_MAJOR_VERSION >= 5
param.depth_mode >= sl::DEPTH_MODE::NEURAL_LIGHT)
#else
param.depth_mode >= sl::DEPTH_MODE::NEURAL)
#endif
{
UERROR("ZED open() returned \"%s\": the optional NEURAL/TensorRT runtime files are "
"likely missing. Install ZED SDK %d.%d, or select a non-NEURAL depth mode "
"(e.g. PERFORMANCE).", toString(r).c_str(), ZED_SDK_MAJOR_VERSION, ZED_SDK_MINOR_VERSION);
// Do NOT delete zed_ here: after a CORRUPTED_SDK_INSTALLATION open failure (NEURAL depth
// mode selected but the TensorRT/neural runtime isn't installed) the ZED SDK is left in a
// bad state and ~sl::Camera() crashes inside sl_zed64.dll. Leak the object (one-time,
// terminal error path) so we fail gracefully with the message above instead of crashing.
zed_ = 0;
return false;
}
#endif
2018-10-01 19:33:56 -04:00
if(r!=sl::ERROR_CODE::SUCCESS)
{
#if ZED_SDK_MAJOR_VERSION >= 4
UERROR("Camera initialization failed: \"%s\": %s", toString(r).c_str(), toVerbose(r).c_str());
#else
2018-10-01 19:33:56 -04:00
UERROR("Camera initialization failed: \"%s\"", toString(r).c_str());
#endif
2018-10-01 19:33:56 -04:00
delete zed_;
zed_ = 0;
return false;
}
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
UINFO("Init ZED: Mode=%d Unit=%d CoordinateSystem=%d Verbose=false device=-1 minDist=-1 self-calibration=%s vflip=false",
quality_, sl::UNIT_METER, sl::COORDINATE_SYSTEM_IMAGE , selfCalibration_?"true":"false");
2020-01-31 20:17:10 +01:00
if(quality_!=sl::DEPTH_MODE_NONE)
{
zed_->setConfidenceThreshold(confidenceThr_);
}
#else
UINFO("Init ZED: Mode=%d Unit=%d CoordinateSystem=%d Verbose=false device=-1 minDist=-1 self-calibration=%s vflip=false",
quality_, sl::UNIT::METER, sl::COORDINATE_SYSTEM::IMAGE , selfCalibration_?"true":"false");
#endif
UDEBUG("");
2018-10-01 19:33:56 -04:00
if (computeOdometry_)
{
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
sl::TrackingParameters tparam;
2020-01-31 20:17:10 +01:00
tparam.enable_spatial_memory=false;
r = zed_->enableTracking(tparam);
#else
sl::PositionalTrackingParameters tparam;
tparam.enable_area_memory=false;
r = zed_->enablePositionalTracking(tparam);
#endif
if(r!=sl::ERROR_CODE::SUCCESS)
2018-10-01 19:33:56 -04:00
{
#if ZED_SDK_MAJOR_VERSION >= 4
UERROR("Camera tracking initialization failed: \"%s\": %s", toString(r).c_str(), toVerbose(r).c_str());
#else
2018-10-01 19:33:56 -04:00
UERROR("Camera tracking initialization failed: \"%s\"", toString(r).c_str());
#endif
2018-10-01 19:33:56 -04:00
}
}
sl::CameraInformation infos = zed_->getCameraInformation();
2023-04-16 18:47:15 -07:00
#if ZED_SDK_MAJOR_VERSION < 4
2018-10-01 19:33:56 -04:00
sl::CalibrationParameters *stereoParams = &(infos.calibration_parameters );
2023-04-16 18:47:15 -07:00
#else
sl::CalibrationParameters *stereoParams = &(infos.camera_configuration.calibration_parameters );
#endif
2018-10-01 19:33:56 -04:00
sl::Resolution res = stereoParams->left_cam.image_size;
2024-01-31 07:17:04 -08:00
#if ZED_SDK_MAJOR_VERSION < 4
stereoModel_ = StereoCameraModel(
stereoParams->left_cam.fx,
stereoParams->left_cam.fy,
stereoParams->left_cam.cx,
stereoParams->left_cam.cy,
stereoParams->T[0],//baseline
this->getLocalTransform(),
cv::Size(res.width, res.height));
#else
2018-10-01 19:33:56 -04:00
stereoModel_ = StereoCameraModel(
stereoParams->left_cam.fx,
stereoParams->left_cam.fy,
stereoParams->left_cam.cx,
stereoParams->left_cam.cy,
2023-04-16 18:47:15 -07:00
stereoParams->getCameraBaseline(),
2018-10-01 19:33:56 -04:00
this->getLocalTransform(),
cv::Size(res.width, res.height));
2024-01-31 07:17:04 -08:00
#endif
2018-10-01 19:33:56 -04:00
2024-01-31 07:17:04 -08:00
#if ZED_SDK_MAJOR_VERSION < 4
UINFO("Calibration: fx=%f, fy=%f, cx=%f, cy=%f, baseline=%f, width=%d, height=%d, transform=%s",
stereoParams->left_cam.fx,
stereoParams->left_cam.fy,
stereoParams->left_cam.cx,
stereoParams->left_cam.cy,
stereoParams->T[0],//baseline
(int)res.width,
(int)res.height,
this->getLocalTransform().prettyPrint().c_str());
#else
2018-10-01 19:33:56 -04:00
UINFO("Calibration: fx=%f, fy=%f, cx=%f, cy=%f, baseline=%f, width=%d, height=%d, transform=%s",
stereoParams->left_cam.fx,
stereoParams->left_cam.fy,
stereoParams->left_cam.cx,
stereoParams->left_cam.cy,
2023-04-16 18:47:15 -07:00
stereoParams->getCameraBaseline(),
2018-10-01 19:33:56 -04:00
(int)res.width,
(int)res.height,
this->getLocalTransform().prettyPrint().c_str());
2024-01-31 07:17:04 -08:00
#endif
2018-10-01 19:33:56 -04:00
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
if(infos.camera_model == sl::MODEL_ZED_M)
2020-01-31 20:17:10 +01:00
#else
if(infos.camera_model != sl::MODEL::ZED)
#endif
{
2023-04-16 18:47:15 -07:00
#if ZED_SDK_MAJOR_VERSION < 4
imuLocalTransform_ = this->getLocalTransform() * zedPoseToTransform(infos.camera_imu_transform).inverse();
2023-04-16 18:47:15 -07:00
#else
imuLocalTransform_ = this->getLocalTransform() * zedPoseToTransform(infos.sensors_configuration.camera_imu_transform).inverse();
#endif
2024-01-31 07:17:04 -08:00
2023-04-16 18:47:15 -07:00
#if ZED_SDK_MAJOR_VERSION < 4
2024-01-31 07:17:04 -08:00
UINFO("IMU local transform: %s (imu2cam=%s))",
imuLocalTransform_.prettyPrint().c_str(),
zedPoseToTransform(infos.camera_imu_transform).prettyPrint().c_str());
2023-04-16 18:47:15 -07:00
#else
2024-01-31 07:17:04 -08:00
UINFO("IMU local transform: %s (imu2cam=%s))",
imuLocalTransform_.prettyPrint().c_str(),
zedPoseToTransform(infos.sensors_configuration.camera_imu_transform).prettyPrint().c_str());
2023-04-16 18:47:15 -07:00
#endif
if(isInterIMUPublishing())
{
imuPublishingThread_ = new ZedIMUThread(200, zed_, this, imuLocalTransform_, true);
imuPublishingThread_->start();
}
}
2018-10-01 19:33:56 -04:00
return true;
#else
UERROR("CameraStereoZED: RTAB-Map is not built with ZED sdk support!");
#endif
return false;
}
bool CameraStereoZed::isCalibrated() const
{
#ifdef RTABMAP_ZED
return stereoModel_.isValidForProjection();
#else
return false;
#endif
}
std::string CameraStereoZed::getSerial() const
{
#ifdef RTABMAP_ZED
if(zed_)
{
return uFormat("%x", zed_->getCameraInformation ().serial_number);
}
#endif
return "";
}
bool CameraStereoZed::odomProvided() const
{
#ifdef RTABMAP_ZED
return computeOdometry_;
#else
return false;
#endif
}
bool CameraStereoZed::getPose(double stamp, Transform & pose, cv::Mat & covariance, double maxWaitTime)
2021-05-23 17:21:12 -04:00
{
#ifdef RTABMAP_ZED
if (computeOdometry_ && zed_)
{
sl::Pose p;
#if ZED_SDK_MAJOR_VERSION < 3
if(!zed_->grab())
{
return false;
}
sl::TRACKING_STATE tracking_state = zed_->getPosition(p);
if (tracking_state == sl::TRACKING_STATE_OK)
#else
if(zed_->grab()!=sl::ERROR_CODE::SUCCESS)
{
return false;
}
sl::POSITIONAL_TRACKING_STATE tracking_state = zed_->getPosition(p);
if (tracking_state == sl::POSITIONAL_TRACKING_STATE::OK)
#endif
{
int trackingConfidence = p.pose_confidence;
// FIXME What does pose_confidence == -1 mean?
if (trackingConfidence>0)
{
pose = zedPoseToTransform(p);
if (!pose.isNull())
{
//transform from:
// x->right, y->down, z->forward
//to:
// x->forward, y->left, z->up
pose = this->getLocalTransform() * pose * this->getLocalTransform().inverse();
if(force3DoF_)
{
pose = pose.to3DoF();
}
if (lost_)
{
covariance = cv::Mat::eye(6, 6, CV_64FC1) * 9999.0f; // don't know transform with previous pose
lost_ = false;
UDEBUG("Init %s (var=%f)", pose.prettyPrint().c_str(), 9999.0f);
}
else
{
covariance = cv::Mat::eye(6, 6, CV_64FC1) * 1.0f / float(trackingConfidence);
UDEBUG("Run %s (var=%f)", pose.prettyPrint().c_str(), 1.0f / float(trackingConfidence));
}
return true;
}
else
{
covariance = cv::Mat::eye(6, 6, CV_64FC1) * 9999.0f; // lost
lost_ = true;
UWARN("ZED lost! (trackingConfidence=%d)", trackingConfidence);
}
}
else
{
covariance = cv::Mat::eye(6, 6, CV_64FC1) * 9999.0f; // lost
lost_ = true;
UWARN("ZED lost! (trackingConfidence=%d)", trackingConfidence);
}
}
else
{
UWARN("Tracking not ok: state=\"%s\"", toString(tracking_state).c_str());
}
}
#endif
return false;
}
SensorData CameraStereoZed::captureImage(SensorCaptureInfo * info)
2018-10-01 19:33:56 -04:00
{
SensorData data;
#ifdef RTABMAP_ZED
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
sl::RuntimeParameters rparam((sl::SENSING_MODE)sensingMode_, quality_ > 0, quality_ > 0, sl::REFERENCE_FRAME_CAMERA);
2023-04-16 18:47:15 -07:00
#elif ZED_SDK_MAJOR_VERSION < 4
sl::RuntimeParameters rparam((sl::SENSING_MODE)sensingMode_, quality_ > 0, confidenceThr_, texturenessConfidenceThr_, sl::REFERENCE_FRAME::CAMERA);
2023-04-16 18:47:15 -07:00
#else
sl::RuntimeParameters rparam(quality_ > 0, sensingMode_ == 1, confidenceThr_, texturenessConfidenceThr_, sl::REFERENCE_FRAME::CAMERA);
2020-01-31 20:17:10 +01:00
#endif
2018-10-01 19:33:56 -04:00
if(zed_)
{
UTimer timer;
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
bool res = zed_->grab(rparam);
while (src_ == CameraVideo::kUsbDevice && res!=sl::SUCCESS && timer.elapsed() < 2.0)
{
// maybe there is a latency with the USB, try again in 10 ms (for the next 2 seconds)
uSleep(10);
res = zed_->grab(rparam);
}
2020-01-31 20:17:10 +01:00
if(res==sl::SUCCESS)
#else
sl::ERROR_CODE res;
sl::Timestamp timestamp;
bool imuReceived = true;
do
{
res = zed_->grab(rparam);
timestamp = zed_->getTimestamp(sl::TIME_REFERENCE::IMAGE);
// If the sensor supports IMU, wait for IMU to be available before sending data.
if(imuPublishingThread_ == 0 && !imuLocalTransform_.isNull())
{
sl::SensorsData imudatatmp;
res = zed_->getSensorsData(imudatatmp, sl::TIME_REFERENCE::IMAGE);
imuReceived = res == sl::ERROR_CODE::SUCCESS && isImuValid(imudatatmp) && imudatatmp.imu.timestamp.getNanoseconds() != 0;
}
}
while(src_ == CameraVideo::kUsbDevice && (res!=sl::ERROR_CODE::SUCCESS || !imuReceived) && timer.elapsed() < 2.0);
2020-01-31 20:17:10 +01:00
// If no valid IMU arrived within the 2 sec startup window, null the IMU transform so
// we don't re-wait 2 sec on every subsequent frame (camera likely has no working IMU).
if(imuPublishingThread_ == 0 && !imuLocalTransform_.isNull() && !imuReceived)
{
UWARN("No valid IMU received within 2 sec; ignoring IMU for the rest of this session.");
imuLocalTransform_.setNull();
}
2020-01-31 20:17:10 +01:00
if(res==sl::ERROR_CODE::SUCCESS)
#endif
2018-10-01 19:33:56 -04:00
{
// get left image
sl::Mat tmp;
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
zed_->retrieveImage(tmp,sl::VIEW_LEFT);
2020-01-31 20:17:10 +01:00
#else
zed_->retrieveImage(tmp,sl::VIEW::LEFT);
#endif
2018-10-01 19:33:56 -04:00
cv::Mat rgbaLeft = slMat2cvMat(tmp);
cv::Mat left;
cv::cvtColor(rgbaLeft, left, cv::COLOR_BGRA2BGR);
if(quality_ > 0)
{
// get depth image
cv::Mat depth;
sl::Mat tmp;
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
zed_->retrieveMeasure(tmp,sl::MEASURE_DEPTH);
2020-01-31 20:17:10 +01:00
#else
zed_->retrieveMeasure(tmp,sl::MEASURE::DEPTH);
#endif
2018-10-01 19:33:56 -04:00
slMat2cvMat(tmp).copyTo(depth);
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
data = SensorData(left, depth, stereoModel_.left(), this->getNextSeqID(), UTimer::now());
#else
data = SensorData(left, depth, stereoModel_.left(), this->getNextSeqID(), double(timestamp.getNanoseconds())/10e8);
#endif
2018-10-01 19:33:56 -04:00
}
else
{
// get right image
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
sl::Mat tmp;zed_->retrieveImage(tmp,sl::VIEW_RIGHT );
2020-01-31 20:17:10 +01:00
#else
sl::Mat tmp;zed_->retrieveImage(tmp,sl::VIEW::RIGHT );
#endif
2018-10-01 19:33:56 -04:00
cv::Mat rgbaRight = slMat2cvMat(tmp);
cv::Mat right;
if(rightGrayScale_)
cv::cvtColor(rgbaRight, right, cv::COLOR_BGRA2GRAY);
else
cv::cvtColor(rgbaRight, right, cv::COLOR_BGRA2BGR);
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
data = SensorData(left, right, stereoModel_, this->getNextSeqID(), UTimer::now());
#else
data = SensorData(left, right, stereoModel_, this->getNextSeqID(), double(timestamp.getNanoseconds())/10e8);
#endif
2018-10-01 19:33:56 -04:00
}
if(imuPublishingThread_ == 0 && !imuLocalTransform_.isNull())
{
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
sl::IMUData imudata;
res = zed_->getIMUData(imudata, sl::TIME_REFERENCE_IMAGE);
if(res == sl::SUCCESS && imudata.valid)
2020-01-31 20:17:10 +01:00
#else
sl::SensorsData imudata;
res = zed_->getSensorsData(imudata, sl::TIME_REFERENCE::IMAGE);
if(res == sl::ERROR_CODE::SUCCESS && isImuValid(imudata))
2020-01-31 20:17:10 +01:00
#endif
{
//ZED-Mini
data.setIMU(zedIMUtoIMU(imudata, imuLocalTransform_));
}
}
2018-10-01 19:33:56 -04:00
if (computeOdometry_ && info)
{
sl::Pose pose;
2020-01-31 20:17:10 +01:00
#if ZED_SDK_MAJOR_VERSION < 3
2018-10-01 19:33:56 -04:00
sl::TRACKING_STATE tracking_state = zed_->getPosition(pose);
if (tracking_state == sl::TRACKING_STATE_OK)
2020-01-31 20:17:10 +01:00
#else
sl::POSITIONAL_TRACKING_STATE tracking_state = zed_->getPosition(pose);
if (tracking_state == sl::POSITIONAL_TRACKING_STATE::OK)
#endif
2018-10-01 19:33:56 -04:00
{
int trackingConfidence = pose.pose_confidence;
// FIXME What does pose_confidence == -1 mean?
2023-01-28 16:33:24 -08:00
info->odomPose = zedPoseToTransform(pose);
if (!info->odomPose.isNull())
2018-10-01 19:33:56 -04:00
{
#if ZED_SDK_MAJOR_VERSION >=3
if(pose.timestamp != timestamp)
{
UWARN("Pose retrieve doesn't have same stamp (%ld) than grabbed image (%ld)", pose.timestamp, timestamp);
}
#endif
2023-01-28 16:33:24 -08:00
//transform from:
// x->right, y->down, z->forward
//to:
// x->forward, y->left, z->up
info->odomPose = this->getLocalTransform() * info->odomPose * this->getLocalTransform().inverse();
if(force3DoF_)
2018-10-01 19:33:56 -04:00
{
2023-01-28 16:33:24 -08:00
info->odomPose = info->odomPose.to3DoF();
2018-10-01 19:33:56 -04:00
}
2023-01-28 16:33:24 -08:00
if (lost_)
{
info->odomCovariance = cv::Mat::eye(6, 6, CV_64FC1) * 9999.0f; // don't know transform with previous pose
lost_ = false;
UDEBUG("Init %s (var=%f)", info->odomPose.prettyPrint().c_str(), 9999.0f);
}
else if(trackingConfidence==0)
2018-10-01 19:33:56 -04:00
{
info->odomCovariance = cv::Mat::eye(6, 6, CV_64FC1) * 9999.0f; // lost
lost_ = true;
UWARN("ZED lost! (trackingConfidence=%d)", trackingConfidence);
}
2023-01-28 16:33:24 -08:00
else
{
info->odomCovariance = cv::Mat::eye(6, 6, CV_64FC1) * 1.0f / float(trackingConfidence);
UDEBUG("Run %s (var=%f)", info->odomPose.prettyPrint().c_str(), 1.0f / float(trackingConfidence));
}
2018-10-01 19:33:56 -04:00
}
else
{
info->odomCovariance = cv::Mat::eye(6, 6, CV_64FC1) * 9999.0f; // lost
lost_ = true;
UWARN("ZED lost! (trackingConfidence=%d)", trackingConfidence);
}
}
else
{
UWARN("Tracking not ok: state=\"%s\"", toString(tracking_state).c_str());
}
}
}
else if(src_ == CameraVideo::kUsbDevice)
{
#if ZED_SDK_MAJOR_VERSION >= 4
UERROR("CameraStereoZed: Failed to grab images after 2 seconds! (%s: %s)", toString(res).c_str(), toVerbose(res).c_str());
#else
2018-10-01 19:33:56 -04:00
UERROR("CameraStereoZed: Failed to grab images after 2 seconds!");
#endif
2018-10-01 19:33:56 -04:00
}
else
{
UWARN("CameraStereoZed: end of stream is reached!");
}
}
#else
UERROR("CameraStereoZED: RTAB-Map is not built with ZED sdk support!");
#endif
return data;
}
void CameraStereoZed::postInterIMUPublic(const IMU & imu, double stamp)
{
postInterIMU(imu, stamp);
}
void CameraStereoZed::setRightGrayScale(bool enabled)
{
#ifdef RTABMAP_ZED
rightGrayScale_ = enabled;
#endif
}
2018-10-01 19:33:56 -04:00
} // namespace rtabmap