// RandomGenerator_t.cpp

#include "RandomGeneratorTest.h"

#include <cassert>
#include <string>
#include <vector>
#include <iostream>

using std::string;
using std::vector;
using std::cout;
using std::cerr;
using std::endl;

//**********************************************************************

int main( ) {

  string component = "RandomGenerator";
  string ok_prefix = component + " (I): ";
  string error_prefix = component + " test (E): ";

  cout << ok_prefix
       << "---------- Testing component " + component
          + ". ----------" << endl;

  // Make sure assert is enabled.
  bool assert_flag = false;
  assert ( ( assert_flag = true, assert_flag ) );
  if ( ! assert_flag ) {
    cerr << "Assert is disabled" << endl;
    return 1;
  }

  //********************************************************************

  // Verify that each generated value is different and is in range.
  cout << ok_prefix << "Test default sequence." << endl;
  double min = 2.0;
  double max = 5.0;
  long seed = 246813579L;
  RandomGeneratorTest rgen(min,max);
  cout << rgen << endl;
  int ntest = 20;
  int itest;
  vector<double> values;
  vector<double>::const_iterator ival;
  for ( itest=0; itest<ntest; ++itest ) {
    double newval;
    if ( itest == 4 || itest == 6 || itest == 7 ) newval = rgen.gauss();
    else newval = rgen.flat();
    cout << newval << endl;
    for ( ival=values.begin(); ival!=values.end(); ++ival ) {
      assert( newval != (*ival) );
    }
    if ( itest != 4 && itest != 6 && itest != 7 ) {
      assert( newval >= min );
      assert( newval <= max );
    }
    values.push_back(newval);
  }

  //********************************************************************

  // Verify that two generators starting with the same seed give the
  // same values.
  cout << ok_prefix << "Test sequence with seed." << endl;
  rgen.set_seed(seed);
  RandomGeneratorTest rgen2(min,max,seed);
  for ( itest=0; itest<ntest; ++itest ) {
    RandomGeneratorTest rgen3(rgen);
    double newval;
    double newval2;
    double newval3;
    if ( itest == 4 || itest == 6 || itest == 7 ) {
      newval = rgen.gauss();
      newval2 = rgen2.gauss();
      newval3 = rgen3.gauss();
    } else {
      newval = rgen.flat();
      newval2 = rgen2.flat();
      newval3 = rgen3.flat();
    }
    cout << newval << ' ' << newval2 << ' ' << newval3 << endl;
    assert( newval == newval2 );
  }

  //********************************************************************

  cout << ok_prefix << "Test copy constructor." << endl;
  // Note that rgen is not at the beginning of its sequence.
  // Copying the state should copy the seed and invoke the
  // the appropriate number of calls so the next value will be
  // the same for both objects.
  RandomGeneratorTest rgen3(rgen);
  for ( itest=0; itest<ntest; ++itest ) {
    double newval;
    double newval2;
    if ( itest == 4 || itest == 6 || itest == 7 ) {
      newval = rgen.gauss();
      newval2 = rgen3.gauss();
    } else {
      newval = rgen.flat();
      newval2 = rgen3.flat();
    }
    cout << newval << ' ' << newval2 << endl;
    assert( newval == newval2 );
  }

  //********************************************************************

  cout << ok_prefix << "Test copying state." << endl;
  rgen.set_seed(seed);
  RandomGeneratorTest rgen4(min,max);
  rgen4.set_state(rgen);
  for ( itest=0; itest<ntest; ++itest ) {
    double newval;
    double newval2;
    if ( itest == 4 || itest == 6 || itest == 7 ) {
      newval = rgen.gauss();
      newval2 = rgen4.gauss();
    } else {
      newval = rgen.flat();
      newval2 = rgen4.flat();
    }
    cout << newval << ' ' << newval2 << endl;
    assert( newval == newval2 );
  }

  //********************************************************************

  cout << ok_prefix
       << "------------- All tests passed. -------------" << endl;
  return 0;

  //********************************************************************

}