@@ -23,14 +23,14 @@ cpdef np.ndarray[np.double_t, ndim=2] mst_linkage_core(
2323 cdef np.ndarray[np.double_t, ndim= 1 ] left
2424 cdef np.ndarray[np.double_t, ndim= 1 ] right
2525 cdef np.ndarray[np.double_t, ndim= 2 ] result
26-
26+
2727 cdef np.ndarray label_filter
28-
28+
2929 cdef np.intp_t current_node
3030 cdef np.intp_t new_node_index
3131 cdef np.intp_t new_node
3232 cdef np.intp_t i
33-
33+
3434 result = np.zeros((distance_matrix.shape[0 ] - 1 , 3 ))
3535 node_labels = np.arange(distance_matrix.shape[0 ], dtype = np.intp)
3636 current_node = 0
@@ -42,14 +42,14 @@ cpdef np.ndarray[np.double_t, ndim=2] mst_linkage_core(
4242 left = current_distances[label_filter]
4343 right = distance_matrix[current_node][current_labels]
4444 current_distances = np.where(left < right, left, right)
45-
45+
4646 new_node_index = np.argmin(current_distances)
4747 new_node = current_labels[new_node_index]
4848 result[i - 1 , 0 ] = < double > current_node
4949 result[i - 1 , 1 ] = < double > new_node
5050 result[i - 1 , 2 ] = current_distances[new_node_index]
5151 current_node = new_node
52-
52+
5353 return result
5454
5555cdef void select_distances(
@@ -93,9 +93,9 @@ cpdef np.ndarray[np.double_t, ndim=2] mst_linkage_core_pdist(
9393 cdef np.ndarray[np.double_t, ndim= 1 ] right
9494 cdef np.ndarray[np.intp_t, ndim= 1 ] col_select
9595 cdef np.ndarray[np.double_t, ndim= 2 ] result
96-
96+
9797 cdef np.ndarray label_filter
98-
98+
9999 cdef np.intp_t current_node
100100 cdef np.intp_t new_node_index
101101 cdef np.intp_t new_node
@@ -104,7 +104,7 @@ cpdef np.ndarray[np.double_t, ndim=2] mst_linkage_core_pdist(
104104
105105 dim = int ((1 + np.sqrt(1 + 8 * pdist_matrix.shape[0 ])) / 2.0 )
106106 col_select = np.cumsum(np.arange(dim - 1 , 0 , - 1 ))
107-
107+
108108 result = np.zeros((dim - 1 , 3 ))
109109 node_labels = np.arange(dim, dtype = np.intp)
110110 current_node = 0
@@ -118,14 +118,14 @@ cpdef np.ndarray[np.double_t, ndim=2] mst_linkage_core_pdist(
118118 # right = fill_row(pdist_matrix, current_node, dim, col_select)[current_labels]
119119 select_distances(pdist_matrix, col_select, current_labels, right, current_node, dim)
120120 current_distances = np.where(left < right[:len (current_labels)], left, right[:len (current_labels)])
121-
121+
122122 new_node_index = np.argmin(current_distances)
123123 new_node = current_labels[new_node_index]
124124 result[i - 1 , 0 ] = < double > current_node
125125 result[i - 1 , 1 ] = < double > new_node
126126 result[i - 1 , 2 ] = current_distances[new_node_index]
127127 current_node = new_node
128-
128+
129129 return result
130130
131131
@@ -244,29 +244,29 @@ cdef class UnionFind (object):
244244 cdef np.intp_t next_label
245245 cdef np.intp_t * parent
246246 cdef np.intp_t * size
247-
247+
248248 def __init__ (self , N ):
249249 self .parent_arr = - 1 * np.ones(2 * N - 1 , dtype = np.intp)
250250 self .next_label = N
251251 self .size_arr = np.hstack((np.ones(N, dtype = np.intp),
252252 np.zeros(N- 1 , dtype = np.intp)))
253253 self .parent = (< np.intp_t * > self .parent_arr.data)
254254 self .size = (< np.intp_t * > self .size_arr.data)
255-
255+
256256 cdef void union (self , np.intp_t m, np.intp_t n):
257257 self .size[self .next_label] = self .size[m] + self .size[n]
258258 self .parent[m] = self .next_label
259259 self .parent[n] = self .next_label
260260 self .size[self .next_label] = self .size[m] + self .size[n]
261261 self .next_label += 1
262-
262+
263263 return
264-
264+
265265 cdef np.intp_t find(self , np.intp_t n):
266266 while self .parent[n] != - 1 :
267267 n = self .parent[n]
268268 return n
269-
269+
270270 cdef np.intp_t fast_find(self , np.intp_t n):
271271 cdef np.intp_t p
272272 p = n
@@ -276,15 +276,15 @@ cdef class UnionFind (object):
276276 while self .parent_arr[p] != n:
277277 p, self .parent_arr[p] = self .parent_arr[p], n
278278 return n
279-
279+
280280cpdef np.ndarray[np.double_t, ndim= 2 ] label(np.ndarray[np.double_t, ndim= 2 ] L):
281281
282282 cdef np.ndarray[np.double_t, ndim= 2 ] result_arr
283283 cdef np.double_t[:, ::1 ] result
284284
285285 cdef np.intp_t N, a, aa, b, bb, idx
286286 cdef np.double_t delta
287-
287+
288288 result_arr = np.zeros((L.shape[0 ], L.shape[1 ] + 1 ))
289289 result = (< np.double_t[:L.shape[0 ], :4 :1 ]> (< np.double_t * > result_arr.data))
290290 N = L.shape[0 ] + 1
0 commit comments