bool CommandGetParents::setup(const std::vector<std::string>& arguments) {
po::options_description opts_cmd{"COMMAND OPTIONS"};
opts_cmd.add_options()
("default-type", po::value<std::string>()->default_value("node"), "Default item type")
("id-file,i", po::value<std::vector<std::string>>(), "Read OSM IDs from text file")
("id-osm-file,I", po::value<std::vector<std::string>>(), "Read OSM IDs from OSM file")
("add-self,s", "Add objects with specified IDs themselves")
("verbose-ids", "Print all requested IDs")
;
po::options_description opts_common{add_common_options()};
po::options_description opts_input{add_single_input_options()};
po::options_description opts_output{add_output_options()};
po::options_description hidden;
hidden.add_options()
("input-filename", po::value<std::string>(), "OSM input file")
("ids", po::value<std::vector<std::string>>(), "OSM IDs")
;
po::options_description desc;
desc.add(opts_cmd).add(opts_common).add(opts_input).add(opts_output);
po::options_description parsed_options;
parsed_options.add(desc).add(hidden);
po::positional_options_description positional;
positional.add("input-filename", 1);
positional.add("ids", -1);
po::variables_map vm;
po::store(po::command_line_parser(arguments).options(parsed_options).positional(positional).run(), vm);
po::notify(vm);
setup_common(vm, desc);
setup_progress(vm);
setup_input_file(vm);
setup_output_file(vm);
if (vm.count("add-self")) {
m_add_self = true;
}
if (vm.count("default-type")) {
m_default_item_type = parse_item_type(vm["default-type"].as<std::string>());
}
if (vm.count("verbose-ids")) {
m_vout.verbose(true);
m_verbose_ids = true;
}
if (vm.count("id-file")) {
for (const std::string& filename : vm["id-file"].as<std::vector<std::string>>()) {
if (filename == "-") {
if (m_input_filename.empty() || m_input_filename == "-") {
throw argument_error{"Can not read OSM input and IDs both from STDIN."};
}
m_vout << "Reading IDs from STDIN...\n";
read_id_file(std::cin, m_ids, m_default_item_type);
} else {
std::ifstream id_file{filename};
if (!id_file.is_open()) {
throw argument_error{"Could not open file '" + filename + "'"};
}
m_vout << "Reading ID file...\n";
read_id_file(id_file, m_ids, m_default_item_type);
}
}
}
if (vm.count("id-osm-file")) {
for (const std::string& filename : vm["id-osm-file"].as<std::vector<std::string>>()) {
m_vout << "Reading OSM ID file...\n";
read_id_osm_file(filename, m_ids);
}
}
if (vm.count("ids")) {
std::string sids;
for (const auto& s : vm["ids"].as<std::vector<std::string>>()) {
sids += s + " ";
}
for (const auto& s : osmium::split_string(sids, "\t ;,/|", true)) {
parse_and_add_id(s, m_ids, m_default_item_type);
}
}
if (no_ids(m_ids)) {
throw argument_error{"Please specify IDs to look for on command line or with option --id-file/-i or --id-osm-file/-I."};
}
return true;
}
bool CommandAddRefs::setup(const std::vector<std::string>& arguments) {
po::options_description cmdline("Available options");
cmdline.add_options()
("id-file,i", po::value<std::string>(), "Read OSM IDs from given file")
("source,s", po::value<std::string>(), "Source file supplying the referenced objects")
("source-format", po::value<std::string>(), "Format of source file")
;
add_common_options(cmdline);
add_multiple_inputs_options(cmdline);
add_output_options(cmdline);
po::options_description hidden("Hidden options");
hidden.add_options()
("input-filenames", po::value<std::vector<std::string>>(), "OSM input files")
;
po::options_description desc("Allowed options");
desc.add(cmdline).add(hidden);
po::positional_options_description positional;
positional.add("input-filenames", -1);
po::variables_map vm;
po::store(po::command_line_parser(arguments).options(desc).positional(positional).run(), vm);
po::notify(vm);
if (vm.count("help")) {
std::cout << "Usage: osmium add-refs [OPTIONS] -s OSM-DATA-FILE OSM-DATA-FILE...\n"
<< " osmium add-refs [OPTIONS] -s OSM-DATA-FILE -i ID-FILE\n\n";
std::cout << cmdline << "\n";
exit(0);
}
setup_common(vm);
setup_input_files(vm, true);
setup_output_file(vm);
if (vm.count("id-file")) {
std::string filename = vm["id-file"].as<std::string>();
std::ifstream id_file{filename};
if (!id_file.is_open()) {
throw argument_error("Could not open file '" + filename + "'");
}
for (std::string line; std::getline(id_file, line); ) {
if (line.empty() || line[0] == '#') {
continue;
}
auto pos = line.find(' ');
if (pos != std::string::npos) {
line = line.erase(pos);
}
parse_and_add_id(line);
}
}
if (vm.count("source")) {
m_source_filename = vm["source"].as<std::string>();
}
if (vm.count("source-format")) {
m_source_format = vm["source-format"].as<std::string>();
}
m_source_file = osmium::io::File{m_source_filename, m_source_format};
return true;
}
