Newer
Older
#include "Configuration.hpp"
#include <iostream>
#include <stdexcept>
#include "MpiEnvironment.hpp"
#include <boost/format.hpp>
#include <boost/program_options.hpp>
auto Configuration::parseArgs(int argc, char* argv[], const MpiEnvironment& env)
-> Configuration {
namespace po = boost::program_options;
Configuration cfg;
po::options_description desc{"Allowed options"};
desc.add_options()("help,h", "produce help message") //
("gridrows,r", po::value<std::size_t>(&cfg.GridRows)->default_value(0),
"number of rows in the grid") //
("gridcols,c",
po::value<std::size_t>(&cfg.GridColumns)->default_value(0),
"number of columns in the grid") //
("file,f", po::value<std::string>(&cfg.InputFilePath)->required(),
"path to wireworld file"); //
po::variables_map vm;
po::store(po::parse_command_line(argc, argv, desc), vm);
if (vm.count("help") > 0 && env.isMaster()) { std::cout << desc << "\n"; }
po::notify(vm);
// if no dimensions given, use MPI_Dims_create
if (cfg.GridColumns < 1 || cfg.GridRows < 1) {
std::array<int, 2> dims{static_cast<int>(cfg.GridColumns),
static_cast<int>(cfg.GridRows)};
MPI_Dims_create(env.worldSize(), // nnodes
2, // ndims
dims.data() // dims
);
cfg.GridColumns = dims[0];
cfg.GridRows = dims[1];
}
// validate
const auto& totalCellCount = cfg.GridColumns * cfg.GridRows;
const auto& worldSize = env.worldSize();
if (totalCellCount != worldSize) {
std::cerr << boost::format("Total number of cells (%d) does not match "
"the MPI World size (%d)") %
totalCellCount % worldSize;
MPI_Abort(MPI_COMM_WORLD, EXIT_FAILURE);
}
return cfg;
}