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

using namespace std;

# include "image_denoise.hpp"

int main ( int argc, char *argv[] );
void test01 ( );
void timestamp ( );

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

int main ( int argc, char *argv[] )

//****************************************************************************80
//
//  Purpose:
//
//    image_denoise_test() tests image_denoise().
//
//  Licensing:
//
//    This code is distributed under the MIT license.
//
//  Modified:
//
//    21 July 2011
//
//  Author:
//
//    John Burkardt
//
{
  timestamp ( );
  cout << "\n";
  cout << "IMAGE_DENOISE_TEST():\n";
  cout << "  C++ version\n";
  cout << "  Test IMAGE_DENOISE().\n";

  test01 ( );
//
//  Terminate.
//
  cout << "\n";
  cout << "IMAGE_DENOISE_TEST():\n";
  cout << "  Normal end of execution.\n";
  cout << "\n";
  timestamp ( );

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

void test01 ( )

//****************************************************************************80
//
//  Purpose:
//
//    TEST01 tests GRAY_MEDIAN_NEWS.
//
//  Licensing:
//
//    This code is distributed under the MIT license.
//
//  Modified:
//
//    22 July 2011
//
//  Author:
//
//    John Burkardt
//
{
  int *g;
  int g_max;
  int *g2;
  string input_filename = "glassware_noisy.ascii.pgm";
  ifstream input_unit;
  int m;
  int n;
  string output_filename = "glassware_median_news.ascii.pgm";

  cout << "\n";
  cout << "TEST01:\n";
  cout << "  GRAY_MEDIAN_NEWS uses a NEWS median filter \n";
  cout << "  on a noisy grayscale image.\n";

  cout << "\n";
  cout << "  The input file is \"" << input_filename << "\".\n";
//
//  Open the input file and read the data.
//
  input_unit.open ( input_filename.c_str ( ) );

  if ( !input_unit )
  {
    cerr << "\n";
    cerr << "TEST01 - Fatal error!\n";
    cerr << "  Could not open the file \"" << input_filename << "\"\n";
    exit ( 1 );
  }

  pgma_read_header ( input_unit, &m, &n, &g_max );

  cout << "\n";
  cout << "  Number of rows =          " << m << "\n";
  cout << "  Number of columns =       " << n << "\n";
  cout << "  Maximum pixel intensity = " << g_max << "\n";

  g = new int[m*n];

  pgma_read_data ( input_unit, m, n, g );

  input_unit.close ( );

  g2 = gray_median_news ( m, n, g );
//
//  Write the denoised images.
//
  pgma_write ( output_filename, m, n, g2 );

  cout << "\n";
  cout << "  Wrote denoised image to \"" << output_filename << "\".\n";

  delete [] g;
  delete [] g2;

  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:
//
//    19 March 2018
//
//  Author:
//
//    John Burkardt
//
{
# 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
}