forked from osm2pgsql-dev/osm2pgsql
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathnode-persistent-cache.cpp
More file actions
83 lines (70 loc) · 2.28 KB
/
Copy pathnode-persistent-cache.cpp
File metadata and controls
83 lines (70 loc) · 2.28 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#define _LARGEFILE64_SOURCE /* See feature_test_macros(7) */
#include "format.hpp"
#include "node-persistent-cache.hpp"
#include "options.hpp"
void node_persistent_cache::set(osmid_t id, osmium::Location location)
{
if (id < 0) {
throw std::runtime_error{"Flatnode store cannot save negative IDs."};
}
m_index->set(static_cast<osmium::unsigned_object_id_type>(id), location);
}
osmium::Location node_persistent_cache::get(osmid_t id) const noexcept
{
if (id < 0) {
return osmium::Location{};
}
return m_index->get_noexcept(
static_cast<osmium::unsigned_object_id_type>(id));
}
std::size_t node_persistent_cache::get_list(osmium::WayNodeList *nodes) const
{
std::size_t count = 0;
for (auto &n : *nodes) {
auto loc = m_ram_cache->get(n.ref());
if (!loc.valid() && n.ref() >= 0) {
loc = m_index->get_noexcept(
static_cast<osmium::unsigned_object_id_type>(n.ref()));
}
n.set_location(loc);
if (loc.valid()) {
++count;
}
}
return count;
}
node_persistent_cache::node_persistent_cache(
const options_t *options, std::shared_ptr<node_ram_cache> ptr)
: m_ram_cache(std::move(ptr))
{
if (!options->flat_node_file) {
throw std::runtime_error{"Unable to set up persistent cache: the name "
"of the flat node file was not set."};
}
m_fname = options->flat_node_file->c_str();
m_remove_file = options->droptemp;
fmt::print(stderr, "Mid: loading persistent node cache from {}\n", m_fname);
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-vararg, hicpp-signed-bitwise)
m_fd = open(m_fname, O_RDWR | O_CREAT, 0644);
if (m_fd < 0) {
fmt::print(stderr, "Cannot open location cache file '{}': {}\n",
m_fname, std::strerror(errno));
throw std::runtime_error{"Unable to open flatnode file\n"};
}
m_index.reset(new index_t{m_fd});
}
node_persistent_cache::~node_persistent_cache() noexcept
{
m_index.reset();
if (m_fd >= 0) {
close(m_fd);
}
if (m_remove_file) {
try {
fmt::print(stderr, "Mid: removing persistent node cache at {}\n",
m_fname);
} catch (...) {
}
unlink(m_fname);
}
}