'''
'''

file = '/air/rclare/CONDOR_OUTPUT/C12/Case12_T-4_no-alter_2005-03-07_06:00:00'

from commands import *						#Library for executing Linux commands
from numpy import *							#Library for math
from matplotlib.pyplot import *				#Library for plotting
from mpl_toolkits.basemap import Basemap	#Library for map projections

dom = [209,143]

#Store values for lat-lon/gridpoint conversion
f = open("/air/rclare/wrf_grid_data/wrf_ll.txt", 'r')
llcontent = f.readlines()		
f.close()

m = linspace(0.0,142,143)					#Latitude linespace
n = linspace(0.0,208,209)					#Longitude linespace

#Conversion info arrays
x_grid, y_grid, longitude, latitude = [], [], [], []
longitude_grid, latitude_grid = [], []
long_store, lat_store = [], []
#Parse the conversion file
l = 0
for i in llcontent:
	mark1 = i.find(",")
	mark2 = i.find("|")
	mark3 = i[mark2:].find(",")
	x_grid.append(int(i[:mark1]))
	y_grid.append(int(i[mark1+2:mark2-1]))
	longitude.append(float64(i[mark2+2:mark2+mark3]))
	latitude.append(float64(i[mark2+mark3+2:-2]))	

#Convert Lat/Lon values to a grid 
l = 0
for i in range(len(m)):
	for j in range(len(n)):
		long_store.append(longitude[l])
		lat_store.append(latitude[l])
		l+=1		
	longitude_grid.append(asarray(long_store))
	latitude_grid.append(asarray(lat_store))
	long_store = []
	lat_store = []
	
	
ncl_content = '\
a = addfile("%s", "r")        \n\
T = a->T2					  \n\
asciiwrite("T2_test.txt", T) \n' % file

f = open("temp_ncl.ncl",'w')				#Creates local temporary ncl script
f.write(ncl_content)						#Writes above contents into file
f.close()

g = getoutput('ncl temp_ncl.ncl; rm temp_ncl.ncl')	#Runs temporary ncl script
print g
f = open("T2_test.txt", "r")
content = map(float, f)
f.close()

content = asarray(content).reshape(dom[1], dom[0])


fig = figure()
			
map = Basemap(llcrnrlon=-126.5696,llcrnrlat=19.3052,urcrnrlon=-46.00888,urcrnrlat=53.53284,\
				rsphere=(6378137.00,6356752.3142),resolution='l',projection='lcc',\
				lat_0=43.5,lon_0=-98.,lat_1=30.,lat_2=60.)
				
map.drawcoastlines()
#map.fillcontinents()
map.drawstates()
map.drawcountries()

X,Y = map(asarray(longitude_grid), asarray(latitude_grid))

levels = linspace(230, 310, 17)								
CP = map.contourf(X,Y,content[:,:],levels)						#Creates contours with colors and data
cb = colorbar(CP)	

show()