B q\P @s^ddlZddlmZddlmZddlmZddl m Z m Z dgZ ddZ Gd ddeZdS) N)Polygon)units)UnitSphericalRepresentation)rotation_matrixmatrix_productSphericalCirclecCsjt||d}tdtjtj| dd}t| dd}t||}|}||}t |}|j |j fS)a& Given a polygon with vertices defined by (lon, lat), rotate the polygon such that the North pole of the spherical coordinates is now at (lon0, lat0). Therefore, to end up with a polygon centered on (lon0, lat0), the polygon should initially be drawn around the North pole. )lonlatg?y)Zaxisz) rrnppiuradianrZ to_cartesianZ transformZfrom_cartesianrr )rr Zlon0Zlat0ZpolygonZm1Zm2Ztransform_matrixrDlib/python3.7/site-packages/astropy/visualization/wcsaxes/patches.py_rotate_polygons    rcs(eZdZdZdejffdd ZZS)ra Create a patch representing a spherical circle - that is, a circle that is formed of all the points that are within a certain angle of the central coordinates on a sphere. Here we assume that latitude goes from -90 to +90 This class is needed in cases where the user wants to add a circular patch to a celestial image, since otherwise the circle will be distorted, because a fixed interval in longitude corresponds to a different angle on the sky depending on the latitude. Parameters ---------- center : tuple or `~astropy.units.Quantity` This can be either a tuple of two `~astropy.units.Quantity` objects, or a single `~astropy.units.Quantity` array with two elements. radius : `~astropy.units.Quantity` The radius of the circle resolution : int, optional The number of points that make up the circle - increase this to get a smoother circle. vertex_unit : `~astropy.units.Unit` The units in which the resulting polygon should be defined - this should match the unit that the transformation (e.g. the WCS transformation) expects as input. Notes ----- Additional keyword arguments are passed to `~matplotlib.patches.Polygon` dc  s|\}}tddtj|dddtj}tdtj|tj|tj} t|| ||\}} ||}| |} t|| g } t j | f|dS)Ngg?) r Zlinspacer rrrepeatZto_valuerZarrayZ transposesuper__init__) selfcenterZradiusZ resolutionZ vertex_unitkwargsZ longitudeZlatituderr Zvertices) __class__rrrGs&$  zSphericalCircle.__init__)__name__ __module__ __qualname____doc__rZdegreer __classcell__rr)rrr(s)Znumpyr Zmatplotlib.patchesrZastropyrrZ"astropy.coordinates.representationrZ$astropy.coordinates.matrix_utilitiesrr__all__rrrrrrs