#!/usr/bin/env python

import numpy
import pylab
import sys
import getopt

import mesh.patch as patch

# plot an output file using the solver's dovis script

def makeplot(myd, solverName, outfile, W, H):

    exec 'import ' + solverName + ' as solver'

    sim = solver.Simulation(None, None)
    sim.cc_data = myd

    pylab.figure(num=1, figsize=(W,H), dpi=100, facecolor='w')

    sim.dovis()
    pylab.savefig(outfile)
    pylab.show()


def usage():
    usage="""
usage: plot.py [-h] [-o image.png] solver filename

positional arguments:
  solver        required inputs: solver name
  filename      required inputs: filename to read from

optional arguments:
  -h, --help    show this help message and exit
  -o image.png  output image name. The extension .png will generate a PNG
                file, .eps will generate an EPS file (default: plot.png).
  -W width      width in inches
  -H height     height in inches
"""
    print usage
    sys.exit()


if __name__== "__main__":

    try: opts, next = getopt.getopt(sys.argv[1:], "o:h:W:H:")
    except getopt.GetoptError:
        sys.exit("invalid calling sequence")

    outfile = "plot.png"

    W = 8.0
    H = 4.5

    for o, a in opts:
        if o == "-h": usage()
        if o == "-o": outfile = a
        if o == "-W": W = float(a)
        if o == "-H": H = float(a)

    try: solver = next[0]
    except: usage()
        
    try: file = next[1]
    except: usage()

    myg, myd = patch.read(file)

    makeplot(myd, solver, outfile, W, H)