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
|
/* 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/geom/mercator_projection.hpp>
#include <osmium/geom/coordinates.hpp>
#include <osmium/geom/haversine.hpp>
#include <osmium/geom/factory.hpp>
#include <osmium/geom/wkb.hpp>
#include <osmium/geom/wkt.hpp>
#include <osmium/geom/geojson.hpp>
#include "cast.h"
#include "osm_base_objects.h"
namespace py = pybind11;
namespace og = osmium::geom;
namespace {
struct WKBFactory : osmium::geom::WKBFactory<>
{
public:
WKBFactory()
: og::WKBFactory<>(og::wkb_type::wkb, og::out_type::hex)
{}
};
using WKTFactory = og::WKTFactory<>;
using GeoJSONFactory = og::GeoJSONFactory<>;
template <typename Factory>
void make_factory_class(py::module_ &m, char const *name)
{
py::class_<Factory>(m, name)
.def(py::init<>())
.def_property_readonly("epsg", &Factory::epsg)
.def_property_readonly("proj_string", &Factory::proj_string)
.def("create_point",
[](Factory &f, py::object const &o) {
if (py::isinstance<osmium::Location>(o)) {
return f.create_point(o.cast<osmium::Location const &>());
}
auto const *node = pyosmium::try_cast<pyosmium::COSMNode>(o);
if (node) {
return f.create_point(*node->get());
}
return f.create_point(o.attr("location").cast<osmium::Location const &>());
},
py::arg("pt"))
.def("create_linestring",
[](Factory &f, py::object const &o, og::use_nodes un, og::direction dir) {
auto const *way = pyosmium::try_cast<pyosmium::COSMWay>(o);
if (way) {
return f.create_linestring(*way->get(), un, dir);
}
return f.create_linestring(pyosmium::cast_list<osmium::WayNodeList>(o), un, dir);
},
py::arg("list"), py::arg("use_nodes")=og::use_nodes::unique,
py::arg("direction")=og::direction::forward)
.def("create_multipolygon",
[](Factory &f, py::object const &o) {
return f.create_multipolygon(*pyosmium::cast<pyosmium::COSMArea>(o).get());
},
py::arg("area"))
;
}
} // namespace
#ifdef Py_GIL_DISABLED
PYBIND11_MODULE(geom, m, py::mod_gil_not_used())
#else
PYBIND11_MODULE(geom, m)
#endif
{
py::enum_<og::use_nodes>(m, "use_nodes")
.value("UNIQUE", og::use_nodes::unique)
.value("ALL", og::use_nodes::all)
.export_values()
;
py::enum_<og::direction>(m, "direction")
.value("BACKWARD", og::direction::backward)
.value("FORWARD", og::direction::forward)
.export_values()
;
py::class_<osmium::geom::Coordinates>(m, "Coordinates")
.def(py::init<>())
.def(py::init<double, double>())
.def(py::init<osmium::Location const &>())
.def_readonly("x", &osmium::geom::Coordinates::x)
.def_readonly("y", &osmium::geom::Coordinates::y)
.def("valid", &osmium::geom::Coordinates::valid)
;
m.def("haversine_distance",
[](py::object const &o) { return og::haversine::distance(pyosmium::cast_list<osmium::WayNodeList>(o)); },
py::arg("list"));
m.def("haversine_distance",
static_cast<double (*)(og::Coordinates const &, og::Coordinates const &)>(&og::haversine::distance));
m.def("haversine_distance",
[](osmium::Location const &l1, osmium::Location const &l2) {
return og::haversine::distance(og::Coordinates(l1), og::Coordinates(l2));
});
m.def("lonlat_to_mercator", &og::lonlat_to_mercator);
m.def("mercator_to_lonlat", &og::mercator_to_lonlat);
make_factory_class<WKBFactory>(m, "WKBFactory");
make_factory_class<WKTFactory>(m, "WKTFactory");
make_factory_class<GeoJSONFactory>(m, "GeoJSONFactory");
}
|
