Program Listing for File laser_scan.hpp

↰ Return to documentation for file (include/beluga_ros/laser_scan.hpp)

// Copyright 2023 Ekumen, 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.

#ifndef BELUGA_ROS_LASER_SCAN_HPP
#define BELUGA_ROS_LASER_SCAN_HPP

#include <range/v3/view/iota.hpp>

#include <beluga/sensor/data/laser_scan.hpp>
#include <beluga/views/take_evenly.hpp>
#include <beluga_ros/messages.hpp>

#include <sophus/se3.hpp>

namespace beluga_ros {

class LaserScan : public beluga::BaseLaserScan<LaserScan> {
 public:
  using Scalar = double;

  explicit LaserScan(
      beluga_ros::msg::LaserScanConstSharedPtr scan,
      Sophus::SE3d origin = Sophus::SE3d(),
      std::size_t max_beams = std::numeric_limits<std::size_t>::max(),
      Scalar min_range = std::numeric_limits<Scalar>::min(),
      Scalar max_range = std::numeric_limits<Scalar>::max())
      : scan_(std::move(scan)),
        origin_(std::move(origin)),
        max_beams_(max_beams),
        min_range_(std::max(static_cast<Scalar>(scan_->range_min), min_range)),
        max_range_(std::min(static_cast<Scalar>(scan_->range_max), max_range)) {
    assert(scan_ != nullptr);
  }

  [[nodiscard]] const auto& origin() const { return origin_; }

  [[nodiscard]] auto angles() const {
    return ranges::views::iota(0, static_cast<int>(scan_->ranges.size())) | beluga::views::take_evenly(max_beams_) |
           ranges::views::transform([this](int i) {
             return static_cast<Scalar>(scan_->angle_min + static_cast<float>(i) * scan_->angle_increment);
           });
  }

  [[nodiscard]] auto ranges() const {
    return scan_->ranges | beluga::views::take_evenly(max_beams_) |
           ranges::views::transform([](auto value) { return static_cast<Scalar>(value); });
  }

  [[nodiscard]] auto min_range() const { return min_range_; }

  [[nodiscard]] auto max_range() const { return max_range_; }

 private:
  beluga_ros::msg::LaserScanConstSharedPtr scan_;
  Sophus::SE3d origin_;
  std::size_t max_beams_;
  Scalar min_range_;
  Scalar max_range_;
};

}  // namespace beluga_ros

#endif  // BELUGA_ROS_LASER_SCAN_HPP