source: branches/0.10-stable/test/rng-mt.cc @ 2936

// $Id: rng-mt.cc 2936 2013-01-01 06:58:24Z peter $

/*
  Copyright (C) 2012 Peter Johansson

  This file is part of the yat library, http://dev.thep.lu.se/yat

  The yat library is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License as
  published by the Free Software Foundation; either version 3 of the
  License, or (at your option) any later version.

  The yat library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with yat. If not, see <http://www.gnu.org/licenses/>.
*/

#include <config.h>

#include "Suite.h"

#include "yat/random/random.h"

#include <boost/thread.hpp>

#include <algorithm>
#include <cstdlib>
#include <iostream>
#include <iterator>
#include <stdexcept>
#include <vector>

using namespace theplu::yat;
using namespace theplu::yat::random;

class Visitor
{
public:
  Visitor(std::vector<int>& x)
    : begin_(x.begin()), end_(x.end()) {}

  void operator()(void)
  {
    try {
      for (std::vector<int>::iterator i=begin_; i!=end_; ++i)
        *i = rnd_(100);
    }
    catch (std::exception& e) {
      std::cerr << e.what() << "\n";
      exit(EXIT_FAILURE);
    }
  }

private:
  DiscreteUniform rnd_;
  std::vector<int>::iterator begin_;
  std::vector<int>::iterator end_;
};

void test1(test::Suite& suite);
void test2(test::Suite& suite);
void test3(test::Suite& suite);

int main(int argc, char* argv[])
{
  theplu::yat::test::Suite suite(argc, argv);
  test1(suite);
  test2(suite);
  test3(suite);

  return suite.return_value();
}


void test1(test::Suite& suite)
{
  suite.out() << "test1\n";
  boost::thread_group threads;

  std::vector<int> x(10);
  std::vector<int> y(x);

  Visitor visitor1(x);
  Visitor visitor2(y);
  threads.create_thread(visitor1);
  threads.create_thread(visitor2);
  threads.join_all();

  suite.out() << "x: ";
  std::copy(x.begin(), x.end(), std::ostream_iterator<int>(suite.out(), " "));
  suite.out() << "\ny: ";
  std::copy(y.begin(), y.end(), std::ostream_iterator<int>(suite.out(), " "));
  suite.out() << std::endl;

  if (x==y) {
    suite.add(false);
    suite.err() << "x and y are equal\n";
  }
}


std::vector<int> rnd_vec(void)
{
  std::vector<int> x(10);
  Visitor visitor1(x);
  boost::thread_group threads;
  threads.create_thread(visitor1);
  threads.join_all();
  return x;
}


void test2(test::Suite& suite)
{
  suite.out() << "test2\n";
  // test that we get same number with same seed
  RNG::instance()->seed(0);
  std::vector<int> y = rnd_vec();
  suite.out() << "y: ";
  std::copy(y.begin(), y.end(), std::ostream_iterator<int>(suite.out(), " "));
  suite.out() << std::endl;

  RNG::instance()->seed(0);
  std::vector<int> x = rnd_vec();
  suite.out() << "x: ";
  std::copy(x.begin(), x.end(), std::ostream_iterator<int>(suite.out(), " "));
  suite.out() << std::endl;
  if (x!=y) {
    suite.add(false);
    suite.err() << "x not equal to y\n";
  }
  else {
    suite.out() << "ok\n";
  }
}


void test3(test::Suite& suite)
{
  suite.out() << "test3\n";
  // test that we get different numbers in thread in which we seed RNG
  // and subsequent thread
  RNG::instance()->seed(1);
  std::vector<int> x(10);
  Visitor visitor(x);
  visitor();
  suite.out() << "x: ";
  std::copy(x.begin(), x.end(), std::ostream_iterator<int>(suite.out(), " "));
  suite.out() << std::endl;

  std::vector<int> y = rnd_vec();
  suite.out() << "y: ";
  std::copy(y.begin(), y.end(), std::ostream_iterator<int>(suite.out(), " "));
  suite.out() << std::endl;

  if (x==y) {
    suite.add(false);
    suite.err() << "x equal to y\n";
  }
  else {
    suite.out() << "ok\n";
  }
}