db7946d762
This change adds support for disabling datasets, such that specific files are not loaded into memory when running OSRM. This enables users to not pay the memory cost for features they do not intend to use. Initially, there are two options: - ROUTE_GEOMETRY, for disabling overview, steps, annotations and waypoints. - ROUTE_STEPS, for disabling steps only. Attempts to query features for which the datasets are disabled will lead to a DisabledDatasetException being returned.
90 lines
2.6 KiB
C++
90 lines
2.6 KiB
C++
#ifndef OSRM_IO_CONFIG_HPP
|
|
#define OSRM_IO_CONFIG_HPP
|
|
|
|
#include "util/exception.hpp"
|
|
|
|
#include <array>
|
|
#include <boost/algorithm/string/predicate.hpp>
|
|
#include <boost/filesystem.hpp>
|
|
#include <boost/filesystem/path.hpp>
|
|
#include <string>
|
|
|
|
namespace osrm::storage
|
|
{
|
|
struct IOConfig
|
|
{
|
|
IOConfig(std::vector<boost::filesystem::path> required_input_files_,
|
|
std::vector<boost::filesystem::path> optional_input_files_,
|
|
std::vector<boost::filesystem::path> output_files_)
|
|
: required_input_files(std::move(required_input_files_)),
|
|
optional_input_files(std::move(optional_input_files_)),
|
|
output_files(std::move(output_files_))
|
|
{
|
|
}
|
|
|
|
bool IsValid() const;
|
|
std::vector<std::string> GetMissingFiles() const;
|
|
boost::filesystem::path GetPath(const std::string &fileName) const
|
|
{
|
|
if (!IsConfigured(fileName, required_input_files) &&
|
|
!IsConfigured(fileName, optional_input_files) && !IsConfigured(fileName, output_files))
|
|
{
|
|
throw util::exception("Tried to access file which is not configured: " + fileName);
|
|
}
|
|
|
|
return {base_path.string() + fileName};
|
|
}
|
|
|
|
bool IsRequiredConfiguredInput(const std::string &fileName) const
|
|
{
|
|
return IsConfigured(fileName, required_input_files);
|
|
}
|
|
|
|
boost::filesystem::path base_path;
|
|
|
|
protected:
|
|
// Infer the base path from the path of the .osrm file
|
|
void UseDefaultOutputNames(const boost::filesystem::path &base)
|
|
{
|
|
// potentially strip off the .osrm (or other) extensions for
|
|
// determining the base path=
|
|
std::string path = base.string();
|
|
|
|
std::array<std::string, 6> known_extensions{
|
|
{".osm.bz2", ".osm.pbf", ".osm.xml", ".pbf", ".osm", ".osrm"}};
|
|
for (const auto &ext : known_extensions)
|
|
{
|
|
const auto pos = path.find(ext);
|
|
if (pos != std::string::npos)
|
|
{
|
|
path.replace(pos, ext.size(), "");
|
|
break;
|
|
}
|
|
}
|
|
|
|
base_path = {path};
|
|
}
|
|
|
|
private:
|
|
static bool IsConfigured(const std::string &fileName,
|
|
const std::vector<boost::filesystem::path> &paths)
|
|
{
|
|
for (auto &path : paths)
|
|
{
|
|
if (boost::algorithm::ends_with(path.string(), fileName))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
std::vector<boost::filesystem::path> required_input_files;
|
|
std::vector<boost::filesystem::path> optional_input_files;
|
|
std::vector<boost::filesystem::path> output_files;
|
|
};
|
|
} // namespace osrm::storage
|
|
|
|
#endif
|