from itertools import product
import numpy as np
import fastplotlib as fpl


def make_circle(center, radius: float, n_points: int = 75) -> np.ndarray:
    theta = np.linspace(0, 2 * np.pi, n_points)
    xs = radius * np.sin(theta)
    ys = radius * np.cos(theta)

    return np.column_stack([xs, ys]) + center


spatial_dims = (100, 100)

circles = list()
for center in product(range(0, spatial_dims[0], 15), range(0, spatial_dims[1], 15)):
    circles.append(make_circle(center, 5, n_points=75))

fig = fpl.Figure()

lines = fig[0, 0].add_line_collection(circles, colors="r", thickness=5)


def highlight_nearest(ev):
    lines.colors = "r"
    
    xy = fig[0, 0].map_screen_to_world(ev)[:-1]
    nearest = fig[0, 0].get_nearest_graphics(xy, method="center", subset=lines)
    
    nearest[0].colors = "w"

fig.renderer.add_event_handler(highlight_nearest, "pointer_move")

fig.show()