import os,os.path,sys
import traceback

"""testMadrigalBuild.py verifies whether Madrigal has been successfully built

Usage: python testMadrigalBuild.py [MADROOT]

If optional argument MADROOT given, search there for test Madrigal file to plot

$Id: testMadrigalBuild.py 3304 2011-01-17 15:25:59Z brideout $
"""

print """This script will attempt to plot some Madrigal test data.
If you see a Pcolor plot that shows electron density versus
altitude, the script succeeded.  Close the display window
when you're done examining it.  This may take a few minutes.\n"""

import warnings
warnings.filterwarnings('ignore', '.*', UserWarning)

try:
    import numpy
    import matplotlib.pylab
    import madrigal
    import madrigal._Madrec
    import madrigal.ui.madrigalPlot
    
except:
    traceback.print_exc()
    print 'FAILURE: REQUIRED MADRIGAL MODULES NOT INSTALLED'
    sys.exit(-1)
    

if len(sys.argv) > 2:
    print 'FAILURE: Illegal usage %s' % (str(sys.argv))
    sys.exit(-1)

if len(sys.argv) == 2:
    madrootDir = sys.argv[1]
    if not os.access(os.path.join(madrootDir, 'experiments'), os.R_OK):
        print 'FAILURE: No experiments directory found in %s' % (madrootDir)
        sys.exit(-1)
else:
    madDB = madrigal.metadata.MadrigalDB()
    madrootDir = madDB.getMadroot()

# get filename to read
filename = os.path.join(madrootDir, 'experiments/1998/mlh/20jan98/mil980120g.003')
if not os.access(filename, os.R_OK):
    filename = os.path.join(madrootDir, 'experiments/1998/mlh/20jan98/mlh980120g.001')
if not os.access(filename, os.R_OK):
    print 'FAILURE: Unable to find basic Madrigal test files installed in experiments/1998/mlh/20jan98'
    sys.exit(-1)

# check that tmpfile can be created
if not os.access('/tmp/', os.W_OK):
    print 'FAILURE: test cannot write to /tmp directory'
    sys.exit(-1)
tmpfile = '/tmp/isprintTest.txt'
tmpplot = '/tmp/isprint.png'

try:
    madrigal._Madrec.getIsprintReport(filename,
                                      '',
                                      ['uth','gdalt','nel'],
                                      ['1'],
                                      ['uth'],
                                      [None],
                                      [1],
                                      [0.0],
                                      [24.0],
                                      0,
                                      0,
                                      0,
                                      'missing',
                                      'assumed',
                                      'knownBad',
                                      tmpfile)
except:
    traceback.print_exc()
    print 'FAILURE: isprint failed'
    sys.exit(-1)

# read isprint string
f = open(tmpfile)
isprintText = f.read()
f.close()
os.remove(tmpfile)

# produce pcolor plot
try:
    madrigal.ui.madrigalPlot.madPcolorPlot(isprintText,
                                           'Nel (log(m^-3)) - Millstone Hill - Test day Jan 20, 1998\nClose this Window if okay',
                                           'Hours since midnight UT Jan 20, 1998',
                                           'Altitude (km)',
                                           tmpplot,
                                           size = 'large',
                                           minColormap = 9,
                                           maxColormap = 12)
except:
    traceback.print_exc()
    print 'FAILURE: plotting failed'
    sys.exit(-1)

os.system('display %s' % (tmpplot))

os.remove(tmpplot)

print 'SUCCESS - Madrigal build completed successfully'