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
84
|
#define _LARGEFILE64_SOURCE /* See feature_test_macrors(7) */
#include "node-persistent-cache.hpp"
#include "options.hpp"
void node_persistent_cache::set(osmid_t id, const osmium::Location &coord)
{
if (id < 0) {
throw std::runtime_error("Flatnode store cannot save negative IDs.");
}
m_index->set(static_cast<osmium::unsigned_object_id_type>(id), coord);
}
osmium::Location node_persistent_cache::get(osmid_t id)
{
if (id >= 0) {
try {
return m_index->get(
static_cast<osmium::unsigned_object_id_type>(id));
} catch (osmium::not_found const &) {
}
}
return osmium::Location();
}
size_t node_persistent_cache::get_list(osmium::WayNodeList *nodes)
{
size_t count = 0;
for (auto &n : *nodes) {
auto loc = m_ram_cache->get(n.ref());
/* Check cache first */
if (!loc.valid() && n.ref() >= 0) {
try {
loc = m_index->get(
static_cast<osmium::unsigned_object_id_type>(n.ref()));
} catch (osmium::not_found const &) {
}
}
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(ptr), m_fd(-1)
{
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;
fprintf(stderr, "Mid: loading persistent node cache from %s\n", m_fname);
m_fd = open(m_fname, O_RDWR | O_CREAT, 0644);
if (m_fd < 0) {
fprintf(stderr, "Cannot open location cache file '%s': %s\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()
{
m_index.reset();
if (m_fd >= 0) {
close(m_fd);
}
if (m_remove_file) {
fprintf(stderr, "Mid: removing persistent node cache at %s\n", m_fname);
unlink(m_fname);
}
}
|
