Skip to content

Commit f50a9ae

Browse files
committed
Add method to a re numbering of network like segment
1 parent 9053746 commit f50a9ae

File tree

2 files changed

+13
-7
lines changed

2 files changed

+13
-7
lines changed

src/py_eddy_tracker/observations/network.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -323,13 +323,19 @@ def close_network(self, other, nb_obs_min=10, **kwargs):
323323
m[other.network_slice(i)] = True
324324
return other.extract_with_mask(m)
325325

326-
def numbering_segment(self):
326+
def numbering_segment(self, start=0):
327327
"""
328328
New numbering of segment
329329
"""
330330
for i, _, _ in self.iter_on("track"):
331331
new_numbering(self.segment[i])
332332

333+
def numbering_network(self, start=1):
334+
"""
335+
New numbering of network
336+
"""
337+
new_numbering(self.track, start)
338+
333339
def only_one_network(self):
334340
"""
335341
Raise a warning or error?
@@ -1169,10 +1175,10 @@ def build_unique_array(id1, id2):
11691175

11701176

11711177
@njit(cache=True)
1172-
def new_numbering(segs):
1178+
def new_numbering(segs, start=0):
11731179
nb = len(segs)
11741180
s0 = segs[0]
1175-
j = 0
1181+
j = start
11761182
for i in range(nb):
11771183
if segs[i] != s0:
11781184
s0 = segs[i]

src/py_eddy_tracker/observations/observation.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -777,10 +777,10 @@ def copy_data_to_zarr(
777777
:param zarr_dataset handler_zarr:
778778
:param array handler_eddies:
779779
:param slice zarr_dataset sl_obs:
780-
:param int zarr_dataset buffer_size:
781-
:param float zarr_dataset factor:
782-
:param bool zarr_dataset raw_data:
783-
:param None,float zarr_dataset scale_factor:
780+
:param int buffer_size:
781+
:param float factor:
782+
:param bool raw_data:
783+
:param None,float scale_factor:
784784
:param None,float add_offset:
785785
"""
786786
i_start, i_stop = sl_obs.start, sl_obs.stop

0 commit comments

Comments
 (0)