/**
 * @file main.cpp
 * @author Conrad Sanderson
 *
 * ensmallen is free software; you may redistribute it and/or modify it under
 * the terms of the 3-clause BSD license.  You should have received a copy of
 * the 3-clause BSD license along with ensmallen.  If not, see
 * http://www.opensource.org/licenses/BSD-3-Clause for more information.
 */

#include <iostream>
#if defined(ENS_USE_COOT)
  #include <armadillo>
  #include <bandicoot>
#endif
#include <ensmallen.hpp>

//#define CATCH_CONFIG_MAIN  // catch.hpp will define main()
#define CATCH_CONFIG_RUNNER  // we will define main()
#include "catch.hpp"

int main(int argc, char** argv)
{
  #ifdef ENS_HAVE_COOT
  coot::get_rt().init(true);
  #endif

  Catch::Session session;
  const int returnCode = session.applyCommandLine(argc, argv);
  // Check for a command line error.
  if (returnCode != 0)
    return returnCode;

  std::cout << "ensmallen version: " << ens::version::as_string() << std::endl;
  std::cout << "armadillo version: " << arma::arma_version::as_string() << std::endl;

  #ifdef ENS_HAVE_COOT
  std::cout << "bandicoot version: " << coot::coot_version::as_string() << std::endl;
  #endif

  // Use Catch2 command-line to set the random seed.
  // -rng-seed <'time'|number>
  // If a number is provided this is used directly as the seed. Alternatively
  // if the keyword 'time' is provided then the result of calling std::time(0)
  // is used.
  const size_t seed = session.config().rngSeed();
  std::cout << "random seed: " << seed << std::endl;
  srand((unsigned int) seed);
  arma::arma_rng::set_seed(seed);

  return session.run();
}
