main.cpp

Go to the documentation of this file.

#include "main.h"

//====================================================================
int main(int argc, char *argv[])
{
  // ###  initial setup  ###
  Bridge::vout.init(std::cout);
  Bridge::vout.ildg_init(std::cout);
  Bridge::VerboseLevel vl = Bridge::GENERAL;

#ifdef USE_TESTMANAGER
  // process list option prior to other initializations.
  if (argc > 1) {
    std::string arg1 = argv[1];

    if ((arg1 == "-l") || (arg1 == "--list")) {
      TestManager::Instance().list();
      return EXIT_SUCCESS;
    } else if ((arg1 == "-h") || (arg1 == "--help")) {
      //- show usage and exit
      vout.general("\n");
      vout.general("usage: %s\n", argv[0]);
      vout.general("  (without args) -- run tests interactively.\n");
      vout.general("  test_names     -- run specified tests.\n");
      vout.general("  -a | --all     -- run all tests.\n");
      vout.general("  -l | --list    -- list registered tests.\n");
      vout.general("  -h | --help    -- show this message.\n");
      vout.general("\n");

      return EXIT_SUCCESS;
    }
  }
#endif

  Communicator::init(&argc, &argv);


  // ####  parameter setup  ####
  unique_ptr<Parameters> params_main(new Parameters_Main);
  unique_ptr<Parameters> params_all(new Parameters);

  params_all->Register_Parameters("Main", params_main);

  // ####  banner  ####
  vout.general(vl, "Bridge++ %s\n\n", BRIDGE_VERSION);


  string filename_input = filename_main_input;
  if (filename_input == "stdin") {
    vout.general(vl, "input filename : ");
    std::cin >> filename_input;
    vout.general(vl, "%s\n", filename_input.c_str());
  } else {
    vout.general(vl, "input filename : %s\n", filename_input.c_str());
  }
  vout.general(vl, "\n");

  ParameterManager::read(filename_input, params_all);

  const std::vector<int> lattice_size     = params_main->get_int_vector("lattice_size");
  const std::vector<int> grid_size        = params_main->get_int_vector("grid_size");
  const int              number_of_thread = params_main->get_int("number_of_thread");
  const int              number_of_color  = params_main->get_int("number_of_color");
  const string           str_logfile      = params_main->get_string("log_filename");
  const string           str_ildg_logfile = params_main->get_string("ildg_log_filename");
  const string           str_vlevel       = params_main->get_string("verbose_level");


  //- initializations
  vl = vout.set_verbose_level(str_vlevel);
  CommonParameters::init_Vlevel(vl);

//  std::ofstream logfile(str_logfile.c_str());
//  std::ofstream ildg_logfile(str_ildg_logfile.c_str());

  std::ofstream logfile;

  if (str_logfile != "stdout") {
    logfile.open(str_logfile.c_str());
    if (!logfile) {
      vout.crucial(vl, "cannot open logfile: %s.\n", str_logfile.c_str());
      exit(EXIT_FAILURE);
    }
    Bridge::vout.init(logfile);
  }

  std::ofstream ildg_logfile;

  if (str_ildg_logfile != "stdout") {
    ildg_logfile.open(str_ildg_logfile.c_str());
    if (!ildg_logfile) {
      vout.crucial(vl, "cannot open logfile for ildg format: %s.\n", str_ildg_logfile.c_str());
      exit(EXIT_FAILURE);
    }
    Bridge::vout.ildg_init(ildg_logfile);
  }

  // CommonParameters::init(lattice_size, grid_size);
  CommonParameters::init(lattice_size, grid_size, number_of_color);
  Communicator::setup();

  ThreadManager_OpenMP::init(number_of_thread);

  int Ndim = CommonParameters::Ndim();
  int Nvol = CommonParameters::Nvol();


  //- print input parameters
  vout.general(vl, "Main: input parameters\n");
  for (int mu = 0; mu < Ndim; ++mu) {
    vout.general(vl, "  lattice_size[%d] = %d\n", mu, lattice_size[mu]);
  }
  for (int mu = 0; mu < Ndim; ++mu) {
    vout.general(vl, "  grid_size[%d]    = %d\n", mu, grid_size[mu]);
  }
  vout.general(vl, "  number of thread = %d\n", number_of_thread);
  vout.general(vl, "  number of color  = %d\n", number_of_color);
  vout.general(vl, "  logfile          = %s\n", str_logfile.c_str());
  vout.general(vl, "  ildg_logfile     = %s\n", str_ildg_logfile.c_str());
  vout.general(vl, "  vlevel           = %s\n", str_vlevel.c_str());
  vout.general(vl, "\n");

  //- input parameter check
  int err = 0;
  err += ParameterCheck::non_NULL(str_logfile);
  err += ParameterCheck::non_NULL(str_ildg_logfile);

  if (err) {
    vout.crucial(vl, "Main: input parameters have not been set.\n");
    exit(EXIT_FAILURE);
  }


  //- timestamp (starting time)
  unique_ptr<Timer> timer(new Timer("Main"));
  timer->timestamp();
  timer->start();


#ifdef USE_TESTMANAGER
  run_testmanager(argc, argv);
#else
  run_test();
#endif


  //- timestamp (end time)
  timer->report();
  timer->timestamp();


  ThreadManager_OpenMP::finalize();
  Communicator::finalize();

  return EXIT_SUCCESS;
}


//====================================================================
//============================================================END=====