IR-eddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd gZejjd d ej_ejjd d ej_ejjd d ej_ejjd d ej_dZGdd eZGddeZdS)N)Polygon)units)SkyCoord)rotation_matrix)SphericalRepresentationUnitSphericalRepresentation)AstropyUserWarning QuadrangleSphericalCirclez `.CapStyle`z``matplotlib._enums.CapStyle``z `.JoinStyle`z``matplotlib._enums.JoinStyle``c>t||}t| dtdtjztjz|z dz}|}||}tj|}|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. )lonlatz)axis?y) rrnppiuradian to_cartesian transformfrom_cartesianr r)r rlon0lat0polygontransform_matrixs Elib/python3.11/site-packages/astropy/visualization/wcsaxes/patches.py_rotate_polygonr$s*cs;;;G'u3777/ + 4 '(s;;; ""$$G 011G)8AAG ; ##c2eZdZdZdejffd ZxZS)r a8 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` ['angle'] This can be either a tuple of two `~astropy.units.Quantity` objects, or a single `~astropy.units.Quantity` array with two elements or a `~astropy.coordinates.SkyCoord` object. radius : `~astropy.units.Quantity` ['angle'] 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 t|trZ|j}t|tt fst jd|dt|j j |j j }}n|\}}tj ddtjz|dzddtjz} tjdtjz|tjz |tjz} t'| | ||\} } | |} | |} tj| | g} t-j| fi|dS)Nz)Received `center` of representation type z. will be converted to SphericalRepresentation gr) isinstancerrepresentation_type issubclassrrwarningswarnr sphericalr rrlinspacerrrrepeatto_valuerarray transposesuper__init__) selfcenterradius resolution vertex_unitkwargsrep_type longitudelatituder rvertices __class__s rr3zSphericalCircle.__init__\st fh ' ' )1H24OP  DDDD& #)"2"68H8LxII"( Ixk#q25y*q.99#2#>Iibe fooah&?&??LLqxW"3YAASll;''ll;''8S#J''1133,,V,,,,,r __name__ __module__ __qualname____doc__rdegreer3 __classcell__r>s@rr r <sP>3618----------r c2eZdZdZdejffd ZxZS)r a Create a patch representing a latitude-longitude quadrangle. The edges of the quadrangle lie on two lines of constant longitude and two lines of constant latitude (or the equivalent component names in the coordinate frame of interest, such as right ascension and declination). Note that lines of constant latitude are not great circles. Unlike `matplotlib.patches.Rectangle`, the edges of this patch will render as curved lines if appropriate for the WCS transformation. Parameters ---------- anchor : tuple or `~astropy.units.Quantity` ['angle'] This can be either a tuple of two `~astropy.units.Quantity` objects, or a single `~astropy.units.Quantity` array with two elements. width : `~astropy.units.Quantity` ['angle'] The width of the quadrangle in longitude (or, e.g., right ascension) height : `~astropy.units.Quantity` ['angle'] The height of the quadrangle in latitude (or, e.g., declination) resolution : int, optional The number of points that make up each side of the quadrangle - increase this to get a smoother quadrangle. vertex_unit : `~astropy.units.Unit` ['angle'] 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` r"c Ttj||\}}||}||}|tjd||dzz} |tjd||dzz} tj| ddtj| d|tj| ddtj| d|g} tjtj| d|| ddtj| d|tj| ddg} tj| | g } tj | fi|dS)Nrr%r&) rQuantityr/rr- concatenater.flipr0r1r2r3)r4anchorwidthheightr7r8r9r;r<lon_seqlat_seqr rr=r>s rr3zQuadrangle.__init__s  j0099+FF 8{++--bk!UJNCCCR[FJNCCCn  '"+z22 $$ '!*j11    n '!*j11  '"+z22 $$    8S#J''1133,,V,,,,,r r?rFs@rr r ~sQD14$-$-$-$-$-$-$-$-$-$-r )r*numpyrmatplotlib.patchesrastropyrrastropy.coordinatesr$astropy.coordinates.matrix_utilitiesr"astropy.coordinates.representationrrastropy.utils.exceptionsr __all__r3rCreplace set_capstyle set_joinstylerr r r rr]s&&&&&&((((((@@@@@@877777 * +#+3;;3#+3;;5 '3;CC3  !( 5 = E E5!! $$$0?-?-?-?-?-g?-?-?-DF-F-F-F-F-F-F-F-F-F-r