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
pass # Get observed timeseries data obs = MesoWest_timeseries.get_mesowest_ts(station, start_date_early, end_date) obs_dates = obs['datetimes'] obs_spd = obs['wind speed'] obs_dir = obs['wind direction'] obs_temp = obs['temperature'] # calculate the U and V components from wind speed obs_u, obs_v = wind_calcs.wind_spddir_to_uv(obs_spd, obs_dir) # Get the WRF timeseries data wrf_dir = '/uufs/chpc.utah.edu/common/home/horel-group4/model/bblaylock/WRF3.7_lake299_ember/WRFV3/test/em_real/' wrf_name = 'lake299' wrf_temp = WRF_timeseries.get_ts_data(wrf_dir + station + '.d02.TS', 't') - 273.15 wrf_u = WRF_timeseries.get_ts_data(wrf_dir + station + '.d02.TS', 'u') wrf_v = WRF_timeseries.get_ts_data(wrf_dir + station + '.d02.TS', 'v') wrf_time = WRF_timeseries.get_ts_data(wrf_dir + station + '.d02.TS', 'ts_hour') # calculate the wind speed and direction from u and v wrf_spd = wind_calcs.wind_uv_to_spd(wrf_u, wrf_v) wrf_dir = wind_calcs.wind_uv_to_dir(wrf_u, wrf_v) # convert wrf_time to datetime wrf_dates = np.array([]) for i in wrf_time: wrf_dates = np.append(wrf_dates, start_date + timedelta(hours=i)) # Get the WRF timeseries data for a second series wrf_dir2 = '/uufs/chpc.utah.edu/common/home/horel-group4/model/bblaylock/WRF3.7_lake303_ember/WRFV3/test/em_real/' wrf_name2 = 'lake303'
print "skipped" pass # Get observed timeseries data obs = MesoWest_timeseries.get_mesowest_ts(station,start_date_early,end_date) obs_dates = obs['datetimes'] obs_spd = obs['wind speed'] obs_dir = obs['wind direction'] obs_temp = obs['temperature'] # calculate the U and V components from wind speed obs_u, obs_v = wind_calcs.wind_spddir_to_uv(obs_spd,obs_dir) # Get the WRF timeseries data wrf_dir = '/uufs/chpc.utah.edu/common/home/horel-group4/model/bblaylock/WRF3.7_lake299_ember/WRFV3/test/em_real/' wrf_name='lake299' wrf_temp = WRF_timeseries.get_ts_data(wrf_dir+station+'.d02.TS','t')-273.15 wrf_u = WRF_timeseries.get_ts_data(wrf_dir+station+'.d02.TS','u') wrf_v = WRF_timeseries.get_ts_data(wrf_dir+station+'.d02.TS','v') wrf_time = WRF_timeseries.get_ts_data(wrf_dir+station+'.d02.TS','ts_hour') # calculate the wind speed and direction from u and v wrf_spd = wind_calcs.wind_uv_to_spd(wrf_u,wrf_v) wrf_dir = wind_calcs.wind_uv_to_dir(wrf_u,wrf_v) # convert wrf_time to datetime wrf_dates = np.array([]) for i in wrf_time: wrf_dates = np.append(wrf_dates,start_date+timedelta(hours=i)) # Get the WRF timeseries data for a second series wrf_dir2 = '/uufs/chpc.utah.edu/common/home/horel-group4/model/bblaylock/WRF3.7_lake303_ember/WRFV3/test/em_real/' wrf_name2='lake303'