98 lines
3.5 KiB
C++
98 lines
3.5 KiB
C++
#ifndef OSRM_PARTITIONER_SERILIZATION_HPP
|
|
#define OSRM_PARTITIONER_SERILIZATION_HPP
|
|
|
|
#include "partitioner/serialization.hpp"
|
|
|
|
#include "storage/io.hpp"
|
|
|
|
namespace osrm::partitioner::files
|
|
{
|
|
|
|
// read .osrm.partition file
|
|
template <typename MultiLevelPartitionT>
|
|
inline void readPartition(const std::filesystem::path &path, MultiLevelPartitionT &mlp)
|
|
{
|
|
static_assert(std::is_same<MultiLevelPartitionView, MultiLevelPartitionT>::value ||
|
|
std::is_same<MultiLevelPartition, MultiLevelPartitionT>::value,
|
|
"");
|
|
|
|
const auto fingerprint = storage::tar::FileReader::VerifyFingerprint;
|
|
storage::tar::FileReader reader{path, fingerprint};
|
|
|
|
serialization::read(reader, "/mld/multilevelpartition", mlp);
|
|
}
|
|
|
|
// writes .osrm.partition file
|
|
template <typename MultiLevelPartitionT>
|
|
inline void writePartition(const std::filesystem::path &path, const MultiLevelPartitionT &mlp)
|
|
{
|
|
static_assert(std::is_same<MultiLevelPartitionView, MultiLevelPartitionT>::value ||
|
|
std::is_same<MultiLevelPartition, MultiLevelPartitionT>::value,
|
|
"");
|
|
|
|
const auto fingerprint = storage::tar::FileWriter::GenerateFingerprint;
|
|
storage::tar::FileWriter writer{path, fingerprint};
|
|
|
|
serialization::write(writer, "/mld/multilevelpartition", mlp);
|
|
}
|
|
|
|
// reads .osrm.cells file
|
|
template <typename CellStorageT>
|
|
inline void readCells(const std::filesystem::path &path, CellStorageT &storage)
|
|
{
|
|
static_assert(std::is_same<CellStorageView, CellStorageT>::value ||
|
|
std::is_same<CellStorage, CellStorageT>::value,
|
|
"");
|
|
|
|
const auto fingerprint = storage::tar::FileReader::VerifyFingerprint;
|
|
storage::tar::FileReader reader{path, fingerprint};
|
|
|
|
serialization::read(reader, "/mld/cellstorage", storage);
|
|
}
|
|
|
|
// writes .osrm.cells file
|
|
template <typename CellStorageT>
|
|
inline void writeCells(const std::filesystem::path &path, CellStorageT &storage)
|
|
{
|
|
static_assert(std::is_same<CellStorageView, CellStorageT>::value ||
|
|
std::is_same<CellStorage, CellStorageT>::value,
|
|
"");
|
|
|
|
const auto fingerprint = storage::tar::FileWriter::GenerateFingerprint;
|
|
storage::tar::FileWriter writer{path, fingerprint};
|
|
|
|
serialization::write(writer, "/mld/cellstorage", storage);
|
|
}
|
|
|
|
// reads .osrm.mldgr file
|
|
template <typename MultiLevelGraphT>
|
|
inline void readGraph(const std::filesystem::path &path,
|
|
MultiLevelGraphT &graph,
|
|
std::uint32_t &connectivity_checksum)
|
|
{
|
|
static_assert(std::is_same<partitioner::MultiLevelEdgeBasedGraph, MultiLevelGraphT>::value, "");
|
|
|
|
storage::tar::FileReader reader{path, storage::tar::FileReader::VerifyFingerprint};
|
|
|
|
reader.ReadInto("/mld/connectivity_checksum", connectivity_checksum);
|
|
serialization::read(reader, "/mld/multilevelgraph", graph);
|
|
}
|
|
|
|
// writes .osrm.mldgr file
|
|
template <typename MultiLevelGraphT>
|
|
inline void writeGraph(const std::filesystem::path &path,
|
|
const MultiLevelGraphT &graph,
|
|
const std::uint32_t connectivity_checksum)
|
|
{
|
|
static_assert(std::is_same<partitioner::MultiLevelEdgeBasedGraph, MultiLevelGraphT>::value, "");
|
|
|
|
storage::tar::FileWriter writer{path, storage::tar::FileWriter::GenerateFingerprint};
|
|
|
|
writer.WriteElementCount64("/mld/connectivity_checksum", 1);
|
|
writer.WriteFrom("/mld/connectivity_checksum", connectivity_checksum);
|
|
serialization::write(writer, "/mld/multilevelgraph", graph);
|
|
}
|
|
} // namespace osrm::partitioner::files
|
|
|
|
#endif
|