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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
/* SPDX-License-Identifier: BSD-2-Clause
*
* This file is part of pyosmium. (https://osmcode.org/pyosmium/)
*
* Copyright (C) 2025 Sarah Hoffmann <lonvia@denofr.de> and others.
* For a full list of authors see the git log.
*/
#include <pybind11/pybind11.h>
#include <osmium/area/assembler.hpp>
#include <osmium/area/multipolygon_manager.hpp>
#include "base_handler.h"
#include "handler_chain.h"
#include "buffer_iterator.h"
namespace py = pybind11;
namespace {
using MpManager = osmium::area::MultipolygonManager<osmium::area::Assembler>;
class AreaManagerSecondPassHandlerBase : public pyosmium::BaseHandler
{
public:
AreaManagerSecondPassHandlerBase(MpManager *mp_manager)
: m_mp_manager(mp_manager)
{
m_enabled_for = osmium::osm_entity_bits::nwr;
}
bool node(pyosmium::PyOSMNode &n) override
{
m_mp_manager->handle_node(*(n.get()));
return false;
}
bool way(pyosmium::PyOSMWay &w) override
{
m_mp_manager->handle_way(*(w.get()));
return false;
}
bool relation(pyosmium::PyOSMRelation &r) override
{
m_mp_manager->handle_relation(*(r.get()));
return false;
}
void flush() override
{
m_mp_manager->flush_output();
}
protected:
MpManager *m_mp_manager;
};
class AreaManagerSecondPassHandler : public AreaManagerSecondPassHandlerBase
{
public:
AreaManagerSecondPassHandler(MpManager *mp_manager, py::args args)
: AreaManagerSecondPassHandlerBase(mp_manager), m_args(args), m_handlers(m_args)
{
m_mp_manager->set_callback([this](osmium::memory::Buffer &&buffer) {
for (auto &obj : buffer.select<osmium::Area>()) {
pyosmium::PyOSMArea area{&obj};
this->m_handlers.area(area);
}
});
}
private:
py::args m_args;
pyosmium::HandlerChain m_handlers;
};
class AreaManagerBufferHandler : public AreaManagerSecondPassHandlerBase
{
public:
AreaManagerBufferHandler(MpManager *mp_manager, pyosmium::BufferIterator *cb)
: AreaManagerSecondPassHandlerBase(mp_manager)
{
m_mp_manager->set_callback([cb](osmium::memory::Buffer &&ab)
{ cb->add_buffer(std::move(ab)); });
}
};
class AreaManager : public pyosmium::BaseHandler
{
public:
AreaManager()
: m_mp_manager(m_assembler_config)
{}
pyosmium::BaseHandler *first_pass_handler() { return this; }
// first-pass-handler
bool relation(pyosmium::PyOSMRelation &r) override
{
m_mp_manager.relation(*(r.get()));
return false;
}
AreaManagerSecondPassHandler *second_pass_handler(py::args args)
{
m_mp_manager.prepare_for_lookup();
return new AreaManagerSecondPassHandler(&m_mp_manager, args);
}
AreaManagerBufferHandler *second_pass_to_buffer(pyosmium::BufferIterator *cb)
{
m_mp_manager.prepare_for_lookup();
return new AreaManagerBufferHandler(&m_mp_manager, cb);
}
private:
osmium::area::Assembler::config_type m_assembler_config;
osmium::area::MultipolygonManager<osmium::area::Assembler> m_mp_manager;
};
} // namespace
#ifdef Py_GIL_DISABLED
PYBIND11_MODULE(area, m, py::mod_gil_not_used())
#else
PYBIND11_MODULE(area, m)
#endif
{
py::class_<AreaManagerSecondPassHandler, pyosmium::BaseHandler>(m,
"AreaManagerSecondPassHandler");
py::class_<AreaManagerBufferHandler, pyosmium::BaseHandler>(m,
"AreaManagerBufferHandler");
py::class_<AreaManager, pyosmium::BaseHandler>(m, "AreaManager")
.def(py::init<>())
.def("first_pass_handler", &AreaManager::first_pass_handler,
py::return_value_policy::reference)
.def("second_pass_handler", &AreaManager::second_pass_handler,
py::return_value_policy::take_ownership, py::keep_alive<1, 2>())
.def("second_pass_to_buffer", &AreaManager::second_pass_to_buffer,
py::keep_alive<1, 2>())
;
}
|
