osrm-backend/third_party/libosmium/benchmarks/osmium_benchmark_count.cpp

62 lines
1.2 KiB
C++

/*
The code in this file is released into the Public Domain.
*/
#include <osmium/handler.hpp>
#include <osmium/io/any_input.hpp>
#include <osmium/visitor.hpp>
#include <cstdint>
#include <cstdlib>
#include <iostream>
#include <string>
struct CountHandler : public osmium::handler::Handler {
uint64_t nodes = 0;
uint64_t ways = 0;
uint64_t relations = 0;
void node(const osmium::Node& /*node*/) {
++nodes;
}
void way(const osmium::Way& /*way*/) {
++ways;
}
void relation(const osmium::Relation& /*relation*/) {
++relations;
}
};
int main(int argc, char* argv[]) {
if (argc != 2) {
std::cerr << "Usage: " << argv[0] << " OSMFILE\n";
return 1;
}
try {
const std::string input_filename{argv[1]};
osmium::io::Reader reader{input_filename};
CountHandler handler;
osmium::apply(reader, handler);
reader.close();
std::cout << "Nodes: " << handler.nodes << '\n';
std::cout << "Ways: " << handler.ways << '\n';
std::cout << "Relations: " << handler.relations << '\n';
} catch (const std::exception& e) {
std::cerr << e.what() << '\n';
return 1;
}
return 0;
}