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

using namespace std;

int main ( );
int i4_min ( int i1, int i2 );
double r8vec_amax ( int n, double r8vec[] );
double *r8vec_even ( int n, double alo, double ahi );
void report ( int step, int step_num, int n, double x[], double t, double u[],
  double stability );
void timestamp ( );
double u_a ( double x, double t );
double u_b ( double x, double t );
void u_init ( int n, double x[], double t, double u[] );

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

int main ( )

//****************************************************************************80
//
//  Purpose:
//
//    FD1D_BURGERS_LAX: nonviscous Burgers equation, Lax-Wendroff method.
//
//  Licensing:
//
//    This code is distributed under the MIT license.
//
//  Modified:
//
//    22 August 2010
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    None
//
{
  double a;
  double b;
  double dt;
  double dx;
  int i;
  int ihi;
  int ilo;
  int n;
  double stability;
  int step;
  int step_num;
  double t;
  double t_init;
  double t_last;
  double *un;
  double *uo;
  double *x;

  timestamp ( );

  cout << "\n";
  cout << "FD1D_BURGERS_LAX:\n";
  cout << "  C++ version\n";
  cout << "  Solve the non-viscous time-dependent Burgers equation,\n";
  cout << "  using the Lax-Wendroff method.\n";
  cout << "\n";
  cout << "  Equation to be solved:\n";
  cout << "\n";
  cout << "    du/dt + u * du/dx = 0\n";
  cout << "\n";
  cout << "  for x in [ a, b ], for t in [t_init, t_last]\n";
  cout << "\n";
  cout << "  with initial conditions:\n";
  cout << "\n";
  cout << "    u(x,o) = u_init\n";
  cout << "\n";
  cout << "  and boundary conditions:\n";
  cout << "\n";
  cout << "    u(a,t) = u_a(t), u(b,t) = u_b(t)\n";
//
//  Set and report the problem parameters.
//
  n = 41;
  a = -1.0;
  b = +1.0;
  dx = ( b - a ) / ( double ) ( n - 1 );
  step_num = 80;
  t_init = 0.0;
  t_last = 1.0;
  dt = ( t_last - t_init ) / ( double ) ( step_num );

  cout << "\n";
  cout << "  " << a << " <= X <= " << b << "\n";
  cout << "  Number of nodes = " << n << "\n";
  cout << "  DX = "  << dx << "\n";
  cout << "\n";
  cout << "  " << t_init << " <= T <= " << t_last << "\n";
  cout << "  Number of time steps = " << step_num << "\n";
  cout << "  DT = " << dt << "\n";

  un = new double[n];
  uo = new double[n];

  x = r8vec_even ( n, a, b );

  cout << "\n";
  cout << "  X:\n";
  cout << "\n";
  for ( ilo = 0; ilo < n; ilo = ilo + 5 )
  {
    ihi = i4_min ( ilo + 5, n - 1 );
    for ( i = ilo; i <= ihi; i++ )
    {
      cout << "  " << setw(14) << x[i];
    }
    cout << "\n";
  }
//
//  Set the initial condition,
//  and apply boundary conditions to first and last entries.
//
  step = 0;
  t = t_init;
  u_init ( n, x, t, un );
  un[0] = u_a ( x[0], t );
  un[n-1] = u_b ( x[n-1], t );

  stability = ( dt / dx ) * r8vec_amax ( n, un );
  report ( step, step_num, n, x, t, un, stability );
//
//  Use the Lax-Wendroff method.
//
  for ( step = 1; step <= step_num; step++ )
  {
    t = ( ( double ) ( step_num - step ) * t_init   
        + ( double ) (            step ) * t_last ) 
        / ( double ) ( step_num        );

    for ( i = 0; i < n; i++ )
    {
      uo[i] = un[i];
    }

    for ( i = 1; i < n - 1; i++ )
    {
      un[i] = uo[i] 
        - ( dt / dx ) * ( pow ( uo[i+1], 2 ) - pow ( uo[i-1], 2 ) ) 
        + 0.5 * ( dt * dt / dx / dx ) * 0.5 *
        (   ( uo[i+1] + uo[i] ) * ( pow ( uo[i+1], 2 ) - pow ( uo[i], 2 ) ) 
          - ( uo[i] + uo[i-1] ) * ( pow ( uo[i], 2 ) - pow ( uo[i-1], 2 ) ) );
    }

    un[0] = u_a ( x[0], t );
    un[n-1] = u_b ( x[n-1], t );

    stability = ( dt / dx ) * r8vec_amax ( n, un );
    report ( step, step_num, n, x, t, un, stability );
  }

  delete [] un;
  delete [] uo;
  delete [] x;
//
//  Terminate.
//
  cout << "\n";
  cout << "FD1D_BURGERS_LAX:\n";
  cout << "  Normal end of execution.\n";

  cout << "\n";
  timestamp ( );

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

int i4_min ( int i1, int i2 )

//****************************************************************************80
//
//  Purpose:
//
//    I4_MIN returns the minimum of two I4's.
//
//  Licensing:
//
//    This code is distributed under the MIT license. 
//
//  Modified:
//
//    13 October 1998
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, int I1, I2, two integers to be compared.
//
//    Output, int I4_MIN, the smaller of I1 and I2.
//
{
  int value;

  if ( i1 < i2 )
  {
    value = i1;
  }
  else
  {
    value = i2;
  }
  return value;
}
//****************************************************************************80

double r8vec_amax ( int n, double a[] )

//****************************************************************************80
//
//  Purpose:
//
//    R8VEC_AMAX returns the maximum absolute value in an R8VEC.
//
//  Discussion:
//
//    An R8VEC is a vector of R8's.
//
//  Licensing:
//
//    This code is distributed under the MIT license. 
//
//  Modified:
//
//    18 September 2005
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, int N, the number of entries in the array.
//
//    Input, double A[N], the array.
//
//    Output, double AMAX, the value of the entry
//    of largest magnitude.
//
{
  double amax;
  int i;

  amax = 0.0;
  for ( i = 0; i < n; i++ )
  {
    if ( amax < fabs ( a[i] ) )
    {
      amax = fabs ( a[i] );
    }
  }

  return amax;
}
//****************************************************************************80

double *r8vec_even ( int n, double alo, double ahi )

//****************************************************************************80
//
//  Purpose:
//
//    R8VEC_EVEN returns an R8VEC of values evenly spaced between ALO and AHI.
//
//  Discussion:
//
//    An R8VEC is a vector of R8's.
//
//  Licensing:
//
//    This code is distributed under the MIT license. 
//
//  Modified:
//
//    18 May 2009
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, int N, the number of values.
//
//    Input, double ALO, AHI, the low and high values.
//
//    Output, double R8VEC_EVEN[N], N evenly spaced values.
//    Normally, A[0] = ALO and A[N-1] = AHI.
//    However, if N = 1, then A[0] = 0.5*(ALO+AHI).
//
{
  double *a;
  int i;

  a = new double[n];

  if ( n == 1 )
  {
    a[0] = 0.5 * ( alo + ahi );
  }
  else
  {
    for ( i = 0; i < n; i++ )
    {
      a[i] = ( ( double ) ( n - i - 1 ) * alo 
             + ( double ) (     i     ) * ahi ) 
             / ( double ) ( n     - 1 );
    }
  }

  return a;
}
//****************************************************************************80

void report ( int step, int step_num, int n, double x[], double t, double u[],
  double stability )

//****************************************************************************80
//
//  Purpose:
//
//    REPORT prints or plots or saves the data at the current time step.
//
//  Licensing:
//
//    This code is distributed under the MIT license.
//
//  Modified:
//
//    22 August 2010
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, int STEP, the index of the current step,
//    between 0 and STEP_NUM.
//
//    Input, int STEP_NUM, the number of steps to take.
//
//    Input, int N, the number of nodes.
//
//    Input, double X[N], the coordinates of the nodes.
//
//    Input, double T, the current time.
//
//    Input, double U[N], the initial values U(X,T).
//
//    Input, double STABILITY, the stability factor, which should be 
//    no greater than 1.
//
{
  int i;
  int ihi;
  int ilo;

  cout << "\n";
  cout << "  STEP = " << step << "\n";
  cout << "  TIME = " << t << "\n";
  cout << "  STABILITY = " << stability << "\n";
  cout << "\n";
  for ( ilo = 0; ilo < n; ilo = ilo + 5 )
  {
    ihi = i4_min ( ilo + 4, n - 1 );
    for ( i = ilo; i <= ihi; i++ )
    {
      cout << "  " << setw(14) << u[i];
    }
    cout << "\n";
  }

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

void timestamp ( )

//****************************************************************************80
//
//  Purpose:
//
//    TIMESTAMP prints the current YMDHMS date as a time stamp.
//
//  Example:
//
//    31 May 2001 09:45:54 AM
//
//  Licensing:
//
//    This code is distributed under the MIT license. 
//
//  Modified:
//
//    08 July 2009
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    None
//
{
# define TIME_SIZE 40

  static char time_buffer[TIME_SIZE];
  const struct std::tm *tm_ptr;
  std::time_t now;

  now = std::time ( NULL );
  tm_ptr = std::localtime ( &now );

  std::strftime ( time_buffer, TIME_SIZE, "%d %B %Y %I:%M:%S %p", tm_ptr );

  std::cout << time_buffer << "\n";

  return;
# undef TIME_SIZE
}
//****************************************************************************80

double u_a ( double x, double t )

//****************************************************************************80
//
//  Purpose:
//
//    U_A sets the boundary condition for U at A.
//
//  Licensing:
//
//    This code is distributed under the MIT license.
//
//  Modified:
//
//    18 August 2010
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, double X, T, the position and time.
//
//    Output, double U_A, the prescribed value of U(X,T).
//
{
  double ua;

  ua = + 0.5;

  return ua;
}
//****************************************************************************80

double u_b ( double x, double t )

//****************************************************************************80
//
//  Purpose:
//
//    U_B sets the boundary condition for U at B.
//
//  Licensing:
//
//    This code is distributed under the MIT license.
//
//  Modified:
//
//    18 August 2010
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, double X, T, the position and time.
//
//    Output, double U_B, the prescribed value of U(X,T).
//
{
  double ub;

  ub = - 0.5;

  return ub;
}
//****************************************************************************80

void u_init ( int n, double x[], double t, double u[] )

//****************************************************************************80
//
//  Purpose:
//
//    U_INIT sets the initial condition for U.
//
//  Licensing:
//
//    This code is distributed under the MIT license.
//
//  Modified:
//
//    18 August 2010
//
//  Author:
//
//    John Burkardt
//
//  Parameters:
//
//    Input, int N, the number of nodes.
//
//    Input, double X[N], the coordinates of the nodes.
//
//    Input, double T, the current time.
//
//    Output, double U[N], the initial values U(X,T).
//
{
  int i;
  double r8_pi = 3.141592653589793;
  double q;
  double r;
  double s;
  double ua;
  double ub;

  ua = u_a ( x[0], t );
  ub = u_b ( x[n-1], t );

  q = 2.0 * ( ua - ub ) / r8_pi;
  r = ( ua + ub ) / 2.0;
//
//  S can be varied.  It is the slope of the initial condition at the midpoint.
//
  s = 1.0;

  for ( i = 0; i < n; i++ )
  {
    u[i] = - q * atan ( s * ( 2.0 * x[i] - x[0] - x[n-1] ) 
      / ( x[n-1] - x[0] ) ) + r;
  }

  return;
}