Add timings

cost_function.py

@@ -78,6 +78,9 @@ def get_SHPC(array_d_ini, date_index):
     assert i_SHPC >= 1
     return i_SHPC - 1
 
+t0 = time.perf_counter()
+timings = open("timings.txt", "w")
+
 if len(sys.argv) == 1:
     sys.exit("Required arguments: SHPC-directory [SHPC-directory] ...")
 
@@ -100,8 +103,6 @@ delta_r_mean = -0.0094709 * 1000 #[m]
 delta_r_std = 8.6953 * 1000
 
 # Load the graph_tool file:
-t0 = time.perf_counter()
-timings = open("timings.txt", "w")
 print('Loading graph...')
 g = graph_tool.Graph()
 
@@ -110,15 +111,15 @@ try:
 except FileNotFoundError:
     g.load('segments.graphml')
 
-t1 = time.perf_counter()
 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()
+t1 = time.perf_counter()
 timings.write(f"loading: {t1 - t0} s\n")
-timings.close()
+t0 = t1
 
 g.set_fast_edge_removal()
 g.vp['pos_first'] = g.new_vp('object')
@@ -235,6 +236,9 @@ for n in g.vertices():
         g.vp.first_av_rad[n] = radii
         g.vp.last_av_rad[n] = radii
 
+t1 = time.perf_counter()
+timings.write(f"iterating on vertices: {t1 - t0} s\n")
+t0 = t1
 print("Iterating on edges...")
 
 for edge in g.edges():
@@ -292,6 +296,12 @@ for edge in g.edges():
     g.ep.nl_cost_function[edge] = cf
 
 
+t1 = time.perf_counter()
+timings.write(f"iterating on edges: {t1 - t0} s\n")
+t0 = t1
 print("Saving...")
 g.save('segments_cost_functions.gt')
 print('All done')
+t1 = time.perf_counter()
+timings.write(f"saving: {t1 - t0} s\n")
+timings.close()