1 #ifndef OSMIUM_IO_READER_HPP
2 #define OSMIUM_IO_READER_HPP
42 #include <system_error>
47 # include <sys/wait.h>
55 #include <osmium/io/detail/input_format.hpp>
56 #include <osmium/io/detail/read_thread.hpp>
57 #include <osmium/io/detail/read_write.hpp>
58 #include <osmium/io/detail/queue_util.hpp>
73 inline size_t get_input_queue_size() noexcept {
78 inline size_t get_osmdata_queue_size() noexcept {
123 detail::future_string_queue_type& input_queue,
124 detail::future_buffer_queue_type& osmdata_queue,
125 std::promise<osmium::io::Header>&& header_promise,
127 std::promise<osmium::io::Header> promise = std::move(header_promise);
128 const auto creator = detail::ParserFactory::instance().get_creator_function(file);
129 const auto parser = creator(input_queue, osmdata_queue, promise, read_which_entities);
145 static int execute(
const std::string& command,
const std::string& filename,
int* childpid) {
147 if (pipe(pipefd) < 0) {
148 throw std::system_error(errno, std::system_category(),
"opening pipe failed");
150 const pid_t pid = fork();
152 throw std::system_error(errno, std::system_category(),
"fork failed");
156 for (
int i = 0; i < 32; ++i) {
157 if (i != pipefd[1]) {
161 if (dup2(pipefd[1], 1) < 0) {
165 ::open(
"/dev/null", O_RDONLY);
166 ::open(
"/dev/null", O_WRONLY);
171 if (::execlp(command.c_str(), command.c_str(),
"-g", filename.c_str(),
nullptr) < 0) {
191 std::string protocol = filename.substr(0, filename.find_first_of(
':'));
192 if (protocol ==
"http" || protocol ==
"https" || protocol ==
"ftp" || protocol ==
"file") {
194 return execute(
"curl", filename, childpid);
196 throw io_error(
"Reading OSM files from the network currently not supported on Windows.");
199 return osmium::io::detail::open_for_reading(filename);
215 m_file(file.check()),
216 m_read_which_entities(read_which_entities),
219 m_input_queue(
detail::get_input_queue_size(),
"raw_input"),
220 m_decompressor(m_file.buffer() ?
221 osmium::io::
CompressionFactory::instance().create_decompressor(file.compression(), m_file.buffer(), m_file.buffer_size()) :
223 m_read_thread_manager(*m_decompressor, m_input_queue),
224 m_osmdata_queue(
detail::get_osmdata_queue_size(),
"parser_results"),
225 m_osmdata_queue_wrapper(m_osmdata_queue),
229 m_file_size(m_decompressor->
file_size()) {
230 std::promise<osmium::io::Header> header_promise;
231 m_header_future = header_promise.get_future();
268 m_read_thread_manager.stop();
270 m_osmdata_queue_wrapper.drain();
273 m_read_thread_manager.close();
281 const pid_t pid = ::waitpid(m_childpid, &status, 0);
282 #pragma GCC diagnostic push
283 #pragma GCC diagnostic ignored "-Wold-style-cast"
284 if (pid < 0 || !WIFEXITED(status) || WEXITSTATUS(status) != 0) {
285 throw std::system_error(errno, std::system_category(),
"subprocess returned error");
287 #pragma GCC diagnostic pop
301 throw io_error(
"Can not get header from reader when in status 'error'");
305 if (m_header_future.valid()) {
306 m_header = m_header_future.
get();
334 throw io_error(
"Can not read from reader when in status 'closed', 'eof', or 'error'");
343 buffer = m_osmdata_queue_wrapper.pop();
344 if (detail::at_end_of_data(buffer)) {
346 m_read_thread_manager.close();
391 return m_decompressor->offset();
404 template <
typename... TArgs>
408 Reader reader(std::forward<TArgs>(args)...);
421 #endif // OSMIUM_IO_READER_HPP
detail::queue_wrapper< osmium::memory::Buffer > m_osmdata_queue_wrapper
Definition: reader.hpp:112
status
Definition: reader.hpp:96
osmium::memory::Buffer read()
Definition: reader.hpp:329
Reader(const osmium::io::File &file, osmium::osm_entity_bits::type read_which_entities=osmium::osm_entity_bits::all)
Definition: reader.hpp:214
type
Definition: entity_bits.hpp:63
int m_childpid
Definition: reader.hpp:103
std::future< osmium::io::Header > m_header_future
Definition: reader.hpp:114
osmium::memory::Buffer read_file(TArgs &&...args)
Definition: reader.hpp:405
enum osmium::io::Reader::status m_status
std::unique_ptr< osmium::io::Decompressor > m_decompressor
Definition: reader.hpp:107
Reader & operator=(const Reader &)=delete
bool eof() const
Definition: reader.hpp:364
object or changeset
Definition: entity_bits.hpp:74
detail::future_string_queue_type m_input_queue
Definition: reader.hpp:105
osmium::io::File m_file
Definition: reader.hpp:93
Namespace for everything in the Osmium library.
Definition: assembler.hpp:73
void add_buffer(const Buffer &buffer)
Definition: buffer.hpp:474
osmium::io::detail::ReadThreadManager m_read_thread_manager
Definition: reader.hpp:109
size_t file_size() const noexcept
Definition: reader.hpp:372
static int execute(const std::string &command, const std::string &filename, int *childpid)
Definition: reader.hpp:145
Definition: reader.hpp:91
~Reader() noexcept
Definition: reader.hpp:249
static void parser_thread(const osmium::io::File &file, detail::future_string_queue_type &input_queue, detail::future_buffer_queue_type &osmdata_queue, std::promise< osmium::io::Header > &&header_promise, osmium::osm_entity_bits::type read_which_entities)
Definition: reader.hpp:122
osmium::io::Header header()
Definition: reader.hpp:299
size_t committed() const noexcept
Definition: buffer.hpp:241
size_t get_max_queue_size(const char *queue_name, size_t default_value) noexcept
Definition: config.hpp:69
Definition: buffer.hpp:97
size_t m_file_size
Definition: reader.hpp:119
osmium::thread::thread_handler m_thread
Definition: reader.hpp:117
static int open_input_file_or_url(const std::string &filename, int *childpid)
Definition: reader.hpp:190
detail::future_buffer_queue_type m_osmdata_queue
Definition: reader.hpp:111
osmium::io::Header m_header
Definition: reader.hpp:115
Definition: entity_bits.hpp:65
Reader(const char *filename, osmium::osm_entity_bits::type read_types=osmium::osm_entity_bits::all)
Definition: reader.hpp:239
Definition: compression.hpp:138
size_t offset() const noexcept
Definition: reader.hpp:390
std::string get(const std::string &key, const std::string &default_value="") const noexcept
Definition: options.hpp:125
void close()
Definition: reader.hpp:265
osmium::osm_entity_bits::type m_read_which_entities
Definition: reader.hpp:94
size_t commit()
Definition: buffer.hpp:335
Reader(const std::string &filename, osmium::osm_entity_bits::type read_types=osmium::osm_entity_bits::all)
Definition: reader.hpp:235