osmium_benchmark_count.cpp

/*

  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 <exception>
#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;
}