make-panel.py

import os, sys
import matplotlib.pyplot as plt
from matplotlib import image
import numpy as np
import getopt

import matplotlib as mpl
mpl.rcParams['figure.dpi']= 500

#=========================================================================
class MakePanelPlot():
  def __init__(self, debug=0, output=0, casename='sfcship_ps',
               varname='surface_pressure'):
    self.debug = debug
    self.output = output
    self.casename = casename
    self.varname = varname

    if(self.debug):
      print('debug = ', debug)

    if(self.debug > 10):
      print('self.casename = ', self.casename)
      print('self.varname = ', self.varname)

    self.genit()

  def genit(self):
   #initialise grid of axes
   #fig, axes = plt.subplots(2, 2, constrained_layout=True)
    fig, axes = plt.subplots(2, 2)
    axes = axes.ravel()

    imgdir = '%s/%s' %(self.casename, self.varname)

   #create fake data
    img = [
        'FAKENAME_30.png',
        'FAKENAME_40.png',
        'FAKENAME_50.png',
        'FAKENAME_60.png',
    ]

   #iterate over axes
    for i, ax in enumerate(axes):
     #imageprefix = '%s_%s_increment_lev' %(casename, varname)
      imageprefix = '%s_lev' %(casename)
      imgname = img[i].replace('FAKENAME', imageprefix)
      im = image.imread(imgname)
      ax.axis('off')
      ax.imshow(im)

   #plt.subplots_adjust(left=0.05, bottom=0.05,
   #                    right=0.95, top=0.95,
   #                    wspace=0.05, hspace=0.05)
   #The parameter meanings (and suggested defaults) are:
   #left  = 0.125  # the left side of the subplots of the figure
   #right = 0.9    # the right side of the subplots of the figure
   #bottom = 0.1   # the bottom of the subplots of the figure
   #top = 0.9      # the top of the subplots of the figure
   #wspace = 0.2   # the amount of width reserved for blank space between subplots
   #hspace = 0.2   # the amount of height reserved for white space between subplots

    plt.tight_layout()

    if(self.output):
      imgname = 'panel-%s-%s' %(casename, varname)
     #fig.savefig(imgname, dpi=300, bbox_inches="tight")
      fig.savefig(imgname, bbox_inches="tight", quality=100)
    else:
      plt.show()

#------------------------------------------------------------------------------
if __name__ == '__main__':
  debug = 1
  output = 0
  casename = 'sfcship_ps'
  varname = 'surface_pressure'

  opts, args = getopt.getopt(sys.argv[1:], '', ['debug=', 'output=',
                             'casename=', 'varname='])

  for o, a in opts:
    if o in ('--debug'):
      debug = int(a)
    elif o in ('--output'):
      output = int(a)
    elif o in ('--casename'):
      casename = a
    elif o in ('--varname'):
      varname = a
   #else:
   #  assert False, 'unhandled option'

  print('debug = ', debug)
  print('output = ', output)

  mpp = MakePanelPlot(debug=debug, output=output,
                      casename=casename,
                      varname=varname)