Skip to content
Snippets Groups Projects
Commit 54e801c0 authored by he7273's avatar he7273
Browse files

aew vortices

parent bc8a5506
No related branches found
No related tags found
No related merge requests found
......@@ -19,7 +19,7 @@ fig_dir = '/home/ws/he7273/phd_all/data/vortices/figs'
# plot_dir = '/home/ws/he7273/phd_all/data/coll_oper/' # '/project/meteo/w2w/C3/fischer/belanger/plots/' # join('/home/ws/he7273/phd_all/data/aew/plots/') # '/project/meteo/w2w/C3/fischer/belanger/plots/'
field = 'rv'
bpf = True
bpf = False
fig_dir = fig_dir + '_' + ('bpf' if bpf else 'no_bpf') + '_' + field + '/'
try:
......
......@@ -258,8 +258,8 @@ class AEWVortexIdentification(IdentificationStrategy):
def postprocess(self, dataset: xr.Dataset, data_desc, **kwargs):
plot_kw_style(dataset, self.json_desc, self.config)
plot_kw_style(dataset, self.json_desc, self.config, lev=700)
plot_kw_style(dataset, self.json_desc, self.config, lev=850)
print(dataset)
dataset.to_netcdf("/home/ws/he7273/phd_all/data/coll_oper/jjaso2021_test.nc")
exit()
......
......@@ -54,14 +54,15 @@ def crop_top_bottom_whitespace(path):
return
def plot_kw_style(dataset, dataset_desc, config):
def plot_kw_style(dataset, dataset_desc, config, lev=700):
set_ = dataset_desc.sets[0]
lv = lev
all_nodes = [e.parent for track in set_.tracks for e in track.edges]
for tidx, t in enumerate(dataset.time.data):
time_dt = datetime.utcfromtimestamp(t.astype(datetime) / 1e9)
fig_name = time_dt.strftime("%Y%m%dT%H.png")
fig_name = time_dt.strftime(str(lv) + "_%Y%m%dT%H.png")
# get WTs which are part of tracks for current ts
wts = get_track_wts_of_time(time_dt, all_nodes)
......@@ -95,7 +96,7 @@ def plot_kw_style(dataset, dataset_desc, config):
# field contour
levels = np.linspace(0, 1e-4, 50)
ds_t[config.field].sel(level=850).plot.contourf(levels=levels, cmap='viridis', subplot_kws={'transform_first': True})
ds_t[config.field].sel(level=lv).plot.contourf(levels=levels, cmap='Blues', subplot_kws={'transform_first': True})
"""
ds_t.sel(level=700).plot.streamplot(x='longitude', y='latitude',
......@@ -106,7 +107,7 @@ def plot_kw_style(dataset, dataset_desc, config):
color='blue') # , transform_first=True not working, or is already implemented. still slow.
"""
ds_t.sel(level=850).plot.streamplot(x='longitude', y='latitude',
ds_t.sel(level=lv).plot.streamplot(x='longitude', y='latitude',
u=config.u_dim, v=config.v_dim,
linewidth=0.3,
arrowsize=0.3,
......@@ -130,7 +131,7 @@ def plot_kw_style(dataset, dataset_desc, config):
# plot vortices
# ds_t.sel(level=700).vortices.plot.contourf(levels=[-0.5,0.5,99], colors=('#00000000', 'blue'), subplot_kws={'transform_first': True}, add_colorbar=False)
ds_t.sel(level=850).vortices.plot.contourf(levels=[-0.5,0.5,99], colors=('#00000000', 'orange'), subplot_kws={'transform_first': True}, add_colorbar=False)
ds_t.sel(level=lv).vortices.plot.contourf(levels=[-0.5,0.5,99], colors=('#00000000', 'orange'), subplot_kws={'transform_first': True}, add_colorbar=False)
# ds_t.prec_rate_rea.plot.contourf(levels=levels_rain, extend='max', subplot_kws={'transform_first': True},
# cmap=rain_cm, norm=norm, add_colorbar=False)
......
......@@ -216,19 +216,12 @@ def interpolate_wavetroughs(json_desc, pb_ref, pb_desc):
child_node_time = pb_str_to_datetime64(child_node.time)
if child_node_time - parent_node_time > np.timedelta64(6, 'h'):
if "2021-09-01T12:00:00" == parent_node.time:
print("a")
if "2021-09-01T18:00:00" == parent_node.time:
print("a")
if "2021-09-02T00:00:00" == parent_node.time:
print("a")
print("Add fake WT at " + parent_node.time)
wt_edges = create_fake_wt_edges(edge, child_idx, pb_ref, pb_desc)
# old_edges.append((edge, child_idx))
new_edges.extend(wt_edges)
# TODO dont remove edges for now, quite complicated with splits. lets have transitives instead
# remove old_edges from this track and from graph
new_edges_sn = [proto_to_simplenamespace(e) for e in new_edges]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment