diff --git a/Overlap/plot_traj.py b/Overlap/plot_traj.py index 2eb06822ffb37796f3371495c91363a1e1d34e45..a2ead9ba5444fd1ef4fac249c31865b3b9bb7a76 100755 --- a/Overlap/plot_traj.py +++ b/Overlap/plot_traj.py @@ -34,6 +34,8 @@ def is_node_type(G, n, label): return G.out_degree[n] >= 2 elif label == "all": return True + elif label == "important": + return G.in_degree[n] == 0 or G.out_degree[n] == 0 or G.degree[n] >= 3 elif isinstance(label, int): return node_to_date_eddy(n, e_overestim, d_init)["days_1950"] == label @@ -102,7 +104,8 @@ if __name__ == "__main__": parser.add_argument("shpc_dir", help = "directory containing SHPC, with " "visible eddies at all dates") parser.add_argument("-l", "--label", - choices = ["root", "leaf", "split", "merge", "all"]) + choices = ["root", "leaf", "split", "merge", "all", + "important"]) parser.add_argument("-s", "--save", metavar = "format", help = "Save file to specified format") parser.add_argument("-t", "--time", action = "store_true",