from rass_thetav import get_thetav
from glob import glob
from rv_utilities import add_colorbar, format_xaxis, format_yaxis
homedir = os.path.expanduser('~')
for r in range(8, 14):
''' case 14 does not have RASS obs '''
print 'case '+str(r)
print '-----------------'
fs = glob(homedir+'/RASS/c'+str(r).zfill(2)+'/*.cns')
fs.sort()
for n, f in enumerate(fs):
T, Tc, Hgt, timestamp = mf.parse_rass(f)
if n == 0:
Tstack = T
Hgtstack = Hgt
ts = np.array(timestamp)
else:
Tstack = np.hstack((Tstack, T))
Hgtstack = np.hstack((Hgtstack, Hgt))
ts = np.hstack((ts, np.array(timestamp)))
thetav = get_thetav(case=r, Tv_array=Tstack,
hgt_array=Hgtstack, homedir=homedir)
titletxt = 'Case{} ({})'
fig, ax = plt.subplots(2, 1, sharex=True)
im = ax[0].imshow(Tstack, interpolation='none', origin='lower')
ax[0].set_title(titletxt.format(str(r), ts[0].strftime('%Y-%b')))
