add humble-navigation2

navigation2/nav2_collision_monitor/src/range.cpp

@@ -0,0 +1,165 @@
+// Copyright (c) 2022 Samsung R&D Institute Russia
+//
+// 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.
+
+#include "nav2_collision_monitor/range.hpp"
+
+#include <math.h>
+#include <cmath>
+#include <functional>
+
+#include "nav2_util/node_utils.hpp"
+
+namespace nav2_collision_monitor
+{
+
+Range::Range(
+  const nav2_util::LifecycleNode::WeakPtr & node,
+  const std::string & source_name,
+  const std::shared_ptr<tf2_ros::Buffer> tf_buffer,
+  const std::string & base_frame_id,
+  const std::string & global_frame_id,
+  const tf2::Duration & transform_tolerance,
+  const rclcpp::Duration & source_timeout,
+  const bool base_shift_correction)
+: Source(
+    node, source_name, tf_buffer, base_frame_id, global_frame_id,
+    transform_tolerance, source_timeout, base_shift_correction),
+  data_(nullptr)
+{
+  RCLCPP_INFO(logger_, "[%s]: Creating Range", source_name_.c_str());
+}
+
+Range::~Range()
+{
+  RCLCPP_INFO(logger_, "[%s]: Destroying Range", source_name_.c_str());
+  data_sub_.reset();
+}
+
+void Range::configure()
+{
+  Source::configure();
+  auto node = node_.lock();
+  if (!node) {
+    throw std::runtime_error{"Failed to lock node"};
+  }
+
+  std::string source_topic;
+
+  getParameters(source_topic);
+
+  rclcpp::QoS range_qos = rclcpp::SensorDataQoS();  // set to default
+  data_sub_ = node->create_subscription<sensor_msgs::msg::Range>(
+    source_topic, range_qos,
+    std::bind(&Range::dataCallback, this, std::placeholders::_1));
+}
+
+void Range::getData(
+  const rclcpp::Time & curr_time,
+  std::vector<Point> & data) const
+{
+  // Ignore data from the source if it is not being published yet or
+  // not being published for a long time
+  if (data_ == nullptr) {
+    return;
+  }
+  if (!sourceValid(data_->header.stamp, curr_time)) {
+    return;
+  }
+
+  // Ignore data, if its range is out of scope of range sensor abilities
+  if (data_->range < data_->min_range || data_->range > data_->max_range) {
+    RCLCPP_DEBUG(
+      logger_,
+      "[%s]: Data range %fm is out of {%f..%f} sensor span. Ignoring...",
+      source_name_.c_str(), data_->range, data_->min_range, data_->max_range);
+    return;
+  }
+
+  tf2::Transform tf_transform;
+  if (base_shift_correction_) {
+    // Obtaining the transform to get data from source frame and time where it was received
+    // to the base frame and current time
+    if (
+      !nav2_util::getTransform(
+        data_->header.frame_id, data_->header.stamp,
+        base_frame_id_, curr_time, global_frame_id_,
+        transform_tolerance_, tf_buffer_, tf_transform))
+    {
+      return;
+    }
+  } else {
+    // Obtaining the transform to get data from source frame to base frame without time shift
+    // considered. Less accurate but much more faster option not dependent on state estimation
+    // frames.
+    if (
+      !nav2_util::getTransform(
+        data_->header.frame_id, base_frame_id_,
+        transform_tolerance_, tf_buffer_, tf_transform))
+    {
+      return;
+    }
+  }
+
+  // Calculate poses and refill data array
+  float angle;
+  for (
+    angle = -data_->field_of_view / 2;
+    angle < data_->field_of_view / 2;
+    angle += obstacles_angle_)
+  {
+    // Transform point coordinates from source frame -> to base frame
+    tf2::Vector3 p_v3_s(
+      data_->range * std::cos(angle),
+      data_->range * std::sin(angle),
+      0.0);
+    tf2::Vector3 p_v3_b = tf_transform * p_v3_s;
+
+    // Refill data array
+    data.push_back({p_v3_b.x(), p_v3_b.y()});
+  }
+
+  // Make sure that last (field_of_view / 2) point will be in the data array
+  angle = data_->field_of_view / 2;
+
+  // Transform point coordinates from source frame -> to base frame
+  tf2::Vector3 p_v3_s(
+    data_->range * std::cos(angle),
+    data_->range * std::sin(angle),
+    0.0);
+  tf2::Vector3 p_v3_b = tf_transform * p_v3_s;
+
+  // Refill data array
+  data.push_back({p_v3_b.x(), p_v3_b.y()});
+}
+
+void Range::getParameters(std::string & source_topic)
+{
+  auto node = node_.lock();
+  if (!node) {
+    throw std::runtime_error{"Failed to lock node"};
+  }
+
+  getCommonParameters(source_topic);
+
+  nav2_util::declare_parameter_if_not_declared(
+    node, source_name_ + ".obstacles_angle", rclcpp::ParameterValue(M_PI / 180));
+  obstacles_angle_ = node->get_parameter(source_name_ + ".obstacles_angle").as_double();
+}
+
+void Range::dataCallback(sensor_msgs::msg::Range::ConstSharedPtr msg)
+{
+  data_ = msg;
+}
+
+}  // namespace nav2_collision_monitor