tomodrgn.so3_grid.get_neighbor#

get_neighbor(quat: ndarray, s2i: int, s1i: int, curr_res: int) → tuple[ndarray, ndarray][source]#

Return the 8 nearest neighbors on SO3 at the next resolution level

Parameters:

• quat – rotation quaternions

• s2i – index of the current resolution level of S2

• s1i – index of the current resolution level of S1

• curr_res – current grid resolution

Returns:

nearest neighbor quaternions, nearest neighbor indices on SO3