Source code for svcco.branch_addition.check
import numba as nb
import numpy as np
[docs]@nb.jit(nopython=True,cache=True,nogil=True)
def check(data,point_set,threshold):
centers = (data[:,0:3] + data[:,3:6])/2
for i in range(point_set.shape[0]):
distances = np.sqrt(np.sum(np.square(centers-point_set[i,:]),axis=1))
if np.all(distances > threshold):
return True, point_set[i,:]
return False,None
[docs]@nb.jit(nopython=True,cache=True,nogil=True)
def check_point(data,point,threshold):
centers = (data[:,0:3] + data[:,3:6])/2
distances = np.sqrt(np.sum(np.square(centers-point),axis=1))
if np.all(distances > threshold):
return True
return False