Remove use of callback_object()

1 files changed, 25 insertions, 13 deletions

diff --git a/tagstats/tagstats_handler.hpp b/tagstats/tagstats_handler.hpp
index dcd23d9..17119fc 100644
--- a/tagstats/tagstats_handler.hpp
+++ b/tagstats/tagstats_handler.hpp
@@ -233,19 +233,7 @@ class TagStatsHandler : public Osmium::Handler::Base {
 
     }
 
-public:
-
-    TagStatsHandler() : Base(), max_timestamp(0) {
-        string_store = new StringStore(string_store_size);
-        db = new Osmium::Sqlite::Database("taginfo-db.db");
-    }
-
-    ~TagStatsHandler() {
-        delete db;
-        delete string_store;
-    }
-
-    void callback_object(Osmium::OSM::Object *object) {
+    void collect_tag_stats(Osmium::OSM::Object *object) {
         if (max_timestamp < object->get_timestamp()) {
             max_timestamp = object->get_timestamp();
         }
@@ -270,6 +258,30 @@ public:
 #endif // TAGSTATS_COUNT_KEY_COMBINATIONS
     }
 
+public:
+
+    TagStatsHandler() : Base(), max_timestamp(0) {
+        string_store = new StringStore(string_store_size);
+        db = new Osmium::Sqlite::Database("taginfo-db.db");
+    }
+
+    ~TagStatsHandler() {
+        delete db;
+        delete string_store;
+    }
+
+    void callback_node(Osmium::OSM::Node *node) {
+        collect_tag_stats(node);
+    }
+
+    void callback_way(Osmium::OSM::Way *way) {
+        collect_tag_stats(way);
+    }
+
+    void callback_relation(Osmium::OSM::Relation *relation) {
+        collect_tag_stats(relation);
+    }
+
     void callback_before_nodes() {
         timer = time(0);
     }