tomodrgn.so3_grid.get_s2_neighbor#

get_s2_neighbor(mini: int, curr_res: int) → tuple[ndarray, ndarray][source]#

Return the 4 nearest neighbors on S2 at the next resolution level

Parameters:

• mini

• curr_res – current grid resolution

Returns:

theta and phi of nearest neighbors on S2, indices of nearest neighbors on S2