225 lines
7.7 KiB
C++
225 lines
7.7 KiB
C++
#ifndef OSRM_EXTRACTOR_SEGMENT_DATA_CONTAINER_HPP_
|
|
#define OSRM_EXTRACTOR_SEGMENT_DATA_CONTAINER_HPP_
|
|
|
|
#include "util/packed_vector.hpp"
|
|
#include "util/typedefs.hpp"
|
|
#include "util/vector_view.hpp"
|
|
|
|
#include "storage/shared_memory_ownership.hpp"
|
|
#include "storage/tar_fwd.hpp"
|
|
#include <filesystem>
|
|
#include <ranges>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
namespace osrm::extractor
|
|
{
|
|
|
|
class CompressedEdgeContainer;
|
|
|
|
namespace detail
|
|
{
|
|
template <storage::Ownership Ownership> class SegmentDataContainerImpl;
|
|
} // namespace detail
|
|
|
|
namespace serialization
|
|
{
|
|
template <storage::Ownership Ownership>
|
|
inline void read(storage::tar::FileReader &reader,
|
|
const std::string &name,
|
|
detail::SegmentDataContainerImpl<Ownership> &segment_data);
|
|
template <storage::Ownership Ownership>
|
|
inline void write(storage::tar::FileWriter &writer,
|
|
const std::string &name,
|
|
const detail::SegmentDataContainerImpl<Ownership> &segment_data);
|
|
} // namespace serialization
|
|
|
|
namespace detail
|
|
{
|
|
template <storage::Ownership Ownership> class SegmentDataContainerImpl
|
|
{
|
|
template <typename T> using Vector = util::ViewOrVector<T, Ownership>;
|
|
template <typename T, std::size_t Bits>
|
|
using PackedVector = util::detail::PackedVector<T, Bits, Ownership>;
|
|
|
|
friend CompressedEdgeContainer;
|
|
|
|
public:
|
|
// FIXME We should change the indexing to Edge-Based-Node id
|
|
using DirectionalGeometryID = std::uint32_t;
|
|
using SegmentOffset = std::uint32_t;
|
|
using SegmentNodeVector = Vector<NodeID>;
|
|
using SegmentWeightVector = PackedVector<SegmentWeight, SEGMENT_WEIGHT_BITS>;
|
|
using SegmentDurationVector = PackedVector<SegmentDuration, SEGMENT_DURATION_BITS>;
|
|
using SegmentDatasourceVector = Vector<DatasourceID>;
|
|
|
|
SegmentDataContainerImpl() = default;
|
|
|
|
SegmentDataContainerImpl(Vector<std::uint32_t> index_,
|
|
SegmentNodeVector nodes_,
|
|
SegmentWeightVector fwd_weights_,
|
|
SegmentWeightVector rev_weights_,
|
|
SegmentDurationVector fwd_durations_,
|
|
SegmentDurationVector rev_durations_,
|
|
SegmentDatasourceVector fwd_datasources_,
|
|
SegmentDatasourceVector rev_datasources_)
|
|
: index(std::move(index_)), nodes(std::move(nodes_)), fwd_weights(std::move(fwd_weights_)),
|
|
rev_weights(std::move(rev_weights_)), fwd_durations(std::move(fwd_durations_)),
|
|
rev_durations(std::move(rev_durations_)), fwd_datasources(std::move(fwd_datasources_)),
|
|
rev_datasources(std::move(rev_datasources_))
|
|
{
|
|
}
|
|
|
|
auto GetForwardGeometry(const DirectionalGeometryID id)
|
|
{
|
|
const auto begin = nodes.begin() + index[id];
|
|
const auto end = nodes.begin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseGeometry(const DirectionalGeometryID id)
|
|
{
|
|
return GetForwardGeometry(id) | std::views::reverse;
|
|
}
|
|
|
|
auto GetForwardDurations(const DirectionalGeometryID id)
|
|
{
|
|
const auto begin = fwd_durations.begin() + index[id] + 1;
|
|
const auto end = fwd_durations.begin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseDurations(const DirectionalGeometryID id)
|
|
{
|
|
const auto begin = rev_durations.begin() + index[id];
|
|
const auto end = rev_durations.begin() + index[id + 1] - 1;
|
|
|
|
return std::ranges::subrange(begin, end) | std::views::reverse;
|
|
}
|
|
|
|
auto GetForwardWeights(const DirectionalGeometryID id)
|
|
{
|
|
const auto begin = fwd_weights.begin() + index[id] + 1;
|
|
const auto end = fwd_weights.begin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseWeights(const DirectionalGeometryID id)
|
|
{
|
|
const auto begin = rev_weights.begin() + index[id];
|
|
const auto end = rev_weights.begin() + index[id + 1] - 1;
|
|
|
|
return std::ranges::subrange(begin, end) | std::views::reverse;
|
|
}
|
|
|
|
auto GetForwardDatasources(const DirectionalGeometryID id)
|
|
{
|
|
const auto begin = fwd_datasources.begin() + index[id] + 1;
|
|
const auto end = fwd_datasources.begin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseDatasources(const DirectionalGeometryID id)
|
|
{
|
|
const auto begin = rev_datasources.begin() + index[id];
|
|
const auto end = rev_datasources.begin() + index[id + 1] - 1;
|
|
|
|
return std::ranges::subrange(begin, end) | std::views::reverse;
|
|
}
|
|
|
|
auto GetForwardGeometry(const DirectionalGeometryID id) const
|
|
{
|
|
const auto begin = nodes.cbegin() + index[id];
|
|
const auto end = nodes.cbegin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseGeometry(const DirectionalGeometryID id) const
|
|
{
|
|
return GetForwardGeometry(id) | std::views::reverse;
|
|
}
|
|
|
|
auto GetForwardDurations(const DirectionalGeometryID id) const
|
|
{
|
|
const auto begin = fwd_durations.cbegin() + index[id] + 1;
|
|
const auto end = fwd_durations.cbegin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseDurations(const DirectionalGeometryID id) const
|
|
{
|
|
const auto begin = rev_durations.cbegin() + index[id];
|
|
const auto end = rev_durations.cbegin() + index[id + 1] - 1;
|
|
|
|
return std::ranges::subrange(begin, end) | std::views::reverse;
|
|
}
|
|
|
|
auto GetForwardWeights(const DirectionalGeometryID id) const
|
|
{
|
|
const auto begin = fwd_weights.cbegin() + index[id] + 1;
|
|
const auto end = fwd_weights.cbegin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseWeights(const DirectionalGeometryID id) const
|
|
{
|
|
const auto begin = rev_weights.cbegin() + index[id];
|
|
const auto end = rev_weights.cbegin() + index[id + 1] - 1;
|
|
|
|
return std::ranges::subrange(begin, end) | std::views::reverse;
|
|
}
|
|
|
|
auto GetForwardDatasources(const DirectionalGeometryID id) const
|
|
{
|
|
const auto begin = fwd_datasources.cbegin() + index[id] + 1;
|
|
const auto end = fwd_datasources.cbegin() + index[id + 1];
|
|
|
|
return std::ranges::subrange(begin, end);
|
|
}
|
|
|
|
auto GetReverseDatasources(const DirectionalGeometryID id) const
|
|
{
|
|
const auto begin = rev_datasources.cbegin() + index[id];
|
|
const auto end = rev_datasources.cbegin() + index[id + 1] - 1;
|
|
|
|
return std::ranges::subrange(begin, end) | std::views::reverse;
|
|
}
|
|
|
|
auto GetNumberOfGeometries() const { return index.size() - 1; }
|
|
auto GetNumberOfSegments() const { return fwd_weights.size(); }
|
|
|
|
friend void
|
|
serialization::read<Ownership>(storage::tar::FileReader &reader,
|
|
const std::string &name,
|
|
detail::SegmentDataContainerImpl<Ownership> &segment_data);
|
|
friend void serialization::write<Ownership>(
|
|
storage::tar::FileWriter &writer,
|
|
const std::string &name,
|
|
const detail::SegmentDataContainerImpl<Ownership> &segment_data);
|
|
|
|
private:
|
|
Vector<std::uint32_t> index;
|
|
SegmentNodeVector nodes;
|
|
SegmentWeightVector fwd_weights;
|
|
SegmentWeightVector rev_weights;
|
|
SegmentDurationVector fwd_durations;
|
|
SegmentDurationVector rev_durations;
|
|
SegmentDatasourceVector fwd_datasources;
|
|
SegmentDatasourceVector rev_datasources;
|
|
};
|
|
} // namespace detail
|
|
|
|
using SegmentDataView = detail::SegmentDataContainerImpl<storage::Ownership::View>;
|
|
using SegmentDataContainer = detail::SegmentDataContainerImpl<storage::Ownership::Container>;
|
|
} // namespace osrm::extractor
|
|
|
|
#endif
|