#! /usr/bin/env python3
#
def grid_fill_contour ( ):

#*****************************************************************************80
#
## grid_fill_contour draws a filled color contour plot from a table Z(X,Y).
#
#  Discussion:
#
#    Here, the grid is a 41x41 array of equally spaced points in [-2,2]x[-2,2].
#
#      x = linspace ( -2.0, +2.0, 41 );
#      y = linspace ( -2.0, +2.0, 41 );
#      [ X, Y ] = meshgrid ( x, y );
#
#    and Z(X,Y) is a function which could be evaluated by:
#
#      Z = exp ( - ( X^2 + Y^2 ) ) ...
#       * cos ( 0.25 * X ) ...
#       * sin ( Y ) ...
#       * cos ( 2 * (X^2 + Y^2 ) );
#
#  Licensing:
#
#    This code is distributed under the GNU LGPL license.
#
#  Modified:
#
#    03 May 2019
#
#  Author:
#
#    John Burkardt
#
  import matplotlib.pyplot as plt
  import numpy as np
  import platform
  from matplotlib import cm

  print ( '' )
  print ( 'grid_fill_contour:' )
  print ( '  Python version: %s' % ( platform.python_version ( ) ) )
  print ( '  Make a contour plot Z(X,Y) for a 41x41 table.' )
#
#  Read the data from the file.
#
  x = []
  y = []
  z = []

  file = open ( 'grid_data.txt', 'r' )

  for line in file:
    line = line.strip ( )
    columns = line.split ( )
    x.append ( float ( columns[0] ) )
    y.append ( float ( columns[1] ) )
    z.append ( float ( columns[2] ) )

  file.close ( )
#
#  The data was stored as a vector.
#  Reshape it to 41x41 arrays.
#
  x = np.reshape ( x, ( 41, 41 ) )
  y = np.reshape ( y, ( 41, 41 ) )
  z = np.reshape ( z, ( 41, 41 ) )
#
#  Form the figure.
#
  levels = 15
  ax = plt.figure ( )
  plt.contourf ( x, y, z, levels, cmap = cm.coolwarm )
  plt.title ( 'Filled contour plot of 41x41 table Z(X,Y)', fontsize = 16 )
  plt.xlabel ( '<--- X --->', fontsize = 16 )
  plt.ylabel ( '<--- Y --->', fontsize = 16 )
  filename = 'grid_fill_contour.png'
  plt.savefig ( filename )
  print ( '' )
  print ( '  Graphics saved as "%s"' % ( filename ) )
  plt.show ( )
#
#  Terminate.
#
  print ( '' )
  print ( 'grid_fill_contour:' )
  print ( '  Normal end of execution.' )
  return

if ( __name__ == '__main__' ):
  from timestamp import timestamp
  timestamp ( )
  grid_fill_contour ( )
  timestamp ( )