From 699ecbf0d9b5275081b836b3551940a3eaf12db5 Mon Sep 17 00:00:00 2001 From: Lionel GUEZ <guez@lmd.ens.fr> Date: Wed, 11 Oct 2023 22:01:45 +0200 Subject: [PATCH] Polish --- Trajectories/trajectories.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/Trajectories/trajectories.py b/Trajectories/trajectories.py index ad9d141e..46b4e320 100755 --- a/Trajectories/trajectories.py +++ b/Trajectories/trajectories.py @@ -29,7 +29,11 @@ def new_traj(ind_traj, traj_prop, n, traj_vert_ind): if len(sys.argv) != 2: sys.exit("Required argument: input-graph") g = graph_tool.load_graph(sys.argv[1]) -print("Loading done") +print("Input graph:") +print("Number of vertices:", g.num_vertices()) +print("Number of edges:", g.num_edges()) +print("Internal properties:") +g.list_properties() if "cost_function" in g.edge_properties: closest_succ = g.new_vertex_property('int') @@ -62,8 +66,8 @@ traj_vert_ind = [] # An element of traj_vert_ind is a trajectory, represented by a list # of vertex indices. -print("Starting loop to define traj_prop...") +# Loop to define traj_prop: for n in topology.topological_sort(g): # First, we check whether we have a phantom pattern. Phantom # eddies may appear in altimetric data because of @@ -169,7 +173,7 @@ for t in traj_vert_ind: traj_segm.append(list_segm) expanded_traj.append(list_eddies_traj) -print("Saving...") +# Save: with open("traj_segm.json", "w") as outfile: json.dump(traj_segm, outfile, separators = (',', ':')) -- GitLab