# include <cstdlib>
# include <iostream>
# include <iomanip>
# include <cmath>
# include <string>
# include <fstream>
# include <ctime>

using namespace std;

# include "colored_noise.hpp"

int main ( );
void test01 ( int n, double q_d, double alpha, int seed );

//****************************************************************************80

int main ( )

//****************************************************************************80
//
//  Purpose:
//
//    MAIN is the main program for COLORED_NOISE_PRB.
//
//  Discussion:
//
//    COLORED_NOISE_PRB tests the COLORED_NOISE library.
//
//  Licensing:
//
//    This code is distributed under the GNU LGPL license. 
//
//  Modified:
//
//    09 June 2010
//
//  Author:
//
//    John Burkardt
//
{
  double alpha;
  int i;
  int n;
  double q_d;
  int seed_init;

  timestamp ( );
  cout << "\n";
  cout << "COLORED_NOISE_PRB\n";
  cout << "  C++ version\n";
  cout << "  Test the COLORED_NOISE library.\n";

  n = 128;
  q_d = 1.0;
  alpha = 0.00;
  seed_init = 123456789;

  for ( i = 0; i <= 8; i++ )
  {
    alpha = 0.25 * i;
    test01 ( n, q_d, alpha, seed_init );
  }
//
//  Terminate.
//
  cout << "\n";
  cout << "COLORED_NOISE_PRB:\n";
  cout << "  Normal end of execution.\n";
  cout << "\n";
  timestamp ( );

  return 0;
}
//****************************************************************************80

void test01 ( int n, double q_d, double alpha, int seed_init )

//****************************************************************************80
//
//  Purpose:
//
//    TEST01 calls F_ALPHA with particular parameters.
//
//  Licensing:
//
//    This code is distributed under the GNU LGPL license. 
//
//  Modified:
//
//    09 June 2010
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, int N, the number of elements of the sequence to generate.
//
//    Input, double Q_D, the variance of the sequence.
//
//    Input, double ALPHA, the exponent of the power law.
//
//    Input, int SEED_INIT, the initial seed for the random number generator.
//
{
  int i;
  string output_filename;
  ofstream output_unit;
  int seed;
  double *x;

  output_filename = "alpha_" + r8_to_string ( alpha, "%4.2f"  ) + ".txt";
//
//  Report parameters.
//
  cout << "\n";
  cout << "TEST01:\n";
  cout << "  Generating " << n << " sample points.\n";
  cout << "  1/F^ALPHA noise has ALPHA = " << alpha << "\n";
  cout << "  Variance is " << q_d << "\n";
  cout << "  Initial random number seed = " << seed_init << "\n";

  seed = seed_init;

  x = f_alpha ( n, q_d, alpha, &seed );
//
//  Print no more than 10 entries of the data.
//
  r8vec_print_part ( n, x, 10, "  Noise sample:" );
//
//  Write the data to a file.
//
  output_unit.open ( output_filename.c_str ( ) );

  for ( i = 0; i < n; i++ )
  {
    output_unit << x[i] << "\n";
  }
  output_unit.close ( );

  cout << "  Data written to file \"" << output_filename << "\"\n";

  delete [] x;

  return;
}