8511256779
git-subtree-dir: third_party/libosmium git-subtree-split: 2282c8450bae55839372a2002db7ca754530d2fc
87 lines
2.6 KiB
C++
87 lines
2.6 KiB
C++
#ifndef CHECK_WKT_HANDLER_HPP
|
|
#define CHECK_WKT_HANDLER_HPP
|
|
|
|
#include <cassert>
|
|
#include <fstream>
|
|
#include <map>
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
#include <osmium/handler.hpp>
|
|
#include <osmium/osm.hpp>
|
|
#include <osmium/osm/types.hpp>
|
|
|
|
class CheckWKTHandler : public osmium::handler::Handler {
|
|
|
|
std::map<osmium::object_id_type, std::string> m_geometries;
|
|
osmium::geom::WKTFactory<> m_factory;
|
|
|
|
void read_wkt_file(const std::string& filename) {
|
|
std::ifstream in(filename, std::ifstream::in);
|
|
if (in) {
|
|
osmium::object_id_type id;
|
|
std::string line;
|
|
while (std::getline(in, line)) {
|
|
size_t pos = line.find_first_of(' ');
|
|
|
|
if (pos == std::string::npos) {
|
|
std::cerr << filename << " not formatted correctly\n";
|
|
exit(1);
|
|
}
|
|
|
|
std::string id_str = line.substr(0, pos);
|
|
std::istringstream iss(id_str);
|
|
iss >> id;
|
|
|
|
if (m_geometries.find(id) != m_geometries.end()) {
|
|
std::cerr << filename + " contains id " << id << "twice\n";
|
|
exit(1);
|
|
}
|
|
|
|
m_geometries[id] = line.substr(pos+1);
|
|
}
|
|
}
|
|
}
|
|
|
|
public:
|
|
|
|
CheckWKTHandler(const std::string& dirname, int test_id) :
|
|
osmium::handler::Handler() {
|
|
|
|
std::string filename = dirname + "/" + std::to_string(test_id / 100) + "/" + std::to_string(test_id) + "/";
|
|
read_wkt_file(filename + "nodes.wkt");
|
|
read_wkt_file(filename + "ways.wkt");
|
|
}
|
|
|
|
~CheckWKTHandler() {
|
|
if (!m_geometries.empty()) {
|
|
for (const auto& geom : m_geometries) {
|
|
std::cerr << "geometry id " << geom.first << " not in data.osm.\n";
|
|
}
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
void node(const osmium::Node& node) {
|
|
const std::string wkt = m_geometries[node.id()];
|
|
assert(wkt != "" && "Missing geometry for node in nodes.wkt");
|
|
|
|
std::string this_wkt = m_factory.create_point(node.location());
|
|
assert(wkt == this_wkt && "wkt geometries don't match");
|
|
m_geometries.erase(node.id());
|
|
}
|
|
|
|
void way(const osmium::Way& way) {
|
|
const std::string wkt = m_geometries[way.id()];
|
|
assert(wkt != "" && "Missing geometry for way in ways.wkt");
|
|
|
|
std::string this_wkt = m_factory.create_linestring(way);
|
|
assert(wkt == this_wkt && "wkt geometries don't match");
|
|
m_geometries.erase(way.id());
|
|
}
|
|
|
|
}; // CheckWKTHandler
|
|
|
|
|
|
#endif // CHECK_WKT_HANDLER_HPP
|