81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
#ifndef OSRM_CUSTOMIZER_FILES_HPP
|
|
#define OSRM_CUSTOMIZER_FILES_HPP
|
|
|
|
#include "customizer/serialization.hpp"
|
|
|
|
#include "storage/tar.hpp"
|
|
|
|
#include "util/integer_range.hpp"
|
|
|
|
#include <unordered_map>
|
|
|
|
namespace osrm
|
|
{
|
|
namespace customizer
|
|
{
|
|
namespace files
|
|
{
|
|
|
|
// reads .osrm.cell_metrics file
|
|
template <typename CellMetricT>
|
|
inline void readCellMetrics(const boost::filesystem::path &path,
|
|
std::unordered_map<std::string, std::vector<CellMetricT>> &metrics)
|
|
{
|
|
static_assert(std::is_same<CellMetricView, CellMetricT>::value ||
|
|
std::is_same<CellMetric, CellMetricT>::value,
|
|
"");
|
|
|
|
const auto fingerprint = storage::tar::FileReader::VerifyFingerprint;
|
|
storage::tar::FileReader reader{path, fingerprint};
|
|
|
|
for (auto &pair : metrics)
|
|
{
|
|
const auto &metric_name = pair.first;
|
|
auto &metric_exclude_classes = pair.second;
|
|
|
|
auto prefix = "/mld/metrics/" + metric_name + "/exclude";
|
|
auto num_exclude_classes = reader.ReadElementCount64(prefix);
|
|
metric_exclude_classes.resize(num_exclude_classes);
|
|
|
|
auto id = 0;
|
|
for (auto &metric : metric_exclude_classes)
|
|
{
|
|
serialization::read(reader, prefix + "/" + std::to_string(id++), metric);
|
|
}
|
|
}
|
|
}
|
|
|
|
// writes .osrm.cell_metrics file
|
|
template <typename CellMetricT>
|
|
inline void
|
|
writeCellMetrics(const boost::filesystem::path &path,
|
|
const std::unordered_map<std::string, std::vector<CellMetricT>> &metrics)
|
|
{
|
|
static_assert(std::is_same<CellMetricView, CellMetricT>::value ||
|
|
std::is_same<CellMetric, CellMetricT>::value,
|
|
"");
|
|
|
|
const auto fingerprint = storage::tar::FileWriter::GenerateFingerprint;
|
|
storage::tar::FileWriter writer{path, fingerprint};
|
|
|
|
for (const auto &pair : metrics)
|
|
{
|
|
const auto &metric_name = pair.first;
|
|
const auto &metric_exclude_classes = pair.second;
|
|
|
|
auto prefix = "/mld/metrics/" + metric_name + "/exclude";
|
|
writer.WriteElementCount64(prefix, metric_exclude_classes.size());
|
|
|
|
auto id = 0;
|
|
for (auto &exclude_metric : metric_exclude_classes)
|
|
{
|
|
serialization::write(writer, prefix + "/" + std::to_string(id++), exclude_metric);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif
|