Add sub-option important

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",