Skip to content
Snippets Groups Projects
Commit e74ae8d9 authored by Lionel GUEZ's avatar Lionel GUEZ
Browse files

Rename vertex property segment to `inst_eddies`

parent 3f9ca4f7
No related branches found
No related tags found
No related merge requests found
......@@ -134,7 +134,7 @@ num_of_days_to_avg = 7 # number of days to average
print("Iterating on vertices...")
for n in g.vertices():
segment = g.vp.segment[n]
segment = g.vp.inst_eddies[n]
num_of_days = len(segment)
# Calculate the date index, the eddy index and the SHPC index of
......
......@@ -35,25 +35,25 @@ t0 = t1
# Processing:
g.vp['segment'] = g.new_vp('object') # this object will be a list
g.vp['inst_eddies'] = g.new_vp('object') # this object will be a list
print('Collapsing into segments...')
g.set_fast_edge_removal()
verts_to_del = []
for v in g.vertices():
if not g.vp.segment[v]:
g.vp.segment[v] = [int(g.vp.name[v])]
if not g.vp.inst_eddies[v]:
g.vp.inst_eddies[v] = [int(g.vp.name[v])]
if v.in_degree() == 1:
v2 = next(v.in_edges()).source()
if v2.out_degree() == 1:
g.remove_edge(g.edge(v2, v))
g.vp.segment[v2].append(int(g.vp.name[v]))
g.vp.inst_eddies[v2].append(int(g.vp.name[v]))
for v3 in v.out_neighbors():
g.remove_edge(g.edge(v, v3))
g.vp.segment[v].append(int(g.vp.name[v3]))
g.vp.inst_eddies[v].append(int(g.vp.name[v3]))
new_edge = g.add_edge(v2, v3)
if(v.in_degree() == 0 and v.out_degree() == 0):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment