from astropy import units as u
from matplotlib import pyplot as plt
import ligo.skymap.plot
import numpy as np

from m4opt import skygrid

n_vertices_target = 1024
vertices = skygrid.geodesic(4 * np.pi * u.sr / n_vertices_target)
n_vertices = len(vertices)

ax = plt.axes(projection='astro globe', center='0d 25d')
plt.suptitle('Class I')
ax.set_title(f'{n_vertices} vertices (goal was {n_vertices_target})')
ax.plot_coord(vertices, '.')
ax.grid()