babbda98a6
40c4a48 Release v2.10.0 6addb2e Search for protozero also in the place where libosmium was found. 9c0d4bb Consistently use lhs and rhs as parameter names in operators etc. 6179759 Update change log. 894eb7d Explicitly use size_t to get no narrowing conversions. e549e73 Fix CMake config for index_lookup. ccf0bc2 Fix bug where some compilers deduce the wrong type. fc3507d Cleaned and documented index example and renamed to osmium_index_lookup. d3c3036 Rename serdump example to dump_internal and document it. 0e9822e Also forward set_uid_from_signed() function. 29ef95c Rename build_taglist function because it was to similar to build_tag_list. c088dd0 Move static constant out of class so clang will compile it. 606cdc4 Fix CMake build script: Path for finding libosmium. 65f91fe Removed unused typedef. 07174f5 Use condition_variable to tell producer when thread queue is not full. b35e957 Some code cleanup in XML parser. c703dff Fix problem with MemberMeta. 1ccbbef Refactoring CompressionFactory. 6561bd6 Use our own exception type for map factory errors. 56e31fa Throw not_found error directly instead of using helper function. 28230c3 Test empty value for node location store, reorganize tests. 2ba316c Reorganize and fix projection tests. f949485 Don't run quite as many tests with projections because they are slow. 1bad16d Add information about build environment to benchmark output. 99617bb Change proj_geojson benchmark into mercator benchmark. 553b946 Allow optional parameters on Reader in any order. dcc3d8f Factor out some common code. 004d8cd Fix forwarding constructor. 9702978 Fix metadata check. 6cfb6c4 Faster implementation of decode_dense_nodes without reading metdata. 4ba4638 Add additional read_metadata parameter to other Reader constructors. d005937 Optionally ignore metadata when reading file. 63961da Mark all CRC32 update functions as noexcept. cc4ca75 Refactor set_user() function to speed it up. 38d19dc Update comments with file sizes. OSM has grown... f7d0824 Add new benchmark that shows performance when main thread is busy. 25070dc Use const ref params. 88e8d96 Mark add_user() as deprecated. f58d9db Refactor some low-level collection code to clean up code. 4680def Add example showing how to create OSM data out of thin air. d42fd50 Add an example showing how tags in OSM files can be changed. 49bf5bc Add additional constructors to Builders taking a reference to parent. 7b91d63 Change Builder::add_item() to take a reference instead of pointer. 2957e48 Some cleanup of examples. d0b458d Calculate size of object at compile time. 3fbb6e7 Use explicit cast. 1851f3d Remove a test that depends on math details. f6a0802 Various cleanups of example programs. ba4921f Rename add_user() to set_user(). d7637c9 Various cleanups related to builders. 07827bc Fix add_user(). 9a5b395 Also refactor OSMObjectBuilder like ChangesetBuilder before. b1f423c Use call chaining on the builder. e49473d Get rid of ObjectBuilder class. 67d70b9 Refactor ChangesetBuilder::add_user(). 8199c33 Make ChangesetBuilder derive directly from Builder. 61d1b67 Simplify some code. d38467a Change derivation hierarchy of some builder classes. b52f8af Refactor Builder code. d012bfa Refactored some code setting attributes on objects using builder instead. 6a05f60 Also forward set_removed function from builders. 8d63b7d Return *this from Builder setter functions and test it. 72a1266 Update catch.hpp to newest version. 3424a74 Check GEOS version is <= 3.5. aee9f9d Cleanup test code. aef198c Improved asserts in Buffer. a98b9bf Code cleanup in tests. a150466 Use GDAL/OGR instead of GEOS to test our WKB implementation. b04a525 Refactor test. 39aa932 Refactored test_factory_with_projection so it works without GEOS. 648f43a Remove unused dependency on geos from tile test. f1748ae Add setters to Builder classes forwarding to underlying objects. 8166879 In debug mode check that Builders are correctly called and destructed. 1c4257e Call commit() on buffer only after all builders are destructed. 2618636 Add functions to check availability of relation members. b45a4d9 Mark RelationMember::set_ref() as deprecated. 7886771 Move "basic" and "buffer" tests to "osm" and "memory", respectively. b664685 Use functions instead of macros in location test for faster compile. b4929ac Add more tests for number parser. 02662a7 Merge pull request #171 from lonvia/fix-long-exponentials 5344a6c fix parsing of numbers in e-notiation with many post-comma digits 3aeaff3 Add some typedefs to NodeRefList and memory::Collection. e750665 Add iterators to IdSetSmall and add docs and tests to IdSet classes. 50ecb2a Add more features to IdSetDense, including unset and iterator. e3dec78 Make IdSet virtual base class with two implementations. 8ea0153 Use C array instead of std::array in IdSet and clear explicitly. 3ba9905 Deprecate osmium::index::BoolVector in favour of new IdSet. 453d1ca Add osmium::index::IdSet. c78254e Add function to (temporarily) disable the progress bar. 4d88a9f Better document osmium::io::Header class. 320e3af Look for protozero includes in CMake config. 838d25e Allow optional checking for libosmium version number in CMake config. 6ce60c1 Fix entity_bits static_assert() tests. f054731 Update change log. 77ac4c2 Make sleep duration for full queues much smaller. 7e39c01 Make some entity_bits functions constexpr. 69ea72f Fix ~ operator on entity_bits and more tests for entity bits. dafe4cf Update embedded Catch unit test header to version 1.5.7. a41c832 Fixed parsing of location coordinates starting with decimal dot. 6523bae README cosmetics. 229acac Add tests for some examples. f1e753d Merge pull request #163 from sebastic/executable-not-elf-or-script ccea2d5 Remove executable bit from .cpp files. af77fb4 Changelog formatting fixes. git-subtree-dir: third_party/libosmium git-subtree-split: 40c4a48f88d25edace6f0b9e079c306308c7760b
638 lines
25 KiB
C++
638 lines
25 KiB
C++
#ifndef OSMIUM_IO_DETAIL_O5M_INPUT_FORMAT_HPP
|
|
#define OSMIUM_IO_DETAIL_O5M_INPUT_FORMAT_HPP
|
|
|
|
/*
|
|
|
|
This file is part of Osmium (http://osmcode.org/libosmium).
|
|
|
|
Copyright 2013-2016 Jochen Topf <jochen@topf.org> and others (see README).
|
|
|
|
Boost Software License - Version 1.0 - August 17th, 2003
|
|
|
|
Permission is hereby granted, free of charge, to any person or organization
|
|
obtaining a copy of the software and accompanying documentation covered by
|
|
this license (the "Software") to use, reproduce, display, distribute,
|
|
execute, and transmit the Software, and to prepare derivative works of the
|
|
Software, and to permit third-parties to whom the Software is furnished to
|
|
do so, all subject to the following:
|
|
|
|
The copyright notices in the Software and this entire statement, including
|
|
the above license grant, this restriction and the following disclaimer,
|
|
must be included in all copies of the Software, in whole or in part, and
|
|
all derivative works of the Software, unless such copies or derivative
|
|
works are solely in the form of machine-executable object code generated by
|
|
a source language processor.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
|
|
SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
|
|
FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
|
|
ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
DEALINGS IN THE SOFTWARE.
|
|
|
|
*/
|
|
|
|
#include <algorithm>
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <future>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
#include <protozero/exception.hpp>
|
|
#include <protozero/varint.hpp>
|
|
|
|
#include <osmium/builder/osm_object_builder.hpp>
|
|
#include <osmium/io/detail/input_format.hpp>
|
|
#include <osmium/io/detail/queue_util.hpp>
|
|
#include <osmium/io/error.hpp>
|
|
#include <osmium/io/file_format.hpp>
|
|
#include <osmium/io/header.hpp>
|
|
#include <osmium/memory/buffer.hpp>
|
|
#include <osmium/osm/box.hpp>
|
|
#include <osmium/osm/entity_bits.hpp>
|
|
#include <osmium/osm/item_type.hpp>
|
|
#include <osmium/osm/location.hpp>
|
|
#include <osmium/osm/node.hpp>
|
|
#include <osmium/osm/object.hpp>
|
|
#include <osmium/osm/relation.hpp>
|
|
#include <osmium/osm/timestamp.hpp>
|
|
#include <osmium/osm/types.hpp>
|
|
#include <osmium/osm/way.hpp>
|
|
#include <osmium/thread/util.hpp>
|
|
#include <osmium/util/cast.hpp>
|
|
#include <osmium/util/delta.hpp>
|
|
|
|
namespace osmium {
|
|
|
|
namespace builder {
|
|
class Builder;
|
|
} // namespace builder
|
|
|
|
/**
|
|
* Exception thrown when the o5m deocder failed. The exception contains
|
|
* (if available) information about the place where the error happened
|
|
* and the type of error.
|
|
*/
|
|
struct o5m_error : public io_error {
|
|
|
|
explicit o5m_error(const char* what) :
|
|
io_error(std::string{"o5m format error: "} + what) {
|
|
}
|
|
|
|
}; // struct o5m_error
|
|
|
|
namespace io {
|
|
|
|
namespace detail {
|
|
|
|
// Implementation of the o5m/o5c file formats according to the
|
|
// description at http://wiki.openstreetmap.org/wiki/O5m .
|
|
|
|
class ReferenceTable {
|
|
|
|
// The following settings are from the o5m description:
|
|
|
|
// The maximum number of entries in this table.
|
|
const uint64_t number_of_entries = 15000;
|
|
|
|
// The size of one entry in the table.
|
|
const unsigned int entry_size = 256;
|
|
|
|
// The maximum length of a string in the table including
|
|
// two \0 bytes.
|
|
const unsigned int max_length = 250 + 2;
|
|
|
|
// The data is stored in this string. It is default constructed
|
|
// and then resized on demand the first time something is added.
|
|
// This is done because the ReferenceTable is in a O5mParser
|
|
// object which will be copied from one thread to another. This
|
|
// way the string is still small when it is copied.
|
|
std::string m_table;
|
|
|
|
unsigned int current_entry = 0;
|
|
|
|
public:
|
|
|
|
void clear() {
|
|
current_entry = 0;
|
|
}
|
|
|
|
void add(const char* string, size_t size) {
|
|
if (m_table.empty()) {
|
|
m_table.resize(entry_size * number_of_entries);
|
|
}
|
|
if (size <= max_length) {
|
|
std::copy_n(string, size, &m_table[current_entry * entry_size]);
|
|
if (++current_entry == number_of_entries) {
|
|
current_entry = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
const char* get(uint64_t index) const {
|
|
if (m_table.empty() || index == 0 || index > number_of_entries) {
|
|
throw o5m_error{"reference to non-existing string in table"};
|
|
}
|
|
auto entry = (current_entry + number_of_entries - index) % number_of_entries;
|
|
return &m_table[entry * entry_size];
|
|
}
|
|
|
|
}; // class ReferenceTable
|
|
|
|
class O5mParser : public Parser {
|
|
|
|
static constexpr int buffer_size = 2 * 1000 * 1000;
|
|
|
|
osmium::io::Header m_header;
|
|
|
|
osmium::memory::Buffer m_buffer;
|
|
|
|
std::string m_input;
|
|
|
|
const char* m_data;
|
|
const char* m_end;
|
|
|
|
ReferenceTable m_reference_table;
|
|
|
|
static int64_t zvarint(const char** data, const char* end) {
|
|
return protozero::decode_zigzag64(protozero::decode_varint(data, end));
|
|
}
|
|
|
|
bool ensure_bytes_available(size_t need_bytes) {
|
|
if ((m_end - m_data) >= long(need_bytes)) {
|
|
return true;
|
|
}
|
|
|
|
if (input_done() && (m_input.size() < need_bytes)) {
|
|
return false;
|
|
}
|
|
|
|
m_input.erase(0, m_data - m_input.data());
|
|
|
|
while (m_input.size() < need_bytes) {
|
|
std::string data = get_input();
|
|
if (input_done()) {
|
|
return false;
|
|
}
|
|
m_input.append(data);
|
|
}
|
|
|
|
m_data = m_input.data();
|
|
m_end = m_input.data() + m_input.size();
|
|
|
|
return true;
|
|
}
|
|
|
|
void check_header_magic() {
|
|
static const unsigned char header_magic[] = { 0xff, 0xe0, 0x04, 'o', '5' };
|
|
|
|
if (std::strncmp(reinterpret_cast<const char*>(header_magic), m_data, sizeof(header_magic))) {
|
|
throw o5m_error{"wrong header magic"};
|
|
}
|
|
|
|
m_data += sizeof(header_magic);
|
|
}
|
|
|
|
void check_file_type() {
|
|
if (*m_data == 'm') { // o5m data file
|
|
m_header.set_has_multiple_object_versions(false);
|
|
} else if (*m_data == 'c') { // o5c change file
|
|
m_header.set_has_multiple_object_versions(true);
|
|
} else {
|
|
throw o5m_error{"wrong header magic"};
|
|
}
|
|
|
|
m_data++;
|
|
}
|
|
|
|
void check_file_format_version() {
|
|
if (*m_data != '2') {
|
|
throw o5m_error{"wrong header magic"};
|
|
}
|
|
|
|
m_data++;
|
|
}
|
|
|
|
void decode_header() {
|
|
if (! ensure_bytes_available(7)) { // overall length of header
|
|
throw o5m_error{"file too short (incomplete header info)"};
|
|
}
|
|
|
|
check_header_magic();
|
|
check_file_type();
|
|
check_file_format_version();
|
|
}
|
|
|
|
void mark_header_as_done() {
|
|
set_header_value(m_header);
|
|
}
|
|
|
|
osmium::util::DeltaDecode<osmium::object_id_type> m_delta_id;
|
|
|
|
osmium::util::DeltaDecode<int64_t> m_delta_timestamp;
|
|
osmium::util::DeltaDecode<osmium::changeset_id_type> m_delta_changeset;
|
|
osmium::util::DeltaDecode<int64_t> m_delta_lon;
|
|
osmium::util::DeltaDecode<int64_t> m_delta_lat;
|
|
|
|
osmium::util::DeltaDecode<osmium::object_id_type> m_delta_way_node_id;
|
|
osmium::util::DeltaDecode<osmium::object_id_type> m_delta_member_ids[3];
|
|
|
|
void reset() {
|
|
m_reference_table.clear();
|
|
|
|
m_delta_id.clear();
|
|
m_delta_timestamp.clear();
|
|
m_delta_changeset.clear();
|
|
m_delta_lon.clear();
|
|
m_delta_lat.clear();
|
|
|
|
m_delta_way_node_id.clear();
|
|
m_delta_member_ids[0].clear();
|
|
m_delta_member_ids[1].clear();
|
|
m_delta_member_ids[2].clear();
|
|
}
|
|
|
|
const char* decode_string(const char** dataptr, const char* const end) {
|
|
if (**dataptr == 0x00) { // get inline string
|
|
(*dataptr)++;
|
|
if (*dataptr == end) {
|
|
throw o5m_error{"string format error"};
|
|
}
|
|
return *dataptr;
|
|
} else { // get from reference table
|
|
auto index = protozero::decode_varint(dataptr, end);
|
|
return m_reference_table.get(index);
|
|
}
|
|
}
|
|
|
|
std::pair<osmium::user_id_type, const char*> decode_user(const char** dataptr, const char* const end) {
|
|
bool update_pointer = (**dataptr == 0x00);
|
|
const char* data = decode_string(dataptr, end);
|
|
const char* start = data;
|
|
|
|
auto uid = protozero::decode_varint(&data, end);
|
|
|
|
if (data == end) {
|
|
throw o5m_error{"missing user name"};
|
|
}
|
|
|
|
const char* user = ++data;
|
|
|
|
if (uid == 0 && update_pointer) {
|
|
m_reference_table.add("\0\0", 2);
|
|
*dataptr = data;
|
|
return std::make_pair(0, "");
|
|
}
|
|
|
|
while (*data++) {
|
|
if (data == end) {
|
|
throw o5m_error{"no null byte in user name"};
|
|
}
|
|
}
|
|
|
|
if (update_pointer) {
|
|
m_reference_table.add(start, data - start);
|
|
*dataptr = data;
|
|
}
|
|
|
|
return std::make_pair(static_cast_with_assert<osmium::user_id_type>(uid), user);
|
|
}
|
|
|
|
void decode_tags(osmium::builder::Builder& parent, const char** dataptr, const char* const end) {
|
|
osmium::builder::TagListBuilder builder{parent};
|
|
|
|
while (*dataptr != end) {
|
|
bool update_pointer = (**dataptr == 0x00);
|
|
const char* data = decode_string(dataptr, end);
|
|
const char* start = data;
|
|
|
|
while (*data++) {
|
|
if (data == end) {
|
|
throw o5m_error{"no null byte in tag key"};
|
|
}
|
|
}
|
|
|
|
const char* value = data;
|
|
while (*data++) {
|
|
if (data == end) {
|
|
throw o5m_error{"no null byte in tag value"};
|
|
}
|
|
}
|
|
|
|
if (update_pointer) {
|
|
m_reference_table.add(start, data - start);
|
|
*dataptr = data;
|
|
}
|
|
|
|
builder.add_tag(start, value);
|
|
}
|
|
}
|
|
|
|
const char* decode_info(osmium::OSMObject& object, const char** dataptr, const char* const end) {
|
|
const char* user = "";
|
|
|
|
if (**dataptr == 0x00) { // no info section
|
|
++*dataptr;
|
|
} else { // has info section
|
|
object.set_version(static_cast_with_assert<object_version_type>(protozero::decode_varint(dataptr, end)));
|
|
auto timestamp = m_delta_timestamp.update(zvarint(dataptr, end));
|
|
if (timestamp != 0) { // has timestamp
|
|
object.set_timestamp(timestamp);
|
|
object.set_changeset(m_delta_changeset.update(zvarint(dataptr, end)));
|
|
if (*dataptr != end) {
|
|
auto uid_user = decode_user(dataptr, end);
|
|
object.set_uid(uid_user.first);
|
|
user = uid_user.second;
|
|
} else {
|
|
object.set_uid(user_id_type(0));
|
|
}
|
|
}
|
|
}
|
|
|
|
return user;
|
|
}
|
|
|
|
void decode_node(const char* data, const char* const end) {
|
|
osmium::builder::NodeBuilder builder{m_buffer};
|
|
|
|
builder.set_id(m_delta_id.update(zvarint(&data, end)));
|
|
|
|
builder.set_user(decode_info(builder.object(), &data, end));
|
|
|
|
if (data == end) {
|
|
// no location, object is deleted
|
|
builder.set_visible(false);
|
|
builder.set_location(osmium::Location{});
|
|
} else {
|
|
auto lon = m_delta_lon.update(zvarint(&data, end));
|
|
auto lat = m_delta_lat.update(zvarint(&data, end));
|
|
builder.set_location(osmium::Location{lon, lat});
|
|
|
|
if (data != end) {
|
|
decode_tags(builder, &data, end);
|
|
}
|
|
}
|
|
}
|
|
|
|
void decode_way(const char* data, const char* const end) {
|
|
osmium::builder::WayBuilder builder{m_buffer};
|
|
|
|
builder.set_id(m_delta_id.update(zvarint(&data, end)));
|
|
|
|
builder.set_user(decode_info(builder.object(), &data, end));
|
|
|
|
if (data == end) {
|
|
// no reference section, object is deleted
|
|
builder.set_visible(false);
|
|
} else {
|
|
auto reference_section_length = protozero::decode_varint(&data, end);
|
|
if (reference_section_length > 0) {
|
|
const char* const end_refs = data + reference_section_length;
|
|
if (end_refs > end) {
|
|
throw o5m_error{"way nodes ref section too long"};
|
|
}
|
|
|
|
osmium::builder::WayNodeListBuilder wn_builder{builder};
|
|
|
|
while (data < end_refs) {
|
|
wn_builder.add_node_ref(m_delta_way_node_id.update(zvarint(&data, end)));
|
|
}
|
|
}
|
|
|
|
if (data != end) {
|
|
decode_tags(builder, &data, end);
|
|
}
|
|
}
|
|
}
|
|
|
|
osmium::item_type decode_member_type(char c) {
|
|
if (c < '0' || c > '2') {
|
|
throw o5m_error{"unknown member type"};
|
|
}
|
|
return osmium::nwr_index_to_item_type(c - '0');
|
|
}
|
|
|
|
std::pair<osmium::item_type, const char*> decode_role(const char** dataptr, const char* const end) {
|
|
bool update_pointer = (**dataptr == 0x00);
|
|
const char* data = decode_string(dataptr, end);
|
|
const char* start = data;
|
|
|
|
auto member_type = decode_member_type(*data++);
|
|
if (data == end) {
|
|
throw o5m_error{"missing role"};
|
|
}
|
|
const char* role = data;
|
|
|
|
while (*data++) {
|
|
if (data == end) {
|
|
throw o5m_error{"no null byte in role"};
|
|
}
|
|
}
|
|
|
|
if (update_pointer) {
|
|
m_reference_table.add(start, data - start);
|
|
*dataptr = data;
|
|
}
|
|
|
|
return std::make_pair(member_type, role);
|
|
}
|
|
|
|
void decode_relation(const char* data, const char* const end) {
|
|
osmium::builder::RelationBuilder builder{m_buffer};
|
|
|
|
builder.set_id(m_delta_id.update(zvarint(&data, end)));
|
|
|
|
builder.set_user(decode_info(builder.object(), &data, end));
|
|
|
|
if (data == end) {
|
|
// no reference section, object is deleted
|
|
builder.set_visible(false);
|
|
} else {
|
|
auto reference_section_length = protozero::decode_varint(&data, end);
|
|
if (reference_section_length > 0) {
|
|
const char* const end_refs = data + reference_section_length;
|
|
if (end_refs > end) {
|
|
throw o5m_error{"relation format error"};
|
|
}
|
|
|
|
osmium::builder::RelationMemberListBuilder rml_builder{builder};
|
|
|
|
while (data < end_refs) {
|
|
auto delta_id = zvarint(&data, end);
|
|
if (data == end) {
|
|
throw o5m_error{"relation member format error"};
|
|
}
|
|
auto type_role = decode_role(&data, end);
|
|
auto i = osmium::item_type_to_nwr_index(type_role.first);
|
|
auto ref = m_delta_member_ids[i].update(delta_id);
|
|
rml_builder.add_member(type_role.first, ref, type_role.second);
|
|
}
|
|
}
|
|
|
|
if (data != end) {
|
|
decode_tags(builder, &data, end);
|
|
}
|
|
}
|
|
}
|
|
|
|
void decode_bbox(const char* data, const char* const end) {
|
|
auto sw_lon = zvarint(&data, end);
|
|
auto sw_lat = zvarint(&data, end);
|
|
auto ne_lon = zvarint(&data, end);
|
|
auto ne_lat = zvarint(&data, end);
|
|
|
|
m_header.add_box(osmium::Box{osmium::Location{sw_lon, sw_lat},
|
|
osmium::Location{ne_lon, ne_lat}});
|
|
}
|
|
|
|
void decode_timestamp(const char* data, const char* const end) {
|
|
auto timestamp = osmium::Timestamp(zvarint(&data, end)).to_iso();
|
|
m_header.set("o5m_timestamp", timestamp);
|
|
m_header.set("timestamp", timestamp);
|
|
}
|
|
|
|
void flush() {
|
|
osmium::memory::Buffer buffer(buffer_size);
|
|
using std::swap;
|
|
swap(m_buffer, buffer);
|
|
send_to_output_queue(std::move(buffer));
|
|
}
|
|
|
|
enum class dataset_type : unsigned char {
|
|
node = 0x10,
|
|
way = 0x11,
|
|
relation = 0x12,
|
|
bounding_box = 0xdb,
|
|
timestamp = 0xdc,
|
|
header = 0xe0,
|
|
sync = 0xee,
|
|
jump = 0xef,
|
|
reset = 0xff
|
|
};
|
|
|
|
void decode_data() {
|
|
while (ensure_bytes_available(1)) {
|
|
dataset_type ds_type = dataset_type(*m_data++);
|
|
if (ds_type > dataset_type::jump) {
|
|
if (ds_type == dataset_type::reset) {
|
|
reset();
|
|
}
|
|
} else {
|
|
ensure_bytes_available(protozero::max_varint_length);
|
|
|
|
uint64_t length = 0;
|
|
try {
|
|
length = protozero::decode_varint(&m_data, m_end);
|
|
} catch (const protozero::end_of_buffer_exception&) {
|
|
throw o5m_error{"premature end of file"};
|
|
}
|
|
|
|
if (! ensure_bytes_available(length)) {
|
|
throw o5m_error{"premature end of file"};
|
|
}
|
|
|
|
switch (ds_type) {
|
|
case dataset_type::node:
|
|
mark_header_as_done();
|
|
if (read_types() & osmium::osm_entity_bits::node) {
|
|
decode_node(m_data, m_data + length);
|
|
m_buffer.commit();
|
|
}
|
|
break;
|
|
case dataset_type::way:
|
|
mark_header_as_done();
|
|
if (read_types() & osmium::osm_entity_bits::way) {
|
|
decode_way(m_data, m_data + length);
|
|
m_buffer.commit();
|
|
}
|
|
break;
|
|
case dataset_type::relation:
|
|
mark_header_as_done();
|
|
if (read_types() & osmium::osm_entity_bits::relation) {
|
|
decode_relation(m_data, m_data + length);
|
|
m_buffer.commit();
|
|
}
|
|
break;
|
|
case dataset_type::bounding_box:
|
|
decode_bbox(m_data, m_data + length);
|
|
break;
|
|
case dataset_type::timestamp:
|
|
decode_timestamp(m_data, m_data + length);
|
|
break;
|
|
default:
|
|
// ignore unknown datasets
|
|
break;
|
|
}
|
|
|
|
if (read_types() == osmium::osm_entity_bits::nothing && header_is_done()) {
|
|
break;
|
|
}
|
|
|
|
m_data += length;
|
|
|
|
if (m_buffer.committed() > buffer_size / 10 * 9) {
|
|
flush();
|
|
}
|
|
}
|
|
}
|
|
|
|
if (m_buffer.committed()) {
|
|
flush();
|
|
}
|
|
|
|
mark_header_as_done();
|
|
}
|
|
|
|
public:
|
|
|
|
O5mParser(future_string_queue_type& input_queue,
|
|
future_buffer_queue_type& output_queue,
|
|
std::promise<osmium::io::Header>& header_promise,
|
|
osmium::io::detail::reader_options options) :
|
|
Parser(input_queue, output_queue, header_promise, options),
|
|
m_header(),
|
|
m_buffer(buffer_size),
|
|
m_input(),
|
|
m_data(m_input.data()),
|
|
m_end(m_data) {
|
|
}
|
|
|
|
~O5mParser() noexcept final = default;
|
|
|
|
void run() final {
|
|
osmium::thread::set_thread_name("_osmium_o5m_in");
|
|
|
|
decode_header();
|
|
decode_data();
|
|
}
|
|
|
|
}; // class O5mParser
|
|
|
|
// we want the register_parser() function to run, setting
|
|
// the variable is only a side-effect, it will never be used
|
|
const bool registered_o5m_parser = ParserFactory::instance().register_parser(
|
|
file_format::o5m,
|
|
[](future_string_queue_type& input_queue,
|
|
future_buffer_queue_type& output_queue,
|
|
std::promise<osmium::io::Header>& header_promise,
|
|
osmium::io::detail::reader_options options) {
|
|
return std::unique_ptr<Parser>(new O5mParser(input_queue, output_queue, header_promise, options));
|
|
});
|
|
|
|
// dummy function to silence the unused variable warning from above
|
|
inline bool get_registered_o5m_parser() noexcept {
|
|
return registered_o5m_parser;
|
|
}
|
|
|
|
} // namespace detail
|
|
|
|
} // namespace io
|
|
|
|
} // namespace osmium
|
|
|
|
#endif // OSMIUM_IO_DETAIL_O5M_INPUT_FORMAT_HPP
|