STN_NAME, STN_ID, LAT, LON = read_tslist.read_tslist(wrf_dir+"tslist")
FIGDIR = '/uufs/chpc.utah.edu/common/home/u0553130/public_html/MS/timeheight/'
if not os.path.exists(FIGDIR):
os.makedirs(FIGDIR)
for s in np.arange(0,len(STN_ID)):
station = STN_ID[s]
st_name = STN_NAME[s]
print 'working on',station
plt.cla()
plt.clf()
plt.close()
# Open the geopotential height file
dates, PH = WRF_timeseries.get_full_vert(wrf_dir+station+'.d02.PH',model_start)
print 'got PH'
TH = WRF_timeseries.get_full_vert(wrf_dir+station+'.d02.TH',model_start)[1]
print 'got TH'
VV = WRF_timeseries.get_full_vert(wrf_dir+station+'.d02.VV',model_start)[1]
print 'got VV'
UU = WRF_timeseries.get_full_vert(wrf_dir+station+'.d02.UU',model_start)[1]
print 'got UU'
QV = WRF_timeseries.get_full_vert(wrf_dir+station+'.d02.QV',model_start)[1]
QV = QV*1000 #convert from kg/kg to g/kg
print 'got QV'
# Since pcolormesh doesn't like datetime on the xaxis I need to get a
# little creative making my own tick labels
# Assign x-axis variables
