2023-09-08 08:55:16 +08:00
|
|
|
/*******************************************************************************
|
2023-09-08 14:22:11 +08:00
|
|
|
* Copyright (c) 2023 Orbbec 3D Technology, Inc
|
|
|
|
|
*
|
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
|
*
|
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
*
|
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
|
* limitations under the License.
|
|
|
|
|
*******************************************************************************/
|
2022-06-08 14:51:00 +08:00
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
#include "orbbec_camera/ob_camera_node.h"
|
|
|
|
|
#include <rclcpp/rclcpp.hpp>
|
|
|
|
|
#include <thread>
|
|
|
|
|
#include <geometry_msgs/msg/transform_stamped.hpp>
|
|
|
|
|
|
|
|
|
|
#include "orbbec_camera/utils.h"
|
2023-02-20 14:46:02 +08:00
|
|
|
#include <filesystem>
|
2024-01-29 11:55:59 +08:00
|
|
|
#include <fstream>
|
2024-04-09 16:08:48 +08:00
|
|
|
#include "diagnostic_msgs/msg/diagnostic_status.hpp"
|
2024-06-21 19:56:57 +08:00
|
|
|
#include "libobsensor/hpp/Utils.hpp"
|
2023-08-25 21:47:13 +08:00
|
|
|
|
|
|
|
|
#if defined(USE_RK_HW_DECODER)
|
|
|
|
|
#include "orbbec_camera/rk_mpp_decoder.h"
|
2023-09-07 16:46:59 +08:00
|
|
|
#elif defined(USE_NV_HW_DECODER)
|
|
|
|
|
#include "orbbec_camera/jetson_nv_decoder.h"
|
2023-08-25 21:47:13 +08:00
|
|
|
#endif
|
|
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
namespace orbbec_camera {
|
|
|
|
|
using namespace std::chrono_literals;
|
2022-06-07 10:46:04 +08:00
|
|
|
|
2026-04-13 10:41:51 +08:00
|
|
|
namespace {
|
|
|
|
|
|
|
|
|
|
int64_t getSystemNowUs() {
|
|
|
|
|
return std::chrono::duration_cast<std::chrono::microseconds>(
|
|
|
|
|
std::chrono::system_clock::now().time_since_epoch())
|
|
|
|
|
.count();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int64_t getSteadyNowUs() {
|
|
|
|
|
return std::chrono::duration_cast<std::chrono::microseconds>(
|
|
|
|
|
std::chrono::steady_clock::now().time_since_epoch())
|
|
|
|
|
.count();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
} // namespace
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
OBCameraNode::OBCameraNode(rclcpp::Node *node, std::shared_ptr<ob::Device> device,
|
2024-08-27 17:58:12 +08:00
|
|
|
std::shared_ptr<Parameters> parameters, bool use_intra_process)
|
2023-02-06 17:26:10 +08:00
|
|
|
: node_(node),
|
|
|
|
|
device_(std::move(device)),
|
|
|
|
|
parameters_(std::move(parameters)),
|
2024-08-27 17:58:12 +08:00
|
|
|
logger_(node->get_logger()),
|
|
|
|
|
use_intra_process_(use_intra_process) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"OBCameraNode: use_intra_process: " << (use_intra_process ? "ON" : "OFF"));
|
2022-06-06 19:00:12 +08:00
|
|
|
is_running_.store(true);
|
2023-02-06 17:26:10 +08:00
|
|
|
stream_name_[COLOR] = "color";
|
|
|
|
|
stream_name_[DEPTH] = "depth";
|
|
|
|
|
stream_name_[INFRA0] = "ir";
|
2023-09-05 17:59:19 +08:00
|
|
|
stream_name_[INFRA1] = "left_ir";
|
|
|
|
|
stream_name_[INFRA2] = "right_ir";
|
2023-04-28 09:44:16 +08:00
|
|
|
stream_name_[ACCEL] = "accel";
|
|
|
|
|
stream_name_[GYRO] = "gyro";
|
2022-06-06 19:00:12 +08:00
|
|
|
|
|
|
|
|
compression_params_.push_back(cv::IMWRITE_PNG_COMPRESSION);
|
|
|
|
|
compression_params_.push_back(0);
|
|
|
|
|
compression_params_.push_back(cv::IMWRITE_PNG_STRATEGY);
|
|
|
|
|
compression_params_.push_back(cv::IMWRITE_PNG_STRATEGY_DEFAULT);
|
2023-02-06 17:26:10 +08:00
|
|
|
setupDefaultImageFormat();
|
2022-06-06 19:00:12 +08:00
|
|
|
setupTopics();
|
2026-04-13 10:41:51 +08:00
|
|
|
if (enable_frame_timestamp_csv_) {
|
|
|
|
|
frame_timestamp_csv_logger_ =
|
|
|
|
|
std::make_unique<FrameTimestampCsvLogger>(true, frame_timestamp_csv_file_, logger_);
|
|
|
|
|
if (!frame_timestamp_csv_logger_->enabled()) {
|
|
|
|
|
frame_timestamp_csv_logger_.reset();
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-08-28 20:50:19 +08:00
|
|
|
#if defined(USE_RK_HW_DECODER)
|
2023-09-07 16:46:59 +08:00
|
|
|
jpeg_decoder_ = std::make_unique<RKJPEGDecoder>(width_[COLOR], height_[COLOR]);
|
|
|
|
|
#elif defined(USE_NV_HW_DECODER)
|
|
|
|
|
jpeg_decoder_ = std::make_unique<JetsonNvJPEGDecoder>(width_[COLOR], height_[COLOR]);
|
2023-08-28 20:50:19 +08:00
|
|
|
#endif
|
2023-02-20 12:11:12 +08:00
|
|
|
if (enable_d2c_viewer_) {
|
|
|
|
|
auto rgb_qos = getRMWQosProfileFromString(image_qos_[COLOR]);
|
|
|
|
|
auto depth_qos = getRMWQosProfileFromString(image_qos_[DEPTH]);
|
|
|
|
|
d2c_viewer_ = std::make_unique<D2CViewer>(node_, rgb_qos, depth_qos);
|
|
|
|
|
}
|
2023-09-07 16:09:35 +08:00
|
|
|
if (enable_stream_[COLOR]) {
|
|
|
|
|
rgb_buffer_ = new uint8_t[width_[COLOR] * height_[COLOR] * 3];
|
|
|
|
|
}
|
2024-06-21 19:56:57 +08:00
|
|
|
if (enable_colored_point_cloud_ && enable_stream_[DEPTH] && enable_stream_[COLOR]) {
|
2024-06-22 15:07:54 +08:00
|
|
|
rgb_point_cloud_buffer_size_ = width_[COLOR] * height_[COLOR] * sizeof(OBColorPoint);
|
|
|
|
|
rgb_point_cloud_buffer_ = new uint8_t[rgb_point_cloud_buffer_size_];
|
2024-06-21 19:56:57 +08:00
|
|
|
xy_table_data_size_ = width_[DEPTH] * height_[DEPTH] * 2;
|
|
|
|
|
xy_table_data_ = new float[xy_table_data_size_];
|
|
|
|
|
}
|
2024-05-09 17:07:43 +08:00
|
|
|
is_camera_node_initialized_ = true;
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
2022-06-08 14:51:00 +08:00
|
|
|
template <class T>
|
2022-06-09 13:52:42 +08:00
|
|
|
void OBCameraNode::setAndGetNodeParameter(
|
2023-08-27 14:06:25 +08:00
|
|
|
T ¶m, const std::string ¶m_name, const T &default_value,
|
|
|
|
|
const rcl_interfaces::msg::ParameterDescriptor ¶meter_descriptor) {
|
2022-06-08 14:51:00 +08:00
|
|
|
try {
|
|
|
|
|
param = parameters_
|
|
|
|
|
->setParam(param_name, rclcpp::ParameterValue(default_value),
|
2023-08-27 14:06:25 +08:00
|
|
|
std::function<void(const rclcpp::Parameter &)>(), parameter_descriptor)
|
2022-06-08 14:51:00 +08:00
|
|
|
.get<T>();
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const rclcpp::ParameterTypeException &ex) {
|
2022-06-08 14:51:00 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to set parameter: " << param_name << ". " << ex.what());
|
|
|
|
|
throw;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-06-21 19:56:57 +08:00
|
|
|
OBCameraNode::~OBCameraNode() noexcept { clean(); }
|
2022-06-08 13:41:31 +08:00
|
|
|
|
2024-07-29 09:50:02 +08:00
|
|
|
void OBCameraNode::rebootDevice() {
|
2024-08-23 11:31:22 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Reboot device");
|
2024-07-29 09:50:02 +08:00
|
|
|
clean();
|
|
|
|
|
if (device_) {
|
|
|
|
|
device_->reboot();
|
|
|
|
|
}
|
2024-08-23 11:31:22 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Reboot device DONE");
|
2024-07-29 09:50:02 +08:00
|
|
|
}
|
|
|
|
|
|
2024-06-21 19:56:57 +08:00
|
|
|
void OBCameraNode::clean() noexcept {
|
2023-10-14 14:03:50 +08:00
|
|
|
std::lock_guard<decltype(device_lock_)> lock(device_lock_);
|
2022-06-08 13:41:31 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Do destroy ~OBCameraNode");
|
|
|
|
|
is_running_.store(false);
|
2026-04-13 10:41:51 +08:00
|
|
|
if (frame_timestamp_csv_logger_) {
|
|
|
|
|
frame_timestamp_csv_logger_->shutdown();
|
|
|
|
|
frame_timestamp_csv_logger_.reset();
|
|
|
|
|
}
|
2024-08-23 11:31:22 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Stop tf thread");
|
2023-10-12 19:16:48 +08:00
|
|
|
if (tf_thread_ && tf_thread_->joinable()) {
|
2022-06-06 19:00:12 +08:00
|
|
|
tf_thread_->join();
|
|
|
|
|
}
|
2024-08-23 11:31:22 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Stop color frame thread");
|
2023-11-02 20:01:48 +08:00
|
|
|
if (colorFrameThread_ && colorFrameThread_->joinable()) {
|
2024-05-09 17:07:43 +08:00
|
|
|
color_frame_queue_cv_.notify_all();
|
2023-11-02 20:01:48 +08:00
|
|
|
colorFrameThread_->join();
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-13 15:34:44 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "stop streams");
|
|
|
|
|
stopStreams();
|
2023-10-13 21:40:32 +08:00
|
|
|
stopIMU();
|
2023-08-27 14:06:25 +08:00
|
|
|
if (rgb_buffer_) {
|
|
|
|
|
delete[] rgb_buffer_;
|
|
|
|
|
rgb_buffer_ = nullptr;
|
|
|
|
|
}
|
2024-06-22 15:07:54 +08:00
|
|
|
if (rgb_point_cloud_buffer_) {
|
|
|
|
|
delete[] rgb_point_cloud_buffer_;
|
|
|
|
|
rgb_point_cloud_buffer_ = nullptr;
|
2024-06-21 19:56:57 +08:00
|
|
|
}
|
|
|
|
|
if (xy_table_data_) {
|
|
|
|
|
delete[] xy_table_data_;
|
|
|
|
|
xy_table_data_ = nullptr;
|
|
|
|
|
}
|
2024-08-23 11:31:22 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Destroy ~OBCameraNode DONE");
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::setupDevices() {
|
2025-09-04 10:46:04 +08:00
|
|
|
if (!depth_filter_config_.empty() && enable_depth_filter_) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Load depth filter config: " << depth_filter_config_);
|
|
|
|
|
TRY_EXECUTE_BLOCK(device_->loadDepthFilterConfig(depth_filter_config_.c_str()));
|
|
|
|
|
} else {
|
|
|
|
|
if (device_->isPropertySupported(OB_PROP_DEPTH_SOFT_FILTER_BOOL, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_DEPTH_SOFT_FILTER_BOOL, enable_soft_filter_);
|
2025-09-28 09:39:13 +08:00
|
|
|
} else {
|
2025-09-04 10:46:04 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Depth filter config is empty and soft filter is not supported");
|
|
|
|
|
}
|
|
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
auto sensor_list = device_->getSensorList();
|
|
|
|
|
for (size_t i = 0; i < sensor_list->count(); i++) {
|
|
|
|
|
auto sensor = sensor_list->getSensor(i);
|
|
|
|
|
auto profiles = sensor->getStreamProfileList();
|
|
|
|
|
for (size_t j = 0; j < profiles->count(); j++) {
|
|
|
|
|
auto profile = profiles->getProfile(j);
|
|
|
|
|
stream_index_pair sip{profile->type(), 0};
|
|
|
|
|
if (sensors_.find(sip) != sensors_.end()) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
sensors_[sip] = sensor;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
for (const auto &[stream_index, enable] : enable_stream_) {
|
2022-06-06 19:00:12 +08:00
|
|
|
if (enable && sensors_.find(stream_index) == sensors_.end()) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
magic_enum::enum_name(stream_index.first)
|
|
|
|
|
<< "sensor isn't supported by current device! -- Skipping...");
|
2023-02-06 17:26:10 +08:00
|
|
|
enable_stream_[stream_index] = false;
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
}
|
2023-03-01 22:01:08 +08:00
|
|
|
auto info = device_->getDeviceInfo();
|
2024-08-13 16:13:50 +08:00
|
|
|
if (retry_on_usb3_detection_failure_ &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_DEVICE_USB3_REPEAT_IDENTIFY_BOOL,
|
2024-07-31 10:59:10 +08:00
|
|
|
OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_DEVICE_USB3_REPEAT_IDENTIFY_BOOL,
|
|
|
|
|
retry_on_usb3_detection_failure_);
|
|
|
|
|
}
|
2025-09-04 10:46:04 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
auto pid = device_info->pid();
|
2025-09-28 09:39:13 +08:00
|
|
|
if (isGemini335PID(pid) && device_->isPropertySupported(OB_PROP_DEPTH_NOISE_REMOVAL_FILTER_BOOL,
|
|
|
|
|
OB_PERMISSION_READ_WRITE)) {
|
2024-08-22 14:30:01 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_DEPTH_NOISE_REMOVAL_FILTER_BOOL,
|
|
|
|
|
enable_noise_removal_filter_);
|
|
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_HEARTBEAT_BOOL, OB_PERMISSION_READ_WRITE)) {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting heartbeat to " << (enable_heartbeat_ ? "ON" : "OFF"));
|
2024-07-31 10:59:10 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_HEARTBEAT_BOOL, enable_heartbeat_);
|
|
|
|
|
}
|
2025-04-22 17:51:40 +08:00
|
|
|
if (!industry_mode_.empty() &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_DEPTH_INDUSTRY_MODE_INT, OB_PERMISSION_READ_WRITE)) {
|
2025-04-03 13:45:32 +08:00
|
|
|
if (industry_mode_ == "default") {
|
|
|
|
|
OBDepthIndustryMode mode = OB_INDUSTRY_DEFAULT;
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_INDUSTRY_MODE_INT, (int32_t)mode);
|
|
|
|
|
} else if (industry_mode_ == "mode1") {
|
|
|
|
|
OBDepthIndustryMode mode = OB_INDUSTRY_MODE1;
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_INDUSTRY_MODE_INT, (int32_t)mode);
|
|
|
|
|
} else if (industry_mode_ == "mode2") {
|
|
|
|
|
OBDepthIndustryMode mode = OB_INDUSTRY_MODE2;
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_INDUSTRY_MODE_INT, (int32_t)mode);
|
|
|
|
|
} else if (industry_mode_ == "mode3") {
|
|
|
|
|
OBDepthIndustryMode mode = OB_INDUSTRY_MODE3;
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_INDUSTRY_MODE_INT, (int32_t)mode);
|
|
|
|
|
} else if (industry_mode_ == "mode4") {
|
|
|
|
|
OBDepthIndustryMode mode = OB_INDUSTRY_MODE4;
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_INDUSTRY_MODE_INT, (int32_t)mode);
|
|
|
|
|
} else if (industry_mode_ == "mode5") {
|
|
|
|
|
OBDepthIndustryMode mode = OB_INDUSTRY_MODE5;
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_INDUSTRY_MODE_INT, (int32_t)mode);
|
|
|
|
|
}
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting industry mode to "
|
|
|
|
|
<< device_->getIntProperty(OB_PROP_DEPTH_INDUSTRY_MODE_INT));
|
|
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
if (max_depth_limit_ > 0 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_MAX_DEPTH_INT, OB_PERMISSION_READ_WRITE)) {
|
2026-01-29 16:39:42 +08:00
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_MAX_DEPTH_INT);
|
|
|
|
|
if (max_depth_limit_ < range.min || max_depth_limit_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Max depth limit is out of range " << range.min << " - " << range.max);
|
|
|
|
|
} else {
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_MAX_DEPTH_INT, max_depth_limit_);
|
|
|
|
|
auto new_max_depth_limit = device_->getIntProperty(OB_PROP_MAX_DEPTH_INT);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting max depth limit to " << new_max_depth_limit);
|
|
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
if (min_depth_limit_ > 0 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_MIN_DEPTH_INT, OB_PERMISSION_READ_WRITE)) {
|
2026-01-29 16:39:42 +08:00
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_MIN_DEPTH_INT);
|
|
|
|
|
if (min_depth_limit_ < range.min || min_depth_limit_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Min depth limit is out of range " << range.min << " - " << range.max);
|
|
|
|
|
} else {
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_MIN_DEPTH_INT, min_depth_limit_);
|
|
|
|
|
auto new_min_depth_limit = device_->getIntProperty(OB_PROP_MIN_DEPTH_INT);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting min depth limit to " << new_min_depth_limit);
|
|
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
if (laser_energy_level_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_LASER_ENERGY_LEVEL_INT, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting laser energy level to " << laser_energy_level_);
|
|
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_LASER_ENERGY_LEVEL_INT);
|
|
|
|
|
if (laser_energy_level_ < range.min || laser_energy_level_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Laser energy level is out of range " << range.min << " - " << range.max);
|
|
|
|
|
} else {
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_LASER_ENERGY_LEVEL_INT, laser_energy_level_);
|
|
|
|
|
auto new_laser_energy_level = device_->getIntProperty(OB_PROP_LASER_ENERGY_LEVEL_INT);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Laser energy level set to " << new_laser_energy_level << " (new value)");
|
2024-05-09 11:34:03 +08:00
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
if (depth_registration_) {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Create align filter");
|
2024-07-31 10:59:10 +08:00
|
|
|
align_filter_ = std::make_unique<ob::Align>(align_target_stream_);
|
|
|
|
|
}
|
|
|
|
|
if (device_->isPropertySupported(OB_PROP_DISPARITY_TO_DEPTH_BOOL, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_DISPARITY_TO_DEPTH_BOOL, enable_hardware_d2d_);
|
|
|
|
|
bool is_hardware_d2d = device_->getBoolProperty(OB_PROP_DISPARITY_TO_DEPTH_BOOL);
|
|
|
|
|
std::string d2d_mode = is_hardware_d2d ? "HW D2D" : "SW D2D";
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Depth process is " << d2d_mode);
|
|
|
|
|
}
|
2024-11-28 15:50:55 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_LDP_BOOL, OB_PERMISSION_WRITE)) {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting LDP to " << (enable_ldp_ ? "ON" : "OFF"));
|
2024-07-31 10:59:10 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_LDP_BOOL, enable_ldp_);
|
|
|
|
|
}
|
|
|
|
|
if (device_->isPropertySupported(OB_PROP_LASER_CONTROL_INT, OB_PERMISSION_READ_WRITE)) {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting laser control to " << enable_laser_);
|
2024-07-31 10:59:10 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_LASER_CONTROL_INT, enable_laser_);
|
|
|
|
|
}
|
|
|
|
|
if (device_->isPropertySupported(OB_PROP_LASER_ON_OFF_MODE_INT, OB_PERMISSION_READ_WRITE)) {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting laser on off mode to " << laser_on_off_mode_);
|
2024-07-31 10:59:10 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_LASER_ON_OFF_MODE_INT, laser_on_off_mode_);
|
|
|
|
|
}
|
|
|
|
|
if (!device_preset_.empty()) {
|
|
|
|
|
try {
|
2024-05-09 11:34:03 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Available presets:");
|
|
|
|
|
auto preset_list = device_->getAvailablePresetList();
|
|
|
|
|
for (uint32_t i = 0; i < preset_list->count(); i++) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Preset " << i << ": " << preset_list->getName(i));
|
|
|
|
|
}
|
2024-05-08 22:28:27 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Load device preset: " << device_preset_);
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_EXECUTE_BLOCK(device_->loadPreset(device_preset_.c_str()));
|
2024-05-15 14:01:07 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Device preset " << device_->getCurrentPresetName() << " loaded");
|
2024-07-31 10:59:10 +08:00
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to load device preset: " << e.getMessage());
|
|
|
|
|
} catch (const std::exception &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to load device preset: " << e.what());
|
|
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to load device preset");
|
2024-05-06 15:39:18 +08:00
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
if (!depth_work_mode_.empty()) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Set depth work mode: " << depth_work_mode_);
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_EXECUTE_BLOCK(device_->switchDepthWorkMode(depth_work_mode_.c_str()));
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
2024-11-29 10:32:54 +08:00
|
|
|
if (!sync_mode_str_.empty()) {
|
2024-07-31 10:59:10 +08:00
|
|
|
auto sync_config = device_->getMultiDeviceSyncConfig();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Current sync mode: " << magic_enum::enum_name(sync_config.syncMode));
|
|
|
|
|
std::transform(sync_mode_str_.begin(), sync_mode_str_.end(), sync_mode_str_.begin(), ::toupper);
|
|
|
|
|
sync_mode_ = OBSyncModeFromString(sync_mode_str_);
|
|
|
|
|
sync_config.syncMode = sync_mode_;
|
|
|
|
|
sync_config.depthDelayUs = depth_delay_us_;
|
|
|
|
|
sync_config.colorDelayUs = color_delay_us_;
|
|
|
|
|
sync_config.trigger2ImageDelayUs = trigger2image_delay_us_;
|
|
|
|
|
sync_config.triggerOutDelayUs = trigger_out_delay_us_;
|
|
|
|
|
sync_config.triggerOutEnable = trigger_out_enabled_;
|
|
|
|
|
sync_config.framesPerTrigger = frames_per_trigger_;
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_EXECUTE_BLOCK(device_->setMultiDeviceSyncConfig(sync_config));
|
2024-07-31 10:59:10 +08:00
|
|
|
sync_config = device_->getMultiDeviceSyncConfig();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Set sync mode: " << magic_enum::enum_name(sync_config.syncMode));
|
|
|
|
|
if (sync_mode_ == OB_MULTI_DEVICE_SYNC_MODE_SOFTWARE_TRIGGERING) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Frames per trigger: " << sync_config.framesPerTrigger);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Software trigger period " << software_trigger_period_.count() << " ms");
|
2024-08-30 10:46:26 +08:00
|
|
|
software_trigger_timer_ = node_->create_wall_timer(
|
|
|
|
|
software_trigger_period_, [this]() { TRY_EXECUTE_BLOCK(device_->triggerCapture()); });
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
}
|
2023-12-20 17:56:37 +08:00
|
|
|
|
2024-09-26 09:45:15 +08:00
|
|
|
if (color_ae_roi_left_ != -1 && color_ae_roi_top_ != -1 && color_ae_roi_right_ != -1 &&
|
|
|
|
|
color_ae_roi_bottom_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_STRUCT_COLOR_AE_ROI, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color AE ROI to "
|
|
|
|
|
<< color_ae_roi_left_ << ", " << color_ae_roi_top_ << ", "
|
|
|
|
|
<< color_ae_roi_right_ << ", " << color_ae_roi_bottom_);
|
|
|
|
|
AE_ROI roi;
|
|
|
|
|
roi.x0_left = color_ae_roi_left_;
|
|
|
|
|
roi.y0_top = color_ae_roi_top_;
|
|
|
|
|
roi.x1_right = color_ae_roi_right_;
|
|
|
|
|
roi.y1_bottom = color_ae_roi_bottom_;
|
|
|
|
|
device_->setStructuredData(OB_STRUCT_COLOR_AE_ROI, &roi, sizeof(AE_ROI));
|
|
|
|
|
}
|
|
|
|
|
// depth ae roi
|
|
|
|
|
if (depth_ae_roi_left_ != -1 && depth_ae_roi_top_ != -1 && depth_ae_roi_right_ != -1 &&
|
|
|
|
|
depth_ae_roi_bottom_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_STRUCT_DEPTH_AE_ROI, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting depth AE ROI to "
|
|
|
|
|
<< depth_ae_roi_left_ << ", " << depth_ae_roi_top_ << ", "
|
|
|
|
|
<< depth_ae_roi_right_ << ", " << depth_ae_roi_bottom_);
|
|
|
|
|
AE_ROI roi;
|
|
|
|
|
roi.x0_left = depth_ae_roi_left_;
|
|
|
|
|
roi.y0_top = depth_ae_roi_top_;
|
|
|
|
|
roi.x1_right = depth_ae_roi_right_;
|
|
|
|
|
roi.y1_bottom = depth_ae_roi_bottom_;
|
|
|
|
|
device_->setStructuredData(OB_STRUCT_DEPTH_AE_ROI, &roi, sizeof(AE_ROI));
|
|
|
|
|
}
|
2025-04-22 16:38:10 +08:00
|
|
|
if (color_rotation_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_ROTATE_INT, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
device_->setIntProperty(OB_PROP_COLOR_ROTATE_INT, color_rotation_);
|
|
|
|
|
RCLCPP_INFO_STREAM(
|
|
|
|
|
logger_, "set color rotation to " << device_->getIntProperty(OB_PROP_COLOR_ROTATE_INT));
|
|
|
|
|
}
|
|
|
|
|
if (depth_rotation_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_DEPTH_ROTATE_INT, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_ROTATE_INT, depth_rotation_);
|
|
|
|
|
RCLCPP_INFO_STREAM(
|
|
|
|
|
logger_, "set depth rotation to " << device_->getIntProperty(OB_PROP_DEPTH_ROTATE_INT));
|
|
|
|
|
}
|
|
|
|
|
if (left_ir_rotation_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_IR_ROTATE_INT, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
device_->setIntProperty(OB_PROP_IR_ROTATE_INT, left_ir_rotation_);
|
|
|
|
|
RCLCPP_INFO_STREAM(
|
|
|
|
|
logger_, "set left ir rotation to " << device_->getIntProperty(OB_PROP_IR_ROTATE_INT));
|
|
|
|
|
}
|
|
|
|
|
if (right_ir_rotation_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_IR_RIGHT_ROTATE_INT, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
device_->setIntProperty(OB_PROP_IR_RIGHT_ROTATE_INT, right_ir_rotation_);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "set right ir rotation to "
|
|
|
|
|
<< device_->getIntProperty(OB_PROP_IR_RIGHT_ROTATE_INT));
|
|
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_DEPTH_PRECISION_LEVEL_INT, OB_PERMISSION_READ_WRITE) &&
|
|
|
|
|
!depth_precision_str_.empty()) {
|
|
|
|
|
auto default_precision_level = device_->getIntProperty(OB_PROP_DEPTH_PRECISION_LEVEL_INT);
|
|
|
|
|
if (default_precision_level != depth_precision_) {
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_PRECISION_LEVEL_INT, depth_precision_);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "set depth precision to " << depth_precision_str_);
|
2023-12-20 17:56:37 +08:00
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
} else if (device_->isPropertySupported(OB_PROP_DEPTH_UNIT_FLEXIBLE_ADJUSTMENT_FLOAT,
|
|
|
|
|
OB_PERMISSION_READ_WRITE) &&
|
|
|
|
|
!depth_precision_str_.empty()) {
|
|
|
|
|
auto depth_unit_flexible_adjustment = depthPrecisionFromString(depth_precision_str_);
|
|
|
|
|
auto range = device_->getFloatPropertyRange(OB_PROP_DEPTH_UNIT_FLEXIBLE_ADJUSTMENT_FLOAT);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Depth unit flexible adjustment range: " << range.min << " - " << range.max);
|
|
|
|
|
if (depth_unit_flexible_adjustment < range.min || depth_unit_flexible_adjustment > range.max) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(
|
|
|
|
|
logger_, "depth unit flexible adjustment value is out of range, please check the value");
|
|
|
|
|
} else {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "set depth unit to " << depth_unit_flexible_adjustment << "mm");
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setFloatProperty, OB_PROP_DEPTH_UNIT_FLEXIBLE_ADJUSTMENT_FLOAT,
|
|
|
|
|
depth_unit_flexible_adjustment);
|
2024-06-17 14:19:02 +08:00
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
2023-12-20 17:56:37 +08:00
|
|
|
|
2024-07-31 10:59:10 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_COLOR_AUTO_WHITE_BALANCE_BOOL, OB_PERMISSION_WRITE)) {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color auto white balance to "
|
|
|
|
|
<< (enable_color_auto_white_balance_ ? "ON" : "OFF"));
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_COLOR_AUTO_WHITE_BALANCE_BOOL,
|
|
|
|
|
enable_color_auto_white_balance_);
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
if (color_exposure_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_EXPOSURE_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_COLOR_EXPOSURE_INT);
|
|
|
|
|
if (color_exposure_ < range.min || color_exposure_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "color exposure value is out of range[%d,%d], please check the value",
|
|
|
|
|
range.min, range.max);
|
|
|
|
|
} else {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color exposure to " << color_exposure_);
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_EXPOSURE_INT, color_exposure_);
|
2023-12-20 17:56:37 +08:00
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
if (color_gain_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_GAIN_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_COLOR_GAIN_INT);
|
|
|
|
|
if (color_gain_ < range.min || color_gain_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "color gain value is out of range[%d,%d], please check the value",
|
|
|
|
|
range.min, range.max);
|
|
|
|
|
} else {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color gain to " << color_gain_);
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_GAIN_INT, color_gain_);
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (color_white_balance_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_WHITE_BALANCE_INT, OB_PERMISSION_WRITE)) {
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_COLOR_AUTO_WHITE_BALANCE_BOOL, false);
|
2024-07-31 10:59:10 +08:00
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_COLOR_WHITE_BALANCE_INT);
|
|
|
|
|
if (color_white_balance_ < range.min || color_white_balance_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR(logger_,
|
|
|
|
|
"color white balance value is out of range[%d,%d], please check the value",
|
|
|
|
|
range.min, range.max);
|
|
|
|
|
} else {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color white balance to " << color_white_balance_);
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_WHITE_BALANCE_INT, color_white_balance_);
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
}
|
2025-10-21 14:41:10 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_COLOR_AUTO_EXPOSURE_PRIORITY_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color auto exposure priority to "
|
|
|
|
|
<< (enable_color_auto_exposure_priority_ ? "true" : "false"));
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_COLOR_AUTO_EXPOSURE_PRIORITY_INT,
|
|
|
|
|
enable_color_auto_exposure_priority_);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"color auto exposure priority is set to "
|
|
|
|
|
<< device_->getBoolProperty(OB_PROP_COLOR_AUTO_EXPOSURE_PRIORITY_INT));
|
|
|
|
|
}
|
2024-12-06 09:33:47 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_COLOR_AUTO_EXPOSURE_BOOL, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(
|
|
|
|
|
logger_, "Setting color auto exposure to " << (enable_color_auto_exposure_ ? "ON" : "OFF"));
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_COLOR_AUTO_EXPOSURE_BOOL,
|
|
|
|
|
enable_color_auto_exposure_);
|
|
|
|
|
}
|
2024-09-23 15:01:34 +08:00
|
|
|
if (color_ae_max_exposure_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_AE_MAX_EXPOSURE_INT, OB_PERMISSION_WRITE)) {
|
2024-09-18 14:13:01 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color AE max exposure to " << color_ae_max_exposure_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_AE_MAX_EXPOSURE_INT, color_ae_max_exposure_);
|
|
|
|
|
}
|
2024-09-23 15:01:34 +08:00
|
|
|
if (color_brightness_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_BRIGHTNESS_INT, OB_PERMISSION_WRITE)) {
|
2024-09-18 14:13:01 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color brightness to " << color_brightness_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_BRIGHTNESS_INT, color_brightness_);
|
|
|
|
|
}
|
2024-11-05 16:18:02 +08:00
|
|
|
if (color_sharpness_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_SHARPNESS_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color sharpness to " << color_sharpness_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_SHARPNESS_INT, color_sharpness_);
|
|
|
|
|
}
|
|
|
|
|
if (color_saturation_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_SATURATION_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color saturation to " << color_saturation_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_SATURATION_INT, color_saturation_);
|
|
|
|
|
}
|
|
|
|
|
if (color_contrast_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_CONTRAST_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color contrast to " << color_contrast_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_CONTRAST_INT, color_contrast_);
|
|
|
|
|
}
|
|
|
|
|
if (color_gamma_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_GAMMA_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color gamma to " << color_gamma_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_GAMMA_INT, color_gamma_);
|
|
|
|
|
}
|
|
|
|
|
if (color_hue_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_COLOR_HUE_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting color hue to " << color_hue_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_COLOR_HUE_INT, color_hue_);
|
|
|
|
|
}
|
|
|
|
|
|
2024-09-23 15:01:34 +08:00
|
|
|
// ir ae max
|
|
|
|
|
if (ir_ae_max_exposure_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_IR_AE_MAX_EXPOSURE_INT, OB_PERMISSION_WRITE)) {
|
2024-09-18 14:13:01 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting IR AE max exposure to " << ir_ae_max_exposure_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_IR_AE_MAX_EXPOSURE_INT, ir_ae_max_exposure_);
|
|
|
|
|
}
|
|
|
|
|
// ir brightness
|
2024-09-23 15:01:34 +08:00
|
|
|
if (ir_brightness_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_IR_BRIGHTNESS_INT, OB_PERMISSION_WRITE)) {
|
2024-09-18 14:13:01 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting IR brightness to " << ir_brightness_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_IR_BRIGHTNESS_INT, ir_brightness_);
|
|
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
if (ir_exposure_ != -1 &&
|
|
|
|
|
device_->isPropertySupported(OB_PROP_IR_EXPOSURE_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_IR_EXPOSURE_INT);
|
|
|
|
|
if (ir_exposure_ < range.min || ir_exposure_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "ir exposure value is out of range[%d,%d], please check the value",
|
|
|
|
|
range.min, range.max);
|
|
|
|
|
} else {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting IR exposure to " << ir_exposure_);
|
2024-08-13 16:13:50 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_IR_EXPOSURE_INT, ir_exposure_);
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (ir_gain_ != -1 && device_->isPropertySupported(OB_PROP_IR_GAIN_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
auto range = device_->getIntPropertyRange(OB_PROP_IR_GAIN_INT);
|
|
|
|
|
if (ir_gain_ < range.min || ir_gain_ > range.max) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "ir gain value is out of range[%d,%d], please check the value",
|
|
|
|
|
range.min, range.max);
|
|
|
|
|
} else {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting IR gain to " << ir_gain_);
|
2024-08-01 17:35:06 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_IR_GAIN_INT, ir_gain_);
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
}
|
2024-12-06 09:33:47 +08:00
|
|
|
// ir auto exposure
|
|
|
|
|
if (device_->isPropertySupported(OB_PROP_IR_AUTO_EXPOSURE_BOOL, OB_PERMISSION_WRITE)) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Setting IR auto exposure to " << (enable_ir_auto_exposure_ ? "ON" : "OFF"));
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_IR_AUTO_EXPOSURE_BOOL, enable_ir_auto_exposure_);
|
|
|
|
|
}
|
2024-07-31 10:59:10 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_IR_LONG_EXPOSURE_BOOL, OB_PERMISSION_WRITE)) {
|
2024-08-30 11:02:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Setting IR long exposure to " << (enable_ir_long_exposure_ ? "ON" : "OFF"));
|
2024-08-13 16:13:50 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_IR_LONG_EXPOSURE_BOOL, enable_ir_long_exposure_);
|
2024-07-31 10:59:10 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (device_->isPropertySupported(OB_PROP_DEPTH_MAX_DIFF_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
auto default_soft_filter_max_diff = device_->getIntProperty(OB_PROP_DEPTH_MAX_DIFF_INT);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "default_soft_filter_max_diff: " << default_soft_filter_max_diff);
|
|
|
|
|
if (soft_filter_max_diff_ != -1 && default_soft_filter_max_diff != soft_filter_max_diff_) {
|
|
|
|
|
device_->setIntProperty(OB_PROP_DEPTH_MAX_DIFF_INT, soft_filter_max_diff_);
|
|
|
|
|
auto new_soft_filter_max_diff = device_->getIntProperty(OB_PROP_DEPTH_MAX_DIFF_INT);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "after set soft_filter_max_diff: " << new_soft_filter_max_diff);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (device_->isPropertySupported(OB_PROP_DEPTH_MAX_SPECKLE_SIZE_INT, OB_PERMISSION_WRITE)) {
|
|
|
|
|
auto default_soft_filter_speckle_size =
|
|
|
|
|
device_->getIntProperty(OB_PROP_DEPTH_MAX_SPECKLE_SIZE_INT);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"default_soft_filter_speckle_size: " << default_soft_filter_speckle_size);
|
|
|
|
|
if (soft_filter_speckle_size_ != -1 &&
|
|
|
|
|
default_soft_filter_speckle_size != soft_filter_speckle_size_) {
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_DEPTH_MAX_SPECKLE_SIZE_INT,
|
|
|
|
|
soft_filter_speckle_size_);
|
2024-07-31 10:59:10 +08:00
|
|
|
auto new_soft_filter_speckle_size =
|
2023-12-20 17:56:37 +08:00
|
|
|
device_->getIntProperty(OB_PROP_DEPTH_MAX_SPECKLE_SIZE_INT);
|
2024-06-13 10:25:25 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
2024-07-31 10:59:10 +08:00
|
|
|
"after set soft_filter_speckle_size: " << new_soft_filter_speckle_size);
|
2023-03-24 14:57:33 +08:00
|
|
|
}
|
2023-03-20 11:18:59 +08:00
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
2024-07-31 11:09:09 +08:00
|
|
|
void OBCameraNode::setupDepthPostProcessFilter() {
|
|
|
|
|
auto depth_sensor = device_->getSensor(OB_SENSOR_DEPTH);
|
|
|
|
|
// set depth sensor to filter
|
|
|
|
|
auto filter_list = depth_sensor->getRecommendedFilters();
|
2024-07-31 11:13:08 +08:00
|
|
|
if (!filter_list) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Failed to get depth sensor filter list");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-07-31 11:09:09 +08:00
|
|
|
for (size_t i = 0; i < filter_list->count(); i++) {
|
|
|
|
|
auto filter = filter_list->getFilter(i);
|
|
|
|
|
std::map<std::string, bool> filter_params = {
|
|
|
|
|
{"DecimationFilter", enable_decimation_filter_},
|
|
|
|
|
{"HDRMerge", enable_hdr_merge_},
|
|
|
|
|
{"SequencedFilter", enable_sequence_id_filter_},
|
|
|
|
|
{"ThresholdFilter", enable_threshold_filter_},
|
|
|
|
|
{"NoiseRemovalFilter", enable_noise_removal_filter_},
|
|
|
|
|
{"SpatialAdvancedFilter", enable_spatial_filter_},
|
|
|
|
|
{"TemporalFilter", enable_temporal_filter_},
|
|
|
|
|
{"HoleFillingFilter", enable_hole_filling_filter_},
|
|
|
|
|
|
|
|
|
|
};
|
|
|
|
|
std::string filter_name = filter->type();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting " << filter_name << "......");
|
|
|
|
|
if (filter_params.find(filter_name) != filter_params.end()) {
|
|
|
|
|
std::string value = filter_params[filter_name] ? "true" : "false";
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "set " << filter_name << " to " << value);
|
|
|
|
|
filter->enable(filter_params[filter_name]);
|
|
|
|
|
filter_status_[filter_name] = filter_params[filter_name];
|
|
|
|
|
}
|
|
|
|
|
if (filter_name == "DecimationFilter" && enable_decimation_filter_) {
|
|
|
|
|
auto decimation_filter = filter->as<ob::DecimationFilter>();
|
|
|
|
|
auto range = decimation_filter->getScaleRange();
|
|
|
|
|
if (decimation_filter_scale_ != -1 && decimation_filter_scale_ < range.max &&
|
|
|
|
|
decimation_filter_scale_ > range.min) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Set decimation filter scale value to " << decimation_filter_scale_);
|
|
|
|
|
decimation_filter->setScaleValue(decimation_filter_scale_);
|
|
|
|
|
}
|
|
|
|
|
if (decimation_filter_scale_ != -1 &&
|
|
|
|
|
(decimation_filter_scale_ < range.min || decimation_filter_scale_ > range.max)) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Decimation filter scale value is out of range "
|
|
|
|
|
<< range.min << " - " << range.max);
|
|
|
|
|
}
|
|
|
|
|
} else if (filter_name == "ThresholdFilter" && enable_threshold_filter_) {
|
|
|
|
|
auto threshold_filter = filter->as<ob::ThresholdFilter>();
|
|
|
|
|
if (threshold_filter_min_ != -1 && threshold_filter_max_ != -1) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Set threshold filter value range to "
|
|
|
|
|
<< threshold_filter_min_ << " - " << threshold_filter_max_);
|
|
|
|
|
threshold_filter->setValueRange(threshold_filter_min_, threshold_filter_max_);
|
|
|
|
|
}
|
|
|
|
|
} else if (filter_name == "SpatialAdvancedFilter" && enable_spatial_filter_) {
|
|
|
|
|
auto spatial_filter = filter->as<ob::SpatialAdvancedFilter>();
|
|
|
|
|
if (spatial_filter_alpha_ != -1.0 && spatial_filter_magnitude_ != -1 &&
|
|
|
|
|
spatial_filter_radius_ != -1 && spatial_filter_diff_threshold_ != -1) {
|
|
|
|
|
OBSpatialAdvancedFilterParams params{};
|
|
|
|
|
params.alpha = spatial_filter_alpha_;
|
|
|
|
|
params.magnitude = spatial_filter_magnitude_;
|
|
|
|
|
params.radius = spatial_filter_radius_;
|
|
|
|
|
params.disp_diff = spatial_filter_diff_threshold_;
|
|
|
|
|
spatial_filter->setFilterParams(params);
|
|
|
|
|
}
|
|
|
|
|
} else if (filter_name == "TemporalFilter" && enable_temporal_filter_) {
|
|
|
|
|
auto temporal_filter = filter->as<ob::TemporalFilter>();
|
|
|
|
|
if (temporal_filter_diff_threshold_ != -1.0 && temporal_filter_weight_ != -1.0) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Set temporal filter value to "
|
|
|
|
|
<< temporal_filter_diff_threshold_ << " - "
|
|
|
|
|
<< temporal_filter_weight_);
|
|
|
|
|
temporal_filter->setDiffScale(temporal_filter_diff_threshold_);
|
|
|
|
|
temporal_filter->setWeight(temporal_filter_weight_);
|
|
|
|
|
}
|
|
|
|
|
} else if (filter_name == "HoleFillingFilter" && enable_hole_filling_filter_ &&
|
|
|
|
|
!hole_filling_filter_mode_.empty()) {
|
|
|
|
|
auto hole_filling_filter = filter->as<ob::HoleFillingFilter>();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Default hole filling filter mode: " << hole_filling_filter_mode_);
|
|
|
|
|
OBHoleFillingMode hole_filling_mode = holeFillingModeFromString(hole_filling_filter_mode_);
|
|
|
|
|
hole_filling_filter->setFilterMode(hole_filling_mode);
|
|
|
|
|
} else if (filter_name == "SequenceIdFilter" && enable_sequence_id_filter_) {
|
|
|
|
|
auto sequenced_filter = filter->as<ob::SequenceIdFilter>();
|
|
|
|
|
if (sequence_id_filter_id_ != -1) {
|
|
|
|
|
sequenced_filter->selectSequenceId(sequence_id_filter_id_);
|
|
|
|
|
}
|
|
|
|
|
} else if (filter_name == "NoiseRemovalFilter" && enable_noise_removal_filter_) {
|
|
|
|
|
auto noise_removal_filter = filter->as<ob::NoiseRemovalFilter>();
|
|
|
|
|
OBNoiseRemovalFilterParams params = noise_removal_filter->getFilterParams();
|
2025-09-28 09:39:13 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Default noise removal filter params: "
|
|
|
|
|
<< "disp_diff: " << params.disp_diff
|
|
|
|
|
<< ", max_size: " << params.max_size);
|
2024-07-31 11:09:09 +08:00
|
|
|
params.disp_diff = noise_removal_filter_min_diff_;
|
|
|
|
|
params.max_size = noise_removal_filter_max_size_;
|
2025-09-28 09:39:13 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Set noise removal filter params: "
|
|
|
|
|
<< "disp_diff: " << params.disp_diff
|
|
|
|
|
<< ", max_size: " << params.max_size);
|
2024-07-31 11:09:09 +08:00
|
|
|
if (noise_removal_filter_min_diff_ != -1 && noise_removal_filter_max_size_ != -1) {
|
|
|
|
|
noise_removal_filter->setFilterParams(params);
|
|
|
|
|
}
|
|
|
|
|
} else if (filter_name == "HDRMerge" && enable_hdr_merge_) {
|
|
|
|
|
if (hdr_merge_exposure_1_ != -1 && hdr_merge_gain_1_ != -1 && hdr_merge_exposure_2_ != -1 &&
|
|
|
|
|
hdr_merge_gain_2_ != -1) {
|
|
|
|
|
auto hdr_merge_filter = filter->as<ob::HdrMerge>();
|
|
|
|
|
hdr_merge_filter->enable(true);
|
2025-09-28 09:39:13 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Set HDR merge filter params: "
|
|
|
|
|
<< "exposure_1: " << hdr_merge_exposure_1_
|
|
|
|
|
<< ", gain_1: " << hdr_merge_gain_1_
|
|
|
|
|
<< ", exposure_2: " << hdr_merge_exposure_2_
|
|
|
|
|
<< ", gain_2: " << hdr_merge_gain_2_);
|
2024-07-31 11:09:09 +08:00
|
|
|
auto config = OBHdrConfig();
|
|
|
|
|
config.enable = true;
|
|
|
|
|
config.exposure_1 = hdr_merge_exposure_1_;
|
|
|
|
|
config.gain_1 = hdr_merge_gain_1_;
|
|
|
|
|
config.exposure_2 = hdr_merge_exposure_2_;
|
|
|
|
|
config.gain_2 = hdr_merge_gain_2_;
|
|
|
|
|
device_->setStructuredData(OB_STRUCT_DEPTH_HDR_CONFIG, &config, sizeof(config));
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Skip setting filter: " << filter_name);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-04-16 20:37:29 +08:00
|
|
|
void OBCameraNode::selectBaseStream() {
|
|
|
|
|
if (enable_stream_[DEPTH]) {
|
|
|
|
|
base_stream_ = DEPTH;
|
|
|
|
|
} else if (enable_stream_[INFRA0]) {
|
|
|
|
|
base_stream_ = INFRA0;
|
|
|
|
|
} else if (enable_stream_[INFRA1]) {
|
|
|
|
|
base_stream_ = INFRA1;
|
|
|
|
|
} else if (enable_stream_[INFRA2]) {
|
|
|
|
|
base_stream_ = INFRA2;
|
|
|
|
|
} else if (enable_stream_[COLOR]) {
|
|
|
|
|
base_stream_ = COLOR;
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-05-08 20:59:49 +08:00
|
|
|
|
|
|
|
|
void OBCameraNode::printSensorProfiles(const std::shared_ptr<ob::Sensor> &sensor) {
|
|
|
|
|
auto profiles = sensor->getStreamProfileList();
|
|
|
|
|
for (size_t i = 0; i < profiles->count(); i++) {
|
|
|
|
|
auto origin_profile = profiles->getProfile(i);
|
|
|
|
|
if (sensor->type() == OB_SENSOR_COLOR) {
|
|
|
|
|
auto profile = origin_profile->as<ob::VideoStreamProfile>();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "color profile: " << profile->width() << "x" << profile->height()
|
|
|
|
|
<< " " << profile->fps() << "fps "
|
|
|
|
|
<< profile->format());
|
|
|
|
|
} else if (sensor->type() == OB_SENSOR_DEPTH) {
|
|
|
|
|
auto profile = origin_profile->as<ob::VideoStreamProfile>();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "depth profile: " << profile->width() << "x" << profile->height()
|
|
|
|
|
<< " " << profile->fps() << "fps "
|
|
|
|
|
<< profile->format());
|
|
|
|
|
} else if (sensor->type() == OB_SENSOR_IR) {
|
|
|
|
|
auto profile = origin_profile->as<ob::VideoStreamProfile>();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "ir profile: " << profile->width() << "x" << profile->height()
|
|
|
|
|
<< " " << profile->fps() << "fps "
|
|
|
|
|
<< profile->format());
|
|
|
|
|
} else if (sensor->type() == OB_SENSOR_ACCEL) {
|
|
|
|
|
auto profile = origin_profile->as<ob::AccelStreamProfile>();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "accel profile: sampleRate " << profile->sampleRate()
|
|
|
|
|
<< " full scale_range "
|
|
|
|
|
<< profile->fullScaleRange());
|
|
|
|
|
} else if (sensor->type() == OB_SENSOR_GYRO) {
|
|
|
|
|
auto profile = origin_profile->as<ob::GyroStreamProfile>();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "gyro profile: sampleRate " << profile->sampleRate()
|
|
|
|
|
<< " full scale_range "
|
|
|
|
|
<< profile->fullScaleRange());
|
|
|
|
|
} else {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "unknown profile: " << magic_enum::enum_name(sensor->type()));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
void OBCameraNode::setupProfiles() {
|
2024-03-15 10:24:38 +08:00
|
|
|
// Image stream
|
2023-08-27 14:06:25 +08:00
|
|
|
for (const auto &elem : IMAGE_STREAMS) {
|
2023-02-06 17:26:10 +08:00
|
|
|
if (enable_stream_[elem]) {
|
2023-08-27 14:06:25 +08:00
|
|
|
const auto &sensor = sensors_[elem];
|
2023-02-16 16:14:38 +08:00
|
|
|
CHECK_NOTNULL(sensor.get());
|
2022-06-06 19:00:12 +08:00
|
|
|
auto profiles = sensor->getStreamProfileList();
|
2023-02-16 16:14:38 +08:00
|
|
|
CHECK_NOTNULL(profiles.get());
|
|
|
|
|
CHECK(profiles->count() > 0);
|
2022-06-06 19:00:12 +08:00
|
|
|
for (size_t i = 0; i < profiles->count(); i++) {
|
2024-08-29 15:17:42 +08:00
|
|
|
auto base_profile = profiles->getProfile(i);
|
|
|
|
|
if (base_profile == nullptr) {
|
|
|
|
|
throw std::runtime_error("Failed to get profile " + std::to_string(i));
|
|
|
|
|
}
|
|
|
|
|
auto profile = base_profile->as<ob::VideoStreamProfile>();
|
|
|
|
|
if (profile == nullptr) {
|
|
|
|
|
throw std::runtime_error("Failed cast profile to VideoStreamProfile");
|
|
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
RCLCPP_DEBUG_STREAM(
|
2025-09-28 09:39:13 +08:00
|
|
|
logger_, "Sensor profile: "
|
|
|
|
|
<< "stream_type: " << magic_enum::enum_name(profile->type())
|
|
|
|
|
<< "Format: " << profile->format() << ", Width: " << profile->width()
|
|
|
|
|
<< ", Height: " << profile->height() << ", FPS: " << profile->fps());
|
2023-02-20 14:19:06 +08:00
|
|
|
supported_profiles_[elem].emplace_back(profile);
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
2023-02-16 16:14:38 +08:00
|
|
|
std::shared_ptr<ob::VideoStreamProfile> selected_profile;
|
|
|
|
|
std::shared_ptr<ob::VideoStreamProfile> default_profile;
|
|
|
|
|
try {
|
2024-05-09 11:34:03 +08:00
|
|
|
if (width_[elem] == 0 && height_[elem] == 0 && fps_[elem] == 0 &&
|
|
|
|
|
format_[elem] == OB_FORMAT_UNKNOWN) {
|
|
|
|
|
selected_profile = profiles->getProfile(0)->as<ob::VideoStreamProfile>();
|
|
|
|
|
} else {
|
|
|
|
|
selected_profile = profiles->getVideoStreamProfile(width_[elem], height_[elem],
|
|
|
|
|
format_[elem], fps_[elem]);
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const ob::Error &ex) {
|
2023-09-11 18:34:11 +08:00
|
|
|
RCLCPP_ERROR_STREAM(
|
2024-05-08 20:59:49 +08:00
|
|
|
logger_, "Failed to get " << stream_name_[elem] << " profile: " << ex.getMessage());
|
2023-02-16 16:14:38 +08:00
|
|
|
RCLCPP_ERROR_STREAM(
|
|
|
|
|
logger_, "Stream: " << magic_enum::enum_name(elem.first)
|
|
|
|
|
<< ", Stream Index: " << elem.second << ", Width: " << width_[elem]
|
|
|
|
|
<< ", Height: " << height_[elem] << ", FPS: " << fps_[elem]
|
|
|
|
|
<< ", Format: " << magic_enum::enum_name(format_[elem]));
|
2024-05-08 20:59:49 +08:00
|
|
|
RCLCPP_ERROR(logger_,
|
|
|
|
|
"Error: The device might be connected via USB 2.0. Please verify your "
|
|
|
|
|
"configuration and try again. The current process will now exit.");
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Available profiles:");
|
|
|
|
|
printSensorProfiles(sensor);
|
2024-08-29 15:17:42 +08:00
|
|
|
RCLCPP_ERROR(logger_, "Because can not set this stream, so exit.");
|
2023-09-05 17:59:19 +08:00
|
|
|
exit(-1);
|
2023-02-16 16:14:38 +08:00
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
|
|
|
|
|
if (!selected_profile) {
|
|
|
|
|
RCLCPP_WARN_STREAM(logger_, "Given stream configuration is not supported by the device! "
|
|
|
|
|
<< " Stream: " << magic_enum::enum_name(elem.first)
|
|
|
|
|
<< ", Stream Index: " << elem.second
|
|
|
|
|
<< ", Width: " << width_[elem]
|
|
|
|
|
<< ", Height: " << height_[elem] << ", FPS: " << fps_[elem]
|
2022-06-09 17:47:12 +08:00
|
|
|
<< ", Format: " << magic_enum::enum_name(format_[elem]));
|
2022-06-06 19:00:12 +08:00
|
|
|
if (default_profile) {
|
|
|
|
|
RCLCPP_WARN_STREAM(logger_, "Using default profile instead.");
|
2022-06-09 17:47:12 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "default FPS " << default_profile->fps());
|
2022-06-06 19:00:12 +08:00
|
|
|
selected_profile = default_profile;
|
|
|
|
|
} else {
|
|
|
|
|
RCLCPP_ERROR_STREAM(
|
|
|
|
|
logger_, " NO default_profile found , Stream: " << magic_enum::enum_name(elem.first)
|
|
|
|
|
<< " will be disable");
|
2023-02-06 17:26:10 +08:00
|
|
|
enable_stream_[elem] = false;
|
2022-06-06 19:00:12 +08:00
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
CHECK_NOTNULL(selected_profile);
|
2023-02-13 15:34:44 +08:00
|
|
|
stream_profile_[elem] = selected_profile;
|
2024-05-09 17:07:43 +08:00
|
|
|
height_[elem] = static_cast<int>(selected_profile->height());
|
|
|
|
|
width_[elem] = static_cast<int>(selected_profile->width());
|
|
|
|
|
fps_[elem] = static_cast<int>(selected_profile->fps());
|
|
|
|
|
format_[elem] = selected_profile->format();
|
|
|
|
|
updateImageConfig(elem);
|
2022-06-06 19:00:12 +08:00
|
|
|
images_[elem] =
|
2023-02-06 17:26:10 +08:00
|
|
|
cv::Mat(height_[elem], width_[elem], image_format_[elem], cv::Scalar(0, 0, 0));
|
2022-06-06 19:00:12 +08:00
|
|
|
RCLCPP_INFO_STREAM(
|
2024-05-09 11:34:03 +08:00
|
|
|
logger_, " stream " << stream_name_[elem]
|
|
|
|
|
<< " is enabled - width: " << selected_profile->width()
|
|
|
|
|
<< ", height: " << selected_profile->height()
|
|
|
|
|
<< ", fps: " << selected_profile->fps() << ", "
|
2022-06-24 17:00:18 +08:00
|
|
|
<< "Format: " << magic_enum::enum_name(selected_profile->format()));
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
// IMU
|
|
|
|
|
for (const auto &stream_index : HID_STREAMS) {
|
|
|
|
|
if (!enable_stream_[stream_index]) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
try {
|
|
|
|
|
auto profile_list = sensors_[stream_index]->getStreamProfileList();
|
|
|
|
|
if (stream_index == ACCEL) {
|
|
|
|
|
auto full_scale_range = fullAccelScaleRangeFromString(imu_range_[stream_index]);
|
|
|
|
|
auto sample_rate = sampleRateFromString(imu_rate_[stream_index]);
|
|
|
|
|
auto profile = profile_list->getAccelStreamProfile(full_scale_range, sample_rate);
|
|
|
|
|
stream_profile_[stream_index] = profile;
|
|
|
|
|
} else if (stream_index == GYRO) {
|
|
|
|
|
auto full_scale_range = fullGyroScaleRangeFromString(imu_range_[stream_index]);
|
|
|
|
|
auto sample_rate = sampleRateFromString(imu_rate_[stream_index]);
|
|
|
|
|
auto profile = profile_list->getGyroStreamProfile(full_scale_range, sample_rate);
|
|
|
|
|
stream_profile_[stream_index] = profile;
|
|
|
|
|
}
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "stream " << stream_name_[stream_index] << " full scale range "
|
|
|
|
|
<< imu_range_[stream_index] << " sample rate "
|
|
|
|
|
<< imu_rate_[stream_index]);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Failed to setup << " << stream_name_[stream_index]
|
|
|
|
|
<< " profile: " << e.getMessage());
|
|
|
|
|
enable_stream_[stream_index] = false;
|
|
|
|
|
stream_profile_[stream_index] = nullptr;
|
|
|
|
|
}
|
|
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
2024-05-09 17:07:43 +08:00
|
|
|
void OBCameraNode::updateImageConfig(const stream_index_pair &stream_index) {
|
|
|
|
|
if (format_[stream_index] == OB_FORMAT_Y8) {
|
|
|
|
|
image_format_[stream_index] = CV_8UC1;
|
|
|
|
|
encoding_[stream_index] = stream_index.first == OB_STREAM_DEPTH
|
|
|
|
|
? sensor_msgs::image_encodings::TYPE_8UC1
|
|
|
|
|
: sensor_msgs::image_encodings::MONO8;
|
|
|
|
|
unit_step_size_[stream_index] = sizeof(uint8_t);
|
|
|
|
|
}
|
|
|
|
|
if (format_[stream_index] == OB_FORMAT_MJPG) {
|
|
|
|
|
if (stream_index.first == OB_STREAM_IR || stream_index.first == OB_STREAM_IR_LEFT ||
|
|
|
|
|
stream_index.first == OB_STREAM_IR_RIGHT) {
|
|
|
|
|
image_format_[stream_index] = CV_8UC1;
|
|
|
|
|
encoding_[stream_index] = sensor_msgs::image_encodings::MONO8;
|
|
|
|
|
unit_step_size_[stream_index] = sizeof(uint8_t);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (format_[stream_index] == OB_FORMAT_Y16 && stream_index == COLOR) {
|
|
|
|
|
image_format_[stream_index] = CV_16UC1;
|
|
|
|
|
encoding_[stream_index] = sensor_msgs::image_encodings::MONO16;
|
|
|
|
|
unit_step_size_[stream_index] = sizeof(uint16_t);
|
|
|
|
|
}
|
|
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
|
2023-02-13 15:34:44 +08:00
|
|
|
void OBCameraNode::startStreams() {
|
2022-06-13 16:02:49 +08:00
|
|
|
if (pipeline_ != nullptr) {
|
|
|
|
|
pipeline_.reset();
|
|
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
pipeline_ = std::make_unique<ob::Pipeline>(device_);
|
2023-02-13 15:34:44 +08:00
|
|
|
try {
|
|
|
|
|
setupPipelineConfig();
|
2023-08-27 14:06:25 +08:00
|
|
|
pipeline_->start(pipeline_config_, [this](const std::shared_ptr<ob::FrameSet> &frame_set) {
|
2023-02-13 15:34:44 +08:00
|
|
|
onNewFrameSetCallback(frame_set);
|
|
|
|
|
});
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const ob::Error &e) {
|
2023-02-13 15:34:44 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to start pipeline: " << e.getMessage());
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "try to disable ir stream and try again");
|
|
|
|
|
enable_stream_[INFRA0] = false;
|
|
|
|
|
setupPipelineConfig();
|
2023-08-27 14:06:25 +08:00
|
|
|
pipeline_->start(pipeline_config_, [this](const std::shared_ptr<ob::FrameSet> &frame_set) {
|
2023-02-13 15:34:44 +08:00
|
|
|
onNewFrameSetCallback(frame_set);
|
|
|
|
|
});
|
2023-10-12 20:52:10 +08:00
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to start pipeline");
|
|
|
|
|
throw std::runtime_error("Failed to start pipeline");
|
2023-02-13 15:34:44 +08:00
|
|
|
}
|
2023-12-27 14:53:26 +08:00
|
|
|
if (enable_stream_[COLOR] && !colorFrameThread_) {
|
|
|
|
|
colorFrameThread_ = std::make_shared<std::thread>([this]() { onNewColorFrameCallback(); });
|
2023-11-02 20:01:48 +08:00
|
|
|
}
|
2023-10-07 17:56:33 +08:00
|
|
|
if (enable_frame_sync_) {
|
2024-05-06 14:08:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Enable frame sync");
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_EXECUTE_BLOCK(pipeline_->enableFrameSync());
|
2024-01-29 09:18:21 +08:00
|
|
|
} else {
|
2024-05-06 14:08:57 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Disable frame sync");
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_EXECUTE_BLOCK(pipeline_->disableFrameSync());
|
2024-01-04 14:46:42 +08:00
|
|
|
}
|
2025-03-24 16:50:06 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_LDP_BOOL, OB_PERMISSION_READ_WRITE)) {
|
2024-11-28 16:15:26 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Setting LDP to " << (enable_ldp_ ? "ON" : "OFF"));
|
2025-03-24 16:50:06 +08:00
|
|
|
if (device_->isPropertySupported(OB_PROP_LASER_CONTROL_INT, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
auto laser_enable = device_->getIntProperty(OB_PROP_LASER_CONTROL_INT);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_LDP_BOOL, enable_ldp_);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_LASER_CONTROL_INT, laser_enable);
|
|
|
|
|
} else if (device_->isPropertySupported(OB_PROP_LASER_BOOL, OB_PERMISSION_READ_WRITE)) {
|
|
|
|
|
if (!enable_ldp_) {
|
|
|
|
|
auto laser_enable = device_->getIntProperty(OB_PROP_LASER_BOOL);
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_LDP_BOOL, enable_ldp_);
|
|
|
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(3));
|
|
|
|
|
TRY_TO_SET_PROPERTY(setIntProperty, OB_PROP_LASER_BOOL, laser_enable);
|
|
|
|
|
} else {
|
|
|
|
|
TRY_TO_SET_PROPERTY(setBoolProperty, OB_PROP_LDP_BOOL, enable_ldp_);
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-11-28 16:15:26 +08:00
|
|
|
}
|
2023-02-13 15:34:44 +08:00
|
|
|
pipeline_started_.store(true);
|
2023-04-28 09:44:16 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-15 10:24:38 +08:00
|
|
|
void OBCameraNode::startIMUSyncStream() {
|
|
|
|
|
if (imuPipeline_ != nullptr) {
|
|
|
|
|
imuPipeline_.reset();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
imuPipeline_ = std::make_unique<ob::Pipeline>(device_);
|
|
|
|
|
if (imu_sync_output_start_) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// ACCEL
|
|
|
|
|
auto accelProfiles = imuPipeline_->getStreamProfileList(OB_SENSOR_ACCEL);
|
|
|
|
|
auto accel_range = fullAccelScaleRangeFromString(imu_range_[ACCEL]);
|
|
|
|
|
auto accel_rate = sampleRateFromString(imu_rate_[ACCEL]);
|
|
|
|
|
auto accelProfile = accelProfiles->getAccelStreamProfile(accel_range, accel_rate);
|
|
|
|
|
// GYRO
|
|
|
|
|
auto gyroProfiles = imuPipeline_->getStreamProfileList(OB_SENSOR_GYRO);
|
|
|
|
|
auto gyro_range = fullGyroScaleRangeFromString(imu_range_[GYRO]);
|
|
|
|
|
auto gyro_rate = sampleRateFromString(imu_rate_[GYRO]);
|
|
|
|
|
auto gyroProfile = gyroProfiles->getGyroStreamProfile(gyro_range, gyro_rate);
|
|
|
|
|
std::shared_ptr<ob::Config> imuConfig = std::make_shared<ob::Config>();
|
|
|
|
|
imuConfig->enableStream(accelProfile);
|
|
|
|
|
imuConfig->enableStream(gyroProfile);
|
2024-08-30 10:46:26 +08:00
|
|
|
TRY_EXECUTE_BLOCK(imuPipeline_->enableFrameSync());
|
2024-03-15 10:24:38 +08:00
|
|
|
imuPipeline_->start(imuConfig, [&](std::shared_ptr<ob::Frame> frame) {
|
|
|
|
|
auto frameSet = frame->as<ob::FrameSet>();
|
|
|
|
|
auto aFrame = frameSet->getFrame(OB_FRAME_ACCEL);
|
|
|
|
|
auto gFrame = frameSet->getFrame(OB_FRAME_GYRO);
|
|
|
|
|
if (aFrame && gFrame) {
|
|
|
|
|
onNewIMUFrameSyncOutputCallback(aFrame, gFrame);
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
imu_sync_output_start_ = true;
|
|
|
|
|
if (!imu_sync_output_start_) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(
|
|
|
|
|
logger_, "Failed to start IMU stream, please check the imu_rate and imu_range parameters.");
|
|
|
|
|
} else {
|
|
|
|
|
RCLCPP_INFO_STREAM(
|
|
|
|
|
logger_, "start accel stream with range: " << fullAccelScaleRangeToString(accel_range)
|
|
|
|
|
<< ",rate:" << sampleRateToString(accel_rate)
|
|
|
|
|
<< ", and start gyro stream with range:"
|
|
|
|
|
<< fullGyroScaleRangeToString(gyro_range)
|
|
|
|
|
<< ",rate:" << sampleRateToString(gyro_rate));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-28 09:44:16 +08:00
|
|
|
void OBCameraNode::startIMU() {
|
2023-12-20 17:56:37 +08:00
|
|
|
if (enable_sync_output_accel_gyro_) {
|
2024-03-15 10:24:38 +08:00
|
|
|
startIMUSyncStream();
|
2023-12-20 17:56:37 +08:00
|
|
|
} else {
|
|
|
|
|
for (const auto &stream_index : HID_STREAMS) {
|
|
|
|
|
if (enable_stream_[stream_index] && !imu_started_[stream_index]) {
|
2024-03-15 10:24:38 +08:00
|
|
|
auto imu_profile = stream_profile_[stream_index];
|
2024-08-30 11:02:57 +08:00
|
|
|
CHECK_NOTNULL(imu_profile);
|
2024-03-15 10:24:38 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "start " << stream_name_[stream_index] << " stream");
|
2024-08-30 11:02:57 +08:00
|
|
|
CHECK_NOTNULL(sensors_[stream_index]);
|
2024-03-15 10:24:38 +08:00
|
|
|
sensors_[stream_index]->start(
|
|
|
|
|
imu_profile, [this, stream_index](const std::shared_ptr<ob::Frame> &frame) {
|
|
|
|
|
onNewIMUFrameCallback(frame, stream_index);
|
|
|
|
|
});
|
2023-12-20 17:56:37 +08:00
|
|
|
}
|
2023-04-28 09:44:16 +08:00
|
|
|
}
|
|
|
|
|
}
|
2023-02-13 15:34:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::stopStreams() {
|
|
|
|
|
if (!pipeline_started_ || !pipeline_) {
|
2024-08-23 11:31:22 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "pipeline not started or not exist, skip stop pipeline");
|
2023-02-13 15:34:44 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
try {
|
|
|
|
|
pipeline_->stop();
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const ob::Error &e) {
|
2025-09-28 10:11:32 +08:00
|
|
|
RCLCPP_WARN_STREAM(logger_, "Failed to stop pipeline: " << e.getMessage());
|
2024-08-23 11:31:22 +08:00
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to stop pipeline");
|
2023-02-13 15:34:44 +08:00
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
2023-04-28 09:44:16 +08:00
|
|
|
void OBCameraNode::stopIMU() {
|
2023-12-20 17:56:37 +08:00
|
|
|
if (enable_sync_output_accel_gyro_) {
|
|
|
|
|
if (!imu_sync_output_start_ || !imuPipeline_) {
|
2024-08-23 11:31:22 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "imu pipeline not started or not exist, skip stop imu pipeline");
|
2023-12-20 17:56:37 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
try {
|
|
|
|
|
imuPipeline_->stop();
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to stop imu pipeline: " << e.getMessage());
|
2024-08-23 11:31:22 +08:00
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to stop imu pipeline");
|
2023-12-20 17:56:37 +08:00
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
for (const auto &stream_index : HID_STREAMS) {
|
|
|
|
|
if (imu_started_[stream_index]) {
|
|
|
|
|
CHECK(sensors_.count(stream_index));
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "stop " << stream_name_[stream_index] << " stream");
|
2024-08-23 11:31:22 +08:00
|
|
|
try {
|
|
|
|
|
sensors_[stream_index]->stop();
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to stop " << stream_name_[stream_index]
|
|
|
|
|
<< " stream: " << e.getMessage());
|
|
|
|
|
}
|
2023-12-20 17:56:37 +08:00
|
|
|
imu_started_[stream_index] = false;
|
|
|
|
|
}
|
2023-04-28 09:44:16 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-06 17:26:10 +08:00
|
|
|
void OBCameraNode::setupDefaultImageFormat() {
|
|
|
|
|
format_[DEPTH] = OB_FORMAT_Y16;
|
|
|
|
|
format_str_[DEPTH] = "Y16";
|
|
|
|
|
image_format_[DEPTH] = CV_16UC1;
|
|
|
|
|
encoding_[DEPTH] = sensor_msgs::image_encodings::TYPE_16UC1;
|
|
|
|
|
unit_step_size_[DEPTH] = sizeof(uint16_t);
|
|
|
|
|
format_[INFRA0] = OB_FORMAT_Y16;
|
|
|
|
|
format_str_[INFRA0] = "Y16";
|
|
|
|
|
image_format_[INFRA0] = CV_16UC1;
|
|
|
|
|
encoding_[INFRA0] = sensor_msgs::image_encodings::MONO16;
|
2023-05-24 18:07:03 +08:00
|
|
|
unit_step_size_[INFRA0] = sizeof(uint16_t);
|
2023-02-06 17:26:10 +08:00
|
|
|
|
2023-09-05 17:59:19 +08:00
|
|
|
format_[INFRA1] = OB_FORMAT_Y16;
|
|
|
|
|
format_str_[INFRA1] = "Y16";
|
|
|
|
|
image_format_[INFRA1] = CV_16UC1;
|
|
|
|
|
encoding_[INFRA1] = sensor_msgs::image_encodings::MONO16;
|
|
|
|
|
unit_step_size_[INFRA1] = sizeof(uint16_t);
|
|
|
|
|
|
|
|
|
|
format_[INFRA2] = OB_FORMAT_Y16;
|
|
|
|
|
format_str_[INFRA2] = "Y16";
|
|
|
|
|
image_format_[INFRA2] = CV_16UC1;
|
|
|
|
|
encoding_[INFRA2] = sensor_msgs::image_encodings::MONO16;
|
|
|
|
|
unit_step_size_[INFRA2] = sizeof(uint16_t);
|
|
|
|
|
|
2023-02-06 17:26:10 +08:00
|
|
|
image_format_[COLOR] = CV_8UC3;
|
|
|
|
|
encoding_[COLOR] = sensor_msgs::image_encodings::RGB8;
|
|
|
|
|
unit_step_size_[COLOR] = 3 * sizeof(uint8_t);
|
|
|
|
|
}
|
|
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
void OBCameraNode::getParameters() {
|
2023-04-28 09:44:16 +08:00
|
|
|
setAndGetNodeParameter<std::string>(camera_name_, "camera_name", "camera");
|
2023-06-13 15:23:58 +08:00
|
|
|
camera_link_frame_id_ = camera_name_ + "_link";
|
2022-06-06 19:00:12 +08:00
|
|
|
for (auto stream_index : IMAGE_STREAMS) {
|
2023-02-06 17:26:10 +08:00
|
|
|
std::string param_name = stream_name_[stream_index] + "_width";
|
2024-05-09 11:34:03 +08:00
|
|
|
setAndGetNodeParameter(width_[stream_index], param_name, 0);
|
2023-02-06 17:26:10 +08:00
|
|
|
param_name = stream_name_[stream_index] + "_height";
|
2024-05-09 11:34:03 +08:00
|
|
|
setAndGetNodeParameter(height_[stream_index], param_name, 0);
|
2023-02-06 17:26:10 +08:00
|
|
|
param_name = stream_name_[stream_index] + "_fps";
|
2024-05-09 11:34:03 +08:00
|
|
|
setAndGetNodeParameter(fps_[stream_index], param_name, 0);
|
2023-02-06 17:26:10 +08:00
|
|
|
param_name = "enable_" + stream_name_[stream_index];
|
2024-08-27 17:58:12 +08:00
|
|
|
if (stream_index == DEPTH) {
|
|
|
|
|
setAndGetNodeParameter(enable_stream_[stream_index], param_name, true);
|
|
|
|
|
} else {
|
|
|
|
|
setAndGetNodeParameter(enable_stream_[stream_index], param_name, false);
|
|
|
|
|
}
|
2023-10-27 18:05:04 +08:00
|
|
|
param_name = "flip_" + stream_name_[stream_index];
|
|
|
|
|
setAndGetNodeParameter(flip_stream_[stream_index], param_name, false);
|
2023-04-28 09:44:16 +08:00
|
|
|
param_name = camera_name_ + "_" + stream_name_[stream_index] + "_frame_id";
|
|
|
|
|
std::string default_frame_id = camera_name_ + "_" + stream_name_[stream_index] + "_frame";
|
2022-06-09 13:52:42 +08:00
|
|
|
setAndGetNodeParameter(frame_id_[stream_index], param_name, default_frame_id);
|
2022-06-06 19:00:12 +08:00
|
|
|
std::string default_optical_frame_id =
|
2023-04-28 09:44:16 +08:00
|
|
|
camera_name_ + "_" + stream_name_[stream_index] + "_optical_frame";
|
2023-02-06 17:26:10 +08:00
|
|
|
param_name = stream_name_[stream_index] + "_optical_frame_id";
|
2022-06-09 13:52:42 +08:00
|
|
|
setAndGetNodeParameter(optical_frame_id_[stream_index], param_name, default_optical_frame_id);
|
2023-02-06 17:26:10 +08:00
|
|
|
param_name = stream_name_[stream_index] + "_format";
|
2022-06-27 10:26:04 +08:00
|
|
|
setAndGetNodeParameter(format_str_[stream_index], param_name, format_str_[stream_index]);
|
|
|
|
|
format_[stream_index] = OBFormatFromString(format_str_[stream_index]);
|
2024-05-09 17:07:43 +08:00
|
|
|
updateImageConfig(stream_index);
|
2023-02-06 17:26:10 +08:00
|
|
|
param_name = stream_name_[stream_index] + "_qos";
|
|
|
|
|
setAndGetNodeParameter<std::string>(image_qos_[stream_index], param_name, "default");
|
|
|
|
|
param_name = stream_name_[stream_index] + "_camera_info_qos";
|
|
|
|
|
setAndGetNodeParameter<std::string>(camera_info_qos_[stream_index], param_name, "default");
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
2023-04-28 09:44:16 +08:00
|
|
|
|
2023-11-23 14:56:39 +08:00
|
|
|
for (auto stream_index : IMAGE_STREAMS) {
|
|
|
|
|
depth_aligned_frame_id_[stream_index] = optical_frame_id_[COLOR];
|
|
|
|
|
}
|
|
|
|
|
|
2023-12-20 17:56:37 +08:00
|
|
|
setAndGetNodeParameter(enable_sync_output_accel_gyro_, "enable_sync_output_accel_gyro", false);
|
2023-08-27 14:06:25 +08:00
|
|
|
for (const auto &stream_index : HID_STREAMS) {
|
2023-04-28 09:44:16 +08:00
|
|
|
std::string param_name = stream_name_[stream_index] + "_qos";
|
|
|
|
|
setAndGetNodeParameter<std::string>(imu_qos_[stream_index], param_name, "default");
|
|
|
|
|
param_name = "enable_" + stream_name_[stream_index];
|
|
|
|
|
setAndGetNodeParameter(enable_stream_[stream_index], param_name, false);
|
2024-01-29 09:18:21 +08:00
|
|
|
if (enable_sync_output_accel_gyro_) {
|
2023-12-20 17:56:37 +08:00
|
|
|
enable_stream_[stream_index] = true;
|
|
|
|
|
}
|
2023-04-28 09:44:16 +08:00
|
|
|
param_name = stream_name_[stream_index] + "_rate";
|
|
|
|
|
setAndGetNodeParameter<std::string>(imu_rate_[stream_index], param_name, "");
|
|
|
|
|
param_name = stream_name_[stream_index] + "_range";
|
|
|
|
|
setAndGetNodeParameter<std::string>(imu_range_[stream_index], param_name, "");
|
|
|
|
|
param_name = camera_name_ + "_" + stream_name_[stream_index] + "_frame_id";
|
|
|
|
|
std::string default_frame_id = camera_name_ + "_" + stream_name_[stream_index] + "_frame";
|
|
|
|
|
setAndGetNodeParameter(frame_id_[stream_index], param_name, default_frame_id);
|
|
|
|
|
std::string default_optical_frame_id =
|
|
|
|
|
camera_name_ + "_" + stream_name_[stream_index] + "_optical_frame";
|
|
|
|
|
param_name = stream_name_[stream_index] + "_optical_frame_id";
|
|
|
|
|
setAndGetNodeParameter(optical_frame_id_[stream_index], param_name, default_optical_frame_id);
|
2024-01-29 09:18:21 +08:00
|
|
|
depth_aligned_frame_id_[stream_index] =
|
|
|
|
|
camera_name_ + "_" + stream_name_[COLOR] + "_optical_frame";
|
2023-04-28 09:44:16 +08:00
|
|
|
}
|
|
|
|
|
|
2024-12-24 20:31:52 +08:00
|
|
|
accel_gyro_frame_id_ = camera_name_ + "_accel_gyro_optical_frame";
|
|
|
|
|
|
2022-06-09 13:52:42 +08:00
|
|
|
setAndGetNodeParameter(publish_tf_, "publish_tf", true);
|
2024-07-13 14:07:29 +08:00
|
|
|
setAndGetNodeParameter(tf_publish_rate_, "tf_publish_rate", 0.0);
|
2023-02-06 17:26:10 +08:00
|
|
|
setAndGetNodeParameter(depth_registration_, "depth_registration", false);
|
2024-07-01 17:32:33 +08:00
|
|
|
setAndGetNodeParameter(enable_point_cloud_, "enable_point_cloud", false);
|
2023-02-06 17:26:10 +08:00
|
|
|
setAndGetNodeParameter<std::string>(ir_info_url_, "ir_info_url", "");
|
|
|
|
|
setAndGetNodeParameter<std::string>(color_info_url_, "color_info_url", "");
|
|
|
|
|
setAndGetNodeParameter(enable_colored_point_cloud_, "enable_colored_point_cloud", false);
|
2024-07-01 17:33:14 +08:00
|
|
|
setAndGetNodeParameter(enable_point_cloud_, "enable_point_cloud", false);
|
2023-02-06 17:26:10 +08:00
|
|
|
setAndGetNodeParameter<std::string>(point_cloud_qos_, "point_cloud_qos", "default");
|
2023-02-20 12:11:12 +08:00
|
|
|
setAndGetNodeParameter(enable_d2c_viewer_, "enable_d2c_viewer", false);
|
2023-03-01 22:01:08 +08:00
|
|
|
setAndGetNodeParameter(enable_hardware_d2d_, "enable_hardware_d2d", true);
|
2024-07-01 17:32:33 +08:00
|
|
|
setAndGetNodeParameter(enable_soft_filter_, "enable_soft_filter", false);
|
2023-12-20 17:56:37 +08:00
|
|
|
setAndGetNodeParameter<std::string>(depth_filter_config_, "depth_filter_config", "");
|
|
|
|
|
if (!depth_filter_config_.empty()) {
|
|
|
|
|
enable_depth_filter_ = true;
|
|
|
|
|
}
|
2023-10-07 17:56:33 +08:00
|
|
|
setAndGetNodeParameter(enable_frame_sync_, "enable_frame_sync", false);
|
2025-10-21 14:41:10 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_color_auto_exposure_priority_,
|
|
|
|
|
"enable_color_auto_exposure_priority", false);
|
2023-03-20 11:18:59 +08:00
|
|
|
setAndGetNodeParameter(enable_color_auto_exposure_, "enable_color_auto_exposure", true);
|
2024-06-17 14:19:02 +08:00
|
|
|
setAndGetNodeParameter(enable_color_auto_white_balance_, "enable_color_auto_white_balance", true);
|
2025-04-22 16:38:10 +08:00
|
|
|
setAndGetNodeParameter<int>(color_rotation_, "color_rotation", -1);
|
2024-06-17 14:19:02 +08:00
|
|
|
setAndGetNodeParameter<int>(color_exposure_, "color_exposure", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_gain_, "color_gain", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_white_balance_, "color_white_balance", -1);
|
2024-09-18 14:13:01 +08:00
|
|
|
setAndGetNodeParameter<int>(color_ae_max_exposure_, "color_ae_max_exposure", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_brightness_, "color_brightness", -1);
|
2024-11-05 16:18:02 +08:00
|
|
|
setAndGetNodeParameter<int>(color_sharpness_, "color_sharpness", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_saturation_, "color_saturation", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_contrast_, "color_contrast", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_gamma_, "color_gamma", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_hue_, "color_hue", -1);
|
2025-04-22 16:38:10 +08:00
|
|
|
setAndGetNodeParameter<int>(depth_rotation_, "depth_rotation", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(left_ir_rotation_, "left_ir_rotation", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(right_ir_rotation_, "right_ir_rotation", -1);
|
2023-03-20 11:18:59 +08:00
|
|
|
setAndGetNodeParameter(enable_ir_auto_exposure_, "enable_ir_auto_exposure", true);
|
2024-06-17 14:19:02 +08:00
|
|
|
setAndGetNodeParameter<int>(ir_exposure_, "ir_exposure", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(ir_gain_, "ir_gain", -1);
|
2024-09-18 14:13:01 +08:00
|
|
|
setAndGetNodeParameter<int>(ir_ae_max_exposure_, "ir_ae_max_exposure", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(ir_brightness_, "ir_brightness", -1);
|
2024-01-15 19:51:41 +08:00
|
|
|
setAndGetNodeParameter(enable_ir_long_exposure_, "enable_ir_long_exposure", true);
|
2023-03-20 11:18:59 +08:00
|
|
|
setAndGetNodeParameter<std::string>(depth_work_mode_, "depth_work_mode", "");
|
2024-05-15 14:01:07 +08:00
|
|
|
setAndGetNodeParameter<std::string>(sync_mode_str_, "sync_mode", "");
|
2023-09-05 17:59:19 +08:00
|
|
|
setAndGetNodeParameter(depth_delay_us_, "depth_delay_us", 0);
|
|
|
|
|
setAndGetNodeParameter(color_delay_us_, "color_delay_us", 0);
|
|
|
|
|
setAndGetNodeParameter(trigger2image_delay_us_, "trigger2image_delay_us", 0);
|
2023-10-10 09:31:44 +08:00
|
|
|
setAndGetNodeParameter(trigger_out_delay_us_, "trigger_out_delay_us", 0);
|
|
|
|
|
setAndGetNodeParameter(trigger_out_enabled_, "trigger_out_enabled", false);
|
2024-05-06 10:34:19 +08:00
|
|
|
setAndGetNodeParameter<std::string>(depth_precision_str_, "depth_precision", "");
|
2024-09-13 10:18:09 +08:00
|
|
|
setAndGetNodeParameter<std::string>(cloud_frame_id_, "cloud_frame_id", "");
|
2024-05-06 10:34:19 +08:00
|
|
|
if (!depth_precision_str_.empty()) {
|
|
|
|
|
depth_precision_ = depthPrecisionLevelFromString(depth_precision_str_);
|
|
|
|
|
}
|
2024-09-09 19:41:42 +08:00
|
|
|
if (enable_colored_point_cloud_ || enable_d2c_viewer_) {
|
2023-02-06 17:26:10 +08:00
|
|
|
depth_registration_ = true;
|
|
|
|
|
}
|
2024-05-24 16:08:14 +08:00
|
|
|
if (!enable_stream_[COLOR]) {
|
|
|
|
|
enable_colored_point_cloud_ = false;
|
|
|
|
|
depth_registration_ = false;
|
|
|
|
|
}
|
2023-03-24 14:57:33 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_ldp_, "enable_ldp", true);
|
|
|
|
|
setAndGetNodeParameter<int>(soft_filter_max_diff_, "soft_filter_max_diff", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(soft_filter_speckle_size_, "soft_filter_speckle_size", -1);
|
2023-04-28 09:44:16 +08:00
|
|
|
setAndGetNodeParameter<double>(liner_accel_cov_, "linear_accel_cov", 0.0003);
|
|
|
|
|
setAndGetNodeParameter<double>(angular_vel_cov_, "angular_vel_cov", 0.02);
|
2023-11-15 11:05:38 +08:00
|
|
|
setAndGetNodeParameter<bool>(ordered_pc_, "ordered_pc", false);
|
2024-01-30 11:21:18 +08:00
|
|
|
setAndGetNodeParameter<int>(max_save_images_count_, "max_save_images_count", 10);
|
2024-02-28 18:27:42 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_depth_scale_, "enable_depth_scale", true);
|
2024-05-06 15:39:18 +08:00
|
|
|
setAndGetNodeParameter<std::string>(device_preset_, "device_preset", "");
|
2024-04-07 15:30:59 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_decimation_filter_, "enable_decimation_filter", false);
|
|
|
|
|
setAndGetNodeParameter<bool>(enable_hdr_merge_, "enable_hdr_merge", false);
|
|
|
|
|
setAndGetNodeParameter<bool>(enable_sequence_id_filter_, "enable_sequence_id_filter", false);
|
|
|
|
|
setAndGetNodeParameter<bool>(enable_threshold_filter_, "enable_threshold_filter", false);
|
|
|
|
|
setAndGetNodeParameter<bool>(enable_noise_removal_filter_, "enable_noise_removal_filter", true);
|
2024-05-15 13:44:53 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_spatial_filter_, "enable_spatial_filter", false);
|
2024-04-07 15:30:59 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_temporal_filter_, "enable_temporal_filter", false);
|
|
|
|
|
setAndGetNodeParameter<bool>(enable_hole_filling_filter_, "enable_hole_filling_filter", false);
|
2024-12-23 16:04:46 +08:00
|
|
|
setAndGetNodeParameter<int>(decimation_filter_scale_, "decimation_filter_scale", -1);
|
2024-05-15 13:44:53 +08:00
|
|
|
setAndGetNodeParameter<int>(sequence_id_filter_id_, "sequence_id_filter_id", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(threshold_filter_max_, "threshold_filter_max", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(threshold_filter_min_, "threshold_filter_min", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(noise_removal_filter_min_diff_, "noise_removal_filter_min_diff", 256);
|
2024-04-07 15:30:59 +08:00
|
|
|
setAndGetNodeParameter<int>(noise_removal_filter_max_size_, "noise_removal_filter_max_size", 80);
|
2024-05-15 13:44:53 +08:00
|
|
|
setAndGetNodeParameter<float>(spatial_filter_alpha_, "spatial_filter_alpha", -1.0);
|
|
|
|
|
setAndGetNodeParameter<int>(spatial_filter_diff_threshold_, "spatial_filter_diff_threshold", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(spatial_filter_magnitude_, "spatial_filter_magnitude", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(spatial_filter_radius_, "spatial_filter_radius", -1);
|
2024-04-07 15:30:59 +08:00
|
|
|
setAndGetNodeParameter<float>(temporal_filter_diff_threshold_, "temporal_filter_diff_threshold",
|
2024-05-15 13:44:53 +08:00
|
|
|
-1.0);
|
|
|
|
|
setAndGetNodeParameter<float>(temporal_filter_weight_, "temporal_filter_weight", -1.0);
|
|
|
|
|
setAndGetNodeParameter<std::string>(hole_filling_filter_mode_, "hole_filling_filter_mode", "");
|
|
|
|
|
setAndGetNodeParameter<int>(hdr_merge_exposure_1_, "hdr_merge_exposure_1", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(hdr_merge_gain_1_, "hdr_merge_gain_1", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(hdr_merge_exposure_2_, "hdr_merge_exposure_2", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(hdr_merge_gain_2_, "hdr_merge_gain_2", -1);
|
2024-04-07 15:51:21 +08:00
|
|
|
setAndGetNodeParameter<std::string>(align_mode_, "align_mode", "HW");
|
2024-04-09 16:08:48 +08:00
|
|
|
setAndGetNodeParameter<double>(diagnostic_period_, "diagnostic_period", 1.0);
|
2024-05-15 13:44:53 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_laser_, "enable_laser", true);
|
2024-04-18 10:19:32 +08:00
|
|
|
setAndGetNodeParameter<int>(laser_on_off_mode_, "laser_on_off_mode", 0);
|
2024-05-06 10:34:19 +08:00
|
|
|
std::string align_target_stream_str_;
|
|
|
|
|
setAndGetNodeParameter<std::string>(align_target_stream_str_, "align_target_stream", "COLOR");
|
|
|
|
|
align_target_stream_ = obStreamTypeFromString(align_target_stream_str_);
|
2024-05-08 20:59:49 +08:00
|
|
|
setAndGetNodeParameter<bool>(retry_on_usb3_detection_failure_, "retry_on_usb3_detection_failure",
|
|
|
|
|
false);
|
2024-06-13 11:08:20 +08:00
|
|
|
setAndGetNodeParameter<int>(laser_energy_level_, "laser_energy_level", -1);
|
2024-07-09 13:55:34 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_3d_reconstruction_mode_, "enable_3d_reconstruction_mode",
|
|
|
|
|
false);
|
2024-07-17 16:56:52 +08:00
|
|
|
setAndGetNodeParameter<int>(min_depth_limit_, "min_depth_limit", 0);
|
|
|
|
|
setAndGetNodeParameter<int>(max_depth_limit_, "max_depth_limit", 0);
|
2024-07-31 11:32:36 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_heartbeat_, "enable_heartbeat", false);
|
2025-04-22 17:51:40 +08:00
|
|
|
setAndGetNodeParameter<std::string>(industry_mode_, "industry_mode", "");
|
2024-07-31 14:30:54 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_color_undistortion_, "enable_color_undistortion", false);
|
2024-09-26 09:45:15 +08:00
|
|
|
setAndGetNodeParameter<int>(color_ae_roi_left_, "color_ae_roi_left", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_ae_roi_top_, "color_ae_roi_top", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_ae_roi_right_, "color_ae_roi_right", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(color_ae_roi_bottom_, "color_ae_roi_bottom", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(depth_ae_roi_left_, "depth_ae_roi_left", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(depth_ae_roi_top_, "depth_ae_roi_top", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(depth_ae_roi_right_, "depth_ae_roi_right", -1);
|
|
|
|
|
setAndGetNodeParameter<int>(depth_ae_roi_bottom_, "depth_ae_roi_bottom", -1);
|
2024-12-23 16:13:44 +08:00
|
|
|
|
2024-07-17 16:56:52 +08:00
|
|
|
setAndGetNodeParameter<std::string>(time_domain_, "time_domain", "device");
|
2026-04-13 10:41:51 +08:00
|
|
|
setAndGetNodeParameter<bool>(enable_frame_timestamp_csv_, "enable_frame_timestamp_csv", false);
|
|
|
|
|
setAndGetNodeParameter<std::string>(frame_timestamp_csv_file_, "frame_timestamp_csv_file", "");
|
2024-07-17 16:56:52 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
CHECK_NOTNULL(device_info.get());
|
|
|
|
|
auto pid = device_info->pid();
|
|
|
|
|
if (isOpenNIDevice(pid)) {
|
|
|
|
|
time_domain_ = "system";
|
|
|
|
|
}
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "current time domain: " << time_domain_);
|
|
|
|
|
setAndGetNodeParameter<int>(frames_per_trigger_, "frames_per_trigger", 2);
|
|
|
|
|
long software_trigger_period = 33;
|
|
|
|
|
setAndGetNodeParameter<long>(software_trigger_period, "software_trigger_period", 33);
|
|
|
|
|
software_trigger_period_ = std::chrono::milliseconds(software_trigger_period);
|
2025-01-13 18:02:22 +08:00
|
|
|
|
|
|
|
|
setAndGetNodeParameter<std::string>(frame_aggregate_mode_, "frame_aggregate_mode", "ANY");
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::setupTopics() {
|
2024-08-01 17:35:06 +08:00
|
|
|
try {
|
2024-08-13 16:13:50 +08:00
|
|
|
getParameters();
|
|
|
|
|
setupDevices();
|
|
|
|
|
setupDepthPostProcessFilter();
|
|
|
|
|
setupProfiles();
|
2024-11-07 11:43:40 +08:00
|
|
|
setupCameraInfo();
|
2024-08-13 16:13:50 +08:00
|
|
|
selectBaseStream();
|
|
|
|
|
setupCameraCtrlServices();
|
|
|
|
|
setupPublishers();
|
|
|
|
|
setupDiagnosticUpdater();
|
2024-08-01 17:35:06 +08:00
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to setup topics: " << e.getMessage());
|
2024-08-29 15:17:42 +08:00
|
|
|
throw std::runtime_error(e.getMessage());
|
2024-08-01 17:35:06 +08:00
|
|
|
} catch (const std::exception &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to setup topics: " << e.what());
|
2024-08-29 15:17:42 +08:00
|
|
|
throw std::runtime_error(e.what());
|
2024-08-01 17:35:06 +08:00
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Failed to setup topics");
|
2024-08-29 15:17:42 +08:00
|
|
|
throw std::runtime_error("Failed to setup topics");
|
2024-08-01 17:35:06 +08:00
|
|
|
}
|
2024-04-09 16:08:48 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::onTemperatureUpdate(diagnostic_updater::DiagnosticStatusWrapper &status) {
|
|
|
|
|
try {
|
|
|
|
|
OBDeviceTemperature temperature;
|
|
|
|
|
uint32_t data_size = sizeof(OBDeviceTemperature);
|
|
|
|
|
device_->getStructuredData(OB_STRUCT_DEVICE_TEMPERATURE, &temperature, &data_size);
|
|
|
|
|
status.add("CPU Temperature", temperature.cpuTemp);
|
|
|
|
|
status.add("IR Temperature", temperature.irTemp);
|
|
|
|
|
status.add("LDM Temperature", temperature.ldmTemp);
|
|
|
|
|
status.add("MainBoard Temperature", temperature.mainBoardTemp);
|
|
|
|
|
status.add("TEC Temperature", temperature.tecTemp);
|
|
|
|
|
status.add("IMU Temperature", temperature.imuTemp);
|
|
|
|
|
status.add("RGB Temperature", temperature.rgbTemp);
|
|
|
|
|
status.add("Left IR Temperature", temperature.irLeftTemp);
|
|
|
|
|
status.add("Right IR Temperature", temperature.irRightTemp);
|
|
|
|
|
status.add("Chip Top Temperature", temperature.chipTopTemp);
|
|
|
|
|
status.add("Chip Bottom Temperature", temperature.chipBottomTemp);
|
|
|
|
|
status.summary(diagnostic_msgs::msg::DiagnosticStatus::OK, "Temperature is normal");
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
status.summary(diagnostic_msgs::msg::DiagnosticStatus::ERROR, e.getMessage());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::setupDiagnosticUpdater() {
|
2026-02-10 14:56:22 +08:00
|
|
|
if (diagnostic_period_ <= 0.0 || isOpenNIDevice(device_->getDeviceInfo()->pid())) {
|
2024-04-09 16:08:48 +08:00
|
|
|
return;
|
|
|
|
|
}
|
2024-05-22 10:21:51 +08:00
|
|
|
try {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Publish diagnostics every " << diagnostic_period_ << " seconds");
|
|
|
|
|
auto info = device_->getDeviceInfo();
|
|
|
|
|
std::string serial_number = info->serialNumber();
|
|
|
|
|
diagnostic_updater_ = std::make_unique<diagnostic_updater::Updater>(node_, diagnostic_period_);
|
|
|
|
|
diagnostic_updater_->setHardwareID(serial_number);
|
|
|
|
|
diagnostic_updater_->add("Temperatures", this, &OBCameraNode::onTemperatureUpdate);
|
2024-05-17 11:39:11 +08:00
|
|
|
} catch (const std::exception &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to setup diagnostic updater: " << e.what());
|
|
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
2023-02-13 15:34:44 +08:00
|
|
|
void OBCameraNode::setupPipelineConfig() {
|
|
|
|
|
if (pipeline_config_) {
|
|
|
|
|
pipeline_config_.reset();
|
|
|
|
|
}
|
|
|
|
|
pipeline_config_ = std::make_shared<ob::Config>();
|
2024-05-14 16:51:21 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
CHECK_NOTNULL(device_info.get());
|
|
|
|
|
auto pid = device_info->pid();
|
|
|
|
|
if (depth_registration_ && enable_stream_[COLOR] && enable_stream_[DEPTH] &&
|
|
|
|
|
!isGemini335PID(pid)) {
|
2024-04-13 16:45:02 +08:00
|
|
|
OBAlignMode align_mode = align_mode_ == "HW" ? ALIGN_D2C_HW_MODE : ALIGN_D2C_SW_MODE;
|
2024-05-06 10:34:19 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "set align mode to " << magic_enum::enum_name(align_mode));
|
2024-04-13 16:45:02 +08:00
|
|
|
pipeline_config_->setAlignMode(align_mode);
|
2024-05-30 10:15:38 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "enable depth scale " << (enable_depth_scale_ ? "ON" : "OFF"));
|
|
|
|
|
pipeline_config_->setDepthScaleRequire(enable_depth_scale_);
|
2023-02-13 15:34:44 +08:00
|
|
|
}
|
2023-08-27 14:06:25 +08:00
|
|
|
for (const auto &stream_index : IMAGE_STREAMS) {
|
2023-02-13 15:34:44 +08:00
|
|
|
if (enable_stream_[stream_index]) {
|
2023-02-16 16:14:38 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Enable " << stream_name_[stream_index] << " stream");
|
2024-05-09 11:34:03 +08:00
|
|
|
auto profile = stream_profile_[stream_index]->as<ob::VideoStreamProfile>();
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_,
|
|
|
|
|
"Stream " << stream_name_[stream_index] << " width: " << profile->width()
|
|
|
|
|
<< " height: " << profile->height() << " fps: " << profile->fps()
|
|
|
|
|
<< " format: " << profile->format());
|
2023-02-13 15:34:44 +08:00
|
|
|
pipeline_config_->enableStream(stream_profile_[stream_index]);
|
|
|
|
|
}
|
|
|
|
|
}
|
2025-01-13 18:02:22 +08:00
|
|
|
|
|
|
|
|
if (frame_aggregate_mode_ == "full_frame") {
|
|
|
|
|
pipeline_config_->setFrameAggregateOutputMode(OB_FRAME_AGGREGATE_OUTPUT_FULL_FRAME_REQUIRE);
|
|
|
|
|
} else if (frame_aggregate_mode_ == "color_frame") {
|
|
|
|
|
pipeline_config_->setFrameAggregateOutputMode(OB_FRAME_AGGREGATE_OUTPUT_COLOR_FRAME_REQUIRE);
|
|
|
|
|
} else if (frame_aggregate_mode_ == "disable") {
|
|
|
|
|
pipeline_config_->setFrameAggregateOutputMode(OB_FRAME_AGGREGATE_OUTPUT_DISABLE);
|
|
|
|
|
} else {
|
|
|
|
|
pipeline_config_->setFrameAggregateOutputMode(OB_FRAME_AGGREGATE_OUTPUT_ANY_SITUATION);
|
|
|
|
|
}
|
2023-02-13 15:34:44 +08:00
|
|
|
}
|
|
|
|
|
|
2024-11-07 11:43:40 +08:00
|
|
|
void OBCameraNode::setupCameraInfo() {
|
|
|
|
|
std::string color_camera_name = camera_name_ + "_color";
|
|
|
|
|
if (!color_info_url_.empty()) {
|
|
|
|
|
color_info_manager_ = std::make_unique<camera_info_manager::CameraInfoManager>(
|
|
|
|
|
node_, color_camera_name, color_info_url_);
|
|
|
|
|
}
|
|
|
|
|
std::string ir_camera_name = camera_name_ + "_ir";
|
|
|
|
|
if (!ir_info_url_.empty()) {
|
|
|
|
|
ir_info_manager_ = std::make_unique<camera_info_manager::CameraInfoManager>(
|
|
|
|
|
node_, ir_camera_name, ir_info_url_);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
void OBCameraNode::setupPublishers() {
|
|
|
|
|
using PointCloud2 = sensor_msgs::msg::PointCloud2;
|
|
|
|
|
using CameraInfo = sensor_msgs::msg::CameraInfo;
|
2023-02-06 17:26:10 +08:00
|
|
|
auto point_cloud_qos_profile = getRMWQosProfileFromString(point_cloud_qos_);
|
2024-08-27 17:58:12 +08:00
|
|
|
if (use_intra_process_) {
|
|
|
|
|
point_cloud_qos_profile = rmw_qos_profile_default;
|
|
|
|
|
}
|
2023-02-06 17:26:10 +08:00
|
|
|
if (enable_colored_point_cloud_) {
|
2023-04-28 09:44:16 +08:00
|
|
|
depth_registration_cloud_pub_ = node_->create_publisher<PointCloud2>(
|
2023-09-07 14:14:34 +08:00
|
|
|
"depth_registered/points",
|
2023-02-06 17:26:10 +08:00
|
|
|
rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(point_cloud_qos_profile),
|
|
|
|
|
point_cloud_qos_profile));
|
|
|
|
|
}
|
|
|
|
|
if (enable_point_cloud_) {
|
2023-04-28 09:44:16 +08:00
|
|
|
depth_cloud_pub_ = node_->create_publisher<PointCloud2>(
|
2023-02-06 17:26:10 +08:00
|
|
|
"depth/points", rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(point_cloud_qos_profile),
|
|
|
|
|
point_cloud_qos_profile));
|
|
|
|
|
}
|
2024-05-07 11:08:38 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
CHECK_NOTNULL(device_info.get());
|
|
|
|
|
auto pid = device_info->pid();
|
2023-08-27 14:06:25 +08:00
|
|
|
for (const auto &stream_index : IMAGE_STREAMS) {
|
2023-02-17 14:43:55 +08:00
|
|
|
if (!enable_stream_[stream_index]) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2023-02-06 17:26:10 +08:00
|
|
|
std::string name = stream_name_[stream_index];
|
2022-06-06 19:00:12 +08:00
|
|
|
std::string topic = name + "/image_raw";
|
2023-02-06 17:26:10 +08:00
|
|
|
auto image_qos = image_qos_[stream_index];
|
|
|
|
|
auto image_qos_profile = getRMWQosProfileFromString(image_qos);
|
2024-08-27 17:58:12 +08:00
|
|
|
if (use_intra_process_) {
|
|
|
|
|
image_qos_profile = rmw_qos_profile_default;
|
|
|
|
|
}
|
2026-04-01 11:01:49 +08:00
|
|
|
const bool is_mjpg_color_stream =
|
|
|
|
|
stream_index == COLOR && format_[stream_index] == OB_FORMAT_MJPG;
|
|
|
|
|
if (use_intra_process_ || is_mjpg_color_stream) {
|
2024-08-27 17:58:12 +08:00
|
|
|
image_publishers_[stream_index] =
|
|
|
|
|
std::make_shared<image_rcl_publisher>(*node_, topic, image_qos_profile);
|
|
|
|
|
} else {
|
|
|
|
|
image_publishers_[stream_index] =
|
|
|
|
|
std::make_shared<image_transport_publisher>(*node_, topic, image_qos_profile);
|
|
|
|
|
}
|
2026-05-27 14:48:47 +08:00
|
|
|
if (is_mjpg_color_stream) {
|
2026-04-01 11:01:49 +08:00
|
|
|
compressed_image_publishers_[stream_index] =
|
|
|
|
|
node_->create_publisher<sensor_msgs::msg::CompressedImage>(
|
|
|
|
|
topic + "/compressed",
|
|
|
|
|
rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(image_qos_profile),
|
|
|
|
|
image_qos_profile));
|
|
|
|
|
}
|
2024-08-27 17:58:12 +08:00
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
topic = name + "/camera_info";
|
2023-02-06 17:26:10 +08:00
|
|
|
auto camera_info_qos = camera_info_qos_[stream_index];
|
|
|
|
|
auto camera_info_qos_profile = getRMWQosProfileFromString(camera_info_qos);
|
2024-08-27 17:58:12 +08:00
|
|
|
if (use_intra_process_) {
|
|
|
|
|
camera_info_qos_profile = rmw_qos_profile_default;
|
|
|
|
|
}
|
2023-02-06 17:26:10 +08:00
|
|
|
camera_info_publishers_[stream_index] = node_->create_publisher<CameraInfo>(
|
|
|
|
|
topic, rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(camera_info_qos_profile),
|
|
|
|
|
camera_info_qos_profile));
|
2024-05-07 11:08:38 +08:00
|
|
|
if (isGemini335PID(pid)) {
|
|
|
|
|
metadata_publishers_[stream_index] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::Metadata>(
|
|
|
|
|
name + "/metadata",
|
|
|
|
|
rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(camera_info_qos_profile),
|
|
|
|
|
camera_info_qos_profile));
|
|
|
|
|
}
|
2024-07-31 14:30:54 +08:00
|
|
|
if (stream_index == COLOR && enable_color_undistortion_) {
|
2024-08-27 17:58:12 +08:00
|
|
|
if (use_intra_process_) {
|
|
|
|
|
color_undistortion_publisher_ = std::make_shared<image_rcl_publisher>(
|
|
|
|
|
*node_, "color/image_undistorted", image_qos_profile);
|
|
|
|
|
} else {
|
|
|
|
|
color_undistortion_publisher_ = std::make_shared<image_transport_publisher>(
|
|
|
|
|
*node_, "color/image_undistorted", image_qos_profile);
|
|
|
|
|
}
|
2024-07-31 14:30:54 +08:00
|
|
|
}
|
2023-02-06 17:26:10 +08:00
|
|
|
}
|
2023-12-20 17:56:37 +08:00
|
|
|
|
|
|
|
|
if (enable_sync_output_accel_gyro_) {
|
2024-03-15 10:24:38 +08:00
|
|
|
std::string topic_name = stream_name_[GYRO] + "_" + stream_name_[ACCEL] + "/sample";
|
2023-12-20 17:56:37 +08:00
|
|
|
auto data_qos = getRMWQosProfileFromString(imu_qos_[GYRO]);
|
2024-08-27 17:58:12 +08:00
|
|
|
if (use_intra_process_) {
|
|
|
|
|
data_qos = rmw_qos_profile_default;
|
|
|
|
|
}
|
2023-12-20 17:56:37 +08:00
|
|
|
imu_gyro_accel_publisher_ = node_->create_publisher<sensor_msgs::msg::Imu>(
|
2024-03-15 10:24:38 +08:00
|
|
|
topic_name, rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(data_qos), data_qos));
|
|
|
|
|
topic_name = stream_name_[GYRO] + "/imu_info";
|
|
|
|
|
imu_info_publishers_[GYRO] = node_->create_publisher<orbbec_camera_msgs::msg::IMUInfo>(
|
|
|
|
|
topic_name, rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(data_qos), data_qos));
|
|
|
|
|
topic_name = stream_name_[ACCEL] + "/imu_info";
|
|
|
|
|
imu_info_publishers_[ACCEL] = node_->create_publisher<orbbec_camera_msgs::msg::IMUInfo>(
|
|
|
|
|
topic_name, rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(data_qos), data_qos));
|
2023-12-20 17:56:37 +08:00
|
|
|
} else {
|
|
|
|
|
for (const auto &stream_index : HID_STREAMS) {
|
|
|
|
|
if (!enable_stream_[stream_index]) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
std::string data_topic_name = stream_name_[stream_index] + "/sample";
|
|
|
|
|
auto data_qos = getRMWQosProfileFromString(imu_qos_[stream_index]);
|
2024-08-27 17:58:12 +08:00
|
|
|
if (use_intra_process_) {
|
|
|
|
|
data_qos = rmw_qos_profile_default;
|
|
|
|
|
}
|
2023-12-20 17:56:37 +08:00
|
|
|
imu_publishers_[stream_index] = node_->create_publisher<sensor_msgs::msg::Imu>(
|
|
|
|
|
data_topic_name, rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(data_qos), data_qos));
|
2024-03-15 10:24:38 +08:00
|
|
|
data_topic_name = stream_name_[stream_index] + "/imu_info";
|
|
|
|
|
imu_info_publishers_[stream_index] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::IMUInfo>(
|
|
|
|
|
data_topic_name,
|
|
|
|
|
rclcpp::QoS(rclcpp::QoSInitialization::from_rmw(data_qos), data_qos));
|
2023-12-20 17:56:37 +08:00
|
|
|
}
|
2023-07-06 15:58:53 +08:00
|
|
|
}
|
2024-08-27 17:58:12 +08:00
|
|
|
|
|
|
|
|
auto extrinsics_qos = rclcpp::QoS(1).transient_local();
|
|
|
|
|
if (use_intra_process_) {
|
|
|
|
|
extrinsics_qos = rclcpp::QoS(1);
|
|
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[INFRA0]) {
|
|
|
|
|
depth_to_other_extrinsics_publishers_[INFRA0] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::Extrinsics>(
|
2024-08-27 17:58:12 +08:00
|
|
|
"/" + camera_name_ + "/depth_to_ir", extrinsics_qos);
|
2024-03-15 10:24:38 +08:00
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[COLOR]) {
|
|
|
|
|
depth_to_other_extrinsics_publishers_[COLOR] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::Extrinsics>(
|
2024-08-27 17:58:12 +08:00
|
|
|
"/" + camera_name_ + "/depth_to_color", extrinsics_qos);
|
2024-03-15 10:24:38 +08:00
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[INFRA1]) {
|
|
|
|
|
depth_to_other_extrinsics_publishers_[INFRA1] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::Extrinsics>(
|
2024-08-27 17:58:12 +08:00
|
|
|
"/" + camera_name_ + "/depth_to_left_ir", extrinsics_qos);
|
2024-03-15 10:24:38 +08:00
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[INFRA2]) {
|
|
|
|
|
depth_to_other_extrinsics_publishers_[INFRA2] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::Extrinsics>(
|
2024-08-27 17:58:12 +08:00
|
|
|
"/" + camera_name_ + "/depth_to_right_ir", extrinsics_qos);
|
2024-03-15 10:24:38 +08:00
|
|
|
}
|
2024-04-28 22:57:54 +08:00
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[ACCEL]) {
|
|
|
|
|
depth_to_other_extrinsics_publishers_[ACCEL] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::Extrinsics>(
|
2024-08-27 17:58:12 +08:00
|
|
|
"/" + camera_name_ + "/depth_to_accel", extrinsics_qos);
|
2024-04-28 22:57:54 +08:00
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[GYRO]) {
|
|
|
|
|
depth_to_other_extrinsics_publishers_[GYRO] =
|
|
|
|
|
node_->create_publisher<orbbec_camera_msgs::msg::Extrinsics>(
|
2024-08-27 17:58:12 +08:00
|
|
|
"/" + camera_name_ + "/depth_to_gyro", extrinsics_qos);
|
2024-04-28 22:57:54 +08:00
|
|
|
}
|
2024-08-27 17:58:12 +08:00
|
|
|
filter_status_pub_ =
|
|
|
|
|
node_->create_publisher<std_msgs::msg::String>("depth_filter_status", extrinsics_qos);
|
2024-04-07 15:30:59 +08:00
|
|
|
std_msgs::msg::String msg;
|
|
|
|
|
msg.data = filter_status_.dump(2);
|
|
|
|
|
filter_status_pub_->publish(msg);
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
2023-12-28 11:02:20 +08:00
|
|
|
void OBCameraNode::publishPointCloud(const std::shared_ptr<ob::FrameSet> &frame_set) {
|
2022-06-09 17:24:59 +08:00
|
|
|
try {
|
2023-12-28 11:02:20 +08:00
|
|
|
if (depth_registration_ || enable_colored_point_cloud_) {
|
|
|
|
|
if (frame_set->depthFrame() != nullptr && frame_set->colorFrame() != nullptr) {
|
|
|
|
|
publishColoredPointCloud(frame_set);
|
2022-06-09 17:47:12 +08:00
|
|
|
}
|
2022-06-24 18:55:31 +08:00
|
|
|
}
|
2023-11-02 20:01:48 +08:00
|
|
|
|
2023-12-28 11:02:20 +08:00
|
|
|
if (enable_point_cloud_ && frame_set->depthFrame() != nullptr) {
|
|
|
|
|
publishDepthPointCloud(frame_set);
|
2022-06-09 17:24:59 +08:00
|
|
|
}
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const ob::Error &e) {
|
2022-06-09 17:24:59 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, e.getMessage());
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const std::exception &e) {
|
2022-06-09 17:24:59 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, e.what());
|
|
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "publishPointCloud with unknown error");
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::publishDepthPointCloud(const std::shared_ptr<ob::FrameSet> &frame_set) {
|
2024-04-07 17:29:28 +08:00
|
|
|
(void)frame_set;
|
2024-06-21 19:56:57 +08:00
|
|
|
if (!depth_cloud_pub_ || depth_cloud_pub_->get_subscription_count() == 0 ||
|
|
|
|
|
!enable_point_cloud_ || !depth_frame_) {
|
2023-09-07 16:09:35 +08:00
|
|
|
return;
|
|
|
|
|
}
|
2024-05-09 17:07:43 +08:00
|
|
|
std::lock_guard<decltype(point_cloud_mutex_)> point_cloud_msg_lock(point_cloud_mutex_);
|
2024-09-23 15:01:34 +08:00
|
|
|
if (!depth_frame_) {
|
2024-09-18 18:50:48 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, "depth frame is null");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-04-07 17:29:28 +08:00
|
|
|
auto depth_frame = depth_frame_->as<ob::DepthFrame>();
|
2023-03-24 19:09:01 +08:00
|
|
|
if (!depth_frame) {
|
2024-04-07 17:29:28 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, "depth frame is null");
|
2023-03-24 19:09:01 +08:00
|
|
|
return;
|
|
|
|
|
}
|
2024-06-21 19:56:57 +08:00
|
|
|
CHECK_NOTNULL(pipeline_);
|
|
|
|
|
auto camera_params = pipeline_->getCameraParam();
|
2024-09-09 14:25:21 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
CHECK_NOTNULL(device_info.get());
|
|
|
|
|
auto pid = device_info->pid();
|
|
|
|
|
if (depth_registration_ || pid == DABAI_MAX_PID) {
|
2024-06-22 15:07:54 +08:00
|
|
|
camera_params.depthIntrinsic = camera_params.rgbIntrinsic;
|
|
|
|
|
}
|
2024-06-21 19:56:57 +08:00
|
|
|
depth_point_cloud_filter_.setCameraParam(camera_params);
|
|
|
|
|
float depth_scale = depth_frame->getValueScale();
|
|
|
|
|
depth_point_cloud_filter_.setPositionDataScaled(depth_scale);
|
|
|
|
|
depth_point_cloud_filter_.setCreatePointFormat(OB_FORMAT_POINT);
|
|
|
|
|
auto result_frame = depth_point_cloud_filter_.process(depth_frame);
|
|
|
|
|
if (!result_frame) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to process depth frame");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
auto point_size = result_frame->dataSize() / sizeof(OBPoint);
|
|
|
|
|
auto *points = static_cast<OBPoint *>(result_frame->data());
|
2023-09-07 16:09:35 +08:00
|
|
|
auto width = depth_frame->width();
|
|
|
|
|
auto height = depth_frame->height();
|
2024-05-16 14:34:55 +08:00
|
|
|
auto point_cloud_msg = std::make_unique<sensor_msgs::msg::PointCloud2>();
|
|
|
|
|
sensor_msgs::PointCloud2Modifier modifier(*point_cloud_msg);
|
2022-06-06 19:00:12 +08:00
|
|
|
modifier.setPointCloud2FieldsByString(1, "xyz");
|
2023-09-07 16:09:35 +08:00
|
|
|
modifier.resize(width * height);
|
2024-05-16 14:34:55 +08:00
|
|
|
point_cloud_msg->width = depth_frame->width();
|
|
|
|
|
point_cloud_msg->height = depth_frame->height();
|
|
|
|
|
point_cloud_msg->row_step = point_cloud_msg->width * point_cloud_msg->point_step;
|
|
|
|
|
point_cloud_msg->data.resize(point_cloud_msg->height * point_cloud_msg->row_step);
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<float> iter_x(*point_cloud_msg, "x");
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<float> iter_y(*point_cloud_msg, "y");
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<float> iter_z(*point_cloud_msg, "z");
|
2024-06-21 19:56:57 +08:00
|
|
|
const static float MIN_DISTANCE = 20.0; // 2cm
|
|
|
|
|
const static float MAX_DISTANCE = 10000.0; // 10m
|
2023-09-07 16:09:35 +08:00
|
|
|
const static float min_depth = MIN_DISTANCE / depth_scale;
|
|
|
|
|
const static float max_depth = MAX_DISTANCE / depth_scale;
|
2024-06-21 19:56:57 +08:00
|
|
|
size_t valid_count = 0;
|
|
|
|
|
for (size_t i = 0; i < point_size; i++) {
|
|
|
|
|
bool valid_point = points[i].z >= min_depth && points[i].z <= max_depth;
|
|
|
|
|
if (valid_point || ordered_pc_) {
|
|
|
|
|
*iter_x = static_cast<float>(points[i].x / 1000.0);
|
|
|
|
|
*iter_y = static_cast<float>(points[i].y / 1000.0);
|
|
|
|
|
*iter_z = static_cast<float>(points[i].z / 1000.0);
|
|
|
|
|
++iter_x, ++iter_y, ++iter_z;
|
|
|
|
|
valid_count++;
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
}
|
2024-05-09 17:07:43 +08:00
|
|
|
if (valid_count == 0) {
|
|
|
|
|
RCLCPP_WARN(logger_, "No valid point in point cloud");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-01-29 09:18:21 +08:00
|
|
|
if (!ordered_pc_) {
|
2024-05-16 14:34:55 +08:00
|
|
|
point_cloud_msg->is_dense = true;
|
|
|
|
|
point_cloud_msg->width = valid_count;
|
|
|
|
|
point_cloud_msg->height = 1;
|
2023-11-15 11:05:38 +08:00
|
|
|
modifier.resize(valid_count);
|
|
|
|
|
}
|
2024-07-17 16:56:52 +08:00
|
|
|
auto frame_timestamp = getFrameTimestampUs(depth_frame);
|
|
|
|
|
auto timestamp = fromUsToROSTime(frame_timestamp);
|
2024-04-07 17:29:28 +08:00
|
|
|
std::string frame_id = depth_registration_ ? optical_frame_id_[COLOR] : optical_frame_id_[DEPTH];
|
2024-09-13 10:18:09 +08:00
|
|
|
if (!cloud_frame_id_.empty()) {
|
|
|
|
|
frame_id = cloud_frame_id_;
|
|
|
|
|
}
|
2024-05-16 14:34:55 +08:00
|
|
|
point_cloud_msg->header.stamp = timestamp;
|
|
|
|
|
point_cloud_msg->header.frame_id = frame_id;
|
2023-02-20 14:46:02 +08:00
|
|
|
if (save_point_cloud_) {
|
|
|
|
|
save_point_cloud_ = false;
|
|
|
|
|
auto now = std::time(nullptr);
|
|
|
|
|
std::stringstream ss;
|
|
|
|
|
ss << std::put_time(std::localtime(&now), "%Y%m%d_%H%M%S");
|
|
|
|
|
auto current_path = std::filesystem::current_path().string();
|
|
|
|
|
std::string filename = current_path + "/point_cloud/points_" + ss.str() + ".ply";
|
|
|
|
|
if (!std::filesystem::exists(current_path + "/point_cloud")) {
|
|
|
|
|
std::filesystem::create_directory(current_path + "/point_cloud");
|
|
|
|
|
}
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Saving point cloud to " << filename);
|
2024-04-13 15:21:27 +08:00
|
|
|
try {
|
2024-05-16 14:34:55 +08:00
|
|
|
saveDepthPointsToPly(point_cloud_msg, filename);
|
2024-04-13 15:21:27 +08:00
|
|
|
} catch (const std::exception &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to save point cloud: " << e.what());
|
|
|
|
|
}
|
2023-02-20 14:46:02 +08:00
|
|
|
}
|
2024-06-04 13:53:02 +08:00
|
|
|
depth_cloud_pub_->publish(std::move(point_cloud_msg));
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::publishColoredPointCloud(const std::shared_ptr<ob::FrameSet> &frame_set) {
|
2024-06-21 19:56:57 +08:00
|
|
|
if (!depth_registration_cloud_pub_ ||
|
|
|
|
|
depth_registration_cloud_pub_->get_subscription_count() == 0 ||
|
|
|
|
|
!enable_colored_point_cloud_ || !depth_frame_) {
|
2024-04-07 17:29:28 +08:00
|
|
|
return;
|
|
|
|
|
}
|
2024-06-21 19:56:57 +08:00
|
|
|
|
2024-04-07 17:29:28 +08:00
|
|
|
CHECK_NOTNULL(depth_frame_.get());
|
2024-05-09 17:07:43 +08:00
|
|
|
std::lock_guard<decltype(point_cloud_mutex_)> point_cloud_msg_lock(point_cloud_mutex_);
|
2024-09-18 18:50:48 +08:00
|
|
|
if (!depth_frame_) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "depth frame is null");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-04-07 17:29:28 +08:00
|
|
|
auto depth_frame = depth_frame_->as<ob::DepthFrame>();
|
2022-06-07 20:59:51 +08:00
|
|
|
auto color_frame = frame_set->colorFrame();
|
2023-03-24 19:09:01 +08:00
|
|
|
if (!depth_frame || !color_frame) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
2023-09-07 16:09:35 +08:00
|
|
|
auto depth_width = depth_frame->width();
|
|
|
|
|
auto depth_height = depth_frame->height();
|
|
|
|
|
auto color_width = color_frame->width();
|
|
|
|
|
auto color_height = color_frame->height();
|
|
|
|
|
if (depth_width != color_width || depth_height != color_height) {
|
2024-05-14 16:51:21 +08:00
|
|
|
RCLCPP_DEBUG(logger_, "Depth (%d x %d) and color (%d x %d) frame size mismatch", depth_width,
|
2024-04-07 17:29:28 +08:00
|
|
|
depth_height, color_width, color_height);
|
2023-09-07 16:09:35 +08:00
|
|
|
return;
|
|
|
|
|
}
|
2024-06-21 19:56:57 +08:00
|
|
|
if (!xy_tables_.has_value()) {
|
|
|
|
|
calibration_param_ = pipeline_->getCalibrationParam(pipeline_config_);
|
|
|
|
|
|
|
|
|
|
uint32_t table_size =
|
|
|
|
|
color_width * color_height * 2; // one for x-coordinate and one for y-coordinate LUT
|
|
|
|
|
if (xy_table_data_size_ != table_size) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Init xy tables with size " << table_size);
|
|
|
|
|
xy_table_data_size_ = table_size;
|
|
|
|
|
delete[] xy_table_data_;
|
|
|
|
|
xy_table_data_ = new float[table_size];
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
xy_tables_ = OBXYTables();
|
|
|
|
|
CHECK_NOTNULL(xy_table_data_);
|
|
|
|
|
if (!ob::CoordinateTransformHelper::transformationInitXYTables(
|
|
|
|
|
*calibration_param_, OB_SENSOR_COLOR, xy_table_data_, &table_size, &(*xy_tables_))) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to init xy tables");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const auto *depth_data = (uint8_t *)depth_frame->data();
|
2023-09-07 16:09:35 +08:00
|
|
|
const auto *color_data = (uint8_t *)(rgb_buffer_);
|
2024-06-22 15:07:54 +08:00
|
|
|
CHECK_NOTNULL(rgb_point_cloud_buffer_);
|
2024-06-21 19:56:57 +08:00
|
|
|
uint32_t point_cloud_buffer_size = color_width * color_height * sizeof(OBColorPoint);
|
2024-06-22 15:07:54 +08:00
|
|
|
if (point_cloud_buffer_size > rgb_point_cloud_buffer_size_) {
|
|
|
|
|
delete[] rgb_point_cloud_buffer_;
|
|
|
|
|
rgb_point_cloud_buffer_ = new uint8_t[point_cloud_buffer_size];
|
|
|
|
|
rgb_point_cloud_buffer_size_ = point_cloud_buffer_size;
|
2024-06-21 19:56:57 +08:00
|
|
|
}
|
2024-06-22 15:07:54 +08:00
|
|
|
memset(rgb_point_cloud_buffer_, 0, rgb_point_cloud_buffer_size_);
|
|
|
|
|
auto *point_cloud = (OBColorPoint *)rgb_point_cloud_buffer_;
|
2024-06-21 19:56:57 +08:00
|
|
|
ob::CoordinateTransformHelper::transformationDepthToRGBDPointCloud(&(*xy_tables_), depth_data,
|
|
|
|
|
color_data, point_cloud);
|
2024-05-16 14:34:55 +08:00
|
|
|
auto point_cloud_msg = std::make_unique<sensor_msgs::msg::PointCloud2>();
|
|
|
|
|
sensor_msgs::PointCloud2Modifier modifier(*point_cloud_msg);
|
2022-06-30 11:39:33 +08:00
|
|
|
modifier.setPointCloud2FieldsByString(1, "xyz");
|
2024-05-16 14:34:55 +08:00
|
|
|
point_cloud_msg->width = color_frame->width();
|
|
|
|
|
point_cloud_msg->height = color_frame->height();
|
2022-06-07 11:58:59 +08:00
|
|
|
std::string format_str = "rgb";
|
2024-05-16 14:34:55 +08:00
|
|
|
point_cloud_msg->point_step =
|
|
|
|
|
addPointField(*point_cloud_msg, format_str, 1, sensor_msgs::msg::PointField::FLOAT32,
|
|
|
|
|
static_cast<int>(point_cloud_msg->point_step));
|
|
|
|
|
point_cloud_msg->row_step = point_cloud_msg->width * point_cloud_msg->point_step;
|
|
|
|
|
point_cloud_msg->data.resize(point_cloud_msg->height * point_cloud_msg->row_step);
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<float> iter_x(*point_cloud_msg, "x");
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<float> iter_y(*point_cloud_msg, "y");
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<float> iter_z(*point_cloud_msg, "z");
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<uint8_t> iter_r(*point_cloud_msg, "r");
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<uint8_t> iter_g(*point_cloud_msg, "g");
|
|
|
|
|
sensor_msgs::PointCloud2Iterator<uint8_t> iter_b(*point_cloud_msg, "b");
|
2022-06-06 19:00:12 +08:00
|
|
|
size_t valid_count = 0;
|
2023-09-07 16:09:35 +08:00
|
|
|
static const float MIN_DISTANCE = 20.0;
|
|
|
|
|
static const float MAX_DISTANCE = 10000.0;
|
|
|
|
|
double depth_scale = depth_frame->getValueScale();
|
|
|
|
|
static float min_depth = MIN_DISTANCE / depth_scale;
|
|
|
|
|
static float max_depth = MAX_DISTANCE / depth_scale;
|
2024-06-21 19:56:57 +08:00
|
|
|
for (size_t i = 0; i < color_width * color_height; i++) {
|
|
|
|
|
bool valid_point = point_cloud[i].z >= min_depth && point_cloud[i].z <= max_depth;
|
|
|
|
|
if (valid_point || ordered_pc_) {
|
|
|
|
|
*iter_x = static_cast<float>(point_cloud[i].x / 1000.0);
|
|
|
|
|
*iter_y = static_cast<float>(point_cloud[i].y / 1000.0);
|
|
|
|
|
*iter_z = static_cast<float>(point_cloud[i].z / 1000.0);
|
|
|
|
|
*iter_r = static_cast<uint8_t>(point_cloud[i].r);
|
|
|
|
|
*iter_g = static_cast<uint8_t>(point_cloud[i].g);
|
|
|
|
|
*iter_b = static_cast<uint8_t>(point_cloud[i].b);
|
|
|
|
|
++iter_x, ++iter_y, ++iter_z, ++iter_r, ++iter_g, ++iter_b;
|
|
|
|
|
++valid_count;
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
}
|
2024-05-09 17:07:43 +08:00
|
|
|
if (valid_count == 0) {
|
|
|
|
|
RCLCPP_WARN(logger_, "No valid points in point cloud");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-01-29 09:18:21 +08:00
|
|
|
if (!ordered_pc_) {
|
2024-05-16 14:34:55 +08:00
|
|
|
point_cloud_msg->is_dense = true;
|
|
|
|
|
point_cloud_msg->width = valid_count;
|
|
|
|
|
point_cloud_msg->height = 1;
|
2023-11-15 11:05:38 +08:00
|
|
|
modifier.resize(valid_count);
|
|
|
|
|
}
|
2024-07-17 16:56:52 +08:00
|
|
|
auto frame_timestamp = getFrameTimestampUs(depth_frame);
|
2024-09-13 10:18:09 +08:00
|
|
|
std::string frame_id = optical_frame_id_[COLOR];
|
|
|
|
|
if (!cloud_frame_id_.empty()) {
|
|
|
|
|
frame_id = cloud_frame_id_;
|
|
|
|
|
}
|
2024-07-17 16:56:52 +08:00
|
|
|
auto timestamp = fromUsToROSTime(frame_timestamp);
|
2024-05-16 14:34:55 +08:00
|
|
|
point_cloud_msg->header.stamp = timestamp;
|
2024-09-13 10:18:09 +08:00
|
|
|
point_cloud_msg->header.frame_id = frame_id;
|
2023-02-20 14:46:02 +08:00
|
|
|
if (save_colored_point_cloud_) {
|
|
|
|
|
save_colored_point_cloud_ = false;
|
|
|
|
|
auto now = std::time(nullptr);
|
|
|
|
|
std::stringstream ss;
|
|
|
|
|
ss << std::put_time(std::localtime(&now), "%Y%m%d_%H%M%S");
|
|
|
|
|
auto current_path = std::filesystem::current_path().string();
|
|
|
|
|
std::string filename = current_path + "/point_cloud/colored_points_" + ss.str() + ".ply";
|
|
|
|
|
if (!std::filesystem::exists(current_path + "/point_cloud")) {
|
|
|
|
|
std::filesystem::create_directory(current_path + "/point_cloud");
|
|
|
|
|
}
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Saving point cloud to " << filename);
|
2024-04-13 15:21:27 +08:00
|
|
|
try {
|
2024-05-16 14:34:55 +08:00
|
|
|
saveRGBPointCloudMsgToPly(point_cloud_msg, filename);
|
2024-04-13 15:21:27 +08:00
|
|
|
} catch (const std::exception &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to save point cloud: " << e.what());
|
|
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Failed to save point cloud");
|
|
|
|
|
}
|
2023-02-20 14:46:02 +08:00
|
|
|
}
|
2024-06-04 13:53:02 +08:00
|
|
|
depth_registration_cloud_pub_->publish(std::move(point_cloud_msg));
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
2022-06-13 14:58:47 +08:00
|
|
|
|
2024-04-07 17:29:28 +08:00
|
|
|
std::shared_ptr<ob::Frame> OBCameraNode::processDepthFrameFilter(
|
|
|
|
|
std::shared_ptr<ob::Frame> &frame) {
|
|
|
|
|
if (frame == nullptr || frame->type() != OB_FRAME_DEPTH) {
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
auto sensor = device_->getSensor(OB_SENSOR_DEPTH);
|
|
|
|
|
CHECK_NOTNULL(sensor.get());
|
|
|
|
|
auto filter_list = sensor->getRecommendedFilters();
|
|
|
|
|
for (size_t i = 0; i < filter_list->count(); i++) {
|
|
|
|
|
auto filter = filter_list->getFilter(i);
|
|
|
|
|
CHECK_NOTNULL(filter.get());
|
2024-05-14 14:36:22 +08:00
|
|
|
if (filter->isEnabled() && frame != nullptr) {
|
2024-04-07 17:29:28 +08:00
|
|
|
frame = filter->process(frame);
|
|
|
|
|
if (frame == nullptr) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Depth filter process failed");
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return frame;
|
|
|
|
|
}
|
|
|
|
|
|
2024-07-17 16:56:52 +08:00
|
|
|
uint64_t OBCameraNode::getFrameTimestampUs(const std::shared_ptr<ob::Frame> &frame) {
|
|
|
|
|
if (frame == nullptr) {
|
|
|
|
|
RCLCPP_WARN(logger_, "getFrameTimestampUs: frame is nullptr, return 0");
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
if (time_domain_ == "device") {
|
|
|
|
|
return frame->timeStampUs();
|
|
|
|
|
} else if (time_domain_ == "global") {
|
|
|
|
|
return frame->globalTimeStampUs();
|
|
|
|
|
} else {
|
|
|
|
|
return frame->systemTimeStampUs();
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-05-06 10:34:19 +08:00
|
|
|
void OBCameraNode::onNewFrameSetCallback(std::shared_ptr<ob::FrameSet> frame_set) {
|
2023-10-14 14:03:50 +08:00
|
|
|
if (!is_running_.load()) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-05-09 17:07:43 +08:00
|
|
|
if (!is_camera_node_initialized_.load()) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
2022-12-28 16:39:11 +08:00
|
|
|
if (frame_set == nullptr) {
|
|
|
|
|
return;
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
2026-05-22 16:38:02 +08:00
|
|
|
if (frame_timestamp_csv_logger_ && frame_timestamp_csv_logger_->enabled()) {
|
|
|
|
|
const auto frame_set_arrival_system_us = getSystemNowUs();
|
|
|
|
|
const auto frame_set_arrival_steady_us = getSteadyNowUs();
|
|
|
|
|
auto final_color_frame = frame_set->getFrame(OB_FRAME_COLOR);
|
|
|
|
|
auto final_depth_frame = frame_set->getFrame(OB_FRAME_DEPTH);
|
|
|
|
|
const bool track_color = enable_stream_[COLOR] && static_cast<bool>(final_color_frame);
|
|
|
|
|
const bool track_depth = enable_stream_[DEPTH] && static_cast<bool>(final_depth_frame);
|
|
|
|
|
const bool color_publish_expected = track_color;
|
|
|
|
|
const bool depth_publish_expected = track_depth;
|
|
|
|
|
|
|
|
|
|
frame_timestamp_csv_logger_->recordFrameSet(
|
|
|
|
|
final_color_frame, final_depth_frame, frame_set_arrival_system_us,
|
|
|
|
|
frame_set_arrival_steady_us, track_color, track_depth, color_publish_expected,
|
|
|
|
|
depth_publish_expected);
|
|
|
|
|
}
|
|
|
|
|
|
2022-12-28 16:39:11 +08:00
|
|
|
try {
|
2023-07-13 18:17:01 +08:00
|
|
|
if (!tf_published_) {
|
|
|
|
|
publishStaticTransforms();
|
|
|
|
|
tf_published_ = true;
|
|
|
|
|
}
|
2024-04-07 17:29:28 +08:00
|
|
|
depth_frame_ = frame_set->getFrame(OB_FRAME_DEPTH);
|
2024-05-07 11:08:38 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
CHECK_NOTNULL(device_info.get());
|
|
|
|
|
auto pid = device_info->pid();
|
2024-05-11 15:28:19 +08:00
|
|
|
auto color_frame = frame_set->getFrame(OB_FRAME_COLOR);
|
2024-08-23 19:55:35 +08:00
|
|
|
has_first_color_frame_ = has_first_color_frame_ || color_frame;
|
2024-12-23 16:04:46 +08:00
|
|
|
if (isGemini335PID(pid) && depth_frame_) {
|
2024-05-14 16:51:21 +08:00
|
|
|
depth_frame_ = processDepthFrameFilter(depth_frame_);
|
2024-08-23 19:55:35 +08:00
|
|
|
if (depth_registration_ && align_filter_ && depth_frame_ && has_first_color_frame_) {
|
2024-05-07 11:08:38 +08:00
|
|
|
auto new_frame = align_filter_->process(frame_set);
|
|
|
|
|
if (new_frame) {
|
|
|
|
|
auto new_frame_set = new_frame->as<ob::FrameSet>();
|
|
|
|
|
CHECK_NOTNULL(new_frame_set.get());
|
|
|
|
|
depth_frame_ = new_frame_set->getFrame(OB_FRAME_DEPTH);
|
2024-05-14 14:36:22 +08:00
|
|
|
} else {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Failed to align depth frame to color frame");
|
2024-05-07 11:08:38 +08:00
|
|
|
}
|
2024-05-14 16:51:21 +08:00
|
|
|
} else {
|
|
|
|
|
RCLCPP_DEBUG(logger_,
|
|
|
|
|
"Depth registration is disabled or align filter is null or depth frame is "
|
|
|
|
|
"null or color frame is null");
|
2024-05-06 11:09:02 +08:00
|
|
|
}
|
2024-05-06 10:34:19 +08:00
|
|
|
}
|
2024-05-09 17:07:43 +08:00
|
|
|
if (enable_stream_[COLOR] && color_frame) {
|
|
|
|
|
std::unique_lock<std::mutex> lock(color_frame_queue_lock_);
|
2024-12-23 16:04:46 +08:00
|
|
|
// if (color_frame_queue_.size() > 2) {
|
|
|
|
|
// color_frame_queue_.pop();
|
|
|
|
|
// }
|
|
|
|
|
color_frame_queue_.push(frame_set);
|
|
|
|
|
color_frame_queue_cv_.notify_all();
|
2024-01-29 09:18:21 +08:00
|
|
|
} else {
|
2023-12-28 11:02:20 +08:00
|
|
|
publishPointCloud(frame_set);
|
|
|
|
|
}
|
2023-11-02 20:01:48 +08:00
|
|
|
|
2023-09-05 17:59:19 +08:00
|
|
|
for (const auto &stream_index : IMAGE_STREAMS) {
|
|
|
|
|
if (enable_stream_[stream_index]) {
|
|
|
|
|
auto frame_type = STREAM_TYPE_TO_FRAME_TYPE.at(stream_index.first);
|
2023-11-02 20:01:48 +08:00
|
|
|
if (frame_type == OB_FRAME_COLOR) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2023-09-05 17:59:19 +08:00
|
|
|
auto frame = frame_set->getFrame(frame_type);
|
|
|
|
|
if (frame == nullptr) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2024-05-06 14:08:57 +08:00
|
|
|
if (stream_index == DEPTH) {
|
2024-12-23 16:04:46 +08:00
|
|
|
frame = depth_frame_;
|
2024-05-06 14:08:57 +08:00
|
|
|
}
|
2024-12-23 16:13:44 +08:00
|
|
|
onNewFrameCallback(frame, stream_index);
|
2023-09-05 17:59:19 +08:00
|
|
|
}
|
|
|
|
|
}
|
2023-11-02 20:01:48 +08:00
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const ob::Error &e) {
|
2022-12-28 16:39:11 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, "onNewFrameSetCallback error: " << e.getMessage());
|
2023-08-27 14:06:25 +08:00
|
|
|
} catch (const std::exception &e) {
|
2022-12-28 16:39:11 +08:00
|
|
|
RCLCPP_ERROR_STREAM(logger_, "onNewFrameSetCallback error: " << e.what());
|
|
|
|
|
} catch (...) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "onNewFrameSetCallback error: unknown error");
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
2022-12-28 16:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
2023-11-06 14:01:41 +08:00
|
|
|
void OBCameraNode::onNewColorFrameCallback() {
|
2023-11-02 20:01:48 +08:00
|
|
|
while (enable_stream_[COLOR] && rclcpp::ok() && is_running_.load()) {
|
2024-05-09 17:07:43 +08:00
|
|
|
std::unique_lock<std::mutex> lock(color_frame_queue_lock_);
|
|
|
|
|
color_frame_queue_cv_.wait(
|
|
|
|
|
lock, [this]() { return !color_frame_queue_.empty() || !(is_running_.load()); });
|
2023-11-02 20:01:48 +08:00
|
|
|
|
2024-01-29 09:18:21 +08:00
|
|
|
if (!rclcpp::ok() || !is_running_.load()) {
|
2023-11-02 20:01:48 +08:00
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
2024-05-09 17:07:43 +08:00
|
|
|
std::shared_ptr<ob::FrameSet> frameSet = color_frame_queue_.front();
|
2023-11-02 20:01:48 +08:00
|
|
|
is_color_frame_decoded_ = decodeColorFrameToBuffer(frameSet->colorFrame(), rgb_buffer_);
|
2023-12-28 11:02:20 +08:00
|
|
|
publishPointCloud(frameSet);
|
2024-05-09 17:07:43 +08:00
|
|
|
onNewFrameCallback(frameSet->colorFrame(), COLOR);
|
|
|
|
|
color_frame_queue_.pop();
|
2023-11-02 20:01:48 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Color frame thread exit!");
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-25 21:47:13 +08:00
|
|
|
std::shared_ptr<ob::Frame> OBCameraNode::softwareDecodeColorFrame(
|
2023-08-27 14:06:25 +08:00
|
|
|
const std::shared_ptr<ob::Frame> &frame) {
|
2023-08-25 21:47:13 +08:00
|
|
|
if (frame == nullptr) {
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
2023-09-07 20:55:25 +08:00
|
|
|
if (frame->format() == OB_FORMAT_RGB || frame->format() == OB_FORMAT_BGR) {
|
2023-08-25 21:47:13 +08:00
|
|
|
return frame;
|
|
|
|
|
}
|
2024-04-07 17:29:28 +08:00
|
|
|
if (frame->format() == OB_FORMAT_RGB || frame->format() == OB_FORMAT_BGR) {
|
|
|
|
|
return frame;
|
|
|
|
|
}
|
|
|
|
|
if (frame->format() == OB_FORMAT_Y16 || frame->format() == OB_FORMAT_Y8) {
|
|
|
|
|
return frame;
|
|
|
|
|
}
|
2023-08-25 21:47:13 +08:00
|
|
|
if (!setupFormatConvertType(frame->format())) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Unsupported color format: %d", frame->format());
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
auto color_frame = format_convert_filter_.process(frame);
|
|
|
|
|
if (color_frame == nullptr) {
|
|
|
|
|
RCLCPP_ERROR_SKIPFIRST_THROTTLE(logger_, *(node_->get_clock()), 1000,
|
|
|
|
|
"Failed to convert frame to RGB format");
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
return color_frame;
|
|
|
|
|
}
|
|
|
|
|
|
2023-09-07 16:09:35 +08:00
|
|
|
bool OBCameraNode::decodeColorFrameToBuffer(const std::shared_ptr<ob::Frame> &frame,
|
|
|
|
|
uint8_t *buffer) {
|
|
|
|
|
if (frame == nullptr) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2023-10-14 14:03:50 +08:00
|
|
|
if (!rgb_buffer_) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(image_publishers_[COLOR]);
|
|
|
|
|
bool has_subscriber = image_publishers_[COLOR]->get_subscription_count() > 0;
|
2023-09-07 16:09:35 +08:00
|
|
|
if (enable_colored_point_cloud_ && depth_registration_cloud_pub_->get_subscription_count() > 0) {
|
|
|
|
|
has_subscriber = true;
|
|
|
|
|
}
|
|
|
|
|
if (!has_subscriber) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2024-05-07 11:08:38 +08:00
|
|
|
if (metadata_publishers_.count(COLOR) &&
|
|
|
|
|
metadata_publishers_[COLOR]->get_subscription_count() > 0) {
|
|
|
|
|
has_subscriber = true;
|
|
|
|
|
}
|
2024-06-21 19:56:57 +08:00
|
|
|
if (camera_info_publishers_.count(COLOR) &&
|
|
|
|
|
camera_info_publishers_[COLOR]->get_subscription_count() > 0) {
|
2024-04-07 17:29:28 +08:00
|
|
|
has_subscriber = true;
|
|
|
|
|
}
|
2023-09-07 16:09:35 +08:00
|
|
|
bool is_decoded = false;
|
|
|
|
|
if (!frame) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2023-09-07 20:55:25 +08:00
|
|
|
|
2023-09-07 16:09:35 +08:00
|
|
|
#if defined(USE_RK_HW_DECODER) || defined(USE_NV_HW_DECODER)
|
|
|
|
|
if (frame && frame->format() != OB_FORMAT_RGB888) {
|
2023-09-07 16:46:59 +08:00
|
|
|
if (frame->format() == OB_FORMAT_MJPG && jpeg_decoder_) {
|
|
|
|
|
CHECK_NOTNULL(jpeg_decoder_.get());
|
2023-09-07 16:09:35 +08:00
|
|
|
CHECK_NOTNULL(rgb_buffer_);
|
|
|
|
|
auto video_frame = frame->as<ob::ColorFrame>();
|
2023-09-07 16:46:59 +08:00
|
|
|
bool ret = jpeg_decoder_->decode(video_frame, rgb_buffer_);
|
2023-09-07 16:09:35 +08:00
|
|
|
if (!ret) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Decode frame failed");
|
|
|
|
|
is_decoded = false;
|
|
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
is_decoded = true;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
if (!is_decoded) {
|
|
|
|
|
auto video_frame = softwareDecodeColorFrame(frame);
|
|
|
|
|
if (!video_frame) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to convert frame to video frame");
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2024-04-07 17:29:28 +08:00
|
|
|
CHECK_NOTNULL(buffer);
|
2023-09-07 16:09:35 +08:00
|
|
|
memcpy(buffer, video_frame->data(), video_frame->dataSize());
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-29 09:18:21 +08:00
|
|
|
std::shared_ptr<ob::Frame> OBCameraNode::decodeIRMJPGFrame(
|
|
|
|
|
const std::shared_ptr<ob::Frame> &frame) {
|
2024-07-09 13:55:34 +08:00
|
|
|
if (frame == nullptr) {
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
2023-10-25 09:52:44 +08:00
|
|
|
if (frame->format() == OB_FORMAT_MJPEG &&
|
|
|
|
|
(frame->type() == OB_FRAME_IR || frame->type() == OB_FRAME_IR_LEFT ||
|
|
|
|
|
frame->type() == OB_FRAME_IR_RIGHT)) {
|
|
|
|
|
auto video_frame = frame->as<ob::IRFrame>();
|
|
|
|
|
|
|
|
|
|
cv::Mat mjpgMat(1, video_frame->dataSize(), CV_8UC1, video_frame->data());
|
|
|
|
|
cv::Mat irRawMat = cv::imdecode(mjpgMat, cv::IMREAD_GRAYSCALE);
|
|
|
|
|
|
|
|
|
|
std::shared_ptr<ob::Frame> irFrame = ob::FrameHelper::createFrame(
|
|
|
|
|
video_frame->type(), video_frame->format(), video_frame->width(), video_frame->height(), 0);
|
|
|
|
|
|
|
|
|
|
uint32_t buffer_size = irRawMat.rows * irRawMat.cols * irRawMat.channels();
|
|
|
|
|
|
2024-01-29 09:18:21 +08:00
|
|
|
if (buffer_size > irFrame->dataSize()) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Insufficient buffer size allocation,failed to decode ir mjpg frame!");
|
2023-10-25 09:52:44 +08:00
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
memcpy(irFrame->data(), irRawMat.data, buffer_size);
|
|
|
|
|
ob::FrameHelper::setFrameDeviceTimestamp(irFrame, video_frame->timeStamp());
|
|
|
|
|
ob::FrameHelper::setFrameDeviceTimestampUs(irFrame, video_frame->timeStampUs());
|
|
|
|
|
ob::FrameHelper::setFrameSystemTimestamp(irFrame, video_frame->systemTimeStamp());
|
|
|
|
|
return irFrame;
|
|
|
|
|
}
|
|
|
|
|
|
2024-07-09 13:55:34 +08:00
|
|
|
return frame;
|
2023-10-25 09:52:44 +08:00
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::onNewFrameCallback(const std::shared_ptr<ob::Frame> &frame,
|
|
|
|
|
const stream_index_pair &stream_index) {
|
2022-12-28 16:39:11 +08:00
|
|
|
if (frame == nullptr) {
|
|
|
|
|
return;
|
2022-06-07 10:46:04 +08:00
|
|
|
}
|
2026-04-01 11:01:49 +08:00
|
|
|
CHECK_NOTNULL(image_publishers_.at(stream_index));
|
|
|
|
|
const bool has_raw_image_subscriber =
|
|
|
|
|
image_publishers_.at(stream_index)->get_subscription_count() > 0;
|
|
|
|
|
const bool has_compressed_image_subscriber = hasCompressedImageSubscriber(stream_index);
|
|
|
|
|
bool has_subscriber = has_raw_image_subscriber || has_compressed_image_subscriber;
|
2024-03-15 10:24:38 +08:00
|
|
|
has_subscriber =
|
|
|
|
|
has_subscriber || camera_info_publishers_[stream_index]->get_subscription_count() > 0;
|
|
|
|
|
has_subscriber =
|
2024-05-07 11:08:38 +08:00
|
|
|
has_subscriber || (metadata_publishers_.count(stream_index) &&
|
|
|
|
|
metadata_publishers_[stream_index]->get_subscription_count() > 0);
|
2023-09-07 16:09:35 +08:00
|
|
|
if (!has_subscriber) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
2023-02-06 17:26:10 +08:00
|
|
|
std::shared_ptr<ob::VideoFrame> video_frame;
|
2023-09-07 16:09:35 +08:00
|
|
|
if (frame->type() == OB_FRAME_COLOR) {
|
2022-12-28 16:39:11 +08:00
|
|
|
video_frame = frame->as<ob::ColorFrame>();
|
|
|
|
|
} else if (frame->type() == OB_FRAME_DEPTH) {
|
|
|
|
|
video_frame = frame->as<ob::DepthFrame>();
|
2023-09-05 17:59:19 +08:00
|
|
|
} else if (frame->type() == OB_FRAME_IR || frame->type() == OB_FRAME_IR_LEFT ||
|
|
|
|
|
frame->type() == OB_FRAME_IR_RIGHT) {
|
2022-12-28 16:39:11 +08:00
|
|
|
video_frame = frame->as<ob::IRFrame>();
|
|
|
|
|
} else {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Unsupported frame type: %d", frame->type());
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (!video_frame) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Failed to convert frame to video frame");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
int width = static_cast<int>(video_frame->width());
|
|
|
|
|
int height = static_cast<int>(video_frame->height());
|
2024-07-17 16:56:52 +08:00
|
|
|
auto frame_timestamp = getFrameTimestampUs(frame);
|
|
|
|
|
auto timestamp = fromUsToROSTime(frame_timestamp);
|
2024-05-07 11:08:38 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
CHECK_NOTNULL(device_info);
|
|
|
|
|
auto pid = device_info->pid();
|
|
|
|
|
OBCameraIntrinsic intrinsic;
|
|
|
|
|
OBCameraDistortion distortion;
|
|
|
|
|
if (isGemini335PID(pid)) {
|
|
|
|
|
auto stream_profile = frame->getStreamProfile();
|
|
|
|
|
CHECK_NOTNULL(stream_profile);
|
|
|
|
|
auto video_stream_profile = stream_profile->as<ob::VideoStreamProfile>();
|
|
|
|
|
CHECK_NOTNULL(video_stream_profile);
|
|
|
|
|
intrinsic = video_stream_profile->getIntrinsic();
|
|
|
|
|
distortion = video_stream_profile->getDistortion();
|
|
|
|
|
} else {
|
|
|
|
|
auto camera_params = pipeline_->getCameraParam();
|
|
|
|
|
intrinsic = stream_index.first == OB_STREAM_COLOR ? camera_params.rgbIntrinsic
|
|
|
|
|
: camera_params.depthIntrinsic;
|
|
|
|
|
distortion = stream_index.first == OB_STREAM_COLOR ? camera_params.rgbDistortion
|
|
|
|
|
: camera_params.depthDistortion;
|
2024-09-09 19:41:42 +08:00
|
|
|
if (pid == DABAI_MAX_PID) {
|
2024-09-09 17:53:32 +08:00
|
|
|
// use color param
|
|
|
|
|
intrinsic = camera_params.rgbIntrinsic;
|
|
|
|
|
distortion = camera_params.rgbDistortion;
|
|
|
|
|
}
|
2024-05-07 11:08:38 +08:00
|
|
|
}
|
2024-07-29 09:05:23 +08:00
|
|
|
std::string frame_id = optical_frame_id_[stream_index];
|
|
|
|
|
if (depth_registration_ && stream_index == DEPTH) {
|
|
|
|
|
frame_id = depth_aligned_frame_id_[stream_index];
|
|
|
|
|
}
|
2024-08-22 21:05:37 +08:00
|
|
|
if (stream_index == COLOR && enable_color_undistortion_) {
|
|
|
|
|
memset(&distortion, 0, sizeof(distortion));
|
|
|
|
|
}
|
2024-11-07 11:43:40 +08:00
|
|
|
sensor_msgs::msg::CameraInfo camera_info{};
|
|
|
|
|
|
|
|
|
|
if (color_info_manager_ && color_info_manager_->isCalibrated() && stream_index == COLOR) {
|
|
|
|
|
camera_info = color_info_manager_->getCameraInfo();
|
|
|
|
|
camera_info.header.stamp = timestamp;
|
|
|
|
|
camera_info.header.frame_id = frame_id;
|
|
|
|
|
camera_info.width = width;
|
|
|
|
|
camera_info.height = height;
|
|
|
|
|
} else if (ir_info_manager_ && ir_info_manager_->isCalibrated() &&
|
|
|
|
|
(stream_index == INFRA1 || stream_index == INFRA2 || stream_index == DEPTH)) {
|
|
|
|
|
camera_info = ir_info_manager_->getCameraInfo();
|
|
|
|
|
camera_info.header.stamp = timestamp;
|
|
|
|
|
camera_info.header.frame_id = frame_id;
|
|
|
|
|
camera_info.width = width;
|
|
|
|
|
camera_info.height = height;
|
|
|
|
|
} else {
|
|
|
|
|
camera_info = convertToCameraInfo(intrinsic, distortion, width);
|
|
|
|
|
camera_info.header.stamp = timestamp;
|
|
|
|
|
camera_info.header.frame_id = frame_id;
|
|
|
|
|
camera_info.width = width;
|
|
|
|
|
camera_info.height = height;
|
|
|
|
|
}
|
2024-04-13 18:10:19 +08:00
|
|
|
if (frame->type() == OB_FRAME_IR_RIGHT && enable_stream_[INFRA1]) {
|
2024-05-07 11:08:38 +08:00
|
|
|
auto stream_profile = frame->getStreamProfile();
|
|
|
|
|
CHECK_NOTNULL(stream_profile);
|
|
|
|
|
auto video_stream_profile = stream_profile->as<ob::VideoStreamProfile>();
|
|
|
|
|
CHECK_NOTNULL(video_stream_profile);
|
2024-04-09 11:24:59 +08:00
|
|
|
auto left_video_profile = stream_profile_[INFRA1]->as<ob::VideoStreamProfile>();
|
2024-05-07 11:08:38 +08:00
|
|
|
CHECK_NOTNULL(left_video_profile);
|
2024-04-09 11:24:59 +08:00
|
|
|
auto ex = video_stream_profile->getExtrinsicTo(left_video_profile);
|
|
|
|
|
float fx = camera_info.k.at(0);
|
|
|
|
|
float fy = camera_info.k.at(4);
|
2024-07-30 10:06:42 +08:00
|
|
|
camera_info.p.at(3) = -fx * ex.trans[0] / 1000.0 + 0.0;
|
|
|
|
|
camera_info.p.at(7) = -fy * ex.trans[1] / 1000.0 + 0.0;
|
2024-04-09 11:24:59 +08:00
|
|
|
}
|
2022-12-28 16:39:11 +08:00
|
|
|
CHECK(camera_info_publishers_.count(stream_index) > 0);
|
2024-12-10 15:14:02 +08:00
|
|
|
if (flip_stream_[stream_index]) {
|
|
|
|
|
// We are performing a horizontal flip (left-right mirror) of the image.
|
|
|
|
|
//
|
|
|
|
|
// After flipping the image, the camera's principal point (cx) in the
|
|
|
|
|
// camera_info must be adjusted so that it still refers to the correct point
|
|
|
|
|
// in the flipped image.
|
|
|
|
|
|
|
|
|
|
// Intrinsic matrix K:
|
|
|
|
|
// K = [ fx 0 cx
|
|
|
|
|
// 0 fy cy
|
|
|
|
|
// 0 0 1 ]
|
|
|
|
|
double &cx = camera_info.k[2]; // K[0,2]
|
|
|
|
|
|
|
|
|
|
// Store the original principal point cx
|
|
|
|
|
double old_cx = cx;
|
|
|
|
|
|
|
|
|
|
// For a horizontal flip, the new cx = (width - 1) - old_cx
|
|
|
|
|
// This effectively mirrors the cx value about the center of the image.
|
|
|
|
|
cx = (width - 1) - old_cx;
|
|
|
|
|
|
|
|
|
|
// Update the projection matrix P:
|
|
|
|
|
// P = [ fx 0 cx Tx
|
|
|
|
|
// 0 fy cy Ty
|
|
|
|
|
// 0 0 1 0 ]
|
|
|
|
|
double &p_cx = camera_info.p[2];
|
|
|
|
|
|
|
|
|
|
double old_p_cx = p_cx;
|
|
|
|
|
p_cx = (width - 1) - old_p_cx;
|
|
|
|
|
|
|
|
|
|
// fx, fy, cy remain the same for a simple horizontal flip.
|
|
|
|
|
// The only changes are cx and p_cx.
|
|
|
|
|
}
|
|
|
|
|
|
2022-12-28 16:39:11 +08:00
|
|
|
camera_info_publishers_[stream_index]->publish(camera_info);
|
2024-05-07 11:08:38 +08:00
|
|
|
if (isGemini335PID(pid)) {
|
|
|
|
|
publishMetadata(frame, stream_index, camera_info.header);
|
|
|
|
|
}
|
2026-04-01 11:01:49 +08:00
|
|
|
if (stream_index == COLOR && frame->format() == OB_FORMAT_MJPG &&
|
|
|
|
|
has_compressed_image_subscriber) {
|
|
|
|
|
publishCompressedColorImage(frame, stream_index, timestamp, frame_id);
|
2026-05-27 14:48:47 +08:00
|
|
|
// Record CSV publish timestamp here only when there is no raw image subscriber.
|
|
|
|
|
// If a raw subscriber also exists, the existing logging at the raw publish site will record
|
|
|
|
|
// it, avoiding a double call that would corrupt publish_steady_delta_us.
|
|
|
|
|
if (!has_raw_image_subscriber && frame_timestamp_csv_logger_ &&
|
|
|
|
|
frame_timestamp_csv_logger_->enabled()) {
|
|
|
|
|
frame_timestamp_csv_logger_->recordPreImagePublish(stream_index.first, frame,
|
|
|
|
|
getSystemNowUs(), getSteadyNowUs());
|
|
|
|
|
}
|
2026-04-01 11:01:49 +08:00
|
|
|
}
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(image_publishers_[stream_index]);
|
|
|
|
|
if (image_publishers_[stream_index]->get_subscription_count() == 0) {
|
2024-05-07 11:08:38 +08:00
|
|
|
return;
|
|
|
|
|
}
|
2023-09-07 16:09:35 +08:00
|
|
|
auto &image = images_[stream_index];
|
|
|
|
|
if (image.empty() || image.cols != width || image.rows != height) {
|
|
|
|
|
image.create(height, width, image_format_[stream_index]);
|
|
|
|
|
}
|
|
|
|
|
if (frame->type() == OB_FRAME_COLOR && !is_color_frame_decoded_) {
|
|
|
|
|
RCLCPP_ERROR(logger_, "color frame is not decoded");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (frame->type() == OB_FRAME_COLOR) {
|
|
|
|
|
memcpy(image.data, rgb_buffer_, video_frame->width() * video_frame->height() * 3);
|
|
|
|
|
} else {
|
|
|
|
|
memcpy(image.data, video_frame->data(), video_frame->dataSize());
|
|
|
|
|
}
|
|
|
|
|
if (stream_index == DEPTH) {
|
|
|
|
|
auto depth_scale = video_frame->as<ob::DepthFrame>()->getValueScale();
|
|
|
|
|
image = image * depth_scale;
|
|
|
|
|
}
|
2024-12-10 15:14:02 +08:00
|
|
|
if (flip_stream_[stream_index]) {
|
|
|
|
|
// flip image
|
|
|
|
|
cv::flip(image, image, 1);
|
|
|
|
|
}
|
2024-08-27 17:58:12 +08:00
|
|
|
sensor_msgs::msg::Image::UniquePtr image_msg(new sensor_msgs::msg::Image());
|
|
|
|
|
|
|
|
|
|
cv_bridge::CvImage(std_msgs::msg::Header(), encoding_[stream_index], image)
|
|
|
|
|
.toImageMsg(*image_msg);
|
2024-05-07 15:55:32 +08:00
|
|
|
CHECK_NOTNULL(image_msg.get());
|
2022-12-28 16:39:11 +08:00
|
|
|
image_msg->header.stamp = timestamp;
|
|
|
|
|
image_msg->is_bigendian = false;
|
|
|
|
|
image_msg->step = width * unit_step_size_[stream_index];
|
2023-09-05 18:02:06 +08:00
|
|
|
image_msg->header.frame_id = frame_id;
|
2022-12-28 16:39:11 +08:00
|
|
|
CHECK(image_publishers_.count(stream_index) > 0);
|
2024-08-27 17:58:12 +08:00
|
|
|
saveImageToFile(stream_index, image, *image_msg);
|
2026-04-13 10:41:51 +08:00
|
|
|
if (frame_timestamp_csv_logger_ && frame_timestamp_csv_logger_->enabled() &&
|
|
|
|
|
(stream_index == COLOR || stream_index == DEPTH)) {
|
|
|
|
|
frame_timestamp_csv_logger_->recordPreImagePublish(stream_index.first, frame, getSystemNowUs(),
|
|
|
|
|
getSteadyNowUs());
|
|
|
|
|
}
|
2024-08-27 17:58:12 +08:00
|
|
|
image_publishers_[stream_index]->publish(std::move(image_msg));
|
2024-07-31 14:30:54 +08:00
|
|
|
if (stream_index == COLOR && enable_color_undistortion_ &&
|
2024-08-27 17:58:12 +08:00
|
|
|
color_undistortion_publisher_->get_subscription_count() > 0) {
|
2024-07-31 14:30:54 +08:00
|
|
|
auto undistorted_image = undistortImage(image, intrinsic, distortion);
|
2024-08-27 17:58:12 +08:00
|
|
|
sensor_msgs::msg::Image::UniquePtr undistorted_image_msg(new sensor_msgs::msg::Image());
|
|
|
|
|
cv_bridge::CvImage(std_msgs::msg::Header(), encoding_[stream_index], undistorted_image)
|
|
|
|
|
.toImageMsg(*undistorted_image_msg);
|
2024-07-31 14:30:54 +08:00
|
|
|
CHECK_NOTNULL(undistorted_image_msg.get());
|
|
|
|
|
undistorted_image_msg->header.stamp = timestamp;
|
|
|
|
|
undistorted_image_msg->is_bigendian = false;
|
|
|
|
|
undistorted_image_msg->step = width * unit_step_size_[stream_index];
|
|
|
|
|
undistorted_image_msg->header.frame_id = frame_id;
|
2024-08-27 17:58:12 +08:00
|
|
|
color_undistortion_publisher_->publish(std::move(undistorted_image_msg));
|
2024-07-31 14:30:54 +08:00
|
|
|
}
|
2023-02-20 14:46:02 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-15 10:24:38 +08:00
|
|
|
void OBCameraNode::publishMetadata(const std::shared_ptr<ob::Frame> &frame,
|
|
|
|
|
const stream_index_pair &stream_index,
|
|
|
|
|
const std_msgs::msg::Header &header) {
|
|
|
|
|
if (metadata_publishers_.count(stream_index) == 0) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
auto metadata_publisher = metadata_publishers_[stream_index];
|
|
|
|
|
if (metadata_publisher->get_subscription_count() == 0) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
orbbec_camera_msgs::msg::Metadata metadata_msg;
|
|
|
|
|
metadata_msg.header = header;
|
|
|
|
|
nlohmann::json json_data;
|
|
|
|
|
|
|
|
|
|
for (int i = 0; i < OB_FRAME_METADATA_TYPE_COUNT; i++) {
|
|
|
|
|
auto meta_data_type = static_cast<OBFrameMetadataType>(i);
|
|
|
|
|
std::string field_name = metaDataTypeToString(meta_data_type);
|
|
|
|
|
if (!frame->hasMetadata(meta_data_type)) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
int64_t value = frame->getMetadataValue(meta_data_type);
|
|
|
|
|
json_data[field_name] = value;
|
|
|
|
|
}
|
|
|
|
|
metadata_msg.json_data = json_data.dump(2);
|
|
|
|
|
metadata_publisher->publish(metadata_msg);
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::saveImageToFile(const stream_index_pair &stream_index, const cv::Mat &image,
|
2024-08-27 17:58:12 +08:00
|
|
|
const sensor_msgs::msg::Image &image_msg) {
|
2024-01-30 11:21:18 +08:00
|
|
|
if (save_images_[stream_index]) {
|
2023-02-20 14:46:02 +08:00
|
|
|
auto now = time(nullptr);
|
|
|
|
|
std::stringstream ss;
|
|
|
|
|
ss << std::put_time(localtime(&now), "%Y%m%d_%H%M%S");
|
|
|
|
|
auto current_path = std::filesystem::current_path().string();
|
|
|
|
|
auto fps = fps_[stream_index];
|
2024-01-30 11:21:18 +08:00
|
|
|
int index = save_images_count_[stream_index];
|
|
|
|
|
std::string file_suffix = stream_index == COLOR ? ".png" : ".raw";
|
2023-02-20 14:46:02 +08:00
|
|
|
std::string filename = current_path + "/image/" + stream_name_[stream_index] + "_" +
|
2024-08-27 17:58:12 +08:00
|
|
|
std::to_string(image_msg.width) + "x" +
|
|
|
|
|
std::to_string(image_msg.height) + "_" + std::to_string(fps) + "hz_" +
|
2024-01-29 10:17:36 +08:00
|
|
|
ss.str() + "_" + std::to_string(index) + file_suffix;
|
2023-02-20 14:46:02 +08:00
|
|
|
if (!std::filesystem::exists(current_path + "/image")) {
|
|
|
|
|
std::filesystem::create_directory(current_path + "/image");
|
|
|
|
|
}
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Saving image to " << filename);
|
2024-01-30 11:21:18 +08:00
|
|
|
if (stream_index.first == OB_STREAM_COLOR) {
|
2023-02-20 14:46:02 +08:00
|
|
|
auto image_to_save =
|
|
|
|
|
cv_bridge::toCvCopy(image_msg, sensor_msgs::image_encodings::BGR8)->image;
|
|
|
|
|
cv::imwrite(filename, image_to_save);
|
2023-09-11 17:57:02 +08:00
|
|
|
} else if (stream_index.first == OB_STREAM_IR || stream_index.first == OB_STREAM_IR_LEFT ||
|
2024-01-30 11:21:18 +08:00
|
|
|
stream_index.first == OB_STREAM_IR_RIGHT || stream_index.first == OB_STREAM_DEPTH) {
|
|
|
|
|
std::ofstream ofs(filename, std::ios::out | std::ios::binary);
|
|
|
|
|
if (!ofs.is_open()) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to open file: " << filename);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (image.isContinuous()) {
|
|
|
|
|
ofs.write(reinterpret_cast<const char *>(image.data), image.total() * image.elemSize());
|
|
|
|
|
} else {
|
|
|
|
|
int rows = image.rows;
|
|
|
|
|
int cols = image.cols * image.channels();
|
|
|
|
|
for (int r = 0; r < rows; ++r) {
|
|
|
|
|
ofs.write(reinterpret_cast<const char *>(image.ptr<uchar>(r)), cols);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
ofs.close();
|
2023-02-20 14:46:02 +08:00
|
|
|
} else {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Unsupported stream type: " << stream_index.first);
|
|
|
|
|
}
|
2024-01-30 11:21:18 +08:00
|
|
|
if (++save_images_count_[stream_index] >= max_save_images_count_) {
|
|
|
|
|
save_images_[stream_index] = false;
|
|
|
|
|
}
|
2023-02-20 14:46:02 +08:00
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
2022-06-13 14:58:47 +08:00
|
|
|
|
2023-12-20 17:56:37 +08:00
|
|
|
void OBCameraNode::onNewIMUFrameSyncOutputCallback(const std::shared_ptr<ob::Frame> &accelframe,
|
|
|
|
|
const std::shared_ptr<ob::Frame> &gryoframe) {
|
2024-05-09 17:07:43 +08:00
|
|
|
if (!is_camera_node_initialized_) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
2023-12-20 17:56:37 +08:00
|
|
|
if (!imu_gyro_accel_publisher_) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "stream Accel Gryo publisher not initialized");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
bool has_subscriber = imu_gyro_accel_publisher_->get_subscription_count() > 0;
|
|
|
|
|
has_subscriber = has_subscriber || imu_info_publishers_[GYRO]->get_subscription_count() > 0;
|
|
|
|
|
has_subscriber = has_subscriber || imu_info_publishers_[ACCEL]->get_subscription_count() > 0;
|
|
|
|
|
if (!has_subscriber) {
|
2023-12-20 17:56:37 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
auto imu_msg = sensor_msgs::msg::Imu();
|
|
|
|
|
setDefaultIMUMessage(imu_msg);
|
|
|
|
|
|
2024-12-24 15:56:13 +08:00
|
|
|
imu_msg.header.frame_id = optical_frame_id_[GYRO];
|
2024-07-17 16:56:52 +08:00
|
|
|
auto frame_timestamp = getFrameTimestampUs(accelframe);
|
|
|
|
|
auto timestamp = fromUsToROSTime(frame_timestamp);
|
2024-12-24 20:31:52 +08:00
|
|
|
imu_msg.header.stamp = timestamp;
|
|
|
|
|
|
2024-04-16 20:37:29 +08:00
|
|
|
auto gyro_info = createIMUInfo(GYRO);
|
|
|
|
|
gyro_info.header = imu_msg.header;
|
|
|
|
|
imu_info_publishers_[GYRO]->publish(gyro_info);
|
2024-12-24 20:31:52 +08:00
|
|
|
|
2024-12-24 15:56:13 +08:00
|
|
|
auto accel_info = createIMUInfo(ACCEL);
|
|
|
|
|
imu_msg.header.frame_id = optical_frame_id_[ACCEL];
|
|
|
|
|
accel_info.header = imu_msg.header;
|
|
|
|
|
imu_info_publishers_[ACCEL]->publish(accel_info);
|
2024-12-24 20:31:52 +08:00
|
|
|
|
|
|
|
|
imu_msg.header.frame_id = accel_gyro_frame_id_;
|
|
|
|
|
|
2024-12-24 15:56:13 +08:00
|
|
|
auto gyro_frame = gryoframe->as<ob::GyroFrame>();
|
2023-12-20 17:56:37 +08:00
|
|
|
auto gyroData = gyro_frame->value();
|
2024-04-16 20:37:29 +08:00
|
|
|
imu_msg.angular_velocity.x = gyroData.x - gyro_info.bias[0];
|
|
|
|
|
imu_msg.angular_velocity.y = gyroData.y - gyro_info.bias[1];
|
|
|
|
|
imu_msg.angular_velocity.z = gyroData.z - gyro_info.bias[2];
|
2024-12-24 20:31:52 +08:00
|
|
|
|
2023-12-20 17:56:37 +08:00
|
|
|
auto accel_frame = accelframe->as<ob::AccelFrame>();
|
|
|
|
|
auto accelData = accel_frame->value();
|
2024-12-24 20:31:52 +08:00
|
|
|
|
2024-04-16 20:37:29 +08:00
|
|
|
imu_msg.linear_acceleration.x = accelData.x - accel_info.bias[0];
|
|
|
|
|
imu_msg.linear_acceleration.y = accelData.y - accel_info.bias[1];
|
|
|
|
|
imu_msg.linear_acceleration.z = accelData.z - accel_info.bias[2];
|
2024-12-24 20:31:52 +08:00
|
|
|
|
2023-12-20 17:56:37 +08:00
|
|
|
imu_gyro_accel_publisher_->publish(imu_msg);
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::onNewIMUFrameCallback(const std::shared_ptr<ob::Frame> &frame,
|
|
|
|
|
const stream_index_pair &stream_index) {
|
2024-05-09 17:07:43 +08:00
|
|
|
if (!is_camera_node_initialized_) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
2023-04-28 09:44:16 +08:00
|
|
|
if (!imu_publishers_.count(stream_index)) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"stream " << stream_name_[stream_index] << " publisher not initialized");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
bool has_subscriber = imu_publishers_[stream_index]->get_subscription_count() > 0;
|
|
|
|
|
has_subscriber =
|
|
|
|
|
has_subscriber || imu_info_publishers_[stream_index]->get_subscription_count() > 0;
|
|
|
|
|
if (!has_subscriber) {
|
2023-04-28 09:44:16 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
auto imu_msg = sensor_msgs::msg::Imu();
|
|
|
|
|
setDefaultIMUMessage(imu_msg);
|
2024-12-24 20:31:52 +08:00
|
|
|
|
2023-04-28 09:44:16 +08:00
|
|
|
imu_msg.header.frame_id = optical_frame_id_[stream_index];
|
2024-03-15 10:24:38 +08:00
|
|
|
auto timestamp = fromUsToROSTime(frame->timeStampUs());
|
2023-04-28 09:44:16 +08:00
|
|
|
imu_msg.header.stamp = timestamp;
|
2024-12-24 20:31:52 +08:00
|
|
|
|
2024-04-16 20:37:29 +08:00
|
|
|
auto imu_info = createIMUInfo(stream_index);
|
|
|
|
|
imu_info.header = imu_msg.header;
|
|
|
|
|
imu_info_publishers_[stream_index]->publish(imu_info);
|
2024-12-24 20:31:52 +08:00
|
|
|
|
2023-04-28 09:44:16 +08:00
|
|
|
if (frame->type() == OB_FRAME_GYRO) {
|
|
|
|
|
auto gyro_frame = frame->as<ob::GyroFrame>();
|
|
|
|
|
auto data = gyro_frame->value();
|
2024-04-16 20:37:29 +08:00
|
|
|
imu_msg.angular_velocity.x = data.x - imu_info.bias[0];
|
|
|
|
|
imu_msg.angular_velocity.y = data.y - imu_info.bias[1];
|
|
|
|
|
imu_msg.angular_velocity.z = data.z - imu_info.bias[2];
|
2023-04-28 09:44:16 +08:00
|
|
|
} else if (frame->type() == OB_FRAME_ACCEL) {
|
|
|
|
|
auto accel_frame = frame->as<ob::AccelFrame>();
|
|
|
|
|
auto data = accel_frame->value();
|
2024-04-16 20:37:29 +08:00
|
|
|
imu_msg.linear_acceleration.x = data.x - imu_info.bias[0];
|
|
|
|
|
imu_msg.linear_acceleration.y = data.y - imu_info.bias[1];
|
|
|
|
|
imu_msg.linear_acceleration.z = data.z - imu_info.bias[2];
|
2023-04-28 09:44:16 +08:00
|
|
|
} else {
|
|
|
|
|
RCLCPP_ERROR(logger_, "Unsupported IMU frame type");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
imu_publishers_[stream_index]->publish(imu_msg);
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::setDefaultIMUMessage(sensor_msgs::msg::Imu &imu_msg) {
|
2023-04-28 09:44:16 +08:00
|
|
|
imu_msg.header.frame_id = "imu_link";
|
|
|
|
|
imu_msg.orientation.x = 0.0;
|
|
|
|
|
imu_msg.orientation.y = 0.0;
|
|
|
|
|
imu_msg.orientation.z = 0.0;
|
2024-09-27 09:20:04 +08:00
|
|
|
imu_msg.orientation.w = 1.0;
|
2023-04-28 09:44:16 +08:00
|
|
|
|
|
|
|
|
imu_msg.orientation_covariance = {-1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0};
|
|
|
|
|
imu_msg.linear_acceleration_covariance = {
|
|
|
|
|
liner_accel_cov_, 0.0, 0.0, 0.0, liner_accel_cov_, 0.0, 0.0, 0.0, liner_accel_cov_};
|
|
|
|
|
imu_msg.angular_velocity_covariance = {
|
|
|
|
|
angular_vel_cov_, 0.0, 0.0, 0.0, angular_vel_cov_, 0.0, 0.0, 0.0, angular_vel_cov_};
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
sensor_msgs::msg::Imu OBCameraNode::createUnitIMUMessage(const IMUData &accel_data,
|
|
|
|
|
const IMUData &gyro_data) {
|
2023-04-28 09:44:16 +08:00
|
|
|
sensor_msgs::msg::Imu imu_msg;
|
|
|
|
|
rclcpp::Time timestamp(gyro_data.timestamp_);
|
|
|
|
|
imu_msg.header.stamp = timestamp;
|
|
|
|
|
imu_msg.angular_velocity.x = gyro_data.data_.x();
|
|
|
|
|
imu_msg.angular_velocity.y = gyro_data.data_.y();
|
|
|
|
|
imu_msg.angular_velocity.z = gyro_data.data_.z();
|
|
|
|
|
|
|
|
|
|
imu_msg.linear_acceleration.x = accel_data.data_.x();
|
|
|
|
|
imu_msg.linear_acceleration.y = accel_data.data_.y();
|
|
|
|
|
imu_msg.linear_acceleration.z = accel_data.data_.z();
|
|
|
|
|
return imu_msg;
|
|
|
|
|
}
|
|
|
|
|
|
2022-06-07 20:59:51 +08:00
|
|
|
std::optional<OBCameraParam> OBCameraNode::findDefaultCameraParam() {
|
|
|
|
|
auto camera_params = device_->getCalibrationCameraParamList();
|
|
|
|
|
for (size_t i = 0; i < camera_params->count(); i++) {
|
|
|
|
|
auto param = camera_params->getCameraParam(i);
|
2022-06-06 19:00:12 +08:00
|
|
|
int depth_w = param.depthIntrinsic.width;
|
|
|
|
|
int depth_h = param.depthIntrinsic.height;
|
|
|
|
|
int color_w = param.rgbIntrinsic.width;
|
|
|
|
|
int color_h = param.rgbIntrinsic.height;
|
|
|
|
|
if ((depth_w * height_[DEPTH] == depth_h * width_[DEPTH]) &&
|
|
|
|
|
(color_w * height_[COLOR] == color_h * width_[COLOR])) {
|
|
|
|
|
return param;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return {};
|
|
|
|
|
}
|
2022-06-07 20:59:51 +08:00
|
|
|
|
2023-02-15 19:50:30 +08:00
|
|
|
std::optional<OBCameraParam> OBCameraNode::getDepthCameraParam() {
|
|
|
|
|
auto camera_params = device_->getCalibrationCameraParamList();
|
2023-10-25 10:23:24 +08:00
|
|
|
for (size_t i = 0; i < camera_params->count(); i++) {
|
|
|
|
|
auto param = camera_params->getCameraParam(i);
|
|
|
|
|
int depth_w = param.depthIntrinsic.width;
|
|
|
|
|
int depth_h = param.depthIntrinsic.height;
|
|
|
|
|
if (depth_w == width_[DEPTH] && depth_h == height_[DEPTH]) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "getCameraDepthParam w: " << depth_w << ",h:" << depth_h);
|
|
|
|
|
return param;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-15 19:50:30 +08:00
|
|
|
for (size_t i = 0; i < camera_params->count(); i++) {
|
|
|
|
|
auto param = camera_params->getCameraParam(i);
|
|
|
|
|
int depth_w = param.depthIntrinsic.width;
|
|
|
|
|
int depth_h = param.depthIntrinsic.height;
|
|
|
|
|
if (depth_w * height_[DEPTH] == depth_h * width_[DEPTH]) {
|
2023-10-25 10:23:24 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "getCameraDepthParam w: " << depth_w << ",h:" << depth_h);
|
2023-02-15 19:50:30 +08:00
|
|
|
return param;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return {};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::optional<OBCameraParam> OBCameraNode::getColorCameraParam() {
|
|
|
|
|
auto camera_params = device_->getCalibrationCameraParamList();
|
2023-10-25 10:23:24 +08:00
|
|
|
for (size_t i = 0; i < camera_params->count(); i++) {
|
|
|
|
|
auto param = camera_params->getCameraParam(i);
|
|
|
|
|
int color_w = param.rgbIntrinsic.width;
|
|
|
|
|
int color_h = param.rgbIntrinsic.height;
|
|
|
|
|
if (color_w == width_[COLOR] && color_h == height_[COLOR]) {
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "getColorCameraParam w: " << color_w << ",h:" << color_h);
|
|
|
|
|
return param;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-02-15 19:50:30 +08:00
|
|
|
for (size_t i = 0; i < camera_params->count(); i++) {
|
|
|
|
|
auto param = camera_params->getCameraParam(i);
|
|
|
|
|
int color_w = param.rgbIntrinsic.width;
|
|
|
|
|
int color_h = param.rgbIntrinsic.height;
|
|
|
|
|
if (color_w * height_[COLOR] == color_h * width_[COLOR]) {
|
2023-10-25 10:23:24 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "getColorCameraParam w: " << color_w << ",h:" << color_h);
|
2023-02-15 19:50:30 +08:00
|
|
|
return param;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return {};
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::publishStaticTF(const rclcpp::Time &t, const tf2::Vector3 &trans,
|
|
|
|
|
const tf2::Quaternion &q, const std::string &from,
|
|
|
|
|
const std::string &to) {
|
2022-06-06 19:00:12 +08:00
|
|
|
geometry_msgs::msg::TransformStamped msg;
|
|
|
|
|
msg.header.stamp = t;
|
|
|
|
|
msg.header.frame_id = from;
|
|
|
|
|
msg.child_frame_id = to;
|
2023-07-14 11:14:17 +08:00
|
|
|
msg.transform.translation.x = trans[2] / 1000.0;
|
|
|
|
|
msg.transform.translation.y = -trans[0] / 1000.0;
|
|
|
|
|
msg.transform.translation.z = -trans[1] / 1000.0;
|
2022-06-06 19:00:12 +08:00
|
|
|
msg.transform.rotation.x = q.getX();
|
|
|
|
|
msg.transform.rotation.y = q.getY();
|
|
|
|
|
msg.transform.rotation.z = q.getZ();
|
|
|
|
|
msg.transform.rotation.w = q.getW();
|
|
|
|
|
static_tf_msgs_.push_back(msg);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::calcAndPublishStaticTransform() {
|
2024-03-15 10:24:38 +08:00
|
|
|
tf2::Quaternion quaternion_optical, zero_rot;
|
2022-06-06 19:00:12 +08:00
|
|
|
zero_rot.setRPY(0.0, 0.0, 0.0);
|
|
|
|
|
quaternion_optical.setRPY(-M_PI / 2, 0.0, -M_PI / 2);
|
2023-07-14 11:14:17 +08:00
|
|
|
tf2::Vector3 zero_trans(0, 0, 0);
|
2024-03-15 10:24:38 +08:00
|
|
|
auto base_stream_profile = stream_profile_[base_stream_];
|
2024-07-08 21:37:53 +08:00
|
|
|
auto device_info = device_->getDeviceInfo();
|
|
|
|
|
CHECK_NOTNULL(device_info);
|
|
|
|
|
auto pid = device_info->pid();
|
2024-04-13 15:35:14 +08:00
|
|
|
if (!base_stream_profile) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to get base stream profile");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
CHECK_NOTNULL(base_stream_profile.get());
|
|
|
|
|
for (const auto &item : stream_profile_) {
|
|
|
|
|
auto stream_index = item.first;
|
2024-07-08 21:37:53 +08:00
|
|
|
|
2024-03-15 10:24:38 +08:00
|
|
|
auto stream_profile = item.second;
|
|
|
|
|
if (!stream_profile) {
|
2023-09-05 18:06:52 +08:00
|
|
|
continue;
|
|
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
OBExtrinsic ex;
|
|
|
|
|
try {
|
|
|
|
|
ex = stream_profile->getExtrinsicTo(base_stream_profile);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_, "Failed to get " << stream_name_[stream_index]
|
|
|
|
|
<< " extrinsic: " << e.getMessage());
|
|
|
|
|
ex = OBExtrinsic({{1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 0, 0}});
|
2023-12-21 16:33:21 +08:00
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
|
|
|
|
|
auto Q = rotationMatrixToQuaternion(ex.rot);
|
|
|
|
|
Q = quaternion_optical * Q * quaternion_optical.inverse();
|
|
|
|
|
tf2::Vector3 trans(ex.trans[0], ex.trans[1], ex.trans[2]);
|
|
|
|
|
auto timestamp = node_->now();
|
2024-07-08 21:37:53 +08:00
|
|
|
if (stream_index.first != base_stream_.first) {
|
2024-07-10 15:38:49 +08:00
|
|
|
if (stream_index.first == OB_STREAM_IR_RIGHT && base_stream_.first == OB_STREAM_DEPTH) {
|
2024-07-09 10:02:16 +08:00
|
|
|
trans[0] = std::abs(trans[0]); // because left and right ir calibration is error
|
|
|
|
|
}
|
2024-07-08 21:37:53 +08:00
|
|
|
publishStaticTF(timestamp, trans, Q, frame_id_[base_stream_], frame_id_[stream_index]);
|
|
|
|
|
}
|
2024-03-15 10:24:38 +08:00
|
|
|
publishStaticTF(timestamp, zero_trans, quaternion_optical, frame_id_[stream_index],
|
2023-09-05 18:06:52 +08:00
|
|
|
optical_frame_id_[stream_index]);
|
2024-07-10 15:38:49 +08:00
|
|
|
RCLCPP_INFO_STREAM(logger_, "Publishing static transform from " << stream_name_[stream_index]
|
|
|
|
|
<< " to "
|
|
|
|
|
<< stream_name_[base_stream_]);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Translation " << trans[0] << ", " << trans[1] << ", " << trans[2]);
|
|
|
|
|
RCLCPP_INFO_STREAM(logger_, "Rotation " << Q.getX() << ", " << Q.getY() << ", " << Q.getZ()
|
|
|
|
|
<< ", " << Q.getW());
|
2023-09-05 18:06:52 +08:00
|
|
|
}
|
2024-07-08 21:37:53 +08:00
|
|
|
|
|
|
|
|
if ((pid == FEMTO_BOLT_PID || pid == FEMTO_MEGA_PID) && enable_stream_[DEPTH] &&
|
|
|
|
|
enable_stream_[COLOR]) {
|
|
|
|
|
// calc depth to color
|
|
|
|
|
CHECK_NOTNULL(stream_profile_[COLOR]);
|
|
|
|
|
auto depth_to_color_extrinsics = base_stream_profile->getExtrinsicTo(stream_profile_[COLOR]);
|
|
|
|
|
auto Q = rotationMatrixToQuaternion(depth_to_color_extrinsics.rot);
|
|
|
|
|
Q = quaternion_optical * Q * quaternion_optical.inverse();
|
|
|
|
|
publishStaticTF(node_->now(), zero_trans, Q, camera_link_frame_id_, frame_id_[base_stream_]);
|
|
|
|
|
} else {
|
|
|
|
|
publishStaticTF(node_->now(), zero_trans, zero_rot, camera_link_frame_id_,
|
|
|
|
|
frame_id_[base_stream_]);
|
|
|
|
|
}
|
2024-04-09 11:24:59 +08:00
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[COLOR]) {
|
|
|
|
|
static const char *frame_id = "depth_to_color_extrinsics";
|
|
|
|
|
OBExtrinsic ex;
|
|
|
|
|
try {
|
|
|
|
|
ex = base_stream_profile->getExtrinsicTo(stream_profile_[COLOR]);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Failed to get " << frame_id << " extrinsic: " << e.getMessage());
|
|
|
|
|
ex = OBExtrinsic({{1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 0, 0}});
|
|
|
|
|
}
|
|
|
|
|
depth_to_other_extrinsics_[COLOR] = ex;
|
|
|
|
|
auto ex_msg = obExtrinsicsToMsg(ex, frame_id);
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(depth_to_other_extrinsics_publishers_[COLOR]);
|
2024-04-09 11:24:59 +08:00
|
|
|
depth_to_other_extrinsics_publishers_[COLOR]->publish(ex_msg);
|
|
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[INFRA0]) {
|
|
|
|
|
static const char *frame_id = "depth_to_ir_extrinsics";
|
|
|
|
|
OBExtrinsic ex;
|
|
|
|
|
try {
|
|
|
|
|
ex = base_stream_profile->getExtrinsicTo(stream_profile_[INFRA0]);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Failed to get " << frame_id << " extrinsic: " << e.getMessage());
|
|
|
|
|
ex = OBExtrinsic({{1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 0, 0}});
|
|
|
|
|
}
|
|
|
|
|
depth_to_other_extrinsics_[INFRA0] = ex;
|
|
|
|
|
auto ex_msg = obExtrinsicsToMsg(ex, frame_id);
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(depth_to_other_extrinsics_publishers_[INFRA0]);
|
2024-04-09 11:24:59 +08:00
|
|
|
depth_to_other_extrinsics_publishers_[INFRA0]->publish(ex_msg);
|
|
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[INFRA1]) {
|
|
|
|
|
static const char *frame_id = "depth_to_left_ir_extrinsics";
|
|
|
|
|
OBExtrinsic ex;
|
|
|
|
|
try {
|
|
|
|
|
ex = base_stream_profile->getExtrinsicTo(stream_profile_[INFRA1]);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Failed to get " << frame_id << " extrinsic: " << e.getMessage());
|
|
|
|
|
ex = OBExtrinsic({{1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 0, 0}});
|
|
|
|
|
}
|
|
|
|
|
depth_to_other_extrinsics_[INFRA1] = ex;
|
|
|
|
|
auto ex_msg = obExtrinsicsToMsg(ex, frame_id);
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(depth_to_other_extrinsics_publishers_[INFRA1]);
|
2024-04-09 11:24:59 +08:00
|
|
|
depth_to_other_extrinsics_publishers_[INFRA1]->publish(ex_msg);
|
|
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[INFRA2]) {
|
|
|
|
|
static const char *frame_id = "depth_to_right_ir_extrinsics";
|
|
|
|
|
OBExtrinsic ex;
|
|
|
|
|
try {
|
|
|
|
|
ex = base_stream_profile->getExtrinsicTo(stream_profile_[INFRA2]);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Failed to get " << frame_id << " extrinsic: " << e.getMessage());
|
|
|
|
|
ex = OBExtrinsic({{1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 0, 0}});
|
|
|
|
|
}
|
2024-07-10 15:38:49 +08:00
|
|
|
ex.trans[0] = -std::abs(ex.trans[0]);
|
2024-04-09 11:24:59 +08:00
|
|
|
depth_to_other_extrinsics_[INFRA2] = ex;
|
|
|
|
|
auto ex_msg = obExtrinsicsToMsg(ex, frame_id);
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(depth_to_other_extrinsics_publishers_[INFRA2]);
|
2024-04-09 11:24:59 +08:00
|
|
|
depth_to_other_extrinsics_publishers_[INFRA2]->publish(ex_msg);
|
|
|
|
|
}
|
2024-04-28 22:57:54 +08:00
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[ACCEL]) {
|
|
|
|
|
static const char *frame_id = "depth_to_accel_extrinsics";
|
|
|
|
|
OBExtrinsic ex;
|
|
|
|
|
try {
|
|
|
|
|
ex = base_stream_profile->getExtrinsicTo(stream_profile_[ACCEL]);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Failed to get " << frame_id << " extrinsic: " << e.getMessage());
|
|
|
|
|
ex = OBExtrinsic({{1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 0, 0}});
|
|
|
|
|
}
|
|
|
|
|
depth_to_other_extrinsics_[ACCEL] = ex;
|
|
|
|
|
auto ex_msg = obExtrinsicsToMsg(ex, frame_id);
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(depth_to_other_extrinsics_publishers_[ACCEL]);
|
2024-04-28 22:57:54 +08:00
|
|
|
depth_to_other_extrinsics_publishers_[ACCEL]->publish(ex_msg);
|
|
|
|
|
}
|
|
|
|
|
if (enable_stream_[DEPTH] && enable_stream_[GYRO]) {
|
|
|
|
|
static const char *frame_id = "depth_to_gyro_extrinsics";
|
|
|
|
|
OBExtrinsic ex;
|
|
|
|
|
try {
|
|
|
|
|
ex = base_stream_profile->getExtrinsicTo(stream_profile_[GYRO]);
|
|
|
|
|
} catch (const ob::Error &e) {
|
|
|
|
|
RCLCPP_ERROR_STREAM(logger_,
|
|
|
|
|
"Failed to get " << frame_id << " extrinsic: " << e.getMessage());
|
|
|
|
|
ex = OBExtrinsic({{1, 0, 0, 0, 1, 0, 0, 0, 1}, {0, 0, 0}});
|
|
|
|
|
}
|
|
|
|
|
depth_to_other_extrinsics_[GYRO] = ex;
|
|
|
|
|
auto ex_msg = obExtrinsicsToMsg(ex, frame_id);
|
2024-08-27 17:58:12 +08:00
|
|
|
CHECK_NOTNULL(depth_to_other_extrinsics_publishers_[GYRO]);
|
2024-04-28 22:57:54 +08:00
|
|
|
depth_to_other_extrinsics_publishers_[GYRO]->publish(ex_msg);
|
|
|
|
|
}
|
2022-06-06 19:00:12 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::publishStaticTransforms() {
|
2023-02-06 17:26:10 +08:00
|
|
|
if (!publish_tf_) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
static_tf_broadcaster_ = std::make_shared<tf2_ros::StaticTransformBroadcaster>(node_);
|
|
|
|
|
dynamic_tf_broadcaster_ = std::make_shared<tf2_ros::TransformBroadcaster>(node_);
|
2022-06-06 19:00:12 +08:00
|
|
|
calcAndPublishStaticTransform();
|
|
|
|
|
if (tf_publish_rate_ > 0) {
|
|
|
|
|
tf_thread_ = std::make_shared<std::thread>([this]() { publishDynamicTransforms(); });
|
|
|
|
|
} else {
|
|
|
|
|
static_tf_broadcaster_->sendTransform(static_tf_msgs_);
|
|
|
|
|
}
|
|
|
|
|
}
|
2022-06-13 14:58:47 +08:00
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
void OBCameraNode::publishDynamicTransforms() {
|
|
|
|
|
RCLCPP_WARN(logger_, "Publishing dynamic camera transforms (/tf) at %g Hz", tf_publish_rate_);
|
|
|
|
|
std::mutex mu;
|
|
|
|
|
std::unique_lock<std::mutex> lock(mu);
|
|
|
|
|
while (rclcpp::ok() && is_running_) {
|
|
|
|
|
tf_cv_.wait_for(lock, std::chrono::milliseconds((int)(1000.0 / tf_publish_rate_)),
|
|
|
|
|
[this] { return (!(is_running_)); });
|
|
|
|
|
{
|
|
|
|
|
rclcpp::Time t = node_->now();
|
2023-08-27 14:06:25 +08:00
|
|
|
for (auto &msg : static_tf_msgs_) {
|
2022-06-06 19:00:12 +08:00
|
|
|
msg.header.stamp = t;
|
|
|
|
|
}
|
|
|
|
|
dynamic_tf_broadcaster_->sendTransform(static_tf_msgs_);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-28 09:44:16 +08:00
|
|
|
template <typename T>
|
2023-08-27 14:06:25 +08:00
|
|
|
T lerp(const T &a, const T &b, const double t) {
|
2023-04-28 09:44:16 +08:00
|
|
|
return a * (1.0 - t) + b * t;
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::FillImuDataLinearInterpolation(const IMUData &imu_data,
|
|
|
|
|
std::deque<sensor_msgs::msg::Imu> &imu_msgs) {
|
2023-04-28 09:44:16 +08:00
|
|
|
imu_history_.push_back(imu_data);
|
|
|
|
|
stream_index_pair steam_index(imu_data.stream_);
|
|
|
|
|
imu_msgs.clear();
|
|
|
|
|
std::deque<IMUData> gyros_data;
|
|
|
|
|
IMUData accel0, accel1, current_imu;
|
|
|
|
|
while (!imu_history_.empty()) {
|
|
|
|
|
current_imu = imu_history_.front();
|
|
|
|
|
imu_history_.pop_front();
|
|
|
|
|
if (accel0.isSet() && current_imu.stream_ == ACCEL) {
|
|
|
|
|
accel0 = current_imu;
|
|
|
|
|
} else if (accel0.isSet() && current_imu.stream_ == ACCEL) {
|
|
|
|
|
accel1 = current_imu;
|
|
|
|
|
const double dt = accel1.timestamp_ - accel0.timestamp_;
|
|
|
|
|
while (!gyros_data.empty()) {
|
|
|
|
|
auto current_gyro = gyros_data.front();
|
|
|
|
|
gyros_data.pop_front();
|
|
|
|
|
const double alpha = (current_gyro.timestamp_ - accel0.timestamp_) / dt;
|
|
|
|
|
IMUData current_accel(ACCEL, lerp(accel0.data_, accel1.data_, alpha),
|
|
|
|
|
current_gyro.timestamp_);
|
|
|
|
|
imu_msgs.push_back((createUnitIMUMessage(current_accel, current_gyro)));
|
|
|
|
|
}
|
|
|
|
|
accel0 = accel1;
|
|
|
|
|
} else if (accel0.isSet() && current_imu.timestamp_ >= accel0.timestamp_ &&
|
|
|
|
|
current_imu.stream_ == GYRO) {
|
|
|
|
|
gyros_data.push_back(current_imu);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
imu_history_.push_back(current_imu);
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-27 14:06:25 +08:00
|
|
|
void OBCameraNode::FillImuDataCopy(const IMUData &imu_data,
|
|
|
|
|
std::deque<sensor_msgs::msg::Imu> &imu_msgs) {
|
2023-04-28 09:44:16 +08:00
|
|
|
stream_index_pair steam_index(imu_data.stream_);
|
|
|
|
|
if (steam_index == ACCEL) {
|
|
|
|
|
accel_data_ = imu_data;
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (accel_data_.isSet()) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
imu_msgs.push_back(createUnitIMUMessage(accel_data_, imu_data));
|
|
|
|
|
}
|
|
|
|
|
|
2022-12-28 16:39:11 +08:00
|
|
|
bool OBCameraNode::setupFormatConvertType(OBFormat format) {
|
|
|
|
|
switch (format) {
|
2022-06-13 17:35:44 +08:00
|
|
|
case OB_FORMAT_RGB888:
|
|
|
|
|
return true;
|
2022-06-09 16:40:35 +08:00
|
|
|
case OB_FORMAT_I420:
|
|
|
|
|
format_convert_filter_.setFormatConvertType(FORMAT_I420_TO_RGB888);
|
|
|
|
|
break;
|
|
|
|
|
case OB_FORMAT_MJPG:
|
|
|
|
|
format_convert_filter_.setFormatConvertType(FORMAT_MJPEG_TO_RGB888);
|
|
|
|
|
break;
|
|
|
|
|
case OB_FORMAT_YUYV:
|
|
|
|
|
format_convert_filter_.setFormatConvertType(FORMAT_YUYV_TO_RGB888);
|
|
|
|
|
break;
|
|
|
|
|
case OB_FORMAT_NV21:
|
|
|
|
|
format_convert_filter_.setFormatConvertType(FORMAT_NV21_TO_RGB888);
|
|
|
|
|
break;
|
|
|
|
|
case OB_FORMAT_NV12:
|
|
|
|
|
format_convert_filter_.setFormatConvertType(FORMAT_NV12_TO_RGB888);
|
|
|
|
|
break;
|
2023-06-15 09:23:26 +08:00
|
|
|
case OB_FORMAT_UYVY:
|
|
|
|
|
format_convert_filter_.setFormatConvertType(FORMAT_UYVY_TO_RGB888);
|
|
|
|
|
break;
|
2022-06-09 16:40:35 +08:00
|
|
|
default:
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
2024-05-07 11:08:38 +08:00
|
|
|
bool OBCameraNode::isGemini335PID(uint32_t pid) {
|
|
|
|
|
return pid == GEMINI_335_PID || pid == GEMINI_330_PID || pid == GEMINI_336_PID ||
|
|
|
|
|
pid == GEMINI_335L_PID || pid == GEMINI_330L_PID || pid == GEMINI_336L_PID ||
|
|
|
|
|
pid == GEMINI_335LG_PID || pid == GEMINI_336LG_PID || pid == GEMINI_335LE_PID ||
|
|
|
|
|
pid == GEMINI_336LE_PID;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-15 10:24:38 +08:00
|
|
|
orbbec_camera_msgs::msg::IMUInfo OBCameraNode::createIMUInfo(
|
|
|
|
|
const stream_index_pair &stream_index) {
|
|
|
|
|
orbbec_camera_msgs::msg::IMUInfo imu_info;
|
|
|
|
|
imu_info.header.frame_id = optical_frame_id_[stream_index];
|
|
|
|
|
imu_info.header.stamp = node_->now();
|
|
|
|
|
auto imu_profile = stream_profile_[stream_index];
|
|
|
|
|
if (stream_index == GYRO) {
|
|
|
|
|
auto gyro_profile = stream_profile_[stream_index]->as<ob::GyroStreamProfile>();
|
|
|
|
|
auto gyro_intrinsics = gyro_profile->getIntrinsic();
|
|
|
|
|
imu_info.noise_density = gyro_intrinsics.noiseDensity;
|
|
|
|
|
imu_info.random_walk = gyro_intrinsics.randomWalk;
|
|
|
|
|
imu_info.reference_temperature = gyro_intrinsics.referenceTemp;
|
|
|
|
|
imu_info.bias = {gyro_intrinsics.bias[0], gyro_intrinsics.bias[1], gyro_intrinsics.bias[2]};
|
|
|
|
|
imu_info.scale_misalignment = {
|
|
|
|
|
gyro_intrinsics.scaleMisalignment[0], gyro_intrinsics.scaleMisalignment[1],
|
|
|
|
|
gyro_intrinsics.scaleMisalignment[2], gyro_intrinsics.scaleMisalignment[3],
|
|
|
|
|
gyro_intrinsics.scaleMisalignment[4], gyro_intrinsics.scaleMisalignment[5],
|
|
|
|
|
gyro_intrinsics.scaleMisalignment[6], gyro_intrinsics.scaleMisalignment[7],
|
|
|
|
|
gyro_intrinsics.scaleMisalignment[8]};
|
|
|
|
|
imu_info.temperature_slope = {
|
|
|
|
|
gyro_intrinsics.tempSlope[0], gyro_intrinsics.tempSlope[1], gyro_intrinsics.tempSlope[2],
|
|
|
|
|
gyro_intrinsics.tempSlope[3], gyro_intrinsics.tempSlope[4], gyro_intrinsics.tempSlope[5],
|
|
|
|
|
gyro_intrinsics.tempSlope[6], gyro_intrinsics.tempSlope[7], gyro_intrinsics.tempSlope[8]};
|
|
|
|
|
} else if (stream_index == ACCEL) {
|
|
|
|
|
auto accel_profile = stream_profile_[stream_index]->as<ob::AccelStreamProfile>();
|
|
|
|
|
auto accel_intrinsics = accel_profile->getIntrinsic();
|
|
|
|
|
imu_info.noise_density = accel_intrinsics.noiseDensity;
|
|
|
|
|
imu_info.random_walk = accel_intrinsics.randomWalk;
|
|
|
|
|
imu_info.reference_temperature = accel_intrinsics.referenceTemp;
|
|
|
|
|
imu_info.bias = {accel_intrinsics.bias[0], accel_intrinsics.bias[1], accel_intrinsics.bias[2]};
|
|
|
|
|
imu_info.gravity = {accel_intrinsics.gravity[0], accel_intrinsics.gravity[1],
|
|
|
|
|
accel_intrinsics.gravity[2]};
|
|
|
|
|
imu_info.scale_misalignment = {
|
|
|
|
|
accel_intrinsics.scaleMisalignment[0], accel_intrinsics.scaleMisalignment[1],
|
|
|
|
|
accel_intrinsics.scaleMisalignment[2], accel_intrinsics.scaleMisalignment[3],
|
|
|
|
|
accel_intrinsics.scaleMisalignment[4], accel_intrinsics.scaleMisalignment[5],
|
|
|
|
|
accel_intrinsics.scaleMisalignment[6], accel_intrinsics.scaleMisalignment[7],
|
|
|
|
|
accel_intrinsics.scaleMisalignment[8]};
|
|
|
|
|
imu_info.temperature_slope = {accel_intrinsics.tempSlope[0], accel_intrinsics.tempSlope[1],
|
|
|
|
|
accel_intrinsics.tempSlope[2], accel_intrinsics.tempSlope[3],
|
|
|
|
|
accel_intrinsics.tempSlope[4], accel_intrinsics.tempSlope[5],
|
|
|
|
|
accel_intrinsics.tempSlope[6], accel_intrinsics.tempSlope[7],
|
|
|
|
|
accel_intrinsics.tempSlope[8]};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return imu_info;
|
|
|
|
|
}
|
|
|
|
|
|
2026-04-01 11:01:49 +08:00
|
|
|
bool OBCameraNode::hasCompressedImageSubscriber(const stream_index_pair &stream_index) const {
|
|
|
|
|
auto it = compressed_image_publishers_.find(stream_index);
|
|
|
|
|
return it != compressed_image_publishers_.end() && it->second &&
|
|
|
|
|
it->second->get_subscription_count() > 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void OBCameraNode::publishCompressedColorImage(const std::shared_ptr<ob::Frame> &frame,
|
|
|
|
|
const stream_index_pair &stream_index,
|
|
|
|
|
const rclcpp::Time ×tamp,
|
|
|
|
|
const std::string &frame_id) {
|
|
|
|
|
auto it = compressed_image_publishers_.find(stream_index);
|
|
|
|
|
if (it == compressed_image_publishers_.end() || !it->second) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
sensor_msgs::msg::CompressedImage msg;
|
|
|
|
|
msg.header.stamp = timestamp;
|
|
|
|
|
msg.header.frame_id = frame_id;
|
|
|
|
|
msg.format = "jpeg";
|
|
|
|
|
const auto *data = static_cast<const uint8_t *>(frame->data());
|
|
|
|
|
msg.data.assign(data, data + frame->dataSize());
|
|
|
|
|
it->second->publish(std::move(msg));
|
|
|
|
|
}
|
|
|
|
|
|
2022-06-06 19:00:12 +08:00
|
|
|
} // namespace orbbec_camera
|