common_mpi_catch_main.cc File Reference

#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"

Include dependency graph for common_mpi_catch_main.cc:

Go to the source code of this file.

Macros
#define	CATCH_CONFIG_RUNNER

Functions
std::unique_ptr< std::mt19937_64 >	mersenne (new std::mt19937_64(0))
int	main (int argc, const char **argv)

Macro Definition Documentation

CATCH_CONFIG_RUNNER

#define CATCH_CONFIG_RUNNER

Definition at line 1 of file common_mpi_catch_main.cc.

Function Documentation

main()

int main	(	int	argc,
		const char **	argv
	)

Definition at line 14 of file common_mpi_catch_main.cc.

                                      {
  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;
}

References cargs, sopt::mpi::init(), and mersenne().

mersenne()

std::unique_ptr<std::mt19937_64> mersenne

(

new

std::mt19937_640

)

Referenced by main().