1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#define CATCH_CONFIG_RUNNER
#include "sopt/config.h"
#include <catch2/catch_all.hpp>
#include <memory>
#include <mpi.h>
#include <random>
#include <regex>
#include "sopt/logging.h"
#include "sopt/mpi/session.h"
std::unique_ptr<std::mt19937_64> mersenne(new std::mt19937_64(0));
int main(int argc, const char **argv) {
Catch::Session session; // There must be exactly once instance
auto const mpi_session = sopt::mpi::init(argc, argv);
// The following mess transforms the input arguments so that output files have different names
// on different processors
std::vector<std::string> arguments(argv, argv + argc);
auto output_opt = std::find_if(arguments.begin(), arguments.end(), [](std::string const &arg) {
if (arg == "-o" or arg == "--out") return true;
auto const N = std::string("--out=").size();
return arg.size() > N and arg.substr(0, N) == "--out=";
});
if (output_opt != arguments.end()) {
if (*output_opt == "-o" or *output_opt == "--out") output_opt += 1;
if (output_opt != arguments.end()) {
int rank;
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
if (rank > 0)
*output_opt =
std::regex_replace(*output_opt, std::regex("\\.xml"), std::to_string(rank) + ".xml");
}
}
// transforms the modified arguments to a C-style array of pointers.
std::vector<char const *> cargs(arguments.size());
std::transform(arguments.begin(), arguments.end(), cargs.begin(),
[](std::string const &c) { return c.c_str(); });
int const returnCode = session.applyCommandLine(argc, const_cast<char **>(cargs.data()));
if (returnCode != 0) // Indicates a command line error
return returnCode;
mersenne.reset(new std::mt19937_64(session.configData().rngSeed));
auto const result = session.run();
return result;
}
|
