cache.hpp

Go to the documentation of this file.

// Copyright 2015 TIER IV, Inc. All rights reserved.
//
// 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 TRAFFIC_SIMULATOR__HDMAP_UTILS__CACHE_HPP_
#define TRAFFIC_SIMULATOR__HDMAP_UTILS__CACHE_HPP_
 
#include <geometry/spline/catmull_rom_spline.hpp>
#include <geometry_msgs/msg/point.hpp>
#include <mutex>
#include <optional>
#include <scenario_simulator_exception/exception.hpp>
#include <traffic_simulator/lanelet_wrapper/lanelet_wrapper.hpp>
#include <unordered_map>
#include <vector>
 
namespace hdmap_utils
{
class RouteCache
{
public:
  auto exists(lanelet::Id from, lanelet::Id to, bool allow_lane_change)
  {
    std::lock_guard lock(mutex_);
    std::tuple<lanelet::Id, lanelet::Id, bool> key = {from, to, allow_lane_change};
    return data_.find(key) != data_.end();
  }
 
  auto getRoute(const lanelet::Id from, const lanelet::Id to, const bool allow_lane_change)
    -> decltype(auto)
  {
    if (!exists(from, to, allow_lane_change)) {
      THROW_SIMULATION_ERROR(
        "route from : ", from, " to : ", to, (allow_lane_change ? " with" : " without"),
        " lane change does not exists on route cache.");
    } else {
      std::lock_guard lock(mutex_);
      return data_.at({from, to, allow_lane_change});
    }
  }
 
  auto appendData(
    lanelet::Id from, lanelet::Id to, const bool allow_lane_change, const lanelet::Ids & route)
    -> void
  {
    std::lock_guard lock(mutex_);
    data_[{from, to, allow_lane_change}] = route;
  }
 
private:
  std::unordered_map<std::tuple<lanelet::Id, lanelet::Id, bool>, lanelet::Ids> data_;
 
  std::mutex mutex_;
};
 
class CenterPointsCache
{
public:
  auto exists(lanelet::Id lanelet_id) -> bool
  {
    std::lock_guard lock(mutex_);
    return data_.find(lanelet_id) != data_.end();
  }
 
  auto getCenterPoints(lanelet::Id lanelet_id) -> decltype(auto)
  {
    if (!exists(lanelet_id)) {
      THROW_SIMULATION_ERROR("center point of : ", lanelet_id, " does not exists on route cache.");
    }
    std::lock_guard lock(mutex_);
    return data_.at(lanelet_id);
  }
 
  auto getCenterPointsSpline(lanelet::Id lanelet_id) -> decltype(auto)
  {
    if (!exists(lanelet_id)) {
      THROW_SIMULATION_ERROR("center point of : ", lanelet_id, " does not exists on route cache.");
    }
    std::lock_guard lock(mutex_);
    return splines_[lanelet_id];
  }
 
  auto appendData(lanelet::Id lanelet_id, const std::vector<geometry_msgs::msg::Point> & route)
  {
    std::lock_guard lock(mutex_);
    data_[lanelet_id] = route;
    splines_[lanelet_id] = std::make_shared<math::geometry::CatmullRomSpline>(route);
  }
 
private:
  std::unordered_map<lanelet::Id, std::vector<geometry_msgs::msg::Point>> data_;
 
  std::unordered_map<lanelet::Id, std::shared_ptr<math::geometry::CatmullRomSpline>> splines_;
 
  std::mutex mutex_;
};
 
class LaneletLengthCache
{
public:
  auto exists(lanelet::Id lanelet_id)
  {
    std::lock_guard lock(mutex_);
    return data_.find(lanelet_id) != data_.end();
  }
 
  auto getLength(lanelet::Id lanelet_id)
  {
    if (!exists(lanelet_id)) {
      THROW_SIMULATION_ERROR("length of : ", lanelet_id, " does not exists on route cache.");
    }
    std::lock_guard lock(mutex_);
    return data_[lanelet_id];
  }
 
  auto appendData(lanelet::Id lanelet_id, double length)
  {
    std::lock_guard lock(mutex_);
    data_[lanelet_id] = length;
  }
 
private:
  std::unordered_map<lanelet::Id, double> data_;
 
  std::mutex mutex_;
};
}  // namespace hdmap_utils
 
#endif  // TRAFFIC_SIMULATOR__HDMAP_UTILS__CACHE_HPP_