Skip to content

Commit 08492a1

Browse files
authored
propage a portion of global args to the controller nodes (#1712)
1 parent ecfec15 commit 08492a1

File tree

3 files changed

+58
-11
lines changed

3 files changed

+58
-11
lines changed

controller_manager/include/controller_manager/controller_manager.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -577,6 +577,7 @@ class ControllerManager : public rclcpp::Node
577577
std::map<std::string, std::vector<std::string>> controller_chained_reference_interfaces_cache_;
578578
std::map<std::string, std::vector<std::string>> controller_chained_state_interfaces_cache_;
579579

580+
rclcpp::NodeOptions cm_node_options_;
580581
std::string robot_description_;
581582
rclcpp::Subscription<std_msgs::msg::String>::SharedPtr robot_description_subscription_;
582583
rclcpp::TimerBase::SharedPtr robot_description_notification_timer_;

controller_manager/src/controller_manager.cpp

Lines changed: 41 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
#include "controller_manager_msgs/msg/hardware_component_state.hpp"
2525
#include "hardware_interface/types/lifecycle_state_names.hpp"
2626
#include "lifecycle_msgs/msg/state.hpp"
27+
#include "rcl/arguments.h"
2728
#include "rclcpp/version.h"
2829
#include "rclcpp_lifecycle/state.hpp"
2930

@@ -197,7 +198,8 @@ ControllerManager::ControllerManager(
197198
kControllerInterfaceNamespace, kControllerInterfaceClassName)),
198199
chainable_loader_(
199200
std::make_shared<pluginlib::ClassLoader<controller_interface::ChainableControllerInterface>>(
200-
kControllerInterfaceNamespace, kChainableControllerInterfaceClassName))
201+
kControllerInterfaceNamespace, kChainableControllerInterfaceClassName)),
202+
cm_node_options_(options)
201203
{
202204
init_controller_manager();
203205
}
@@ -217,7 +219,8 @@ ControllerManager::ControllerManager(
217219
kControllerInterfaceNamespace, kControllerInterfaceClassName)),
218220
chainable_loader_(
219221
std::make_shared<pluginlib::ClassLoader<controller_interface::ChainableControllerInterface>>(
220-
kControllerInterfaceNamespace, kChainableControllerInterfaceClassName))
222+
kControllerInterfaceNamespace, kChainableControllerInterfaceClassName)),
223+
cm_node_options_(options)
221224
{
222225
init_controller_manager();
223226
}
@@ -234,7 +237,8 @@ ControllerManager::ControllerManager(
234237
kControllerInterfaceNamespace, kControllerInterfaceClassName)),
235238
chainable_loader_(
236239
std::make_shared<pluginlib::ClassLoader<controller_interface::ChainableControllerInterface>>(
237-
kControllerInterfaceNamespace, kChainableControllerInterfaceClassName))
240+
kControllerInterfaceNamespace, kChainableControllerInterfaceClassName)),
241+
cm_node_options_(options)
238242
{
239243
init_controller_manager();
240244
}
@@ -2819,29 +2823,56 @@ rclcpp::NodeOptions ControllerManager::determine_controller_node_options(
28192823

28202824
rclcpp::NodeOptions controller_node_options = controller.c->define_custom_node_options();
28212825
std::vector<std::string> node_options_arguments = controller_node_options.arguments();
2822-
const std::string ros_args_arg = "--ros-args";
2826+
2827+
for (const std::string & arg : cm_node_options_.arguments())
2828+
{
2829+
if (arg.find("__ns") != std::string::npos || arg.find("__node") != std::string::npos)
2830+
{
2831+
if (
2832+
node_options_arguments.back() == RCL_REMAP_FLAG ||
2833+
node_options_arguments.back() == RCL_SHORT_REMAP_FLAG)
2834+
{
2835+
node_options_arguments.pop_back();
2836+
}
2837+
continue;
2838+
}
2839+
2840+
node_options_arguments.push_back(arg);
2841+
}
2842+
28232843
if (controller.info.parameters_file.has_value())
28242844
{
2825-
if (!check_for_element(node_options_arguments, ros_args_arg))
2845+
if (!check_for_element(node_options_arguments, RCL_ROS_ARGS_FLAG))
28262846
{
2827-
node_options_arguments.push_back(ros_args_arg);
2847+
node_options_arguments.push_back(RCL_ROS_ARGS_FLAG);
28282848
}
2829-
node_options_arguments.push_back("--params-file");
2849+
node_options_arguments.push_back(RCL_PARAM_FILE_FLAG);
28302850
node_options_arguments.push_back(controller.info.parameters_file.value());
28312851
}
28322852

28332853
// ensure controller's `use_sim_time` parameter matches controller_manager's
28342854
const rclcpp::Parameter use_sim_time = this->get_parameter("use_sim_time");
28352855
if (use_sim_time.as_bool())
28362856
{
2837-
if (!check_for_element(node_options_arguments, ros_args_arg))
2857+
if (!check_for_element(node_options_arguments, RCL_ROS_ARGS_FLAG))
28382858
{
2839-
node_options_arguments.push_back(ros_args_arg);
2859+
node_options_arguments.push_back(RCL_ROS_ARGS_FLAG);
28402860
}
2841-
node_options_arguments.push_back("-p");
2861+
node_options_arguments.push_back(RCL_PARAM_FLAG);
28422862
node_options_arguments.push_back("use_sim_time:=true");
28432863
}
28442864

2865+
std::string arguments;
2866+
arguments.reserve(1000);
2867+
for (const auto & arg : node_options_arguments)
2868+
{
2869+
arguments.append(arg);
2870+
arguments.append(" ");
2871+
}
2872+
RCLCPP_INFO(
2873+
get_logger(), "Controller '%s' node arguments: %s", controller.info.name.c_str(),
2874+
arguments.c_str());
2875+
28452876
controller_node_options = controller_node_options.arguments(node_options_arguments);
28462877
controller_node_options.use_global_arguments(false);
28472878
return controller_node_options;

controller_manager/src/ros2_control_node.cpp

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,22 @@ int main(int argc, char ** argv)
4141
std::make_shared<rclcpp::executors::MultiThreadedExecutor>();
4242
std::string manager_node_name = "controller_manager";
4343

44-
auto cm = std::make_shared<controller_manager::ControllerManager>(executor, manager_node_name);
44+
rclcpp::NodeOptions cm_node_options = controller_manager::get_cm_node_options();
45+
std::vector<std::string> node_arguments = cm_node_options.arguments();
46+
for (int i = 1; i < argc; ++i)
47+
{
48+
if (node_arguments.empty() && std::string(argv[i]) != "--ros-args")
49+
{
50+
// A simple way to reject non ros args
51+
continue;
52+
}
53+
RCLCPP_INFO(rclcpp::get_logger("CM args"), "Adding argument: %s", argv[i]);
54+
node_arguments.push_back(argv[i]);
55+
}
56+
cm_node_options.arguments(node_arguments);
57+
58+
auto cm = std::make_shared<controller_manager::ControllerManager>(
59+
executor, manager_node_name, "", cm_node_options);
4560

4661
RCLCPP_INFO(cm->get_logger(), "update rate is %d Hz", cm->get_update_rate());
4762

0 commit comments

Comments
 (0)