Add color decimation filter in gemini2.launch.py and gemini2L.launch.py

This commit is contained in:
jj
2025-06-06 16:37:05 +08:00
parent b47501658b
commit 928e96f672
3 changed files with 31 additions and 5 deletions
+2
View File
@@ -43,6 +43,8 @@ def generate_launch_description():
DeclareLaunchArgument("enable_color_auto_white_balance", default_value="true"), DeclareLaunchArgument("enable_color_auto_white_balance", default_value="true"),
DeclareLaunchArgument("color_white_balance", default_value="-1"), DeclareLaunchArgument("color_white_balance", default_value="-1"),
DeclareLaunchArgument("color_brightness", default_value="-1"), DeclareLaunchArgument("color_brightness", default_value="-1"),
DeclareLaunchArgument('enable_color_decimation_filter', default_value='false'),
DeclareLaunchArgument('color_decimation_filter_scale', default_value='-1'),
DeclareLaunchArgument("depth_width", default_value="0"), DeclareLaunchArgument("depth_width", default_value="0"),
DeclareLaunchArgument("depth_height", default_value="0"), DeclareLaunchArgument("depth_height", default_value="0"),
DeclareLaunchArgument("depth_fps", default_value="0"), DeclareLaunchArgument("depth_fps", default_value="0"),
+2
View File
@@ -39,6 +39,8 @@ def generate_launch_description():
DeclareLaunchArgument("enable_color_auto_white_balance", default_value="true"), DeclareLaunchArgument("enable_color_auto_white_balance", default_value="true"),
DeclareLaunchArgument("color_white_balance", default_value="-1"), DeclareLaunchArgument("color_white_balance", default_value="-1"),
DeclareLaunchArgument("color_brightness", default_value="-1"), DeclareLaunchArgument("color_brightness", default_value="-1"),
DeclareLaunchArgument('enable_color_decimation_filter', default_value='false'),
DeclareLaunchArgument('color_decimation_filter_scale', default_value='-1'),
DeclareLaunchArgument("depth_width", default_value="1280"), DeclareLaunchArgument("depth_width", default_value="1280"),
DeclareLaunchArgument("depth_height", default_value="800"), DeclareLaunchArgument("depth_height", default_value="800"),
DeclareLaunchArgument("depth_fps", default_value="30"), DeclareLaunchArgument("depth_fps", default_value="30"),
+27 -5
View File
@@ -199,7 +199,7 @@ void OBCameraNode::setupDevices() {
"Laser energy level set to " << new_laser_energy_level << " (new value)"); "Laser energy level set to " << new_laser_energy_level << " (new value)");
} }
} }
if (depth_registration_) { if (depth_registration_ && align_mode_ == "SW") {
RCLCPP_INFO_STREAM(logger_, "Create align filter"); RCLCPP_INFO_STREAM(logger_, "Create align filter");
align_filter_ = std::make_unique<ob::Align>(align_target_stream_); align_filter_ = std::make_unique<ob::Align>(align_target_stream_);
} }
@@ -738,10 +738,10 @@ void OBCameraNode::setupDevices() {
void OBCameraNode::setupColorPostProcessFilter() { void OBCameraNode::setupColorPostProcessFilter() {
auto color_sensor = device_->getSensor(OB_SENSOR_COLOR); auto color_sensor = device_->getSensor(OB_SENSOR_COLOR);
color_filter_list_ = color_sensor->createRecommendedFilters(); color_filter_list_ = color_sensor->createRecommendedFilters();
if (color_filter_list_.empty()) { if (color_filter_list_.empty()) {
RCLCPP_WARN_STREAM(logger_, "Failed to get color sensor filter list"); RCLCPP_WARN_STREAM(logger_, "Failed to get color sensor filter list");
return; // return;
} }
for (size_t i = 0; i < color_filter_list_.size(); i++) { for (size_t i = 0; i < color_filter_list_.size(); i++) {
auto filter = color_filter_list_[i]; auto filter = color_filter_list_[i];
std::map<std::string, bool> filter_params = { std::map<std::string, bool> filter_params = {
@@ -770,6 +770,28 @@ void OBCameraNode::setupColorPostProcessFilter() {
} }
} }
} }
auto device_info = device_->getDeviceInfo();
CHECK_NOTNULL(device_info);
auto pid = device_info->getPid();
if (pid == GEMINI2_PID || pid == GEMINI2L_PID) {
if (enable_color_decimation_filter_) {
auto decimation_filter = std::make_shared<ob::DecimationFilter>();
decimation_filter->enable(true);
color_filter_list_.push_back(decimation_filter);
auto range = decimation_filter->getScaleRange();
if (color_decimation_filter_scale_ != -1 && color_decimation_filter_scale_ <= range.max &&
color_decimation_filter_scale_ >= range.min) {
RCLCPP_INFO_STREAM(logger_, "Set color decimation filter scale value to "
<< color_decimation_filter_scale_);
decimation_filter->setScaleValue(color_decimation_filter_scale_);
}
if (color_decimation_filter_scale_ != -1 && (color_decimation_filter_scale_ < range.min ||
color_decimation_filter_scale_ > range.max)) {
RCLCPP_ERROR_STREAM(logger_, "Color Decimation filter scale value is out of range "
<< range.min << " - " << range.max);
}
}
}
} }
void OBCameraNode::setupLeftIrPostProcessFilter() { void OBCameraNode::setupLeftIrPostProcessFilter() {
auto device_info = device_->getDeviceInfo(); auto device_info = device_->getDeviceInfo();