def point_fcst_according_to_3D_field(
model='ECMWF',
output_dir=None,
t_range=[0,60],
t_gap=3,
points={'lon':[116.3833], 'lat':[39.9], 'altitude':[1351]},
initTime=None,obs_ID=54511,day_back=0,
extra_info={
'output_head_name':' ',
'output_tail_name':' ',
'point_name':' ',
'drw_thr':True,
'levels_for_interp':[1000, 950, 925, 900, 850, 800, 700, 600, 500]}
):
try:
dir_rqd=[utl.Cassandra_dir(data_type='high',data_source=model,var_name='HGT',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='UGRD',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='VGRD',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='TMP',lvl=''),
utl.Cassandra_dir(data_type='surface',data_source=model,var_name='RAIN'+str(t_gap).zfill(2))]
except KeyError:
raise ValueError('Can not find all required directories needed')
#-get all the directories needed
if(initTime == None):
initTime = get_latest_initTime(dir_rqd[0][0:-1]+'/850')
#initTime=utl.filename_day_back_model(day_back=day_back,fhour=0)[0:8]
directory=dir_rqd[0][0:-1]
fhours = np.arange(t_range[0], t_range[1], t_gap)
filenames = [initTime+'.'+str(fhour).zfill(3) for fhour in fhours]
HGT_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
directory=dir_rqd[1][0:-1]
U_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
directory=dir_rqd[2][0:-1]
V_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
directory=dir_rqd[3][0:-1]
TMP_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
rn=utl.get_model_points_gy(dir_rqd[4], filenames, points,allExists=False)
directory=dir_rqd[3][0:-1]
coords_info_2D=utl.get_model_points_gy(directory+str(extra_info['levels_for_interp'][0])+'/',
points=points,filenames=filenames,allExists=False)
delt_xy=HGT_4D['lon'].values[1]-HGT_4D['lon'].values[0]
mask = (HGT_4D['lon']<(points['lon']+2*delt_xy))&(HGT_4D['lon']>(points['lon']-2*delt_xy))&(HGT_4D['lat']<(points['lat']+2*delt_xy))&(HGT_4D['lat']>(points['lat']-2*delt_xy))
HGT_4D_sm=HGT_4D['data'].where(mask,drop=True)
U_4D_sm=U_4D['data'].where(mask,drop=True)
V_4D_sm=V_4D['data'].where(mask,drop=True)
TMP_4D_sm=U_4D['data'].where(mask,drop=True)
lon_md=np.squeeze(HGT_4D_sm['lon'].values)
lat_md=np.squeeze(HGT_4D_sm['lat'].values)
alt_md=np.squeeze(HGT_4D_sm.values*10).flatten()
time_md=np.squeeze(HGT_4D_sm['forecast_period'].values)
coords = np.zeros((len(time_md),len(extra_info['levels_for_interp']),len(lat_md),len(lon_md),4))
coords[...,0]=time_md.reshape((len(time_md),1,1,1))
coords[...,2] = lat_md.reshape((1,1,len(lat_md),1))
coords[...,3] = lon_md.reshape((1,1,1,len(lon_md)))
coords = coords.reshape((alt_md.size,4))
coords[:,1]=alt_md
interpolator_U = LinearNDInterpolator(coords,U_4D_sm.values.reshape((U_4D_sm.values.size)),rescale=True)
interpolator_V = LinearNDInterpolator(coords,V_4D_sm.values.reshape((V_4D_sm.values.size)),rescale=True)
interpolator_TMP = LinearNDInterpolator(coords,TMP_4D_sm.values.reshape((TMP_4D_sm.values.size)),rescale=True)
coords2 = np.zeros((len(time_md),1,1,1,4))
coords2[...,0]=time_md.reshape((len(time_md),1,1,1))
coords2[...,1]=points['altitude'][0]
coords2[...,2] = points['lat'][0]
coords2[...,3] = points['lon'][0]
coords2 = coords2.reshape((time_md.size,4))
U_interped=np.squeeze(interpolator_U(coords2))
V_interped=np.squeeze(interpolator_V(coords2))
TMP_interped=np.squeeze(interpolator_TMP(coords2))
U_interped_xr=coords_info_2D.copy()
U_interped_xr['data'].values=U_interped.reshape(U_interped.size,1,1)
V_interped_xr=coords_info_2D.copy()
V_interped_xr['data'].values=V_interped.reshape(V_interped.size,1,1)
TMP_interped_xr=coords_info_2D.copy()
TMP_interped_xr['data'].values=TMP_interped.reshape(TMP_interped.size,1,1)
sta_graphics.draw_point_fcst(t2m=TMP_interped_xr,u10m=U_interped_xr,v10m=V_interped_xr,rn=rn,
model=model,
output_dir=output_dir,
points=points,
extra_info=extra_info
)
def point_wind_time_fcst_according_to_3D_wind(
model='ECMWF',
output_dir=None,
t_range=[0,60],
t_gap=3,
points={'lon':[116.3833], 'lat':[39.9], 'altitude':[1351]},
initTime=None,draw_obs=True,obs_ID=54511,day_back=0,
extra_info={
'output_head_name':' ',
'output_tail_name':' ',
'point_name':' ',
'drw_thr':True,
'levels_for_interp':[1000, 950, 925, 900, 850, 800, 700, 600, 500]}
):
#+get all the directories needed
try:
dir_rqd=[utl.Cassandra_dir(data_type='high',data_source=model,var_name='HGT',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='UGRD',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='VGRD',lvl='')]
except KeyError:
raise ValueError('Can not find all required directories needed')
#-get all the directories needed
if(initTime == None):
initTime = get_latest_initTime(dir_rqd[0][0:-1]+'/850')
#initTime=utl.filename_day_back_model(day_back=day_back,fhour=0)[0:8]
directory=dir_rqd[0][0:-1]
fhours = np.arange(t_range[0], t_range[1], t_gap)
filenames = [initTime+'.'+str(fhour).zfill(3) for fhour in fhours]
HGT_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
directory=dir_rqd[1][0:-1]
U_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
directory=dir_rqd[2][0:-1]
V_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
#obs
if(draw_obs == True):
initial_time=pd.to_datetime(str(V_4D['forecast_reference_time'].values)).replace(tzinfo=None).to_pydatetime()
sign=0
for ifhour in V_4D['forecast_period'].values:
temp=(initial_time+timedelta(hours=ifhour))
filenames_obs=temp.strftime("%Y%m%d%H")+'0000.000'
try:
obs_data=get_station_data('SURFACE/PLOT/',filename=filenames_obs)
except:
break
if(obs_data is not None):
temp=obs_data.where(obs_data['ID']==obs_ID).dropna(how='all')
if ((ifhour == V_4D['forecast_period'].values[0]) or ((ifhour > V_4D['forecast_period'].values[0]) and (sign==0))):
if(len(temp) > 0):
sta_obs_data=obs_data.where(obs_data['ID']==obs_ID).dropna(how='all').reset_index()
sign=1
else:
if(len(temp) > 0):
sta_obs_data=sta_obs_data.append(temp).reset_index()
if(obs_data is None):
break
try:
sta_obs_data
except:
draw_obs=False
delt_xy=HGT_4D['lon'].values[1]-HGT_4D['lon'].values[0]
mask = (HGT_4D['lon']<(points['lon']+2*delt_xy))&(HGT_4D['lon']>(points['lon']-2*delt_xy))&(HGT_4D['lat']<(points['lat']+2*delt_xy))&(HGT_4D['lat']>(points['lat']-2*delt_xy))
HGT_4D_sm=HGT_4D['data'].where(mask,drop=True)
U_4D_sm=U_4D['data'].where(mask,drop=True)
V_4D_sm=V_4D['data'].where(mask,drop=True)
lon_md=np.squeeze(HGT_4D_sm['lon'].values)
lat_md=np.squeeze(HGT_4D_sm['lat'].values)
alt_md=np.squeeze(HGT_4D_sm.values*10).flatten()
time_md=np.squeeze(HGT_4D_sm['forecast_period'].values)
coords = np.zeros((len(time_md),len(extra_info['levels_for_interp']),len(lat_md),len(lon_md),4))
coords[...,0]=time_md.reshape((len(time_md),1,1,1))
coords[...,2] = lat_md.reshape((1,1,len(lat_md),1))
coords[...,3] = lon_md.reshape((1,1,1,len(lon_md)))
coords = coords.reshape((alt_md.size,4))
coords[:,1]=alt_md
interpolator_U = LinearNDInterpolator(coords,U_4D_sm.values.reshape((U_4D_sm.values.size)),rescale=True)
interpolator_V = LinearNDInterpolator(coords,V_4D_sm.values.reshape((V_4D_sm.values.size)),rescale=True)
coords2 = np.zeros((len(time_md),1,1,1,4))
coords2[...,0]=time_md.reshape((len(time_md),1,1,1))
coords2[...,1]=points['altitude'][0]
coords2[...,2] = points['lat'][0]
coords2[...,3] = points['lon'][0]
coords2 = coords2.reshape((time_md.size,4))
U_interped=np.squeeze(interpolator_U(coords2))
V_interped=np.squeeze(interpolator_V(coords2))
time_info=HGT_4D_sm.coords
sta_graphics.draw_point_wind(U=U_interped,V=V_interped,
model=model,
output_dir=output_dir,
points=points,
time_info=time_info,
extra_info=extra_info
)
def Time_Crossection_rh_uv_Temp(
initTime=None,
model='ECMWF',
points={
'lon': [116.3833],
'lat': [39.9]
},
levels=[1000, 950, 925, 900, 850, 800, 700, 600, 500, 400, 300, 200],
t_gap=3,
t_range=[0, 48],
output_dir=None):
fhours = np.arange(t_range[0], t_range[1], t_gap)
# 读数据
try:
data_dir = [
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='TMP',
lvl=''),
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='UGRD',
lvl=''),
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='VGRD',
lvl=''),
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='RH',
lvl=''),
utl.Cassandra_dir(data_type='surface',
data_source=model,
var_name='PSFC')
]
except KeyError:
raise ValueError('Can not find all directories needed')
if (initTime == None):
initTime = get_latest_initTime(data_dir[0][0:-1] + "850")
filenames = [initTime + '.' + str(fhour).zfill(3) for fhour in fhours]
TMP_4D = get_model_3D_grids(directory=data_dir[0][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
TMP_2D = TMP_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
u_4D = get_model_3D_grids(directory=data_dir[1][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
u_2D = u_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
v_4D = get_model_3D_grids(directory=data_dir[2][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
v_2D = v_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
rh_4D = get_model_3D_grids(directory=data_dir[3][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
rh_2D = rh_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
rh_2D.attrs['model'] = model
rh_2D.attrs['points'] = points
Psfc_3D = get_model_grids(directory=data_dir[4][0:-1],
filenames=filenames,
allExists=False)
Psfc_1D = Psfc_3D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
v_2D2, pressure_2D = xr.broadcast(v_2D['data'], v_2D['level'])
v_2D2, Psfc_2D = xr.broadcast(v_2D['data'], Psfc_1D['data'])
terrain_2D = pressure_2D - Psfc_2D
crossection_graphics.draw_Time_Crossection_rh_uv_Temp(
rh_2D=rh_2D,
u_2D=u_2D,
v_2D=v_2D,
TMP_2D=TMP_2D,
terrain_2D=terrain_2D,
t_range=t_range,
model=model,
output_dir=output_dir)
def Time_Crossection_rh_uv_theta_e(
initTime=None,
model='ECMWF',
points={
'lon': [116.3833],
'lat': [39.9]
},
levels=[1000, 950, 925, 900, 850, 800, 700, 600, 500, 400, 300, 200],
t_gap=3,
t_range=[0, 48],
output_dir=None):
fhours = np.arange(t_range[0], t_range[1], t_gap)
# 读数据
try:
data_dir = [
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='TMP',
lvl=''),
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='UGRD',
lvl=''),
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='VGRD',
lvl=''),
utl.Cassandra_dir(data_type='high',
data_source=model,
var_name='RH',
lvl='')
]
except KeyError:
raise ValueError('Can not find all directories needed')
if (initTime == None):
initTime = get_latest_initTime(data_dir[0][0:-1] + "850")
filenames = [initTime + '.' + str(fhour).zfill(3) for fhour in fhours]
TMP_4D = get_model_3D_grids(directory=data_dir[0][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
TMP_2D = TMP_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
filenames = [initTime + '.' + str(fhour).zfill(3) for fhour in fhours]
u_4D = get_model_3D_grids(directory=data_dir[1][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
u_2D = u_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
filenames = [initTime + '.' + str(fhour).zfill(3) for fhour in fhours]
v_4D = get_model_3D_grids(directory=data_dir[2][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
v_2D = v_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
filenames = [initTime + '.' + str(fhour).zfill(3) for fhour in fhours]
rh_4D = get_model_3D_grids(directory=data_dir[3][0:-1],
filenames=filenames,
levels=levels,
allExists=False)
rh_2D = rh_4D.interp(lon=('points', points['lon']),
lat=('points', points['lat']))
rh_2D.attrs['model'] = model
rh_2D.attrs['points'] = points
Td_2D = mpcalc.dewpoint_rh(TMP_2D['data'].values * units.celsius,
rh_2D['data'].values * units.percent)
rh, pressure = xr.broadcast(rh_2D['data'], rh_2D['level'])
Theta_e = mpcalc.equivalent_potential_temperature(
pressure, TMP_2D['data'].values * units.celsius, Td_2D)
theta_e_2D = xr.DataArray(np.array(Theta_e),
coords=rh_2D['data'].coords,
dims=rh_2D['data'].dims,
attrs={'units': Theta_e.units})
crossection_graphics.draw_Time_Crossection_rh_uv_theta_e(
rh_2D=rh_2D,
u_2D=u_2D,
v_2D=v_2D,
theta_e_2D=theta_e_2D,
t_range=t_range,
output_dir=output_dir)
def point_fcst_tmp_according_to_3D_field_box_line(
output_dir=None,
t_range=[0,60],
t_gap=3,
points={'lon':[116.3833], 'lat':[39.9], 'altitude':[1351]},
initTime=None,obs_ID=54511,day_back=0,
extra_info={
'output_head_name':' ',
'output_tail_name':' ',
'point_name':' ',
'drw_thr':True,
'levels_for_interp':[1000, 925, 850, 700, 500,300,200]}
):
try:
dir_rqd=[utl.Cassandra_dir(data_type='high',data_source='ECMWF_ENSEMBLE',var_name='HGT_RAW',lvl=''),
utl.Cassandra_dir(data_type='high',data_source='ECMWF_ENSEMBLE',var_name='TMP_RAW',lvl='')]
#utl.Cassandra_dir(data_type='surface',data_source=model,var_name='RAIN'+str(t_gap).zfill(2)+'_RAW')]
except KeyError:
raise ValueError('Can not find all required directories needed')
#-get all the directories needed
if(initTime == None):
initTime = get_latest_initTime(dir_rqd[0][0:-1]+'/850')
#initTime=utl.filename_day_back_model(day_back=day_back,fhour=0)[0:8]
directory=dir_rqd[0][0:-1]
if(t_range[1] > 72):
fhours = np.append(np.arange(t_range[0], 72, t_gap),np.arange(72,t_range[1],6))
else:
fhours = np.arange(t_range[0], t_range[1], t_gap)
filenames = [initTime+'.'+str(fhour).zfill(3) for fhour in fhours]
HGT_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
directory=dir_rqd[0][0:-1]
directory=dir_rqd[1][0:-1]
TMP_4D=get_model_3D_grids(directory=directory,filenames=filenames,levels=extra_info['levels_for_interp'], allExists=False)
#rn=utl.get_model_points_gy(dir_rqd[4], filenames, points,allExists=False)
directory=dir_rqd[1][0:-1]
coords_info_2D=utl.get_model_points_gy(directory+str(extra_info['levels_for_interp'][0])+'/',
points=points,filenames=filenames,allExists=False)
delt_xy=HGT_4D['lon'].values[1]-HGT_4D['lon'].values[0]
mask = (HGT_4D['lon']<(points['lon']+2*delt_xy))&(HGT_4D['lon']>(points['lon']-2*delt_xy))&(HGT_4D['lat']<(points['lat']+2*delt_xy))&(HGT_4D['lat']>(points['lat']-2*delt_xy))
HGT_4D_sm=HGT_4D['data'].where(mask,drop=True)
TMP_4D_sm=TMP_4D['data'].where(mask,drop=True)
lon_md=np.squeeze(HGT_4D_sm['lon'].values)
lat_md=np.squeeze(HGT_4D_sm['lat'].values)
alt_md=np.squeeze(HGT_4D_sm.values*10).flatten()
time_md=np.squeeze(HGT_4D_sm['forecast_period'].values)
number_md=np.squeeze(HGT_4D_sm['number'].values)
'''
coords = np.zeros((len(time_md),len(number_md),len(extra_info['levels_for_interp']),len(lat_md),len(lon_md),5))
coords[...,0]=time_md.reshape((len(time_md),1,1,1,1))
coords[...,1]=number_md.reshape((1,len(number_md),1,1,1))
coords[...,3] = lat_md.reshape((1,1,1,len(lat_md),1))
coords[...,4] = lon_md.reshape((1,1,1,1,len(lon_md)))
coords = coords.reshape((alt_md.size,5))
coords[:,2]=alt_md
interpolator_TMP = LinearNDInterpolator(coords,TMP_4D_sm.values.reshape((TMP_4D_sm.values.size)),rescale=True)
coords2 = np.zeros((len(time_md),len(number_md),1,1,1,5))
coords2[...,0]=time_md.reshape((len(time_md),1,1,1,1))
coords2[...,1]=number_md.reshape(1,(len(number_md),1,1,1))
coords2[...,2]=points['altitude'][0]
coords2[...,3] = points['lat'][0]
coords2[...,4] = points['lon'][0]
coords2 = coords2.reshape((time_md.size,5))
TMP_interped=np.squeeze(interpolator_TMP(coords2))
'''
TMP_interped=np.zeros((len(time_md),len(number_md)))
for it in range(0,len(time_md)):
for inum in range(0,len(number_md)):
alt_md=np.squeeze(HGT_4D_sm.values[it,inum,:,:,:]*10).flatten()
coords = np.zeros((len(extra_info['levels_for_interp']),len(lat_md),len(lon_md),3))
coords[...,1] = lat_md.reshape((1,len(lat_md),1))
coords[...,2] = lon_md.reshape((1,1,len(lon_md)))
coords = coords.reshape((alt_md.size,3))
coords[:,0]=alt_md
interpolator_TMP = LinearNDInterpolator(coords,TMP_4D_sm.values[it,inum,:,:,:].reshape((TMP_4D_sm.values[it,inum,:,:,:].size)),rescale=True)
coords2 = np.zeros((1,1,1,3))
coords2[...,0]=points['altitude'][0]
coords2[...,1] = points['lat'][0]
coords2[...,2] = points['lon'][0]
coords2 = coords2.reshape((1,3))
TMP_interped[it,inum]=np.squeeze(interpolator_TMP(coords2))
TMP_interped_xr=coords_info_2D.copy()
TMP_interped_xr['data'].values[:,:,0,0]=TMP_interped
TMP_interped_xr.attrs['model']='ECMWF_ENSEMBLE'
Ensemble_graphics.box_line_temp(TMP=TMP_interped_xr,
points=points,
extra_info=extra_info,output_dir=output_dir)
def Time_Crossection_rh_uv_Temp(initTime=None,model='ECMWF',points={'lon':[116.3833], 'lat':[39.9]},
levels=[1000, 950, 925, 900, 850, 800, 700,600,500,400,300,200],data_source='MICAPS',
t_gap=3,t_range=[0,48],lw_ratio=[16,9],output_dir=None,**kwargs):
fhours = np.arange(t_range[0], t_range[1], t_gap)
# 读数据
if(data_source == 'MICAPS'):
try:
data_dir = [utl.Cassandra_dir(data_type='high',data_source=model,var_name='TMP',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='UGRD',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='VGRD',lvl=''),
utl.Cassandra_dir(data_type='high',data_source=model,var_name='RH',lvl=''),
utl.Cassandra_dir(data_type='surface',data_source=model,var_name='PSFC')]
except KeyError:
raise ValueError('Can not find all directories needed')
if(initTime==None):
initTime = get_latest_initTime(data_dir[0][0:-1]+"850")
filenames = [initTime+'.'+str(fhour).zfill(3) for fhour in fhours]
TMP_4D=get_model_3D_grids(directory=data_dir[0][0:-1],filenames=filenames,levels=levels, allExists=False)
u_4D=get_model_3D_grids(directory=data_dir[1][0:-1],filenames=filenames,levels=levels, allExists=False)
v_4D=get_model_3D_grids(directory=data_dir[2][0:-1],filenames=filenames,levels=levels, allExists=False)
rh_4D=get_model_3D_grids(directory=data_dir[3][0:-1],filenames=filenames,levels=levels, allExists=False)
Psfc_3D=get_model_grids(directory=data_dir[4][0:-1],filenames=filenames,allExists=False)
if(data_source == 'CIMISS'):
if(initTime != None):
filename = utl.model_filename(initTime, 0,UTC=True)
else:
filename=utl.filename_day_back_model(day_back=0,fhour=0,UTC=True)
try:
rh_4D=CMISS_IO.cimiss_model_3D_grids(init_time_str='20'+filename[0:8],valid_times=fhours,
data_code=utl.CMISS_data_code(data_source=model,var_name='RHU'),
fcst_levels=levels, fcst_ele="RHU", units='%',pbar=True)
u_4D=CMISS_IO.cimiss_model_3D_grids(init_time_str='20'+filename[0:8],valid_times=fhours,
data_code=utl.CMISS_data_code(data_source=model,var_name='WIU'),
fcst_levels=levels, fcst_ele="WIU", units='m/s',pbar=True)
v_4D=CMISS_IO.cimiss_model_3D_grids(init_time_str='20'+filename[0:8],valid_times=fhours,
data_code=utl.CMISS_data_code(data_source=model,var_name='WIV'),
fcst_levels=levels, fcst_ele="WIV", units='m/s',pbar=True)
TMP_4D=CMISS_IO.cimiss_model_3D_grids(init_time_str='20'+filename[0:8],valid_times=fhours,
data_code=utl.CMISS_data_code(data_source=model,var_name='TEM'),
fcst_levels=levels, fcst_ele="TEM", units='K',pbar=True)
TMP_4D['data'].values=TMP_4D['data'].values-273.15
Psfc_3D=CMISS_IO.cimiss_model_grids(init_time_str='20'+filename[0:8], valid_times=fhours,
data_code=utl.CMISS_data_code(data_source=model,var_name='PRS'),
fcst_level=0, fcst_ele="PRS", units='Pa',pbar=True)
except KeyError:
raise ValueError('Can not find all data needed')
TMP_2D=TMP_4D.interp(lon=('points', points['lon']), lat=('points', points['lat']))
u_2D=u_4D.interp(lon=('points', points['lon']), lat=('points', points['lat']))
v_2D=v_4D.interp(lon=('points', points['lon']), lat=('points', points['lat']))
rh_2D=rh_4D.interp(lon=('points', points['lon']), lat=('points', points['lat']))
rh_2D.attrs['model']=model
rh_2D.attrs['points']=points
Psfc_1D=Psfc_3D.interp(lon=('points', points['lon']), lat=('points', points['lat']))
v_2D2,pressure_2D = xr.broadcast(v_2D['data'],v_2D['level'])
v_2D2,Psfc_2D = xr.broadcast(v_2D['data'],Psfc_1D['data'])
terrain_2D=pressure_2D-Psfc_2D
crossection_graphics.draw_Time_Crossection_rh_uv_Temp(
rh_2D=rh_2D, u_2D=u_2D, v_2D=v_2D,TMP_2D=TMP_2D,terrain_2D=terrain_2D,
t_range=t_range,model=model,lw_ratio=lw_ratio,output_dir=output_dir)