aew vortices

enstools/feature/identification/aew_vortices/configuration.py

@@ -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:

enstools/feature/identification/aew_vortices/identification.py

@@ -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()

enstools/feature/identification/aew_vortices/plotting.py

@@ -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)

enstools/feature/identification/aew_vortices/util.py

@@ -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]