Source code for svcco.branch_addition.add_depths

import numba as nb

[docs]@nb.jit(nopython=True,cache=True,nogil=True) def add_depths(data, edge): search = [int(edge)] while len(search) > 0: for idx in search: if data[idx, 15].item() >= 0 or data[idx, 16].item() >= 0: left = int(data[idx, 15].item()) right = int(data[idx, 16].item()) data[left, 26] += 1 data[right, 26] += 1 search.remove(idx) search.append(left) search.append(right) else: search.remove(idx)