Source code for cosmogrb.utils.plotting.skyplot

from astropy.coordinates import SkyCoord
import matplotlib.pyplot as plt
from cosmogrb.utils.plotting.projections import *


[docs]def skyplot(ax=None, projection="astro degrees mollweide", center=None, radius=10.0): if ax is None: assert projection in [ "astro degrees aitoff", "astro degrees mollweide", "astro hours aitoff", "astro hours mollweide", "astro globe", "astro zoom", ] skw_dict = dict(projection=projection) if projection in ["astro globe", "astro zoom"]: if center is None: center = SkyCoord(0, 0, unit="deg") skw_dict = dict(projection=projection, center=center) if projection == "astro zoom": assert radius is not None, "you must specify a radius" skw_dict = dict(projection=projection, center=center, radius=radius) fig, ax = plt.subplots(subplot_kw=skw_dict) else: fig = ax.get_figure() ax.grid() return ax