main.cpp File Reference

#include <PeriDEMConfig.h>
#include <iostream>
#include <algorithm>
#include <ctime>
#include <filesystem>
#include "inp/input.h"
#include "periDEMModel.h"
#include "util/io.h"
#include "util/json.h"
#include "util/parallelUtil.h"
#include "util/vecMethods.h"
#include <format>

Include dependency graph for main.cpp:

Go to the source code of this file.

Functions
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 28 of file main.cpp.

                                 {
 
  // init parallel
  util::parallel::initMpi(argc, argv);
  int mpiSize = util::parallel::mpiSize(), mpiRank = util::parallel::mpiRank();
  util::io::print(std::format("Initialized MPI. MPI size = {}, MPI rank = {}\n", mpiSize, mpiRank));
  util::io::print(util::parallel::getMpiStatus()->printStr());
 
  util::io::InputParser input(argc, argv);
 
  if (input.cmdOptionExists("-h") or !input.cmdOptionExists("-i")) {
    // print help
    std::cout << "Syntax to run PeriDEM: PeriDEM -i <input file> -nThreads <number of threads>" << std::endl;
    std::cout << "Example: PeriDEM -i input.json -nThreads 4" << std::endl;
    exit(EXIT_FAILURE);
  }
 
  unsigned int nThreads;
  if (input.cmdOptionExists("-nThreads")) nThreads = std::stoi(input.getCmdOption("-nThreads"));
  else {
    nThreads = std::thread::hardware_concurrency();
    util::io::print(std::format("Running test with default number of threads = {}\n", nThreads));
  }
  // set number of threads
  util::parallel::initNThreads(nThreads);
  util::io::print(std::format("Number of threads = {}\n", util::parallel::getNThreads()));
  
  // print program version
  std::cout << "PeriDEM"
            << " (Version " << MAJOR_VERSION << "." << MINOR_VERSION << "."
            << UPDATE_VERSION << ")" << std::endl;
 
  // current time
  auto begin = steady_clock::now();
 
  // read input data
  std::string filename = input.getCmdOption("-i");
  if (!std::filesystem::exists(filename)) {
    throw std::runtime_error(std::format("Input file {} does not exist.", filename));
  }
  std::ifstream f(filename);
  auto j = json::parse(f);
  auto deck = std::make_shared<inp::Input>(j);
 
  // run model
  if (deck->isPeriDEM()) {
    PeriDEMModel dem(deck);
    dem.run(deck);
  } else {
    std::cout << "PeriDEM model not found in input file.\n";
  }
 
  // get time elapsed
  auto end = steady_clock::now();
 
  std::cout << "Total simulation time (s) = " 
            << util::methods::timeDiff(begin, end, "seconds") 
            << std::endl;
 
  util::parallel::finalizeMpi();
  return 0;
}

References util::io::InputParser::cmdOptionExists(), util::parallel::finalizeMpi(), util::io::InputParser::getCmdOption(), util::parallel::getMpiStatus(), util::parallel::getNThreads(), util::parallel::initMpi(), util::parallel::initNThreads(), util::parallel::mpiRank(), util::parallel::mpiSize(), util::io::print(), PeriDEMModel::run(), and util::methods::timeDiff().

Here is the call graph for this function: