#include <stdio.h>
#include <gsl/gsl_rng.h>
#include <gsl/gsl_randist.h>
#include <time.h>
#include "mkl_vsl.h"
#include <iostream>
using namespace std;

int main (int argc, char *argv[])
{
  timespec tS, tE;
  VSLStreamStatePtr stream;
  gsl_rng *r;                         /* pointer to RNG */
  int i,n;
  double u, t;

  n=atoi(argv[1]);

/* GSL generator */

  r = gsl_rng_alloc(gsl_rng_mt19937); /* allocate generator */
  gsl_rng_set(r,5489);                /* seed the generator */

  clock_gettime(CLOCK_REALTIME, &tS);
  for (i=0;i<n;i++)
    {
      u=gsl_rng_uniform(r);           /* generate random numbers */
    }
  clock_gettime(CLOCK_REALTIME, &tE);
  t= tE.tv_sec - tS.tv_sec +
            (tE.tv_nsec - tS.tv_nsec) / 1000000000.;
  cout << "Time taken is: " << t << endl;

  gsl_rng_free(r);                    /* delete generator */

/* MKL generator */

  vslNewStream( &stream, VSL_BRNG_MT19937, 5489 ); /* seed the generator */

  clock_gettime(CLOCK_REALTIME, &tS);
  for (i=0;i<n;i++)
    {
      vdRngUniform(VSL_RNG_METHOD_UNIFORM_STD, stream, 1, &u, 0.0, 1.0);    
    }
  clock_gettime(CLOCK_REALTIME, &tE);  
  t= tE.tv_sec - tS.tv_sec +
            (tE.tv_nsec - tS.tv_nsec) / 1000000000.;
  cout << "Time taken is: " << t << endl;   

  vslDeleteStream( &stream );                     /* delete generator */

  vslNewStream( &stream, VSL_BRNG_SFMT19937, 5489 ); /* seed the generator */

  clock_gettime(CLOCK_REALTIME, &tS);
  for (i=0;i<n;i++)
    {
      vdRngUniform(VSL_RNG_METHOD_UNIFORM_STD, stream, 1, &u, 0.0, 1.0);    
    }
  clock_gettime(CLOCK_REALTIME, &tE);  
  t= tE.tv_sec - tS.tv_sec +
            (tE.tv_nsec - tS.tv_nsec) / 1000000000.;
  cout << "Time taken is: " << t << endl;   

  vslDeleteStream( &stream );                     /* delete generator */

  return(0);
}