Skip to content

Commit

Permalink
feat(rtc_interface): apply multithread for rtc_interface (tier4#1275)
Browse files Browse the repository at this point in the history
Signed-off-by: Fumiya Watanabe <[email protected]>
  • Loading branch information
rej55 authored and boyali committed Oct 3, 2022
1 parent c0a11f6 commit 016ed20
Show file tree
Hide file tree
Showing 10 changed files with 42 additions and 23 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ class AvoidanceModule : public SceneModuleInterface
rtc_interface_right_.publishCooperateStatus(clock_->now());
}

bool isActivated() const override
bool isActivated() override
{
if (rtc_interface_left_.isRegistered(uuid_left_)) {
return rtc_interface_left_.isActivated(uuid_left_);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ class LaneChangeModule : public SceneModuleInterface
rtc_interface_right_.publishCooperateStatus(clock_->now());
}

bool isActivated() const override
bool isActivated() override
{
if (rtc_interface_left_.isRegistered(uuid_left_)) {
return rtc_interface_left_.isActivated(uuid_left_);
Expand All @@ -124,6 +124,8 @@ class LaneChangeModule : public SceneModuleInterface
UUID uuid_left_;
UUID uuid_right_;

bool is_activated_ = false;

void waitApprovalLeft(const double distance)
{
rtc_interface_left_.updateCooperateStatus(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -203,7 +203,7 @@ class SceneModuleInterface
/**
* @brief Return true if the activation command is received
*/
virtual bool isActivated() const
virtual bool isActivated()
{
if (!rtc_interface_ptr_) {
return true;
Expand Down Expand Up @@ -270,7 +270,7 @@ class SceneModuleInterface

void clearWaitingApproval() { is_waiting_approval_ = false; }

UUID generateUUID()
static UUID generateUUID()
{
// Generate random number
UUID uuid;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,8 @@ AvoidanceModule::AvoidanceModule(
const std::string & name, rclcpp::Node & node, const AvoidanceParameters & parameters)
: SceneModuleInterface{name, node},
parameters_{parameters},
rtc_interface_left_(node, "avoidance_left"),
rtc_interface_right_(node, "avoidance_right"),
rtc_interface_left_(&node, "avoidance_left"),
rtc_interface_right_(&node, "avoidance_right"),
uuid_left_{generateUUID()},
uuid_right_{generateUUID()}
{
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,8 @@ LaneChangeModule::LaneChangeModule(
const std::string & name, rclcpp::Node & node, const LaneChangeParameters & parameters)
: SceneModuleInterface{name, node},
parameters_{parameters},
rtc_interface_left_(node, "lane_change_left"),
rtc_interface_right_(node, "lane_change_right"),
rtc_interface_left_(&node, "lane_change_left"),
rtc_interface_right_(&node, "lane_change_right"),
uuid_left_{generateUUID()},
uuid_right_{generateUUID()}
{
Expand All @@ -50,6 +50,7 @@ BehaviorModuleOutput LaneChangeModule::run()
{
RCLCPP_DEBUG(getLogger(), "Was waiting approval, and now approved. Do plan().");
current_state_ = BT::NodeStatus::RUNNING;
is_activated_ = isActivated();
const auto output = plan();
const auto turn_signal_info = output.turn_signal_info;
if (turn_signal_info.turn_signal.command == TurnIndicatorsCommand::ENABLE_LEFT) {
Expand Down Expand Up @@ -475,7 +476,7 @@ bool LaneChangeModule::isAbortConditionSatisfied() const
return false;
}

if (!isActivated()) {
if (!is_activated_) {
return false;
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ PullOutModule::PullOutModule(
const std::string & name, rclcpp::Node & node, const PullOutParameters & parameters)
: SceneModuleInterface{name, node}, parameters_{parameters}
{
rtc_interface_ptr_ = std::make_shared<RTCInterface>(node, "pull_out");
rtc_interface_ptr_ = std::make_shared<RTCInterface>(&node, "pull_out");
}

BehaviorModuleOutput PullOutModule::run()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ PullOverModule::PullOverModule(
const std::string & name, rclcpp::Node & node, const PullOverParameters & parameters)
: SceneModuleInterface{name, node}, parameters_{parameters}
{
rtc_interface_ptr_ = std::make_shared<RTCInterface>(node, "pull_over");
rtc_interface_ptr_ = std::make_shared<RTCInterface>(&node, "pull_over");
}

BehaviorModuleOutput PullOverModule::run()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -269,7 +269,7 @@ class SceneModuleManagerInterfaceWithRTC : public SceneModuleManagerInterface
{
public:
SceneModuleManagerInterfaceWithRTC(rclcpp::Node & node, const char * module_name)
: SceneModuleManagerInterface(node, module_name), rtc_interface_(node, module_name)
: SceneModuleManagerInterface(node, module_name), rtc_interface_(&node, module_name)
{
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
#include "tier4_rtc_msgs/srv/cooperate_commands.hpp"
#include <unique_identifier_msgs/msg/uuid.hpp>

#include <mutex>
#include <string>
#include <vector>

Expand All @@ -43,14 +44,14 @@ using unique_identifier_msgs::msg::UUID;
class RTCInterface
{
public:
RTCInterface(rclcpp::Node & node, const std::string & name);
RTCInterface(rclcpp::Node * node, const std::string & name);
void publishCooperateStatus(const rclcpp::Time & stamp);
void updateCooperateStatus(
const UUID & uuid, const bool safe, const double distance, const rclcpp::Time & stamp);
void removeCooperateStatus(const UUID & uuid);
void clearCooperateStatus();
bool isActivated(const UUID & uuid) const;
bool isRegistered(const UUID & uuid) const;
bool isActivated(const UUID & uuid);
bool isRegistered(const UUID & uuid);

private:
void onCooperateCommandService(
Expand All @@ -61,6 +62,8 @@ class RTCInterface
rclcpp::Publisher<CooperateStatusArray>::SharedPtr pub_statuses_;
rclcpp::Service<CooperateCommands>::SharedPtr srv_commands_;

std::mutex mutex_;
rclcpp::CallbackGroup::SharedPtr callback_group_;
rclcpp::Logger logger_;
Module module_;
CooperateStatusArray registered_status_;
Expand Down
29 changes: 21 additions & 8 deletions planning/rtc_interface/src/rtc_interface.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -68,26 +68,30 @@ Module getModuleType(const std::string & module_name)

namespace rtc_interface
{
RTCInterface::RTCInterface(rclcpp::Node & node, const std::string & name)
: logger_{node.get_logger().get_child("RTCInterface[" + name + "]")}
RTCInterface::RTCInterface(rclcpp::Node * node, const std::string & name)
: logger_{node->get_logger().get_child("RTCInterface[" + name + "]")}
{
using std::placeholders::_1;
using std::placeholders::_2;

// Publisher
pub_statuses_ = node.create_publisher<CooperateStatusArray>("~/" + name + "/cooperate_status", 1);
pub_statuses_ =
node->create_publisher<CooperateStatusArray>("~/" + name + "/cooperate_status", 1);

// Service
srv_commands_ = node.create_service<CooperateCommands>(
callback_group_ = node->create_callback_group(rclcpp::CallbackGroupType::MutuallyExclusive);
srv_commands_ = node->create_service<CooperateCommands>(
"~/" + name + "/cooperate_commands",
std::bind(&RTCInterface::onCooperateCommandService, this, _1, _2));
std::bind(&RTCInterface::onCooperateCommandService, this, _1, _2),
rmw_qos_profile_services_default, callback_group_);

// Module
module_ = getModuleType(name);
}

void RTCInterface::publishCooperateStatus(const rclcpp::Time & stamp)
{
std::lock_guard<std::mutex> lock(mutex_);
registered_status_.stamp = stamp;
pub_statuses_->publish(registered_status_);
}
Expand All @@ -96,6 +100,7 @@ void RTCInterface::onCooperateCommandService(
const CooperateCommands::Request::SharedPtr request,
const CooperateCommands::Response::SharedPtr responses)
{
std::lock_guard<std::mutex> lock(mutex_);
for (const auto & command : request->commands) {
CooperateResponse response;
response.uuid = command.uuid;
Expand All @@ -122,6 +127,7 @@ void RTCInterface::onCooperateCommandService(
void RTCInterface::updateCooperateStatus(
const UUID & uuid, const bool safe, const double distance, const rclcpp::Time & stamp)
{
std::lock_guard<std::mutex> lock(mutex_);
// Find registered status which has same uuid
auto itr = std::find_if(
registered_status_.statuses.begin(), registered_status_.statuses.end(),
Expand All @@ -148,6 +154,7 @@ void RTCInterface::updateCooperateStatus(

void RTCInterface::removeCooperateStatus(const UUID & uuid)
{
std::lock_guard<std::mutex> lock(mutex_);
// Find registered status which has same uuid and erase it
const auto itr = std::find_if(
registered_status_.statuses.begin(), registered_status_.statuses.end(),
Expand All @@ -163,10 +170,15 @@ void RTCInterface::removeCooperateStatus(const UUID & uuid)
"[removeCooperateStatus] uuid : " << to_string(uuid) << " is not found." << std::endl);
}

void RTCInterface::clearCooperateStatus() { registered_status_.statuses.clear(); }
void RTCInterface::clearCooperateStatus()
{
std::lock_guard<std::mutex> lock(mutex_);
registered_status_.statuses.clear();
}

bool RTCInterface::isActivated(const UUID & uuid) const
bool RTCInterface::isActivated(const UUID & uuid)
{
std::lock_guard<std::mutex> lock(mutex_);
const auto itr = std::find_if(
registered_status_.statuses.begin(), registered_status_.statuses.end(),
[uuid](auto & s) { return s.uuid == uuid; });
Expand All @@ -180,8 +192,9 @@ bool RTCInterface::isActivated(const UUID & uuid) const
return false;
}

bool RTCInterface::isRegistered(const UUID & uuid) const
bool RTCInterface::isRegistered(const UUID & uuid)
{
std::lock_guard<std::mutex> lock(mutex_);
const auto itr = std::find_if(
registered_status_.statuses.begin(), registered_status_.statuses.end(),
[uuid](auto & s) { return s.uuid == uuid; });
Expand Down

0 comments on commit 016ed20

Please sign in to comment.