112 lines
4.3 KiB
C++
112 lines
4.3 KiB
C++
#ifndef OSRM_EXTRACTOR_BEARING_CONTAINER_HPP
|
|
#define OSRM_EXTRACTOR_BEARING_CONTAINER_HPP
|
|
|
|
#include "storage/shared_memory_ownership.hpp"
|
|
#include "storage/tar_fwd.hpp"
|
|
|
|
#include "util/guidance/bearing_class.hpp"
|
|
#include "util/range_table.hpp"
|
|
#include "util/vector_view.hpp"
|
|
|
|
#include <numeric>
|
|
|
|
namespace osrm
|
|
{
|
|
namespace extractor
|
|
{
|
|
namespace detail
|
|
{
|
|
template <storage::Ownership Ownership> class IntersectionBearingsContainer;
|
|
}
|
|
|
|
namespace serialization
|
|
{
|
|
template <storage::Ownership Ownership>
|
|
void read(storage::tar::FileReader &reader,
|
|
const std::string &name,
|
|
detail::IntersectionBearingsContainer<Ownership> &turn_data);
|
|
|
|
template <storage::Ownership Ownership>
|
|
void write(storage::tar::FileWriter &writer,
|
|
const std::string &name,
|
|
const detail::IntersectionBearingsContainer<Ownership> &turn_data);
|
|
} // namespace serialization
|
|
|
|
namespace detail
|
|
{
|
|
template <storage::Ownership Ownership> class IntersectionBearingsContainer
|
|
{
|
|
template <typename T> using Vector = util::ViewOrVector<T, Ownership>;
|
|
template <unsigned size> using RangeTable = util::RangeTable<size, Ownership>;
|
|
|
|
public:
|
|
IntersectionBearingsContainer() = default;
|
|
IntersectionBearingsContainer(IntersectionBearingsContainer &&) = default;
|
|
IntersectionBearingsContainer(const IntersectionBearingsContainer &) = default;
|
|
IntersectionBearingsContainer &operator=(IntersectionBearingsContainer &&) = default;
|
|
IntersectionBearingsContainer &operator=(const IntersectionBearingsContainer &) = default;
|
|
|
|
IntersectionBearingsContainer(std::vector<BearingClassID> node_to_class_id_,
|
|
const std::vector<util::guidance::BearingClass> &bearing_classes)
|
|
: node_to_class_id(std::move(node_to_class_id_))
|
|
{
|
|
std::vector<unsigned> bearing_counts(bearing_classes.size());
|
|
std::transform(bearing_classes.begin(),
|
|
bearing_classes.end(),
|
|
bearing_counts.begin(),
|
|
[](const auto &bearings) { return bearings.getAvailableBearings().size(); });
|
|
// NOLINTNEXTLINE(bugprone-fold-init-type)
|
|
auto total_bearings = std::accumulate(bearing_counts.begin(), bearing_counts.end(), 0);
|
|
class_id_to_ranges_table = RangeTable<16>{bearing_counts};
|
|
|
|
values.reserve(total_bearings);
|
|
for (const auto &bearing_class : bearing_classes)
|
|
{
|
|
const auto &bearings = bearing_class.getAvailableBearings();
|
|
values.insert(values.end(), bearings.begin(), bearings.end());
|
|
}
|
|
}
|
|
|
|
IntersectionBearingsContainer(Vector<DiscreteBearing> values_,
|
|
Vector<BearingClassID> node_to_class_id_,
|
|
RangeTable<16> class_id_to_ranges_table_)
|
|
: values(std::move(values_)), node_to_class_id(std::move(node_to_class_id_)),
|
|
class_id_to_ranges_table(std::move(class_id_to_ranges_table_))
|
|
{
|
|
}
|
|
|
|
// Returns the bearing class for an intersection node
|
|
util::guidance::BearingClass GetBearingClass(const NodeID node) const
|
|
{
|
|
auto class_id = node_to_class_id[node];
|
|
auto range = class_id_to_ranges_table.GetRange(class_id);
|
|
util::guidance::BearingClass result;
|
|
std::for_each(values.begin() + range.front(),
|
|
values.begin() + range.back() + 1,
|
|
[&](const DiscreteBearing &bearing) { result.add(bearing); });
|
|
return result;
|
|
}
|
|
|
|
friend void serialization::read<Ownership>(storage::tar::FileReader &reader,
|
|
const std::string &name,
|
|
IntersectionBearingsContainer &turn_data_container);
|
|
friend void
|
|
serialization::write<Ownership>(storage::tar::FileWriter &writer,
|
|
const std::string &name,
|
|
const IntersectionBearingsContainer &turn_data_container);
|
|
|
|
private:
|
|
Vector<DiscreteBearing> values;
|
|
Vector<BearingClassID> node_to_class_id;
|
|
RangeTable<16> class_id_to_ranges_table;
|
|
};
|
|
} // namespace detail
|
|
|
|
using IntersectionBearingsContainer =
|
|
detail::IntersectionBearingsContainer<storage::Ownership::Container>;
|
|
using IntersectionBearingsView = detail::IntersectionBearingsContainer<storage::Ownership::View>;
|
|
} // namespace extractor
|
|
} // namespace osrm
|
|
|
|
#endif
|